intrinsics를 사용하여 배열에서 0이 아닌 다음을 찾습니다.

Question

int 배열 [10000]을 가지고 있으며 다음 위치에서 반복하여 0이 아닌 인덱스를 찾고자합니다. 현재 내가 기본 while 루프를 사용

while(array[i] == 0){ 
    pos++; 
} 

등

내가 한 번에 제로 4의 정수를 테스트 할 수있는 내장으로 알고 있지만,의 벡터 인덱스를 나타내는 뭔가를 반환하는 방법이입니다 " 첫 번째 "0이 아닌?

Answer 1

그것은이 일을 매우 간단하지만 (배열이 이미 캐시되지 않은 경우) 당신은 아마 메모리 대역폭에 의해 제한되기 때문에 처리량 개선, 중대하지 않을 수 있습니다

int index = -1; 
for (i = 0; i < n; i += 4) 
{ 
    __m128i v = _mm_load_si128(&A[i]); 
    __m128i vcmp = _mm_cmpeq_epi32(v, _mm_setzero_si128()); 
    int mask = _mm_movemask_epi8(vcmp); 
    if (mask != 0xffff) 
    { 
     break; 
    } 
} 
if (i < n) 
{ 
    for (j = i; j < i + 4; ++j) 
    { 
     if (A[j] != 0) 
     { 
      index = j; 
      break; 
     } 
    } 
} 

이 있다고 가정합니다 배열 A 는 16 바이트 정렬이며, 크기는 n이며 4의 배수이며 int는 32 비트입니다.

특히 입력 데이터가 크고/또는 희소성이 낮은 경우 루프를 푸는 것이 도움이 될 수 있습니다. 당신이 (나중에 하 스웰과) AVX2이있는 경우

int index = -1; 
for (i = 0; i < n; i += 8) 
{ 
    __m128i v0 = _mm_load_si128(&A[i]); 
    __m128i v1 = _mm_load_si128(&A[i + 4]); 
    __m128i vcmp0 = _mm_cmpeq_epi32(v0, _mm_setzero_si128()); 
    __m128i vcmp1 = _mm_cmpeq_epi32(v1, _mm_setzero_si128()); 
    int mask0 = _mm_movemask_epi8(vcmp0); 
    int mask1 = _mm_movemask_epi8(vcmp1); 
    if ((mask0 | mask1) != 0xffff) 
    { 
     break; 
    } 
} 
if (i < n) 
{ 
    for (j = i; j < i + 8; ++j) 
    { 
     if (A[j] != 0) 
     { 
      index = j; 
      break; 
     } 
    } 
} 

당신은 4