csrc/flash_attn/fmha_api.cpp · 635f159ee32c78319d7ab41b160be998bc4b91b0 · 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

fmha_api.cpp 32.4 KB