[SPMD] Move a comment (#55877)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/55877
Address a comment in: https://github.com/pytorch/pytorch/pull/55353/files/10bc1dae408260e34d678e02a56c4d231ad8787b#r610930244
ghstack-source-id: 126369525
Test Plan: N/A
Reviewed By: rohan-varma
Differential Revision: D27729567
fbshipit-source-id: 5509ebfba2b741cd3532c69044227e5af0fb54fc
