From a4d88fb6420cba41c701378b1ed56fe4f9ec45f1 Mon Sep 17 00:00:00 2001 From: minqiyang Date: Thu, 31 May 2018 11:41:08 +0800 Subject: [PATCH] 1. remove simple distranspiler from transpiler/ repo 2. remvoe comment line --- python/paddle/fluid/__init__.py | 4 ++-- .../fluid/tests/unittests/test_simple_dist_transpiler.py | 1 - 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/python/paddle/fluid/__init__.py b/python/paddle/fluid/__init__.py index 859605d00..d53a96a7a 100644 --- a/python/paddle/fluid/__init__.py +++ b/python/paddle/fluid/__init__.py @@ -44,8 +44,8 @@ import transpiler from param_attr import ParamAttr, WeightNormParamAttr from data_feeder import DataFeeder from core import LoDTensor, CPUPlace, CUDAPlace, CUDAPinnedPlace -from transpiler import DistributeTranspiler, SimpleDistributeTranspiler, \ - InferenceTranspiler, memory_optimize, release_memory +from transpiler import DistributeTranspiler, InferenceTranspiler, \ + memory_optimize, release_memory from concurrency import (Go, make_channel, channel_send, channel_recv, channel_close, Select) from lod_tensor import create_lod_tensor, create_random_int_lodtensor diff --git a/python/paddle/fluid/tests/unittests/test_simple_dist_transpiler.py b/python/paddle/fluid/tests/unittests/test_simple_dist_transpiler.py index d51e356a5..60f99f412 100644 --- a/python/paddle/fluid/tests/unittests/test_simple_dist_transpiler.py +++ b/python/paddle/fluid/tests/unittests/test_simple_dist_transpiler.py @@ -62,7 +62,6 @@ class TestSimpleDistTranspiler(unittest.TestCase): self.assertEqual([op.type for op in pserver.blocks[1].ops], ["sum", "scale", "sgd"]) - print("xxx", [op.output_arg_names for op in startup.global_block().ops]) # confirm startup program self.assertEqual([op.type for op in startup.global_block().ops], ["fill_constant", "uniform_random", "uniform_random"]) -- GitLab