ci: test sm_86

punica-ai · Nov 28, 2023 · e523846 · e523846
1 parent 4d5b49f
commit e523846
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/gpu-test.yml b/.github/workflows/gpu-test.yml
@@ -20,6 +20,8 @@ jobs:
         include:
           - { arch: sm80, torch: "2.1.0", cuda_major: 12, cuda_minor: 1 }
           - { arch: sm80, torch: "2.1.0", cuda_major: 11, cuda_minor: 8 }
+          - { arch: sm86, torch: "2.1.0", cuda_major: 12, cuda_minor: 1 }
+          - { arch: sm86, torch: "2.1.0", cuda_major: 11, cuda_minor: 8 }
     runs-on: [self-hosted, "${{ matrix.arch }}"]
     steps:
       - name: Checkout