flash-attention/README.md

12 lines
165 B
Markdown
Raw Normal View History

2022-05-21 05:21:58 +08:00
Alpha release of FlashAttention.
To compile:
```
cd csrc/stream_attn
python setup.py install
```
Interface: `streaming_attention.py`
Contact: `trid@stanford.edu`