| .. |
|
e2e
|
Revert "[DCP] Makes fsspec public (#121508)"
|
2024-03-12 17:02:43 +00:00 |
|
fsdp
|
[FSDP2][DCP][DSD] Add FSDP2 model state dict unit test with distributed state dict (#120680)
|
2024-02-28 01:40:04 +00:00 |
|
test_checkpoint.py
|
[DCP] Enable filesystem/fsspec auto detection (#118888)
|
2024-02-08 16:38:04 +00:00 |
|
test_compatibility.py
|
[DCP] Removes no_dist and coordinator_rank from public DCP API's (#121317)
|
2024-03-08 02:14:12 +00:00 |
|
test_dedup_tensors.py
|
[DCP][BE] Apply ufmt to DCP and turn on lintrunner for DCP (#115302)
|
2023-12-13 10:32:36 +00:00 |
|
test_dtensor_checkpoint.py
|
[DCP][BE] Apply ufmt to DCP and turn on lintrunner for DCP (#115302)
|
2023-12-13 10:32:36 +00:00 |
|
test_dtensor_resharding.py
|
[DTensor] Remove assert to allow tensor sharding dimension < Shard(x).ndim (#115114)
|
2023-12-07 21:57:30 +00:00 |
|
test_file_system_checkpoint_cpu.py
|
Enables load balancing duplicates in DCP (#116469)
|
2024-01-26 22:34:14 +00:00 |
|
test_file_system_checkpoint.py
|
[DCP][BE] Apply ufmt to DCP and turn on lintrunner for DCP (#115302)
|
2023-12-13 10:32:36 +00:00 |
|
test_format_utils.py
|
[DCP] Adds storage reader and planner classes for online loading/sharding of models in torch.save format (#119816)
|
2024-03-01 00:21:05 +00:00 |
|
test_fsdp_model_state.py
|
[DCP][BE] Apply ufmt to DCP and turn on lintrunner for DCP (#115302)
|
2023-12-13 10:32:36 +00:00 |
|
test_fsdp_optim_state.py
|
[DCP] Add test for planner option for load_sharded_optimizer_state_dict (#112891)
|
2023-11-03 22:37:49 +00:00 |
|
test_fsdp_tp_checkpoint_conversion.py
|
[DCP][BE] Move DCP._state_dict_utils out from DCP (#115523)
|
2023-12-13 08:59:48 +00:00 |
|
test_fsspec.py
|
Revert "[DCP] Makes fsspec public (#121508)"
|
2024-03-12 17:02:43 +00:00 |
|
test_hsdp_checkpoint.py
|
[DCP] Add HSDP checkpiont unit tests (#111399)
|
2023-10-17 17:59:42 +00:00 |
|
test_nested_dict.py
|
[DCP][BE] Apply ufmt to DCP and turn on lintrunner for DCP (#115302)
|
2023-12-13 10:32:36 +00:00 |
|
test_planner.py
|
Enables load balancing duplicates in DCP (#116469)
|
2024-01-26 22:34:14 +00:00 |
|
test_save_load_api.py
|
[DCP] Enable filesystem/fsspec auto detection (#118888)
|
2024-02-08 16:38:04 +00:00 |
|
test_state_dict_utils.py
|
[DCP][state_dict] Let _offload_state_dict_to_cpu to return the companion_obj if it exist. (#121273)
|
2024-03-08 00:24:29 +00:00 |
|
test_state_dict.py
|
Revert "[DCP] Makes fsspec public (#121508)"
|
2024-03-12 17:02:43 +00:00 |
|
test_tp_checkpoint.py
|
[DCP] Adds support for meta tensor loading for DCP.load_state_dict() (#113319)
|
2024-01-17 00:23:29 +00:00 |
|
test_traverse.py
|
[DCP][BE] Apply ufmt to DCP and turn on lintrunner for DCP (#115302)
|
2023-12-13 10:32:36 +00:00 |
|
test_utils.py
|
[DCP][BE] Apply ufmt to DCP and turn on lintrunner for DCP (#115302)
|
2023-12-13 10:32:36 +00:00 |
0398dc9e8e