From 251b5145309e941a32c6c7691062544f2fd243fb Mon Sep 17 00:00:00 2001 From: David Mansell Date: Thu, 2 Feb 2023 09:25:37 +0000 Subject: gemm_interleaved: Set up the accumulation buffer properly in alternate threading mode. Resolves: COMPMID-5844 Change-Id: Iceb0018114bbca2bfdac4d4406936f9b260539e9 Signed-off-by: David Mansell Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9070 Benchmark: Arm Jenkins Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins Reviewed-by: Gian Marco Iodice Reviewed-by: Jakub Sujak --- src/core/NEON/kernels/arm_gemm/gemm_interleaved.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/core/NEON/kernels/arm_gemm/gemm_interleaved.hpp') diff --git a/src/core/NEON/kernels/arm_gemm/gemm_interleaved.hpp b/src/core/NEON/kernels/arm_gemm/gemm_interleaved.hpp index 442739b55e..c2fd0b0e8c 100644 --- a/src/core/NEON/kernels/arm_gemm/gemm_interleaved.hpp +++ b/src/core/NEON/kernels/arm_gemm/gemm_interleaved.hpp @@ -818,8 +818,8 @@ public: (last_pass ? _act : Activation()), !first_pass, // Pass in quantization parameters for requantizing kernels (others will ignore) _os, col_bias + (multi * _Nsize), - // Accumulation buffer (not yet implemented on this path) - static_cast(nullptr)); + // Accumulation buffer + get_accumulation_buffer(start_row, start_x, batch, multi)); /* Increment to the next block */ start_row += strategy::out_height(); -- cgit v1.2.1