CANN/Ascend C Conv3DBackpropInput Tiling构造函数
Conv3DBackpropInput Tiling构造函数【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明用于创建一个Conv3DBackpropInput单核Tiling对象。函数原型带参构造函数需要传入硬件平台信息推荐使用这类构造函数来获得更好的兼容性。使用PlatformAscendC类传入信息explicit Conv3DBpInputTiling(const platform_ascendc::PlatformAscendC ascendcPlatform)使用PlatformInfo传入信息当platform_ascendc::PlatformAscendC无法在Tiling运行时获取时需要用户自己构造PlatformInfo结构体透传给Conv3DBpInputTiling构造函数。explicit Conv3DBpInputTiling(const PlatformInfo platform)无参构造函数Conv3DBpInputTiling()基类构造函数Conv3DBpInputTiling继承自基类Conv3DBpInputTilingBase其构造函数如下Conv3DBpInputTilingBase()explicit Conv3DBpInputTilingBase(const platform_ascendc::PlatformAscendC ascendcPlatform)explicit Conv3DBpInputTilingBase(const PlatformInfo platform)参数说明表1参数说明参数名输入/输出描述ascendcPlatform输入传入硬件平台的信息PlatformAscendC定义请参见构造及析构函数。platform输入传入硬件版本以及AI Core中各个硬件单元提供的内存大小。PlatformInfo构造时通过构造及析构函数获取。PlatformInfo结构定义如下代码所示socVersion通过GetSocVersion获取并透传各类硬件存储空间大小通过GetCoreMemSize获取并透传。不推荐通过直接填值构造PlatformInfo的方式调用构造函数例如PlatformInfo(, 1024, 1024, ..);struct PlatformInfo { platform_ascendc::SocVersion socVersion; uint64_t l1Size 0; uint64_t l0CSize 0; uint64_t ubSize 0; uint64_t l0ASize 0; uint64_t l0BSize 0; };在实现Host侧的Tiling函数时platform_ascendc::PlatformAscendC用于获取一些硬件平台的信息来支撑Tiling的计算比如获取硬件平台的核数等信息。PlatformAscendC类提供获取这些平台信息的功能。与platform_ascendc::PlatformAscendC不同的是PlatformInfo则用于获取芯片版本以及AI Core中各个硬件单元提供的内存大小等只针对单个AI Core的信息。约束说明无调用示例无参构造函数ConvBackpropApi::Conv3DBpInputTiling tiling; tiling.SetWeightType(ConvCommonApi::TPosition::GM,ConvCommonApi::ConvFormat::FRACTAL_Z_3D,ConvCommonApi::ConvDtype::FLOAT16); ... optiling::Conv3DBackpropInputTilingData tilingData; int ret tiling.GetTiling(tilingData); // if ret -1, gen tiling failed ...带参构造函数auto ascendcPlatform platform_ascendc::PlatformAscendC(context-GetPlatformInfo()); ConvBackpropApi::Conv3DBpInputTiling tiling(ascendcPlatform); tiling.SetWeightType( ConvCommonApi::TPosition::GM, ConvCommonApi::ConvFormat::FRACTAL_Z_3D, ConvCommonApi::ConvDtype::FLOAT16); ... optiling::Conv3DBackpropInputTilingData tilingData; int ret tiling.GetTiling(tilingData); // if ret -1, gen tiling failed【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考