mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
b5c006acac
Part of #123062 - #123062 Pull Request resolved: https://github.com/pytorch/pytorch/pull/128865 Approved by: https://github.com/ezyang
28 lines
787 B
Python
28 lines
787 B
Python
# mypy: allow-untyped-defs
|
|
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
|