Tryign out H100 for float8

pytorch · Nov 6, 2024 · ef39a09 · ef39a09
1 parent 71a442a
commit ef39a09
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/.github/workflows/float8_test.yml b/.github/workflows/float8_test.yml
@@ -28,6 +28,11 @@ jobs:
             torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu121'
             gpu-arch-type: "cuda"
             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'
+            gpu-arch-type: "cuda"
+            gpu-arch-version: "12.1"
 
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with: