OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00
Code Issues Actions 52 Packages Projects Releases Wiki Activity
Files
661fb534494e88da84d025ee3da2ca362b81bfcd
pytorch/test
History
Abhishek Nandy 4e1b772103 Fix: Improve fallback behavior in deserialize_torch_artifact and relocate test into TestSaveLoad (#158247) 
This is a follow-up to [#154333](https://github.com/pytorch/pytorch/pull/154333), where I initially introduced a fallback mechanism in deserialize_torch_artifact.

In this revised PR:

Cleaned up commit history for clarity and reproducibility.

Relocated the test into the TestSaveLoad class in test_serialize.py.

There were some issues with last PR so opened this PR

The previous PR had inconsistencies due to local branch issues and was closed in favor of this cleaner submission.

Feedback is very welcome
Pull Request resolved: https://github.com/pytorch/pytorch/pull/158247
Approved by: https://github.com/angelayi
2025-11-17 21:14:37 +00:00
..
ao/sparsity
[5/N] Use key in dict for existence checks (#167311)
2025-11-08 01:43:12 +00:00
autograd
…
backends/xeon
Add missing super().setUp() (#167163)
2025-11-06 17:55:23 +00:00
benchmark_utils
…
compiled_autograd_skips
…
cpp
Move isQIntType to headeronly (#167772)
2025-11-17 18:05:36 +00:00
cpp_api_parity
…
cpp_extensions
[STABLE ABI] Add mutable_data_ptr() and const_data_ptr() methods to torch::stable::Tensor. (#161891)
2025-11-17 18:05:36 +00:00
custom_backend
[2/N] Use context managers (#167404)
2025-11-09 13:38:14 +00:00
custom_operator
Fix longstanding race condition around getAllOperatorsFor (#167860)
2025-11-17 17:37:02 +00:00
distributed
Revert "[ROCm][CI] Upgrade ROCm CI to 7.1 (#166743)"
2025-11-17 18:25:59 +00:00
distributions
Enable PLW0127 in ruff (#165851)
2025-10-21 03:30:57 +00:00
dynamo
deprecate check_is_size and guard_size_oblivious (#167198)
2025-11-17 05:47:40 +00:00
dynamo_expected_failures
Cpython test_math.FMATests (#167217)
2025-11-11 19:26:18 +00:00
dynamo_skips
…
error_messages
…
expect
Reland "Add model code stack trace to torch.profile (#166677)" (#167110)
2025-11-06 01:14:27 +00:00
export
Fix: Improve fallback behavior in deserialize_torch_artifact and relocate test into TestSaveLoad (#158247)
2025-11-17 21:14:37 +00:00
forward_backward_compatibility
…
functorch
Revert "Ops convolution_backward optional flag bug (#165008)"
2025-11-15 01:50:09 +00:00
fx
Add missing super().setUp() (#167163)
2025-11-06 17:55:23 +00:00
higher_order_ops
[HOP][print] Add functionalization (make sure ordering) for print (#167016)
2025-11-15 05:06:05 +00:00
inductor
Support SymInt placeholder in wrapper fxir (#167757)
2025-11-17 21:10:55 +00:00
inductor_expected_failures
…
inductor_skips
…
jit
[5/N] Use key in dict for existence checks (#167311)
2025-11-08 01:43:12 +00:00
jit_hooks
…
lazy
Revert "Back out "Do not decompose in functionalization/proxy tensor if autograd wouldn't have decomposed (#164939)" (#165910)" (#166812)
2025-11-03 15:06:11 +00:00
mobile
[2/N][Fix] Fix typo in test folder (#166374)
2025-10-29 03:02:07 +00:00
nn
[CUDA][Test] Add serialTest() to some largeTensorTest tests (#167471)
2025-11-14 17:13:14 +00:00
onnx
[5/N] Use key in dict for existence checks (#167311)
2025-11-08 01:43:12 +00:00
optim
[4/N] Remove unused loop variables in tests (#166690)
2025-10-31 10:20:48 +00:00
package
[2/N][Fix] Fix typo in test folder (#166374)
2025-10-29 03:02:07 +00:00
profiler
[xpu][test] Enable profiler test for XPU (#165423)
2025-11-10 04:02:59 +00:00
quantization
[5/N] Use key in dict for existence checks (#167311)
2025-11-08 01:43:12 +00:00
scripts
…
strobelight/examples
…
test_img
…
torch_np
[dynamo] [3.14] Update broken numpy test (#167681)
2025-11-13 04:27:55 +00:00
typing
[13/N] Apply ruff UP035 rule (#167048)
2025-11-09 01:47:38 +00:00
xpu
[xpu][fix] Fix conv1d precision error (#162944)
2025-11-14 01:12:21 +00:00
_test_bazel.py
…
allowlist_for_publicAPI.json
…
bench_mps_ops.py
…
conftest.py
[CI] manually gen json for segfaults (#167250)
2025-11-12 19:16:12 +00:00
create_dummy_torchscript_model.py
…
HowToWriteTestsUsingFileCheck.md
…
linear.py
…
load_torchscript_model.py
…
minioptest_failures_dict.json
…
mkl_verbose.py
…
mkldnn_verbose.py
…
pytest_shard_custom.py
…
run_doctests.sh
…
run_test.py
Add new CI jobs to run dynamo tests on all python versions supported (#166978)
2025-11-15 00:15:36 +00:00
simulate_nccl_errors.py
…
slow_tests.json
Update slow tests (#166844)
2025-11-10 12:39:27 +00:00
test_accelerator.py
[xpu][feature] Add XPU support on torch.accelerator.get_memory_info (#162564)
2025-11-13 11:03:17 +00:00
test_ao_sparsity.py
…
test_appending_byte_serializer.py
…
test_as_strided.py
Add some code for exploring the space of accessible size/stride configs via plain views (#167076)
2025-11-05 18:25:19 +00:00
test_autocast.py
[AMP][Refactor] Autocast dtype handling to simplify device-specific c… (#165221)
2025-10-28 06:21:29 +00:00
test_autograd_fallback.py
Make PT2 compile backprop through custom op without autograd key a hard error (#166367)
2025-11-11 03:16:30 +00:00
test_autograd.py
Rework PyObject preservation (v2) (#167564)
2025-11-17 14:52:02 +00:00
test_autoload.py
…
test_binary_ufuncs.py
Logaddexp complex inconsistent bw cpu and cuda (#163509)
2025-11-17 20:30:51 +00:00
test_bundled_images.py
…
test_bundled_inputs.py
[4/N] Use key in dict for existence checks (#167285)
2025-11-07 09:47:17 +00:00
test_ci_sanity_check_fail.py
…
test_comparison_utils.py
…
test_compile_benchmark_util.py
…
test_complex.py
…
test_content_store.py
…
test_cpp_api_parity.py
…
test_cpp_extensions_aot.py
…
test_cpp_extensions_jit.py
Fix command injection vulnerability in PCH compilation (#167502)
2025-11-11 07:27:41 +00:00
test_cpp_extensions_mtia_backend.py
…
test_cpp_extensions_stream_and_event.py
…
test_cuda_expandable_segments.py
…
test_cuda_multigpu.py
…
test_cuda_nvml_based_avail.py
…
test_cuda_primary_ctx.py
Add missing super().setUp() (#167163)
2025-11-06 17:55:23 +00:00
test_cuda_sanitizer.py
Add missing super().setUp() (#167163)
2025-11-06 17:55:23 +00:00
test_cuda_trace.py
Add missing super().setUp() (#167163)
2025-11-06 17:55:23 +00:00
test_cuda.py
Fix flaky memory profiler test [2] (#167268)
2025-11-10 19:51:38 +00:00
test_custom_ops.py
[4/N] Use key in dict for existence checks (#167285)
2025-11-07 09:47:17 +00:00
test_dataloader.py
Fix dataloader tests failing on python 3.14 (#167429)
2025-11-17 18:10:26 +00:00
test_datapipe.py
[8/N] Fix unused loop variables in tests (#166921)
2025-11-06 12:20:00 +00:00
test_decomp.py
address DDE in matmul decomp (#166541)
2025-11-13 23:50:00 +00:00
test_determination.py
…
test_dispatch.py
…
test_dlpack.py
[Fix] Completely remove stride normalization on DLPack Tensor (#164161)
2025-10-14 17:17:11 +00:00
test_dynamic_shapes.py
deprecate check_is_size and guard_size_oblivious (#167198)
2025-11-17 05:47:40 +00:00
test_expanded_weights.py
…
test_extension_utils.py
[user-streams] Track symbolic current stream (#165212)
2025-11-02 11:55:22 +00:00
test_fake_tensor.py
[4/N] Use key in dict for existence checks (#167285)
2025-11-07 09:47:17 +00:00
test_file_check.py
…
test_flop_counter.py
[5/N] Remove unused loop variables in tests (#166716)
2025-10-31 20:47:57 +00:00
test_foreach.py
[3/N] Fix unused loop variables (#166509)
2025-10-30 20:13:51 +00:00
test_function_schema.py
…
test_functional_autograd_benchmark.py
…
test_functional_optim.py
…
test_functionalization_of_rng_ops.py
…
test_functionalization.py
…
test_futures.py
…
test_fx_experimental.py
[6/N] Remove unused loop variables in tests (#166785)
2025-11-03 03:52:52 +00:00
test_fx_passes.py
…
test_fx_reinplace_pass.py
…
test_fx.py
Move enrich_profiler_metadata config import out of gm.recompile() (#167114)
2025-11-07 16:12:47 +00:00
test_hop_infra.py
…
test_hub.py
…
test_import_stats.py
…
test_indexing.py
[xpu][test]port embedding indexing and native_mha test files for Intel GPU (#165886)
2025-11-13 08:17:23 +00:00
test_itt.py
Enable PLW0127 in ruff (#165851)
2025-10-21 03:30:57 +00:00
test_jit_autocast.py
[3.14, jit] skip jit tests on 3.14+, add jit deprecation warnings to user-facing API (#167669)
2025-11-14 01:00:43 +00:00
test_jit_disabled.py
[3.14, jit] skip jit tests on 3.14+, add jit deprecation warnings to user-facing API (#167669)
2025-11-14 01:00:43 +00:00
test_jit_fuser_legacy.py
[3.14, jit] skip jit tests on 3.14+, add jit deprecation warnings to user-facing API (#167669)
2025-11-14 01:00:43 +00:00
test_jit_fuser_te.py
[3.14, jit] skip jit tests on 3.14+, add jit deprecation warnings to user-facing API (#167669)
2025-11-14 01:00:43 +00:00
test_jit_fuser.py
[3.14, jit] skip jit tests on 3.14+, add jit deprecation warnings to user-facing API (#167669)
2025-11-14 01:00:43 +00:00
test_jit_legacy.py
[3.14, jit] skip jit tests on 3.14+, add jit deprecation warnings to user-facing API (#167669)
2025-11-14 01:00:43 +00:00
test_jit_llga_fuser.py
[3.14, jit] skip jit tests on 3.14+, add jit deprecation warnings to user-facing API (#167669)
2025-11-14 01:00:43 +00:00
test_jit_profiling.py
[3.14, jit] skip jit tests on 3.14+, add jit deprecation warnings to user-facing API (#167669)
2025-11-14 01:00:43 +00:00
test_jit_simple.py
[3.14, jit] skip jit tests on 3.14+, add jit deprecation warnings to user-facing API (#167669)
2025-11-14 01:00:43 +00:00
test_jit_string.py
[3.14, jit] skip jit tests on 3.14+, add jit deprecation warnings to user-facing API (#167669)
2025-11-14 01:00:43 +00:00
test_jit.py
[3.14, jit] skip jit tests on 3.14+, add jit deprecation warnings to user-facing API (#167669)
2025-11-14 01:00:43 +00:00
test_jiterator.py
[3/N] Fix unused loop variables (#166509)
2025-10-30 20:13:51 +00:00
test_kernel_launch_checks.py
…
test_legacy_vmap.py
…
test_license.py
…
test_linalg.py
Logaddexp complex inconsistent bw cpu and cuda (#163509)
2025-11-17 20:30:51 +00:00
test_logging.py
…
test_masked.py
…
test_maskedtensor.py
…
test_matmul_cuda.py
fixes a few issues with out_dtype overload for addmm/baddbmm (#167931)
2025-11-17 20:50:30 +00:00
test_meta.py
Revert "Ops convolution_backward optional flag bug (#165008)"
2025-11-15 01:50:09 +00:00
test_metal.py
…
test_mkl_verbose.py
…
test_mkldnn_fusion.py
…
test_mkldnn_verbose.py
…
test_mkldnn.py
[2/N][Fix] Fix typo in test folder (#166374)
2025-10-29 03:02:07 +00:00
test_mobile_optimizer.py
…
test_model_exports_to_core_aten.py
…
test_module_tracker.py
…
test_modules.py
[2/N][Fix] Fix typo in test folder (#166374)
2025-10-29 03:02:07 +00:00
test_monitor.py
Add missing super().setUp() (#167163)
2025-11-06 17:55:23 +00:00
test_mps.py
[MPS] Fix repeat_interleave with slices (#167961)
2025-11-17 17:10:59 +00:00
test_multiprocessing_spawn.py
Change forkserver test to only run below 3.13.8 (#165667)
2025-10-16 19:34:10 +00:00
test_multiprocessing.py
[3/N] Fix unused loop variables (#166509)
2025-10-30 20:13:51 +00:00
test_namedtensor.py
Add missing device to namedtensor tests (#166717)
2025-10-31 20:04:41 +00:00
test_namedtuple_return_api.py
…
test_native_functions.py
…
test_native_mha.py
[xpu][test]port embedding indexing and native_mha test files for Intel GPU (#165886)
2025-11-13 08:17:23 +00:00
test_nestedtensor.py
[CI] Move the periodic debug tests to newer runner (#165158)
2025-10-21 21:28:12 +00:00
test_nn.py
[CUDA] Large max pool fix (#167427)
2025-11-13 20:11:41 +00:00
test_nnapi.py
Add missing super().setUp() (#167163)
2025-11-06 17:55:23 +00:00
test_numa_binding.py
Replace NUMA inheritance approach (#166026)
2025-10-29 03:58:44 +00:00
test_numba_integration.py
…
test_numpy_interop.py
[dynamo] [3.14] Support np._CopyMode (#167619)
2025-11-13 04:27:55 +00:00
test_opaque_obj_v2.py
deprecate check_is_size and guard_size_oblivious (#167198)
2025-11-17 05:47:40 +00:00
test_opaque_obj.py
[opaque obj] Allow non-effectful scriptobjs (#163714)
2025-11-13 00:35:20 +00:00
test_openmp.py
…
test_ops_fwd_gradients.py
…
test_ops_gradients.py
…
test_ops_jit.py
…
test_ops.py
Bugfix to forward autodiff causing different datatype 2 (#165784)
2025-11-06 01:59:53 +00:00
test_optim.py
…
test_out_dtype_op.py
…
test_overrides.py
…
test_package.py
…
test_per_overload_api.py
…
test_prims.py
…
test_privateuseone_python_backend.py
…
test_proxy_tensor.py
ProxyTorchDispatchMode: Decomposing missing sympy.SymExpr should handle constant literals (#167585)
2025-11-13 00:27:10 +00:00
test_pruning_op.py
Enable all SIM rules except disabled ones (#164645)
2025-10-17 07:27:11 +00:00
test_public_bindings.py
[FlexFlash] CuteDSL flat indexer needs to be colexigraphic in coordinate space (#166657)
2025-11-01 21:18:51 +00:00
test_python_dispatch.py
[2/N][Fix] Fix typo in test folder (#166374)
2025-10-29 03:02:07 +00:00
test_pytree.py
[pytree][dynamo] add test to ensure tree_map preserves dict order (#166236)
2025-11-05 17:04:40 +00:00
test_quantization.py
[2/N][Fix] Fix typo in test folder (#166374)
2025-10-29 03:02:07 +00:00
test_reductions.py
Handle only a Tensor for IntList parsing (#167606)
2025-11-13 17:39:38 +00:00
test_rename_privateuse1_to_existing_device.py
[user-streams] Track symbolic current stream (#165212)
2025-11-02 11:55:22 +00:00
test_scaled_matmul_cuda.py
Disable CUDA MXFP4 on non-B200 GPUs (#167857)
2025-11-17 17:14:53 +00:00
test_scatter_gather_ops.py
don't produce invalid grid configs (#166974)
2025-11-05 00:20:27 +00:00
test_schema_check.py
…
test_segment_reductions.py
Enable all SIM rules except disabled ones (#164645)
2025-10-17 07:27:11 +00:00
test_serialization.py
[inductor, 3.14] catch pickle.PicklingError exceptions (#167383)
2025-11-13 21:47:14 +00:00
test_set_default_mobile_cpu_allocator.py
…
test_shape_ops.py
[2/N][Fix] Fix typo in test folder (#166374)
2025-10-29 03:02:07 +00:00
test_show_pickle.py
…
test_sort_and_select.py
…
test_sparse_csr.py
[3/N] Fix unused loop variables (#166509)
2025-10-30 20:13:51 +00:00
test_sparse_semi_structured.py
[CI] Move the periodic debug tests to newer runner (#165158)
2025-10-21 21:28:12 +00:00
test_sparse.py
[MPS] mm out sparse (#167908)
2025-11-17 20:44:58 +00:00
test_spectral_ops.py
Correctly use test parameters (#166726)
2025-11-01 04:43:31 +00:00
test_stateless.py
[2/N][Fix] Fix typo in test folder (#166374)
2025-10-29 03:02:07 +00:00
test_static_runtime.py
[4/N] Use key in dict for existence checks (#167285)
2025-11-07 09:47:17 +00:00
test_subclass.py
…
test_sympy_utils.py
[Dynamo] Support for xor (#166065)
2025-11-11 05:44:08 +00:00
test_tensor_creation_ops.py
[random] Add generator arg to rand*_like APIs (#166160)
2025-11-03 19:58:45 +00:00
test_tensorboard.py
Add back Windows and macOS to tensorboard tests (#166389)
2025-10-29 04:34:57 +00:00
test_tensorexpr_pybind.py
…
test_tensorexpr.py
[3/N] Fix unused loop variables (#166509)
2025-10-30 20:13:51 +00:00
test_testing.py
[2/N] Use key in dict for existence checks (#167174)
2025-11-06 12:13:47 +00:00
test_throughput_benchmark.py
[5/N] Remove unused loop variables in tests (#166716)
2025-10-31 20:47:57 +00:00
test_torch.py
Rework PyObject preservation (v2) (#167564)
2025-11-17 14:52:02 +00:00
test_torchfuzz_repros.py
Add missing super().setUp() (#167163)
2025-11-06 17:55:23 +00:00
test_transformers.py
[7/N] Fix unused loop variables in tests (#167043)
2025-11-06 03:36:59 +00:00
test_type_hints.py
fix test_type_hints (#163150)
2025-11-04 00:29:22 +00:00
test_type_info.py
Enable PLW0127 in ruff (#165851)
2025-10-21 03:30:57 +00:00
test_type_promotion.py
Enable PLW0127 in ruff (#165851)
2025-10-21 03:30:57 +00:00
test_typing.py
…
test_unary_ufuncs.py
Fix mvlgamma_ FPE crash on x86 with integer input (#164230)
2025-11-14 17:09:10 +00:00
test_utils_config_module.py
…
test_utils_filelock.py
…
test_utils.py
[xpu][test] Migrated two test files to XPU (#166684)
2025-11-16 14:15:28 +00:00
test_varlen_attention.py
bwd pass (#164504)
2025-10-30 03:46:37 +00:00
test_view_ops.py
Enable PLW0127 in ruff (#165851)
2025-10-21 03:30:57 +00:00
test_vulkan.py
…
test_weak.py
[4/N] Use key in dict for existence checks (#167285)
2025-11-07 09:47:17 +00:00
test_xnnpack_integration.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_xpu_expandable_segments.py
[xpu][test] Add UT for expandable segments (#166495)
2025-11-04 08:01:35 +00:00
test_xpu.py
[xpu][test] Enable profiler test for XPU (#165423)
2025-11-10 04:02:59 +00:00