diff --git a/python/paddle/v2/fluid/__init__.py b/python/paddle/v2/fluid/__init__.py index 5e01b8719806f4bb0c0d985373a5f4b076e05bd5..a14422ee92f753060c2a4d8868992e888e44f9e9 100644 --- a/python/paddle/v2/fluid/__init__.py +++ b/python/paddle/v2/fluid/__init__.py @@ -18,13 +18,14 @@ from param_attr import ParamAttr from data_feeder import DataFeeder from core import LoDTensor, CPUPlace, CUDAPlace from distribute_transpiler import DistributeTranspiler +from distribute_transpiler_simple import SimpleDistributeTranspiler import clip Tensor = LoDTensor __all__ = framework.__all__ + executor.__all__ + [ 'io', 'initializer', 'layers', 'nets', 'optimizer', 'backward', 'regularizer', 'LoDTensor', 'CPUPlace', 'CUDAPlace', 'Tensor', 'ParamAttr' - 'DataFeeder', 'clip', 'DistributeTranspiler' + 'DataFeeder', 'clip', 'SimpleDistributeTranspiler', 'DistributeTranspiler' ] diff --git a/python/paddle/v2/fluid/distribute_transpiler_simple.py b/python/paddle/v2/fluid/distribute_transpiler_simple.py index 49ece7b725e318d7526d58fe54c97cbe20200a7d..32db3df9aa2a9381d39809e5bcb18a558704c7ed 100644 --- a/python/paddle/v2/fluid/distribute_transpiler_simple.py +++ b/python/paddle/v2/fluid/distribute_transpiler_simple.py @@ -48,7 +48,7 @@ def round_robin(params_grads, pserver_endpoints): return param_grad_map -class DistributeTranspiler: +class SimpleDistributeTranspiler: def transpile(self, optimize_ops, params_grads,