下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

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

输入

设置的溢出模式。

  • 0:饱和模式,为默认值。
  • 1:INF/NAN模式,推荐用户使用该模式。

针对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)
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词