From d93991e290618a685b67506c78090350e6aee43f Mon Sep 17 00:00:00 2001 From: David Mansell 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 Tested-by: Jenkins --- .../kernels/arm_gemm/kernels/a64_gemm_u16_12x8.hpp | 27 ++++++++++++---------- 1 file changed, 15 insertions(+), 12 deletions(-) (limited to 'src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u16_12x8.hpp') diff --git a/src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u16_12x8.hpp b/src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u16_12x8.hpp index 13dd570677..d2692ba77f 100644 --- a/src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u16_12x8.hpp +++ b/src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u16_12x8.hpp @@ -25,6 +25,8 @@ #ifdef __aarch64__ +#include "../std_transforms_fixed.hpp" + namespace arm_gemm { // Actual kernel implementations @@ -45,20 +47,21 @@ public: typedef void (*kern_type)(const uint16_t *, const uint16_t *, uint32_t *, int, int, int); - /* Describes the data layout for A input */ - static const int A_interleave = 8; - static const int A_block = 1; - static const int A_transpose = 0; + /* Kernel blocking parameters */ + static int out_width() { + return 12; + } + + static int out_height() { + return 8; + } - /* Same for B input */ - static const int B_interleave = 12; - static const int B_block = 1; - static const int B_transpose = 1; + static int k_unroll() { + return 1; + } - /* Kernel blocking parameters */ - static const int out_width = 12; - static const int out_height = 8; - static const int k_unroll = 1; + // Use the standard fixed size transforms. + StdTransformsFixed transforms = {}; kern_type kernel = a64_gemm_u16_asimd_12x8; -- cgit v1.2.1