diff options
author | David Mansell <David.Mansell@arm.com> | 2023-02-02 09:25:37 +0000 |
---|---|---|
committer | Viet-Hoa Do <viet-hoa.do@arm.com> | 2023-03-21 11:59:27 +0000 |
commit | 251b5145309e941a32c6c7691062544f2fd243fb (patch) | |
tree | 2238fe7eece060a6af52a0e5014723e11896882c | |
parent | a3e57c20a0b7a174f0c357676a4da40a248d04db (diff) | |
download | ComputeLibrary-251b5145309e941a32c6c7691062544f2fd243fb.tar.gz |
gemm_interleaved: Set up the accumulation buffer properly in alternate
threading mode.
Resolves: COMPMID-5844
Change-Id: Iceb0018114bbca2bfdac4d4406936f9b260539e9
Signed-off-by: David Mansell <David.Mansell@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9070
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-by: Jakub Sujak <jakub.sujak@arm.com>
-rw-r--r-- | src/core/NEON/kernels/arm_gemm/gemm_interleaved.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
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<Tab *>(nullptr)); + // Accumulation buffer + get_accumulation_buffer(start_row, start_x, batch, multi)); /* Increment to the next block */ start_row += strategy::out_height(); |