函数:set_stream_failure_mode

C函数原型

aclError aclrtSetStreamFailureMode(aclrtStream stream, uint64_t mode)

Python函数

ret = acl.rt.set_stream_failure_mode(stream, mode)

函数功能

当一个Stream上下发了多个任务时,可通过本接口指定任务调度模式,以便控制某个任务失败后是否继续执行下一个任务。

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

输入说明

stream:int,待操作Stream的指针地址。

“stream”参数不支持指定默认Stream(即向该参数传入0)的任务调度模式。

mode:int,当一个Stream上下发了多个任务时,可通过本参数指定任务调度模式,以便控制某个任务失败后是否继续执行下一个任务。

取值范围如下:

  • ACL_CONTINUE_ON_FAILURE = 0:默认值,某个任务失败后,继续执行下一个任务。
  • ACL_STOP_ON_FAILURE = 1:某个任务失败后,停止执行后续的任务。

返回值说明

ret:int,错误码。

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

约束说明