[object Object][object Object]

aclgraph_dump 提供 [object Object] 接口，适用于aclgraph场景，可以将张量保存为 [object Object] 文件。

[object Object]

环境准备

1. 安装 msProbe 工具，详见《》。

2. 源码编译安装时需包含 [object Object] 模块：

   [object Object]

3. 安装并正确配置 Ascend Extension for PyTorch（torch_npu）和 CANN （同msProbe安装要求）环境。

约束

• 仅支持 PyTorch 框架。
• 构建 [object Object] 需要 torch_npu 参与编译；若未包含该模块，将无法导入 [object Object]。

[object Object]

下面示例展示如何在前向过程中保存某个张量：

[object Object][object Object]

[object Object] 用于保存张量数据，调用后会保存[object Object]文件。

[object Object]

函数原型

参数说明

[object Object]undefined

返回值

返回一个与输入形状一致的张量。无实际意义，仅用于触发保存操作。

[object Object][object Object][object Object]

会生成：[object Object]、[object Object]、[object Object]。

[object Object][object Object]

调用 [object Object] 后，会在 [object Object] 指定目录下生成 [object Object] 文件。

[object Object]

保存格式为 PyTorch [object Object] 文件（pickle 序列化），可通过 [object Object] 读取：

[object Object][object Object]

1. 导入报错：Failed to import msprobe.lib.aclgraph_dump_ext

请确认：

• 编译安装时已包含 [object Object]；
• 已安装 torch_npu 且环境变量配置正确；
• 当前系统为 Linux。

2. [object Object]问题

CANN 8.5 以下（不含8.5版本）可能会出现[object Object]问题，这是由于老版本存在SQ不复用的问题，建议可以将ccsrc/aclgraph_dump/aclgraph_dump.cpp中的[object Object]改为[object Object]规避，或者升级至 CANN 8.5.0+ 版本。