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

Skip test_batch_norm in test_jit_fuser_te for asan (#98016)

Browse Source 
it takes 10+ minutes on asan?

Pull Request resolved: https://github.com/pytorch/pytorch/pull/98016
Approved by: https://github.com/huydhn
This commit is contained in:
Catherine Lee 2023-03-30 20:58:37 +00:00 committed by PyTorch MergeBot
parent 7bb5fb3c6d
commit 1f85390eb2
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_jit_fuser_te.py
View File

@ -21,7 +21,7 @@ torch._C._jit_set_profiling_executor(True)
torch._C._get_graph_executor_optimize(True)
from torch.testing._internal.common_utils import run_tests, ProfilingMode, GRAPH_EXECUTOR, \
enable_profiling_mode_for_profiling_tests, slowTest, skipIfTorchDynamo
enable_profiling_mode_for_profiling_tests, slowTest, skipIfTorchDynamo, TEST_WITH_ASAN
from torch.testing._internal.jit_utils import JitTestCase, \
RUN_CUDA, RUN_CUDA_HALF, RUN_CUDA_MULTI_GPU, warmup_backward, set_fusion_group_inlining, \
clone_inputs, get_traced_sample_variant_pairs, TensorExprTestOptions, NoTracerWarnContextManager
@ -2199,6 +2199,7 @@ class TestTEFuser(JitTestCase):
x = torch.ones((8, 1))
torch.testing.assert_close(eager(x), script(x))
@unittest.skipIf(TEST_WITH_ASAN, "takes 10+ minutes on asan")
def test_batch_norm(self):
def test(fn, args):
trace = torch.jit.trace(fn, args)