Commit Graph

3 Commits

Author SHA1 Message Date
jayhshah
a5a75274bc
FA3 kvcache + split kv + gqa parallelization (#1236) 2024-10-15 00:21:22 -07:00
Ying Zhang
dfe1a59e4b
Add var-seq-len to FA3 fp16 / bf16 fwd (#1072)
* fwd var-seq-len

* fixes

* benchmark

* fixes

---------

Co-authored-by: Tri Dao <tridao@users.noreply.github.com>
2024-07-22 21:32:41 -07:00
Tri Dao
7f67966cc7 FA3 initial code release 2024-07-11 09:53:36 -07:00