AddrReg【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit产品支持情况Ascend 950PR/Ascend 950DT支持Atlas A3 训练系列产品/Atlas A3 推理系列产品不支持Atlas A2 训练系列产品/Atlas A2 推理系列产品不支持Atlas 200I/500 A2 推理产品不支持Atlas 推理系列产品AI Core不支持Atlas 推理系列产品Vector Core不支持Atlas 训练系列产品不支持功能说明头文件引用路径为 basic_api/reg_compute/kernel_reg_compute_addrreg_intf.h。AddrReg是用于Reg矢量计算接口管理地址偏移量的专用寄存器。在循环外定义后在循环内调用模板函数CreateAddrReg 接口赋值并绑定每层循环轴索引index与步长stride能够在多维循环中逐层累加地址偏移。CreateAddrReg接口支持绑定1-4层for循环返回的地址偏移由index*stride的决定在循环中index每次递增1AddrReg的偏移量就自动增加对应的stride具体的偏移量满足如下伪代码所示// 1层循环: AscendC::Reg::AddrReg aReg; for (uint16_t index0 0; index0 loopNum0; index0){ aReg AscendC::Reg::CreateAddrRegT(index0, stride0); // Offset index0 * stride0 } // 2层循环 AscendC::Reg::AddrReg aReg; for (uint16_t index0 0; index0 loopNum0; index0){ for (uint16_t index1 0; index1 loopNum1; index1){ aReg AscendC::Reg::CreateAddrRegT(index0, stride0, index1, stride1); // Offset index0 * stride0 index1 * stride1 } } // 3层循环 AscendC::Reg::AddrReg aReg; for (uint16_t index0 0; index0 loopNum0; index0){ for (uint16_t index1 0; index1 loopNum1; index1){ for (uint16_t index2 0; index2 loopNum2; index2){ aReg AscendC::Reg::CreateAddrRegT(index0, stride0, index1, stride1, index2, stride2); // Offset index0 * stride0 index1 * stride1 index2 * stride2 } } } // 4层循环 AscendC::Reg::AddrReg aReg; for (uint16_t index0 0; index0 loopNum0; index0){ for (uint16_t index1 0; index1 loopNum1; index1){ for (uint16_t index2 0; index2 loopNum2; index2){ for (uint16_t index3 0; index3 loopNum3; index3){ aReg AscendC::Reg::CreateAddrRegT(index0, stride0, index1, stride1, index2, stride2, index3, stride3); // Offset index0 * stride0 index1 * stride1 index2 * stride2 index3 * stride3 } } } }AddrReg应该通过CreateAddrReg API初始化然后在循环中使用AddrReg存储地址偏移量。AddrReg在每层循环中根据所设置的stride进行自增。函数原型// offset index0 * stride0 template typename T __simd_callee__ inline AddrReg CreateAddrReg(uint16_t index0, uint32_t stride0) // offset index0 * stride0 index1 * stride1 template typename T __simd_callee__ inline AddrReg CreateAddrReg(uint16_t index0, uint32_t stride0, uint16_t index1, uint32_t stride1) // offset index0 * stride0 index1 * stride1 index2 * stride2 template typename T __simd_callee__ inline AddrReg CreateAddrReg(uint16_t index0, uint32_t stride0, uint16_t index1, uint32_t stride1, uint16_t index2, uint32_t stride2) // offset index0 * stride0 index1 * stride1 index2 * stride2 index3 * stride3 template typename T __simd_callee__ inline AddrReg CreateAddrReg(uint16_t index0, uint32_t stride0, uint16_t index1, uint32_t stride1, uint16_t index2, uint32_t stride2, uint16_t index3, uint32_t stride3)参数说明参数含义T模板参数支持的数据类型为b8、b16、b32、b64。index0计算偏移量时作为最外层循环轴。index1计算偏移量时作为第二层循环轴。index2计算偏移量时作为第三层循环轴。index3计算偏移量时作为第四层循环轴。stride0每次循环最外层循环轴对应的地址偏移量单位为element。stride1每次循环第二层循环轴对应的地址偏移量单位为element。stride2每次循环第三层循环轴对应的地址偏移量单位为element。stride3每次循环第四层循环轴对应的地址偏移量单位为element。支持的型号Ascend 950PR/Ascend 950DT约束说明AddrReg 为地址偏移量寄存器仅支持部分搬运指令使用请根据Reg数据搬运接口函数原型选择除了通过AddrReg设置地址偏移支持用户自行累加地址或通过PostUpdate模式进行地址自增完整特性见连续对齐搬入关键特性。通过AddrReg地址偏移进行搬运时需要满足对应搬运指令的地址对齐约束。AddrReg 数量上限为8。由于硬件循环(HardwareLoop)限制AddrReg最多支持4层循环轴。调用示例__simd_vf__ inline void CreateAddrRegVF(__ubuf__ T* dstAddr, __ubuf__ T* srcAddr, uint16_t oneRepeatSize, uint16_t repeatTimes) { AscendC::Reg::MaskReg mask AscendC::Reg::CreateMaskT(); AscendC::Reg::AddrReg aReg; for (uint16_t i 0; i repeatTimes; i) { aReg AscendC::Reg::CreateAddrRegT(i, oneRepeatSize); AscendC::Reg::LoadAlign(mask, srcAddr, aReg); AscendC::Reg::StoreAlign(dstAddr, mask, aReg); } }【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考