Skip to content

Commit 201d25b

Browse files
authored
Add flash-linear-attention and causal-conv1d to Helion Docker image (#453)
These are dependencies needed by the gated_deltanet and causal_conv kernel benchmarks.
1 parent a0e0094 commit 201d25b

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

docker/helion.Dockerfile

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,12 @@ RUN sudo uv pip install --system \
5353
# Helion
5454
RUN sudo uv pip install --system helion
5555

56+
# Flash Linear Attention
57+
RUN sudo uv pip install --system flash-linear-attention
58+
59+
# Causal Conv1d (Dao-AILab reference for benchmarking)
60+
RUN sudo uv pip install --system causal-conv1d
61+
5662
# # tinygrad
5763
# RUN sudo uv pip install --system tinygrad~=0.10
5864

0 commit comments

Comments
 (0)