SetScaleAType【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明MxMatmul场景设置scaleA矩阵的位置、数据格式、是否转置等信息这些信息需要和Kernel侧的设置保持一致。如果不调用本接口scaleA矩阵的信息将与SetAType中设置的A矩阵的信息保持一致。函数原型int32_t SetScaleAType(TPosition scalePos, CubeFormat scaleType, bool isScaleTrans false)参数说明表1参数说明参数名输入/输出描述scalePos输入scaleA矩阵的内存逻辑位置。针对Ascend 950PR/Ascend 950DTscaleA矩阵可设置为TPosition::GMTPosition::VECOUTTPosition::TSCM。scaleType输入scaleA矩阵的物理排布格式。针对Ascend 950PR/Ascend 950DTscaleA矩阵可设置为CubeFormat::NDCubeFormat::NZ。isScaleTrans输入scaleA矩阵是否转置。参数支持的取值如下truescaleA矩阵转置false默认值scaleA矩阵不转置。返回值说明-1表示设置失败 0表示设置成功。约束说明无调用示例auto ascendcPlatform platform_ascendc::PlatformAscendC(context-GetPlatformInfo()); matmul_tiling::MatmulApiTiling tiling(ascendcPlatform); tiling.SetAType(matmul_tiling::TPosition::GM, matmul_tiling::CubeFormat::ND, matmul_tiling::DataType::DT_FLOAT16); // 设置scaleA矩阵buffer位置为GM数据格式为ND不转置 tiling.SetScaleAType(matmul_tiling::TPosition::GM, matmul_tiling::CubeFormat::ND, false);【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考