is this the right way?

Author: adamomainz

Diff for: .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