我不明白使用sse的代码在哪里有问题

时间:2019-12-30 08:31:03

标签: c++ sse avx

我是sse编程的新手。我想编写代码,将向量v中的4个连续的数字相加,然后将此和的结果写入ans向量中。我想使用sse编写优化的代码。但是,当我设置大小等于4时,我的程序正在运行。但是,当我将大小设置为8时,我的程序将无法运行,并且出现以下错误消息: “抛出异常:读取访问冲突。

ans是0x1110112。

如果有用于此异常的处理程序,则该程序可以安全地继续。” 我不明白我在哪里有问题。我分配了内存权,在哪里我有问题。有人可以帮助我,我将非常感激。

#include <iostream>
#include <immintrin.h>
#include <pmmintrin.h>
#include <vector>
#include <math.h>  
using namespace std;
arith_t = double
void init(arith_t *&v, size_t size) {
    for (int i = 0; i < size; ++i) {
        v[i] = i / 10.0;
    }
}
//accumulate with sse
void sub_func_sse(arith_t *v, size_t size, int start_idx, arith_t *ans, size_t start_idx_ans) {
    __m128d first_part = _mm_loadu_pd(v + start_idx);
    __m128d second_part = _mm_loadu_pd(v + start_idx + 2);

    __m128d sum = _mm_add_pd(first_part, second_part);
    sum = _mm_hadd_pd(sum, sum);
    _mm_store_pd(ans + start_idx_ans, sum);
}
int main() {
    const size_t size = 8;
    arith_t *v = new arith_t[size];
    arith_t *ans_sse = new arith_t[size / 4];
    init(v, size);
    init(ans_sse, size / 4);
    int num_repeat = 1;
    arith_t total_time_sse = 0;
    for (int p = 0; p < num_repeat; ++p) {
        for (int idx = 0, ans_idx = 0; idx < size; idx += 4, ans_idx++) {
            sub_func_sse(v, size, idx, ans_sse, ans_idx);
        }
    }

    for (size_t i = 0; i < size / 4; ++i) {
        cout << *(ans_sse + i) << endl;
    }
    delete[] ans_sse;
    delete[] v;
}

1 个答案:

答案 0 :(得分:2)

您正在使用未对齐的内存,这需要特殊版本的加载和存储功能。您已正确使用_mm_loadu_pd,但是_mm_store_pd不适合使用未对齐的内存,因此应将其更改为_mm_storeu_pd。还可以考虑使用对齐的内存,这将带来更好的性能。