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 37 Packages Projects Releases Wiki Activity
a0207c8471
pytorch/test/distributed/checkpoint
History
Chien-Chin Huang 1d9fefff19 [DCP] Fixes the stateless optimizer issue of distributed state_dict (#135535) 
Some optimizers don't have states that can cause get_state_dict/set_state_dict behave incorrectly. This PR fixes the issues.

fixes: https://github.com/pytorch/pytorch/issues/133415

Pull Request resolved: https://github.com/pytorch/pytorch/pull/135535
Approved by: https://github.com/wz337
2024-09-10 03:10:00 +00:00
..
e2e Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
fsdp [DTensor] Extend implicit replication to replicate DTensor for foreach ops so model doesn't have to be fully tp-ed when using 2D (#134551) 2024-08-29 09:01:31 +00:00
test_checkpoint.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_compatibility.py [DCP] Fixes the BC issue where the traversal doesn't support versions before 2.4 (#134158) 2024-08-28 16:31:44 +00:00
test_dedup_tensors.py [DCP][BE] Apply ufmt to DCP and turn on lintrunner for DCP (#115302) 2023-12-13 10:32:36 +00:00
test_dtensor_checkpoint.py [DCP][BE] Apply ufmt to DCP and turn on lintrunner for DCP (#115302) 2023-12-13 10:32:36 +00:00
test_dtensor_resharding.py
test_file_system_checkpoint_cpu.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_file_system_checkpoint.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_format_utils.py Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
test_fsdp_model_state.py [Distributed] Change function call in test to non-deprecated to eliminate warning (#134938) 2024-09-06 03:25:09 +00:00
test_fsdp_optim_state.py Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
test_fsdp_tp_checkpoint_conversion.py [Distributed] Change function call in test to non-deprecated to eliminate warning (#134938) 2024-09-06 03:25:09 +00:00
test_fsspec.py Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
test_hsdp_checkpoint.py [Distributed] Change function call in test to non-deprecated to eliminate warning (#134938) 2024-09-06 03:25:09 +00:00
test_nested_dict.py [DCP] Always flatten mapping even if no tensors present (#125335) 2024-05-07 17:08:49 +00:00
test_planner.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_save_load_api.py Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
test_state_dict_utils.py [reland][dtensor] move DTensor to public namespace (#134203) 2024-09-08 17:08:40 +00:00
test_state_dict.py [DCP] Fixes the stateless optimizer issue of distributed state_dict (#135535) 2024-09-10 03:10:00 +00:00
test_tp_checkpoint.py [DCP] Fix meta tensor loading (#133256) 2024-08-14 21:07:11 +00:00
test_traverse.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_utils.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00