diff options
Diffstat (limited to 'src/backends/reference/workloads/BatchMatMulImpl.cpp')
| -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; |