CANN/asc-devkit L1到Fixpipe数据搬运API

asc_copy_l12fb

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

产品支持情况

• Ascend 950PR/Ascend 950DT：不支持

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

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

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

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

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

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

功能说明

头文件路径："c_api/cube_datamove/cube_datamove.h"。

将数据从L1 Buffer搬运到Fixpipe Buffer中，Fixpipe Buffer用于存放量化参数。

函数原型

• 前n个数据搬运

  __aicore__ inline void asc_copy_l12fb(__fbuf__ void* dst, __cbuf__ void* src, uint32_t size)

• 高维切分搬运

  __aicore__ inline void asc_copy_l12fb(__fbuf__ void* dst, __cbuf__ void* src, uint16_t n_burst, uint16_t len_burst, uint16_t src_gap_size, uint16_t dst_gap_size)

• 同步搬运

  __aicore__ inline void asc_copy_l12fb_sync(__fbuf__ void* dst, __cbuf__ void* src, uint32_t size)

参数说明

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

返回值说明

无

流水类型

PIPE_MTE1

约束说明

• dst、src的起始地址需要32字节对齐。
• 操作数地址重叠约束请参考通用地址重叠约束。
• 当采用前n个数据搬运接口时，搬运数据大小要求32字节对齐。

调用示例

constexpr uint16_t n_burst = 1; constexpr uint16_t len_burst = 1; constexpr uint16_t src_gap_size = 0; constexpr uint16_t dst_gap_size = 1; __cbuf__ half src[256]; __fbuf__ half dst[256]; asc_copy_l12fb(dst, src, n_burst, len_burst, src_gap_size, dst_gap_size);

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