已关闭
[Bug] aclgraph_dump/_meta.py register_fake 算子命名格式错误(ns.op → 应为 ns::op),fake 实现注册静默失败 #580
Menba创建于  9月4日关闭于  9月15日
Menba
Menba成员
9月4日 创建

问题描述

python/msprobe/pytorch/aclgraph_dump/_meta.py 中使用 torch.library.register_fake 为 aclgraph dump 自定义算子注册 fake(meta) 实现时,算子名称使用了点号分隔:

@torch.library.register_fake("my_ns.acl_save")
@torch.library.register_fake("my_ns.acl_tensor_save")
@torch.library.register_fake("my_ns.acl_stat")

而 torch.library.register_fake 要求算子名称使用 ns::op 双冒号格式。点号格式会导致注册时报错,且该异常被外层 except Exception: pass 静默吞掉,导致三个算子的 fake 实现实际上从未注册成功,meta tensor / FakeTensorMode 追踪场景下行为不符合预期。

复现方式

  1. 构造包含 acl_save / acl_stat 调用的计算图,使用 meta tensor 或 FakeTensorMode 追踪;
  2. 观察 fake 实现未被调用(注册失败被静默忽略)。

修复建议

将三个注册调用的算子名称从 my_ns.xxx 改为 my_ns::xxx。

likedislike
捡到宝。
捡到宝。成员
9月4日 评论:

👋 您好,欢迎向 MindStudio Probe 提交 Issue!
我们已收到您的反馈,感谢你对开源社区的支持。🎉

📅处理时效: 维护团队将在24小时内 查看并回复您的问题(工作日)。
🔍自助查询: 在等待期间,建议您先查阅以下资料,可能已有解决方案:

📖 MindStudio Probe官方文档
📝 贡献者指南

请确保 Issue 描述清晰,包含复现步骤和日志,这将帮助我们更快定位问题。谢谢!

likedislike
捡到宝。捡到宝。成员
9月4日 添加了label:triage-review
ascend-robotascend-robot成员
9月4日 添加了label:bug
MenbaMenba成员
9月4日 关联了pull request:fix: 修正 aclgraph_dump/_meta.py 中 register_fake 算子命名格式(ns.op → ns::op)
Menba
Menba成员
9月4日 评论:

/label add resolved

likedislike
ascend-robotascend-robot成员
9月4日 添加了label:resolved
ascend-robot
ascend-robot成员
9月11日 评论:

您好,当前Issue标记为resolved且有一段时间未进一步更新,因此我们将其标记为'stale'(闲置)状态。若您认为这是误操作,可通过添加任意评论来去除'stale'标签。标记为stale的Issue在4天内无更新活动将自动关闭。

likedislike
ascend-robotascend-robot成员
9月11日 添加了label:stale
ascend-robotascend-robot成员
9月15日 关闭了 issue
ascend-robotascend-robot成员
9月15日 issue状态由 TODO 改变为 DONE