函数：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 800I A2推理产品上，单进程内调用acl.rt.subscribe_report接口订阅的线程数量如果超过1024个，则接口返回失败。
  • 在Atlas 推理系列产品上，单进程内调用acl.rt.subscribe_report接口订阅的线程数量如果超过1024个，则接口返回失败。
  • 在Atlas 200I/500 A2推理产品上，单进程内调用acl.rt.subscribe_report接口订阅的线程数量如果超过1024个，则接口返回失败。

  • 考虑操作系统的线程切换性能开销，建议调用acl.rt.subscribe_report接口订阅的线程数量控制在32个以下（包括32）。

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

参考资源
父主题： 执行控制