fixup! [CI] Upgrade CI envs to torch~=2.7.0 (#450)

Author: zhiyuan1i

.github/workflows/nvidia-4090.yml

@@ -62,16 +62,15 @@ jobs:
       - name: Setup python
         if: steps.find-dependent-tests.outputs.test_files && steps.check_skip.outputs.skip_tests == 'false'
         run: |
-          pip uninstall -y flash-linear-attention
           # Installed by hand to avoid issues with pip
           # pip install -U pytest setuptools wheel ninja torch triton
           # MAX_JOBS=4 pip install -U flash-attn --no-build-isolation
           # pip install git+https://github.com/Dao-AILab/causal-conv1d.git --no-build-isolation
           pip install -U uv
           pip uninstall -y flash-linear-attention
-          uv pip install git+https://github.com/Dao-AILab/causal-conv1d.git -U --no-cache-dir --no-build-isolation
+          uv pip install -U pytest setuptools wheel ninja
+          uv pip install torch~=2.7.0 triton -U --index-url https://download.pytorch.org/whl/cu128
           uv pip install flash-attn -U --no-cache-dir --no-build-isolation
-          uv pip install torch~=2.7.0 triton pytest setuptools wheel ninja -U --index-url https://download.pytorch.org/whl/cu128
           pip install .
 
       - name: Check GPU status

.github/workflows/nvidia-a100.yml

@@ -68,9 +68,10 @@ jobs:
           # pip install git+https://github.com/Dao-AILab/causal-conv1d.git --no-build-isolation
           pip install -U uv
           pip uninstall -y flash-linear-attention
+          uv pip install -U pytest setuptools wheel ninja
+          uv pip install torch~=2.7.0 triton -U --index-url https://download.pytorch.org/whl/cu128
           uv pip install git+https://github.com/Dao-AILab/causal-conv1d.git -U --no-cache-dir --no-build-isolation
           uv pip install flash-attn -U --no-cache-dir --no-build-isolation
-          uv pip install torch~=2.7.0 triton pytest setuptools wheel ninja -U --index-url https://download.pytorch.org/whl/cu128
           pip install .
 
       - name: Check GPU status

.github/workflows/nvidia-h100.yml

@@ -68,9 +68,10 @@ jobs:
           # pip install git+https://github.com/Dao-AILab/causal-conv1d.git --no-build-isolation
           pip install -U uv
           pip uninstall -y flash-linear-attention
+          uv pip install -U pytest setuptools wheel ninja
+          uv pip install torch~=2.7.0 triton -U --index-url https://download.pytorch.org/whl/cu128
           uv pip install git+https://github.com/Dao-AILab/causal-conv1d.git -U --no-cache-dir --no-build-isolation
           uv pip install flash-attn -U --no-cache-dir --no-build-isolation
-          uv pip install torch~=2.7.0 triton pytest setuptools wheel ninja -U --index-url https://download.pytorch.org/whl/cu128
           pip install .
 
       - name: Check GPU status