SSE4内存与差异位置

时间:2017-09-03 21:24:19

标签: c++ optimization sse

在尝试使用SSE4优化跟随时,我自己的实现会让我感到厌烦:

std::distance(byteptr, std::mismatch(byteptr, ptr + lenght, dataptr).first)

这会比较byteptr和data并返回字节不匹配的索引。 我确实需要原始速度,因为我处理了这么多内存,RAM速度已经成为瓶颈。通过SSE4获取和比较16个字节可以提高速度,因为比较16个字节的速度更快。

这是我目前的代码,我无法正常工作。 它使用GCC SSE内在函数,需要SSE4.2:

// define SIMD 128-bit type of bytes.
typedef char v128i __attribute__ ((vector_size(16)));
// mask of four low bits set.
const uintptr_t aligned_16_imask = (uintptr_t)15;
// mask of four low bits unset.
const uintptr_t aligned_16_mask = ~aligned_16_imask;

inline unsigned int cmp_16b_sse4(v128i *a, v128i *b) {
    return __builtin_ia32_pcmpistri128(__builtin_ia32_lddqu((char*)a), *b, 0x18);  
}

size_t memcmp_pos(const char * ptr1, const char * ptr2, size_t lenght)
{
    size_t nro = 0;
    size_t cmpsz;
    size_t alignlen = lenght & aligned_16_mask;
    // process 16-bytes at time.
    while(nro < alignlen) {
        cmpsz = cmp_16b_sse4((v128i*)ptr1, (v128i*)ptr2);
        ptr1 += cmpsz;
        ptr2 += cmpsz;
        nro += cmpsz;
        // if compare failed return now.
        if(cmpsz < 16)
            return nro;
        if(cmpsz != 16)
            break;
    }
    // process remainder 15 bytes:
    while( *ptr1 == *ptr2 && nro < lenght) {
        ++nro;
        ++ptr1;
        ++ptr2;
    }
    return nro;
}

在测试上述功能时,它大部分时间都有效,但在某些情况下会失败。

1 个答案:

答案 0 :(得分:2)

pcmpistri的一个已知问题是它总是读取完整的16个字节 - 甚至超出变量的末尾。这在页面边界上,在分配给未分配的内存的边界上成为问题。请参阅here (scroll down to "Renat Saifutdinov")

即使支持未对齐的读取,see this SO answer也可以使用源的对齐读取来避免这种情况。

这可能是您的代码失败的可能性之一。