Remove self refs in softmax for-loop#1924

tridao merged 2 commits into

Dao-AILab:mainDao-AILab/flash-attention:mainfrom

kevin-tong-augment:kevin/softmax-loop-local-varkevin-tong-augment/flash-attention:kevin/softmax-loop-local-varCopy head branch name to clipboard