Skip to content
版 本

SetHF32Mode

产 品 支 持 情 况

产 品

是 否 支 持

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

功 能 说 明

用 于 设 置Mmad计 算 是 否 开 启HF32模 式,开 启 该 模 式 后L0A/L0B中 的FP32数 据 将 在 参 与Mmad计 算 之 前 被 舍 入 为HF32。

函 数 原 型

Text
__aicore__ inline void SetHF32Mode(HF32Mode mode)

参 数 说 明

表 1 参 数 说 明

参 数 名

输 入/输 出

描 述

mode

输 入

Mmad HF32模 式 控 制 入 参,HF32Mode枚 举 类 型。支 持 如 下 两 种 取 值:

  • ENABLE:L0A/L0B中 的FP32数 据 将 在 矩 阵 乘 法 之 前 被 舍 入 为HF32。
  • DISABLE:将 执 行 常 规 的FP32矩 阵 乘 法。

返 回 值 说 明

约 束 说 明

调 用 示 例

Text
AscendC::SetHF32Mode(HF32Mode::ENABLE); // 控 制mmad计 算 时 是 否 使 用HF32精 度 进 行 计 算。

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