有沒有辦法在不知道每個進程中每個數組的大小的情況下執行 MPI_Gatherv ？

Question

我正在嘗試為 int 數組執行 MPI_Gatherv 而不知道每個進程中每個數組的大小。 有沒有辦法做到這一點？

這是我的代碼片段。

#include "mpi.h"
#include <iostream>

using namespace std;

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

    int *local_arr;
    int *arr;
    int local_n;
    int n;

    if (rank == 0) {

        // do gatherv without knowing the element sizes and displacements

    } else {
        // fill array with random size
        srand(rank);
        local_n = (rand()%10)+1;
        local_arr = new int[local_n];

        for (int i = 0; i < local_n; i++) {
            local_arr[i] = i;
        }

        // do gatherv without knowing the element sizes and displacements
    }
}

Answer 1

你真的不能那樣做。 displacements和element_sizes參數的存在表明了這一點。 如果確實無法知道（無法在根處計算）根將接收到的計數，您將需要從每個節點執行單個int的MPI_Gather並將其用作MPI_Gatherv displacements 。

Answer 2

當您調用Gatherv ，您必須事先知道大小和位移，因為您只需要在目標等級上分配足夠的空間。

這就是為什么你會經常看到一個Gather調用來獲得每個等級的大小，然后是一個減少以獲得位移，然后是Gatherv調用。 如果在不同等級上有幾個固定大小的后者，則將在預處理步驟中調用Gather 。