[Pytorch][Attention][CP][CI] Make 3221 changes robust - #3438

Merged

sudhakarsingh27 merged 7 commits into

NVIDIA:mainfrom

sudhakarsingh27:sudhakars/make_3221_robust

Aug 31, 2026

Commits on Aug 28, 2026

Support no-load-balance THD with FlashAttention
sudhakarsingh27
committed
d9d2240
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
9c8ca62
Clarify no-load-balance backend constraints
sudhakarsingh27
committed
fb02936
Add no-load-balance FlashAttention regression coverage
sudhakarsingh27
committed
5e14311
Use backend selection for FlashAttention CP test
sudhakarsingh27
committed
15af9b8
Clarify forced FA2 regression coverage
sudhakarsingh27
committed
11c205c
Rely on existing L1 attention runs
sudhakarsingh27
committed
905de9d