commit 02acf014b2840a3b644bfb783b8396e79cafd3a3
author Georgios Pinitas <georgios.pinitas@arm.com> Tue Mar 19 10:49:03 2019 +0000
committer Georgios Pinitas <georgios.pinitas@arm.com> Tue Mar 19 11:53:59 2019 +0000
tree bc610105f2dd98ecf0522848c0e115b1ed08fe6a
parent 146138378c1587b7297d245b7177641315f6180b

COMPMID-2065: Fix validation issue in NEGEMMLowpMMCore

Sets correctly the quantization info in the auxilary tensors.

Change-Id: I35355d0aa2ab9716df3a91cbbf4609365fb109e4
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/872
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>

src/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.cpp

diff --git a/src/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.cpp b/src/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.cpp
index 85e49fd..0247a39 100644
--- a/src/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.cpp
+++ b/src/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.cpp
@@ -104,9 +104,9 @@
         matrix_b = &_tmp_b;
 
         // The interleaved output matrix will have the following shape: [ a_height * 4, ceil(a_width / 4.0f) ]
-        TensorInfo a_info(compute_interleaved_shape(*a->info()), 1, a->info()->data_type());
+        TensorInfo a_info(compute_interleaved_shape(*a->info()), 1, a->info()->data_type(), a->info()->quantization_info());
         // The transpose1xW output matrix will have the following shape: [ b_height * 16, ceil(b_width / 16.0f) ]
-        TensorInfo b_info(compute_transpose1xW_shape(*b->info()), 1, b->info()->data_type());
+        TensorInfo b_info(compute_transpose1xW_shape(*b->info()), 1, b->info()->data_type(), b->info()->quantization_info());
         _tmp_a.allocator()->init(a_info);
         _tmp_b.allocator()->init(b_info);
         _memory_group.manage(&_tmp_a);