使用MPI Scatter求二维数组唯一元素时C6001警告的原因及修复

警告情况 #

Severity    Code    Description Project File    Line    Suppression State   Detail Description
Warning C6001   使用未初始化的内存'*unique_counts'.    ConsoleApplication15    C:\Users\netd3en\source\repos\ConsoleApplication15\ConsoleApplication15\ConsoleApplication15.cpp    29  

问题分析 #

C6001警告源于两点：

1. 非0进程分配了unique_counts数组但从未使用，编译器判定这是未初始化内存的无效占用；
2. 原代码存在逻辑错误：0进程收集完各进程的唯一元素数量后，试图用本地的local_arr拼接全局唯一元素，但local_arr仅存储当前进程的分片数据，其他进程的唯一元素并未传递过来，不仅触发警告，还会导致结果错误。

修复方案 #

核心修复点 #

• 仅在0进程中分配和使用unique_counts，避免非0进程的无效内存分配；
• 使用MPI_Gatherv替代MPI_Gather，实现变长数据（各进程唯一元素数量可能不同）的收集，完整获取所有进程的唯一元素后再全局去重。

修复后完整代码 #

#include <iostream>
#include <unordered_set>
#include <vector>
#include <mpi.h>

int main(int argc, char* argv[]) {
    int size, rank;
    MPI_Init(&argc, &argv);
    MPI_Comm_size(MPI_COMM_WORLD, &size);
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);

    int arr[3][3] = { {1, 2, 3}, {4, 5, 6}, {7, 8, 9} };
    int* local_arr = new int[3];
    MPI_Scatter(arr, 3, MPI_INT, local_arr, 3, MPI_INT, 0, MPI_COMM_WORLD);

    // 本地计算唯一元素
    std::unordered_set<int> unique_elements;
    for (int i = 0; i < 3; i++) {
        unique_elements.insert(local_arr[i]);
    }
    int local_unique_count = unique_elements.size();
    // 将unordered_set转为数组，方便MPI传递
    std::vector<int> local_unique_vec(unique_elements.begin(), unique_elements.end());

    if (rank == 0) {
        int* unique_counts = new int[size];
        // 收集每个进程的唯一元素数量
        MPI_Gather(&local_unique_count, 1, MPI_INT, unique_counts, 1, MPI_INT, 0, MPI_COMM_WORLD);

        // 计算接收所有元素的位移量和总大小
        int total_unique = 0;
        int* displacements = new int[size];
        displacements[0] = 0;
        total_unique += unique_counts[0];
        for (int i = 1; i < size; i++) {
            displacements[i] = displacements[i-1] + unique_counts[i-1];
            total_unique += unique_counts[i];
        }

        // 收集所有进程的唯一元素
        std::vector<int> all_unique_elements(total_unique);
        MPI_Gatherv(local_unique_vec.data(), local_unique_count, MPI_INT,
                    all_unique_elements.data(), unique_counts, displacements, MPI_INT,
                    0, MPI_COMM_WORLD);

        // 全局去重
        std::unordered_set<int> global_unique(all_unique_elements.begin(), all_unique_elements.end());

        // 输出结果
        std::cout << "Unique elements:";
        for (int num : global_unique) {
            std::cout << " " << num;
        }
        std::cout << std::endl;

        // 释放内存
        delete[] unique_counts;
        delete[] displacements;
    } else {
        // 非0进程发送本地唯一元素数量
        MPI_Gather(&local_unique_count, 1, MPI_INT, nullptr, 0, MPI_INT, 0, MPI_COMM_WORLD);
        // 非0进程发送本地唯一元素
        MPI_Gatherv(local_unique_vec.data(), local_unique_count, MPI_INT,
                    nullptr, nullptr, nullptr, MPI_INT, 0, MPI_COMM_WORLD);
    }

    // 释放本地内存
    delete[] local_arr;
    MPI_Finalize();

    return 0;
}

修复说明 #

1. 消除警告：将unique_counts的分配和使用限制在0进程内，非0进程不再分配该数组，彻底解决未初始化内存的问题；
2. 修正逻辑错误：通过MPI_Gatherv实现变长数据收集，先获取各进程的唯一元素数量，计算位移后收集所有元素，最后在0进程完成全局去重，得到正确结果；
3. 内存优化：用std::vector管理动态数组，减少手动内存管理风险，同时在合适时机释放内存。

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