2

我已经编写了一些在 Haswell i7 处理器上运行的 AVX2 代码。相同的代码库也用于非 Haswell 处理器，其中相同的代码应替换为其 SSE 等效项。我想知道编译器是否有办法忽略非 Haswell 处理器上的 AVX2 指令。我需要类似的东西：

public void useSSEorAVX(...){
    IF (compiler directive detected AVX2)
        AVX2 code (this part is ready)
    ELSE
        SSE code  (this part is also ready)
    }
}

现在我在编译之前注释掉相关代码，但必须有一些更有效的方法来做到这一点。我正在使用 Ubuntu 和 gcc。谢谢你的帮助。

4

2 回答 2

16

除非必须，否则我认为制作单独的可执行文件不是一个好主意。在您的情况下，您可以制作一个 CPU 调度程序。我最近为 GCC 和 Visual Studio 做了这个。

假设您有一个调用productSSE 和 AVX 的函数。您将 SSE 版本放在文件 product_SSE.cpp 中，将 AVX2 版本放在文件 product_AVX2.cpp 中。您分别编译每一个（例如，使用-msse2and -mavx2）。然后制作一个这样的模块：

extern "C" void product_SSE(float *a, float *b, float *c, int n);
extern "C" void product_AVX2(float *a, float *b, float *c, int n); 
           void product_dispatch(float *a, float *b, float *c, int n); 
void (*fp)(float* a, float *b, float *c, int n) = product_dispatch;

inline void product_dispatch(float *a, float *b, float *c, int n) {
    int iset = instrset_detect();
    if(iset==8) {
        fp = product_AVX2
    }
    else {
        fp = product_SSE
    }
    fp(a,b,c,n);
}

inline void product(float *a, float *b, float*c, int bs) {
    fp(a,b,c,n);
}

您使用较低的通用指令集（例如使用 SSE2）编译该模块。现在，当您调用 product 时，它首先调用product_dispatch将函数指针fp设置为product_AVX2orproduct_SSE然后从函数指针调用函数。第二次调用product它会直接跳转到product_AVX2or product_SSE。这样您就不必拥有单独的可执行文件。

于 2014-05-15T12:13:26.080 回答
5

如果您只想在编译时执行此操作，则可以执行以下操作：

#ifdef __AVX2__
    // AVX2 code
#elif __SSE__
    // SSE code
#else
    // scalar code
#endif

请注意，当您编译时gcc -mavx2 ...then__AVX2__会自动定义。同样对于__SSE__。（另请注意，您可以使用 incanation 检查编译器为任何给定的命令行切换预定义的内容gcc -dM -E -mavx2 - < /dev/null。）

但是，如果您想进行运行时分派，那就有点复杂了。

于 2014-05-15T11:24:59.900 回答