[FSDP] Refactor casting of grad to full param dtype (#81574)
I noticed a comment was repeated, so I wanted to refactor the duplication. Refer to the comment for the explanation.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81574
Approved by: https://github.com/rohan-varma