[luci/pass] Add RmsNorm to VerifyQuantizedNodeType (#14044)

This commit adds CircleRmsNorm to VerifyQuantizedNodeType. ONE-DCO-1.0-Signed-off-by: Seockho Kim [email protected]
Samsung · Sep 23, 2024 · 17e9508 · 17e9508
1 parent 07978d5
commit 17e9508
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/compiler/luci/pass/src/VerifyQuantizedNodeType.cpp b/compiler/luci/pass/src/VerifyQuantizedNodeType.cpp
@@ -364,6 +364,12 @@ bool VerifyQuantizedNodeTypeBase<Qtype, Btype>::visit(const luci::CircleResizeNe
   return true;
 }
 
+template <loco::DataType Qtype, loco::DataType Btype>
+bool VerifyQuantizedNodeTypeBase<Qtype, Btype>::visit(const luci::CircleRmsNorm *node)
+{
+  return group_has_type(node, Qtype);
+}
+
 template <loco::DataType Qtype, loco::DataType Btype>
 bool VerifyQuantizedNodeTypeBase<Qtype, Btype>::visit(const luci::CircleRsqrt *node)
 {

diff --git a/compiler/luci/pass/src/VerifyQuantizedNodeType.h b/compiler/luci/pass/src/VerifyQuantizedNodeType.h
@@ -110,6 +110,7 @@ class VerifyQuantizedNodeTypeBase : public luci::CircleNodeVisitor<bool>,
   bool visit(const luci::CircleReshape *node);
   bool visit(const luci::CircleResizeBilinear *node);
   bool visit(const luci::CircleResizeNearestNeighbor *node);
+  bool visit(const luci::CircleRmsNorm *node);
   bool visit(const luci::CircleRsqrt *node);
   bool visit(const luci::CircleSlice *node);
   bool visit(const luci::CircleSpaceToBatchND *node);