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

[experiment] More procs in CI (#98098)

Browse Source 
experiment with more procs but only in master so prs dont get affected

Pull Request resolved: https://github.com/pytorch/pytorch/pull/98098
Approved by: https://github.com/huydhn
This commit is contained in:
Catherine Lee
2023-04-07 17:21:32 +00:00
committed by PyTorch MergeBot
parent e302f083bb
commit 9fd3eba6ce
2 changed files with 4 additions and 3 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 2
NUM_PROCS = 1 if IS_MEM_LEAK_CHECK else 3 if os.getenv("BRANCH", "") == "master" else 2
THRESHOLD = 60 * 10 # 10 minutes
# See Note [ROCm parallel CI testing]