flash-attention/csrc/flash_attn
Nicolas Patry 5bf201966a
Fixing argument checking when using seqlenq_ngroups_swapped. (#976)
When user send `out` as a parameter of the function
`seqlenq_ngroups_swapped` with parameters that trigger,
the CHECK_SHAPE is incorrect (since q shape is modified.)
2024-06-30 22:39:22 -07:00
..
src remove swizzle part of sV.data() to get a completely non-swizzle sVtNoSwizzle (#984) 2024-06-30 22:38:44 -07:00
flash_api.cpp Fixing argument checking when using seqlenq_ngroups_swapped. (#976) 2024-06-30 22:39:22 -07:00