已合并
register_device_interface_for_triton #43622
cuiduo创建于 8月3日
register_device_interface_for_triton #43622
已合并
共 2 个文件变更+1-1
| @@ -1,7 +1,6 @@ | |||
| 1 | import torch | 1 | import torch |
| 2 | from torch.testing._internal.common_utils import run_tests, parametrize, instantiate_parametrized_tests | 2 | from torch.testing._internal.common_utils import run_tests, parametrize, instantiate_parametrized_tests |
| 3 | from testutils import TestUtils | 3 | from testutils import TestUtils |
| 4 | -import torch_npu | ||
| 5 | 4 | ||
| 6 | 5 | ||
| 7 | class TestAdd(TestUtils): | 6 | class TestAdd(TestUtils): |
| @@ -516,6 +516,7 @@ def has_triton() -> bool: | |||
| 516 | } | 516 | } |
| 517 | 517 | ||
| 518 | def is_device_compatible_with_triton(): | 518 | def is_device_compatible_with_triton(): |
| 519 | + _dynamo_register_interface_for_device() | ||
| 519 | for device, extra_check in triton_supported_devices.items(): | 520 | for device, extra_check in triton_supported_devices.items(): |
| 520 | device_interface = get_interface_for_device(device) | 521 | device_interface = get_interface_for_device(device) |
| 521 | if device_interface.is_available() and extra_check(device_interface): | 522 | if device_interface.is_available() and extra_check(device_interface): |