2

ブロードキャスターが実行時に決定される場合、MPI_Bcast を実行するにはどうすればよいですか? この場合、ルート ノードを指定するにはどうすればよいですか?

配列内の個別の番号を検索しようとしています。ノードが番号を見つけた場合、その場所を他のすべてのノードにブロードキャストする必要があります。ただし、事前にファインダーがわからないため、「ルート」値は次のようになります。

int MPI_Bcast ( void *buffer, int count, MPI_Datatype datatype, 
                int root, MPI_Comm comm );
4

1 に答える 1

2

全員がコレクティブに入る前に「ルート」に同意する必要があるため、事前にある程度のコラボレーションが必要です。これは簡単なアプローチの 1 つです。関連するデータを持っているかどうかを示すフラグを全員が送信し、誰から受信するかについて全員が同意できるようにします。これにより、送信者候補が複数存在する場合や存在しない場合に対処できます。

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

int main(int argc, char **argv) {
    int rank, size, ierr;
    int bcaster;
    int bcasterflag, *allflags;
    int consensus;
    int globaldata, mydata;

    ierr = MPI_Init(&argc, &argv);
    ierr|= MPI_Comm_size(MPI_COMM_WORLD,&size);
    ierr|= MPI_Comm_rank(MPI_COMM_WORLD,&rank);

    if (argc != 2) {
        if (rank == 0) fprintf(stderr,"Usage: %s rank-to-broadcast\n",argv[0]);
        MPI_Abort(MPI_COMM_WORLD,1);
    }

    bcaster = atoi(argv[1]);
    if (bcaster < 0    ) bcaster = 0;
    if (bcaster >= size) bcaster = size-1;

    /* pretend the processes didn't all know the above and had to
     * rely solely on local info to decide the broadcaster
     */

    bcasterflag = 0;     /* not the broadcaster yet */

    mydata = rank*rank;   /* the local data */
    if (mydata == bcaster*bcaster) {
        bcasterflag = 1;
        globaldata = mydata;
    }


    /* collect these local decisions */

    allflags = (int *)malloc(size * sizeof(int));
    ierr = MPI_Allgather(&bcasterflag, 1, MPI_INT,
                         allflags, 1, MPI_INT, MPI_COMM_WORLD);

    consensus = -1;
    for (int i=0; i<size; i++)
        if (allflags[i] != 0) consensus = i;

    if (consensus == -1) {
       if (rank == 0) {
          fprintf(stderr,"Error: no one found to do the broadcast.\n");
       }
    } else {
        ierr = MPI_Bcast(&globaldata, 1, MPI_INT, consensus, MPI_COMM_WORLD);
    }

    printf("%d: Received data %d from %d\n",
            rank, globaldata, consensus);

    return 0;
}
于 2011-04-17T13:44:31.333 に答える