Google Git
Sign in
android / platform / external / pytorch / e696fa1ee7 / . / test / distributed / checkpoint
tree: 3182f177ab93bec4517bd33897c1cbbf86197107 [path history] [tgz]
  1. e2e/
  2. test_checkpoint.py
  3. test_compatibility.py
  4. test_dedup_tensors.py
  5. test_dtensor_checkpoint.py
  6. test_dtensor_resharding.py
  7. test_file_system_checkpoint.py
  8. test_file_system_checkpoint_cpu.py
  9. test_fsdp_model_state.py
  10. test_fsdp_optim_state.py
  11. test_fsdp_tp_checkpoint_conversion.py
  12. test_fsspec.py
  13. test_hsdp_checkpoint.py
  14. test_nested_dict.py
  15. test_planner.py
  16. test_state_dict.py
  17. test_state_dict_utils.py
  18. test_torch_save_to_dcp.py
  19. test_tp_checkpoint.py
  20. test_traverse.py
  21. test_utils.py