diff options
| author | Samuel Yap <samuel.yap@arm.com> | 2022-08-02 09:12:02 +0100 |
|---|---|---|
| committer | Nikhil Raj <nikhil.raj@arm.com> | 2022-08-05 15:52:13 +0100 |
| commit | ca7bbd6a7eb2d7a32bf8896eb8e2760a6600fff4 (patch) | |
| tree | 5a44940dd8d677df22bc2c715c65b70b33ccd7a4 | |
| parent | d301c0a56061c050199d60206ba5c2c89a6d099e (diff) | |
| download | armnn-ca7bbd6a7eb2d7a32bf8896eb8e2760a6600fff4.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; |