文件最后提交记录最后更新时间
4 个月前
1 个月前
1 个月前
4 个月前
4 个月前
4 个月前
README

1_adump_callback

描述

本用例展示了单算子执行场景下如何通过回调函数接收Dump数据块。示例会注册 acldumpRegCallback 回调,在回调中解析 acldumpChunk 的文件名、偏移、分块标记、flag 和数据预览,并输出汇总统计信息。

产品支持情况

本样例支持以下产品:

产品 是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品

编译运行

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

运行步骤如下:

# ${install_root} 替换为 CANN 安装根目录,默认安装在`/usr/local/Ascend`目录
source ${install_root}/cann/set_env.sh

# 自动识别 SOC_VERSION 和 ASCENDC_CMAKE_DIR
source ${git_clone_path}/example/set_sample_env.sh

# 编译运行
bash run.sh

CANN RUNTIME API

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

  • 初始化
    • 调用aclInit接口初始化AscendCL配置。
    • 调用acldumpRegCallback接口注册Dump算子信息回调函数。
    • 调用acldumpGetPath接口查询当前Dump输出路径。
    • 调用acldumpUnregCallback接口去注册Dump算子信息回调函数。
    • 调用aclFinalize接口实现AscendCL去初始化。
  • Device管理
    • 调用aclrtSetDevice接口指定用于运算的Device。
    • 调用aclrtSynchronizeDevice接口阻塞等待正在运算中的Device完成运算。
    • 调用aclrtResetDeviceForce接口强制复位当前运算的Device,回收Device上的资源。
  • Stream管理
    • 调用aclrtCreateStream接口创建Stream。
    • 调用aclrtSynchronizeStream接口阻塞等待Stream上任务的完成。
    • 调用aclrtDestroyStream接口销毁Stream。
  • 内存管理
    • 调用aclrtMalloc接口申请Device上的内存。
    • 调用aclrtFree接口释放Device上的内存。
  • 数据传输
    • 调用aclrtMemcpy接口通过内存复制的方式实现数据传输。

本示例新增覆盖

  • 回调数据解析:在回调中解析 acldumpChunk 的文件名、偏移、是否最后一块、flag 和数据预览。
  • acldumpGetPath:查询当前Dump输出路径,便于与回调收到的文件名做对应。

示例输出

[INFO]  acldumpGetPath returned dump path: ...
[INFO]  Get workspace size...
[INFO]  Begin to add...
[INFO]  Receive dump tensor data success. file=..., bufLen=..., isLastChunk=..., offset=..., flag=..., preview=...
[INFO]  result[0] is: 1.000000
[INFO]  result[1] is: 2.000000
[INFO]  result[2] is: 3.000000
[INFO]  result[3] is: 5.000000
[INFO]  result[4] is: 6.000000
[INFO]  result[5] is: 7.000000
[INFO]  result[6] is: 10.000000
[INFO]  result[7] is: 11.000000
[INFO]  Dump callback summary: total chunks=..., total bytes=..., total files=...
[INFO]  Run the device_normal sample successfully.

已知issue

暂无