1# mypy: allow-untyped-defs 2from typing_extensions import deprecated 3 4from torch.nn.parallel.data_parallel import data_parallel, DataParallel 5from torch.nn.parallel.distributed import DistributedDataParallel 6from torch.nn.parallel.parallel_apply import parallel_apply 7from torch.nn.parallel.replicate import replicate 8from torch.nn.parallel.scatter_gather import gather, scatter 9 10 11__all__ = [ 12 "replicate", 13 "scatter", 14 "parallel_apply", 15 "gather", 16 "data_parallel", 17 "DataParallel", 18 "DistributedDataParallel", 19] 20 21 22@deprecated( 23 "`torch.nn.parallel.DistributedDataParallelCPU` is deprecated, " 24 "please use `torch.nn.parallel.DistributedDataParallel` instead.", 25 category=FutureWarning, 26) 27class DistributedDataParallelCPU(DistributedDataParallel): 28 pass 29