diff --git a/.github/workflows/regression_test.yml b/.github/workflows/regression_test.yml index cc474ff9e7..456822e4f2 100644 --- a/.github/workflows/regression_test.yml +++ b/.github/workflows/regression_test.yml @@ -77,6 +77,12 @@ jobs: gpu-arch-type: "cuda" gpu-arch-version: "12.6" dev-requirements-overrides: "" + - name: CUDA 2.9 + runs-on: linux.g5.12xlarge.nvidia.gpu + torch-spec: 'torch==2.9.0' + gpu-arch-type: "cuda" + gpu-arch-version: "12.6" + dev-requirements-overrides: "" - name: CPU 2.6 runs-on: linux.4xlarge @@ -96,6 +102,12 @@ jobs: gpu-arch-type: "cpu" gpu-arch-version: "" dev-requirements-overrides: "" + - name: CPU 2.9 + runs-on: linux.4xlarge + torch-spec: 'torch==2.9.0 --index-url https://download.pytorch.org/whl/cpu' + gpu-arch-type: "cpu" + gpu-arch-version: "" + dev-requirements-overrides: "" uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main with: