已合并
fix(jit): fix test for jit api: torch.jit.script、torch.jit.trace、torch.jit.save、torch.jit.load #32381
dinglaiping创建于 3月25日
fix(jit): fix test for jit api: torch.jit.script、torch.jit.trace、torch.jit.save、torch.jit.load #32381
已合并
从已删除 :fixtest-jit-api-2.7.1合入到Ascend/pytorchv2.7.1
共 1 个文件变更+3-3
| @@ -107,10 +107,10 @@ from torch.testing._internal.jit_utils import JitTestCase, enable_cpu_fuser, dis | |||
| 107 | from torch.testing._internal.jit_metaprogramming_utils import ( | 107 | from torch.testing._internal.jit_metaprogramming_utils import ( |
| 108 | get_script_args, | 108 | get_script_args, |
| 109 | create_input, unpack_variables, | 109 | create_input, unpack_variables, |
| 110 | - additional_module_tests, EXCLUDE_SCRIPT_MODULES, | 110 | + get_all_nn_module_tests, EXCLUDE_SCRIPT_MODULES, |
| 111 | get_nn_module_name_from_kwargs, get_nn_mod_test_name, script_method_template) | 111 | get_nn_module_name_from_kwargs, get_nn_mod_test_name, script_method_template) |
| 112 | 112 | ||
| 113 | -from torch.testing._internal.common_nn import module_tests, new_module_tests, criterion_tests | 113 | +from torch.testing._internal.common_nn import criterion_tests |
| 114 | 114 | ||
| 115 | # For testing truediv in python 2 | 115 | # For testing truediv in python 2 |
| 116 | from torch.testing._internal.test_module.future_div import div_int_future, div_float_future | 116 | from torch.testing._internal.test_module.future_div import div_int_future, div_float_future |
| @@ -16159,7 +16159,7 @@ class TestProducerVersion(TestCase): | |||
| 16159 | # issue gh-32561 | 16159 | # issue gh-32561 |
| 16160 | self.assertTrue(torch.__version__.startswith(torch.onnx.producer_version)) | 16160 | self.assertTrue(torch.__version__.startswith(torch.onnx.producer_version)) |
| 16161 | 16161 | ||
| 16162 | -for test in module_tests + new_module_tests + additional_module_tests: | 16162 | +for test in get_all_nn_module_tests(): |
| 16163 | add_nn_module_test(**test) | 16163 | add_nn_module_test(**test) |
| 16164 | 16164 | ||
| 16165 | for test in criterion_tests: | 16165 | for test in criterion_tests: |