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