Skip to content

Commit 4aa54b2

Browse files
xuzhao9facebook-github-bot
authored andcommitted
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
1 parent 8ecdf1a commit 4aa54b2

File tree

4 files changed

+9
-9
lines changed

4 files changed

+9
-9
lines changed

.github/workflows/v2-bisection.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,11 +15,11 @@ jobs:
1515
BISECT_DIR: ".torchbench/v2-bisection-ci"
1616
BISECT_BRANCH: "v2.0"
1717
PYTHON_VER: "3.10"
18-
CUDA_VER: "11.8"
18+
CUDA_VER: "12.1"
1919
NUMPY_VER: "1.21.2"
2020
CMAKE_VER: "3.26"
2121
MKL_VER: "2021.2.0"
22-
MAGMA_VER: "magma-cuda118"
22+
MAGMA_VER: "magma-cuda121"
2323
if: ${{ github.repository_owner == 'pytorch' }}
2424
runs-on: [self-hosted, bm-runner]
2525
timeout-minutes: 2880 # 48 hours
@@ -62,10 +62,10 @@ jobs:
6262
- name: Create the github issue
6363
if: env.TORCHBENCH_PERF_BISECTION_NONEMPTY_SIGNAL
6464
uses: peter-evans/create-issue-from-file@v4
65-
token: ${{ secrets.TORCHBENCH_ACCESS_TOKEN }}
6665
with:
6766
title: V2 Performance Signal Detected by TorchBench CI on ${{ env.TORCHBENCH_PERF_BISECTION_NONEMPTY_SIGNAL }}
6867
content-filepath: ./bisection-result/gh-issue.md
68+
token: ${{ secrets.TORCHBENCH_ACCESS_TOKEN }}
6969
labels: |
7070
torchbench-perf-report
7171
- name: Upload artifact

.github/workflows/v2-nightly.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,12 +11,12 @@ jobs:
1111
TORCHBENCH_VER: "v2"
1212
CONFIG_VER: "v2"
1313
PYTHON_VER: "3.10"
14-
CUDA_VER: "11.8"
15-
MAGMA_VERSION: "magma-cuda118"
14+
CUDA_VER: "12.1"
15+
MAGMA_VERSION: "magma-cuda121"
1616
CONDA_ENV_NAME: "torchbench-v2-nightly-ci"
1717
OUTPUT_DIR: ".torchbench/v2-nightly-ci"
1818
BISECTION_ROOT: ".torchbench/v2-bisection-ci"
19-
CUDA_VERSION: "cu118"
19+
CUDA_VERSION: "cu121"
2020
SCRIBE_GRAPHQL_ACCESS_TOKEN: ${{ secrets.SCRIBE_GRAPHQL_ACCESS_TOKEN }}
2121
AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY_ID }}
2222
AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_SECRET_ACCESS_KEY }}

.github/workflows/v2-sweep.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,8 @@ jobs:
1919
CONFIG_VER: "v2"
2020
SWEEP_DIR: ".torchbench/v2-sweep-ci"
2121
PYTHON_VER: "3.10"
22-
CUDA_VER: "11.8"
23-
MAGMA_VERSION: "magma-cuda118"
22+
CUDA_VER: "12.1"
23+
MAGMA_VERSION: "magma-cuda121"
2424
AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY_ID }}
2525
AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
2626
AWS_DEFAULT_REGION: us-east-1

.github/workflows/v3-bisection.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,10 +85,10 @@ jobs:
8585
continue-on-error: true
8686
if: env.TORCHBENCH_BISECTION_COMMIT_FOUND_OR_FAILED
8787
uses: peter-evans/create-issue-from-file@v4
88-
token: ${{ secrets.TORCHBENCH_ACCESS_TOKEN }}
8988
with:
9089
title: V3 Performance Signal Detected by TorchBench Userbenchmark "torch-nightly" on ${{ env.TORCHBENCH_BISECTION_COMMIT_FOUND_OR_FAILED }}
9190
content-filepath: ./benchmark/gh-issue.md
91+
token: ${{ secrets.TORCHBENCH_ACCESS_TOKEN }}
9292
labels: |
9393
torchbench-perf-report
9494
- name: Upload artifact

0 commit comments

Comments
 (0)