From fa591272c05d2a24412b4b5b4398ded17be0912e Mon Sep 17 00:00:00 2001
From: TatWai Chong <tatwai.chong@arm.com>
Date: Thu, 29 Feb 2024 00:17:29 -0800
Subject: Change the shift operand of mul to be available to all data types.

Change-Id: I436912af95e4aef1b67b140079070168d158ff49
Signed-off-by: TatWai Chong <tatwai.chong@arm.com>
---
 src/TosaSerialize.cpp | 19 ++++---------------
 1 file changed, 4 insertions(+), 15 deletions(-)

(limited to 'src/TosaSerialize.cpp')

diff --git a/src/TosaSerialize.cpp b/src/TosaSerialize.cpp
index 5b0d2bd..0d5e044 100644
--- a/src/TosaSerialize.cpp
+++ b/src/TosaSerialize.cpp
@@ -1286,22 +1286,11 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::MulOp>(
   std::string input0_name = GetTensorName(mul_op.getInput1());
   std::string input1_name = GetTensorName(mul_op.getInput2());
   std::string output_name = GetTensorName(mul_op.getOutput());
+  std::string shift_name = GetTensorName(mul_op.getShift());
 
-  std::vector<std::string> operands;
-  if (mul_op.getOutput()
-          .getType()
-          .cast<mlir::TensorType>()
-          .getElementType()
-          .isInteger(32)) {
-    std::string shift_name = GetTensorName(mul_op.getShift());
-    operands = {input0_name, input1_name, shift_name};
-  } else {
-    operands = {input0_name, input1_name};
-  }
-
-  TosaSerializationOperator *tyop =
-      new TosaSerializationOperator(Op_MUL, Attribute_NONE, nullptr, operands,
-                                    std::vector<std::string>{output_name});
+  TosaSerializationOperator *tyop = new TosaSerializationOperator(
+      Op_MUL, Attribute_NONE, nullptr, {input0_name, input1_name, shift_name},
+      std::vector<std::string>{output_name});
 
   return tyop;
 }
-- 
cgit v1.2.1