From 4aa54b245bdc2722d7dab5024fb53d06a1fa58d0 Mon Sep 17 00:00:00 2001
From: Xu Zhao <xzhao9@meta.com>
Date: Wed, 31 Jan 2024 09:25:53 -0800
Subject: [PATCH] Upgrade to CUDA 12.1 for v2 workflows (#2143)

Summary:
As titled.

Pull Request resolved: https://github.com/pytorch/benchmark/pull/2143

Test Plan: https://github.com/pytorch/benchmark/actions/runs/7712419504

Reviewed By: aaronenyeshi

Differential Revision: D53230805

Pulled By: xuzhao9

fbshipit-source-id: 83f01e3228d7599afa8f2ac3d7f8518c22b2ec83
---
 .github/workflows/v2-bisection.yml | 6 +++---
 .github/workflows/v2-nightly.yml   | 6 +++---
 .github/workflows/v2-sweep.yml     | 4 ++--
 .github/workflows/v3-bisection.yml | 2 +-
 4 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/.github/workflows/v2-bisection.yml b/.github/workflows/v2-bisection.yml
index 1c05cf0840..f1452dc9a0 100644
--- a/.github/workflows/v2-bisection.yml
+++ b/.github/workflows/v2-bisection.yml
@@ -15,11 +15,11 @@ jobs:
       BISECT_DIR: ".torchbench/v2-bisection-ci"
       BISECT_BRANCH: "v2.0"
       PYTHON_VER: "3.10"
-      CUDA_VER: "11.8"
+      CUDA_VER: "12.1"
       NUMPY_VER: "1.21.2"
       CMAKE_VER: "3.26"
       MKL_VER: "2021.2.0"
-      MAGMA_VER: "magma-cuda118"
+      MAGMA_VER: "magma-cuda121"
     if: ${{ github.repository_owner == 'pytorch' }}
     runs-on: [self-hosted, bm-runner]
     timeout-minutes: 2880 # 48 hours
@@ -62,10 +62,10 @@ jobs:
       - name: Create the github issue
         if: env.TORCHBENCH_PERF_BISECTION_NONEMPTY_SIGNAL
         uses: peter-evans/create-issue-from-file@v4
-        token: ${{ secrets.TORCHBENCH_ACCESS_TOKEN }}
         with:
           title: V2 Performance Signal Detected by TorchBench CI on ${{ env.TORCHBENCH_PERF_BISECTION_NONEMPTY_SIGNAL }}
           content-filepath: ./bisection-result/gh-issue.md
+          token: ${{ secrets.TORCHBENCH_ACCESS_TOKEN }}
           labels: |
             torchbench-perf-report
       - name: Upload artifact
diff --git a/.github/workflows/v2-nightly.yml b/.github/workflows/v2-nightly.yml
index 9f615ba5dc..52f3044ecc 100644
--- a/.github/workflows/v2-nightly.yml
+++ b/.github/workflows/v2-nightly.yml
@@ -11,12 +11,12 @@ jobs:
       TORCHBENCH_VER: "v2"
       CONFIG_VER: "v2"
       PYTHON_VER: "3.10"
-      CUDA_VER: "11.8"
-      MAGMA_VERSION: "magma-cuda118"
+      CUDA_VER: "12.1"
+      MAGMA_VERSION: "magma-cuda121"
       CONDA_ENV_NAME:  "torchbench-v2-nightly-ci"
       OUTPUT_DIR: ".torchbench/v2-nightly-ci"
       BISECTION_ROOT: ".torchbench/v2-bisection-ci"
-      CUDA_VERSION: "cu118"
+      CUDA_VERSION: "cu121"
       SCRIBE_GRAPHQL_ACCESS_TOKEN: ${{ secrets.SCRIBE_GRAPHQL_ACCESS_TOKEN }}
       AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY_ID }}
       AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
diff --git a/.github/workflows/v2-sweep.yml b/.github/workflows/v2-sweep.yml
index 424c69cd7b..d773c355ae 100644
--- a/.github/workflows/v2-sweep.yml
+++ b/.github/workflows/v2-sweep.yml
@@ -19,8 +19,8 @@ jobs:
       CONFIG_VER: "v2"
       SWEEP_DIR: ".torchbench/v2-sweep-ci"
       PYTHON_VER: "3.10"
-      CUDA_VER: "11.8"
-      MAGMA_VERSION: "magma-cuda118"
+      CUDA_VER: "12.1"
+      MAGMA_VERSION: "magma-cuda121"
       AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY_ID }}
       AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
       AWS_DEFAULT_REGION: us-east-1
diff --git a/.github/workflows/v3-bisection.yml b/.github/workflows/v3-bisection.yml
index e07f2360d2..3d95417911 100644
--- a/.github/workflows/v3-bisection.yml
+++ b/.github/workflows/v3-bisection.yml
@@ -85,10 +85,10 @@ jobs:
         continue-on-error: true
         if: env.TORCHBENCH_BISECTION_COMMIT_FOUND_OR_FAILED
         uses: peter-evans/create-issue-from-file@v4
-        token: ${{ secrets.TORCHBENCH_ACCESS_TOKEN }}
         with:
           title: V3 Performance Signal Detected by TorchBench Userbenchmark "torch-nightly" on ${{ env.TORCHBENCH_BISECTION_COMMIT_FOUND_OR_FAILED }}
           content-filepath: ./benchmark/gh-issue.md
+          token: ${{ secrets.TORCHBENCH_ACCESS_TOKEN }}
           labels: |
             torchbench-perf-report
       - name: Upload artifact