two pass to infer shape (#1200)

Former-commit-id: ece6957b

two pass to infer shape (#1200)
Former-commit-id: ece6957b
dd9be365 · Li Xinqi · Jinhui Yuan · 27630a89 · dd9be365
隐藏空白更改
内联并排

Showing with 6 addition and 10 deletion

oneflow/core/job/compiler.cpp oneflow/core/job/compiler.cpp +6 -10

未找到文件。
--- a/oneflow/core/job/compiler.cpp
+++ b/oneflow/core/job/compiler.cpp
@@ -101,16 +101,12 @@ Plan Compiler::DoCompile() {
  task_gph->ForEachNode(std::bind(&TaskNode::ProduceAllRegstsAndBindEdges, _1));
  task_gph->ForEachNode(std::bind(&TaskNode::ConsumeAllRegsts, _1));
  task_gph->ForEachNode(std::bind(&TaskNode::PinConsumedRegst, _1));
-  if (job_desc->IsTrain()) {
-    task_gph->AcyclicTopoForEachNode([](TaskNode* node) { node->Build(); });
-  } else {
-    task_gph->AcyclicTopoForEachNode([](TaskNode* node) {
-      if (node->GetTaskType() != kNormalMdUpdt) { node->Build(); }
-    });
-    task_gph->AcyclicTopoForEachNode([](TaskNode* node) {
-      if (node->GetTaskType() == kNormalMdUpdt) { node->Build(); }
-    });
-  }
+  task_gph->AcyclicTopoForEachNode([](TaskNode* node) {
+    if (node->GetTaskType() != kNormalMdUpdt) { node->Build(); }
+  });
+  task_gph->AcyclicTopoForEachNode([](TaskNode* node) {
+    if (node->GetTaskType() == kNormalMdUpdt) { node->Build(); }
+  });
  task_gph->RemoveEmptyRegsts();
  task_gph->AddOrderingCtrlEdgeInSameChain();
  if (job_desc->IsTrain() && job_desc->enable_mem_sharing()) {