AVX2在栈与堆内存使用中的差异及段错误问题排查

我用AVX2实现两个数组相加，数组在栈内存声明时运行正常；但用堆内存分配时，编译能通过但运行触发段错误，编译还会出现警告：AVX vector return without AVX enabled changes the ABI [-Wpsabi]。测试环境是支持AVX2的i7-8550U，编译参数为-march=native -mavx2。

栈内存（静态内存）可用示例代码 #

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

int main()
{
    __attribute__((aligned(32))) uint32_t buffer1[8];
    __attribute__((aligned(32))) uint32_t buffer2[8];
    __attribute__((aligned(32))) uint32_t buffer3[8];

    __m256i* buffer1_s = (__m256i*)buffer1;
    __m256i* buffer2_s = (__m256i*)buffer2;
    __m256i* buffer3_s = (__m256i*)buffer3;

    for (uint32_t i = 0; i < 8; ++i) {
        buffer1[i] = rand() % 64;
        buffer2[i] = rand() % 64;
    }

    *buffer3_s = _mm256_add_epi32(*buffer1_s, *buffer2_s);

    for (uint32_t i = 0; i < 8; ++i) {
        printf("%d ", buffer1[i]);
        printf("%d ", buffer2[i]);
        printf("%d ", buffer3[i]);
        printf("\n");
    }
}

堆内存（动态内存）分配错误示例代码 #

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

int main()
{
    __attribute__((aligned(32))) uint32_t* buffer1 = malloc(8 * sizeof(uint32_t));
    __attribute__((aligned(32))) uint32_t* buffer2 = malloc(8 * sizeof(uint32_t));
    __attribute__((aligned(32))) uint32_t* buffer3 = malloc(8 * sizeof(uint32_t));

    __m256i* buffer1_s = (__m256i*)buffer1;
    __m256i* buffer2_s = (__m256i*)buffer2;
    __m256i* buffer3_s = (__m256i*)buffer3;

    for (uint32_t i = 0; i < 8; ++i) {
        buffer1[i] = rand() % 64;
        buffer2[i] = rand() % 64;
    }

    *buffer3_s = _mm256_add_epi32(*buffer1_s, *buffer2_s);

    for (uint32_t i = 0; i < 8; ++i) {
        printf("%d ", buffer1[i]);
        printf("%d ", buffer2[i]);
        printf("%d ", buffer3[i]);
        printf("\n");
    }

    free(buffer1);
    free(buffer2);
    free(buffer3);
}

编译命令 #

gcc -O0 -g -march=native -mavx2 -o test_avx2 test_avx2.c

问题解答 #

1. 是什么导致了段错误？ #

AVX2指令要求操作的内存必须按32字节对齐，但malloc仅能保证默认的内存对齐（通常为8或16字节），无法满足AVX2的对齐要求。而代码中给指针添加的__attribute__((aligned(32)))属性无效——该属性只能约束变量本身的存储对齐（比如栈上的数组），无法改变malloc返回内存块的实际对齐方式。当AVX2指令访问未对齐的内存时，就会触发段错误。

2. 在malloc分配的内存上正确使用AVX2的方式是什么？ #

需要使用支持指定对齐的内存分配函数，比如aligned_alloc（C11标准）或posix_memalign（POSIX标准），确保分配的内存按32字节对齐。

原错误代码行：

__attribute__((aligned(32))) uint32_t* buffer1 = malloc(8 * sizeof(uint32_t));

替换为正确的写法：

uint32_t* buffer1 = aligned_alloc(32, 8 * sizeof(uint32_t));

注意：使用aligned_alloc时，分配的大小必须是对齐值的整数倍；使用posix_memalign时，需要传入指针的指针，且对齐值必须是2的幂次方。

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