Add optional output and logsumexp tensors to attention functions#289

LoserCheems merged 1 commit into

mainHKUSTDial/flash-sparse-attention:mainfrom

quantHKUSTDial/flash-sparse-attention:quantCopy head branch name to clipboard