pytorch/torch/distributed/checkpoint
2025-07-25 02:56:34 +00:00
..
_experimental fix forced loglevel in pytorch oss code (#158820) 2025-07-24 00:40:28 +00:00
examples [DCP] OSS Zero Overhead Checkpointing Implementation (#156207) 2025-06-29 03:19:48 +00:00
__init__.py Clean up HF components (#155707) 2025-06-24 00:07:37 +00:00
_async_executor.py [DCP] OSS Zero Overhead Checkpointing Implementation (#156207) 2025-06-29 03:19:48 +00:00
_async_process_executor.py [DCP] OSS Zero Overhead Checkpointing Implementation (#156207) 2025-06-29 03:19:48 +00:00
_async_thread_executor.py Name threads in caffe2/torch/distributed/checkpoint AsyncCheckpointExecutor (#158612) 2025-07-18 17:33:12 +00:00
_checkpointer.py [DCP] OSS Zero Overhead Checkpointing Implementation (#156207) 2025-06-29 03:19:48 +00:00
_consolidate_hf_safetensors.py [oss][hf][bug fix] Remove buggy consolidation logic (#158380) 2025-07-17 13:05:06 +00:00
_dedup_save_plans.py Changes to HFStorageWriter to support saving shards of tensors (#154742) (#155566) 2025-06-10 23:37:47 +00:00
_dedup_tensors.py
_extension.py
_fsspec_filesystem.py [BE][5/16] fix typos in torch/ (torch/distributed/) (#156315) 2025-06-23 02:57:28 +00:00
_hf_utils.py DCP safetensors test fix (#158685) 2025-07-20 22:52:54 +00:00
_nested_dict.py
_pg_transport.py [DCP] Add support for ShardedTensor to PgTransport (#158573) 2025-07-21 21:04:23 +00:00
_sharded_tensor_utils.py
_state_dict_stager.py fix forced loglevel in pytorch oss code (#158820) 2025-07-24 00:40:28 +00:00
_storage_utils.py [BE][5/16] fix typos in torch/ (torch/distributed/) (#156315) 2025-06-23 02:57:28 +00:00
_traverse.py
_version.py
api.py
default_planner.py Clean up HF components (#155707) 2025-06-24 00:07:37 +00:00
filesystem.py [oss] Add version to metadata (#155343) 2025-07-07 20:57:30 +00:00
format_utils.py
hf_storage.py DCP safetensors test fix (#158685) 2025-07-20 22:52:54 +00:00
logger.py Revert "[BE]: Enable RUFF TRY400 rule - log.exception (#153473)" 2025-05-16 08:29:26 +00:00
logging_handlers.py
metadata.py [oss] Add version to metadata (#155343) 2025-07-07 20:57:30 +00:00
optimizer.py [BE][5/16] fix typos in torch/ (torch/distributed/) (#156315) 2025-06-23 02:57:28 +00:00
planner_helpers.py
planner.py
resharding.py remove allow-untyped-defs from torch/distributed/checkpoint/resharding.py (#154626) 2025-05-30 07:43:04 +00:00
staging.py [DCP] OSS Zero Overhead Checkpointing Implementation (#156207) 2025-06-29 03:19:48 +00:00
state_dict_loader.py [BE][5/16] fix typos in torch/ (torch/distributed/) (#156315) 2025-06-23 02:57:28 +00:00
state_dict_saver.py [BE][1/16] fix typos in torch/ (#156311) 2025-07-09 11:02:22 +00:00
state_dict.py [BE][5/16] fix typos in torch/ (torch/distributed/) (#156315) 2025-06-23 02:57:28 +00:00
stateful.py [BE]: Backport runtime_checkable perf improvements/behavior from 3.12 (#155130) 2025-06-06 13:28:05 +00:00
storage.py [BE][5/16] fix typos in torch/ (torch/distributed/) (#156315) 2025-06-23 02:57:28 +00:00
utils.py [BE] add noqa for flake8 rule B036: found except BaseException without re-raising (#159043) 2025-07-25 02:56:34 +00:00