diff --git a/test/test_dataloader.py b/test/test_dataloader.py index 0cdba26a303..6ece8afac85 100644 --- a/test/test_dataloader.py +++ b/test/test_dataloader.py @@ -40,6 +40,7 @@ from torch.testing._internal.common_utils import ( TEST_WITH_ROCM, TEST_WITH_TSAN, TestCase, + xfailIfLinux, ) from torch.utils.data import ( _utils, @@ -1382,6 +1383,8 @@ except RuntimeError as e: del loader1_it del loader2_it + # https://github.com/pytorch/pytorch/issues/128551 + @xfailIfLinux def test_segfault(self): p = ErrorTrackingProcess(target=_test_segfault) p.start() diff --git a/torch/testing/_internal/common_utils.py b/torch/testing/_internal/common_utils.py index 4fb13abb5b9..ea3485249bb 100644 --- a/torch/testing/_internal/common_utils.py +++ b/torch/testing/_internal/common_utils.py @@ -1507,6 +1507,10 @@ def xfailIfTorchDynamo(func): return unittest.expectedFailure(func) if TEST_WITH_TORCHDYNAMO else func +def xfailIfLinux(func): + return unittest.expectedFailure(func) if IS_LINUX and not TEST_WITH_ROCM else func + + def skipIfTorchDynamo(msg="test doesn't currently work with dynamo"): """ Usage: