Skip to content

Commit bb43ced

Browse files
davidberard98pytorchmergebot
davidberard98
authored and
pytorchmergebot
committed
Inductor unit tests: cuda 12.6 -> 12.8 (pytorch#155056)
Fixes pytorch#154938 When we update the Triton version in CI, we'll require cuda >= 12.8 for certain AOTI tests to pass: these AOTI tests try to run nvcc on the triton-generated PTX, and triton-generated PTX is PTX 8.7, which requires CUDA 12.8 Pull Request resolved: pytorch#155056 Approved by: https://github.com/atalman, https://github.com/nWEIdia ghstack dependencies: pytorch#155167
1 parent 1c82878 commit bb43ced

File tree

3 files changed

+56
-30
lines changed

3 files changed

+56
-30
lines changed

‎.ci/docker/build.sh‎

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,30 @@ case "$tag" in
121121
TRITON=yes
122122
INDUCTOR_BENCHMARKS=yes
123123
;;
124+
pytorch-linux-jammy-cuda12.8-cudnn9-py3.12-gcc9-inductor-benchmarks)
125+
CUDA_VERSION=12.8.1
126+
CUDNN_VERSION=9
127+
ANACONDA_PYTHON_VERSION=3.12
128+
GCC_VERSION=9
129+
VISION=yes
130+
KATEX=yes
131+
UCX_COMMIT=${_UCX_COMMIT}
132+
UCC_COMMIT=${_UCC_COMMIT}
133+
TRITON=yes
134+
INDUCTOR_BENCHMARKS=yes
135+
;;
136+
pytorch-linux-jammy-cuda12.8-cudnn9-py3.13-gcc9-inductor-benchmarks)
137+
CUDA_VERSION=12.8.1
138+
CUDNN_VERSION=9
139+
ANACONDA_PYTHON_VERSION=3.13
140+
GCC_VERSION=9
141+
VISION=yes
142+
KATEX=yes
143+
UCX_COMMIT=${_UCX_COMMIT}
144+
UCC_COMMIT=${_UCC_COMMIT}
145+
TRITON=yes
146+
INDUCTOR_BENCHMARKS=yes
147+
;;
124148
pytorch-linux-focal-cuda12.6-cudnn9-py3-gcc9)
125149
CUDA_VERSION=12.6.3
126150
CUDNN_VERSION=9

‎.github/workflows/docker-builds.yml‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,8 @@ jobs:
5454
pytorch-linux-jammy-cuda12.6-cudnn9-py3.12-gcc9-inductor-benchmarks,
5555
pytorch-linux-jammy-cuda12.6-cudnn9-py3.13-gcc9-inductor-benchmarks,
5656
pytorch-linux-jammy-cuda12.8-cudnn9-py3-gcc9-inductor-benchmarks,
57+
pytorch-linux-jammy-cuda12.8-cudnn9-py3.12-gcc9-inductor-benchmarks,
58+
pytorch-linux-jammy-cuda12.8-cudnn9-py3.13-gcc9-inductor-benchmarks,
5759
pytorch-linux-focal-cuda12.8-cudnn9-py3-gcc9,
5860
pytorch-linux-focal-py3.9-clang10,
5961
pytorch-linux-focal-py3.11-clang10,

‎.github/workflows/inductor-unittest.yml‎

Lines changed: 30 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -26,13 +26,13 @@ jobs:
2626
curr_ref_type: ${{ github.ref_type }}
2727
opt_out_experiments: lf
2828

29-
linux-jammy-cuda12_6-py3_10-gcc9-inductor-build:
30-
name: cuda12.6-py3.10-gcc9-sm86
29+
linux-jammy-cuda12_8-py3_10-gcc9-inductor-build:
30+
name: cuda12.8-py3.10-gcc9-sm86
3131
uses: ./.github/workflows/_linux-build.yml
3232
needs: get-label-type
3333
with:
34-
build-environment: linux-jammy-cuda12.6-py3.10-gcc9-sm86
35-
docker-image-name: ci-image:pytorch-linux-jammy-cuda12.6-cudnn9-py3-gcc9-inductor-benchmarks
34+
build-environment: linux-jammy-cuda12.8-py3.10-gcc9-sm86
35+
docker-image-name: ci-image:pytorch-linux-jammy-cuda12.8-cudnn9-py3-gcc9-inductor-benchmarks
3636
cuda-arch-list: '8.6'
3737
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
3838
test-matrix: |
@@ -46,23 +46,23 @@ jobs:
4646
allow-reuse-old-whl: true
4747
secrets: inherit
4848

49-
linux-jammy-cuda12_6-py3_10-gcc9-inductor-test:
50-
name: cuda12.6-py3.10-gcc9-sm86
49+
linux-jammy-cuda12_8-py3_10-gcc9-inductor-test:
50+
name: cuda12.8-py3.10-gcc9-sm86
5151
uses: ./.github/workflows/_linux-test.yml
52-
needs: linux-jammy-cuda12_6-py3_10-gcc9-inductor-build
52+
needs: linux-jammy-cuda12_8-py3_10-gcc9-inductor-build
5353
with:
54-
build-environment: linux-jammy-cuda12.6-py3.10-gcc9-sm86
55-
docker-image: ${{ needs.linux-jammy-cuda12_6-py3_10-gcc9-inductor-build.outputs.docker-image }}
56-
test-matrix: ${{ needs.linux-jammy-cuda12_6-py3_10-gcc9-inductor-build.outputs.test-matrix }}
54+
build-environment: linux-jammy-cuda12.8-py3.10-gcc9-sm86
55+
docker-image: ${{ needs.linux-jammy-cuda12_8-py3_10-gcc9-inductor-build.outputs.docker-image }}
56+
test-matrix: ${{ needs.linux-jammy-cuda12_8-py3_10-gcc9-inductor-build.outputs.test-matrix }}
5757
secrets: inherit
5858

