“8f156b5e2e222167172f1f2dae32ead2adcf9001”上不存在“develop/api_doc/v2/data.html”
提交 b1cc1c3b 编写于 作者: S seiriosPlus

fix recv

上级 91f98315
...@@ -223,22 +223,6 @@ def init_from_server_pass(program, config): ...@@ -223,22 +223,6 @@ def init_from_server_pass(program, config):
fetch_barrier_out = program.global_block().create_var( fetch_barrier_out = program.global_block().create_var(
name=framework.generate_control_dev_var_name()) name=framework.generate_control_dev_var_name())
recv_ctx = config.get_communicator_recv_context(recv_type=1)
recv_varnames = []
for name, ctxs in recv_ctx.items():
recv_varnames.extend(ctxs.origin_varnames())
program.global_block().append_op(
type="recv",
inputs={"X": []},
outputs={"Out": []},
attrs={
"recv_varnames": recv_varnames,
"trainer_id": config.get_role_id(),
RPC_OP_ROLE_ATTR_NAME: RPC_OP_ROLE_ATTR_VALUE
})
program.global_block().append_op( program.global_block().append_op(
type="fetch_barrier", type="fetch_barrier",
inputs={}, inputs={},
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册