diff options
| author | Tai Ly <tai.ly@arm.com> | 2023-07-07 23:36:04 +0000 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2023-07-12 15:20:51 +0000 |
| commit | 787606544db8664ff42a31958f554fe7088427b5 (patch) | |
| tree | 1abcacc8fb205bf313c66c197c3e99fa0764818b /src/TosaDeserialize.cpp | |
| parent | 9a57b9fe6f9832fa0406daac367fd3fc09afa018 (diff) | |
| download | tosa_mlir_translator-787606544db8664ff42a31958f554fe7088427b5.tar.gz | |
[tosa_mlir_translator] Fix Pad const attr type
This fixes serialization and deserialization of Pad Operator's
constant value attributes that is integer data types other than I32
also fixed serialization of F16 constants to not crash, found in
testing
Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: Ic3987aad0e11b612de591eaeecd308d599d174e1
Diffstat (limited to 'src/TosaDeserialize.cpp')
| -rw-r--r-- | src/TosaDeserialize.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/TosaDeserialize.cpp b/src/TosaDeserialize.cpp index a4b7eda..153f16f 100644 --- a/src/TosaDeserialize.cpp +++ b/src/TosaDeserialize.cpp @@ -809,10 +809,12 @@ TosaMlirOperatorBuilder::build<Op_PAD>(TosaSerializationOperator *op) const { padding_value); } else { // create a const value for pad_const input + const auto input_element_type = + input_val.getType().cast<mlir::ShapedType>().getElementType(); mlir::Value pad_const_value; if (pad_const_int != 0) { auto pad_const_int_type = - mlir::RankedTensorType::get({}, op_builder->getI32Type()); + mlir::RankedTensorType::get({}, input_element_type); auto pad_const_int_attr = mlir::DenseElementsAttr::get(pad_const_int_type, {pad_const_int}); mlir::Operation *pad_const_int_op = @@ -822,7 +824,7 @@ TosaMlirOperatorBuilder::build<Op_PAD>(TosaSerializationOperator *op) const { pad_const_value = pad_const_int_op->getResult(0); } else if (pad_const_fp != 0) { auto pad_const_fp_type = - mlir::RankedTensorType::get({}, op_builder->getF32Type()); + mlir::RankedTensorType::get({}, input_element_type); auto pad_const_fp_attr = mlir::DenseElementsAttr::get(pad_const_fp_type, {pad_const_fp}); mlir::Operation *pad_const_fp_op = |