diff options
author | Jerry Ge <jerry.ge@arm.com> | 2023-06-23 13:54:21 -0700 |
---|---|---|
committer | Jerry Ge <jerry.ge@arm.com> | 2023-06-23 13:54:21 -0700 |
commit | aff58dc10ffa3968c43a0f1a5239d43c0e5fcdb2 (patch) | |
tree | 32bc0eb9ceb1b69c10e2ec11cb5e57bc1e62c001 | |
parent | e4673b91ede85918209297c525f36c4d9dc7e2aa (diff) | |
download | tosa_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.cpp | 7 |
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); |