From 9121c479ef2b11b52d7e77dc1b4dccd8f55b0db0 Mon Sep 17 00:00:00 2001
From: Jerry Ge <jerry.ge@arm.com>
Date: Tue, 20 Jun 2023 14:57:58 -0700
Subject: Add INT4 support for ConstOp deserialization

Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: Ie5687f9eda2208e86a5d1d50d073d59edf434524
---
 src/TosaDeserialize.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/TosaDeserialize.cpp b/src/TosaDeserialize.cpp
index ab60a26..d55585e 100644
--- a/src/TosaDeserialize.cpp
+++ b/src/TosaDeserialize.cpp
@@ -425,6 +425,13 @@ TosaMlirOperatorBuilder::build<Op_CONST>(TosaSerializationOperator *op) const {
         mlir::DenseElementsAttr::get(output_type, llvm::ArrayRef(float_data));
     break;
   }
+  case DType_INT4: {
+    std::vector<int8_t> int4_data;
+    TosaSerializationHandler::ConvertU8toI4(data, out_size, int4_data);
+    value_attr =
+        mlir::DenseElementsAttr::get(output_type, llvm::ArrayRef(int4_data));
+    break;
+  }
   case DType_INT8: {
     std::vector<int8_t> int8_data;
     TosaSerializationHandler::ConvertU8toI8(data, out_size, int8_data);
-- 
cgit v1.2.1