diff options
author | Tai Ly <tai.ly@arm.com> | 2023-07-06 20:28:57 +0000 |
---|---|---|
committer | Tai Ly <tai.ly@arm.com> | 2023-07-10 21:03:45 +0000 |
commit | 0720dfa614ff041d1530e7bb2b7155d1047c71ff (patch) | |
tree | d8addab5629e39ac9472344355e64137104a962f | |
parent | 289fd33b43f5260dba2faab9c372e42b4b71f83d (diff) | |
download | tosa_mlir_translator-0720dfa614ff041d1530e7bb2b7155d1047c71ff.tar.gz |
[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 <tai.ly@arm.com>
Change-Id: I16b6e4ef8c7e0db988f1ccec814face70302efa1
-rw-r--r-- | src/TosaDeserialize.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
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<TosaAxisAttribute *>(op->GetAttribute()); - auto axis = op_builder->getI64IntegerAttr(attr->axis()); + auto axis = op_builder->getI32IntegerAttr(attr->axis()); mlir::Operation *mlir_op = op_builder->create<MLIR_OP>(loc, output_type, input_val, axis); @@ -711,7 +711,7 @@ TosaMlirOperatorBuilder::build<Op_CONCAT>(TosaSerializationOperator *op) const { Attribute_AxisAttribute); // double check attribute type TosaAxisAttribute *attr = static_cast<TosaAxisAttribute *>(op->GetAttribute()); - auto axis = op_builder->getI64IntegerAttr(attr->axis()); + auto axis = op_builder->getI32IntegerAttr(attr->axis()); mlir::Operation *mlir_op = op_builder->create<mlir::tosa::ConcatOp>( loc, output_type, input_values, axis); |