pytorch/torch/distributed/fsdp
2025-10-30 12:22:25 +00:00
..
_fully_shard [DeviceMesh] Remove slicing submesh warning messages and clean up in fsdp params (#166466) 2025-10-29 20:52:49 +00:00
__init__.py [FSDP2] provide public API to share cuda streams across roots (#165024) 2025-10-14 17:50:46 +00:00
_common_utils.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_debug_utils.py [distributed] Replace 164 assert statements in fsdp directory (#165235) 2025-10-14 18:04:57 +00:00
_dynamo_utils.py
_exec_order_utils.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_flat_param.py [1/N] Remove unused loop variables (#166258) 2025-10-30 12:22:25 +00:00
_fsdp_extensions.py [distributed] Replace 164 assert statements in fsdp directory (#165235) 2025-10-14 18:04:57 +00:00
_init_utils.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_limiter_utils.py
_optim_utils.py Fix existing pyrefly errors on main (#166312) 2025-10-27 19:03:06 +00:00
_runtime_utils.py [distributed] Replace 164 assert statements in fsdp directory (#165235) 2025-10-14 18:04:57 +00:00
_shard_utils.py [distributed] Replace 164 assert statements in fsdp directory (#165235) 2025-10-14 18:04:57 +00:00
_state_dict_utils.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_trace_utils.py [distributed] Replace 164 assert statements in fsdp directory (#165235) 2025-10-14 18:04:57 +00:00
_traversal_utils.py [BE][5/16] fix typos in torch/ (torch/distributed/) (#156315) 2025-06-23 02:57:28 +00:00
_unshard_param_utils.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_wrap_utils.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
api.py
fully_sharded_data_parallel.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
sharded_grad_scaler.py [distributed] Replace 164 assert statements in fsdp directory (#165235) 2025-10-14 18:04:57 +00:00
wrap.py [distributed] Replace 164 assert statements in fsdp directory (#165235) 2025-10-14 18:04:57 +00:00