aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/cpu/operators/CpuGemmLowpMatrixMultiplyCore.cpp
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2021-08-12 07:42:51 +0100
committerSiCongLi <sicong.li@arm.com>2021-08-13 10:07:58 +0100
commitbcbd7068b8b6800f0f5c34239fbd26d5f6135d28 (patch)
tree675bb531117fa1f402ee37bfda8eb606f47a43ba /src/runtime/cpu/operators/CpuGemmLowpMatrixMultiplyCore.cpp
parentf2dde586bd785af5b8f3924b513fdb62f639ed0f (diff)
downloadComputeLibrary-bcbd7068b8b6800f0f5c34239fbd26d5f6135d28.tar.gz
Ensure correct transformed matrices are used in CpuGemmConvolution
Resolves: COMPMID-4763 Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com> Change-Id: Iae2e093cfb7d2c7172603897afe1c6a2e5d1caa3 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/c/VisualCompute/ComputeLibrary/+/349725 Tested-by: bsgcomp <bsgcomp@arm.com> Reviewed-by: Pablo Tello <pablo.tello@arm.com> Comments-Addressed: bsgcomp <bsgcomp@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6101 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Pablo Marquez Tello <pablo.tello@arm.com>
Diffstat (limited to 'src/runtime/cpu/operators/CpuGemmLowpMatrixMultiplyCore.cpp')
-rw-r--r--src/runtime/cpu/operators/CpuGemmLowpMatrixMultiplyCore.cpp1
1 files changed, 1 insertions, 0 deletions
diff --git a/src/runtime/cpu/operators/CpuGemmLowpMatrixMultiplyCore.cpp b/src/runtime/cpu/operators/CpuGemmLowpMatrixMultiplyCore.cpp
index f22446863c..7affc3f506 100644
--- a/src/runtime/cpu/operators/CpuGemmLowpMatrixMultiplyCore.cpp
+++ b/src/runtime/cpu/operators/CpuGemmLowpMatrixMultiplyCore.cpp
@@ -529,6 +529,7 @@ void CpuGemmLowpMatrixMultiplyCore::run(ITensorPack &tensors)
};
NEScheduler::get().schedule_op(_convert_to_signed_asymm.get(), Window::DimY, _convert_to_signed_asymm->window(), pack);
a_to_use = signed_a.get();
+ matrix_a = signed_a.get();
}
// Run GEMM