From 900b78f599ea5997d60e7538831a906b92265ae0 Mon Sep 17 00:00:00 2001 From: Gian Marco Date: Mon, 20 Nov 2017 15:01:12 +0000 Subject: COMPMID-556 - Fixed static window in NEGEMMMatrixAccumulateBiasesKernel Change-Id: Ic94eebe220b5f400740f10a3ed9fae65ab7b6c47 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/96604 Tested-by: Kaizen Reviewed-by: Georgios Pinitas --- src/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'src/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.cpp') diff --git a/src/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.cpp b/src/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.cpp index 683510879b..3dd59bddd6 100644 --- a/src/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.cpp +++ b/src/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.cpp @@ -29,6 +29,7 @@ #include "arm_compute/core/ITensor.h" #include "arm_compute/core/NEON/NEFixedPoint.h" #include "arm_compute/core/Types.h" +#include "arm_compute/core/Utils.h" #include "arm_compute/core/Validate.h" #include "arm_compute/core/Window.h" @@ -60,7 +61,7 @@ void NEGEMMMatrixAccumulateBiasesKernel::configure(ITensor *accum, const ITensor update_window_and_padding(win, AccessWindowHorizontal(accum->info(), 0, num_elems_processed_per_iteration), - AccessWindowStatic(biases->info(), 0, 0, win.x().end(), biases->info()->tensor_shape().y())); + AccessWindowStatic(biases->info(), 0, 0, ceil_to_multiple(biases->info()->dimension(0), num_elems_processed_per_iteration), biases->info()->tensor_shape().y())); AccessWindowHorizontal output_access(accum->info(), 0, num_elems_processed_per_iteration); -- cgit v1.2.1