Google Git
Sign in
android / platform / external / pytorch / 35aeb6aa8533cf9bb88c0f8375b9ba8be20b2425 / . / test / distributed / fsdp
tree: fb4f1ce30cd865ac24942a8cfeb8fbf12228d4b5 [path history] [tgz]
  1. test_checkpoint_wrapper.py
  2. test_distributed_checkpoint.py
  3. test_fsdp_apply.py
  4. test_fsdp_backward_prefetch.py
  5. test_fsdp_checkpoint.py
  6. test_fsdp_clip_grad_norm.py
  7. test_fsdp_comm.py
  8. test_fsdp_comm_hooks.py
  9. test_fsdp_core.py
  10. test_fsdp_dtensor_state_dict.py
  11. test_fsdp_exec_order.py
  12. test_fsdp_fine_tune.py
  13. test_fsdp_flatten_params.py
  14. test_fsdp_freezing_weights.py
  15. test_fsdp_fx.py
  16. test_fsdp_grad_acc.py
  17. test_fsdp_hybrid_shard.py
  18. test_fsdp_ignored_modules.py
  19. test_fsdp_input.py
  20. test_fsdp_memory.py
  21. test_fsdp_meta.py
  22. test_fsdp_misc.py
  23. test_fsdp_mixed_precision.py
  24. test_fsdp_multiple_forward.py
  25. test_fsdp_multiple_wrapping.py
  26. test_fsdp_optim_state.py
  27. test_fsdp_overlap.py
  28. test_fsdp_pure_fp16.py
  29. test_fsdp_sharded_grad_scaler.py
  30. test_fsdp_state_dict.py
  31. test_fsdp_tp_integration.py
  32. test_fsdp_traversal.py
  33. test_fsdp_uneven.py
  34. test_fsdp_unshard_params.py
  35. test_fsdp_use_orig_params.py
  36. test_hsdp_dtensor_state_dict.py
  37. test_shard_utils.py
  38. test_utils.py
  39. test_wrap.py