0_simple_callback
描述
本样例展示了如何为同一个 Stream 同时注册 Report 回调线程和 HostFunc 处理线程,并通过 aclrtLaunchCallback 与 aclrtLaunchHostFunc 观察两类回调在用户指定线程上的执行行为。
产品支持情况
本样例关键接口在不同产品上的支持情况如下:
| 接口 | Atlas A3 训练系列产品/Atlas A3 推理系列产品 | Atlas A2 训练系列产品/Atlas A2 推理系列产品 |
|---|---|---|
| aclrtSubscribeReport | √ | √ |
| aclrtProcessReport | √ | √ |
| aclrtUnSubscribeReport | √ | √ |
| aclrtLaunchCallback | √ | √ |
| aclrtSubscribeHostFunc | x | x |
| aclrtProcessHostFunc | x | x |
| aclrtUnSubscribeHostFunc | x | x |
| 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/aclFinalizeaclrtSetDevice/aclrtResetDeviceForceaclrtCreateContext/aclrtDestroyContextaclrtCreateStream/aclrtDestroyStreamForceaclrtSetStreamFailureMode
- 控制回调
aclrtSubscribeReport/aclrtProcessReport/aclrtUnSubscribeReportaclrtLaunchCallbackaclrtSubscribeHostFunc/aclrtProcessHostFunc/aclrtUnSubscribeHostFuncaclrtLaunchHostFunc
- 内存与数据传输
aclrtMalloc/aclrtFreeaclrtMemcpyaclrtSynchronizeStream
已知 issue
暂无。