support flashattention bwd#52

Closed

chengchao23 wants to merge 4 commits intoNVIDIA:mainfrom

chengchao23:main

Commits on Jan 7, 2026

flashattention-2 fwd and bwd support (not support causal and gqa)

程超
authored and
程超
committed
flashattention-2 bwd support causal mask

程超
authored and
程超
committed
support dq dk dv computation in accuracy atol=1e-2, rtol=5e-2

程超
authored and
程超
committed
support gqa

程超
authored and
程超
committed