CANN卷积Tiling接口文档

GetTiling

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言，原生支持C和C++标准规范，主要由类库和语言扩展层构成，提供多层级API，满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit

功能说明

获取Tiling参数。

函数原型

int64_t GetTiling(optiling::Conv3DBackpropInputTilingData& tiling)

int64_t GetTiling(AscendC::tiling::Conv3DBackpropInputTilingData& tiling)

参数说明

表1参数说明

返回值说明

如果返回值不为-1，则代表Tiling计算成功，用户可以使用该Tiling结构的值。如果返回值为-1，则代表Tiling计算失败，该Tiling结果无法使用。

约束说明

无

调用示例

// 构建Conv3dBackpropInput算子tiling的调用示例 auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo()); ConvBackpropApi::Conv3DBpInputTiling tiling(ascendcPlatform); tiling.SetWeightShape(256, 128, 3, 3, 3); tiling.SetInputShape(1, 128, 11, 128, 128); tiling.SetGradOutputShape(1, 256, 9, 128, 128); tiling.SetPadding(0, 0, 1, 1, 1, 1); tiling.SetDilation(1, 1, 1); tiling.SetStride(1, 1, 1); optiling::Conv3DBackpropInputTilingData tilingData; int ret = tiling.GetTiling(tilingData); // 获取Tiling参数 AscendC::tiling::Conv3DBackpropInputTilingData tilingDataNotOp; ret = tiling.GetTiling(tilingDataNotOp); // 使用AscendC::tiling::Conv3DBackpropInputTilingData获取Tiling参数

// 构建Conv3dTranspose算子tiling的调用示例 auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo()); ConvBackpropApi::Conv3DBpInputTiling tiling(ascendcPlatform); tiling.SetWeightShape(256, 128, 3, 3, 3); tiling.SetGradOutputShape(256, 9, 128, 128); // 等价于Conv3dTranspose的输入X tiling.SetPadding(0, 0, 1, 1, 1, 1); tiling.SetDilation(1, 1, 1); tiling.SetStride(1, 1, 1); tiling.SetOutputPadding(0, 0, 0); // 对Conv3dTranspose的输出Y进行padding optiling::Conv3DBackpropInputTilingData tilingData; int ret = tiling.GetTiling(tilingData); // 获取Tiling参数

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言，原生支持C和C++标准规范，主要由类库和语言扩展层构成，提供多层级API，满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit