如何在C++中结合使用constexpr与向量化代码？

核心依赖C++20标准引入的std::is_constant_evaluated()（定义于<type_traits>头文件），该函数可以判断当前代码是否处于常量求值上下文，且运行期会被优化为常量false，无额外开销。

实现要点 #

• 不能使用if constexpr匹配分支，否则编译器会要求两个分支都符合constexpr语法规则，导致非constexpr的intrinsic调用直接报错
• 常量求值分支写纯C++可移植实现，无需加编译制导语句、intrinsic调用
• 运行期分支根据架构选择intrinsic实现或OpenMP向量化实现
• 编译器会自动丢弃未匹配的分支，无任何运行时性能损耗

适配示例代码 #

首先是你提到的FA结构体的实现：

#include <type_traits>

struct FA {
    float c[4];
};

inline constexpr FA add(FA a, FA b) noexcept {
    if (std::is_constant_evaluated()) {
        // 编译期走纯C++参考实现，无任何平台相关代码
        FA result{};
        for (int i = 0; i < 4; ++i) {
            result.c[i] = a.c[i] + b.c[i];
        }
        return result;
    } else {
        // 运行期走向量化实现，可自由用OpenMP或编译器扩展
        FA result{};
        #pragma omp simd
        for (int i = 0; i < 4; ++i) {
            result.c[i] = a.c[i] + b.c[i];
        }
        return result;
    }
}

然后是带SSE/Neon intrinsic的FA2实现，可通过宏做架构适配：

#ifdef __x86_64__
#include <xmmintrin.h>
struct FA2 {
    __m128 c;
};
#elif defined(__ARM_NEON)
#include <arm_neon.h>
struct FA2 {
    float32x4_t c;
};
#endif

inline constexpr FA2 add2(FA2 a, FA2 b) noexcept {
    if (std::is_constant_evaluated()) {
        // 编译期走纯C++实现，转成数组操作后再转回对应向量类型
        alignas(alignof(FA2)) float a_arr[4], b_arr[4], res_arr[4];
        __builtin_memcpy(a_arr, &a.c, sizeof(a_arr));
        __builtin_memcpy(b_arr, &b.c, sizeof(b_arr));
        for (int i = 0; i < 4; ++i) {
            res_arr[i] = a_arr[i] + b_arr[i];
        }
        FA2 res{};
        __builtin_memcpy(&res.c, res_arr, sizeof(res.c));
        return res;
    } else {
        // 运行期直接调用对应架构intrinsic
        FA2 res{};
    #ifdef __x86_64__
        res.c = _mm_add_ps(a.c, b.c);
    #elif defined(__ARM_NEON)
        res.c = vaddq_f32(a.c, b.c);
    #endif
        return res;
    }
}

兼容性说明 #

• 所有支持C++20的编译器（GCC 9+, Clang 10+, MSVC 2019 16.6+）均支持该特性
• 运行期分支的intrinsic调用不会触发constexpr语法检查，完美解决GCC下intrinsic非constexpr的问题
• 常量求值分支的纯C++实现保证全平台可移植性

验证方式 #

// 编译期求值，走参考实现
constexpr FA compile_time_res = add(FA{1,2,3,4}, FA{5,6,7,8});

// 运行期求值，走向量化/intrinsic实现
FA runtime_add(FA x, FA y) {
    return add(x, y);
}

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