提交 8bda4ab2 编写于 作者: Q Qiao Longfei

parameter recv can run

上级 fbd186bd
...@@ -518,13 +518,12 @@ class DistributeTranspiler(object): ...@@ -518,13 +518,12 @@ class DistributeTranspiler(object):
self._update_remote_sparse_update_op( self._update_remote_sparse_update_op(
param_varname, height_sections, eps, table_names) param_varname, height_sections, eps, table_names)
else: else:
all_recv_outputs.extend(splited_var)
recv_varnames = [] recv_varnames = []
if self.config.runtime_split_send_recv: if self.config.runtime_split_send_recv:
orig_param = program.global_block().vars[param_varname] orig_param = program.global_block().vars[param_varname]
recv_varnames = [var.name for var in splited_vars] recv_varnames = [var.name for var in splited_var]
splited_var = [orig_param] splited_var = [orig_param]
all_recv_outputs.extend(splited_var)
program.global_block().append_op( program.global_block().append_op(
type="recv", type="recv",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册