Skip to content
版 本

SetLoadDataRepeat

产 品 支 持 情 况

产 品

是 否 支 持

Ascend 950PR/Ascend 950DT

Atlas A3 训 练 系 列 产 品/Atlas A3 推 理 系 列 产 品

Atlas A2 训 练 系 列 产 品/Atlas A2 推 理 系 列 产 品

Atlas 200I/500 A2 推 理 产 品

Atlas 推 理 系 列 产 品AI Core

x

Atlas 推 理 系 列 产 品Vector Core

x

Atlas 训 练 系 列 产 品

x

功 能 说 明

用 于 设 置Load3Dv2接 口的repeat参 数。设 置repeat参 数 后,可 以 通 过 调 用 一 次Load3Dv2接 口 完 成 多 个 迭 代 的 数 据 搬 运。

函 数 原 型

Text
__aicore__ inline void SetLoadDataRepeat(const LoadDataRepeatParam& repeatParams)

参 数 说 明

表 1 参 数 说 明

参 数 名 称

输 入/输 出

含 义

repeatParams

输 入

设 置Load3Dv2接 口 的repeat参 数,类 型 为LoadDataRepeatParam。

具 体 定 义 请 参 考${INSTALL_DIR}/include/ascendc/basic_api/interface/kernel_struct_mm.h,${INSTALL_DIR}请 替 换 为CANN软 件 安 装 后 文 件 存 储 路 径。

参 数 说 明 请 参 考表2

表 2 LoadDataRepeatParam结 构 体 参 数 说 明

参 数 名 称

含 义

repeatStride

height/width方 向 上 的 前 一 个 迭 代 与 后 一 个 迭 代 起 始 地 址 的 距 离,取 值 范 围:n∈[0, 65535],默 认 值 为0。

  • repeatMode为0,repeatStride的 单 位 为16个 元 素。
  • repeatMode为1,repeatStride的 单 位 和 具 体 型 号 有 关。下 文 中 的data_type指Load3Dv2中 源 操 作 数 的 数 据 类 型。

    Ascend 950PR/Ascend 950DT,repeatStride的 单 位 为32/sizeof(data_type)个 元 素 。

    Atlas A2 训 练 系 列 产 品/Atlas A2 推 理 系 列 产 品,repeatStride的 单 位 为32/sizeof(data_type)个 元 素 。

    Atlas A3 训 练 系 列 产 品/Atlas A3 推 理 系 列 产 品,repeatStride的 单 位 为32/sizeof(data_type)个 元 素 。

    Atlas 200I/500 A2 推 理 产 品,repeatStride的 单 位 为64/sizeof(data_type)个 元 素。

repeatTime

height/width方 向 上 的 迭 代 次 数,取 值 范 围:repeatTime ∈[0, 255] 。默 认 值 为1。

repeatMode

控 制repeat迭 代 的 方 向,取 值 范 围:k∈[0, 1] 。默 认 值 为0。

0:迭 代 沿height方 向;

1:迭 代 沿width方 向。

调 用 示 例

参 考调 用 示 例

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