flash-attention/csrc/flash_attn
2022-10-24 17:29:36 -07:00
..
cutlass@319a389f42 Add Cutlass as submodule 2022-06-02 09:54:16 -07:00
src Get rid of o_rows_are_valid since we don't have headdim=16 anymore 2022-10-24 17:29:36 -07:00
fmha_api.cpp Support all head dims that are multiples of 8, up to 128 2022-10-24 16:04:21 -07:00