From d216f570750b8ccde3754c4aef53fc20a90cb32d Mon Sep 17 00:00:00 2001 From: Freddie Liardet Date: Tue, 3 Aug 2021 15:57:32 +0100 Subject: Update cpu depthwise kernels Resolves: COMPMID-4688 Signed-off-by: Freddie Liardet Change-Id: I9e22f967f5b7ccaebff2fc49f0253f621d62d820 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6030 Tested-by: Arm Jenkins Reviewed-by: Pablo Marquez Tello Reviewed-by: Georgios Pinitas Comments-Addressed: Arm Jenkins --- .../arm_conv/depthwise/depthwise_implementation_constraints.hpp | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation_constraints.hpp') diff --git a/src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation_constraints.hpp b/src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation_constraints.hpp index 6526d001b3..419872744d 100644 --- a/src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation_constraints.hpp +++ b/src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation_constraints.hpp @@ -33,7 +33,7 @@ #pragma once #include "arm_gemm.hpp" -#include "depthwise.hpp" +#include "src/core/NEON/kernels/assembly/depthwise.hpp" namespace arm_conv { @@ -97,6 +97,12 @@ bool cpu_has_sve2(const DepthwiseArgs &args, const void *) return args.cpu_info->has_sve2(); } +bool cpu_has_fp16(const DepthwiseArgs &args, const void *) __attribute__ ((unused)); +bool cpu_has_fp16(const DepthwiseArgs &args, const void *) +{ + return args.cpu_info->has_fp16(); +} + bool has_no_channel_multiplier(const DepthwiseArgs &args, const void *) __attribute__ ((unused)); bool has_no_channel_multiplier(const DepthwiseArgs &args, const void *) { -- cgit v1.2.1