函数：set_stream_overflow_switch

Atlas A3 训练系列产品/Atlas A3 推理系列产品

√

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

√

Atlas 训练系列产品

x

Atlas 推理系列产品

x

Atlas 200I/500 A2 推理产品

x

Atlas 200/300/500 推理产品

x

aclError aclrtSetStreamOverflowSwitch(aclrtStream stream, uint32_t flag)

ret = acl.rt.set_stream_overflow_switch(stream, flag)

饱和模式下，对接上层训练框架时（例如：PyTorch），针对指定Stream，打开或关闭溢出检测开关，关闭后无法通过溢出检测算子获取任务是否溢出。

stream：int，指定Stream的指针地址，若传入0，则操作默认Stream。

flag：int，溢出检测开关，取值范围如下。

• 0：关闭
• 1：打开

ret：int，错误码。

• 返回0表示成功。
• 返回其它值表示失败。

• 在调用本接口前，可调用acl.rt.set_device_sat_mode接口设置饱和模式。
• 调用该接口打开或关闭溢出检测开关后，仅对后续新下发的任务生效，已下发的任务仍维持原样。