Checking H100

Author: jainapurva

.github/workflows/float8_test.yml

@@ -23,16 +23,16 @@ jobs:
       fail-fast: false
       matrix:
         include:
-          - name: SM-89
-            runs-on: linux.g6.4xlarge.experimental.nvidia.gpu
-            torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu121'
-            gpu-arch-type: "cuda"
-            gpu-arch-version: "12.1"
+          # - name: SM-89
+          #   runs-on: linux.g6.4xlarge.experimental.nvidia.gpu
+          #   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"
+            gpu-arch-version: "12.4"
 
     permissions:
       id-token: write
@@ -52,4 +52,5 @@ jobs:
         pip install ${{ matrix.torch-spec }}
         pip install -r dev-requirements.txt
         pip install .
-        pytest test/float8 --verbose -s
+        python -c "import torch; major, minor = torch.cuda.get_device_capability(); print(f'CUDA Arch Version: {major}.{minor}')"
+        # pytest test/float8 --verbose -s