From bc2d5a65b2518aca2130908c65a35f1b7fd74969 Mon Sep 17 00:00:00 2001 From: Samuel Yap Date: Tue, 2 Aug 2022 09:12:02 +0100 Subject: IVGCVSW-7145: BatchMatMul Fix for Raspi Cross Compile Failing * Changed long variable declaration to int Signed-off-by: Samuel Yap Change-Id: I2df6f8f6df8780e48e09f7e68c04626a8a8a207d --- src/backends/reference/workloads/BatchMatMulImpl.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/backends/reference/workloads/BatchMatMulImpl.cpp b/src/backends/reference/workloads/BatchMatMulImpl.cpp index 74a358cc5c..6693f15760 100644 --- a/src/backends/reference/workloads/BatchMatMulImpl.cpp +++ b/src/backends/reference/workloads/BatchMatMulImpl.cpp @@ -73,7 +73,8 @@ void BatchMatMul::RecurseBMM(std::vector& curIdx, unsigned int cur void BatchMatMul::AdjustAxesToMulForUnequalRanks( std::pair, std::pair>& axesToMul) { - long rankDiff = static_cast(inputXInfo.GetNumDimensions()) - inputYInfo.GetNumDimensions(); + int rankDiff = static_cast(inputXInfo.GetNumDimensions()) - + static_cast(inputYInfo.GetNumDimensions()); if(rankDiff == 0) { return; -- cgit v1.2.1