Fix dynamic weights for CPU connected layer
Resolves: COMPMID-5995
Signed-off-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Change-Id: I707b8918bebee7e70d4de5207ef555c806e7a305
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9405
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Reviewed-by: Jakub Sujak <jakub.sujak@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
diff --git a/src/cpu/operators/CpuGemmConv2d.cpp b/src/cpu/operators/CpuGemmConv2d.cpp
index 9bf6ed1..ebf2ebc 100644
--- a/src/cpu/operators/CpuGemmConv2d.cpp
+++ b/src/cpu/operators/CpuGemmConv2d.cpp
@@ -169,7 +169,7 @@
{
// Configure matrix multiply function
_mm_gemm = std::make_unique<CpuGemm>();
- _mm_gemm->configure(src, weights, biases, dst, 1.0f, 0.0f, gemm_info);
+ _mm_gemm->configure(src, weights, biases, dst, 1.0f, 1.0f, gemm_info);
auto mm_mem_req = _mm_gemm->workspace();
for(unsigned int cont = 0; cont < mm_mem_req.size(); ++cont)
{
@@ -235,7 +235,7 @@
else
{
// Perform validation step on Matrix multiply function
- return CpuGemm::validate(src, weights, nullptr, dst, 1.0f, 0.0f, gemm_info);
+ return CpuGemm::validate(src, weights, biases, dst, 1.0f, 1.0f, gemm_info);
}
}