frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove unused Python variables in torch/[b-z]* (#136963)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/136963
Approved by: https://github.com/ezyang
This commit is contained in:
Tom Ritchford
2024-10-19 13:25:28 +00:00
committed by PyTorch MergeBot
parent fb44658415
commit c0582fd0f8
152 changed files with 376 additions and 528 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/distributed/fsdp/_state_dict_utils.py
View File

@ -297,7 +297,7 @@ def _full_pre_state_dict_hook(
``nn.Module``.
"""
if getattr(fsdp_state, "_device_mesh", False):
root_mesh = _mesh_resources.get_root_mesh(fsdp_state._device_mesh)
_mesh_resources.get_root_mesh(fsdp_state._device_mesh)
_common_pre_state_dict_hook(module, fsdp_state)
_common_unshard_pre_state_dict_hook(