diff options
| author | Samuel Yap <samuel.yap@arm.com> | 2022-08-02 09:12:02 +0100 |
|---|---|---|
| committer | Nikhil Raj Arm <nikhil.raj@arm.com> | 2022-08-03 08:40:01 +0000 |
| commit | bc2d5a65b2518aca2130908c65a35f1b7fd74969 (patch) | |
| tree | 5a44940dd8d677df22bc2c715c65b70b33ccd7a4 | |
| parent | c347b1727e6afb415241c77f997e311954021651 (diff) | |
| download | armnn-bc2d5a65b2518aca2130908c65a35f1b7fd74969.tar.gz | |
IVGCVSW-7145: BatchMatMul Fix for Raspi Cross Compile Failing
* Changed long variable declaration to int
Signed-off-by: Samuel Yap <samuel.yap@arm.com>
Change-Id: I2df6f8f6df8780e48e09f7e68c04626a8a8a207d
| -rw-r--r-- | src/backends/reference/workloads/BatchMatMulImpl.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
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<unsigned int>& curIdx, unsigned int cur void BatchMatMul::AdjustAxesToMulForUnequalRanks( std::pair<std::pair<unsigned int, unsigned int>, std::pair<unsigned int, unsigned int>>& axesToMul) { - long rankDiff = static_cast<long>(inputXInfo.GetNumDimensions()) - inputYInfo.GetNumDimensions(); + int rankDiff = static_cast<int>(inputXInfo.GetNumDimensions()) - + static_cast<int>(inputYInfo.GetNumDimensions()); if(rankDiff == 0) { return; |