[neurun] Fix Dfs Iterator (#3721)

As mentioned from #3617 by @d-poshshoev, Visit checking assumed ID range
as vector which is wrong. This commit changes variable `visit` to be a
map so it can correctly check nodes for sure.

Signed-off-by: Hanjoung Lee <hanjoung.lee@samsung.com>

diff --git a/runtimes/neurun/src/graph/Graph.cc b/runtimes/neurun/src/graph/Graph.cc
index aa26e92..33aa25c 100644 (file)
--- a/runtimes/neurun/src/graph/Graph.cc
+++ b/runtimes/neurun/src/graph/Graph.cc
@@ -281,13 +281,15 @@ void Graph::PostDfsIterator<is_const>::iterate(GraphRef graph, const IterFn &fn)
 {
   assert(!graph.isBuildingPhase()); // Restrict iteration condition
 
-  std::vector<bool> visited(graph.operations().size(), false);
+  std::unordered_map<operation::Index, bool> visited;
+  graph.operations().iterate(
+      [&](const operation::Index &index, NodeRef) { visited[index] = false; });
 
   std::function<void(const operation::Index &, NodeRef)> dfs_recursive =
       [&](const operation::Index &index, NodeRef node) -> void {
-    if (visited[index.asInt()])
+    if (visited[index])
       return;
-    visited[index.asInt()] = true;
+    visited[index] = true;
 
     for (auto output : node.getOutputs())
     {
@@ -304,7 +306,8 @@ void Graph::PostDfsIterator<is_const>::iterate(GraphRef graph, const IterFn &fn)
   graph.operations().iterate(dfs_recursive);
 
   // All of the operations(nodes) must have been visited.
-  assert(std::all_of(visited.begin(), visited.end(), [](bool v) { return v; }));
+  assert(std::all_of(visited.begin(), visited.end(),
+                     [](const std::pair<const operation::Index, bool> &v) { return v.second; }));
 }
 
 } // namespace graph