saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Increase timeout for slow tests (#109206)

Browse source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/109206
Approved by: https://github.com/huydhn
This commit is contained in:
Catherine Lee 2023-09-26 16:18:38 +00:00 committed by PyTorch MergeBot
parent 0dcea70bfd
commit 47adcd412f
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
test/run_test.py
View file

@ -601,8 +601,13 @@ def run_test(
and not RERUN_DISABLED_TESTS
and not options.continue_through_error
)
is_slow = "slow" in os.environ.get("TEST_CONFIG", "") or "slow" in os.environ.get(
"BUILD_ENVRIONMENT", ""
)
timeout = (
THRESHOLD * 3
THRESHOLD * 6
if is_slow
else THRESHOLD * 3
if should_file_rerun
and isinstance(test_module, ShardedTest)
and test_module.time is not None