flash-attention/csrc/flash_attn
2022-06-29 23:16:24 -07:00
..
cutlass@319a389f42 Add Cutlass as submodule 2022-06-02 09:54:16 -07:00
src Support batch size > 64K by swapping grid.x and grid.y 2022-06-29 23:16:24 -07:00
fmha_api.cpp Add type check (fp16) in the forward pass 2022-06-26 11:41:30 -07:00