asc_copy
产品支持情况
- Ascend 950PR/Ascend 950DT:支持
- Atlas A3 训练系列产品/Atlas A3 推理系列产品:不支持
- Atlas A2 训练系列产品/Atlas A2 推理系列产品:不支持
- Atlas 200I/500 A2 推理产品:不支持
- Atlas 推理系列产品AI Core:不支持
- Atlas 推理系列产品Vector Core:不支持
- Atlas 训练系列产品:不支持
功能说明
完成数据在寄存器间的搬运,支持以下三种模式:
矢量数据寄存器复制到矢量数据寄存器:将src中被mask筛选的有效元素复制到dst对应位置,未被mask筛选的位置保持dst原值。
掩码寄存器复制到掩码寄存器:将src的所有bit位复制到dst中。支持带mask和不带mask两种重载:
- 带mask:将src中被mask筛选的bit复制到dst。对于b8模式,mask的每个bit控制src的一个bit;对于b16模式,mask的每2bit为一组、仅LSB控制src的一个bit;对于b32模式,mask的每4bit为一组、仅LSB控制src的一个bit。
- 不带mask:将src的所有bit复制到dst。
矢量数据寄存器复制到掩码寄存器:
- 数据类型为uint16_t时,矢量数据寄存器(256B)分成16个数据块(16B),用户通过参数可选择某一个数据块,数据块的每个bit broadcast成2bit后变为32B,即输出的mask。
- 数据类型为uint32_t时,矢量数据寄存器(256B)分成32个数据块(8B),用户通过参数可选择某一个数据块,数据块的每个bit broadcast成4bit后变为32B,即输出的mask。
具体搬运原理请参考关键特性说明。
函数原型
矢量数据寄存器复制到矢量数据寄存器
C++__simd_callee__ inline void asc_copy(vector_int8_t& dst, vector_int8_t src, vector_bool mask) __simd_callee__ inline void asc_copy(vector_uint8_t& dst, vector_uint8_t src, vector_bool mask) __simd_callee__ inline void asc_copy(vector_int16_t& dst, vector_int16_t src, vector_bool mask) __simd_callee__ inline void asc_copy(vector_uint16_t& dst, vector_uint16_t src, vector_bool mask) __simd_callee__ inline void asc_copy(vector_half& dst, vector_half src, vector_bool mask) __simd_callee__ inline void asc_copy(vector_bfloat16_t& dst, vector_bfloat16_t src, vector_bool mask) __simd_callee__ inline void asc_copy(vector_int32_t& dst, vector_int32_t src, vector_bool mask) __simd_callee__ inline void asc_copy(vector_uint32_t& dst, vector_uint32_t src, vector_bool mask) __simd_callee__ inline void asc_copy(vector_float& dst, vector_float src, vector_bool mask)掩码寄存器复制到掩码寄存器
C++__simd_callee__ inline void asc_copy(vector_bool& dst, vector_bool src, vector_bool mask) __simd_callee__ inline void asc_copy(vector_bool& dst, vector_bool src)矢量数据寄存器复制到掩码寄存器
C++__simd_callee__ inline void asc_copy(vector_bool& dst, vector_uint16_t src, int16_t part) __simd_callee__ inline void asc_copy(vector_bool& dst, vector_uint32_t src, int16_t part)
参数说明
表1 参数说明(矢量数据寄存器复制到矢量数据寄存器)
| 参数名 | 输入/输出 | 描述 |
|---|---|---|
| dst | 输出 | 目的操作数(矢量数据寄存器)。 |
| src | 输入 | 源操作数(矢量数据寄存器)。 |
| mask | 输入 | 源操作数掩码(掩码寄存器),用于指示在计算过程中哪些元素参与计算。对应位置为1时参与计算,为0时不参与计算。mask未筛选的元素在输出中保持原值。 |
表2 参数说明(掩码寄存器复制到掩码寄存器)
| 参数名 | 输入/输出 | 描述 |
|---|---|---|
| dst | 输出 | 目的操作数(掩码寄存器)。 |
| src | 输入 | 源操作数(掩码寄存器)。 |
| mask | 输入 | 源操作数掩码(掩码寄存器),用于指示在计算过程中哪些元素参与计算。对应位置为1时参与计算,为0时不参与计算。mask未筛选的元素在输出中置零。 |
表3 参数说明(矢量数据寄存器复制到掩码寄存器)
| 参数名 | 输入/输出 | 描述 |
|---|---|---|
| dst | 输出 | 目的操作数(掩码寄存器)。 |
| src | 输入 | 源操作数(矢量数据寄存器)。 |
| part | 输入 | part决定了src中需要搬运的数据块,具体描述请参考关键特性说明。 • 当数据类型为uint16_t,part∈[0, 15]。 • 当数据类型为uint32_t,part∈[0, 31]。 |
矢量数据寄存器和掩码寄存器的详细说明请参见reg数据类型定义。
返回值说明
无
约束说明
- 针对矢量数据寄存器复制到矢量数据寄存器:dst中未被mask筛选的位置保持原值。
- 针对掩码寄存器复制到掩码寄存器:带mask的接口dst中未被mask筛选的位置填0。
- 针对矢量寄存器复制到掩码寄存器:当数据类型为uint16_t,part∈[0, 15],当数据类型为uint32_t,part∈[0, 31]。
关键特性说明
针对于矢量数据寄存器复制到掩码寄存器:
由掩码寄存器原理可知,当操作数类型为b16时,每一个element对应2bit的掩码寄存器,且仅2bit中的最低位是有效的;当操作数类型为b32时,每一个element对应4bit的掩码寄存器,且仅4bit中的最低位是有效的。
- 当操作数类型为uint16_t时,矢量数据寄存器的一个数据块大小为16B,每个bit经过broadcast成2bit后变为32B,即一个MaskReg(32B)。此外,一个矢量数据寄存器有256B/16B = 16个数据块,偏移量part取值范围为[0, 15]。
- 当操作数类型为b32时,矢量数据寄存器的一个数据块大小为8B,每个bit经过broadcast成4bit后变为32B,即一个MaskReg(32B)。此外,一个矢量数据寄存器有256B/8B = 32个数据块,偏移量part取值范围为[0, 31]。
调用示例
矢量数据寄存器复制到矢量数据寄存器
C++__simd_vf__ inline void copy_vf(__ubuf__ half* dst_addr, __ubuf__ half* src_addr, uint32_t count, uint16_t one_repeat_size, uint16_t repeat_time) { vector_half dst; vector_half src; vector_bool mask; for (uint16_t i = 0; i < repeat_time; ++i) { mask = asc_update_mask_b16(count); asc_loadalign(src, src_addr + i * one_repeat_size); asc_copy(dst, src, mask); asc_storealign(dst_addr + i * one_repeat_size, dst, mask); } }掩码寄存器复制到掩码寄存器
C++__simd_vf__ inline void copy_mask_vf(__ubuf__ uint32_t* dst_addr, __ubuf__ uint32_t* src_addr, uint32_t count, uint16_t one_block_size, uint16_t repeat_time) { vector_bool dst; vector_bool src; vector_bool mask; for (uint16_t i = 0; i < repeat_time; ++i) { mask = asc_update_mask_b16(count); asc_loadalign_postupdate(src, src_addr); asc_copy(dst, src, mask); asc_storealign_postupdate(dst_addr, dst, one_block_size); } }矢量数据寄存器复制到掩码寄存器
C++__simd_vf__ inline void copy_vector_to_mask_vf(__ubuf__ uint16_t* dst_addr, __ubuf__ uint16_t* src_addr) { vector_uint16_t src; vector_bool mask; asc_loadalign(src, src_addr); asc_copy(mask, src, 0); asc_storealign(dst_addr, mask); }

