cp-shutdown-bench.c 1.38 KB
Newer Older
1 2 3 4 5
/*
 *  (C) 2009 by Argonne National Laboratory.
 *      See COPYRIGHT in top-level directory.
 */

6 7 8 9 10 11 12 13 14 15 16
/* Arguments: an integer specifying the number of iterations to run of each
 * test phase
 */

#include <stdio.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#include <mpi.h>
17 18 19 20 21 22 23

/* NOTE: we deliberately provide our own function declaration here; there is
 * no header installed with the instrumentation package that defines the
 * benchmarking hooks for us.  This should only be used by special-purpose
 * benchmarking tools.
 */
void darshan_shutdown_bench(int argc, char** argv, int rank, int nprocs);
24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61

int main(int argc, char **argv) 
{
    int nprocs;
    int mynod;
    int ret;
    int iters;

    MPI_Init(&argc, &argv);
    MPI_Comm_rank(MPI_COMM_WORLD, &mynod);
    MPI_Comm_size(MPI_COMM_WORLD, &nprocs);

    /* parse argument for number of iterations */
    if(argc != 2)
    {
        fprintf(stderr, "Usage: %s <number of iterations>\n", argv[0]);
        MPI_Finalize();
        return(-1);
    }

    ret = sscanf(argv[1], "%d", &iters);
    if(ret != 1)
    {
        fprintf(stderr, "Usage: %s <number of iterations>\n", argv[0]);
        MPI_Finalize();
        return(-1);
    }

    if(mynod == 0)
    {
        printf("ignoring iters argument for now.\n");
    }

    darshan_shutdown_bench(argc, argv, mynod, nprocs);

    MPI_Finalize();
    return(0);
}