[simplefsdp] fix region ac in zero2-style FSDP#1970

ruisizhang123 wants to merge 1 commit intomainpytorch/torchtitan:mainfrom

ruisi/zero2_fixpytorch/torchtitan:ruisi/zero2_fixCopy head branch name to clipboard