Skip to content

Commit 28a86ae

Browse files
committed
Fix Parallel Exe(VarHandel's version)
1 parent 2fdbc1c commit 28a86ae

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

paddle/fluid/framework/details/multi_devices_graph_builder.cc

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -345,7 +345,7 @@ void MultiDevSSAGraphBuilder::InsertAllReduceOp(SSAGraph *result,
345345
auto &prev_grad = vars.back();
346346
op_handle->AddInput(prev_grad.get());
347347

348-
auto var = new VarHandle(vars.size() - 1, i, og, p);
348+
auto var = new VarHandle(vars.size(), i, og, p);
349349
vars.emplace_back(var);
350350
op_handle->AddOutput(var);
351351
}
@@ -442,8 +442,7 @@ VarHandle *MultiDevSSAGraphBuilder::CreateReduceOp(SSAGraph *result,
442442
op_handle->AddInput(prev_grad.get());
443443
}
444444
auto &vars = result->vars_[dst_dev_id][og];
445-
auto var =
446-
new VarHandle(vars.size() - 1, dst_dev_id, og, places_[dst_dev_id]);
445+
auto var = new VarHandle(vars.size(), dst_dev_id, og, places_[dst_dev_id]);
447446
vars.emplace_back(var);
448447
op_handle->AddOutput(var);
449448
return var;

0 commit comments

Comments
 (0)