已合并
fix transfer_to_npu has_triton patch bug #33753
kkjocker创建于 4月15日
fix transfer_to_npu has_triton patch bug #33753
已合并
共 2 个文件变更+3-0
| @@ -522,6 +522,7 @@ class TestPublicBindings(TestCase): | |||
| 522 | "torch.utils.tensorboard._pytorch_graph", | 522 | "torch.utils.tensorboard._pytorch_graph", |
| 523 | "torch.utils.tensorboard._utils", | 523 | "torch.utils.tensorboard._utils", |
| 524 | "torch.distributed._tools.sac_ilp", | 524 | "torch.distributed._tools.sac_ilp", |
| 525 | + "torch_npu.contrib.transfer_to_npu", | ||
| 525 | "torch_npu.dynamo.torchair._tf_concrete_graph.fx2tf_converter", | 526 | "torch_npu.dynamo.torchair._tf_concrete_graph.fx2tf_converter", |
| 526 | "torch_npu.dynamo.torchair.core._abi_compat_ge_apis", | 527 | "torch_npu.dynamo.torchair.core._abi_compat_ge_apis", |
| 527 | "torch_npu.dynamo.torchair.core._backend", | 528 | "torch_npu.dynamo.torchair.core._backend", |
| @@ -492,6 +492,8 @@ def _init(): | |||
| 492 | _do_wrapper_libraries_func(_load_json_file(config_path)) | 492 | _do_wrapper_libraries_func(_load_json_file(config_path)) |
| 493 | 493 | ||
| 494 | setattr(torch.utils._triton, 'has_triton', _patch_has_triton) | 494 | setattr(torch.utils._triton, 'has_triton', _patch_has_triton) |
| 495 | + setattr(torch._dynamo.utils, 'has_triton', _patch_has_triton) | ||
| 496 | + setattr(torch._inductor.runtime.autotune_cache, 'has_triton', _patch_has_triton) | ||
| 495 | 497 | ||
| 496 | setattr(torch._utils, '_get_available_device_type', _patch_get_available_device_type) | 498 | setattr(torch._utils, '_get_available_device_type', _patch_get_available_device_type) |
| 497 | setattr(torch.distributed.checkpoint.filesystem._OverlappingCpuLoader, '__init__', | 499 | setattr(torch.distributed.checkpoint.filesystem._OverlappingCpuLoader, '__init__', |