已合并
register_device_interface_for_triton #43622
cuiduo创建于 8月3日
register_device_interface_for_triton #43622
已合并
Pull Request已成功合入, 合并人@ascend-robot
(感谢 cuiduo 的贡献)atomgit-bot
8月3日 评论:
8月3日 评论:
变更摘要
此 PR 主要修复了 torch_npu 中 Triton 设备兼容性检查的注册流程问题,并清理了测试文件中的未使用导入。核心改动是在 is_device_compatible_with_triton() 函数中增加了对 _dynamo_register_interface_for_device() 的调用,确保在检查设备是否兼容 Triton 之前完成 device interface 的注册;同时移除了 test_torch_mlir.py 中多余的 import torch_npu 语句。
主要改动
- 在
is_device_compatible_with_triton()中增加设备接口注册调用:于torch_npu/utils/_dynamo.py的is_device_compatible_with_triton函数内部新增_dynamo_register_interface_for_device()调用,确保在执行设备兼容性检查前已正确注册对应的 device interface。 - 移除测试文件中的未使用导入:删除了
test/_inductor/test_torch_mlir.py中的import torch_npu语句,以消除 flake8 F401(未使用导入)警告。


ascend-robot
8月3日 评论:
8月3日 评论:
atomgit-bot
8月3日 评论:
8月3日 评论:
8月3日 添加了label:ascend-cla/no
此处折叠了46条消息 查看更多
29 天前 添加了label:approved
daifu1234567
29 天前 评论:
29 天前 评论:
/lgtm


29 天前 添加了label:lgtm
29 天前 合入了pull request
ascend-robot
29 天前 评论:
29 天前 评论:
流水线 pytorch_gitcode_PR_multiVersion#13589 [ commitID:9e13bc06 ] 已完成


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