Merge pull request #11839 from velconia/fix_reader_op_in_pserver

Do NOT clone input vars if op NOT in pserver_program

Author: velconia

python/paddle/fluid/transpiler/distribute_transpiler.py

@@ -309,10 +309,10 @@ def get_trainer_program(self):
     def get_pserver_program(self, endpoint):
         """
         Get parameter server side program.
-        
+
         Args:
             endpoint (str): current parameter server endpoint.
-        
+
         Returns:
             Program: the program for current parameter server to run.
         """
@@ -516,7 +516,7 @@ def get_startup_program(self, endpoint, pserver_program):
             endpoint (str): current pserver endpoint.
             pserver_program (Program): call get_pserver_program first and
                 pass the result here.
-        
+
         Returns:
             Program: parameter server side startup program.
         """
@@ -552,10 +552,10 @@ def _get_splited_name_and_shape(varname):
                     op_on_pserver = True
                     new_outputs[key] = pserver_vars[op.output(key)[0]]
 
-            # most startup program ops have no inputs
-            new_inputs = self._get_input_map_from_op(pserver_vars, op)
-
             if op_on_pserver:
+                # most startup program ops have no inputs
+                new_inputs = self._get_input_map_from_op(pserver_vars, op)
+
                 if op.type in [
                         "gaussian_random", "fill_constant", "uniform_random"
                 ]: