pytorch/torch/distributed/checkpoint
2025-10-30 12:22:25 +00:00
..
_experimental Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
examples Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
__init__.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
_async_executor.py [DCP][OSS] Rank local checkpointing in DCP without collectives (#147758) 2025-08-13 16:20:28 +00:00
_async_process_executor.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
_async_thread_executor.py [DCP][OSS] Rank local checkpointing in DCP without collectives (#147758) 2025-08-13 16:20:28 +00:00
_checkpointer.py [3/N] fix typo in other folders (#166606) 2025-10-30 10:30:40 +00:00
_consolidate_hf_safetensors.py Add pg argument to consolidate_safetensors_files_on_every_rank (#161421) 2025-08-29 13:31:11 +00:00
_dedup_save_plans.py [distributed] Replace assert statements in distributed checkpoint with explicit checks (#165256) 2025-10-17 20:14:35 +00:00
_dedup_tensors.py
_extension.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
_fsspec_filesystem.py [distributed] Replace assert statements in distributed checkpoint with explicit checks (#165256) 2025-10-17 20:14:35 +00:00
_hf_utils.py Use only safetensors APIs in HFStorageReader (#159681) 2025-08-07 17:23:03 +00:00
_nested_dict.py
_pg_transport.py Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
_sharded_tensor_utils.py
_state_dict_stager.py state dict staging fixes (#166025) 2025-10-22 23:32:41 +00:00
_storage_utils.py
_traverse.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
_version.py
api.py
default_planner.py [distributed] Replace assert statements in distributed checkpoint with explicit checks (#165256) 2025-10-17 20:14:35 +00:00
filesystem.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
format_utils.py [3/N] fix typo in other folders (#166606) 2025-10-30 10:30:40 +00:00
hf_storage.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
logger.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
logging_handlers.py
metadata.py [oss] Add version to metadata (#155343) 2025-07-07 20:57:30 +00:00
optimizer.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
planner_helpers.py [3/N] Import Callable from collections.abc in torch/distributed (#164104) 2025-09-30 00:28:53 +00:00
planner.py
quantized_hf_storage.py [distributed] Replace assert statements in distributed checkpoint with explicit checks (#165256) 2025-10-17 20:14:35 +00:00
resharding.py
staging.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
state_dict_loader.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
state_dict_saver.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
state_dict.py [1/N] Remove unused loop variables (#166258) 2025-10-30 12:22:25 +00:00
stateful.py
storage.py [DCP][OSS] Rank local checkpointing in DCP without collectives (#147758) 2025-08-13 16:20:28 +00:00
utils.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00