Skip to content

Commit 7aa190c

Browse files
authored
feat: add a naive CuTe flash-attn (#314)
* [flash-attn] add naive CuTe implementation * [flash-attn] dispatch headdim and add to test script * [chore] modify macros and comments * [lint] run pre-commit hooks * [chore] remove redundant cutlass include path
1 parent d3fa094 commit 7aa190c

File tree

3 files changed

+537
-0
lines changed

3 files changed

+537
-0
lines changed

0 commit comments

Comments
 (0)