INF_NAN_MODE_ENABLE

功能描述

用于控制芯片对输入数据为Inf/NaN的处理能力，即控制芯片使用饱和模式还是INF_NAN模式。

默认值为0。配置为0时为饱和模式，配置为1时为INF_NAN模式。

饱和模式：Inf置为max，NaN置为0。

torch.exp(torch.tensor([12], dtype=torch.float16).npu()) 
# tensor([65504.], device='npu:0', dtype=torch.float16)

INF_NAN模式：IEEE 754标准模式，与GPU行为一致。

torch.exp(torch.tensor([12], dtype=torch.float16).npu()) 
# tensor([inf], device='npu:0', dtype=torch.float16)

配置示例

export INF_NAN_MODE_ENABLE=1