25 #if defined(ARM_COMPUTE_ENABLE_SME2)
26 #include "../std_transforms_sme.hpp"
29 const __fp16 *, const __fp16 *, \
30 __fp16 *, size_t, size_t, \
31 const __fp16 *, Activation, bool
35 void sme2_gemv_fp16fp32fp16_dot_16VL( ARGLIST );
37 class cls_sme2_gemv_fp16fp32fp16_dot_16VL
40 typedef __fp16 operand_type;
41 typedef __fp16 result_type;
43 typedef void (*kern_type)( ARGLIST );
45 static unsigned int out_width()
47 return sme::get_vector_length<float>() * 16;
50 static constexpr
unsigned int k_unroll()
55 static constexpr
bool supports_accumulate()
60 static constexpr
bool supports_bias()
65 static constexpr
bool supports_activation()
71 StdTransformsSME<operand_type, result_type, 1, 16, 2> transforms = {};
75 kern_type kernel=sme2_gemv_fp16fp32fp16_dot_16VL;
76 cls_sme2_gemv_fp16fp32fp16_dot_16VL(
const CPUInfo *)
85 #endif // defined(ARM_COMPUTE_ENABLE_SME2)