Checking H100

pytorch · Jan 18, 2025 · 029f193 · 029f193
1 parent 46a7b0c
commit 029f193
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/float8_test.yml b/.github/workflows/float8_test.yml
@@ -30,9 +30,9 @@ jobs:
             gpu-arch-version: "12.1"
           - name: H100
             runs-on: linux.aws.h100
-            torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu121'
+            torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu124'
             gpu-arch-type: "cuda"
-            gpu-arch-version: "12.1"
+            gpu-arch-version: "12.4"
 
     permissions:
       id-token: write