From 1b4f4a2d61827bc590abe73f536b6335ce2b3d44 Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Wed, 7 May 2025 21:06:34 -0700 Subject: [PATCH 1/3] Update to use torch2.7 for nv-torch-latest --- .github/workflows/nv-torch-latest-v100.yml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/nv-torch-latest-v100.yml b/.github/workflows/nv-torch-latest-v100.yml index ab6da53acf23..d0e37ad2c2c4 100644 --- a/.github/workflows/nv-torch-latest-v100.yml +++ b/.github/workflows/nv-torch-latest-v100.yml @@ -29,7 +29,7 @@ jobs: - name: Install pytorch run: | - pip install -U --cache-dir $TORCH_CACHE torch torchvision --index-url https://download.pytorch.org/whl/cu124 + pip install -U --cache-dir $TORCH_CACHE torch torchvision --index-url https://download.pytorch.org/whl/cu126 python -c "import torch; print('torch:', torch.__version__, torch)" python -c "import torch; print('CUDA available:', torch.cuda.is_available())" @@ -55,5 +55,5 @@ jobs: run: | unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch cd tests - 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" + pytest $PYTEST_OPTS --forked -n 8 unit/ --torch_ver="2.7" --cuda_ver="12.4" + pytest $PYTEST_OPTS --forked -m 'sequential' unit/ --torch_ver="2.7" --cuda_ver="12.4" From b79673620f9fc30633dcf93f9b2f32de5ab91d9d Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Thu, 8 May 2025 08:37:34 -0700 Subject: [PATCH 2/3] Update cuda version --- .github/workflows/nv-torch-latest-v100.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/nv-torch-latest-v100.yml b/.github/workflows/nv-torch-latest-v100.yml index d0e37ad2c2c4..5b1f34a19d39 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 8 unit/ --torch_ver="2.7" --cuda_ver="12.4" - pytest $PYTEST_OPTS --forked -m 'sequential' unit/ --torch_ver="2.7" --cuda_ver="12.4" + pytest $PYTEST_OPTS --forked -n 8 unit/ --torch_ver="2.7" --cuda_ver="12.6" + pytest $PYTEST_OPTS --forked -m 'sequential' unit/ --torch_ver="2.7" --cuda_ver="12.6" From a02f46771d80be274e36ae2abcf198d3c8eaa4d2 Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Fri, 30 May 2025 16:04:10 -0700 Subject: [PATCH 3/3] Update version to pull to 128 --- .github/workflows/nv-torch-latest-v100.yml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/nv-torch-latest-v100.yml b/.github/workflows/nv-torch-latest-v100.yml index c248e33ff604..c1f3e5390d1d 100644 --- a/.github/workflows/nv-torch-latest-v100.yml +++ b/.github/workflows/nv-torch-latest-v100.yml @@ -29,7 +29,7 @@ jobs: - name: Install pytorch run: | - pip install -U --cache-dir $TORCH_CACHE torch torchvision --index-url https://download.pytorch.org/whl/cu126 + pip install -U --cache-dir $TORCH_CACHE torch torchvision --index-url https://download.pytorch.org/whl/cu128 python -c "import torch; print('torch:', torch.__version__, torch)" python -c "import torch; print('CUDA available:', torch.cuda.is_available())" @@ -56,5 +56,5 @@ jobs: run: | unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch cd tests - pytest -x $PYTEST_OPTS --instafail --timeout 600 --forked -n 8 unit/ --torch_ver="2.7" --cuda_ver="12.6" - pytest $PYTEST_OPTS --instafail --timeout 600 --forked -m 'sequential' unit/ --torch_ver="2.7" --cuda_ver="12.6" + pytest -x $PYTEST_OPTS --instafail --timeout 600 --forked -n 8 unit/ --torch_ver="2.7" --cuda_ver="12.8" + pytest $PYTEST_OPTS --instafail --timeout 600 --forked -m 'sequential' unit/ --torch_ver="2.7" --cuda_ver="12.8"