OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[FSDP1] print fqns when debug FlatParamHandle (#151336)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/151336
Approved by: https://github.com/awgu, https://github.com/Skylion007
This commit is contained in:
Wei Feng 2025-04-15 11:46:11 -07:00 committed by PyTorch MergeBot
parent 2a58d2a155
commit 2102b3b4c5
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
torch/distributed/fsdp/_flat_param.py
View File

@ -593,6 +593,9 @@ class FlatParamHandle:
)
self._use_unsharded_views(as_params=False)
def __repr__(self):
return f"FlatParamHandle(flat_param.fqns={self.flat_param._fqns})"
def _init_setattr_fns(self):
use_unsafe_setattr = os.environ.get(_FSDP_USE_UNSAFE_SETATTR, "") == "1"
self._setattr_tensor: Callable[[nn.Module, str, Tensor], None]