From 7a74294786e548277fee0f2f873a11be9e473dcd Mon Sep 17 00:00:00 2001 From: Aart Bik Date: Wed, 14 Aug 2024 21:58:20 +0000 Subject: [PATCH] [sparse] enable meta tests (#133379) The skip for dynamo is no longer needed. Pull Request resolved: https://github.com/pytorch/pytorch/pull/133379 Approved by: https://github.com/ezyang --- test/test_sparse.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/test/test_sparse.py b/test/test_sparse.py index 3823fcf848e..6b35b782007 100644 --- a/test/test_sparse.py +++ b/test/test_sparse.py @@ -4415,7 +4415,6 @@ class TestSparseMeta(TestCase): @all_sparse_layouts('layout', include_strided=False) @parametrize("dtype", [torch.float64]) - @skipIfTorchDynamo("TODO(pearu,sparse-team) : investigate dynamo fail") def test_to_meta(self, dtype, layout): index_dtype = torch.int64 device = 'cpu' @@ -4426,7 +4425,6 @@ class TestSparseMeta(TestCase): @all_sparse_layouts('layout', include_strided=False) @parametrize("dtype", [torch.float64]) - @skipIfTorchDynamo("TODO(pearu,sparse-team) : investigate dynamo fail") def test_zeros_like_meta(self, dtype, layout): index_dtype = torch.int64 device = 'cpu' @@ -4437,7 +4435,6 @@ class TestSparseMeta(TestCase): @all_sparse_layouts('layout', include_strided=False) @parametrize("dtype", [torch.float64]) - @skipIfTorchDynamo("TODO(pearu,sparse-team) : investigate dynamo fail") def test_fake(self, dtype, layout): from torch._subclasses.fake_tensor import FakeTensorMode, FakeTensor fake_mode = FakeTensorMode() @@ -4454,7 +4451,6 @@ class TestSparseMeta(TestCase): @all_sparse_layouts('layout', include_strided=False) @parametrize("dtype", [torch.float64]) - @skipIfTorchDynamo("TODO(pearu,sparse-team) : investigate dynamo fail") def test_zeros_like_fake(self, dtype, layout): from torch._subclasses.fake_tensor import FakeTensorMode, FakeTensor from torch.utils._mode_utils import no_dispatch @@ -4483,7 +4479,6 @@ class TestSparseMeta(TestCase): @all_sparse_layouts('layout', include_strided=False) @parametrize("dtype", [torch.float64]) - @skipIfTorchDynamo("TODO(pearu,sparse-team) : investigate dynamo fail") def test_add_meta(self, dtype, layout): device = 'cpu' index_dtype = torch.int64