配置任务遇错即停
CANN支持遇错即停模式(ACL_STOP_ON_FAILURE)和遇错继续模式(ACL_CONTINUE_ON_FAILURE),以支持不同应用对任务执行失败的差异化控制。默认为遇错继续模式。
当Stream上的任务执行失败时,如果配置了遇错即停模式(ACL_STOP_ON_FAILURE),则会停止执行该Context中所有Stream上的任务;如果配置了遇错继续模式(ACL_CONTINUE_ON_FAILURE),则会继续执行Stream上的后续任务。
调用aclrtSetStreamFailureMode接口指定调度模式的示例代码如下,不可以直接拷贝编译运行,仅供参考:
1 2 3 4 5 6 | aclrtStream stream; aclrtCreateStream(&stream); // 设置遇错即停模式 aclrtSetStreamFailureMode(stream, ACL_STOP_ON_FAILURE); ...... |
也可以调用aclrtSetStreamAttribute接口指定调度模式的示例代码如下,不可以直接拷贝编译运行,仅供参考:
1 2 3 4 5 6 | aclrtStream stream; aclrtCreateStream(&stream); // 设置遇错继续模式 aclrtSetStreamAttribute(stream, ACL_STREAM_ATTR_FAILURE_MODE, ACL_CONTINUE_ON_FAILURE); ...... |
父主题: Stream管理