Skip to content

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_gap, uint16_t dst_gap)
  • 同步计算
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输入待搬运的每个连续传输数据块的长度,单位为DataBlock(32字节)。取值范围:[1, 65535]。
src_gap输入源操作数相邻连续数据块的间隔(前面一个数据块的尾与后面一个数据块的头的间隔)。
单位为DataBlock(32字节)。
dst_gap输入目的操作数相邻连续数据块的间隔(前面一个数据块的尾与后面一个数据块的头的间隔)。
单位为DataBlock(32字节)。

返回值说明

流水类型

PIPE_MTE2

约束说明

  • 各存储单元的空间大小和对齐要求请参考存储单元说明
  • 如果需要执行多条asc_copy_gm2ub指令,且asc_copy_gm2ub指令的目的地址存在重叠,需要插入同步指令,保证多个asc_copy_gm2ub指令的串行化,防止出现异常数据。
  • 同步计算包含同步等待。
  • 当采用前n个数据搬运接口时,搬运数据大小要求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));

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