From c8bdeaacc2f02625b900be7b6ccb26a4b663eead Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Tue, 26 May 2020 09:59:18 +0100 Subject: COMPMID-3069: Fix reduction in GEMMLowp We shouldn't perform reduction on matrix b when we have a configured fused assembly kernel Change-Id: I1f26c2afb387ee6ebbd54263e7255dab276ea08f Signed-off-by: Michalis Spyrou Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3261 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Comments-Addressed: Arm Jenkins --- src/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.cpp') diff --git a/src/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.cpp b/src/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.cpp index f0fa61657b..993907c29a 100644 --- a/src/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.cpp +++ b/src/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.cpp @@ -574,7 +574,7 @@ void NEGEMMLowpMatrixMultiplyCore::prepare() } // Run matrix B reduction kernel only if _a_offset is not equal to 0 - if(_a_offset != 0 && _reshape_b_only_on_first_run) + if(!_fused_assembly_path && _a_offset != 0 && _reshape_b_only_on_first_run) { _vector_sum_col.allocator()->allocate(); NEScheduler::get().schedule(&_mtx_b_reduction_kernel, Window::DimX); -- cgit v1.2.1