pytorch/test/distributed/fsdp
2025-05-09 20:40:54 +00:00
..
test_checkpoint_wrapper.py
test_distributed_checkpoint.py
test_fsdp_apply.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
test_fsdp_backward_prefetch.py
test_fsdp_checkpoint.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
test_fsdp_clip_grad_norm.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
test_fsdp_comm_hooks.py
test_fsdp_comm.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
test_fsdp_core.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
test_fsdp_dtensor_state_dict.py [BE][DTensor] move torch.distributed._tensor import to torch.distributed.tensor in test files (#153225) 2025-05-09 20:40:54 +00:00
test_fsdp_exec_order.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
test_fsdp_fine_tune.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
test_fsdp_flatten_params.py
test_fsdp_freezing_weights.py
test_fsdp_fx.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
test_fsdp_grad_acc.py [ROCm] Enable several fsdp related UTs (#149369) 2025-03-31 16:15:57 +00:00
test_fsdp_hybrid_shard.py
test_fsdp_ignored_modules.py
test_fsdp_input.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
test_fsdp_memory.py
test_fsdp_meta.py
test_fsdp_misc.py
test_fsdp_mixed_precision.py
test_fsdp_multiple_forward.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
test_fsdp_multiple_wrapping.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
test_fsdp_optim_state.py Enable ruff F841 on distributed tests (#146131) 2025-02-01 03:06:16 +00:00
test_fsdp_overlap.py
test_fsdp_pure_fp16.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
test_fsdp_sharded_grad_scaler.py
test_fsdp_state_dict.py
test_fsdp_tp_integration.py [BE][DTensor] move torch.distributed._tensor import to torch.distributed.tensor in test files (#153225) 2025-05-09 20:40:54 +00:00
test_fsdp_traversal.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
test_fsdp_uneven.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
test_fsdp_unshard_params.py
test_fsdp_use_orig_params.py
test_hsdp_dtensor_state_dict.py [BE][DTensor] move torch.distributed._tensor import to torch.distributed.tensor in test files (#153225) 2025-05-09 20:40:54 +00:00
test_shard_utils.py
test_utils.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
test_wrap.py Fix bug in FSDP wrapped module with zero argument (#147771) 2025-02-26 01:40:53 +00:00