flash-attention/README.md
2022-05-20 14:21:58 -07:00

12 lines
165 B
Markdown

Alpha release of FlashAttention.
To compile:
```
cd csrc/stream_attn
python setup.py install
```
Interface: `streaming_attention.py`
Contact: `trid@stanford.edu`