flash-attention/csrc/flash_attn
2022-10-24 15:59:49 -07:00
..
cutlass@319a389f42 Add Cutlass as submodule 2022-06-02 09:54:16 -07:00
src Attempt to use atomicCAS to replace atomicAdd(bfloat16) 2022-10-23 16:22:43 -07:00
fmha_api.cpp Cast q.get_device() to char to avoid compiler warning (narrowing) 2022-10-24 15:59:49 -07:00