Merge pull request #10247 from putcn/fix-transpiler

add missing grad_to_block_id

Merge pull request #10247 from putcn/fix-transpiler
add missing grad_to_block_id
87d86ee3 · Wu Yi · GitHub · e498e1fc · b1334259 · 87d86ee3
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

python/paddle/fluid/distribute_transpiler.py python/paddle/fluid/distribute_transpiler.py +2 -1

未找到文件。
--- a/python/paddle/fluid/distribute_transpiler.py
+++ b/python/paddle/fluid/distribute_transpiler.py
@@ -444,7 +444,8 @@ class DistributeTranspiler:
            opt_state_block = pserver_program.create_block(
                pserver_program.num_blocks - 1)
            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:
        #