Do not return lse if return_softmax == false - Githubissues

ROCm / flash-attention

Fast and memory-efficient exact attention

BSD 3-Clause "New" or "Revised" License

141 stars 46 forks source link

Do not return lse if return_softmax == false #59

Closed rocking5566 closed 5 months ago