Conv3D Tiling构造函数【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明用于创建一个Conv3D单核Tiling对象。函数原型带参构造函数需要传入硬件平台信息推荐使用这类构造函数来获得更好的兼容性。使用PlatformAscendC类传入信息explicit Conv3dTiling(const platform_ascendc::PlatformAscendC ascendcPlatform)使用PlatformInfo传入信息当platform_ascendc::PlatformAscendC无法在Tiling运行时获取时需要用户自己构造PlatformInfo结构体透传给Conv3dTiling构造函数。explicit Conv3dTiling(const PlatformInfo platform)基类构造函数Conv3dTiling继承自基类Conv3dTilingBase其构造函数如下explicit Conv3dTilingBase(const platform_ascendc::PlatformAscendC ascendcPlatform)explicit Conv3dTilingBase(const PlatformInfo platform)参数说明表1参数说明参数名输入/输出描述ascendcPlatform输入传入硬件平台的信息PlatformAscendC定义请参见构造及析构函数。platform输入传入硬件版本以及AI Core中各个硬件单元提供的内存大小。PlatformInfo构造时通过构造及析构函数获取。PlatformInfo结构定义如下方代码所示socVersion通过GetSocVersion获取并透传各类硬件存储空间大小通过GetCoreMemSize获取并透传。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; uint64_t btSize 0; uint64_t fbSize 0; };约束说明无调用示例// 实例化Conv3d Api auto ascendcPlatform platform_ascendc::PlatformAscendC(context-GetPlatformInfo()); Conv3dTilingApi::Conv3dTiling conv3dApiTiling(ascendcPlatform); conv3dApiTiling.SetGroups(groups); conv3dApiTiling.SetOrgWeightShape(cout, kd, kh, kw); ... conv3dApiTiling.GetTiling(conv3dCustomTilingData.conv3dApiTilingData);【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考