如何在SSE编译模块中按需使用AVX/AVX512内联函数？

GCC & Clang #

可以通过局部指令集控制，让编译器仅在指定代码块/函数中允许使用高级SIMD intrinsics，全局保持低版本指令集编译：

方法1：用#pragma GCC target局部指定指令集 #

将不同SIMD分支的代码包裹在单独代码块中，通过pragma临时开启对应指令集支持：

void my_math_func(void *raw_pointers_to_data)
{
    if (CpuSupportsAvx512()) {
        #pragma GCC target("avx512f")
        {
            // 此处可直接使用__m512及_mm512_*系列intrinsics
            __m512 vec = _mm512_loadu_ps((float*)raw_pointers_to_data);
            // ... 后续AVX512处理逻辑
        }
    } else if(CpuSupportsAvx2()) {
        #pragma GCC target("avx2")
        {
            // AVX2专属处理代码
            __m256i vec = _mm256_loadu_si256((__m256i*)raw_pointers_to_data);
            // ...
        }
    } else {
        // SSE4.2代码，全局编译选项-msse4.2已支持
        __m128i vec = _mm_loadu_si128((__m128i*)raw_pointers_to_data);
        // ...
    }
}

编译时仅需全局指定-msse4.2，pragma会让编译器在对应代码块中允许使用指定intrinsics，且不会全局生成高级指令集代码。

方法2：拆分辅助函数并添加指令集属性 #

把各分支逻辑拆为静态辅助函数，用__attribute__((target("指令集")))标记：

static __attribute__((target("avx512f"))) void avx512_process(void *data)
{
    // AVX512处理逻辑
}

static __attribute__((target("avx2"))) void avx2_process(void *data)
{
    // AVX2处理逻辑
}

static void sse4_process(void *data)
{
    // SSE4.2处理逻辑
}

void my_math_func(void *raw_pointers_to_data)
{
    if (CpuSupportsAvx512()) {
        avx512_process(raw_pointers_to_data);
    } else if(CpuSupportsAvx2()) {
        avx2_process(raw_pointers_to_data);
    } else {
        sse4_process(raw_pointers_to_data);
    }
}

编译时全局用-msse4.2，编译器会为带target属性的函数生成对应指令集代码，不影响其他函数的指令集级别。

MSVC (VS2019) #

MSVC通过__declspec(target)标记函数，实现局部指令集控制：

static __declspec(target("avx512f")) void avx512_process(void *data)
{
    // 此处使用AVX512 intrinsics
    __m512 vec = _mm512_loadu_ps((float*)data);
    // ...
}

static __declspec(target("avx2")) void avx2_process(void *data)
{
    // AVX2处理逻辑
    __m256i vec = _mm256_loadu_si256((__m256i*)data);
    // ...
}

static void sse4_process(void *data)
{
    // SSE4.2处理逻辑
    __m128i vec = _mm_loadu_si128((__m128i*)data);
    // ...
}

void my_math_func(void *raw_pointers_to_data)
{
    if (CpuSupportsAvx512()) {
        avx512_process(raw_pointers_to_data);
    } else if(CpuSupportsAvx2()) {
        avx2_process(raw_pointers_to_data);
    } else {
        sse4_process(raw_pointers_to_data);
    }
}

编译时全局指定/arch:SSE4.2，带__declspec(target)的函数会生成对应指令集代码，其余部分保持SSE4.2指令级别。

• 必须确保CpuSupportsAvx512()、CpuSupportsAvx2()等检测逻辑准确，只有在CPU支持时才执行对应指令集代码，否则会触发非法指令错误。
• 全局编译仅需指定最低兼容的指令集（如SSE4.2），无需全局开启AVX/AVX512。

内容的提问来源于stack exchange，提问作者FedyuninV