diff options
Diffstat (limited to 'reference_model/src')
-rw-r--r-- | reference_model/src/ops/ewise_binary.cc | 2 |
1 files changed, 1 insertions, 1 deletions
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, |