Visual Studio 2017 循环自动矢量化问题

Question

我正在尝试使用 Visual C++ 2017 自动矢量化器将以下循环矢量化 (/arch:AVX2)：

void fun(char* data, char* threshold, char* output, int64_t len)
{
    // Assumes output filled with 0

    for (int64_t c = 0, mm = len; c < mm; ++c)
    {
        output[c] = (data[c] < threshold[c])
                            ? (threshold[c] - data[c]) 
                            : output[c];
    }
}

此代码用于比较2个数组（数据和阈值），如果数据

此循环不会自动矢量化：

info C5002：由于原因“1100”，循环未矢量化

含义：循环包含控制流，例如“if”或“?”。

好的，我明白了，我需要重写我的循环，以便删除控制流或为编译器简化它。但是：

• GCC 对它进行矢量化没有问题

• 如果我以这种方式更改代码，Visual Studio 接受对其进行矢量化：

代码：

for (int64_t c = 0, mm = len; c < mm; ++c)
{
    output[c] = (data[c] < threshold[c])
                        ? (char)(threshold[c] - data[c]) 
                        : output[c];
}

为什么这个 (char) cast 会改变 Visual Studio 自动矢量化器的任何内容？这是自动矢量化器的错误还是我遗漏了什么？

此外，如果我将输出数组的类型从 char 更改为 int，我无法再让 Visual Studio 对我的循环进行矢量化，而 GCC 可以：

void fun(char* data, char* threshold, int* output, int64_t len)
{
    // Assumes output filled with 0
    for (int64_t c = 0, mm = len; c < mm; ++c)
    {
        output[c] = (data[c] < threshold[c])
                            ? (int)(threshold[c] - data[c]) 
                            : output[c];
    }
}

与 GCC 相比是否缺少 Visual Studio 2017 自动矢量化器？还是我在尝试做一些我不应该做的事情？

Answer 1

这只是一个错失的优化机会。 VS 矢量化器自 2012 年以来有了很大的改进，但与 gcc 或 clang 相比仍然相当缺乏。请记住，他们的编译器基于古老的代码库，例如直到最近，他们甚至都没有SSA 的代表。