aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJerry Ge <jerry.ge@arm.com>2023-06-23 13:54:21 -0700
committerJerry Ge <jerry.ge@arm.com>2023-06-23 13:54:21 -0700
commitaff58dc10ffa3968c43a0f1a5239d43c0e5fcdb2 (patch)
tree32bc0eb9ceb1b69c10e2ec11cb5e57bc1e62c001
parente4673b91ede85918209297c525f36c4d9dc7e2aa (diff)
downloadtosa_mlir_translator-v0.60.tar.gz
Add INT4 support for ConstOp deserializationv0.60
Signed-off-by: Jerry Ge <jerry.ge@arm.com> Change-Id: I6eda561327635d3c5ec4793f3ae3a1483f2c018c
-rw-r--r--src/TosaDeserialize.cpp7
1 files changed, 7 insertions, 0 deletions
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);