From 600f5f0b49264f44deba3fb07f63e0c3acfb12d6 Mon Sep 17 00:00:00 2001 From: qiaolongfei Date: Fri, 13 Jul 2018 10:52:31 +0800 Subject: [PATCH] simpilify distribute transpiler --- .../fluid/transpiler/distribute_transpiler.py | 16 ++-------------- 1 file changed, 2 insertions(+), 14 deletions(-) diff --git a/python/paddle/fluid/transpiler/distribute_transpiler.py b/python/paddle/fluid/transpiler/distribute_transpiler.py index 92cdff04a00..121c36e4773 100644 --- a/python/paddle/fluid/transpiler/distribute_transpiler.py +++ b/python/paddle/fluid/transpiler/distribute_transpiler.py @@ -1284,20 +1284,8 @@ class DistributeTranspiler(object): # If one op's input is another op's output or # one op's output is another op's input, we say # the two operator is connected. - def _append_inname(varname_list): - op_input_names = [] - for in_name in varname_list: - op_input_names.append(in_name) - return op_input_names - - op1_input_names = _append_inname(op1.desc.input_arg_names()) - op1_output_names = op1.desc.output_arg_names() - - op2_input_names = _append_inname(op2.desc.input_arg_names()) - op2_output_names = op2.desc.output_arg_names() - - if set(op1_output_names) & set(op2_input_names) or \ - set(op1_input_names) & set(op2_output_names): + if set(op1.desc.output_arg_names()) & set(op2.desc.input_arg_names()) or \ + set(op1.desc.input_arg_names()) & set(op2.desc.output_arg_names()): return True return False -- GitLab