search

ROCm / flash-attention

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

Update ck backend #84

Closed rocking5566 closed 1 month ago

rocking5566 commented 1 month ago

image Pass all the CI test in MI200 and MI300