fix: use LSE accum strides from params instead of hardcoded ones#2388

tridao merged 1 commit into

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

ZeronSix:fix-lse-stridesZeronSix/flash-attention:fix-lse-stridesCopy head branch name to clipboard