[loco] Add bool return type for shape infer to() (#7131)

This will set boolean return type for ShapeInferenceSession to() method

Signed-off-by: SaeHie Park <saehie.park@samsung.com>

diff --git a/compiler/loco/include/loco/Service/ShapeInference.h b/compiler/loco/include/loco/Service/ShapeInference.h
index 405eff7..f7bc5d4 100644 (file)
--- a/compiler/loco/include/loco/Service/ShapeInference.h
+++ b/compiler/loco/include/loco/Service/ShapeInference.h
@@ -42,7 +42,7 @@ public:
   }
 
 public:
-  void to(Graph *g) const;
+  bool to(Graph *g) const;
 
 private:
   const ShapeInferenceRule *_rule;

diff --git a/compiler/loco/src/Service/ShapeInference.cpp b/compiler/loco/src/Service/ShapeInference.cpp
index d8eb545..8b86d48 100644 (file)
--- a/compiler/loco/src/Service/ShapeInference.cpp
+++ b/compiler/loco/src/Service/ShapeInference.cpp
@@ -47,8 +47,10 @@ private:
 namespace loco
 {
 
-void ShapeInferenceSession::to(Graph *g) const
+bool ShapeInferenceSession::to(Graph *g) const
 {
+  bool changed = false;
+
   for (auto node : loco::postorder_traversal(loco::output_nodes(g)))
   {
     if (_rule->recognize(node->dialect()))
@@ -58,9 +60,12 @@ void ShapeInferenceSession::to(Graph *g) const
       if (_rule->infer(node, shape))
       {
         node->annot(stdex::make_unique<ShapeAnnotation>(shape));
+        changed = true;
       }
     }
   }
+
+  return changed;
 }
 
 bool ShapeInference::known(const Node *node) { return node->annot<ShapeAnnotation>() != nullptr; }