frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make JOIN_TIMEOUT longer for ppc64le (#14107)

Browse Source 
Summary:
This should resolve the issue on ppc64le getting FAIL: test_proper_exit (__main__.TestDataLoader). This only happens when the CI build machine is very busy and fails with a timeout.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14107

Differential Revision: D13103859

Pulled By: soumith

fbshipit-source-id: 268be80b59840853c5025f3211af272f68608fe5
This commit is contained in:
Freddie Mendoza
2018-11-16 12:05:27 -08:00
committed by Facebook Github Bot
parent c192788188
commit 2c21de2007
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_dataloader.py
View File

@ -15,7 +15,7 @@ from torch import multiprocessing as mp
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
@ -34,7 +34,7 @@ if not NO_MULTIPROCESSING_SPAWN:
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):