Blame test/mpi/topo/dims3.c

Packit 0848f5
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
Packit 0848f5
/*
Packit 0848f5
 *  (C) 2003 by Argonne National Laboratory.
Packit 0848f5
 *      See COPYRIGHT in top-level directory.
Packit 0848f5
 */
Packit 0848f5
#include "mpi.h"
Packit 0848f5
#include <stdio.h>
Packit 0848f5
#include "mpitest.h"
Packit 0848f5
Packit 0848f5
static inline void print_err(int *dims, int ndims)
Packit 0848f5
{
Packit 0848f5
    int i;
Packit 0848f5
Packit 0848f5
    printf("[ ");
Packit 0848f5
    for (i = 0; i < ndims; i++)
Packit 0848f5
        printf("%d ", dims[i]);
Packit 0848f5
    printf("] Suboptimal distribution!\n");
Packit 0848f5
}
Packit 0848f5
Packit 0848f5
int main(int argc, char **argv)
Packit 0848f5
{
Packit 0848f5
    int errs = 0;
Packit 0848f5
    int dims[4], ndims, nnodes;
Packit 0848f5
Packit 0848f5
    MTest_Init(&argc, &argv);
Packit 0848f5
Packit 0848f5
    for (ndims = 3; ndims <= 4; ndims++) {
Packit 0848f5
        for (nnodes = 2; nnodes <= 4096; nnodes *= 2) {
Packit 0848f5
            int i;
Packit 0848f5
            for (i = 0; i < ndims; i++)
Packit 0848f5
                dims[i] = 0;
Packit 0848f5
Packit 0848f5
            MPI_Dims_create(nnodes, ndims, dims);
Packit 0848f5
Packit 0848f5
            /* Checking */
Packit 0848f5
            for (i = 0; i < ndims - 1; i++)
Packit 0848f5
                if (dims[i] / 2 > dims[i + 1]) {
Packit 0848f5
                    print_err(dims, ndims);
Packit 0848f5
                    ++errs;
Packit 0848f5
                    break;
Packit 0848f5
                }
Packit 0848f5
        }
Packit 0848f5
    }
Packit 0848f5
Packit 0848f5
    MTest_Finalize(errs);
Packit 0848f5
    MPI_Finalize();
Packit 0848f5
Packit 0848f5
    return 0;
Packit 0848f5
}