| .. |
|
_experimental
|
[dcp_poc] add async checkpointing tests (#161034)
|
2025-08-21 00:08:53 +00:00 |
|
e2e
|
Fix test_fsdp_ep.py due to _MeshEnv API change (#158695)
|
2025-08-08 22:36:47 +00:00 |
|
fsdp
|
[BE][DTensor] move torch.distributed._tensor import to torch.distributed.tensor in test files (#153225)
|
2025-05-09 20:40:54 +00:00 |
|
test_async_process_executor.py
|
[DCP] Improve error handling for process based async checkpointing (#159374)
|
2025-07-30 16:25:28 +00:00 |
|
test_checkpoint.py
|
Revert "Enable all SIM rules except disabled ones (#164645)"
|
2025-10-05 19:32:21 +00:00 |
|
test_compatibility.py
|
|
|
|
test_consolidate_hf_safetensors.py
|
[dcp][hf] Fix multi-rank consolidation for no files to process case (#160660)
|
2025-08-21 21:18:03 +00:00 |
|
test_dedup_tensors.py
|
|
|
|
test_dtensor_checkpoint.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_dtensor_resharding.py
|
UT/Examples for resharding checkpoint save/loads for distributed tensors with uneven shards. (#160533)
|
2025-10-03 22:15:02 +00:00 |
|
test_file_system_checkpoint_cpu.py
|
|
|
|
test_file_system_checkpoint.py
|
|
|
|
test_format_utils.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_model_state.py
|
|
|
|
test_fsdp_optim_state.py
|
|
|
|
test_fsdp_tp_checkpoint_conversion.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_fsspec.py
|
[1/N] Apply UP035 rule in tests (#163947)
|
2025-09-29 01:42:01 +00:00 |
|
test_hf_safetensor_e2e.py
|
[DCP][HuggingFace] Add Support for dequantization of SafeTensors checkpoints (#160682)
|
2025-09-04 01:09:53 +00:00 |
|
test_hf_storage.py
|
Use only safetensors APIs in HFStorageReader (#159681)
|
2025-08-07 17:23:03 +00:00 |
|
test_hsdp_checkpoint.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_nested_dict.py
|
|
|
|
test_pg_transport.py
|
[C10D] Fix spelling of MultiProcContinuousTest (#160892)
|
2025-08-19 20:17:19 +00:00 |
|
test_planner.py
|
[oss] Add version to metadata (#155343)
|
2025-07-07 20:57:30 +00:00 |
|
test_quantized_hf_storage.py
|
[DCP] DTensor slice dequantization with proper block alignment (#163532)
|
2025-09-23 16:48:16 +00:00 |
|
test_save_load_api.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_state_dict_stager.py
|
[DCP][Prototype] Checkpoint replication via PGTransport (#157963) (#159801)
|
2025-08-06 16:52:03 +00:00 |
|
test_state_dict_utils.py
|
fix-unpin-memory-tensor-param (#160992)
|
2025-08-26 21:55:25 +00:00 |
|
test_state_dict.py
|
[1/N] Apply UP035 rule in tests (#163947)
|
2025-09-29 01:42:01 +00:00 |
|
test_tp_checkpoint.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_traverse.py
|
|
|
|
test_utils.py
|
[DCP][Ez]: Fix broadcast_object bug in DCP utils (#155912)
|
2025-06-14 12:14:14 +00:00 |
