pytorch/torch/distributed/fsdp
2025-06-11 23:00:52 +00:00
..
_fully_shard [FSDP2] keep root unsharded when not specifying reshard_after_forward (#155319) 2025-06-06 20:29:31 +00:00
__init__.py
_common_utils.py [BE][Ez]: Use itertools.chain.from_iterable when possible (#148190) 2025-03-06 20:37:06 +00:00
_debug_utils.py
_dynamo_utils.py
_exec_order_utils.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
_flat_param.py [FSDP1] print fqns when debug FlatParamHandle (#151336) 2025-04-24 04:49:24 +00:00
_fsdp_extensions.py
_init_utils.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
_limiter_utils.py
_optim_utils.py [BE]: Apply ruff PERF403 to use dict comprehensions more often (#149257) 2025-03-18 00:46:07 +00:00
_runtime_utils.py
_shard_utils.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
_state_dict_utils.py Fix space typo in warning message (#143473) 2025-03-31 07:38:02 +00:00
_trace_utils.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
_traversal_utils.py
_unshard_param_utils.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
_wrap_utils.py
api.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
fully_sharded_data_parallel.py Convert rst files to md (#155369) 2025-06-11 23:00:52 +00:00
sharded_grad_scaler.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
wrap.py Use Python 3.9 typing (#148157) 2025-03-04 03:09:55 +00:00