xref: /aosp_15_r20/external/pytorch/torch/cuda/comm.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 # The functions here have been moved to torch.nn.parallel.comm
2 from torch.nn.parallel.comm import (
3     broadcast,
4     broadcast_coalesced,
5     gather,
6     reduce_add,
7     reduce_add_coalesced,
8     scatter,
9 )
10 
11 
12 __all__ = [
13     "broadcast",
14     "broadcast_coalesced",
15     "reduce_add",
16     "reduce_add_coalesced",
17     "scatter",
18     "gather",
19 ]
20