企业官网建设流程全解析

asc_copy_ub2gm

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言，原生支持C和C++标准规范，主要由类库和语言扩展层构成，提供多层级API，满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit

产品支持情况

功能说明

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

函数原型

• 前n个数据搬运

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

• 高维切分搬运

__aicore__ inline void asc_copy_ub2gm(__gm__ 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_ub2gm_sync(__gm__ void* dst, __ubuf__ void* src, uint32_t size)

参数说明

返回值说明

无

流水类型

PIPE_MTE3

约束说明

• dst的起始地址要求按照对应数据类型所占字节数对齐。
• src的起始地址要求32字节对齐。
• 如果需要执行多条asc_copy_ub2gm指令，且asc_copy_ub2gm指令的目的地址存在重叠，需要插入同步指令，保证多个asc_copy_ub2gm指令的串行化，防止出现异常数据。
• 同步计算包含同步等待。
• 当采用前n个数据搬运接口时，搬运数据大小要求32字节对齐。

调用示例

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

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言，原生支持C和C++标准规范，主要由类库和语言扩展层构成，提供多层级API，满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit