From c7ecb1efcdd333c8be50393a6c5f1eb77f0d6085 Mon Sep 17 00:00:00 2001 From: jainapurva Date: Fri, 7 Nov 2025 11:51:47 -0800 Subject: [PATCH] Add torch2.9 in regression tests --- .github/workflows/regression_test.yml | 12 ++++++++++++ 1 file changed, 12 insertions(+) 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: