[neurun] Workaround for operand::Set iteration (#3196)

`OperandPass` sometimes manipulate the set while iteration, this
workaround forces the `iterate` method to iterate only elements that
has been existed from the beginning of the iteration.

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

diff --git a/runtimes/neurun/src/graph/operand/Set.cc b/runtimes/neurun/src/graph/operand/Set.cc
index 60dad23..dff690a 100644 (file)
--- a/runtimes/neurun/src/graph/operand/Set.cc
+++ b/runtimes/neurun/src/graph/operand/Set.cc
@@ -57,9 +57,25 @@ void Set::iterate(const std::function<void(const Index &, const Object &)> &fn)
 
 void Set::iterate(const std::function<void(const Index &, Object &)> &fn)
 {
+  // TODO Remove this workaround
+  //      This implementation is a workaround in case of adding operands while iteration
+  //
+  //      // Original Implementation (We probably should be back to this)
+  //      for (auto &e : _objects)
+  //      {
+  //        fn(e.first, *e.second);
+  //      }
+
+  std::list<Index> l;
+
   for (auto &e : _objects)
   {
-    fn(e.first, *e.second);
+    l.push_back(e.first);
+  }
+
+  for (auto index : l)
+  {
+    fn(index, *_objects[index]);
   }
 }