Question

这是问题所在。我有一些“小”数组，我不想MPI_Gather成一个大的，但我只想在root（0）线程上分配一个大的。

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

int main(int argc, char **argv) {

    int rank, numprocs;

    MPI_Init(&argc, &argv);
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    MPI_Comm_size(MPI_COMM_WORLD, &numprocs);
    MPI_Status status;

    int N = 5;
    int x[N] = {1,2,3,4,5};

    int big_x[numprocs*N];
    MPI_Gather(x, N, MPI_INT, big_x, N, MPI_INT, 0, MPI_COMM_WORLD);

    if (rank == 0) {
        for (int i=0; i<numprocs*N; ++i)
            std::cout << big_x[i] << std::endl;
    }


    MPI_Finalize();
    return 0;
}

这是正常工作的代码，但正如您所看到的，我已经在每个线程上分配了big_x。如果我想在一个线程中分配它，该怎么办？我会收到范围错误。我应该只使用动态内存吗？

Answer 1

是的，你应该使用动态内存。

int *big_x = NULL;
if (rank == 0) big_x = new int[numprocs*N];
...
if (rank == 0) delete [] big_x;

静态数组的大小应该是常量。所以int big_x[numprocs*N]是一个错误，因为它在numprocs初始化之前在函数start上分配。

另外

int N = 5;
int x[N] = {1,2,3,4,5};

是错误，因为N不是常量。请改用const int N = 5或#define N 5。

在MPI，C ++上分配单个堆栈数组

1 个答案: