CANN/runtime回调机制示例

📅 2026/7/9 0:59:44 👁️ 阅读次数 📝 编程学习
CANN/runtime回调机制示例

0_simple_callback

【免费下载链接】runtime本项目提供CANN运行时组件和维测功能组件。项目地址: https://gitcode.com/cann/runtime

描述

本样例展示了如何为同一个 Stream 同时注册 Report 回调线程和 HostFunc 处理线程,并通过aclrtLaunchCallbackaclrtLaunchHostFunc观察两类回调在用户指定线程上的执行行为。

产品支持情况

本样例关键接口在不同产品上的支持情况如下:

接口Atlas A3 训练系列产品/Atlas A3 推理系列产品Atlas A2 训练系列产品/Atlas A2 推理系列产品
aclrtSubscribeReport
aclrtProcessReport
aclrtUnSubscribeReport
aclrtLaunchCallback
aclrtSubscribeHostFuncxx
aclrtProcessHostFuncxx
aclrtUnSubscribeHostFuncxx
aclrtLaunchHostFunc

编译运行

环境安装详情以及运行详情请见 example 目录下的 README。

运行步骤如下:

# ${install_root} 替换为 CANN 安装根目录,默认安装在`/usr/local/Ascend`目录 source ${install_root}/cann/set_env.sh export ASCEND_INSTALL_PATH=${install_root}/cann # ${ascend_name} 替换为昇腾AI处理器的型号,可通过 npu-smi info 查看 Name 字段并去掉空格获得,例如 ascend910b3 export SOC_VERSION=${ascend_name} # 部分样例中涉及调用AscendC算子,需配置AscendC编译器ascendc.cmake所在的路径,如 ${install_root}/cann/aarch64-linux/tikcpp/ascendc_kernel_cmake # 可在CANN包安装路径下查找ascendc_kernel_cmake,例如find ./ -name ascendc_kernel_cmake,并将${cmake_path}替换为ascendc_kernel_cmake所在路径 export ASCENDC_CMAKE_DIR=${cmake_path} # 编译运行 bash run.sh

CANN RUNTIME API

在该 Sample 中,涉及的关键功能点及其关键接口如下:

  • 初始化与资源管理
    • aclInit/aclFinalize
    • aclrtSetDevice/aclrtResetDeviceForce
    • aclrtCreateContext/aclrtDestroyContext
    • aclrtCreateStream/aclrtDestroyStreamForce
    • aclrtSetStreamFailureMode
  • 控制回调
    • aclrtSubscribeReport/aclrtProcessReport/aclrtUnSubscribeReport
    • aclrtLaunchCallback
    • aclrtSubscribeHostFunc/aclrtProcessHostFunc/aclrtUnSubscribeHostFunc
    • aclrtLaunchHostFunc
  • 内存与数据传输
    • aclrtMalloc/aclrtFree
    • aclrtMemcpy
    • aclrtSynchronizeStream

已知 issue

暂无。

【免费下载链接】runtime本项目提供CANN运行时组件和维测功能组件。项目地址: https://gitcode.com/cann/runtime

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考