aclrtSetStreamFailureMode

产品支持情况

产品 是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品

功能说明

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

函数原型

aclError aclrtSetStreamFailureMode(aclrtStream stream, uint64_t mode)

参数说明

参数名 输入/输出 说明
stream 输入 待操作Stream。
不支持指定默认Stream(即该参数传入NULL)的任务调度模式。
mode 输入 当一个Stream上下发了多个任务时,可通过本参数指定任务调度模式,以便控制某个任务失败后是否继续执行下一个任务。
取值范围如下:

- ACL_CONTINUE_ON_FAILURE:默认值,某个任务失败后,继续执行下一个任务;
- ACL_STOP_ON_FAILURE:某个任务失败后,停止执行后续任务,通常称作遇错即停。触发遇错即停之后,不支持再下发新任务。

返回值说明

返回0表示成功,返回其他值表示失败,请参见aclError

约束说明

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

  • 当Stream上设置了遇错即停模式,该Stream所在的Context下的其它Stream也是遇错即停 。该约束适用于以下产品型号:

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

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