Test on torch latest version

Author: jainapurva

.github/workflows/float8_test.yml

@@ -23,7 +23,6 @@ jobs:
         include:
           - name: SM-89
             runs-on: amz2023.linux.g6.4xlarge.experimental.nvidia.gpu
-            torch-spec: '--pre torch==2.5.0.dev20240728+cu121 --index-url https://download.pytorch.org/whl/nightly/cu121'
             gpu-arch-type: "cuda"
             gpu-arch-version: "12.1"
 
@@ -40,7 +39,7 @@ jobs:
         yum install -y  devtoolset-10-binutils
         export PATH=/opt/rh/devtoolset-10/root/usr/bin/:$PATH
         python -m pip install --upgrade pip
-        pip install ${{ matrix.torch-spec }}
+        pip install torch
         pip install -r dev-requirements.txt
         pip install .
         pytest test/float8 --verbose -s

test/float8/test_fsdp2/test_fsdp2.py

@@ -37,7 +37,8 @@
 )
 
 is_H100 = torch.cuda.is_available() and torch.cuda.get_device_capability() >= (9, 0)
-if not is_H100:
+is_cuda_8_9 = torch.cuda.is_available() and torch.cuda.get_device_capability() >= (8, 9)
+if not is_cuda_8_9:
     pytest.skip("Unsupported CUDA device capability version", allow_module_level=True)
 
 class TestFloat8Common: