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/arm_gemm/gemm_int8.cpp | |
parent | f1adf11c776aebaa8da1b8644a4ba2453afd2b81 (diff) | |
download | ComputeLibrary-421405b6a21b124288a750e2da26dc01eb7391cb.tar.gz |
COMPMID-1675: Add SVE support
Change-Id: I86679adff556b6ffc9929b35cbf1b59b3958bdb1
Diffstat (limited to 'src/core/NEON/kernels/arm_gemm/gemm_int8.cpp')
-rw-r--r-- | src/core/NEON/kernels/arm_gemm/gemm_int8.cpp | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/gemm_int8.cpp b/src/core/NEON/kernels/arm_gemm/gemm_int8.cpp index dffa056adc..627d8abdb9 100644 --- a/src/core/NEON/kernels/arm_gemm/gemm_int8.cpp +++ b/src/core/NEON/kernels/arm_gemm/gemm_int8.cpp @@ -31,9 +31,21 @@ #include "kernels/a64_gemm_s16_12x8.hpp" #include "kernels/a64_gemm_s8_12x8.hpp" #include "kernels/a64_gemm_s8_4x4.hpp" +#include "kernels/sve_interleaved_s8s32_dot_3VLx8.hpp" namespace arm_gemm { +#ifdef __ARM_FEATURE_SVE +class GemmImpl_gemm_s8_interleaved_dot : public GemmImplementation<int8_t, int32_t> { +public: + UniqueGemmCommon<int8_t, int32_t> instantiate(const GemmArgs<int32_t> &args) override { + return UniqueGemmCommon<int8_t, int32_t>(new GemmInterleaved<interleaved_s8s32_dot_3VLx8, int8_t, int32_t>(args)); + } + + GemmImpl_gemm_s8_interleaved_dot() : GemmImplementation<int8_t, int32_t>(GemmMethod::GEMM_INTERLEAVED_DOT) { } +}; +#else + class GemmImpl_gemm_s8_interleaved_dot : public GemmImplementation<int8_t, int32_t> { public: bool is_supported(const GemmArgs<int32_t> &args) override { @@ -47,6 +59,8 @@ public: GemmImpl_gemm_s8_interleaved_dot() : GemmImplementation<int8_t, int32_t>(GemmMethod::GEMM_INTERLEAVED_DOT) { } }; +#endif + class GemmImpl_gemm_s8_interleaved : public GemmImplementation<int8_t, int32_t> { public: UniqueGemmCommon<int8_t, int32_t> instantiate(const GemmArgs<int32_t> &args) override { |