cpu_asimddp.c 432 B

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