Skip to content

Commit 703784c

Browse files
committed
fixup! [CI] Upgrade CI envs to torch~=2.7.0 (#450)
1 parent c1ac8d0 commit 703784c

File tree

3 files changed

+6
-4
lines changed

3 files changed

+6
-4
lines changed

.github/workflows/nvidia-4090.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,9 +69,10 @@ jobs:
6969
# pip install git+https://github.com/Dao-AILab/causal-conv1d.git --no-build-isolation
7070
pip install -U uv
7171
pip uninstall -y flash-linear-attention
72+
uv pip install -U pytest setuptools wheel ninja
73+
uv pip install torch~=2.7.0 triton -U --index-url https://download.pytorch.org/whl/cu128
7274
uv pip install git+https://github.com/Dao-AILab/causal-conv1d.git -U --no-cache-dir --no-build-isolation
7375
uv pip install flash-attn -U --no-cache-dir --no-build-isolation
74-
uv pip install torch~=2.7.0 triton pytest setuptools wheel ninja -U --index-url https://download.pytorch.org/whl/cu128
7576
pip install .
7677
7778
- name: Check GPU status

.github/workflows/nvidia-a100.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,9 +68,9 @@ jobs:
6868
# pip install git+https://github.com/Dao-AILab/causal-conv1d.git --no-build-isolation
6969
pip install -U uv
7070
pip uninstall -y flash-linear-attention
71-
uv pip install git+https://github.com/Dao-AILab/causal-conv1d.git -U --no-cache-dir --no-build-isolation
71+
uv pip install -U pytest setuptools wheel ninja
72+
uv pip install torch~=2.7.0 triton -U --index-url https://download.pytorch.org/whl/cu128
7273
uv pip install flash-attn -U --no-cache-dir --no-build-isolation
73-
uv pip install torch~=2.7.0 triton pytest setuptools wheel ninja -U --index-url https://download.pytorch.org/whl/cu128
7474
pip install .
7575
7676
- name: Check GPU status

.github/workflows/nvidia-h100.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,9 +68,10 @@ jobs:
6868
# pip install git+https://github.com/Dao-AILab/causal-conv1d.git --no-build-isolation
6969
pip install -U uv
7070
pip uninstall -y flash-linear-attention
71+
uv pip install -U pytest setuptools wheel ninja
72+
uv pip install torch~=2.7.0 triton -U --index-url https://download.pytorch.org/whl/cu128
7173
uv pip install git+https://github.com/Dao-AILab/causal-conv1d.git -U --no-cache-dir --no-build-isolation
7274
uv pip install flash-attn -U --no-cache-dir --no-build-isolation
73-
uv pip install torch~=2.7.0 triton pytest setuptools wheel ninja -U --index-url https://download.pytorch.org/whl/cu128
7475
pip install .
7576
7677
- name: Check GPU status

0 commit comments

Comments
 (0)