MPI——生产者和消费者

Question

如何简单地制作一个生产者和消费者应用程序。生产者制作物品，将其发送给消费者，而消费者等待直到他拥有该物品。他使用它，项目不见了，他向生产者发送创建新项目的请求。再来一遍。

我有一些 MPI_send 和 MPI_recv 组合的模式，但它只运行一次。生产者生产一件商品，消费者消费一件商品，应用程序陷入僵局。我应该使用非阻塞接收和发送吗？

int count=10;
if(myrank==0){     //server
 for(i=0;i<10;i++){
    MPI_Recv(&a,1,MPI_INT,1,99,MPI_COMM_WORLD,&status);
    if (a==0){
      a=produced(); //here it returns 1
      MPI_Send(&a,1,MPI_INT,1,99,MPI_COMM_WORLD);
    }
  }

}
else{//client
 for(i=0;i<10;i++){
    if(a==0){
       a=0;
       MPI_Send(&a,1,MPI_INT,0,99,MPI_COMM_WORLD);
    }else{
       MPI_Recv(&a,1,MPI_INT,0,99,MPI_COMM_WORLD,&status);
       a=consumed();
       n++;
    }
    if(n==count){
       MPI_Finalize();

    }

 } 

}

编辑：

int produced(){
   sleep(1);
   printf("Produced item\n");
   return 1;

}

int consumed(){
   sleep(1);
   printf("Consumed item\n");
   return 0;

}

Answer 1

你不应该为此需要非阻塞 io。问题是没有任何东西改变客户端状态，所以它永远不会收到任何东西。试试：

else { //client
  a=0;
  for (i=0;i<10;i++) {
    MPI_Send(&a,1,MPI_INT,0,99,MPI_COMM_WORLD);
    do {
      MPI_Recv(&a,1,MPI_INT,0,99,MPI_COMM_WORLD,&status);
    } while (a != 1);
    a=consumed();
  }

  MPI_Finalize();
}

请注意，上面实际上并不需要 while 循环，但是当您检查了服务器接收到的数据时，我假设您想要检查接收到的客户端数据。

编辑：更改为反映consumed() 和produced() 的源代码