flash_attn/flash_attn_interface.py · 7d25a4ec4f4ecef5df1a208ec74ca34455dcb439 · gaoqiong / flash-attention · GitLab

Find file Blame History Permalink

Handle FlashAttnQKVPackedSplitFunc by making rng_state optional in backward · 7d25a4ec
Kirthi Shankar Sivamani authored Apr 13, 2023
```
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
```
7d25a4ec

flash_attn_interface.py 20 KB