Oneflow-Inc · chengtbf · Apr 28, 2022 · Apr 19, 2022 · Apr 19, 2022 · Apr 19, 2022
diff --git a/oneflow/core/graph/boxing/naive_b2p_sub_task_graph_builder.cpp b/oneflow/core/graph/boxing/naive_b2p_sub_task_graph_builder.cpp
@@ -62,7 +62,7 @@ Maybe<SubTskGphBuilderStatus> NaiveB2PSubTskGphBuilder::Build(
         zeros_node->Init(out_machine_id, thrd_id, lbi, logical_blob_desc.shape(),
                          logical_blob_desc.data_type(), time_shape);
         nearest_in_node->BuildCtrlRegstDesc(zeros_node);
-        Connect<TaskNode>(nearest_in_node, ctx->task_graph()->NewEdge(), zeros_node);
+        ctx->task_graph()->ConnectWithLbi(nearest_in_node, zeros_node, lbi);
         sorted_out_tasks->emplace_back(zeros_node);
       }
     }