You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
19 lines
343 B
19 lines
343 B
5 months ago
|
# 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",
|
||
|
|
]
|