删除 MPI_Bcast()

Question

所以我有一些代码，我使用 MPI_Bcast 将信息从根节点发送到所有节点，但我想让我的 P0 将数组块发送到各个进程。

如何使用 MPI_Send 和 MPI_Receive 执行此操作？

我以前从未使用过它们，我不知道我是否需要循环我的 MPI_Receive 以有效地发送所有内容或什么。

我在需要替换我的 MPI_Bcast() 的代码中放置了巨大的大写锁定注释，对于代码的瀑布提前抱歉。

代码：

#include "mpi.h"
#include <stdio.h>
#include <math.h>

#define MAXSIZE 10000000

int add(int *A, int low, int high)
{
  int res = 0, i;

  for(i=low; i<=high; i++)
    res += A[i];

  return(res);
}

int main(argc,argv)
int argc;
char *argv[];
{
    int myid, numprocs, x;
    int data[MAXSIZE];
    int i, low, high, myres, res;
    double elapsed_time;

    MPI_Init(&argc,&argv);
    MPI_Comm_size(MPI_COMM_WORLD,&numprocs);
    MPI_Comm_rank(MPI_COMM_WORLD,&myid);

        if (myid == 0)
          {
                for(i=0; i<MAXSIZE; i++)
                  data[i]=1;
          }

/* star the timer */
        elapsed_time = -MPI_Wtime();

//THIS IS WHERE I GET CONFUSED ABOUT MPI_SEND AND MPI_RECIEVE!!!
        MPI_Bcast(data, MAXSIZE, MPI_INT, 0, MPI_COMM_WORLD);

            x = MAXSIZE/numprocs;
            low = myid * x;
            high = low + x - 1;
        if (myid == numprocs - 1) 
        high = MAXSIZE-1;

            myres = add(data, low, high);
            printf("I got %d from %d\n", myres, myid);

        MPI_Reduce(&myres, &res, 1, MPI_INT, MPI_SUM, 0, MPI_COMM_WORLD);
/* stop the timer*/
        elapsed_time += MPI_Wtime();

        if (myid == 0)
            printf("The sum is %d, time taken = %f.\n", res,elapsed_time);

    MPI_Barrier(MPI_COMM_WORLD);

            printf("The sum is %d at process %d.\n", res,myid);

    MPI_Finalize();
    return 0;
}

Answer 1

你需要MPI_Scatter。这里有一个很好的介绍：http://mpitutorial.com/tutorials/mpi-scatter-gather-and-allgather/

我认为在您的代码中它可能如下所示：

elements_per_proc = MAXSIZE/numprocs;

// Create a buffer that will hold a chunk of the global array
int *data_chunk = malloc(sizeof(int) * elements_per_proc);

MPI_Scatter(data, elements_per_proc, MPI_INT, data_chunk,
            elements_per_proc, MPI_INT, 0, MPI_COMM_WORLD);

Answer 2

如果你真的想使用 MPI_Send 和 MPI_Recv，那么你可以这样使用：

int x = MAXSIZE / numprocs;
int *procData = new int[x];

if (rank == 0) {
    for (int i = 1; i < num; i++) {
        MPI_Send(data + i*x, x, MPI_INT, i, 0, MPI_COMM_WORLD);
    }
} else {
    MPI_Recv(procData, x, MPI_INT, 0, 0, MPI_COMM_WORLD, &status);
}

删除 MPI_Bcast()

Removing MPI_Bcast()

mpi