未验证 提交 87d86ee3 编写于 作者: W Wu Yi 提交者: GitHub

Merge pull request #10247 from putcn/fix-transpiler

add missing grad_to_block_id
...@@ -444,7 +444,8 @@ class DistributeTranspiler: ...@@ -444,7 +444,8 @@ class DistributeTranspiler:
opt_state_block = pserver_program.create_block( opt_state_block = pserver_program.create_block(
pserver_program.num_blocks - 1) pserver_program.num_blocks - 1)
for glb_op in global_ops: for glb_op in global_ops:
__append_optimize_op__(glb_op, opt_state_block) __append_optimize_op__(glb_op, opt_state_block,
grad_to_block_id)
# NOT USED: single block version: # NOT USED: single block version:
# #
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册