aboutsummaryrefslogtreecommitdiff
path: root/tests/TensorLibrary.h
diff options
context:
space:
mode:
authorPablo Tello <pablo.tello@arm.com>2017-06-28 17:27:56 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-09-17 14:15:39 +0100
commit221f38176b0d4dbc212441779d9bbac3cc0eecfa (patch)
treee838d673b35c5b40d4b484a3645cc7ae3c9d3ecc /tests/TensorLibrary.h
parent6410fb2a14427713251f5d97144ac5d4f17c988c (diff)
downloadComputeLibrary-221f38176b0d4dbc212441779d9bbac3cc0eecfa.tar.gz
COMPMID-421: Fixed FP16 support in Neon GEMM.
Fixed GEMM FP16 problem with matrices that are not multiple of 32. Added a new test suite NEON/GEMM/Float16/SmallGEMM. Implemented FP16 function to multiply vector by a matrix. Change-Id: Ie6c692885a48d0206bd6fe748332fa83bc286d67 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/79118 Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com> Reviewed-by: Moritz Pflanzer <moritz.pflanzer@arm.com>
Diffstat (limited to 'tests/TensorLibrary.h')
-rw-r--r--tests/TensorLibrary.h2
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/TensorLibrary.h b/tests/TensorLibrary.h
index 4d7143a206..69b2381171 100644
--- a/tests/TensorLibrary.h
+++ b/tests/TensorLibrary.h
@@ -505,7 +505,7 @@ void TensorLibrary::fill_tensor_uniform(T &&tensor, std::random_device::result_t
fill(tensor, distribution_f16, seed_offset);
break;
}
-#endif
+#endif /* ARM_COMPUTE_ENABLE_FP16 */
case DataType::F32:
{
// It doesn't make sense to check [-inf, inf], so hard code it to a big number