CANN/asc-devkit MakeLayout API

MakeLayout

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

产品支持情况

功能说明

将传入的Shape和Stride数据打包成Layout数据结构。

函数原型

template <typename ShapeType, typename StrideType>
__aicore__ inline constexpr auto MakeLayout(const ShapeType& shape, const StrideType& stride)

参数说明

表 1 模板参数说明

表 2 参数说明

返回值说明

返回Layout对象。

约束说明

构造Layout对象时传入的Shape和Stride结构，需是Std::tuple结构类型，且满足Std::tuple结构类型的使用约束。

调用示例

// 初始化Layout数据结构，获取对应数值
using namespace AscendC::Te;

auto shape = MakeShape(10, 20, 30);
auto stride = MakeStride(1, 100, 200);
auto layoutMake = MakeLayout(shape, stride);
Layout<decltype(shape), decltype(stride)> layoutInit(shape, stride);

int value = Std::get<0>(layoutMake.Shape()); // value = 10
value = Std::get<1>(layoutMake.Shape()); // value = 20
value = Std::get<2>(layoutMake.Shape()); // value = 30

value = Std::get<0>(layoutInit.Stride()); // value = 1
value = Std::get<1>(layoutInit.Stride()); // value = 100
value = Std::get<2>(layoutInit.Stride()); // value = 200

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