[FA4][hd256] Fix layout of non-contiguous qkv in backward kernel#2545

Johnsonms merged 1 commit into

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

wangsiyu:fused_qga_backwardwangsiyu/flash-attention:fused_qga_backwardCopy head branch name to clipboard