diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-10-26 19:05:32 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-11-08 12:00:31 +0000 |
commit | 421405b6a21b124288a750e2da26dc01eb7391cb (patch) | |
tree | 35f5655ce9d8b5921cb03630534f532e4eb47bf5 /src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h | |
parent | f1adf11c776aebaa8da1b8644a4ba2453afd2b81 (diff) | |
download | ComputeLibrary-421405b6a21b124288a750e2da26dc01eb7391cb.tar.gz |
COMPMID-1675: Add SVE support
Change-Id: I86679adff556b6ffc9929b35cbf1b59b3958bdb1
Diffstat (limited to 'src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h')
-rw-r--r-- | src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h b/src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h index 00e483c0f8..69842fec80 100644 --- a/src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h +++ b/src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h @@ -37,6 +37,10 @@ #include "../arm_gemm/kernels/a64_gemm_u8_4x4.hpp" #include "../arm_gemm/kernels/a64_hgemm_24x8.hpp" #include "../arm_gemm/kernels/a64_sgemm_12x8.hpp" +#include "../arm_gemm/kernels/sve_interleaved_fp16_mla_3VLx8.hpp" +#include "../arm_gemm/kernels/sve_interleaved_fp32_mla_3VLx8.hpp" +#include "../arm_gemm/kernels/sve_interleaved_s8s32_dot_3VLx8.hpp" +#include "../arm_gemm/kernels/sve_interleaved_u8u32_dot_3VLx8.hpp" namespace arm_compute { @@ -54,6 +58,29 @@ struct Kernel #define DEFINE_STRATEGY(strat) \ DEFINE_STRATEGY_SUFFIX(strat, "") +#ifdef __ARM_FEATURE_SVE +template <> +struct Kernel<float, false> +{ + DEFINE_STRATEGY(interleaved_fp32_mla_3VLx8) +}; +template <> +struct Kernel<float16_t, false> +{ + DEFINE_STRATEGY(interleaved_fp16_mla_3VLx8) +}; +template <bool use_dot> +struct Kernel<int8_t, use_dot> +{ + DEFINE_STRATEGY(interleaved_s8s32_dot_3VLx8) +}; +template <bool use_dot> +struct Kernel<uint8_t, use_dot> +{ + DEFINE_STRATEGY(interleaved_u8u32_dot_3VLx8) +}; +#else /* __ARM_FEATURE_SVE */ + #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC template <> struct Kernel<float16_t, false> @@ -96,6 +123,7 @@ struct Kernel<float, false> DEFINE_STRATEGY(sgemm_8x6) }; #endif /* __aarch64__ */ +#endif /* __ARM_FEATURE_SVE */ #undef DEFINE_STRATEGY #undef DEFINE_STRATEGY_SUFFIX |