From 5f707736413aeac77818c42838296966f8dc6761 Mon Sep 17 00:00:00 2001 From: Anthony Barbier Date: Tue, 3 Jul 2018 16:22:02 +0100 Subject: COMPMID-1369: Revert accidental formatting of RSH's repo Pulled latest fixes from David's repo: commit f43ebe932c84083332b0b1a0348241b69dda63a7 Author: David Mansell Date: Tue Jul 3 18:09:01 2018 +0100 Whitespace tidying, fixed comment in gemv_batched imported from ACL. Change-Id: Ie37a623f44e90d88072236cb853ac55ac82d5f51 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/138530 Tested-by: Jenkins Reviewed-by: Georgios Pinitas Reviewed-by: Gian Marco Iodice Reviewed-by: David Mansell Reviewed-by: Anthony Barbier --- src/core/NEON/kernels/arm_gemm/gemm_fp32.cpp | 27 +++++++++++++-------------- 1 file changed, 13 insertions(+), 14 deletions(-) (limited to 'src/core/NEON/kernels/arm_gemm/gemm_fp32.cpp') diff --git a/src/core/NEON/kernels/arm_gemm/gemm_fp32.cpp b/src/core/NEON/kernels/arm_gemm/gemm_fp32.cpp index c093761614..99f061bde8 100644 --- a/src/core/NEON/kernels/arm_gemm/gemm_fp32.cpp +++ b/src/core/NEON/kernels/arm_gemm/gemm_fp32.cpp @@ -29,15 +29,15 @@ #include "gemv_native_transposed.hpp" #include "gemv_pretransposed.hpp" -#include "kernels/a32_sgemm_8x6.hpp" #include "kernels/a64_sgemm_12x8.hpp" -#include "kernels/a64_sgemm_native_16x4.hpp" -#include "kernels/a64_sgemv_pretransposed.hpp" +#include "kernels/a32_sgemm_8x6.hpp" #include "kernels/a64_sgemv_trans.hpp" +#include "kernels/a64_sgemv_pretransposed.hpp" +#include "kernels/a64_sgemm_native_16x4.hpp" -namespace arm_gemm -{ -template <> +namespace arm_gemm { + +template<> UniqueGemmCommon gemm(const CPUInfo &ci, const unsigned int M, const unsigned int N, const unsigned int K, const unsigned int nbatches, const unsigned int nmulti, const bool trA, const bool trB, const float alpha, const float beta, @@ -46,18 +46,17 @@ UniqueGemmCommon gemm(const CPUInfo &ci, const unsig if (M==1 && nbatches>1) { return UniqueGemmCommon (new GemvBatched(ci, M, N, K, nbatches, nmulti, trA, trB, alpha, beta, maxthreads, pretransposed_hint)); } + #ifdef __aarch64__ /* Cases in priority order */ /* GemvPretransposed: requires M=1, alpha=1, and transposed hint set. nbatches must be 1 or we would have returned above so don't test. */ - if(M == 1 && alpha == 1.0f && pretransposed_hint) - { - return UniqueGemmCommon(new GemvPretransposed(&ci, N, K, nmulti, trB, beta)); + if (M==1 && alpha==1.0f && pretransposed_hint) { + return UniqueGemmCommon (new GemvPretransposed(&ci, N, K, nmulti, trB, beta)); } /* GemvNativeTransposed: requires M=1, no trA or trB, doesn't handle alpha */ - if(M == 1 && alpha == 1.0f && !trA && !trB) - { - return UniqueGemmCommon(new GemvNativeTransposed(&ci, N, K, nmulti, beta)); + if (M==1 && alpha==1.0f && !trA && !trB) { + return UniqueGemmCommon (new GemvNativeTransposed(&ci, N, K, nmulti, beta)); } /* Native GEMM: requires K at least 4, N a multiple of 16, doesn't @@ -69,9 +68,9 @@ UniqueGemmCommon gemm(const CPUInfo &ci, const unsig } /* Blocked GEMM, handles all cases. */ - return UniqueGemmCommon(new GemmInterleaved(&ci, M, N, K, nbatches, nmulti, trA, trB, alpha, beta, maxthreads, pretransposed_hint)); + return UniqueGemmCommon (new GemmInterleaved(&ci, M, N, K, nbatches, nmulti, trA, trB, alpha, beta, maxthreads, pretransposed_hint)); #else - return UniqueGemmCommon(new GemmInterleaved(&ci, M, N, K, nbatches, nmulti, trA, trB, alpha, beta, maxthreads, pretransposed_hint)); + return UniqueGemmCommon (new GemmInterleaved(&ci, M, N, K, nbatches, nmulti, trA, trB, alpha, beta, maxthreads, pretransposed_hint)); #endif } -- cgit v1.2.1