/ lib / numpy / distutils / checks / cpu_asimddp.c
cpu_asimddp.c
 1  #ifdef _MSC_VER
 2      #include <Intrin.h>
 3  #endif
 4  #include <arm_neon.h>
 5  
 6  int main(int argc, char **argv)
 7  {
 8      unsigned char *src = (unsigned char*)argv[argc-1];
 9      uint8x16_t v1 = vdupq_n_u8(src[0]), v2 = vdupq_n_u8(src[1]);
10      uint32x4_t va = vdupq_n_u32(3);
11      int ret = (int)vgetq_lane_u32(vdotq_u32(va, v1, v2), 0);
12  #ifdef __aarch64__
13      ret += (int)vgetq_lane_u32(vdotq_laneq_u32(va, v1, v2, 0), 0);
14  #endif
15      return ret;
16  }