【TorchAir】遇到ERR03001 GRAPH invalid parameter问题怎么解决？

原创

已于 2025-08-27 15:43:15 修改 · 348 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #TorchAir #torch_npu #graph

于 2025-08-27 15:25:15 首次发布

背景

在测试昇腾（Ascend）TorchAir的图结构dump功能的时候，通过其举例的demo，编写脚本，遇到ERR03001 GRAPH invalid parameter。
昇腾社区文档参考如下：https://www.hiascend.com/document/detail/zh/Pytorch/710/modthirdparty/torchairuseguide/torchair_00012.html

完整代码：

# 导包（必须先导torch_npu再导torchair）
import torch
import torch_npu
import torchair

# Patch方式实现集合通信入图（可选）
from torchair import patch_for_hcom
patch_for_hcom()

# 自定义Model
class Model(torch.nn.Module):
    def __init__(self):
        super().__init__()
    def forward(self, x, y):
        return torch.add(x, y)
model = Model().npu()

# 配置图模式config
config = torchair.CompilerConfig()
config.debug.graph_dump.type