跳到主要内容

aclrtSetStreamOverflowSwitch

aclrtDestroyStream

函数功能

销毁指定Stream,销毁通过aclrtCreateStreamaclrtCreateStreamWithConfigaclrtCreateStreamV2接口创建的Stream,若Stream上有未完成的任务,会等待任务完成后再销毁Stream。同步接口。

约束说明

  • 在调用aclrtDestroyStream接口销毁指定Stream前,需要先调用aclrtSynchronizeStream接口确保Stream中的任务都已完成。
  • 调用aclrtDestroyStream接口销毁指定Stream时,需确保该Stream在当前Context下。
  • 在调用aclrtDestroyStream接口销毁指定Stream时,需确保其它接口没有正在使用该Stream。

函数原型

aclError aclrtDestroyStream(aclrtStream stream)

参数说明

参数名输入/输出说明
stream输入待销毁的Stream。

返回值说明

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

参考资源

接口调用流程及示例,参见运行管理资源申请与释放

父主题: Stream管理

aclrtDestroyStreamForce

函数功能

销毁指定Stream,销毁通过aclrtCreateStreamaclrtCreateStreamWithConfig接口创建的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管理

在线提单