asc_set_ub2gm_loop_size
【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言，原生支持C和C++标准规范，主要由类库和语言扩展层构成，提供多层级API，满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
功能说明
将数据从Unified Buffer (UB)搬运到Global Memory (GM)时，通过调用该接口设置内层循环和外层循环的次数。
以源操作数搬运场景为例，如下图所示。
函数原型
__aicore__ inline void asc_set_ub2gm_loop_size(uint32_t loop1_size, uint32_t loop2_size)参数说明
| 参数名 | 输入/输出 | 描述 |
|---|---|---|
| loop1_size | 输入 | 用于设置内层循环的循环次数，取值范围为[0, 2^21)。 |
| loop2_size | 输入 | 用于设置外层循环的循环次数，取值范围为[0, 2^21)。 |
返回值说明
无
流水类型
PIPE_S
约束说明
每次设置循环相关参数后，需要进行寄存器的复位（循环次数设置为1），否则会影响下一次搬运的使用。
调用示例
uint32_t loop1_size = 2; uint32_t loop2_size = 2; uint64_t loop1_src_stride = 96; uint64_t loop1_dst_stride = 128; uint64_t loop2_src_stride = 192; uint64_t loop2_dst_stride = 288; asc_set_ub2gm_loop_size(loop1_size, loop2_size); asc_set_ub2gm_loop1_stride(loop1_src_stride, loop1_dst_stride); asc_set_ub2gm_loop2_stride(loop2_src_stride, loop2_dst_stride); asc_copy_ub2gm_align(dst, src, 2, 48 * sizeof(int8_t), 0, 0, 0); asc_set_ub2gm_loop_size(1, 1);【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言，原生支持C和C++标准规范，主要由类库和语言扩展层构成，提供多层级API，满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit
创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
