mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
3bf922a6ce
Signed-off-by: Edward Z. Yang <ezyang@meta.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/106249 Approved by: https://github.com/Skylion007
19 lines
343 B
Python
19 lines
343 B
Python
# The functions here have been moved to torch.nn.parallel.comm
|
|
from torch.nn.parallel.comm import (
|
|
broadcast,
|
|
broadcast_coalesced,
|
|
gather,
|
|
reduce_add,
|
|
reduce_add_coalesced,
|
|
scatter,
|
|
)
|
|
|
|
__all__ = [
|
|
"broadcast",
|
|
"broadcast_coalesced",
|
|
"reduce_add",
|
|
"reduce_add_coalesced",
|
|
"scatter",
|
|
"gather",
|
|
]
|