1

我有以下代码旨在使用 c++ 中的流式 SIMD 扩展通过数组执行线性搜索：

#include <iostream>
#include <emmintrin.h>

using namespace std;

bool sse2_search_array(int* arr, int size, int key) {
    int iterations;
    if (size % 16 == 0) {
        iterations = size / 16;
    }
    else {
        iterations = size / 16 + 1;
    }
    __m128i* arr_ = reinterpret_cast<__m128i*>(arr);  /*Cast to corresponding int type for 128 bit registers. Each __m128i
                                occupies 8 bits, so 16 integers can be processed simultaneously.*/
    __declspec(align(16)) int key_arr[16];
    fill_n(key_arr, 16, key);  /*fill key array with 16 keys (for SSE comparisons)*/
    __m128i* key_arr_ = reinterpret_cast<__m128i*>(key_arr);

    int result;
    /*Actual search begins here.*/
    for (int i = 0; i < iterations; i++, arr_++) {
        result = _mm_movemask_epi8(_mm_cmpeq_epi8( *key_arr_, *arr_));  /*Comparison of 2 16 bit arrays simultaneously.*/
        cout << "result: " << result << endl;
        if (result != 0) { return true; }
    }
    return false;

}

int main() {
    __declspec(align(16)) int example_array[16] = { 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6 };

    cout << "found: " << sse2_search_array(example_array, 16, 128);
    cin.get();
}

它可以工作，但主函数中的示例应该返回 false，因为 128 不在example_array，但sse2_search_array似乎总是返回 true，result示例中的值为 1110111011101110b 或 61166，这对我来说没有意义，因为我期待它为0。那么有人可以告诉我问题是什么以及如何解决它吗？我对 C++ 不是很有经验，对 SSE 知之甚少。

4

1 回答 1

3

两个主要问题：

永远不要填充临时数组，以便您可以将其作为向量加载：

__declspec(align(16)) int key_arr[16];
fill_n(key_arr, 16, key);  /*fill key array with 16 keys (for SSE comparisons)*/
__m128i* key_arr_ = reinterpret_cast<__m128i*>(key_arr);

相反，使用__m128i keyvec = _mm_set1_epi8(key);. 与将 16 次标量存储到内存然后进行向量加载（这将遭受存储转发停顿）相比，将一个字节广播到向量的所有位置要快得多。让编译器通过使用_mm_set内在函数而不是写入本地数组来为您选择。

int是 4 个字节（在所有现代 x86 编译器上），但您显然希望使用单字节元素数组，因为您使用的是_mm_cmpeq_epi8. 你example_array实际上是 16 * 4 字节长：

__declspec(align(16)) int example_array[16] = { 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6 };
// equivalent byte array (on little-endian x86):
uint8_t byte_array[16*sizeof(int)] = { 1,0,0,0,  2,0,0,0,  3,0,0,0, ... };

您的评论通常是完全错误的，例如Comparison of 2 16 bit arrays simultaneously。也许您的意思是“字节”？

如果您真的想搜索 的数组int，请使用_mm_set1_epi32(key)和_mm_cmpeq_epi32。一个 16 字节的向量包含四个ints。movemask 结果仍然基于字节，但结果中的每组 4 位将是相同的。

另请参阅标签 wiki 和标签 wiki 以获得有用的链接。标签 wiki 有很多关于语言的好东西，因为你说你也是新手。

IDK 为什么你得到 key=128 的点击；除非您的代码有更多我没有注意到的错误，否则这似乎没有意义。

您的调试器应该能够向您显示__m128i变量中的内容。在变量中存储一些临时变量将使使用 C++ 源代码级调试器更容易查看它们，而不是单步执行 asm 代码。

于 2016-06-25T23:20:17.047 回答