Merge pull request #11839 from velconia/fix_reader_op_in_pserver

Do NOT clone input vars if op NOT in pserver_program
revert-12383-port_py3_syntax
Qiyang Min 7 years ago committed by GitHub
commit 201928d005
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -309,10 +309,10 @@ class DistributeTranspiler(object):
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 @@ class DistributeTranspiler(object):
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 @@ class DistributeTranspiler(object):
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"
]:

Loading…
Cancel
Save