CANN asc-devkit Zn布局创建函数

MakeZnLayout【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√功能说明创建Zn格式的布局外层矩阵按列优先存储内层矩阵按行优先存储。函数原型template typename T, typename U, typename S __aicore__ inline decltype(auto) MakeZnLayout(U row, S column)参数说明表 1模板参数说明参数名输入/输出描述T输入数据类型支持fp4x2_e2m1_t、fp4x2_e1m2_t、int8_t、uint8_t、int16_t、uint16_t、half、bfloat16_t、int32_t、uint32_t、float、complex32、int64_t、uint64_t。U输入行数类型size_t或Int整型常量。S输入列数类型size_t或Int整型常量。表 2参数说明参数名输入/输出描述row输入矩阵的总行数。column输入矩阵的总列数。返回值说明输入为编译时常量时返回Zn格式的Layout类型。输入为整型变量时返回Zn格式的Layout对象。返回对齐后的Layout对齐方式及对应位置的参数大小说明详见Layout和层次化表述法。约束说明参数row和column需为size_t类型或Int整型常量。调用示例// 创建Zn格式Layout using namespace AscendC::Te; // 编译时常量传参构造Layout auto staticLayout MakeZnLayouthalf(Std::Int32{}, Std::Int32{}); // 运行时变量传参构造Layout auto layout MakeZnLayouthalf(32, 32);【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考