From aff58dc10ffa3968c43a0f1a5239d43c0e5fcdb2 Mon Sep 17 00:00:00 2001
From: Jerry Ge <jerry.ge@arm.com>
Date: Fri, 23 Jun 2023 13:54:21 -0700
Subject: Add INT4 support for ConstOp deserialization

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

diff --git a/src/TosaDeserialize.cpp b/src/TosaDeserialize.cpp
index f6f78fc..da92c19 100644
--- a/src/TosaDeserialize.cpp
+++ b/src/TosaDeserialize.cpp
@@ -384,6 +384,13 @@ std::vector<mlir::Value> TosaMlirOperatorBuilder::build<Op_CONST>(TosaSerializat
         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