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

Disable mem leak check (#88373)

Browse Source 
tbh at this point it might be easier to make a new workflow and copy the relevant jobs...

Changes:
* Disable cuda mem leak check except for on scheduled workflows
* Make pull and trunk run on a schedule which will run the memory leak check
* Periodic will always run the memory leak check -> periodic does not have parallelization anymore
* Concurrency check changed to be slightly more generous
Pull Request resolved: https://github.com/pytorch/pytorch/pull/88373
Approved by: https://github.com/ZainRizvi, https://github.com/huydhn
This commit is contained in:
Catherine Lee
2022-11-04 20:47:42 +00:00
committed by PyTorch MergeBot
parent 093e220836
commit d632d94cc7
13 changed files with 36 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tools/testing/test_selections.py
View File

@ -5,7 +5,7 @@ from typing import Callable, Dict, List, Optional, Tuple
from tools.stats.import_test_stats import get_disabled_tests, get_slow_tests
NUM_PROCS = 2
NUM_PROCS = 1 if os.getenv("PYTORCH_TEST_CUDA_MEM_LEAK_CHECK", "0") == "1" else 2
class ShardJob: