SetSaturationFlag(ISASI)

产品支持情况

• Ascend 950PR/Ascend 950DT：不支持

• Atlas A3 训练系列产品/Atlas A3 推理系列产品：支持

• Atlas A2 训练系列产品/Atlas A2 推理系列产品：支持

• Atlas 200I/500 A2 推理产品：不支持

• Atlas 推理系列产品AI Core：不支持

• Atlas 推理系列产品Vector Core：不支持

• Atlas 训练系列产品：不支持

• Kirin X90：不支持

• Kirin 9030：不支持

功能说明

头文件路径为："basic_api/kernel_operator_common_intf.h"。

设置饱和开关的标志位，以控制计算/精度转换时饱和模式的开启与关闭。

函数原型

C++

template <SaturationMode mode>
__aicore__ inline void SetSaturationFlag(bool enableSat)

参数说明

表1 模板参数说明

参数名	描述
mode	配置的对应饱和模式。 当mode配置为FLOAT时，开启饱和模式，inf输出会被饱和为±MAX，NaN输出会被饱和为0；关闭饱和模式，inf/NaN保持原输出。 当mode配置为CAST时，开启饱和模式，溢出值会被饱和为±MAX；关闭饱和模式，溢出值会按目标数据类型位数截断，保留低位，舍弃高位。 enum class SaturationMode : uint8_t { FLOAT, // 控制浮点数计算和浮点数精度转换时的饱和模式,浮点数数据类型仅支持half、bfloat16_t; CAST // 控制浮点数转整数或整数转整数时的精度转换饱和模式; };

表2 参数说明

参数名	输入/输出	描述
enableSat	输入	是否开启对应饱和模式。true表示开启饱和模式，false表示关闭饱和模式。

返回值说明

无

约束说明

无

调用示例

如下示例中开启整数转整数时精度转换的饱和模式。

C++

AscendC::SetSaturationFlag<AscendC::SaturationMode::CAST>(true);