[CI] Use CUDA 11.8 to run inductor benchmark tests (#96059)

Pull Request resolved: #96059 Approved by: https://github.com/malfet, https://github.com/huydhn
pytorch · Mar 6, 2023 · 483e193 · 483e193
1 parent 000d9ec
commit 483e193
Showing 1 changed file with 20 additions and 20 deletions.
diff --git a/.github/workflows/inductor.yml b/.github/workflows/inductor.yml
@@ -14,12 +14,12 @@ concurrency:
   cancel-in-progress: true
 
 jobs:
-  linux-bionic-cuda11_7-py3_10-gcc7-inductor-build:
-    name: cuda11.7-py3.10-gcc7-sm86
+  linux-bionic-cuda11_8-py3_10-gcc7-inductor-build:
+    name: cuda11.8-py3.10-gcc7-sm86
     uses: ./.github/workflows/_linux-build.yml
     with:
-      build-environment: linux-bionic-cuda11.7-py3.10-gcc7-sm86
-      docker-image-name: pytorch-linux-bionic-cuda11.7-cudnn8-py3-gcc7
+      build-environment: linux-bionic-cuda11.8-py3.10-gcc7-sm86
+      docker-image-name: pytorch-linux-bionic-cuda11.8-cudnn8-py3-gcc7
       cuda-arch-list: '8.6'
       test-matrix: |
         { include: [
@@ -31,35 +31,35 @@ jobs:
           { config: "inductor_distributed", shard: 1, num_shards: 1, runner: "linux.g5.12xlarge.nvidia.gpu" },
         ]}
 
-  linux-bionic-cuda11_7-py3_10-gcc7-inductor-test:
-    name: cuda11.7-py3.10-gcc7-sm86
+  linux-bionic-cuda11_8-py3_10-gcc7-inductor-test:
+    name: cuda11.8-py3.10-gcc7-sm86
     uses: ./.github/workflows/_linux-test.yml
-    needs: linux-bionic-cuda11_7-py3_10-gcc7-inductor-build
+    needs: linux-bionic-cuda11_8-py3_10-gcc7-inductor-build
     with:
-      build-environment: linux-bionic-cuda11.7-py3.10-gcc7-sm86
-      docker-image: ${{ needs.linux-bionic-cuda11_7-py3_10-gcc7-inductor-build.outputs.docker-image }}
-      test-matrix: ${{ needs.linux-bionic-cuda11_7-py3_10-gcc7-inductor-build.outputs.test-matrix }}
+      build-environment: linux-bionic-cuda11.8-py3.10-gcc7-sm86
+      docker-image: ${{ needs.linux-bionic-cuda11_8-py3_10-gcc7-inductor-build.outputs.docker-image }}
+      test-matrix: ${{ needs.linux-bionic-cuda11_8-py3_10-gcc7-inductor-build.outputs.test-matrix }}
 
-  linux-bionic-cuda11_7-py3_10-gcc7-inductor-build-gcp:
-    name: cuda11.7-py3.10-gcc7-sm80
+  linux-bionic-cuda11_8-py3_10-gcc7-inductor-build-gcp:
+    name: cuda11.8-py3.10-gcc7-sm80
     uses: ./.github/workflows/_linux-build.yml
     with:
-      build-environment: linux-bionic-cuda11.7-py3.10-gcc7-sm80
-      docker-image-name: pytorch-linux-bionic-cuda11.7-cudnn8-py3-gcc7
+      build-environment: linux-bionic-cuda11.8-py3.10-gcc7-sm80
+      docker-image-name: pytorch-linux-bionic-cuda11.8-cudnn8-py3-gcc7
       cuda-arch-list: '8.0'
       test-matrix: |
         { include: [
           { config: "inductor_torchbench_smoketest_perf", shard: 1, num_shards: 1, runner: "linux.gcp.a100" },
         ]}
 
-  linux-bionic-cuda11_7-py3_10-gcc7-inductor-test-gcp:
-    name: cuda11.7-py3.10-gcc7-sm80
+  linux-bionic-cuda11_8-py3_10-gcc7-inductor-test-gcp:
+    name: cuda11.8-py3.10-gcc7-sm80
     uses: ./.github/workflows/_linux-test.yml
-    needs: linux-bionic-cuda11_7-py3_10-gcc7-inductor-build-gcp
+    needs: linux-bionic-cuda11_8-py3_10-gcc7-inductor-build-gcp
     with:
-      build-environment: linux-bionic-cuda11.7-py3.10-gcc7-sm80
-      docker-image: ${{ needs.linux-bionic-cuda11_7-py3_10-gcc7-inductor-build-gcp.outputs.docker-image }}
-      test-matrix: ${{ needs.linux-bionic-cuda11_7-py3_10-gcc7-inductor-build-gcp.outputs.test-matrix }}
+      build-environment: linux-bionic-cuda11.8-py3.10-gcc7-sm80
+      docker-image: ${{ needs.linux-bionic-cuda11_8-py3_10-gcc7-inductor-build-gcp.outputs.docker-image }}
+      test-matrix: ${{ needs.linux-bionic-cuda11_8-py3_10-gcc7-inductor-build-gcp.outputs.test-matrix }}
       use-gha: anything-non-empty-to-use-gha
 
   linux-focal-cpu-py3_8-gcc7-inductor-build: