已合并
register_device_interface_for_triton #43624
cuiduo创建于 18 天前
register_device_interface_for_triton #43624
已合并
Pull Request已成功合入, 合并人@ascend-robot
(感谢 cuiduo 的贡献)atomgit-bot
18 天前 评论:
18 天前 评论:
变更摘要
此 PR 主要修复 has_triton() 和 is_device_compatible_with_triton() 中遗漏的设备接口注册问题,并在测试文件中清理无用的 torch_npu 导入。
主要改动
- 在
is_device_compatible_with_triton()中补充设备接口注册:在torch_npu/utils/_dynamo.py的is_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 警告。


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


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


【合入来源】
【修改方案】
【资料变更】
【接口变更】
【功能验证】
【CheckList】