提交 f6496272 编写于 作者: D dongzhihong

relauch ci

上级 800085fe
...@@ -144,24 +144,17 @@ static std::unique_ptr<OperatorBase> BackwardRecursive( ...@@ -144,24 +144,17 @@ static std::unique_ptr<OperatorBase> BackwardRecursive(
// //
// one variable is shared between multiple operators. // one variable is shared between multiple operators.
// insert add operator one by one, then add it to output // insert add operator one by one, then add it to output
if (dup_outputs.size() == 2) {
insert_position.push_back(
{dup_op.back(),
OpRegistry::CreateOp(
"add", {{"X", {dup_outputs[0]}}, {"Y", {dup_outputs[1]}}},
{{"Out", {name}}}, {})});
} else {
for (size_t output_idx = 0; output_idx < dup_outputs.size() - 1; for (size_t output_idx = 0; output_idx < dup_outputs.size() - 1;
++output_idx) { ++output_idx) {
auto insert_add_x = dup_outputs[output_idx]; auto insert_add_x = dup_outputs[output_idx];
auto insert_add_y = dup_outputs[output_idx]; auto insert_add_y = dup_outputs[output_idx];
auto insert_add_out = name + "@SHARED@" + std::to_string(output_idx); auto insert_add_out = name + "@SHARED@" + std::to_string(output_idx);
// first add op inserted // first add op inserted
if (output_idx == dup_outputs.size() - 1) { if (output_idx == dup_outputs.size() - 2) {
insert_add_out = name; insert_add_out = name;
} }
if (output_idx != 0) { if (output_idx != 0) {
insert_add_y = name + "@SHARED@" + std::to_string(output_idx-1); insert_add_y = name + "@SHARED@" + std::to_string(output_idx - 1);
} }
insert_position.push_back( insert_position.push_back(
{dup_op.back(), {dup_op.back(),
...@@ -170,7 +163,6 @@ static std::unique_ptr<OperatorBase> BackwardRecursive( ...@@ -170,7 +163,6 @@ static std::unique_ptr<OperatorBase> BackwardRecursive(
{{"Out", {insert_add_out}}}, {})}); {{"Out", {insert_add_out}}}, {})});
} }
} }
}
// make sure the inserted `add` ops follow the BFS order. // make sure the inserted `add` ops follow the BFS order.
insert_position.sort( insert_position.sort(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册