函数:set_stream_overflow_switch

C函数原型

aclError aclrtSetStreamOverflowSwitch(aclrtStream stream, uint32_t flag)

Python函数

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

函数功能

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

Atlas 200/300/500 推理产品,不支持该接口。

Atlas 训练系列产品,不支持该接口。

Atlas 推理系列产品,不支持该接口。

输入说明

stream:int,操作Stream,若传入0,则操作默认Stream。

flag:int,溢出检测开关,取值范围如下:

  • 0:关闭
  • 1:打开

返回值说明

ret:int,错误码。

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

约束说明

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