small fix.

XinYao1994 · May 22, 2018 · ae9340d · ae9340d
1 parent 471e6d2
commit ae9340d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/executor/graph_executor.cc b/src/executor/graph_executor.cc
@@ -1557,7 +1557,7 @@ GraphExecutor::CachedSegOpr GraphExecutor::CreateCachedSegOpr(size_t topo_start,
     if (inode.source->is_variable()) continue;
     // We shouldn't add control flow operators to a segment.
     // We can't execute these operators in the engine.
-    if (op_node.exec->HasSubgraph()) continue;
+    if (op_node.exec->HasSubgraph()) return ret;
     if (op_node.exec->exec_type() != ExecType::kSync) {
       return ret;
     }