diff --git a/test/test_jit_fuser.py b/test/test_jit_fuser.py index 8698f39cc2be..2ea4c81db4e1 100644 --- a/test/test_jit_fuser.py +++ b/test/test_jit_fuser.py @@ -8,6 +8,7 @@ import torch import torch.nn as nn import torch.nn.functional as F from torch.testing import FileCheck +from unittest import skipIf from torch.testing._internal.common_utils import run_tests, IS_SANDCASTLE, ProfilingMode, GRAPH_EXECUTOR, \ enable_profiling_mode_for_profiling_tests, IS_WINDOWS, TemporaryDirectoryName, shell @@ -38,6 +39,7 @@ def warmup_forward(f, *args): return results +@skipIf(GRAPH_EXECUTOR == ProfilingMode.LEGACY, "skip due to SIGIOT failures, #67646") class TestFuser(JitTestCase): def assertAllFused(self, graph, except_for=()):