59-
linux-jammy-cuda12_6-py3_12-gcc9-inductor-build:
60-
name: cuda12.6-py3.12-gcc9-sm86
59+
linux-jammy-cuda12_8-py3_12-gcc9-inductor-build:
60+
name: cuda12.8-py3.12-gcc9-sm86
6161
uses: ./.github/workflows/_linux-build.yml
6262
needs: get-label-type
6363
with:
64-
build-environment: linux-jammy-cuda12.6-py3.12-gcc9-sm86
65-
docker-image-name: ci-image:pytorch-linux-jammy-cuda12.6-cudnn9-py3.12-gcc9-inductor-benchmarks
64+
build-environment: linux-jammy-cuda12.8-py3.12-gcc9-sm86
65+
docker-image-name: ci-image:pytorch-linux-jammy-cuda12.8-cudnn9-py3.12-gcc9-inductor-benchmarks
6666
cuda-arch-list: '8.6'
6767
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
6868
test-matrix: |
@@ -73,14 +73,14 @@ jobs:
7373
allow-reuse-old-whl: true
7474
secrets: inherit
7575

76-
linux-jammy-cuda12_6-py3_12-gcc9-inductor-test:
77-
name: cuda12.6-py3.12-gcc9-sm86
76+
linux-jammy-cuda12_8-py3_12-gcc9-inductor-test:
77+
name: cuda12.8-py3.12-gcc9-sm86
7878
uses: ./.github/workflows/_linux-test.yml
79-
needs: linux-jammy-cuda12_6-py3_12-gcc9-inductor-build
79+
needs: linux-jammy-cuda12_8-py3_12-gcc9-inductor-build
8080
with:
81-
build-environment: linux-jammy-cuda12.6-py3.12-gcc9-sm86
82-
docker-image: ${{ needs.linux-jammy-cuda12_6-py3_12-gcc9-inductor-build.outputs.docker-image }}
83-
test-matrix: ${{ needs.linux-jammy-cuda12_6-py3_12-gcc9-inductor-build.outputs.test-matrix }}
81+
build-environment: linux-jammy-cuda12.8-py3.12-gcc9-sm86
82+
docker-image: ${{ needs.linux-jammy-cuda12_8-py3_12-gcc9-inductor-build.outputs.docker-image }}
83+
test-matrix: ${{ needs.linux-jammy-cuda12_8-py3_12-gcc9-inductor-build.outputs.test-matrix }}
8484
secrets: inherit
8585

8686
linux-jammy-cpu-py3_12-inductor-halide-build:
@@ -161,13 +161,13 @@ jobs:
161161
test-matrix: ${{ needs.linux-jammy-cpu-py3_9-gcc11-inductor-build.outputs.test-matrix }}
162162
secrets: inherit
163163

164-
linux-jammy-cuda12_6-py3_13-gcc9-inductor-build:
165-
name: cuda12.6-py3.13-gcc9-sm86
164+
linux-jammy-cuda12_8-py3_13-gcc9-inductor-build:
165+
name: cuda12.8-py3.13-gcc9-sm86
166166
uses: ./.github/workflows/_linux-build.yml
167167
needs: get-label-type
168168
with:
169-
build-environment: linux-jammy-cuda12.6-py3.13-gcc9-sm86
170-
docker-image-name: ci-image:pytorch-linux-jammy-cuda12.6-cudnn9-py3.13-gcc9-inductor-benchmarks
169+
build-environment: linux-jammy-cuda12.8-py3.13-gcc9-sm86
170+
docker-image-name: ci-image:pytorch-linux-jammy-cuda12.8-cudnn9-py3.13-gcc9-inductor-benchmarks
171171
cuda-arch-list: '8.6'
172172
test-matrix: |
173173
{ include: [
@@ -177,12 +177,12 @@ jobs:
177177
allow-reuse-old-whl: true
178178
secrets: inherit
179179

180-
linux-jammy-cuda12_6-py3_13-gcc9-inductor-test:
181-
name: cuda12.6-py3.13-gcc9-sm86
180+
linux-jammy-cuda12_8-py3_13-gcc9-inductor-test:
181+
name: cuda12.8-py3.13-gcc9-sm86
182182
uses: ./.github/workflows/_linux-test.yml
183-
needs: linux-jammy-cuda12_6-py3_13-gcc9-inductor-build
183+
needs: linux-jammy-cuda12_8-py3_13-gcc9-inductor-build
184184
with:
185-
build-environment: linux-jammy-cuda12.6-py3.13-gcc9-sm86
186-
docker-image: ${{ needs.linux-jammy-cuda12_6-py3_13-gcc9-inductor-build.outputs.docker-image }}
187-
test-matrix: ${{ needs.linux-jammy-cuda12_6-py3_13-gcc9-inductor-build.outputs.test-matrix }}
185+
build-environment: linux-jammy-cuda12.8-py3.13-gcc9-sm86
186+
docker-image: ${{ needs.linux-jammy-cuda12_8-py3_13-gcc9-inductor-build.outputs.docker-image }}
187+
test-matrix: ${{ needs.linux-jammy-cuda12_8-py3_13-gcc9-inductor-build.outputs.test-matrix }}
188188
secrets: inherit

0 commit comments

Comments
 (0)