saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test/distributed/fsdp
History
Aaron Orenstein 99dbc5b0e2 PEP585 update - test (#145176) 
See #145101 for details.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/145176
Approved by: https://github.com/bobrenjc93
2025-01-22 04:48:28 +00:00
..
test_checkpoint_wrapper.py Tests Generelization for multiple accelerator devices (#139184) 2025-01-07 09:04:38 +00:00
test_distributed_checkpoint.py [CI][CUDA][Distributed][FSDP] Remove hardcoded world size of 2 (#145195) 2025-01-21 20:25:52 +00:00
test_fsdp_apply.py [CI][CUDA][Distributed][FSDP] Remove hardcoded world size of 2 (#145195) 2025-01-21 20:25:52 +00:00
test_fsdp_backward_prefetch.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_fsdp_checkpoint.py Tests Generelization for multiple accelerator devices (#139184) 2025-01-07 09:04:38 +00:00
test_fsdp_clip_grad_norm.py Tests Generelization for multiple accelerator devices (#139184) 2025-01-07 09:04:38 +00:00
test_fsdp_comm.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_fsdp_comm_hooks.py
test_fsdp_core.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_fsdp_dtensor_state_dict.py Tests Generelization for multiple accelerator devices (#139184) 2025-01-07 09:04:38 +00:00
test_fsdp_exec_order.py Tests Generelization for multiple accelerator devices (#139184) 2025-01-07 09:04:38 +00:00
test_fsdp_fine_tune.py Tests Generelization for multiple accelerator devices (#139184) 2025-01-07 09:04:38 +00:00
test_fsdp_flatten_params.py
test_fsdp_freezing_weights.py
test_fsdp_fx.py Tests Generelization for multiple accelerator devices (#139184) 2025-01-07 09:04:38 +00:00
test_fsdp_grad_acc.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_fsdp_hybrid_shard.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_fsdp_ignored_modules.py
test_fsdp_input.py Tests Generelization for multiple accelerator devices (#139184) 2025-01-07 09:04:38 +00:00
test_fsdp_memory.py Tests Generelization for multiple accelerator devices (#139184) 2025-01-07 09:04:38 +00:00
test_fsdp_meta.py
test_fsdp_misc.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_fsdp_mixed_precision.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_fsdp_multiple_forward.py Tests Generelization for multiple accelerator devices (#139184) 2025-01-07 09:04:38 +00:00
test_fsdp_multiple_wrapping.py Tests Generelization for multiple accelerator devices (#139184) 2025-01-07 09:04:38 +00:00
test_fsdp_optim_state.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_fsdp_overlap.py Tests Generelization for multiple accelerator devices (#139184) 2025-01-07 09:04:38 +00:00
test_fsdp_pure_fp16.py Tests Generelization for multiple accelerator devices (#139184) 2025-01-07 09:04:38 +00:00
test_fsdp_sharded_grad_scaler.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_fsdp_state_dict.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_fsdp_tp_integration.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_fsdp_traversal.py [CI][CUDA][Distributed][FSDP] Remove hardcoded world size of 2 (#145195) 2025-01-21 20:25:52 +00:00
test_fsdp_uneven.py Tests Generelization for multiple accelerator devices (#139184) 2025-01-07 09:04:38 +00:00
test_fsdp_unshard_params.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_fsdp_use_orig_params.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_hsdp_dtensor_state_dict.py Tests Generelization for multiple accelerator devices (#139184) 2025-01-07 09:04:38 +00:00
test_shard_utils.py
test_utils.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_wrap.py