OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
cd68559d04
pytorch/docs/source/distributed._dist2.md
Tristan Rice 0d77364ee3 dist2: cleanup non-option methods on PG (missing, timeouts) (#158123) 
This updates the ProcessGroup.* API to include timeouts on all non-option based overloaded methods. This also adds 2 missing ones `alltoall_base` and `barrier`.

Following design in: https://docs.google.com/document/d/13R-1t_yESTvmAjcCN-wQjQQadIEu0JNIdS65uZawZzY/edit?tab=t.0#heading=h.3ctbqqopzc89

Test plan:

```
pytest test/distributed/test_dist2.py
```

Pull Request resolved: https://github.com/pytorch/pytorch/pull/158123
Approved by: https://github.com/Skylion007, https://github.com/fduwjj
2025-07-12 00:06:37 +00:00

210 B
Raw Blame History 

.. role:: hidden
    :class: hidden-section

.. automodule:: torch.distributed._dist2
    :members:
    :undoc-members:
    :show-inheritance:
    :exclude-members: ReduceOp