set_device_sat_mode

函数原型

def set_device_sat_mode(mode)

功能说明

设置针对浮点计算的进程级溢出模式。

• 饱和模式：计算出现溢出时，计算结果会饱和为浮点数极值（+-MAX）。
• INF/NAN模式：遵循IEEE 754标准，根据定义输出INF/NAN的计算结果。

针对Atlas 训练系列产品，仅支持“饱和模式”。

针对Atlas A2 训练系列产品，支持两种溢出模式：饱和模式与INF/NAN模式。

使用约束

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

参数说明

返回值

无

调用示例

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

1
2
3
4
5
6
7

import tensorflow as tf
from npu_bridge.npu_init import *

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