CANN/asc-devkit MakeNDLayout函数

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