OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[1/N] Don't skip ASAN on some tests (#138571)

Browse Source 
Clang15's ASAN is new enough so that it's possible to re-evaluate the disabled ASAN  tests.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/138571
Approved by: https://github.com/ezyang
This commit is contained in:
cyy 2024-10-23 02:38:42 +00:00 committed by PyTorch MergeBot
parent d437df342b
commit 86cca3fb05
2 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
test/distributed/_tensor/test_dtensor_ops.py
View File

@ -14,11 +14,7 @@ from torch.testing._internal.common_device_type import (
ops, ops,
) )
from torch.testing._internal.common_methods_invocations import DecorateInfo, op_db from torch.testing._internal.common_methods_invocations import DecorateInfo, op_db
from torch.testing._internal.common_utils import ( from torch.testing._internal.common_utils import run_tests, suppress_warnings
run_tests,
suppress_warnings,
TEST_WITH_ASAN,
)
from torch.testing._internal.distributed._tensor.common_dtensor import ( from torch.testing._internal.distributed._tensor.common_dtensor import (
DTensorConverter, DTensorConverter,
DTensorOpTestBase, DTensorOpTestBase,
@ -528,7 +524,6 @@ class TestDTensorOps(DTensorOpTestBase):
# only allow float dytpe for now, we can relax this constraint # only allow float dytpe for now, we can relax this constraint
# when feel necessary later (i.e when adding quantization support). # when feel necessary later (i.e when adding quantization support).
@unittest.skipIf(TEST_WITH_ASAN, "Skipped under ASAN")
@suppress_warnings @suppress_warnings
@ops(op_db, allowed_dtypes=(torch.float,)) @ops(op_db, allowed_dtypes=(torch.float,))
@skipOps("TestDTensorOps", "test_dtensor_op_db", dtensor_fails) @skipOps("TestDTensorOps", "test_dtensor_op_db", dtensor_fails)

3
test/test_openmp.py
View File

@ -4,7 +4,7 @@ import collections
import unittest import unittest
import torch import torch
from torch.testing._internal.common_utils import run_tests, TEST_WITH_ASAN, TestCase from torch.testing._internal.common_utils import run_tests, TestCase
try: try:
@ -27,7 +27,6 @@ class Network(torch.nn.Module):
@unittest.skipIf(not HAS_PSUTIL, "Requires psutil to run") @unittest.skipIf(not HAS_PSUTIL, "Requires psutil to run")
@unittest.skipIf(TEST_WITH_ASAN, "Cannot test with ASAN")
class TestOpenMP_ParallelFor(TestCase): class TestOpenMP_ParallelFor(TestCase):
batch = 20 batch = 20
channels = 1 channels = 1