aclrtSetStreamOverflowSwitch

函数功能

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

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

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

Atlas 推理系列产品(Ascend 310P处理器),不支持该接口。

约束说明

函数原型

aclError aclrtSetStreamOverflowSwitch(aclrtStream stream, uint32_t flag)

参数说明

参数名

输入/输出

说明

stream

输入

待操作Stream,若传入NULL,则操作默认Stream。

flag

输入

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

  • 0:关闭
  • 1:打开

返回值说明

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