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

Revert "[experiment] More procs in CI (#98098)"

Browse Source 
This reverts commit 9fd3eba6ceb048cfdcb430e34f9168eda888b4c8.

Reverted https://github.com/pytorch/pytorch/pull/98098 on behalf of https://github.com/clee2000 due to I think theres a bug
This commit is contained in:
PyTorch MergeBot
2023-04-07 19:50:54 +00:00
parent 5210d7c423
commit 55724a5ec9
2 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tools/testing/test_selections.py
View File

@ -8,7 +8,7 @@ from tools.stats.import_test_stats import get_disabled_tests, get_slow_tests
IS_MEM_LEAK_CHECK = os.getenv("PYTORCH_TEST_CUDA_MEM_LEAK_CHECK", "0") == "1"
NUM_PROCS = 1 if IS_MEM_LEAK_CHECK else 3 if os.getenv("BRANCH", "") == "master" else 2
NUM_PROCS = 1 if IS_MEM_LEAK_CHECK else 2
THRESHOLD = 60 * 10 # 10 minutes
# See Note [ROCm parallel CI testing]