From cef5b93f280e0591d20a21135eee7c3ab6484f97 Mon Sep 17 00:00:00 2001 From: Catherine Lee Date: Mon, 22 Jan 2024 23:11:41 +0000 Subject: [PATCH] [ez] Serial when NUM_PROCS is 1 (#117977) Makes it easier to understand whats going on Pull Request resolved: https://github.com/pytorch/pytorch/pull/117977 Approved by: https://github.com/huydhn --- test/run_test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/run_test.py b/test/run_test.py index 5aec0357445..4f90b6ded11 100755 --- a/test/run_test.py +++ b/test/run_test.py @@ -1321,6 +1321,7 @@ def must_serial(file: Union[str, ShardedTest]) -> bool: or file in CI_SERIAL_LIST or file in JIT_EXECUTOR_TESTS or file in ONNX_SERIAL_LIST + or NUM_PROCS == 1 )