frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[sparse] enable meta tests (#133379)

Browse Source 
The skip for dynamo is no longer needed.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/133379
Approved by: https://github.com/ezyang
This commit is contained in:
Aart Bik
2024-08-14 21:58:20 +00:00
committed by PyTorch MergeBot
parent 3965f11837
commit 7a74294786
1 changed files with 0 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/test_sparse.py
View File

@ -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