CANN/Ascend C Conv3D Tiling构造函数

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参数说明

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