Blame test/mpi/pt2pt/many_isend.c

Packit 0848f5
/* -*- Mode: C; c-basic-offset:4 ; -*- */
Packit 0848f5
/*
Packit 0848f5
 * (C) 2015 by Argonne National Laboratory.
Packit 0848f5
 *     See COPYRIGHT in top-level directory.
Packit 0848f5
 */
Packit 0848f5
#include <stdio.h>
Packit 0848f5
#include <stdlib.h>
Packit 0848f5
#include <string.h>
Packit 0848f5
#include <mpi.h>
Packit 0848f5
Packit 0848f5
#define ITER 5
Packit 0848f5
#define BUF_COUNT 4096
Packit 0848f5
Packit 0848f5
int rank, nprocs;
Packit 0848f5
char recvbuf[BUF_COUNT], sendbuf[BUF_COUNT];
Packit 0848f5
Packit 0848f5
int main(int argc, char *argv[])
Packit 0848f5
{
Packit 0848f5
    int x, i;
Packit 0848f5
    MPI_Status *sendstats = NULL;
Packit 0848f5
    MPI_Request *sendreqs = NULL;
Packit 0848f5
Packit 0848f5
    MPI_Init(&argc, &argv);
Packit 0848f5
Packit 0848f5
    MPI_Comm_rank(MPI_COMM_WORLD, &rank;;
Packit 0848f5
    MPI_Comm_size(MPI_COMM_WORLD, &nprocs);
Packit 0848f5
Packit 0848f5
    sendreqs = (MPI_Request *) malloc(nprocs * sizeof(MPI_Request));
Packit 0848f5
    sendstats = (MPI_Status *) malloc(nprocs * sizeof(MPI_Status));
Packit 0848f5
Packit 0848f5
    for (x = 0; x < ITER; x++) {
Packit 0848f5
        MPI_Barrier(MPI_COMM_WORLD);
Packit 0848f5
Packit 0848f5
        /* all to all send */
Packit 0848f5
        for (i = 0; i < nprocs; i++) {
Packit 0848f5
            MPI_Isend(sendbuf, BUF_COUNT, MPI_CHAR, i, 0, MPI_COMM_WORLD, &sendreqs[i]);
Packit 0848f5
        }
Packit 0848f5
Packit 0848f5
        /* receive one by one */
Packit 0848f5
        for (i = 0; i < nprocs; i++) {
Packit 0848f5
            MPI_Recv(recvbuf, BUF_COUNT, MPI_CHAR, MPI_ANY_SOURCE, MPI_ANY_TAG, MPI_COMM_WORLD,
Packit 0848f5
                     MPI_STATUS_IGNORE);
Packit 0848f5
        }
Packit 0848f5
Packit 0848f5
        /* ensure all send request completed */
Packit 0848f5
        MPI_Waitall(nprocs, sendreqs, sendstats);
Packit 0848f5
    }
Packit 0848f5
Packit 0848f5
    MPI_Barrier(MPI_COMM_WORLD);
Packit 0848f5
    if (rank == 0)
Packit 0848f5
        printf(" No Errors\n");
Packit 0848f5
Packit 0848f5
    free(sendreqs);
Packit 0848f5
    free(sendstats);
Packit 0848f5
Packit 0848f5
    MPI_Finalize();
Packit 0848f5
Packit 0848f5
    return 0;
Packit 0848f5
}