Google Git
Sign in
android / platform / external / pytorch / refs/heads/sdk-release / . / test / distributed / checkpoint
tree: 158ddc562d049544068396509b1c2d2f38a20e1a [path history] [tgz]
  1. e2e/
  2. fsdp/
  3. test_checkpoint.py
  4. test_compatibility.py
  5. test_dedup_tensors.py
  6. test_dtensor_checkpoint.py
  7. test_dtensor_resharding.py
  8. test_file_system_checkpoint.py
  9. test_file_system_checkpoint_cpu.py
  10. test_format_utils.py
  11. test_fsdp_model_state.py
  12. test_fsdp_optim_state.py
  13. test_fsdp_tp_checkpoint_conversion.py
  14. test_fsspec.py
  15. test_hsdp_checkpoint.py
  16. test_nested_dict.py
  17. test_planner.py
  18. test_save_load_api.py
  19. test_state_dict.py
  20. test_state_dict_utils.py
  21. test_tp_checkpoint.py
  22. test_traverse.py
  23. test_utils.py