search

ROCm / flash-attention

Fast and memory-efficient exact attention
BSD 3-Clause "New" or "Revised" License
109 stars 33 forks source link

Integrate ck tile backward #65

Closed rocking5566 closed 6 days ago

rocking5566 commented 1 week ago

This PR add mha_bwd() and mha_varlen_bwd()