VS2022下OpenMP并行遍历STL unordered_map问题：稀疏矩阵自乘失效

问题描述 #

尝试并行化稀疏方阵的自乘运算 R = A * A，其中R和A均为稀疏矩阵，采用二维std::unordered_map存储（A[i][j]对应矩阵的(i,j)元素）。但在smatrix类的smul成员函数中，OpenMP的#pragma omp parallel for未生效：

std::pair<int, int> shape;
std::unordered_map<int, std::unordered_map<int, mpreal>> nzel;

smatrix smul(void)
{
    smatrix result(shape);
    #pragma omp parallel for
    for (int row = 0; row < shape.first; row++)
    {
        auto& c = result.nzel[row];
        for (const auto& [k1, v1] : nzel[row])
            for (const auto& [k2, v2] : nzel[k1])
                c[k2] += v1 * v2;
    }
    return result;
}

问题排查与解决方案 #

1. 编译未启用OpenMP支持（最常见原因） #

OpenMP需要编译器显式启用才能生效，不同编译器的启用方式：

• GCC/Clang：编译时添加-fopenmp参数，例如：g++ -fopenmp your_code.cpp -o your_program
• MSVC：编译时添加/openmp参数，或在Visual Studio项目属性中开启“OpenMP支持”选项

如果未添加该选项，编译器会直接忽略#pragma omp指令，自然无法触发并行执行。

2. 循环迭代存在大量空执行 #

原代码循环遍历所有行（从0到shape.first-1），但稀疏矩阵中大部分行没有非零元素，导致大量空循环迭代，既浪费资源，也可能让你误以为并行未生效。

优化方式：只遍历存在非零元素的行，避免空迭代：

smatrix smul(void)
{
    smatrix result(shape);
    // 先收集所有存在非零元素的行
    std::vector<int> valid_rows;
    valid_rows.reserve(nzel.size());
    for (const auto& [row, _] : nzel) {
        valid_rows.push_back(row);
    }
    #pragma omp parallel for
    for (size_t idx = 0; idx < valid_rows.size(); idx++)
    {
        int row = valid_rows[idx];
        auto& c = result.nzel[row];
        // 使用at()确保访问存在的行，避免默认构造空map
        for (const auto& [k1, v1] : nzel.at(row))
            for (const auto& [k2, v2] : nzel.at(k1))
                c[k2] += v1 * v2;
    }
    return result;
}

3. C++特性与OpenMP版本兼容性问题 #

原代码使用了C17的结构化绑定（auto& [k1, v1]），需要确保编译器支持C17，且OpenMP版本（如OpenMP 4.5+）兼容该特性。如果编译器版本过旧，可能导致OpenMP指令无法正确解析。

解决方式：升级编译器到支持C++17和对应OpenMP版本的版本（例如GCC 7+、MSVC 2017+）。

4. 验证并行是否生效 #

可以在并行循环中添加线程编号输出，确认多线程是否启动：

#pragma omp parallel for
for (size_t idx = 0; idx < valid_rows.size(); idx++)
{
    int row = valid_rows[idx];
    // 打印线程编号，仅用于调试
    printf("Thread %d processing row %d\n", omp_get_thread_num(), row);
    // ... 原有计算代码 ...
}

如果输出中出现多个不同的线程编号，说明并行已生效。

内容的提问来源于stack exchange，提问作者M. R. Omar