Truncate(ISASI)

产 品 支 持 情 况

产 品	是 否 支 持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训 练 系 列 产 品/Atlas A3 推 理 系 列 产 品	x
Atlas A2 训 练 系 列 产 品/Atlas A2 推 理 系 列 产 品	x
Atlas 200I/500 A2 推 理 产 品	x
Atlas 推 理 系 列 产 品AI Core	x
Atlas 推 理 系 列 产 品Vector Core	x
Atlas 训 练 系 列 产 品	x

功 能 说 明

将 源 操 作 数 的 浮 点 数 元 素 截 断 到 整 数 位，同 时 源 操 作 数 的 数 据 类 型 保 持 不 变。

函 数 原 型

Text

template <typename T, RoundMode roundMode>
__aicore__ inline void Truncate(const LocalTensor<T> &dst, const LocalTensor<T> &src, const uint32_t count)

参 数 说 明

表 1 模 板 参 数 说 明

参 数 名	描 述
T	操 作 数 数 据 类 型。 Ascend 950PR/Ascend 950DT，支 持 的 数 据 类 型 为：half、float、bfloat16_t。
roundMode	舍 入 模 式，取 值 如 下。 RoundMode::CAST_RINT：返 回 最 接 近 的 整 数，若 存 在 两 个 同 样 接 近 的 整 数，则 选 择 偶 数。 RoundMode::CAST_ROUND：四 舍 五 入 取 整。 RoundMode::CAST_FLOOR：向 下 取 整。 RoundMode::CAST_CEIL：向 上 取 整。 RoundMode::CAST_TRUNC：截 断 取 整（直 接 舍 弃 小 数 部 分）。

表 2 参 数 说 明

参 数 名	输 入/输 出	描 述
dst	输 出	目 的 操 作 数。 类 型 为LocalTensor，支 持 的TPosition为VECIN/VECCALC/VECOUT。 LocalTensor的 起 始 地 址 需 要32字 节 对 齐。
src	输 入	源 操 作 数。 类 型 为LocalTensor，支 持 的TPosition为VECIN/VECCALC/VECOUT。 LocalTensor的 起 始 地 址 需 要32字 节 对 齐。 源 操 作 数 的 数 据 类 型 需 要 与 目 的 操 作 数 保 持 一 致。
count	输 入	参 与 计 算 的 元 素 个 数。

返 回 值 说 明

无

约 束 说 明

• 操 作 数 地 址 对 齐 要 求 请 参 见通 用 地 址 对 齐 约 束。

调 用 示 例

Text

AscendC::Truncate<T, RoundMode::CAST_FLOOR>(dstLocal, srcLocal, 512);

结 果 示 例 如 下：

Text

输 入 数 据srcLocal：[-37.063736    64.17277    -33.448692 ... ]
输 出 数 据dstLocal：[-38  64 -34 ... ]