asc_copy_gm2ub

产品支持情况

• Ascend 950PR/Ascend 950DT：支持

• Atlas A3 训练系列产品/Atlas A3 推理系列产品：不支持

• Atlas A2 训练系列产品/Atlas A2 推理系列产品：不支持

• Atlas 200I/500 A2 推理产品：不支持

• Atlas 推理系列产品AI Core：不支持

• Atlas 推理系列产品Vector Core：不支持

• Atlas 训练系列产品：不支持

功能说明

将数据从Global Memory (GM) 搬运到 Unified Buffer (UB)。

函数原型

• 前n个数据搬运

C++

__aicore__ inline void asc_copy_gm2ub(__ubuf__ void* dst, __gm__ void* src, uint32_t size)

• 高维切分搬运

C++

__aicore__ inline void asc_copy_gm2ub(__ubuf__ void* dst, __gm__ void* src, uint16_t n_burst, uint16_t len_burst, uint16_t src_stride, uint16_t dst_stride)

• 同步计算

C++

__aicore__ inline void asc_copy_gm2ub_sync(__ubuf__ void* dst, __gm__ void* src, uint32_t size)

参数说明

表1 参数说明

参数名	输入/输出	描述
dst	输出	目的UB地址。
src	输入	源GM地址。
size	输入	搬运数据大小（字节）。
n_burst	输入	待搬运的连续传输数据块个数。取值范围：[1, 4095]。
len_burst	输入	待搬运的每个连续传输数据块的长度，单位为字节。取值范围：[1, 2097151]。
src_stride	输入	源操作数相邻连续数据块的间隔（前面一个数据块的头与后面一个数据块的头的间隔）。 单位为字节。
dst_stride	输入	目的操作数相邻连续数据块的间隔（前面一个数据块的头与后面一个数据块的头的间隔）。 单位为字节。

返回值说明

无

流水类型

PIPE_MTE2

约束说明

• 各存储单元的空间大小和对齐要求请参考存储单元说明。
• 如果需要执行多条asc_copy_gm2ub指令，且asc_copy_gm2ub指令的目的地址存在重叠，需要插入同步指令，保证多个asc_copy_gm2ub指令的串行化，防止出现异常数据。
• 同步计算包含同步等待。
• 当采用前n个数据搬运接口时，搬运数据大小要求32字节对齐。
• 当dst_stride不等于len_burst时，dst_stride要求32字节对齐。

调用示例

C++

//total_length指参与搬运的数据总长度。src是外部输入的half类型的GM内存。
constexpr uint64_t total_length = 128;
__ubuf__ half dst[total_length];
asc_copy_gm2ub(dst, src, total_length * sizeof(half));