Add more optimal dk order optional arg#2512

drisspg wants to merge 1 commit into

mainDao-AILab/flash-attention:mainfrom

drisspg/stack/37Dao-AILab/flash-attention:drisspg/stack/37Copy head branch name to clipboard