From 4c634e005eee65466eee607c4e287a8d5a0ee57d Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Tue, 1 Dec 2020 02:17:19 +0000 Subject: Initialize batches variable in NEGEMMAssemblyDispatch Resolves: COMPMID-4027 Signed-off-by: Georgios Pinitas Change-Id: I54e141a3521d9b2c5f93dbcb3ed78a259b96358e Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4627 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Comments-Addressed: Arm Jenkins --- src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp b/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp index 394f970e54..b54389cf5f 100644 --- a/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp +++ b/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp @@ -62,9 +62,10 @@ Params extract_parameters(const ITensor *a, const ITensor *b, const ITensor *d, p.M = d->info()->tensor_shape().y(); p.K = a->info()->tensor_shape().x(); p.N = d->info()->tensor_shape().x(); + p.batches = 1; p.multis = 1; - p.indirect = false; p.sections = 1; + p.indirect = false; if(info.method == AsmConvMethod::Conv || info.method == AsmConvMethod::Indirect) { -- cgit v1.2.1