aclrtSetStreamOverflowSwitch
aclrtDestroyStream
函数功能
销毁指定Stream,销毁通过aclrtCreateStream或aclrtCreateStreamWithConfig或aclrtCreateStreamV2接口创建的Stream,若Stream上有未完成的任务,会等待任务完成后再销毁Stream。同步接口。
约束说明
- 在调用aclrtDestroyStream接口销毁指定Stream前,需要先调用aclrtSynchronizeStream接口确保Stream中的任务都已完成。
- 调用aclrtDestroyStream接口销毁指定Stream时,需确保该Stream在当前Context下。
- 在调用aclrtDestroyStream接口销毁指定Stream时,需确保其它接口没有正在使用该Stream。
函数原型
aclError aclrtDestroyStream(aclrtStream stream)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| stream | 输入 | 待销毁的Stream。 |
返回值说明
返回0表示成功,返回其它值表示失败。
参考资源
接口调用流程及示例,参见运行管理资源申请与释放。
父主题: Stream管理
aclrtDestroyStreamForce
函数功能
销毁指定Stream,销毁通过aclrtCreateStream或aclrtCreateStreamWithConfig接口创建的Stream,若Stream上有未完成的任务,不会等待任务完成,直接强制销毁Stream。同步接口。
约束说明
调用本接口销毁指定Stream时,需确保该Stream在当前Context下。
函数原型
aclError aclrtDestroyStreamForce(aclrtStream stream)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| stream | 输入 | 待销毁的Stream。 |
返回值说明
返回0表示成功,返回其它值表示失败。
父主题: Stream管理
aclrtSetStreamOverflowSwitch
函数功能
饱和模式下,对接上层训练框架时(例如PyTorch),针对指定Stream,打开或关闭溢出检测开关,关闭后无法通过溢出检测算子获取任务是否溢出,同步接口。
Atlas 200/500 A2推理产品,不支持该接口。
约束说明
- 在调用本接口前,可调用aclrtSetDeviceSatMode接口设置饱和模式。
- 调用该接口打开或关闭溢出检测开关后,仅对后续新下的任务生效,已下发的任务仍维持原样。
函数原型
aclError aclrtSetStreamOverflowSwitch(aclrtStream stream, uint32_t flag)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| stream | 输入 | 待操作Stream,若传入NULL,则操作默认Stream。 |
| flag | 输入 | 溢出检测开关,取值范围如下: - 0:关闭 - 1:打开 |
返回值说明
返回0表示成功,返回其它值表示失败。
父主题: Stream管理
aclrtGetStreamOverflowSwitch
函数功能
针对指定Stream,获取其当前溢出检测开关是否打开。
Atlas 200/500 A2推理产品,不支持该接口。
函数原型
aclError aclrtGetStreamOverflowSwitch(aclrtStream stream, uint32_t *flag)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| stream | 输入 | 待操作Stream,若传入NULL,则操作默认Stream。 |
| flag | 输出 | 溢出检测开关,取值范围如下: - 0:关闭 - 1:打开 |
返回值说明
返回0表示成功,返回其它值表示失败。
父主题: Stream管理
aclrtSetStreamFailureMode
函数功能
当一个Stream上下发了多个任务时,可通过本接口指定任务调度模式,以便控制某个任务失败后是否继续执行下一个任务。同步接口。
函数原型
aclError aclrtSetStreamFailureMode(****aclrtStream stream, uint64_t mode)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| stream | 输入 | 待操作Stream。 不支持指定默认Stream(即该参数传入NULL)的任务调度模式。 |
| mode | 输入 | 当一个Stream上下发了多个任务时,可通过本参数指定任务调度模式,以便控制某个任务失败后是否继续执行下一个任务。 取值范围如下: - ACL_CONTINUE_ON_FAILURE:默认值,某个任务失败后,继续执行下一个任务; - ACL_STOP_ON_FAILURE:某个任务失败后,停止执行后续的任务。 |
返回值说明
返回0表示成功,返回其它值表示失败。
父主题: Stream管理
aclrtStreamQuery
函数功能
查询指定Stream上的所有任务是否已经执行完成。同步接口。
函数原型
aclError aclrtStreamQuery(****aclrtStream stream, aclrtStreamStatus *status)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| stream | 输入 | Stream的指针。 |
| status | 输出 | Stream上的任务状态。 |
返回值说明
返回0表示成功,返回其它值表示失败。
父主题: Stream管理
在线提单