From e342a67089304531c0e8172c02b4f0d520fc5e92 Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Wed, 26 Feb 2025 14:28:35 -0800 Subject: [PATCH 1/2] Update parallelism for nv-torch-latest/nightly due to more GPUs per runner Signed-off-by: Logan Adams --- .github/workflows/nv-torch-latest-v100.yml | 2 +- .github/workflows/nv-torch-nightly-v100.yml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/nv-torch-latest-v100.yml b/.github/workflows/nv-torch-latest-v100.yml index 3884663d9e55..46d13e0efe58 100644 --- a/.github/workflows/nv-torch-latest-v100.yml +++ b/.github/workflows/nv-torch-latest-v100.yml @@ -55,5 +55,5 @@ jobs: run: | unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch cd tests - pytest $PYTEST_OPTS --forked -n 4 unit/ --torch_ver="2.6" --cuda_ver="12.4" + pytest $PYTEST_OPTS --forked -n 8 unit/ --torch_ver="2.6" --cuda_ver="12.4" pytest $PYTEST_OPTS --forked -m 'sequential' unit/ --torch_ver="2.6" --cuda_ver="12.4" diff --git a/.github/workflows/nv-torch-nightly-v100.yml b/.github/workflows/nv-torch-nightly-v100.yml index cb8c15f0d709..9ce6c8b3853d 100644 --- a/.github/workflows/nv-torch-nightly-v100.yml +++ b/.github/workflows/nv-torch-nightly-v100.yml @@ -54,7 +54,7 @@ jobs: run: | unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch cd tests - pytest $PYTEST_OPTS --forked -n 4 unit/ + pytest $PYTEST_OPTS --forked -n 8 unit/ pytest $PYTEST_OPTS --forked -m 'sequential' unit/ - name: Open GitHub issue if nightly CI fails From 41b18fc29364dd60ca7c13f7f1ba799c235e613b Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Wed, 26 Feb 2025 14:31:13 -0800 Subject: [PATCH 2/2] Fix path trigger on nv-torch-nightly-v100 --- .github/workflows/nv-torch-nightly-v100.yml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/nv-torch-nightly-v100.yml b/.github/workflows/nv-torch-nightly-v100.yml index 9ce6c8b3853d..1ab73a9597cd 100644 --- a/.github/workflows/nv-torch-nightly-v100.yml +++ b/.github/workflows/nv-torch-nightly-v100.yml @@ -2,11 +2,11 @@ name: nv-torch-nightly-v100 on: workflow_dispatch: - pull_request: - paths: - - '.github/workflows/nv-torch-nightly-v100.yml' schedule: - cron: "0 0 * * *" + pull_request: + paths: + - '.github/workflows/nv-torch-nightly-v100.yml' concurrency: group: ${{ github.workflow }}-${{ github.ref }}