Checking H100

Author: jainapurva

.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

.github/workflows/regression_test.yml

@@ -74,6 +74,11 @@ jobs:
             torch-spec: 'torch==2.5.1 --index-url https://download.pytorch.org/whl/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.4"
 
           - name: CPU 2.3
             runs-on: linux.4xlarge
@@ -90,11 +95,6 @@ jobs:
             torch-spec: 'torch==2.5.1 --index-url https://download.pytorch.org/whl/cpu'
             gpu-arch-type: "cpu"
             gpu-arch-version: ""
-          - 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:

test/float8/test_compile.py

@@ -469,7 +469,7 @@ def test_dynamic_scale_numeric_parity(dtype: torch.dtype):
 
 
 @unittest.skipIf(
-    not is_sm_at_least_89() or not is_fbcode(),
+    not is_sm_at_least_89() or is_fbcode(),
     "CUDA with float8 support not available; or not on fbcode (the test needs be run with the latest pytorch package)",
 )
 @pytest.mark.parametrize("dtype", [torch.bfloat16, torch.float16, torch.float32])