Fix "inf" norm gradient clipping for FSDP2#3199

wz337 wants to merge 1 commit intopytorch:mainpytorch/torchrec:mainfrom

wz337:export-D78326114wz337/torchrec:export-D78326114Copy head branch name to clipboard