frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "list_stored_sd_metadata API. (#160610)"

Browse Source 
This reverts commit da903b6a8be422529d47649e89c0d50bb95c37ca.

Reverted https://github.com/pytorch/pytorch/pull/160610 on behalf of https://github.com/jeffdaily due to broke ROCm CI, but flaky also on CUDA CI https://hud.pytorch.org/failure?name=periodic%20%2F%20linux-jammy-rocm-py3.10%20%2F%20test%20(distributed%2C%202%2C%203%2C%20linux.rocm.gpu.mi250.4%2C%20module%3Arocm%2C%20oncall%3Adistributed)&jobName=undefined&failureCaptures=distributed%2Fcheckpoint%2Ftest_list_stored_state_dict.py%3A%3ATestListStateDict%3A%3Atest_list_stored_sd_metadata ([comment](https://github.com/pytorch/pytorch/pull/160610#issuecomment-3382023022))
This commit is contained in:
PyTorch MergeBot
2025-10-08 15:10:38 +00:00
parent b5e93ffdcf
commit fd4bde430a
4 changed files with 0 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
docs/source/distributed.checkpoint.md
View File

@ -64,16 +64,6 @@ The entrypoints to load and save a checkpoint are the following:
.. autofunction:: load_state_dict
```
Following APIs can be used for inspecting the metadata of a checkpoint/stored state-dict.
```{eval-rst}
.. automodule:: torch.distributed.checkpoint.metadata_utils
```
```{eval-rst}
.. autofunction:: list_stored_state_dict
```
The following module is also useful for additional customization of the staging mechanisms used for asynchronous checkpointing (`torch.distributed.checkpoint.async_save`):
```{eval-rst}