set_device_sat_mode
函数原型
def set_device_sat_mode(mode)
功能说明
设置针对浮点计算的进程级溢出模式,当前可支持两种溢出模式:饱和模式与INF/NAN模式。
- 饱和模式:计算出现溢出时,计算结果会饱和为浮点数极值(+-MAX)。
- INF/NAN模式:遵循IEEE 754标准,根据定义输出INF/NAN的计算结果。
针对Atlas 训练系列产品,仅支持“饱和模式”。
针对Atlas A2 训练系列产品,支持两种溢出模式:饱和模式与INF/NAN模式。
使用约束
无。
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
mode |
输入 |
设置的溢出模式。
针对Atlas 训练系列产品,默认值“0”,且仅支持配置为“0”。 针对Atlas A2 训练系列产品,默认值为“1”,且推荐保持默认值“1”。 |
返回值
无
调用示例
以下示例仅针对Atlas A2 训练系列产品,其他型号处理器,开发者无需显式调用此接口。
1 2 3 4 5 6 |
import tensorflow as tf import npu_device as npu # 初始化NPU为默认设备 npu.open().as_default() # 针对Atlas A2 训练系列产品,网络执行时调用如下接口进行溢出模式的设置 npu.npu_device.set_device_sat_mode(1) |
父主题: TF Adapter 2.x