[5/N] Don't skip ASAN on some tests (#139265)

Fixes #ISSUE_NUMBER

Pull Request resolved: https://github.com/pytorch/pytorch/pull/139265
Approved by: https://github.com/ezyang
This commit is contained in:
cyy
2024-10-31 02:49:03 +00:00
committed by PyTorch MergeBot
parent d7411c0cc1
commit 29297731bb

View File

@ -1038,7 +1038,6 @@ instantiate_device_type_tests(TestDecomp, globals())
class DecompOneOffTests(TestCase):
@unittest.skipIf(TEST_WITH_ASAN, "Skipped under ASAN")
@onlyNativeDeviceTypes
@skipIfCrossRef
def test_contiguous_softmax(self, device):
@ -1053,7 +1052,6 @@ class DecompOneOffTests(TestCase):
res = torch._decomp.decompositions._softmax(x, -1, False)
self.assertEqual(ref.stride(), res.stride())
@unittest.skipIf(TEST_WITH_ASAN, "Skipped under ASAN")
@onlyNativeDeviceTypes
@skipIfCrossRef
def test_contiguous_log_softmax(self, device):
@ -1119,7 +1117,6 @@ class DecompOneOffTests(TestCase):
self.assertEqual(a.stride(), b.stride())
self.assertEqual(a.dtype, b.dtype)
@unittest.skipIf(TEST_WITH_ASAN, "Skipped under ASAN")
@onlyNativeDeviceTypes
@skipIfCrossRef
def test_elu_backward(self, device):
@ -1132,7 +1129,6 @@ class DecompOneOffTests(TestCase):
res = torch._decomp.decompositions.elu_backward(grad_out, 1.0, 1, 1, True, out)
self.assertEqual(ref, res)
@unittest.skipIf(TEST_WITH_ASAN, "Skipped under ASAN")
@onlyNativeDeviceTypes
@skipIfCrossRef
def test_threshold_backward_dtype(self, device):
@ -1143,7 +1139,6 @@ class DecompOneOffTests(TestCase):
res = torch._decomp.decompositions.threshold_backward(grad, input_tensor, 1)
self.assertEqual(ref.dtype, res.dtype)
@unittest.skipIf(TEST_WITH_ASAN, "Skipped under ASAN")
@onlyNativeDeviceTypes
@skipIfCrossRef
def test_weight_norm_interface(self, device):
@ -1163,7 +1158,6 @@ class DecompOneOffTests(TestCase):
torch._decomp.decompositions._weight_norm_interface(inp, inp2),
)
@unittest.skipIf(TEST_WITH_ASAN, "Skipped under ASAN")
@onlyCPU
@skipIfCrossRef
@skipOps(