diff options
Diffstat (limited to 'src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_fp32_mla_8x4.hpp')
-rw-r--r-- | src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_fp32_mla_8x4.hpp | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_fp32_mla_8x4.hpp b/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_fp32_mla_8x4.hpp index 2b5cdae652..957754ad68 100644 --- a/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_fp32_mla_8x4.hpp +++ b/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_fp32_mla_8x4.hpp @@ -36,9 +36,9 @@ namespace arm_gemm { - // Actual kernel implementations void a64_hybrid_fp32_mla_8x4( ARGLIST ); +void a64_hybrid_fp32_mla_8x4_a55( ARGLIST ); class cls_a64_hybrid_fp32_mla_8x4 { @@ -73,9 +73,16 @@ public: // Default to the generic kernel kern_type kernel=a64_hybrid_fp32_mla_8x4; - - cls_a64_hybrid_fp32_mla_8x4(const CPUInfo *) + cls_a64_hybrid_fp32_mla_8x4(const CPUInfo *ci) { + switch(ci->get_cpu_model()) { + default: + break; + case CPUModel::A55r1: + case CPUModel::A53: + kernel=a64_hybrid_fp32_mla_8x4_a55; + break; + } } }; |