Upgrade to CUDA 12.1 for v2 workflows (#2143)

Summary:
As titled.

Pull Request resolved: #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

@@ -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

.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 }}

.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

.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