candle/candle-flash-attn/tests
Michael Feil a594ef669c
Flash-Attn upgrade / SoftCap Candle-FlashAttn [2/n] (#2689)
* update flash-attn v1

* restore: hdim224

* add 224 flash_fwd_template

* remove whitespace

* softcap is working, including test and api.

* make softcap test case better

---------

Co-authored-by: laurent <laurent.mazare@gmail.com>
2024-12-31 09:41:23 +01:00
..
flash_attn_tests.rs Flash-Attn upgrade / SoftCap Candle-FlashAttn [2/n] (#2689) 2024-12-31 09:41:23 +01:00