diff options
Diffstat (limited to 'src/core/NEON/kernels/arm_gemm/kernels/a32_sgemm_8x6.hpp')
-rw-r--r-- | src/core/NEON/kernels/arm_gemm/kernels/a32_sgemm_8x6.hpp | 27 |
1 files changed, 15 insertions, 12 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/kernels/a32_sgemm_8x6.hpp b/src/core/NEON/kernels/arm_gemm/kernels/a32_sgemm_8x6.hpp index 01bf1f9297..06e62456dc 100644 --- a/src/core/NEON/kernels/arm_gemm/kernels/a32_sgemm_8x6.hpp +++ b/src/core/NEON/kernels/arm_gemm/kernels/a32_sgemm_8x6.hpp @@ -25,6 +25,8 @@ #ifdef __arm__ +#include "../std_transforms_fixed.hpp" + namespace arm_gemm { // Actual kernel implementations @@ -47,20 +49,21 @@ public: typedef void (*kern_type)(const float *, const float *, float *, int, int, int); - /* Describes the data layout for A input */ - static const int A_interleave = 6; - static const int A_block = 1; - static const int A_transpose = 0; + /* Kernel blocking parameters */ + static int out_width() { + return 8; + } - /* Same for B input */ - static const int B_interleave = 8; - static const int B_block = 1; - static const int B_transpose = 1; + static int out_height() { + return 6; + } - /* Kernel blocking parameters */ - static const int out_width = 8; - static const int out_height = 6; - static const int k_unroll = 1; + static int k_unroll() { + return 1; + } + + // Use the standard fixed size transforms. + StdTransformsFixed<operand_type, result_type, 6, 8> transforms = {}; kern_type kernel = a32_sgemm_8x6; |