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