mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
03991798ca
Pull Request resolved: https://github.com/pytorch/pytorch/pull/143437 Approved by: https://github.com/oulgen
28 lines
760 B
Python
28 lines
760 B
Python
from typing_extensions import deprecated
|
|
|
|
from torch.nn.parallel.data_parallel import data_parallel, DataParallel
|
|
from torch.nn.parallel.distributed import DistributedDataParallel
|
|
from torch.nn.parallel.parallel_apply import parallel_apply
|
|
from torch.nn.parallel.replicate import replicate
|
|
from torch.nn.parallel.scatter_gather import gather, scatter
|
|
|
|
|
|
__all__ = [
|
|
"replicate",
|
|
"scatter",
|
|
"parallel_apply",
|
|
"gather",
|
|
"data_parallel",
|
|
"DataParallel",
|
|
"DistributedDataParallel",
|
|
]
|
|
|
|
|
|
@deprecated(
|
|
"`torch.nn.parallel.DistributedDataParallelCPU` is deprecated, "
|
|
"please use `torch.nn.parallel.DistributedDataParallel` instead.",
|
|
category=FutureWarning,
|
|
)
|
|
class DistributedDataParallelCPU(DistributedDataParallel):
|
|
pass
|