set_device_sat_mode

函数原型

def set_device_sat_mode(mode)

功能说明

设置针对浮点计算的进程级溢出模式，当前可支持两种溢出模式：饱和模式与INF/NAN模式。
  • 饱和模式：计算出现溢出时，计算结果会饱和为浮点数极值（+-MAX）。
  • Inf/NaN模式：遵循IEEE 754标准，根据定义输出Inf/NaN的计算结果。

使用约束

该接口需要在运行时配置，网络训练脚本执行前调用。

参数说明

参数名

输入/输出

描述

mode

输入

设置的溢出模式。

  • 0：饱和模式，为默认值。
  • 1：INF-NAN模式，推荐用户使用该模式。

针对Atlas 训练系列产品，仅支持设置为默认值“0”。

返回值

无

调用示例

以下示例仅针对Atlas A2训练系列产品，针对Atlas 训练系列产品，开发者无需显式调用此接口。

import tensorflow as tf
from npu_bridge.estimator.npu import npu_plugin

......
# 针对Atlas A2训练系列产品，网络执行时调用如下接口进行溢出模式的设置
npu_plugin.set_device_sat_mode(1)
sess.run(xxx)
父主题： npu_bridge.estimator.npu.npu_plugin