c++ - 在 C++ 数组快速 SIMD 版本中寻找短值

Question

我的程序中有一个算法运行良好。我推测是否有可能加快思考速度:

unsigned short c;   
bool found = false;
unsigned short* arrIterator = arr;
while(( c = *arrIterator & mask) != stopValue)
{       
    if(c == next)
    {               
        found= true;
        break;                  
    }
    arrIterator ++;
}   

是否可以将此类算法重写为 SIMD 指令？

Answer 1

假设 arr 是 16 对齐的(让它如此)，你可以做这样的事情(未测试)

__m128i vstop = _mm_set1_epi16(stopValue);
__m128i vnext = _mm_set1_epi16(next);
int found_mask = 0;
int stop_mask = 0;
do
{
    __m128i data = _mm_load_si128(arrIterator++);
    __m128i contains_next = _mm_cmpeq_epi16(data, vnext);
    __m128i contains_stop = _mm_cmpeq_epi16(data, vstop);
    found_mask = _mm_movemask_epi8(contains_next);
    stopmask = found_mask | _mm_movemask_epi8(contains_stop);
} while (stopmask == 0);

然后，您可以通过对 found_mask 和一些具有迭代器当前值的内容进行位扫描来告诉索引在哪里找到了 next。