from torch.utils.data import Dataset, TensorDataset, DataLoader, ConcatDataset
from torch.utils.data.dataset import random_split
from torch.utils.data.dataloader import default_collate, ExceptionWrapper, MP_STATUS_CHECK_INTERVAL
-from common_utils import (TestCase, run_tests, TEST_NUMPY, IS_WINDOWS, NO_MULTIPROCESSING_SPAWN,
+from common_utils import (TestCase, run_tests, TEST_NUMPY, IS_WINDOWS, IS_PPC, NO_MULTIPROCESSING_SPAWN,
skipIfRocm, load_tests)
# load_tests from common_utils is used to automatically filter tests for
mp = mp.get_context(method='spawn')
-JOIN_TIMEOUT = 17.0 if IS_WINDOWS else 8.5
+JOIN_TIMEOUT = 17.0 if IS_WINDOWS or IS_PPC else 8.5
class TestDatasetRandomSplit(TestCase):