C++ 中的 OpenMP 分段错误

Question

我有一个非常简单的函数，可以计算 N by N 2D 矩阵（由指针 arr 表示）的内部条目数低于 a特定阈值，并更新通过引用传递的计数器 below_threshold：

void count(float *arr, const int N, const float threshold, int &below_threshold) {
    below_threshold = 0;  // make sure it is reset
    bool comparison;
    float temp;
    
    #pragma omp parallel for shared(arr, N, threshold) private(temp, comparison) reduction(+:below_threshold)
    for (int i = 1; i < N-1; i++)  // count only the inner N-2 rows
    {
        for (int j = 1; j < N-1; j++)  // count only the inner N-2 columns
        {
            temp = *(arr + i*N + j);
            comparison = (temp < threshold);
            below_threshold += comparison;
        }
    }
}

当我不使用 OpenMP 时，它运行良好（因此，分配和初始化已正确完成）。

当我使用 N 小于 40000 左右的 OpenMP 时，它运行良好。

然而，一旦我开始在 OpenMP 中使用更大的 N ，它就会不断给我带来分段错误（我目前正在使用 N = 50000 进行测试，并希望最终能够将其恢复）至~100000）。

在软件层面上这有什么问题吗？

---

PS 分配是动态完成的（ float *arr = new float [N*N] ），这里是用于随机初始化整个矩阵的代码，这对于 OpenMP 来说没有任何问题Large N：

void initialize(float *arr, const int N)
{
    #pragma omp parallel for
    for (int i = 0; i < N; i++)
    {
        for (int j = 0; j < N; j++)
        {
            *(arr + i*N + j) = static_cast <float> (rand()) / static_cast <float> (RAND_MAX);
        }
    }

}

---

更新：

我尝试将 i、j 和 N 更改为 long long int，它仍然没有修复我的分段错误。如果这是问题所在，为什么它在没有 OpenMP 的情况下仍然可以工作？只有当我添加 #pragma omp ... 时，它才会失败。

Answer 1

我认为，这是因为，你的值（50000*50000 = 2500000000）在c++中达到了INT_MAX（2147483647）以上。因此，数组访问行为将是未定义的。

因此，您应该使用 UINT_MAX 或一些适合您的用例的其他类型。