numpy/distutils/checks/cpu_asimdfhm.c


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

#ifdef _MSC_VER
    #include <Intrin.h>
#endif
#include <arm_neon.h>

int main(void)
{
    float16x8_t vhp  = vdupq_n_f16((float16_t)1);
    float16x4_t vlhp = vdup_n_f16((float16_t)1);
    float32x4_t vf   = vdupq_n_f32(1.0f);
    float32x2_t vlf  = vdup_n_f32(1.0f);

    int ret  = (int)vget_lane_f32(vfmlal_low_f16(vlf, vlhp, vlhp), 0);
        ret += (int)vgetq_lane_f32(vfmlslq_high_f16(vf, vhp, vhp), 0);

    return ret;
}