From e28cf395b57a091d0850cd28cecc81046153b843 Mon Sep 17 00:00:00 2001
From: Georgios Pinitas <georgios.pinitas@arm.com>
Date: Sun, 31 Jan 2021 05:18:43 +0000
Subject: Regenerate kernels and update A55 versions

Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: I9eae76c77db03b8806af65729da34ab2d77f95f2
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4965
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
---
 .../arm_gemm/kernels/a64_hybrid_fp16_mla_6x32.hpp       | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

(limited to 'src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_fp16_mla_6x32.hpp')

diff --git a/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_fp16_mla_6x32.hpp b/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_fp16_mla_6x32.hpp
index ca2696bebd..674d71d626 100644
--- a/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_fp16_mla_6x32.hpp
+++ b/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_fp16_mla_6x32.hpp
@@ -37,9 +37,9 @@
 
 namespace arm_gemm
 {
-
 // Actual kernel implementations
 void a64_hybrid_fp16_mla_6x32( ARGLIST );
+void a64_hybrid_fp16_mla_6x32_a55( ARGLIST );
 
 class cls_a64_hybrid_fp16_mla_6x32
 {
@@ -72,10 +72,11 @@ public:
 
     StdTransformsFixed<operand_type, result_type, 6, 32, 1> transforms = {};
 
-    static PerformanceParameters get_performance_parameters(const CPUInfo *ci) {
+    static PerformanceParameters get_performance_parameters(const CPUInfo *ci)
+    {
         switch (ci->get_cpu_model()) {
             case CPUModel::A55r1:
-                return { 5.22 };
+                return { 6.94 };
             default:
                 return { 14.53 };
         }
@@ -83,9 +84,15 @@ public:
 
     // Default to the generic kernel
     kern_type kernel=a64_hybrid_fp16_mla_6x32;
-
-    cls_a64_hybrid_fp16_mla_6x32(const CPUInfo *)
+    cls_a64_hybrid_fp16_mla_6x32(const CPUInfo *ci)
     {
+        switch(ci->get_cpu_model()) {
+            default:
+                break;
+            case CPUModel::A55r1:
+                kernel=a64_hybrid_fp16_mla_6x32_a55;
+                break;
+        }
     }
 };
 
-- 
cgit v1.2.1