diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-11-21 02:49:11 +0000 |
---|---|---|
committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2020-11-23 11:47:38 +0000 |
commit | 2e44e39846c8b52b81ccf5f5f6d94816082269cd (patch) | |
tree | 054c8dd44b915fd90f11b4281fde147415ce030a | |
parent | 0052078c7329a1bef7705c7d1e7603075e2a1a20 (diff) | |
download | ComputeLibrary-2e44e39846c8b52b81ccf5f5f6d94816082269cd.tar.gz |
Initialize unitialized M parameter in NEGEMMAssemblyDispatch
Resolves: COMPMID-3991
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: I353ad84833ce25e1adadb980f46b95de63dd6c62
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4522
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
(cherry picked from commit 71696baa30fd2cb9ac7493937e65abfb16daa16e)
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4168
Tested-by: Michele Di Giorgio <michele.digiorgio@arm.com>
-rw-r--r-- | src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp b/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp index 400fa64438..f6739ee925 100644 --- a/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp +++ b/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp @@ -61,6 +61,7 @@ Params extract_parameters(const ITensor *a, const ITensor *b, const ITensor *d, ARM_COMPUTE_ERROR_ON_NULLPTR(a, b, d); Params p; + p.M = d->info()->tensor_shape().y(); p.K = a->info()->tensor_shape().x(); p.N = d->info()->tensor_shape().x(); p.multis = 1; @@ -74,7 +75,6 @@ Params extract_parameters(const ITensor *a, const ITensor *b, const ITensor *d, } else { - p.M = d->info()->tensor_shape().y(); p.multis = b->info()->tensor_shape().z(); p.batches = d->info()->tensor_shape().total_size_upper(2) / p.multis; //COMPMID-1423: Agree on and document the layout of gemm inputs/outputs } |