c++ - 在矩阵乘法 C++ SIMD 编程中检测到 Stack Smashing

Question

我刚刚使用 C++ 学习了 simd 编程，我很容易安静地执行加法和减法，但我发现矩阵乘法有问题。

当我编译它时使用:gcc -o 自动 vector auto-vector.cpp -lstdc++

它已编译，但是当我尝试运行它时，它说:耗时:3e-06 秒* 检测到堆栈粉碎 *:已终止

它说检测到堆栈粉碎但它也测量耗时。

我的代码编译了吗？

//gcc -o auto-vector auto-vector.cpp -lstdc++
#include "xmmintrin.h"
#include <chrono> // for high_resolution_clock
#include <iostream>

int main()
{
  float A[4][4] = {{16, 2, 77, 40}, {16, 2, 77, 40}, {16, 2, 77, 40}, {16, 2, 77, 40}};
  float B[4][4] = {{16, 2, 77, 40}, {16, 2, 77, 40}, {16, 2, 77, 40}, {16, 2, 77, 40}};
  float C[4][4] = {};

  __m128 a_vec, b_vec, c_vec;
  int N = 4;

  // Record start time
  auto start = std::chrono::high_resolution_clock::now();

  for (int i = 0; i < N; i++)
  {
    for (int j = 0; j < N; j++)
    {
      c_vec = _mm_set1_ps(0);

      for (int k = 0; k < N; k++)
      {
        a_vec = _mm_set1_ps(A[i][k]);
        b_vec = _mm_loadu_ps(&B[k][j]);

        c_vec = _mm_add_ps(_mm_mul_ps(a_vec, b_vec), c_vec);
      }

      _mm_storeu_ps(&C[i][j], c_vec);
    }
  }

  // Record end time
  auto finish = std::chrono::high_resolution_clock::now();

  std::chrono::duration<double> elapsed = finish - start;

  std::cout << "Elapsed time: " << elapsed.count() << " s\n";

  return 0;
}

Answer 1

这意味着你存储了一个堆栈数组的末尾，你很幸运它在你的堆栈框架的顶部所以-fstack-protector-strong可以为您捕获错误，而不仅仅是踩到其他本地人。

例如16 字节存储到 C[3][3]接触 C[3][3 .. 7] .

您正在对 j 进行自动矢量化但是你忘了增加 j由 4 而不是 1，并使用 j < N - 3作为你的循环界限。

---

此外，如果您想要有意义的计时结果，请确保使用 gcc -O3 进行编译.或者最好 gcc -O3 -march=native -ffast-math -flto和 -fprofile-generate/-fprofile-use .

此外，您正在测试手动矢量化，但您的评论说“自动矢量化”。

---

编码风格:声明你的__m128 vars 当你第一次使用它们时。喜欢
__m128 c_vec = _mm_setzero_ps();