aboutsummaryrefslogtreecommitdiff
path: root/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_fp32_mla_8x4.hpp
diff options
context:
space:
mode:
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.hpp13
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;
+ }
}
};