From 9257fd5775b49e206de33536a623e34cc1cecb42 Mon Sep 17 00:00:00 2001
From: Kevin Cheng <kevin.cheng@arm.com>
Date: Wed, 14 Apr 2021 15:55:31 -0700
Subject: Fix OpMul for 32-bit x 32-bit inputs

Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: I068d3ce0bffe1b49845f0c8cd39130060cceaf19
---
 reference_model/src/ops/ewise_binary.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/reference_model/src/ops/ewise_binary.cc b/reference_model/src/ops/ewise_binary.cc
index 7ed8374..fc587f1 100644
--- a/reference_model/src/ops/ewise_binary.cc
+++ b/reference_model/src/ops/ewise_binary.cc
@@ -436,7 +436,7 @@ int OpMul<Rank, InDtype, OutDtype>::register_fcn()
                 if (shift > 0)
                 {
                     int64_t round = 1L << (shift - 1);
-                    result        = a * b + round;
+                    result        = static_cast<int64_t>(a) * static_cast<int64_t>(b) + round;
                     result        = result >> shift;
 
                     ASSERT_MSG_NODE(result >= QMin && result <= QMax,
-- 
cgit v1.2.1