Skip to content

asc_storeunalign_post

产品支持情况

  • Ascend 950PR/Ascend 950DT:支持
  • Atlas A3 训练系列产品/Atlas A3 推理系列产品:不支持
  • Atlas A2 训练系列产品/Atlas A2 推理系列产品:不支持
  • Atlas 200I/500 A2 推理产品:不支持
  • Atlas 推理系列产品AI Core:不支持
  • Atlas 推理系列产品Vector Core:不支持
  • Atlas 训练系列产品:不支持

功能说明

reg计算数据搬运接口,适用于从矢量数据寄存器连续非32B对齐的起始地址连续搬出到UB的尾块场景。

需要先调用下列接口之一后,再调用本接口。

搬运原理如下: 记目的操作数的起始地址为dst_start,结束地址为dst_end,尾块元素个数为unalign_count = (dst_end - dst_end / 32 * 32) / sizeof(T)。 则搬运的数据会分为两部分,分别是:

  • 32B对齐的主块部分UB[dst_start, dst_end - unalign_count * sizeof(T)]。
  • 非32B对齐的尾块部分UB[dst_end - unalign_count * sizeof(T), dst_end]。 store_unalign或store_unalign_postupdate接口执行时,会将主块搬出至UB,尾块暂存至非对齐寄存器[0, unalign_count]。 本接口执行时,会将尾块从非对齐寄存器搬出至UB。

函数原型

  • 不指定偏移量,配合asc_storeunalign_postupdate使用不指定存储偏移量的接口。

    C++
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int8_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint8_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp4x2_e2m1_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp4x2_e1m2_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int4b_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e8m0_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e5m2_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e4m3fn_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int16_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint16_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ half* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ bfloat16_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int32_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint32_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ float* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int64_t* dst, vector_store_unalign src)
    
  • 使用int32_t作为偏移量,配合asc_storeunalign接口。

    C++
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int8_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint8_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp4x2_e2m1_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp4x2_e1m2_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int4b_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e8m0_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e5m2_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e4m3fn_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ hifloat8_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int16_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint16_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ half* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ bfloat16_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int32_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint32_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ float* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int64_t* dst, vector_store_unalign src, int32_t offset)
    
  • 使用addr_reg作为偏移量,配合asc_storeunalign_postupdate使用addr_reg作为存储偏移量的接口。

    C++
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int8_t* dst, vector_store_unalign src, addr_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint8_t* dst, vector_store_unalign src, addr_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp4x2_e2m1_t* dst, vector_store_unalign src, addr_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp4x2_e1m2_t* dst, vector_store_unalign src, addr_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int4b_t* dst, vector_store_unalign src, addr_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e8m0_t* dst, vector_store_unalign src, addr_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e5m2_t* dst, vector_store_unalign src, addr_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e4m3fn_t* dst, vector_store_unalign src, addr_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int16_t* dst, vector_store_unalign src, addr_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint16_t* dst, vector_store_unalign src, addr_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ half* dst, vector_store_unalign src, addr_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ bfloat16_t* dst, vector_store_unalign src, addr_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int32_t* dst, vector_store_unalign src, addr_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint32_t* dst, vector_store_unalign src, addr_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ float* dst, vector_store_unalign src, addr_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int64_t* dst, vector_store_unalign src, addr_reg offset)
    

参数说明

  • 不指定偏移量

    表1 参数说明

    参数名输入/输出描述
    dst输出目的操作数(矢量)的起始地址。
    src输入非对齐寄存器,用于保存非对齐数据,长度32B。
  • 使用int32_t作为存储偏移量

    表2 参数说明

    参数名输入/输出描述
    dst输出目的操作数(矢量)的起始地址。
    src输入非对齐寄存器,用于保存非对齐数据,长度32B。
    offset输入目的操作数结束地址的偏移,类型为int32_t,单位为元素个数。
  • 使用addr_reg作为存储偏移量

    表3 参数说明

    参数名输入/输出描述
    dst输出目的操作数(矢量)的起始地址。
    src输入非对齐寄存器,用于保存非对齐数据,长度32B。
    offset输入目的操作数结束地址的偏移,类型为地址寄存器,单位为元素个数。

