From 0720dfa614ff041d1530e7bb2b7155d1047c71ff Mon Sep 17 00:00:00 2001 From: Tai Ly Date: Thu, 6 Jul 2023 20:28:57 +0000 Subject: [tosa_mlir_translator] Change axis to I32Attr Fix deserialization issues due to Axis attribute type changing from I64Attr to I32Attr to match the TOSA spec. Signed-off-by: Tai Ly Change-Id: I16b6e4ef8c7e0db988f1ccec814face70302efa1 --- src/TosaDeserialize.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src') diff --git a/src/TosaDeserialize.cpp b/src/TosaDeserialize.cpp index db52d67..335a997 100644 --- a/src/TosaDeserialize.cpp +++ b/src/TosaDeserialize.cpp @@ -338,7 +338,7 @@ TosaMlirOperatorBuilder::BuildReductionOp(TosaSerializationOperator *op) const { TosaAxisAttribute *attr = static_cast(op->GetAttribute()); - auto axis = op_builder->getI64IntegerAttr(attr->axis()); + auto axis = op_builder->getI32IntegerAttr(attr->axis()); mlir::Operation *mlir_op = op_builder->create(loc, output_type, input_val, axis); @@ -711,7 +711,7 @@ TosaMlirOperatorBuilder::build(TosaSerializationOperator *op) const { Attribute_AxisAttribute); // double check attribute type TosaAxisAttribute *attr = static_cast(op->GetAttribute()); - auto axis = op_builder->getI64IntegerAttr(attr->axis()); + auto axis = op_builder->getI32IntegerAttr(attr->axis()); mlir::Operation *mlir_op = op_builder->create( loc, output_type, input_values, axis); -- cgit v1.2.1