xref: /aosp_15_r20/external/pytorch/torch/cuda/comm.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1# The functions here have been moved to torch.nn.parallel.comm
2from 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