Question

如何使用SSE反转字符串？这个概念对我来说是新的，所以请给我一些有关它的信息。原因是有人说使用SSE会加快代码和运行时间。

我搜索了SSE，这是_mm128，但实际上并不知道如何将它们分解。

谢谢

Answer 1

如果你可以假设最小的SSSE3然后使用_mm_shuffle_epi8非常容易，但在你这样做之前你应该确保（a）你当前的实现是性能瓶颈而且（b）你有尽你所能，尽可能快地完成当前的实现，因为这是一个非常简单的操作，它的性能应该只受到任何体面实现的内存带宽的限制。

无论如何，这是一个简单的实现和测试工具：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <tmmintrin.h> // SSSE3

void reverse(char *begin, char *end)
{
    while (begin < end)
    {
        const char c = *begin;
        *begin = *end;
        *end = c;
        ++begin;
        --end;
    }
}

void vreverse(char *begin, char *end)
{
    const __m128i vrev = _mm_set_epi8(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15);

    ssize_t len = end - begin + 1;

    while (len >= 16)
    {
        __m128i v1 = _mm_loadu_si128((__m128i *)begin);
        __m128i v2 = _mm_loadu_si128((__m128i *)(end - 15));
        v1 = _mm_shuffle_epi8(v1, vrev);
        v2 = _mm_shuffle_epi8(v2, vrev);
        _mm_storeu_si128((__m128i *)(end - 15), v1);
        _mm_storeu_si128((__m128i *)begin, v2);
        begin += 16;
        end -= 16;
        len -= 32;
    }
    if (len > 1)
    {
        reverse(begin, end);
    }
}

int main(void)
{
    const size_t MAX_LEN = 64;
    char s1[MAX_LEN + 1], s2[MAX_LEN + 1];
    size_t i, len;

    for (len = 0; len < MAX_LEN; ++len)
    {
        for (i = 0; i < len; ++i)
        {
            s1[i] = s2[i] = (char)('a' + rand() % 26);
        }
        s1[len] = s2[len] = '\0';
        reverse(s1, s1 + len - 1);
        vreverse(s2, s2 + len - 1);
        if (memcmp(s1, s2, len) != 0)
        {
            printf("FAIL: len = %zu\n", len);
            printf("FAIL: s1 = %s\n", s1);
            printf("FAIL: s2 = %s\n", s2);
        }
        else
        {
            //printf("PASS: len = %zu\n", len);
        }
    }
    return 0;
}

测试它：

$ gcc -Wall -mssse3 -O3 vreverse.c && ./a.out
$

使用SSE反转字符串

1 个答案: