Skip to content

Commit 694b9f9

Browse files
authored
[train] Document ray.train.collective (#54340)
Signed-off-by: Timothy Seah <[email protected]> Co-authored-by: Timothy Seah <[email protected]>
1 parent 9e90f97 commit 694b9f9

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

doc/source/train/api/api.rst

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -147,6 +147,14 @@ Ray Train Utilities
147147
~train.get_dataset_shard
148148
~train.report
149149

150+
**Collective**
151+
152+
.. autosummary::
153+
:nosignatures:
154+
:toctree: doc/
155+
156+
~train.collective.barrier
157+
~train.collective.broadcast_from_rank_zero
150158

151159
Ray Train Output
152160
----------------

0 commit comments

Comments
 (0)