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

[dynamo, 3.12] enable tests disabled due to missing dynamo 3.12 support (#123300)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/123300
Approved by: https://github.com/jansel, https://github.com/malfet, https://github.com/zou3519
This commit is contained in:
William Wen
2024-04-04 16:58:26 -07:00
committed by PyTorch MergeBot
parent ae6f8d923c
commit cbde0f048b
26 changed files with 6 additions and 132 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_sparse_semi_structured.py
View File

@ -2,7 +2,6 @@
import itertools
import random
import unittest
import sys
import torch
from torch import nn
@ -197,7 +196,6 @@ class SparseSemiStructuredTensorCompileTest(torch._dynamo.test_case.TestCase):
assert sparse_result.stride() == sparse_compile_result.stride()
@unittest.skipIf(IS_WINDOWS, "torch.compile not supported on windows")
@unittest.skipIf(sys.version_info >= (3, 12), "torch.compile is not supported on python 3.12+")
@unittest.skipIf("cusparselt" not in SEMI_STRUCTURED_SUPPORTED_BACKENDS, "cusparselt not supported on this machine")
def test_mlp_contiguous_relu_compile_cusparselt(self):
"""
@ -207,7 +205,6 @@ class SparseSemiStructuredTensorCompileTest(torch._dynamo.test_case.TestCase):
SparseSemiStructuredTensorCompileTest._test_mlp_contiguous_relu_compile("cusparselt", dense_input_shape)
@unittest.skipIf(IS_WINDOWS, "torch.compile not supported on windows")
@unittest.skipIf(sys.version_info >= (3, 12), "torch.compile is not supported on python 3.12+")
def test_mlp_contiguous_relu_compile_cutlass(self):
"""
test for CUTLASS meta registrations (_sparse_semi_structured_linear) + torch.compile