Saurabh Mishra b90503d9ae [DCP] Unit Test to validate the stateful and non-stateful loads (#139251) ... Summary: Unit Test to validate the stateful and non-stateful loads. This test is a follow up to the fix in [#138575](https://github.com/pytorch/pytorch/pull/138575) which addresses an issue in stateful dict's in-place updates in distributed checkpoint loading. Also, added additional code comments regarding the stateful and non-stateful loads. Test Plan: ``` buck2 test //caffe2/test/distributed/checkpoint/e2e:test_e2e_save_and_load ``` https://www.internalfb.com/intern/testinfra/testrun/8162774562859797 Differential Revision: D65188659 Pull Request resolved: https://github.com/pytorch/pytorch/pull/139251 Approved by: https://github.com/LucasLLC, https://github.com/fegin		2024-10-31 01:12:51 +00:00
..
examples
__init__.py
_checkpointer.py
_dedup_save_plans.py
_dedup_tensors.py
_fsspec_filesystem.py
_nested_dict.py
_sharded_tensor_utils.py
_storage_utils.py
_traverse.py
_version.py
api.py
default_planner.py
filesystem.py
format_utils.py
logger.py
logging_handlers.py
metadata.py
optimizer.py
planner.py
planner_helpers.py
resharding.py
staging.py
state_dict.py
state_dict_loader.py	[DCP] Unit Test to validate the stateful and non-stateful loads (#139251)	2024-10-31 01:12:51 +00:00
state_dict_saver.py
stateful.py
storage.py
utils.py