CANN asc-devkit解交织加载API

📅 2026/7/16 19:06:35
CANN asc-devkit解交织加载API
asc_loadalign_deintlv【免费下载链接】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 训练系列产品不支持功能说明头文件路径c_api/reg_compute/reg_load.h。asc_loadalign_deintlv用于从UB中读取以32B对齐地址为起始位置的连续2 x VL长度数据。数据按元素类型宽度进行解释在Load过程中完成解交织后分别写入两个目的矢量数据寄存器。以float数据类型、Ascend 950PR/Ascend 950DT产品为例若VL 256B则单次操作覆盖64个float元素。调用asc_loadalign_deintlv后数据从UB到Reg的排布如下图所示。提示使用本接口需要包含头文件reg_load.h。函数原型asc_loadalign_deintlv按源地址的寻址方式分为以下三类接口基址寻址接口、标量偏移寻址接口、地址寄存器偏移寻址接口。支持的数据类型请参见数据类型章节。基址寻址接口__simd_callee__ inline void asc_loadalign_deintlv(vector_int4x2_t dst0, vector_int4x2_t dst1, __ubuf__ int4b_t* src) __simd_callee__ inline void asc_loadalign_deintlv(vector_int8_t dst0, vector_int8_t dst1, __ubuf__ int8_t* src) __simd_callee__ inline void asc_loadalign_deintlv(vector_uint8_t dst0, vector_uint8_t dst1, __ubuf__ uint8_t* src) __simd_callee__ inline void asc_loadalign_deintlv(vector_fp4x2_e2m1_t dst0, vector_fp4x2_e2m1_t dst1, __ubuf__ fp4x2_e2m1_t* src) __simd_callee__ inline void asc_loadalign_deintlv(vector_fp4x2_e1m2_t dst0, vector_fp4x2_e1m2_t dst1, __ubuf__ fp4x2_e1m2_t* src) __simd_callee__ inline void asc_loadalign_deintlv(vector_hifloat8_t dst0, vector_hifloat8_t dst1, __ubuf__ hifloat8_t* src) __simd_callee__ inline void asc_loadalign_deintlv(vector_fp8_e8m0_t dst0, vector_fp8_e8m0_t dst1, __ubuf__ fp8_e8m0_t* src) __simd_callee__ inline void asc_loadalign_deintlv(vector_fp8_e5m2_t dst0, vector_fp8_e5m2_t dst1, __ubuf__ fp8_e5m2_t* src) __simd_callee__ inline void asc_loadalign_deintlv(vector_fp8_e4m3fn_t dst0, vector_fp8_e4m3fn_t dst1, __ubuf__ fp8_e4m3fn_t* src) __simd_callee__ inline void asc_loadalign_deintlv(vector_int16_t dst0, vector_int16_t dst1, __ubuf__ int16_t* src) __simd_callee__ inline void asc_loadalign_deintlv(vector_uint16_t dst0, vector_uint16_t dst1, __ubuf__ uint16_t* src) __simd_callee__ inline void asc_loadalign_deintlv(vector_half dst0, vector_half dst1, __ubuf__ half* src) __simd_callee__ inline void asc_loadalign_deintlv(vector_bfloat16_t dst0, vector_bfloat16_t dst1, __ubuf__ bfloat16_t* src) __simd_callee__ inline void asc_loadalign_deintlv(vector_int32_t dst0, vector_int32_t dst1, __ubuf__ int32_t* src) __simd_callee__ inline void asc_loadalign_deintlv(vector_uint32_t dst0, vector_uint32_t dst1, __ubuf__ uint32_t* src) __simd_callee__ inline void asc_loadalign_deintlv(vector_float dst0, vector_float dst1, __ubuf__ float* src)标量偏移寻址接口源地址通过src offset计算得到且计算结果必须满足32B对齐要求。offset的单位为元素个数实际地址偏移字节数为offset * sizeof(data_type)。__simd_callee__ inline void asc_loadalign_deintlv(vector_int8_t dst0, vector_int8_t dst1, __ubuf__ int8_t* src, int32_t offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_uint8_t dst0, vector_uint8_t dst1, __ubuf__ uint8_t* src, int32_t offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_fp4x2_e2m1_t dst0, vector_fp4x2_e2m1_t dst1, __ubuf__ fp4x2_e2m1_t* src, int32_t offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_fp4x2_e1m2_t dst0, vector_fp4x2_e1m2_t dst1, __ubuf__ fp4x2_e1m2_t* src, int32_t offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_fp8_e8m0_t dst0, vector_fp8_e8m0_t dst1, __ubuf__ fp8_e8m0_t* src, int32_t offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_fp8_e5m2_t dst0, vector_fp8_e5m2_t dst1, __ubuf__ fp8_e5m2_t* src, int32_t offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_fp8_e4m3fn_t dst0, vector_fp8_e4m3fn_t dst1, __ubuf__ fp8_e4m3fn_t* src, int32_t offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_hifloat8_t dst0, vector_hifloat8_t dst1, __ubuf__ hifloat8_t* src, int32_t offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_int16_t dst0, vector_int16_t dst1, __ubuf__ int16_t* src, int32_t offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_uint16_t dst0, vector_uint16_t dst1, __ubuf__ uint16_t* src, int32_t offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_half dst0, vector_half dst1, __ubuf__ half* src, int32_t offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_bfloat16_t dst0, vector_bfloat16_t dst1, __ubuf__ bfloat16_t* src, int32_t offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_int32_t dst0, vector_int32_t dst1, __ubuf__ int32_t* src, int32_t offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_uint32_t dst0, vector_uint32_t dst1, __ubuf__ uint32_t* src, int32_t offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_float dst0, vector_float dst1, __ubuf__ float* src, int32_t offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_int4x2_t dst0, vector_int4x2_t dst1, __ubuf__ int4b_t* src, int32_t offset)地址寄存器偏移寻址接口该类接口通过地址寄存器addr_reg offset生成实际访问地址。地址寄存器通常在循环场景中使用需配合asc_create_addr_reg完成初始化。实际访问地址必须满足32B对齐要求。__simd_callee__ inline void asc_loadalign_deintlv(vector_fp4x2_e1m2_t dst0, vector_fp4x2_e1m2_t dst1, __ubuf__ fp4x2_e1m2_t* src, addr_reg offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_fp4x2_e2m1_t dst0, vector_fp4x2_e2m1_t dst1, __ubuf__ fp4x2_e2m1_t* src, addr_reg offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_uint8_t dst0, vector_uint8_t dst1, __ubuf__ uint8_t* src, addr_reg offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_int8_t dst0, vector_int8_t dst1, __ubuf__ int8_t* src, addr_reg offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_fp8_e4m3fn_t dst0, vector_fp8_e4m3fn_t dst1, __ubuf__ fp8_e4m3fn_t* src, addr_reg offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_fp8_e5m2_t dst0, vector_fp8_e5m2_t dst1, __ubuf__ fp8_e5m2_t* src, addr_reg offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_fp8_e8m0_t dst0, vector_fp8_e8m0_t dst1, __ubuf__ fp8_e8m0_t* src, addr_reg offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_hifloat8_t dst0, vector_hifloat8_t dst1, __ubuf__ hifloat8_t* src, addr_reg offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_uint16_t dst0, vector_uint16_t dst1, __ubuf__ uint16_t* src, addr_reg offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_int16_t dst0, vector_int16_t dst1, __ubuf__ int16_t* src, addr_reg offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_bfloat16_t dst0, vector_bfloat16_t dst1, __ubuf__ bfloat16_t* src, addr_reg offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_half dst0, vector_half dst1, __ubuf__ half* src, addr_reg offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_uint32_t dst0, vector_uint32_t dst1, __ubuf__ uint32_t* src, addr_reg offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_int32_t dst0, vector_int32_t dst1, __ubuf__ int32_t* src, addr_reg offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_float dst0, vector_float dst1, __ubuf__ float* src, addr_reg offset) __simd_callee__ inline void asc_loadalign_deintlv(vector_int4x2_t dst0, vector_int4x2_t dst1, __ubuf__ int4b_t* src, addr_reg offset)参数说明参数名输入/输出说明dst0输出第一个目的向量寄存器。dst1输出第二个目的向量寄存器。src输入源数据在UB中的起始地址。对于不同寻址接口src分别表示基址或参与地址计算的基地址。offset输入地址偏移量。对于标量偏移寻址接口单位为元素个数对于地址寄存器偏移寻址接口表示地址寄存器。寄存器类型的详细说明请参见data_type_definition.md。数据类型asc_loadalign_deintlv支持的数据类型如下表所示。三类寻址接口均支持表中所列数据类型。源数据类型src目的寄存器类型dst0/dst1int4b_tvector_int4x2_tfp4x2_e1m2_tvector_fp4x2_e1m2_tfp4x2_e2m1_tvector_fp4x2_e2m1_tint8_tvector_int8_tuint8_tvector_uint8_tfp8_e4m3fn_tvector_fp8_e4m3fn_tfp8_e5m2_tvector_fp8_e5m2_tfp8_e8m0_tvector_fp8_e8m0_thifloat8_tvector_hifloat8_tint16_tvector_int16_tuint16_tvector_uint16_thalfvector_halfbfloat16_tvector_bfloat16_tint32_tvector_int32_tuint32_tvector_uint32_tfloatvector_float返回值说明无。流水类型PIPE_V约束说明对于基址寻址接口src必须为32B对齐地址且访问范围不能超过UB有效地址上界减去2 x VL。对于标量偏移寻址接口和地址寄存器偏移寻址接口实际访问地址必须为32B对齐且访问范围不能超过UB有效地址上界减去2 x VL。使用地址寄存器偏移寻址接口时需要先通过asc_create_addr_reg完成地址寄存器初始化。调用示例从UB中连续读取以32B对齐地址src为起始位置的float数据。基址寻址接口for (uint16_t i 0; i repeat_times; i) { asc_loadalign_deintlv(reg0, reg1, src i * one_repeat_size); ... }标量偏移寻址接口// 方式1通过offset控制偏移 for (uint16_t i 0; i repeat_times; i) { asc_loadalign_deintlv(reg0, reg1, src, i * one_repeat_size); ... } // 方式2通过src控制偏移 for (uint16_t i 0; i repeat_times; i) { asc_loadalign_deintlv(reg0, reg1, src i * one_repeat_size, 0); ... }地址寄存器偏移寻址接口for (uint16_t i 0; i repeat_times; i) { addr_reg addr_reg asc_create_addr_reg_b32(one_repeat_size); asc_loadalign_deintlv(reg0, reg1, src, addr_reg); ... }【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考