[Add] Add FlashAttention 4 context parallel support.#3259

Baibaifan wants to merge 2 commits into

NVIDIA:mainNVIDIA/TransformerEngine:mainfrom

Baibaifan:support_fa4_with_cpBaibaifan/TransformerEngine:support_fa4_with_cpCopy head branch name to clipboard