diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-12-01 02:17:19 +0000 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-12-01 10:46:32 +0000 |
| commit | 4c634e005eee65466eee607c4e287a8d5a0ee57d (patch) | |
| tree | 2f537643e026d81ecb306cfa6678240089f448f0 /src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp | |
| parent | 40f51a63c8e7258db15269427ae4fe1ad199c550 (diff) | |
| download | ComputeLibrary-4c634e005eee65466eee607c4e287a8d5a0ee57d.tar.gz | |
Initialize batches variable in NEGEMMAssemblyDispatch
Resolves: COMPMID-4027
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: I54e141a3521d9b2c5f93dbcb3ed78a259b96358e
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4627
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp')
| -rw-r--r-- | src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
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) { |