asc_copy_ub2l1

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT

功能说明

将数据从Unified Buffer (UB) 搬运到L1 Buffer。

函数原型

  • 前n个数据搬运
__aicore__ inline void asc_copy_ub2l1(__cbuf__ void* dst, __ubuf__ void* src, uint32_t size)
  • 高维切分搬运
__aicore__ inline void asc_copy_ub2l1(__cbuf__ void* dst, __ubuf__ void* src, uint16_t n_burst, uint16_t len_burst, uint16_t src_gap, uint16_t dst_gap)
  • 同步搬运
__aicore__ inline void asc_copy_ub2l1_sync(__cbuf__ void* dst, __ubuf__ void* src, uint32_t size)

参数说明

参数名 输入/输出 描述
dst 输出 目的L1 Buffer地址。
src 输入 源UB地址。
size 输入 搬运数据大小(字节)。
n_burst 输入 待搬运的连续传输数据块个数。取值范围:[1, 4095]。
len_burst 输入 待搬运的每个连续传输数据块的长度,单位为DataBlock(32字节)。取值范围:[1, 65535]。
src_gap 输入 源操作数相邻连续数据块的间隔(前面一个数据块的尾与后面一个数据块的头的间隔)。
单位为DataBlock(32字节)。
dst_gap 输入 目的操作数相邻连续数据块的间隔(前面一个数据块的尾与后面一个数据块的头的间隔)。
单位为DataBlock(32字节)。

返回值说明

流水类型

PIPE_MTE3

约束说明

  • dst的起始地址要求按照对应数据类型所占字节数对齐。
  • src的起始地址要求32字节对齐。
  • 如果需要执行多条asc_copy_ub2l1指令,且asc_copy_ub2l1指令的目的地址存在重叠,需要插入同步指令,保证多个asc_copy_ub2l1指令的串行化,防止出现异常数据。
  • 同步计算包含同步等待。

调用示例

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