From 30c5006e9283233ee53f2e12688d42c79ed8bae7 Mon Sep 17 00:00:00 2001 From: drisspg Date: Fri, 8 Nov 2024 09:47:04 -0800 Subject: [PATCH] [CI/CD] switch to 2.5.1 for our 2.5 regression tests stack-info: PR: https://github.com/pytorch/ao/pull/1247, branch: drisspg/stack/19 --- .github/workflows/regression_test.yml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/regression_test.yml b/.github/workflows/regression_test.yml index 2726ced40a..e34031f6c4 100644 --- a/.github/workflows/regression_test.yml +++ b/.github/workflows/regression_test.yml @@ -33,9 +33,9 @@ jobs: torch-spec: 'torch==2.4.0' gpu-arch-type: "cuda" gpu-arch-version: "12.1" - - name: CUDA 2.5 + - name: CUDA 2.5.1 runs-on: linux.g5.12xlarge.nvidia.gpu - torch-spec: 'torch==2.5.0 --index-url https://download.pytorch.org/whl/cu121' + torch-spec: 'torch==2.5.1 --index-url https://download.pytorch.org/whl/cu121' gpu-arch-type: "cuda" gpu-arch-version: "12.1" - name: CUDA Nightly @@ -54,9 +54,9 @@ jobs: torch-spec: 'torch==2.4.0 --index-url https://download.pytorch.org/whl/cpu' gpu-arch-type: "cpu" gpu-arch-version: "" - - name: CPU 2.5 + - name: CPU 2.5.1 runs-on: linux.4xlarge - torch-spec: 'torch==2.5.0 --index-url https://download.pytorch.org/whl/cpu' + torch-spec: 'torch==2.5.1 --index-url https://download.pytorch.org/whl/cpu' gpu-arch-type: "cpu" gpu-arch-version: "" - name: CPU Nightly