MPI_Probe-Dynamically allocate memory

最新推荐文章于 2024-11-12 13:00:28 发布

原创最新推荐文章于 2024-11-12 13:00:28 发布 · 992 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#MPI并行编程 #动态分配内存 #MPI-Probe

在MPI并行编程中，当接收方不确定消息大小时，可以使用MPI_Probe先探测消息尺寸，然后根据探测到的尺寸动态分配内存，再进行接收。通过这种方法，可以避免预定义过大内存空间。例如，进程1向其他进程发送不同数量的数据，其他进程通过MPI_Probe获取实际数据量，然后接收。输出显示了各个进程接收的数据数量。

某些情况下，接收者不知道消息尺寸，此时当然可以用一个足够大的空间去接收，但我们可以用MPI_Probe去首先探测消息尺寸，然后再分配相应空间，然后再正式接收消息，可以认为MPI_Probe和MPI_Recv很类似，只是不真正执行接收操作

MPI_Probe(
int source, //source rank, or MPI_ANY_SOURCE (integer)
int tag, //tag value or MPI_ANY_TAG (integer)
MPI_Comm comm, //communicator (handle)
MPI_Status *status)

例子，设进程1随机决定发送给其它进程的数据个数，其他进程需要先探测真实尺寸，然后接收：

#include <mpi.h>
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <vector>
int main(int argc, char** argv) {
  MPI_Init(NULL, NULL);

  int nProcs,Rank;
  int i;
  MPI_Status status;
  MPI_Comm_size(MPI_COMM_WORLD, &nProcs);
  MPI_Comm_rank(MPI_COMM_WORLD, &Rank);

  int NumToBeSent;

  if (Rank == 0) {

    const int MaxNum = 6000;
    std::vector<double> Data;
    srand(time(NULL));

        for(i=0;i<MaxNum;i++)Data.resize(MaxNum);
        for(i=0;i<MaxNum;i++)Data[i] = i;

    for(i=1;i<nProcs;i++)
    {   
        NumToBeSent = rand() % MaxNum;
        MPI_Send(&Data[0],NumToBeSent, MPI_DOUBLE, i, 0, MPI_COMM_WORLD);
        printf("0 sent %d numbers to %d\n", NumToBeSent,i);
    }
  }
  else {

     MPI_Probe(0, 0, MPI_COMM_WORLD, &status);
    //When MPI_Probe returns, the size information is included, the MPI_Get_count function can extract size information
     MPI_Get_count(&status, MPI_DOUBLE, &NumToBeSent);
      // Now we can allocate suitable memory for the receive buffer
    double* recvbuf = (double*)malloc(sizeof(double) * NumToBeSent);
    // Now the message can be received properly
    MPI_Recv(recvbuf, NumToBeSent, MPI_DOUBLE, 0, 0, MPI_COMM_WORLD,&status);
    printf("%d Received %d Data From 0.\n", Rank, NumToBeSent);
    free(recvbuf);
  }
  system("pause");
  MPI_Finalize();
}