6 lines
293 B
Python
6 lines
293 B
Python
# The functions here have been moved to torch.nn.parallel.comm
|
|
from torch.nn.parallel.comm import broadcast, broadcast_coalesced, reduce_add, \
|
|
reduce_add_coalesced, scatter, gather
|
|
|
|
__all__ = ['broadcast', 'broadcast_coalesced', 'reduce_add', 'reduce_add_coalesced', 'scatter', 'gather']
|