|
|
7f4c3e7d2f
|
distributed/serialization: support zero sized tensors (#164198)
Fixes
```
[4] ValueError: both buffer length (0) and count (-1) must not be 0
```
Test plan:
```
pytest test/distributed/test_serialization.py
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/164198
Approved by: https://github.com/amirafzali
|
2025-09-30 08:11:29 +00:00 |
|
|
|
4ccc0381de
|
[BE][5/16] fix typos in torch/ (torch/distributed/) (#156315)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/156315
Approved by: https://github.com/Skylion007, https://github.com/albanD
ghstack dependencies: #156313, #156314
|
2025-06-23 02:57:28 +00:00 |
|
|
|
145d4cdc11
|
Revert "[BE][5/16] fix typos in torch/ (torch/distributed/) (#156315)"
This reverts commit c2f0292bd5b4b3206f5b295e96f81cd6c178eb18.
Reverted https://github.com/pytorch/pytorch/pull/156315 on behalf of https://github.com/atalman due to export/test_torchbind.py::TestCompileTorchbind::test_compile_error_on_input_aliasing_contents_backend_aot_eager [GH job link](https://github.com/pytorch/pytorch/actions/runs/15804799771/job/44548489912) [HUD commit link](c95f7fa874) ([comment](https://github.com/pytorch/pytorch/pull/156313#issuecomment-2994171213))
|
2025-06-22 12:31:57 +00:00 |
|
|
|
c2f0292bd5
|
[BE][5/16] fix typos in torch/ (torch/distributed/) (#156315)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/156315
Approved by: https://github.com/Skylion007, https://github.com/albanD
ghstack dependencies: #156313, #156314
|
2025-06-22 08:43:26 +00:00 |
|
|
|
db4ce78d46
|
PEP585: More UP006 fixes (#146392)
This should be the final PR before we can enable RUFF UP006.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/146392
Approved by: https://github.com/justinchuby, https://github.com/albanD, https://github.com/Skylion007
|
2025-02-20 06:18:13 +00:00 |
|
|
|
23af9dde4d
|
distributed/serialization: add experimental streaming torch.save/load methods (#146555)
Summary:
This is intended for use with torchft when we need to do a streaming state dict transfer. This is strictly superior to the prior streaming method in torchft as this supports all tensor subclasses such as DTensor.
This supports 100% of the inputs to torch.save/load but is not wire compatible nor intended to have any backwards compatibility.
Security wise this fully supports weights_only and defaults to True. It does use pickle for some metadata but uses weights_only for the metadata.
Adapted from:
https://github.com/pytorch/torchft/pull/101
https://github.com/pytorch/torchft/pull/54
Test Plan:
pytest test/distributed/test_serialization.py
Pull Request resolved: https://github.com/pytorch/pytorch/pull/146555
Approved by: https://github.com/fegin, https://github.com/mikaylagawarecki
Co-authored-by: Krishn Parasar <76171905+Krishn1412@users.noreply.github.com>
|
2025-02-07 18:08:11 +00:00 |
|
