1 files changed, 14 insertions, 0 deletions
diff --git a/compiler/luci/pass/src/VerifyQuantizedNodeGranularity.h b/compiler/luci/pass/src/VerifyQuantizedNodeGranularity.h
index 6bf7ff698..cc618bf0e 100644
--- a/compiler/luci/pass/src/VerifyQuantizedNodeGranularity.h
+++ b/compiler/luci/pass/src/VerifyQuantizedNodeGranularity.h
@@ -298,6 +298,13 @@ private:
     return true;
   }
 
+  bool visit(const luci::CircleSum *node)
+  {
+    RETURN_FALSE_UNLESS(is_lwq(node));
+    RETURN_FALSE_UNLESS(is_lwq(node->input()));
+    return true;
+  }
+
   bool visit(const luci::CircleArgMax *node)
   {
     // node's output is index, thus not quantized
@@ -333,6 +340,13 @@ private:
     return true;
   }
 
+  bool visit(const luci::CircleGelu *node)
+  {
+    RETURN_FALSE_UNLESS(is_lwq(node));
+    RETURN_FALSE_UNLESS(is_lwq(node->features()));
+    return true;
+  }
+
   bool visit(const luci::CircleGreater *node)
   {
     RETURN_FALSE_UNLESS(is_lwq(node->x()));