如何仅对源代码的特定部分打开-mavx2？

Question

我可以强制编译器在代码的某些特定部分中在-march设置之外编译一些内部函数吗？

当然，其余的将保留在-march设置之内。

是否可以仅在源代码的特定部分启用-mavx2？

还是我必须分别编译-mavx2部分的唯一方法？

Answer 1

尝试__attribute__((target("avx2")))。 GCC和Clang都支持。

示例：

#include <stdlib.h>
#include <stdio.h>
#include <immintrin.h>

__attribute__((target("avx2")))
int add_with_avx2(int a, int b) {
    __m256i av = _mm256_set_epi32(a, 0, 0, 0, 0, 0, 0, 0);
    __m256i bv = _mm256_set_epi32(b, 0, 0, 0, 0, 0, 0, 0);
    __m256i result = _mm256_add_epi32(av, bv);
    return ((int*)&result)[7];
}

int main(void) {
    return add_with_avx2(5, 6);
}

但是，如果需要使用没有此功能的编译器，将需要内部函数的函数放在单独的文件中可能是一个更好的主意。