Commit Graph

3 Commits

Author SHA1 Message Date
Ying Zhang
3669b25206
bwd benchmark + small fixes (#1129) 2024-08-05 21:27:52 -07:00
Ying Zhang
c7f20a2d31 add cudnn benchmark for var-len 2024-07-31 22:33:29 -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