pytorch
diff --git a/‎torch/distributed/_state_dict_utils.py
Copy file name to clipboardExpand all lines: torch/distributed/_state_dict_utils.py
+2-1Lines changed: 2 additions & 1 deletion b/‎torch/distributed/_state_dict_utils.py
Copy file name to clipboardExpand all lines: torch/distributed/_state_dict_utils.py
+2-1Lines changed: 2 additions & 1 deletion
@@ -172,8 +172,9 @@ def dtensor_func(value, pg, device):
             device_mesh=value.device_mesh,
             placements=placements,
         )
-        # Call `wait()` to force the tensor is synchronous with respect
+        # Call `wait()` to force the tensor to be synchronous with respect
         # to the main stream.
+        # See the discussion in https://github.com/pytorch/pytorch/pull/117799.
         return value.to_local().wait()
 
     return _iterate_state_dict(