OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
173f126068
pytorch/torch/distributed/fsdp
History
William Wen 3ac5a499dd [dynamo] add dynamo disable reasons to codebase (#150440) 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/150440
Approved by: https://github.com/jansel, https://github.com/zou3519
ghstack dependencies: #150341
2025-04-02 04:26:48 +00:00
..
_fully_shard [dynamo] add dynamo disable reasons to codebase (#150440) 2025-04-02 04:26:48 +00:00
__init__.py [FSDP2] Move to public torch.distributed.fsdp (#141868) 2024-12-07 01:24:28 +00:00
_common_utils.py [BE][Ez]: Use itertools.chain.from_iterable when possible (#148190) 2025-03-06 20:37:06 +00:00
_debug_utils.py PEP585 update - torch/distributed/fsdp (#145162) 2025-01-19 20:04:05 +00:00
_dynamo_utils.py PEP585 update - torch/distributed/fsdp (#145162) 2025-01-19 20:04:05 +00:00
_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 Use Python 3.9 typing (#148157) 2025-03-04 03:09:55 +00:00
_fsdp_extensions.py PEP585 update - torch/distributed/fsdp (#145162) 2025-01-19 20:04:05 +00:00
_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 PEP585 update - torch/distributed/fsdp (#145162) 2025-01-19 20:04:05 +00:00
_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 Fix bug in FSDP wrapped module with zero argument (#147771) 2025-02-26 01:40:53 +00:00
_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 PEP585 update - torch/distributed/fsdp (#145162) 2025-01-19 20:04:05 +00:00
_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 PEP585 update - torch/distributed/fsdp (#145162) 2025-01-19 20:04:05 +00:00
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 [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +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