MPI_Waitall

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

インターフェース

#include <mpi.h>
int MPI_Waitall(int count, MPI_Request* prequests, MPI_Status* pstatuses);
配列prequestsで指定された識別データを持つ非同期通信すべてが完了するまで待機し、 各非同期通信の結果を配列pstatusesに返します。
成功した場合の戻り値はMPI_SUCCESSです。失敗した場合はそれ以外のエラー値が返されます。
引数内容
count対象とする非同期通信の数です。
prequests対象とする非同期通信の識別データです。MPI_Requestの配列として与えます。
pstatuses非同期通信の結果を返すバッファです。MPI_Status型の配列です。

サンプルプログラム

ランク番号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)
{
    std::vector<int32_t> ichigobuf;
    std::vector<MPI_Request> req;
    std::vector<MPI_Status> status;
    int ichigocount;
    int ret;

    ichigobuf.resize(num_processes);
    req.resize(num_processes);
    status.resize(num_processes);

    ichigobuf[RANK_MASTER] = 10;
    for ( ichigocount = 0; ichigocount < num_processes-1;
          ichigocount++ ) {
        ret = MPI_Irecv(
            &ichigobuf[ichigocount+1], 1, MPI_INT32_T,
            ichigocount+1, TAG_ICHIGO, MPI_COMM_WORLD,
            &req[ichigocount]);
        if (ret != MPI_SUCCESS) {
            // error
            std::cout << "ichigosample: Irecv error" << std::endl;
            MPI_Abort(MPI_COMM_WORLD, ABORT_ICHIGO);
            return;
        }
    }

    ret = MPI_Waitall(num_processes-1, &req[0], &status[0]);
    if (ret != MPI_SUCCESS) {
        // error
        std::cout << "ichigosample: Waitall error" << std::endl;
        MPI_Abort(MPI_COMM_WORLD, ABORT_ICHIGO);
        return;
    }

    for ( ichigocount = 0; ichigocount < num_processes;
          ichigocount++ ) {
        std::cout << "ichigosample: data "
            << ichigobuf[ichigocount] << std::endl;
    }
}

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

    ret = MPI_Send(&ichigodata, 1, MPI_INT32_T,
        RANK_MASTER, TAG_ICHIGO, MPI_COMM_WORLD);
    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: data 10
ichigosample: data 16
ichigosample: data 17
ichigosample: data 18

関連ページ

  • MPIの解説 目次