is this the right way?

pytorch · Sep 6, 2024 · f20f6f1 · f20f6f1
1 parent 0818a09
commit f20f6f1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.github/workflows/inductor.yml b/.github/workflows/inductor.yml
@@ -22,6 +22,9 @@ jobs:
     timeout-minutes: 30
     steps:
       - uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
+        with:
+          gpu-arch-type: cuda
+          gpu-arch-version: "12.1"
       - name: "Checkout"
         run: |
           set -x