MPI_Wait

いちごパック > MPIの解説 > MPI_Wait

インターフェース

#include <mpi.h>
int MPI_Wait(MPI_Request* prequest, MPI_Status* pstatus);
prequestを識別データとする非同期通信が完了するまで待機し、 非同期通信の結果をpstatusに返します。
成功した場合の戻り値はMPI_SUCCESSです。失敗した場合はそれ以外のエラー値が返されます。
引数内容
prequestこの非同期通信の識別データです。
pstatus非同期通信の追加情報です。

サンプルプログラム

ランク番号0を持つプロセスにそれ以外のプロセスが(ランク番号+15)を送信し、 ランク番号0を持つプロセスはそれらを受け取って表示するサンプルソースコードを示します。

#include <mpi.h>
#include <stdint.h>
#include <iostream>
#include <vector>

#define RANK_MASTER 0

#define ABORT_INIT 1
#define ABORT_ICHIGO 15

#define TAG_ICHIGO 15

void process_master(int num_processes)
{
    int32_t ichigodata;
    std::vector<int32_t> ichigobuf;
    int ichigocount;
    int source_rank;
    MPI_Status status;
    int ret;
    bool ichigoerror = false;

    ichigobuf.resize(num_processes);
    ichigobuf[RANK_MASTER] = 10;

    for ( ichigocount = 0; ichigocount < num_processes-1;
          ichigocount++ ) {
        ret = MPI_Recv(
            &ichigodata, 1, MPI_INT32_T,
            MPI_ANY_SOURCE, TAG_ICHIGO, MPI_COMM_WORLD, &status);
        if (ret == MPI_SUCCESS) {
            source_rank = status.MPI_SOURCE;
            ichigobuf[source_rank] = ichigodata;
            std::cout << "ichigosample: rank "
                << source_rank << " recv "
                << ichigodata << std::endl;
        } else {
            // error
            ichigoerror = true;
            break;
        }
    }
    if ( !ichigoerror ) {
        for ( ichigocount = 0; ichigocount < num_processes;
              ichigocount++ ) {
            std::cout << "ichigosample: data "
                << ichigobuf[ichigocount] << std::endl;
        }
    } else {
        std::cout << "ichigosample: error" << std::endl;
        MPI_Abort(MPI_COMM_WORLD, ABORT_ICHIGO);
    }
}

void process_others(int num_processes, int process_rank)
{
    MPI_Request req;
    MPI_Status status;
    int32_t ichigodata = process_rank + 15;
    int ret;

    ret = MPI_Isend(&ichigodata, 1, MPI_INT32_T,
        RANK_MASTER, TAG_ICHIGO, MPI_COMM_WORLD, &req);
    if (ret != MPI_SUCCESS) {
        MPI_Abort(MPI_COMM_WORLD, ABORT_ICHIGO);
    }

    // ...

    ret = MPI_Wait(&req, &status);
    if (ret != MPI_SUCCESS) {
        MPI_Abort(MPI_COMM_WORLD, ABORT_ICHIGO);
    }
}

int main( int argc, char** argv )
{
    int num_processes = 1;
    int process_rank = 0;
    int ret;

    ret = MPI_Init(&argc, &argv);
    if (ret == MPI_SUCCESS)
    ret = MPI_Comm_size(MPI_COMM_WORLD, &num_processes);
    if (ret == MPI_SUCCESS)
        ret = MPI_Comm_rank(MPI_COMM_WORLD, &process_rank);
    if (ret != MPI_SUCCESS) {
        // error
        MPI_Abort(MPI_COMM_WORLD, ABORT_INIT);
    }

    if (process_rank == RANK_MASTER) {
        process_master(num_processes);
    } else {
        process_others(num_processes, process_rank);
    }

    MPI_Finalize();
    return 0;
}

4つのプロセスで動作させたときの出力結果の1例は次の通りです。
ichigosample: rank 3 recv 18
ichigosample: rank 2 recv 17
ichigosample: rank 1 recv 16
ichigosample: data 10
ichigosample: data 16
ichigosample: data 17
ichigosample: data 18

関連ページ

  • MPIの解説 目次