From c7b183ab741650653289f8ce3bdeb4926521fdbd Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Fri, 6 Mar 2020 18:12:09 +0000 Subject: COMPMID-3160: Add Bfloat16 support in NEGEMMConvolutionLayer Signed-off-by: Georgios Pinitas Change-Id: I0e449306c138a562ffc1455e76ec44b2fd059d85 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2860 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Comments-Addressed: Arm Jenkins --- src/core/NEON/kernels/arm_gemm/gemm_fp32.cpp | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) (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 35493a609c..96e3ce832c 100644 --- a/src/core/NEON/kernels/arm_gemm/gemm_fp32.cpp +++ b/src/core/NEON/kernels/arm_gemm/gemm_fp32.cpp @@ -44,6 +44,7 @@ #include "kernels/sve_hybrid_fp32_mla_4VLx4.hpp" #include "kernels/sve_interleaved_fp32_mla_3VLx8.hpp" #include "kernels/sve_native_fp32_mla_4VLx4.hpp" +#include "kernels/sve_smallK_hybrid_fp32_mla_1VLx8.hpp" namespace arm_gemm { @@ -73,7 +74,14 @@ static const GemmImplementation gemm_fp32_methods[] = }, #ifdef __ARM_FEATURE_SVE -// SVE native / hybrid methods +// SVE smallk / native / hybrid methods +{ + GemmMethod::GEMM_HYBRID, + "smallK_hybrid_fp32_mla_1VLx8", + [](const GemmArgs &args) { return (args._Ksize <= 24) && !args._trA && args._pretransposed_hint; }, + nullptr, + [](const GemmArgs &args) { return new GemmHybrid(args); } +}, { GemmMethod::GEMM_HYBRID, "hybrid_fp32_mla_4VLx4", -- cgit v1.2.1