fix(CP, FA): the conditional logic in the FA version contains a vulnerability when processing the output of Flash Attn forward pass#2825

Open

zhujian19891203 wants to merge 1 commit intoNVIDIA:mainfrom

021ai:CP_FA_fix

Commits on Apr 2, 2026

fix(CP, FA): when processing the output of Flash Attn forward pass, the conditional logic in the FA version contains a vulnerability, fix it
zhujian19891203
committed