非对齐寄存器和地址寄存器的详细说明请参见reg数据类型定义

返回值说明

流水类型

PIPE_V

约束说明

  • 该接口中的dst不需要32B对齐,但数据类型为T的dst需要sizeof(T) Byte对齐。
  • 调用本接口之前,需要调用asc_storeunalign接口,或者asc_storeunalign_postupdate接口,且本接口与前序接口的非对齐寄存器要保持一致。
  • 需要保证目的操作数的地址加上offset对应的偏移地址,结果等于数据搬运的结束地址,具体见调用示例。

调用示例

  • 不指定偏移量

    C++
    // dst地址为8,非32B对齐。范围为[8:520]。
    __ubuf__ uint32_t* dst = (__ubuf__ uint32_t*)asc_get_phy_buf_addr(8);
    vector_store_unalign ureg;
    vector_uint32_t src;
    asc_storeunalign_postupdate(dst, ureg, src);
    asc_storeunalign_post(dst, ureg);
    
  • 使用int32_t作为存储偏移量

    C++
    // dst地址为8,非32B对齐,占用的地址为UB[8:520]。
    __ubuf__ uint32_t* dst = (__ubuf__ uint32_t*)asc_get_phy_buf_addr(8);
    vector_store_unalign ureg;
    vector_uint32_t src;
    uint32_t count = 64;
    uint32_t repeat = 2;
    for (uint32_t i = 0; i < repeat; i++) {
        // 其他reg操作
    
        // 第一次:src[0:62] 写入dst[0:62](即UB[8:256]);同时src[62:64] 写入ureg[0:2]
        // 第二次:ureg[0:2] + src[0:62] 写入dst[62:126](即UB[256:512]);同时src[62:64] 写入ureg[0:2]
        asc_storeunalign(dst + i*count, ureg, src, count);
    }
    // dst结束地址为520,即dst + repeat * count * sizeof(uint32_t),所以offset要配置为repeat * count
    int32_t offset = repeat * count;
    // ureg[0:2] 写入dst[126:128](即UB[512:520])
    asc_storeunalign_post(dst, ureg, offset);
    
  • 使用addr_reg作为存储偏移量

    C++
    // dst地址为8,非32B对齐。占用的地址为[8:520]。
    __ubuf__ uint32_t* dst = (__ubuf__ uint32_t*)asc_get_phy_buf_addr(8);
    vector_store_unalign ureg;
    vector_uint32_t src;
    uint32_t repeat = 2;
    for (uint32_t i = 0; i < repeat; i++) {
        addr_reg count = asc_update_addr_reg_b32(64);
        // 其他reg操作
    
        // 第一次:src[0:62] 写入dst[0:62](即UB[8:256]);同时src[62:64] 写入ureg[0:2];同时dst自增为dst[64](即UB[264])
        // 第二次:ureg[0:2] + src[0:62] 写入dst[-2:62](即UB[256:512]);同时src[62:64] 写入ureg[0:2];同时dst自增为dst[128](即UB[520])
        asc_storeunalign_postupdate(dst, ureg, src, count);
    }
    // dst结束地址为520,但此时dst已经自增到了520,所以offset要配置为0 
    addr_reg offset = asc_update_addr_reg_b32(0);
    // ureg[0:2] 写入dst[-2:0](即UB[512:520])
    asc_storeunalign_post(dst, ureg, offset);
    

免责声明:本站内容由 asc-devkit 仓 master 分支自动编译生成,属于持续开发版本,可能存在缺陷,仅供预览与参考。如需稳定及商用资料,请查阅官方 昇腾社区