diff options
Diffstat (limited to 'src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation_constraints.hpp')
-rw-r--r-- | src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation_constraints.hpp | 8 |
1 files changed, 7 insertions, 1 deletions
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 *) { |