diff options
Diffstat (limited to 'testpar/t_mdset.c')
-rw-r--r-- | testpar/t_mdset.c | 58 |
1 files changed, 58 insertions, 0 deletions
diff --git a/testpar/t_mdset.c b/testpar/t_mdset.c new file mode 100644 index 0000000..77354d6 --- /dev/null +++ b/testpar/t_mdset.c @@ -0,0 +1,58 @@ +#include <testphdf5.h> + +#define DIM 2 +#define SIZE 10 +#define NUMITEMS 500 /* 988 */ + +void multiple_dset_write(char *filename) +{ + int i, j, nprocs, rank; + hid_t iof, plist, dataset, memspace, filespace; + hssize_t chunk_origin [DIM]; + hsize_t chunk_dims [DIM], file_dims [DIM]; + double outme [SIZE][SIZE]; + + MPI_Comm_rank (MPI_COMM_WORLD, &rank); + MPI_Comm_size (MPI_COMM_WORLD, &nprocs); + + chunk_origin [0] = 0; + chunk_origin [1] = rank * (SIZE / nprocs); + chunk_dims [0] = SIZE; + chunk_dims [1] = SIZE / nprocs; + + for (i = 0; i < DIM; i++) + file_dims [i] = SIZE; + + for (i = 0; i < SIZE; i++) + for (j = 0; j < SIZE; j++) + outme [i][j] = rank; + + plist = H5Pcreate (H5P_FILE_ACCESS); + H5Pset_fapl_mpio(plist, MPI_COMM_WORLD, MPI_INFO_NULL); + iof = H5Fcreate (filename, H5F_ACC_TRUNC, H5P_DEFAULT, plist); + H5Pclose (plist); + + memspace = H5Screate_simple (DIM, chunk_dims, NULL); + filespace = H5Screate_simple (DIM, file_dims, NULL); + H5Sselect_hyperslab (filespace, H5S_SELECT_SET, chunk_origin, NULL, chunk_dims +, NULL); + + for (i = 0; i < NUMITEMS; i++) { + char dname [100]; + + sprintf (dname, "dataset %d", i); + dataset = H5Dcreate (iof, dname, H5T_NATIVE_FLOAT, filespace, H5P_DEFAULT); + if (dataset < 0) { + fprintf (stderr, "proc %d: failed to create dataset %d\n", rank, i); + exit (-1); + } + + H5Dwrite (dataset, H5T_NATIVE_DOUBLE, memspace, filespace, H5P_DEFAULT, outme); + + H5Dclose (dataset); + } + + H5Sclose (filespace); + H5Sclose (memspace); + H5Fclose (iof); +} |