diff options
Diffstat (limited to 'compiler/luci/pass/src/VerifyQuantizedNodeType.cpp')
-rw-r--r-- | compiler/luci/pass/src/VerifyQuantizedNodeType.cpp | 8 |
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); } |