如何使用LLVM生成含AVX SIMD内联函数的LLVM IR？

要生成对应AVX SIMD内联函数的LLVM IR，核心是在C代码中显式使用AVX内联函数，并配合指定AVX指令集的编译选项，具体步骤如下：

1. 在C代码中使用AVX内联函数 #

AVX的内联函数定义在<immintrin.h>头文件中，你可以直接调用这类函数编写AVX逻辑，比如用AVX2的256位整数加法实现向量操作：

#include <immintrin.h>

void avx_vector_add(int *A, int *B, int length) {
    // 按AVX2的256位向量长度（8个int）循环处理数据
    for (int i = 0; i < length; i += 8) {
        // 从内存加载非对齐的256位向量
        __m256i b_vec = _mm256_loadu_si256((__m256i*)&B[i]);
        // 给向量每个元素加1
        __m256i add_result = _mm256_add_epi32(b_vec, _mm256_set1_epi32(1));
        // 将结果存回内存
        _mm256_storeu_si256((__m256i*)&A[i], add_result);
    }
}

2. 编译时指定AVX指令集 #

Clang默认不会启用AVX指令集，你需要通过编译选项明确告知编译器目标架构支持AVX/AVX2，否则内联函数可能会被降级为标量代码或其他通用SIMD指令。

对应的编译命令：

clang -O3 -mavx2 -S -emit-llvm test.c -o test.ll

• -mavx2：启用AVX2指令集支持；若仅需基础AVX（128位），替换为-mavx即可
• -O3：保持优化级别，确保内联函数被正确展开为对应LLVM IR指令

3. 生成的LLVM IR示例 #

编译完成后，你会在test.ll中看到对应AVX内联函数的LLVM IR指令，比如_mm256_add_epi32会被转换为：

%add_result = call <8 x i32> @llvm.x86.avx2.vpadd.d(<8 x i32> %b_vec, <8 x i32> %1)

这类指令直接对应AVX2的硬件加法操作，和自动向量化生成的通用SIMD指令有明确区别。

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