From 17e9508773e2d45917dad45cea063c92f0850f7c Mon Sep 17 00:00:00 2001
From: seockho-kim <seockho.kim@samsung.com>
Date: Mon, 23 Sep 2024 18:54:32 +0900
Subject: [PATCH] [luci/pass] Add RmsNorm to VerifyQuantizedNodeType (#14044)

This commit adds CircleRmsNorm to VerifyQuantizedNodeType.

ONE-DCO-1.0-Signed-off-by: Seockho Kim seockho.kim@samsung.com
---
 compiler/luci/pass/src/VerifyQuantizedNodeType.cpp | 6 ++++++
 compiler/luci/pass/src/VerifyQuantizedNodeType.h   | 1 +
 2 files changed, 7 insertions(+)

diff --git a/compiler/luci/pass/src/VerifyQuantizedNodeType.cpp b/compiler/luci/pass/src/VerifyQuantizedNodeType.cpp
index 4bad9522b85..1f0ff43b779 100644
--- 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
index 03f1e1d8640..15ec384413c 100644
--- 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);