昇腾社区首页
中文
注册

函数:set_stream_failure_mode

产品支持情况

产品

是否支持

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

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

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 200/300/500 推理产品

x

功能说明

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

函数原型

  • C函数原型
    1
    aclError aclrtSetStreamFailureMode(aclrtStream stream, uint64_t mode)
    
  • python函数
    1
    ret = acl.rt.set_stream_failure_mode(stream, mode)
    

参数说明

参数名

说明

stream

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

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

mode

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

取值范围如下:

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

返回值说明

返回值

说明

ret

int,错误码,返回0表示成功,返回其它值表示失败。

约束说明

针对指定Stream只能调用一次本接口设置任务调度模式。