Skip to content
版 本

SetFixPipeAddr

产 品 支 持 情 况

产 品

是 否 支 持

Ascend 950PR/Ascend 950DT

x

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

x

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

x

Atlas 200I/500 A2 推 理 产 品

Atlas 推 理 系 列 产 品AI Core

x

Atlas 推 理 系 列 产 品Vector Core

x

Atlas 训 练 系 列 产 品

x

功 能 说 明

DataCopy(CO1->GM)过 程 中 进 行 随 路 量 化 后,通 过 调 用 该 接 口 设 置Elementwise操 作 时LocalTensor的 地 址。

函 数 原 型

Text
template <typename T>
__aicore__ inline void SetFixPipeAddr(const LocalTensor<T>& eleWiseData, uint16_t c0ChStride)

参 数 说 明

表 1 参 数 说 明

参 数 名 称

输 入/输 出

含 义

eleWiseData

输 入

L1 Buffer上 的 源 操 作 数。类 型 为LocalTensor。

支 持 的TPosition为A1/B1/C1。起 始 地 址 需 要 保 证32字 节 对 齐,仅 支 持half数 据 类 型。

c0ChStride

输 入

L1 Buffer上 的C0 channel stride,单 位 是C0_SIZE(32B)。

eleWiseData沿N方 向 以C0为 单 位 切 分 得 到 的 数 据 块 称 为C0 channel,两 块C0 channel的 间 隔 称 之 为C0 channel stride。

约 束 说 明

返 回 值 说 明

调 用 示 例

完 整 示 例 可 参 考完 整 示 例

DataCopy随 路 量 化 搬 运 后,可 以 逐 个 元 素 加/减 一 个 大 小 为mSize * nSize的LocalTensor,具 体LocalTensor地 址 相 关 参 数 需 要 调 用SetFixPipeAddr来 设 置。

Text
__aicore__inline void SetEleSrcPara(const LocalTensor <half>& eleWiseData, uint16_t c0ChStride)
{
    AscendC::SetFixPipeAddr(eleWiseData, c0ChStride);
}

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