c - MPI_Send 到单变量目的地

Question

我想将 MPI_Send 消息发送到单个但可变的主机。我的意思是，我在编译时不知道要发送消息的主机的级别。

很自然地，我写的内容如下:

MPI_Send(&myIntData, 1, MPI_INT, target_host, MY_MSGTAG, MPI_COMM_WORLD);

其中 target_host 是一个整数，包含目标的排名。

当我执行此操作时，MPI 以超时结束。如果我在发送之前和之后放置一个 printf，则仅显示之前发生的 printf，这意味着我的程序卡在 MPI_Send 语句中。

例如，如果我将 target_host 更改为 1，则显示第二个 printf 并且一切正常。

那么.. mpi 不支持目标排名标识符中的变量吗？

这里有什么问题吗？

Answer 1

MPI 确实允许在排名参数中使用变量。考虑以下示例

#include <stdio.h>
#include <stdlib.h>
#include <mpi.h>
#include <unistd.h>
#include <string.h>

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

    int rank = 0;
    int comm_size = 0;

    MPI_Init(&argc, &argv);

    MPI_Comm_size(MPI_COMM_WORLD, &comm_size);
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);

    printf("Rank %d starting\n", rank);
    unsigned int i;

    if(rank == 0){

        for(i = 1; i < comm_size; i++){

            char myString[] = "STRING";
            int len = strlen(myString);
            printf("Sending to %d\n", i);
            MPI_Send(myString, len, MPI_CHAR, i, 0, MPI_COMM_WORLD);

        }
    }
    else{
        char rbuf[4096];            
        MPI_Recv((void*)rbuf, 4096, MPI_CHAR, MPI_ANY_SOURCE, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
        printf("[%d] Got %s in Recv\n", rank, rbuf);
    }


    MPI_Finalize();

}

使用 mpicc -o my_program example.c 进行编译并确保使用 mpiexec -n 2 my_program 运行程序或大于二的数字 -n参数(即要使用的进程数)。在您的情况下，您需要将主机作为消息发送到排名 0。然后，在等级 0 中，您将需要一个循环来从每个等级接收。类似于下面的内容

if(rank != 0){
    MPI_Send(host, host_msg_len, MPI_CHAR, 0, 0, MPI_COMM_WORLD);
}
else{
    for(i = 1; i < comm_size; i++){
        char rbuf[4096];            
        MPI_Recv((void*)rbuf, 4096, MPI_CHAR, i, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE);

        if(strcmp(rbuf, target_host) == 0){
            target_rank = i;
            break;
        }
    }
}