unpin torch

Author: jainapurva

.github/workflows/float8_test.yml

@@ -23,6 +23,7 @@ jobs:
         include:
           - name: SM-89
             runs-on: amz2023.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"
 
@@ -39,7 +40,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 torch
+        pip install ${{ matrix.torch-spec }}
         pip install -r dev-requirements.txt
         pip install .
         pytest test/float8 --verbose -s