From d93991e290618a685b67506c78090350e6aee43f Mon Sep 17 00:00:00 2001
From: David Mansell <David.Mansell@arm.com>
Date: Fri, 6 Jul 2018 14:52:52 +0100
Subject: COMPMID-1380: Pre-work for SVE support.

This patch makes the needed infrastructure changes to allow SVE
kernels to be added later on.

Change-Id: Ide5bccac2f47278e93fff3d648231aee2d5f8c2e
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/139070
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
---
 .../kernels/arm_gemm/kernels/a64_gemm_u8_12x8.hpp     | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

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

diff --git a/src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u8_12x8.hpp b/src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u8_12x8.hpp
index c67aed7275..a252abfd3e 100644
--- a/src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u8_12x8.hpp
+++ b/src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u8_12x8.hpp
@@ -27,6 +27,8 @@
 
 #include "arm_gemm.hpp"
 
+#include "../std_transforms_fixed.hpp"
+
 namespace arm_gemm {
 
 // Load the actual kernel
@@ -51,9 +53,20 @@ public:
     static const bool B_transpose = true;
 
     /* Kernel blocking parameters */
-    static const int out_width = 12;
-    static const int out_height = 8;
-    static const int k_unroll = 4;
+    static int out_width() {
+        return 12;
+    }
+
+    static int out_height() {
+        return 8;
+    }
+
+    static int k_unroll() {
+        return 4;
+    }
+
+    // Use the standard fixed sized transforms.
+    StdTransformsFixed<operand_type, result_type, 8, 12, 4> transforms = {};
 
     kern_type kernel = a64_gemm_u8_12x8;
 
-- 
cgit v1.2.1