| .. |
|
__init__.py
|
[FSDP] Introduce ModuleWrapPolicy for simplicity (#88450)
|
2022-11-12 04:14:32 +00:00 |
|
_common_utils.py
|
[FSDP()] Fix fully_shard fwd hook registration (#90201)
|
2022-12-06 06:09:03 +00:00 |
|
_dynamo_utils.py
|
Revert D41682843: Multisect successfully blamed D41682843 for test or build failures (#90132)
|
2022-12-04 05:35:17 +00:00 |
|
_exec_order_utils.py
|
[FSDP] Limit all gather after pre-unshard (#89057)
|
2022-11-30 02:27:02 +00:00 |
|
_fsdp_extensions.py
|
[FSDP] ufmt /fsdp (#87811)
|
2022-10-27 04:25:55 +00:00 |
|
_init_utils.py
|
[FSDP] Save _stream_to_name for debugging (#90611)
|
2022-12-11 03:46:18 +00:00 |
|
_limiter_utils.py
|
[FSDP()][8/N] Refactor limiter's _FreeEventQueue (#87922)
|
2022-10-31 16:45:24 +00:00 |
|
_optim_utils.py
|
[FSDP][optim_state_dict][2/N] Add _get_fqn_to_fsdp_param_info to map from original FQN to flat_param (#89899)
|
2022-12-07 19:40:47 +00:00 |
|
_runtime_utils.py
|
[FSDP] Fix _pre_forward type annotation (#90621)
|
2022-12-11 06:39:38 +00:00 |
|
_shard_utils.py
|
[FSDP] ufmt /fsdp (#87811)
|
2022-10-27 04:25:55 +00:00 |
|
_state_dict_utils.py
|
Adopt state_dict_pre_hook in FSDP (#90436)
|
2022-12-11 03:54:26 +00:00 |
|
_trace_utils.py
|
[FSDP] Slightly refactor fx symbolic tracer (#89917)
|
2022-11-30 20:31:55 +00:00 |
|
_unshard_param_utils.py
|
[FSDP][state_dict][4/N] Move the core logic of summon full parameters to _unshard_params_utils.py (#88636)
|
2022-11-11 18:30:57 +00:00 |
|
_utils.py
|
Deprecate TypedStorage, its derived classes, and all of their public methods (#85303)
|
2022-11-08 18:11:01 +00:00 |
|
_wrap_utils.py
|
Fix fully_shard error when policy is not provided (#90151)
|
2022-12-05 15:21:47 +00:00 |
|
api.py
|
[FSDP] Allow MixedPrecision to skip inputs (#90620)
|
2022-12-11 06:39:38 +00:00 |
|
flat_param.py
|
[Reland][FSDP] Another fix for DTensor, use_orig_params=True (#90562)
|
2022-12-10 22:50:30 +00:00 |
|
fully_sharded_data_parallel.py
|
Adopt state_dict_pre_hook in FSDP (#90436)
|
2022-12-11 03:54:26 +00:00 |
|
sharded_grad_scaler.py
|
[FSDP] ufmt /fsdp (#87811)
|
2022-10-27 04:25:55 +00:00 |
|
wrap.py
|
[FSDP] Include module classes in ModuleWrapPolicy.__repr__ (#89058)
|
2022-11-30 02:27:02 +00:00 |
7ec1cb8553