commit b5a450a1acc1149f99f7bb06b10694fba554f4e3
author Michalis Spyrou <michalis.spyrou@arm.com> Wed Jan 06 17:40:30 2021 +0000
committer Michalis Spyrou <michalis.spyrou@arm.com> Mon Jan 18 14:03:46 2021 +0000
tree ac6ddf556c1656fd697a53b6ad5e961938f7d85d
parent d2447bb039c268aa21a5ca358cc2d91abe4f4d21

Add SVE for Softmax

Implements COMPMID-3875

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

src/core/NEON/wrapper/traits.h

diff --git a/src/core/NEON/wrapper/traits.h b/src/core/NEON/wrapper/traits.h
index eafbeef..0bb55af 100644
--- a/src/core/NEON/wrapper/traits.h
+++ b/src/core/NEON/wrapper/traits.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2018-2020 Arm Limited.
+ * Copyright (c) 2018-2021 Arm Limited.
  *
  * SPDX-License-Identifier: MIT
  *
@@ -26,6 +26,10 @@
 
 #include <arm_neon.h>
 
+#if defined(__ARM_FEATURE_SVE)
+#include <arm_sve.h>
+#endif /* defined(__ARM_FEATURE_SVE) */
+
 namespace arm_compute
 {
 namespace wrapper
@@ -110,6 +114,16 @@
 template <> struct neon_bitvector<float16_t, BitWidth::W64>{ using type = float16x4_t; using tag_type = vector_64_tag; };
 template <> struct neon_bitvector<float16_t, BitWidth::W128>{ using type = float16x8_t; using tag_type = vector_128_tag; };
 #endif // __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
+
+
+#if defined(__ARM_FEATURE_SVE)
+/** Create the appropriate SVE vector given its type */
+template <typename T> struct sve_vector;
+
+template <> struct sve_vector<uint8_t>{ using scalar_type = uint8_t; using type = svuint8_t; };
+template <> struct sve_vector<int8_t>{ using scalar_type = int8_t; using type = svint8_t; };
+#endif /* defined(__ARM_FEATURE_SVE) */
+
 #endif /* DOXYGEN_SKIP_THIS */
 
 /**  Helper type template to get the type of a neon vector */