summaryrefslogtreecommitdiff
path: root/compiler/luci/pass/src/VerifyQuantizedNodeType.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'compiler/luci/pass/src/VerifyQuantizedNodeType.cpp')
-rw-r--r--compiler/luci/pass/src/VerifyQuantizedNodeType.cpp8
1 files changed, 8 insertions, 0 deletions
diff --git a/compiler/luci/pass/src/VerifyQuantizedNodeType.cpp b/compiler/luci/pass/src/VerifyQuantizedNodeType.cpp
index cf86acabe..3ce32555b 100644
--- a/compiler/luci/pass/src/VerifyQuantizedNodeType.cpp
+++ b/compiler/luci/pass/src/VerifyQuantizedNodeType.cpp
@@ -47,6 +47,10 @@ namespace luci
template <loco::DataType Qtype, loco::DataType Btype>
bool VerifyQuantizedNodeTypeBase<Qtype, Btype>::visit(const luci::CircleAdd *node)
{
+ // Allow add of indices
+ if (group_has_type(node, loco::DataType::S32) or group_has_type(node, loco::DataType::S64))
+ return true;
+
return group_has_type(node, Qtype);
}
@@ -240,6 +244,10 @@ bool VerifyQuantizedNodeTypeBase<Qtype, Btype>::visit(const luci::CircleMirrorPa
template <loco::DataType Qtype, loco::DataType Btype>
bool VerifyQuantizedNodeTypeBase<Qtype, Btype>::visit(const luci::CircleMul *node)
{
+ // Allow mul of indices
+ if (group_has_type(node, loco::DataType::S32) or group_has_type(node, loco::DataType::S64))
+ return true;
+
return group_has_type(node, Qtype);
}