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