allocmem.c

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

/* -*- Mode: C; c-basic-offset:4 ; -*- */
/*
 *
 *  (C) 2003 by Argonne National Laboratory.
 *      See COPYRIGHT in top-level directory.
 */
#include "mpi.h"
#include <stdio.h>
#include "mpitest.h"

static char MTEST_Descrip[] = "Simple test that alloc_mem and free_mem work together";

int main( int argc, char *argv[] )
{
    int errs = 0, err;
    int j, count;
    char *ap;

    MTest_Init( &argc, &argv );

    MPI_Errhandler_set( MPI_COMM_WORLD, MPI_ERRORS_RETURN );
    for (count=1; count < 128000; count *= 2) {
	
	err = MPI_Alloc_mem( count, MPI_INFO_NULL, &ap );
	if (err) {
	    int errclass;
	    /* An error of  MPI_ERR_NO_MEM is allowed */
	    MPI_Error_class( err, &errclass );
	    if (errclass != MPI_ERR_NO_MEM) {
		errs++;
		MTestPrintError( err );
	    }
	    
	}
	else {
	    /* Access all of this memory */
	    for (j=0; j<count; j++) {
		ap[j] = (char)(j & 0x7f);
	    }
	    MPI_Free_mem( ap );
	}
    }

    MTest_Finalize( errs );
    MPI_Finalize();
    return 0;
}