Skip to content
版 本

asc_set_ub2gm_loop2_stride

产 品 支 持 情 况

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

功 能 说 明

将 数 据 从Unified Buffer (UB)搬 运 到Global Memory (GM)时,通 过 调 用 该 接 口 设 置 外 层 循 环 中 源 操 作 数 在 相 邻 迭 代 间 的 数 据 块 间 隔,以 及 目 的 操 作 数 在 相 邻 迭 代 间 的 数 据 块 间 隔。

以 源 操 作 数 搬 运 场 景 为 例,如 下 图 所 示。

源操作数搬运场景示例

函 数 原 型

C++
__aicore__ inline void asc_set_ub2gm_loop2_stride(uint64_t loop2_src_stride, uint64_t loop2_dst_stride)

参 数 说 明

参 数 名输 入/输 出描 述
loop2_src_stride输 入用 于 设 置 外 层 循 环 中 相 邻 迭 代 源 操 作 数 的 数 据 块 间 的 间 隔,单 位 为Byte,取 值 范 围 为[0, 2^21),并 且loop2_src_stride必 须32B对 齐。
loop2_dst_stride输 入用 于 设 置 外 层 循 环 中 相 邻 迭 代 目 的 操 作 数 的 数 据 块 间 的 间 隔,单 位 为Byte,取 值 范 围 为[0, 2^40)。

返 回 值 说 明

流 水 类 型

PIPE_S

约 束 说 明

每 次 设 置 循 环 相 关 参 数 后,需 要 进 行 寄 存 器 的 复 位(循 环 次 数 设 置 为1),否 则 会 影 响 下 一 次 搬 运 的 使 用。

调 用 示 例

C++
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 仓 master 分 支 自 动 编 译 生 成,属 于 持 续 开 发 版 本,可 能 存 在 缺 陷,仅 供 预 览 与 参 考。如 需 稳 定 及 商 用 资 料,请 查 阅 官 方 昇 腾 社 区