commit 5255672226de359aa5357815047ae95ec027c07d
author Michele Di Giorgio <michele.digiorgio@arm.com> Mon Dec 23 16:35:12 2019 +0000
committer Michele Di Giorgio <michele.digiorgio@arm.com> Thu Jan 02 12:36:01 2020 +0000
tree 61a1597e456a33ff3fc4d724acbd6deb738fcb2d
parent c6d9a8ba3706f1a0eea3fa8d243e2210e137a9c8

COMPMID-2998: NEON/GEMMConvolutionLayer/Quantized/QASYMM8_SIGNED failing on 32-bit

Change-Id: If8a705ba09f184e264c37453ffbee591f45e7cba
Signed-off-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-on: https://review.mlplatform.org/c/2524
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>

src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp

diff --git a/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp b/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp
index 6551518..65d800c 100644
--- a/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp
+++ b/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp
@@ -465,7 +465,7 @@
     ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR(a, b, d);
     ARM_COMPUTE_RETURN_ERROR_ON_CPU_F16_UNSUPPORTED(a);
 #ifndef __aarch64__
-    ARM_COMPUTE_RETURN_ERROR_ON_MSG(a->data_type() == DataType::U8 || a->data_type() == DataType::S8 || a->data_type() == DataType::QASYMM8, "8bit integer types only supported for aarch64");
+    ARM_COMPUTE_RETURN_ERROR_ON_MSG(a->element_size() == 1, "8bit integer types only supported for aarch64");
 #endif /* __aarch64__ */
     ARM_COMPUTE_RETURN_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(a, 1, DataType::U8, DataType::QASYMM8, DataType::QASYMM8_SIGNED, DataType::S8,
                                                          DataType::F16, DataType::F32);