mirror/netcdf-c
2
0
Fork 0
mirror of https://github.com/Unidata/netcdf-c.git synced 2025-01-30 16:10:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

test development

Browse Source
This commit is contained in:
Ed Hartnett 2018-05-14 10:06:35 -06:00
parent 868df56d8f
commit 984f112580
1 changed files with 25 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
nc_test4/tst_parallel5.c
View File

@ -12,7 +12,9 @@
#include <mpi.h> #include <mpi.h>
#define FILE "tst_parallel5.nc" #define FILE "tst_parallel5.nc"
#define NDIMS 3 #define VAR_NAME "Odysseus"
#define DIM_NAME "height"
#define NDIMS1 1
#define DIMSIZE 4 #define DIMSIZE 4
#define NUM_PROC 4 #define NUM_PROC 4
#define NUM_SLABS 10 #define NUM_SLABS 10
@ -23,12 +25,10 @@ main(int argc, char **argv)
int mpi_size, mpi_rank; int mpi_size, mpi_rank;
MPI_Comm comm = MPI_COMM_WORLD; MPI_Comm comm = MPI_COMM_WORLD;
MPI_Info info = MPI_INFO_NULL; MPI_Info info = MPI_INFO_NULL;
int ncid, v1id, dimids[NDIMS]; int ncid, v1id, dimid;
size_t start[NDIMS], count[NDIMS]; size_t start[NDIMS1] = {0}, count[NDIMS1] = {0};
int data = TEST_VAL_42; int data = TEST_VAL_42;
int i, res; int data_in = TEST_VAL_42 + 1;
int slab_data[DIMSIZE * DIMSIZE / 4]; /* one slab */
char file_name[NC_MAX_NAME + 1];
/* Initialize MPI. */ /* Initialize MPI. */
MPI_Init(&argc,&argv); MPI_Init(&argc,&argv);
@ -46,35 +46,35 @@ main(int argc, char **argv)
/* Create a parallel netcdf-4 file. */ /* Create a parallel netcdf-4 file. */
/*nc_set_log_level(3);*/ /*nc_set_log_level(3);*/
sprintf(file_name, "%s/%s", TEMP_LARGE, FILE); if (nc_create_par(FILE, NC_NETCDF4|NC_MPIIO, comm, info, &ncid)) ERR;
if (nc_create_par(file_name, NC_NETCDF4|NC_MPIIO, comm, info, &ncid)) ERR;
/* Create a dimension. */ /* Create a dimension. */
if (nc_def_dim(ncid, "d1", DIMSIZE, dimids)) ERR; if (nc_def_dim(ncid, DIM_NAME, DIMSIZE, &dimid)) ERR;
if (nc_def_dim(ncid, "d2", DIMSIZE, &dimids[1])) ERR;
if (nc_def_dim(ncid, "d3", NUM_SLABS, &dimids[2])) ERR;
/* Create one var. */ /* Create one var. */
if (nc_def_var(ncid, "v1", NC_INT, NDIMS, dimids, &v1id)) ERR; if (nc_def_var(ncid, VAR_NAME, NC_INT, NDIMS1, &dimid, &v1id)) ERR;
/* Write metadata to file. */ /* Write metadata to file. */
if (nc_enddef(ncid)) ERR; if (nc_enddef(ncid)) ERR;
/* /\* Set up slab for this process. *\/ */ /* Set up slab for this process. */
/* start[0] = mpi_rank * DIMSIZE/mpi_size; */ if (!mpi_rank)
/* start[1] = 0; */ count[0] = 1;
/* count[0] = DIMSIZE/mpi_size; */
/* count[1] = DIMSIZE; */
/* count[2] = 1; */
/* /\*printf("mpi_rank=%d start[0]=%d start[1]=%d count[0]=%d count[1]=%d\n", */
/* mpi_rank, start[0], start[1], count[0], count[1]);*\/ */
/* if (nc_var_par_access(ncid, v1id, NC_COLLECTIVE)) ERR; */ if (nc_var_par_access(ncid, v1id, NC_COLLECTIVE)) ERR;
/* /\* if (nc_var_par_access(ncid, v1id, NC_INDEPENDENT)) ERR;*\/ */ /* if (nc_var_par_access(ncid, v1id, NC_INDEPENDENT)) ERR;*/
/* /\* Write slabs of phoney data. *\/ */ /* Write phoney data. */
/* for (start[2] = 0; start[2] < NUM_SLABS; start[2]++) */ if (nc_put_vara_int(ncid, v1id, start, count, &data)) ERR;
/* if (nc_put_vara_int(ncid, v1id, start, count, slab_data)) ERR; */
if (nc_sync(ncid)) ERR;
/* Read phoney data. */
if (nc_get_vara_int(ncid, v1id, start, count, &data_in)) ERR;
/* Task 0 has TEST_VAL_42, the others have data_in remaining, as
* initialized, at TEST_VAL_42 + 1. */
if (data_in != (mpi_rank ? (TEST_VAL_42 + 1) : TEST_VAL_42)) ERR;
/* Close the netcdf file. */ /* Close the netcdf file. */
if (nc_close(ncid)) ERR; if (nc_close(ncid)) ERR;