[CuTe, Bwd] Return zero gradients for empty Q/K workloads - #2776

[CuTe, Bwd] Return zero gradients for empty Q/K workloads#2776

guoriyue wants to merge 2 commits into

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

guoriyue:agent/cute-empty-backwardguoriyue/flash-attention:agent/cute-empty-backwardCopy head branch name to clipboard