From 8109bccb9f58de334fec539738dfba1956fade80 Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Mon, 23 Sep 2024 14:54:25 +0200 Subject: [PATCH] windows torchrun --- .github/workflows/test_cli_cuda_vllm.yaml | 4 ++-- tests/test_cli.py | 3 +++ 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/.github/workflows/test_cli_cuda_vllm.yaml b/.github/workflows/test_cli_cuda_vllm.yaml index c95771ef..2c3ae862 100644 --- a/.github/workflows/test_cli_cuda_vllm.yaml +++ b/.github/workflows/test_cli_cuda_vllm.yaml @@ -45,9 +45,9 @@ jobs: run: | pip install -e .[testing] - - name: Run tests + - name: Run tests (sequential) run: | - FORCE_SERIAL=1 pytest tests/test_cli.py -x -s -k "cli and cuda and vllm and not (tp or pp)" + FORCE_SEQUENTIAL=1 pytest tests/test_cli.py -x -s -k "cli and cuda and vllm and not (tp or pp)" run_cli_cuda_vllm_multi_gpu_tests: if: ${{ diff --git a/tests/test_cli.py b/tests/test_cli.py index 7816b70c..dca4ff0f 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -72,6 +72,9 @@ def test_cli_exit_code_0(launcher): @pytest.mark.parametrize("launcher", ["inline", "process", "torchrun"]) def test_cli_exit_code_1(launcher): + if sys.platform == "win32": + os.environ["USE_LIBUV"] = "0" + args_1 = [ "optimum-benchmark", "--config-dir",