diff --git a/benchmarks/dynamo/pr_time_benchmarks/expected_results.csv b/benchmarks/dynamo/pr_time_benchmarks/expected_results.csv index 6a90a5bfacbe..7d0fe1305fef 100644 --- a/benchmarks/dynamo/pr_time_benchmarks/expected_results.csv +++ b/benchmarks/dynamo/pr_time_benchmarks/expected_results.csv @@ -1,4 +1,4 @@ -add_loop_eager,compile_time_instruction_count,2953000000,0.015 +add_loop_eager,compile_time_instruction_count,2937000000,0.015 @@ -18,15 +18,15 @@ add_loop_inductor_gpu,compile_time_instruction_count,25900000000,0.015 -basic_modules_ListOfLinears_eager,compile_time_instruction_count,939900000,0.015 +basic_modules_ListOfLinears_eager,compile_time_instruction_count,952700000,0.015 -basic_modules_ListOfLinears_inductor,compile_time_instruction_count,18270000000,0.015 +basic_modules_ListOfLinears_inductor,compile_time_instruction_count,18390000000,0.015 -basic_modules_ListOfLinears_inductor_gpu_force_shape_pad,compile_time_instruction_count,16310000000,0.015 +basic_modules_ListOfLinears_inductor_gpu_force_shape_pad,compile_time_instruction_count,16450000000,0.015 @@ -34,11 +34,11 @@ basic_modules_ListOfLinears_inductor_gpu,compile_time_instruction_count,10370000 -update_hint_regression,compile_time_instruction_count,1700000000,0.02 +update_hint_regression,compile_time_instruction_count,1661000000,0.02 -float_args,compile_time_instruction_count,452500000,0.015 +float_args,compile_time_instruction_count,455500000,0.015 @@ -62,7 +62,7 @@ aotdispatcher_inference_subclass_cpu,compile_time_instruction_count,6022000000,0 -aotdispatcher_partitioner_cpu,compile_time_instruction_count,8672000000,0.015 +aotdispatcher_partitioner_cpu,compile_time_instruction_count,8724000000,0.015 @@ -70,7 +70,7 @@ aotdispatcher_partitioner_cpu2,compile_time_instruction_count,1917000000,0.015 -aotdispatcher_training_nosubclass_cpu,compile_time_instruction_count,3859000000,0.015 +aotdispatcher_training_nosubclass_cpu,compile_time_instruction_count,3838000000,0.015 diff --git a/test/test_serialization.py b/test/test_serialization.py index 94632b1a0ffc..5f7642e80e89 100644 --- a/test/test_serialization.py +++ b/test/test_serialization.py @@ -413,6 +413,7 @@ class SerializationMixin: def test_serialization_sparse_safe(self): self._test_serialization(True) + @unittest.skipIf(True, "Temporary skip due to gh-153143") def test_serialization_sparse_invalid(self): x = torch.zeros(3, 3) x[1][1] = 1 @@ -443,6 +444,7 @@ class SerializationMixin: "size is inconsistent with indices"): y = torch.load(f, weights_only=weights_only) + @unittest.skipIf(True, "Temporary skip due to gh-153143") def test_serialization_sparse_invalid_legacy_ctor(self): # This is set in test class setup but would not be check when running user code prev_invariant_check_enabled = torch.sparse.check_sparse_tensor_invariants.is_enabled() @@ -515,18 +517,22 @@ class SerializationMixin: f"`{compressed_indices_name}[[]..., 0[]] == 0` is not satisfied."): y = torch.load(f) + @unittest.skipIf(True, "Temporary skip due to gh-153143") def test_serialization_sparse_csr_invalid(self): self._test_serialization_sparse_compressed_invalid( torch.Tensor.to_sparse_csr, torch.Tensor.crow_indices, torch.Tensor.col_indices) + @unittest.skipIf(True, "Temporary skip due to gh-153143") def test_serialization_sparse_csc_invalid(self): self._test_serialization_sparse_compressed_invalid( torch.Tensor.to_sparse_csc, torch.Tensor.ccol_indices, torch.Tensor.row_indices) + @unittest.skipIf(True, "Temporary skip due to gh-153143") def test_serialization_sparse_bsr_invalid(self): self._test_serialization_sparse_compressed_invalid( lambda x: x.to_sparse_bsr((1, 1)), torch.Tensor.crow_indices, torch.Tensor.col_indices) + @unittest.skipIf(True, "Temporary skip due to gh-153143") def test_serialization_sparse_bsc_invalid(self): self._test_serialization_sparse_compressed_invalid( lambda x: x.to_sparse_bsc((1, 1)), torch.Tensor.ccol_indices, torch.Tensor.row_indices) diff --git a/torch/_utils.py b/torch/_utils.py index e813c082ba90..31c63896e787 100644 --- a/torch/_utils.py +++ b/torch/_utils.py @@ -276,7 +276,11 @@ _sparse_tensors_to_validate: list["torch.Tensor"] = [] def _validate_loaded_sparse_tensors(): try: for t in _sparse_tensors_to_validate: - if t.layout is torch.sparse_coo: + if True: + # Temporarily disable sparse tensor validation due to + # gh-153143. + pass + elif t.layout is torch.sparse_coo: torch._validate_sparse_coo_tensor_args( t._indices(), t._values(), t.size(), t.is_coalesced() )