Skip to content

Commit bc41a18

Browse files
author
cenxi.lx
committed
chore: improve comments in README
1 parent 2b0a52d commit bc41a18

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

README.md

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,8 @@ dq, dk, dv, db, dg, dh0 = chunk_gated_delta_rule_bwd(
7070
## Tests
7171

7272
```bash
73-
pip install flash_linear_attention==0.5.0 # require flash linear attention for comparison
73+
# require flash linear attention for comparison
74+
pip install flash_linear_attention==0.5.0
7475

7576
cd tests
7677
python test_gdr.py --set develop
@@ -92,7 +93,8 @@ Specifically, the forward (FWD) benchmarks measure single-kernel latency for dif
9293
More detail in [./benchmark/benchmark_results_H200.txt].
9394

9495
```bash
95-
pip install flash_linear_attention==0.5.0 flashinfer-python==0.6.9 # require flash linear attention and flashinfer for comparison
96+
# require flash linear attention and flashinfer for comparison
97+
pip install flash_linear_attention==0.5.0 flashinfer-python==0.6.9
9698

9799
cd benchmark
98100
python bench_gated_delta_rule.py

0 commit comments

Comments
 (0)