Merge pull request #108 from pytorch/update-ci-but-better-tested

Update to linux_job_v2 but tested this time
pytorch · Jan 22, 2025 · 8fe0de2 · 8fe0de2
2 parents 8ae1d54 + cb143fc
commit 8fe0de2
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -16,12 +16,15 @@ jobs:
         runner: ["linux.12xlarge"]
         gpu-arch-type: ["cpu"]
         include:
-          - python-version: 3.8
+          - python-version: 3.13
             runner: linux.g5.4xlarge.nvidia.gpu
             gpu-arch-type: cuda
-            gpu-arch-version: "11.8"
+            gpu-arch-version: "12.4"
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
+    permissions:
+      id-token: write
+      contents: read
     with:
       repository: pytorch/extension-cpp
       runner: ${{ matrix.runner }}