dimsbalanced.c

来自「fortran并行计算包」· C语言 代码 · 共 60 行

/* This is a simple test program to see how well balanced the output from
   MPI_Dims_create is.  This test can be run with a single process
   and does not need mpiexec.

   There is no standard result.  Instead, the output should be examined to 
   see if the results are acceptable.  For example, make sure that Dims_create
   of 2*2*2 in three dimensions returns 2,2,2 (and not 1,1,8).
 */

#include "mpi.h"
#include <stdio.h>

static int verbose = 1;

int main( int argc, char *argv[] )
{
    int i, j, dims[3], size;

    MPI_Init( &argc, &argv );
    
    for (i=2; i<12; i++) {
	for (j=0; j<2; j++) dims[j] = 0;
	MPI_Dims_create( i * i, 2, dims );
	if (verbose) printf( "Dims_create(%d,2) = (%d,%d)\n", 
			     i*i, dims[0], dims[1]);
	for (j=0; j<3; j++) dims[j] = 0;
	MPI_Dims_create( i * i * i, 3, dims );
	if (verbose) printf( "Dims_create(%d,3) = (%d,%d,%d)\n", 
			     i*i*i, dims[0], dims[1], dims[2]);
    }
    size = 2*5*7*11;
    for (j=0; j<3; j++) dims[j] = 0;
    MPI_Dims_create( size, 2, dims );
    if (verbose) printf( "Dims_create(%d,2) = (%d,%d)\n", 
			 size, dims[0], dims[1]);
    for (j=0; j<3; j++) dims[j] = 0;
    MPI_Dims_create( size, 3, dims );
    if (verbose) printf( "Dims_create(%d,3) = (%d,%d,%d)\n", 
			 size, dims[0], dims[1], dims[2]);
    size = 5*5*2*7*7*7;
    for (j=0; j<3; j++) dims[j] = 0;
    MPI_Dims_create( size, 2, dims );
    if (verbose) printf( "Dims_create(%d,2) = (%d,%d)\n", 
			 size, dims[0], dims[1]);
    for (j=0; j<3; j++) dims[j] = 0;
    MPI_Dims_create( size, 3, dims );
    if (verbose) printf( "Dims_create(%d,3) = (%d,%d,%d)\n", 
			 size, dims[0], dims[1], dims[2]);
    
    size = 12;
    for (j=0; j<3; j++) dims[j] = 0;
    MPI_Dims_create( size, 2, dims );
    if (verbose) printf( "Dims_create(%d,2) = (%d,%d)\n", 
			 size, dims[0], dims[1]);

    MPI_Finalize();

    return 0;
}