已合并
API Conflict Fix #30595
cuiduo创建于 2月6日
API Conflict Fix #30595
已合并
共 2 个文件变更+5-7
| @@ -6,7 +6,6 @@ from torch.testing._internal.common_utils import ( | |||
| 6 | instantiate_parametrized_tests, | 6 | instantiate_parametrized_tests, |
| 7 | ) | 7 | ) |
| 8 | from testutils import TestUtils | 8 | from testutils import TestUtils |
| 9 | -import torch_npu | ||
| 10 | 9 | ||
| 11 | 10 | ||
| 12 | class TestEvent(TestUtils): | 11 | class TestEvent(TestUtils): |
| @@ -39,6 +39,11 @@ torch_non_c_binding_in_graph_functions_npu = dict.fromkeys( | |||
| 39 | "torch.npu.random.seed", | 39 | "torch.npu.random.seed", |
| 40 | "torch.npu.set_sync_debug_mode", | 40 | "torch.npu.set_sync_debug_mode", |
| 41 | "torch.npu._set_rng_state_offset", | 41 | "torch.npu._set_rng_state_offset", |
| 42 | + "torch.npu._get_generator", | ||
| 43 | + "torch.npu._memory_viz._frames_fmt", | ||
| 44 | + "torch.npu._memory_viz._frame_fmt", | ||
| 45 | + "torch.npu.amp.autocast_mode.custom_bwd", | ||
| 46 | + "torch.npu.amp.autocast_mode.custom_fwd", | ||
| 42 | ], | 47 | ], |
| 43 | TorchInGraphFunctionVariable, | 48 | TorchInGraphFunctionVariable, |
| 44 | ) | 49 | ) |
| @@ -68,12 +73,6 @@ torch_c_binding_in_graph_functions_npu = dict.fromkeys( | |||
| 68 | skip_functions_npu = dict.fromkeys( | 73 | skip_functions_npu = dict.fromkeys( |
| 69 | [ | 74 | [ |
| 70 | "torch.npu.set_device", | 75 | "torch.npu.set_device", |
| 71 | - "torch.npu._get_generator", | ||
| 72 | - "torch.npu._memory_viz._frames_fmt", | ||
| 73 | - "torch.npu._memory_viz._frame_fmt", | ||
| 74 | - "torch.npu.amp.autocast_mode.custom_bwd", | ||
| 75 | - "torch.npu.amp.autocast_mode.custom_fwd", | ||
| 76 | - | ||
| 77 | ], | 76 | ], |
| 78 | SkipFunctionVariable | 77 | SkipFunctionVariable |
| 79 | ) | 78 | ) |