文档
注册

函数:subscribe_report

C函数原型

aclError aclrtSubscribeReport(uint64_t threadId, aclrtStream stream)

Python函数

ret = acl.rt.subscribe_report(thread_id, stream)

函数功能

异步任务场景下,指定处理Stream上回调函数的线程。同步接口。

本接口需与以下其它接口配合使用,以便实现异步场景下的Callback功能:
  1. 新建线程,在线程函数内,调用acl.rt.process_report接口设置超时时间(需循环调用),等待acl.rt.launch_callback接口下发的函数回调任务。
  2. 调用acl.rt.subscribe_report接口建立第1步中的线程和Stream的绑定关系,该Stream下发的函数回调任务将在绑定的线程中执行。
  3. 在指定Stream上执行异步任务(例如异步推理任务)。
  4. 定义并实现回调函数,调用acl.rt.launch_callback接口在Stream的任务队列中下发函数回调任务,触发acl.rt.subscribe_report订阅的线程处理回调函数,每调用一次acl.rt.launch_callback接口,就会下发一个回调函数任务。
  5. 异步任务全部执行完成后,调用acl.rt.unsubscribe_report接口取消线程订阅,解除线程和Stream的绑定关系。

输入说明

thread_id:int,指定线程id。

stream:int,指定需要处理的Stream的指针地址。

返回值说明

ret:int,错误码。

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

约束说明

  • 支持多次调用acl.rt.subscribe_report接口给多个Stream(仅支持同一Device内的多个Stream)订阅同一个处理回调函数的线程。

  • 为确保Stream内的任务按调用顺序执行,不支持调用acl.rt.subscribe_report接口给同一个Stream订阅多个处理回调函数的线程。

  • Atlas 200/300/500 推理产品,单进程内调用acl.rt.subscribe_report接口订阅的线程数量如果超过128个,则接口返回失败。
  • Atlas 训练系列产品,单进程内调用acl.rt.subscribe_report接口订阅的线程数量如果超过1024个,则接口返回失败。
  • Atlas A2训练系列产品/Atlas 300I A2推理产品,单进程内调用acl.rt.subscribe_report接口订阅的线程数量如果超过1024个,则接口返回失败。
  • Atlas 推理系列产品(Ascend 310P处理器),单进程内调用acl.rt.subscribe_report接口订阅的线程数量如果超过1024个,则接口返回失败。
  • Atlas 200/500 A2推理产品,单进程内调用acl.rt.subscribe_report接口订阅的线程数量如果超过1024个,则接口返回失败。
  • 考虑操作系统的线程切换性能开销,建议调用acl.rt.subscribe_report接口订阅的线程数量控制在32个以下(包括32)。

  • 同一个进程内,在不同的Device上订阅回调函数的线程时,不能指定同一个线程ID。

参考资源

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词