CANN asc-devkit 3D卷积反向传播Tiling接口
GetTiling【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明获取Tiling参数。函数原型int64_t GetTiling(optiling::Conv3DBackpropFilterTilingData tiling)int64_t GetTiling(AscendC::tiling::Conv3DBackpropFilterTilingData tiling)参数说明表1参数说明参数名输入/输出描述tiling输出TConv3DBpFilterTiling的Tiling结构体用于存储最终的Tiling结果。TConv3DBpFilterTiling结构介绍请参考TConv3DBpFilterTiling结构体。返回值说明如果返回值不为-1则代表Tiling计算成功用户可以使用该Tiling结构的值。如果返回值为-1则代表Tiling计算失败该Tiling结果无法使用。约束说明无调用示例#include tiling/conv_backprop/conv3d_bp_filter_tiling.h optiling::Conv3DBackpropFilterTilingData tilingData; auto ascendcPlatform platform_ascendc::PlatformAscendCManager::GetInstance(); ConvBackpropApi::Conv3dBpFilterTiling conv3dBpDwTiling(*ascendcPlatform); conv3dBpDwTiling.SetWeightType( ConvCommonApi::TPosition::GM, ConvCommonApi::ConvFormat::FRACTAL_Z_3D, ConvCommonApi::ConvDtype::FLOAT32); conv3dBpDwTiling.SetInputType( ConvCommonApi::TPosition::GM, ConvCommonApi::ConvFormat::NDC1HWC0, ConvCommonApi::ConvDtype::FLOAT16); conv3dBpDwTiling.SetGradOutputType( ConvCommonApi::TPosition::GM, ConvCommonApi::ConvFormat::NDC1HWC0, ConvCommonApi::ConvDtype::FLOAT16); conv3dBpDwTiling.SetInputShape(n, c, d, h, w); conv3dBpDwTiling.SetGradOutputShape(n, c, d, h, w); conv3dBpDwTiling.SetWeightShape(cout, cin, d, h, w); conv3dBpDwTiling.SetPadding(padFront, padBack, padUp, padDown, padLeft, padRight); conv3dBpDwTiling.SetStride(strideD, strideH, strideW); conv3dBpDwTiling.SetDilation(dilationD, dilationH, dilationW); int ret conv3dBpDwTiling.GetTiling(tilingData); // 获取tiling参数 // 使用AscendC::tiling命名空间中的Tiling结构体获取tiling参数 AscendC::tiling::Conv3DBackpropFilterTilingData tilingDataNotOp; ret conv3dBpDwTiling.GetTiling(tilingDataNotOp);【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考