已合并
Fix test_pytorch_onnx_no_runtime #22404
dilililiwhy创建于 2025年6月28日
Fix test_pytorch_onnx_no_runtime #22404
已合并
从refs/pull/22404/head合入到master
共 2 个文件变更+2-3
| @@ -380,7 +380,7 @@ class TestONNXExport(pytorch_test_common.ExportTestCase): | |||
| 380 | f = io.BytesIO() | 380 | f = io.BytesIO() |
| 381 | torch.onnx.export(mod, (x_in,), f) | 381 | torch.onnx.export(mod, (x_in,), f) |
| 382 | 382 | ||
| 383 | - with self.assertRaisesRegex(RuntimeError, r"DictConstruct.+is not supported."): | 383 | + with self.assertRaisesRegex(RuntimeError, r"DictConstruct.+is not supported"): |
| 384 | f = io.BytesIO() | 384 | f = io.BytesIO() |
| 385 | torch.onnx.export(torch.jit.script(mod), (x_in,), f) | 385 | torch.onnx.export(torch.jit.script(mod), (x_in,), f) |
| 386 | 386 | ||
| @@ -31607,6 +31607,5 @@ | |||
| 31607 | "test_fake_autocast_fft_fftshift_npu_float32 (__main__.TestFakeTensorPRIVATEUSE1)": ["", [""]], | 31607 | "test_fake_autocast_fft_fftshift_npu_float32 (__main__.TestFakeTensorPRIVATEUSE1)": ["", [""]], |
| 31608 | "test_fake_autocast_fft_ifftshift_npu_float32 (__main__.TestFakeTensorPRIVATEUSE1)": ["", [""]], | 31608 | "test_fake_autocast_fft_ifftshift_npu_float32 (__main__.TestFakeTensorPRIVATEUSE1)": ["", [""]], |
| 31609 | "test_replicate (__main__.TestDataParallel)": ["", [""]], | 31609 | "test_replicate (__main__.TestDataParallel)": ["", [""]], |
| 31610 | - "test_replicate_buffers (__main__.TestDataParallel)": ["", [""]], | 31610 | + "test_replicate_buffers (__main__.TestDataParallel)": ["", [""]] |
| 31611 | - "test_export_dict (__main__.TestONNXExport)": ["", [""]] | ||
| 31612 | } | 31611 | } |