已合并
register_device_interface_for_triton #43624
cuiduo创建于 18 天前
register_device_interface_for_triton #43624
已合并
cuiduo创建于 18 天前
cuiduo
cuiduo成员
18 天前

【合入来源】

如有社区issue,请关联issue链接
请勿携带内部流程信息(需求链接、问题单、内部issue等)

【修改方案】

image.png
修改上述报错,增加npuinterface注册

【资料变更】

不涉及

【接口变更】

不涉及

【功能验证】

本地验证通过

【CheckList】

PR提交人对以下CheckList自检项进行全量自检,自检通过或不涉及,均修改 [ ] 为 [x]

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 cuiduo 的贡献)
cuiduocuiduo成员
18 天前 创建了 pull request,commit db76491d
atomgit-bot
atomgit-bot
18 天前 评论:

变更摘要

此 PR 主要修复 has_triton()is_device_compatible_with_triton() 中遗漏的设备接口注册问题,并在测试文件中清理无用的 torch_npu 导入。

主要改动

  • is_device_compatible_with_triton() 中补充设备接口注册:在 torch_npu/utils/_dynamo.pyis_device_compatible_with_triton() 函数开头新增 _dynamo_register_interface_for_device() 调用,确保在遍历 triton_supported_devices 之前完成设备接口注册,避免因未注册导致 get_interface_for_device() 查找失败。
  • 移除 test_torch_mlir.py 中的冗余导入:删除 test/_inductor/test_torch_mlir.py 中未使用的 import torch_npu 语句,消除 F401 级别的 lint 警告。
likedislike
atomgit-bot
atomgit-bot
18 天前 评论:

代码审查

✅ 未发现问题

likedislike
ascend-robotascend-robot成员
18 天前 添加了label:ascend-cla/no
此处折叠了45条消息 查看更多
daifu1234567成员
17 天前 评论:

/lgtm

likedislike
ascend-robotascend-robot成员
17 天前 添加了label:lgtm
ascend-robotascend-robot成员
17 天前 关闭了关联的issue
ascend-robotascend-robot成员
17 天前 合入了pull request
ascend-robot
ascend-robot成员
17 天前 评论:
流水线 pytorch_gitcode_PR_multiVersion#13591 [ commitID:2b96bbaf ] 已完成
likedislike