summaryrefslogtreecommitdiffstats
path: root/testpar/t_coll_chunk.c
diff options
context:
space:
mode:
authorMuQun Yang <ymuqun@hdfgroup.org>2004-07-20 21:41:44 (GMT)
committerMuQun Yang <ymuqun@hdfgroup.org>2004-07-20 21:41:44 (GMT)
commit3176090a0381967a6c2e4fdd22fd203bb335ed7a (patch)
treea3ba2c1718c9f0bd4d6cc25fa038947104ef004f /testpar/t_coll_chunk.c
parent8933c596946b9ba88fb2ca66a3f46f1f1d209aed (diff)
downloadhdf5-3176090a0381967a6c2e4fdd22fd203bb335ed7a.zip
hdf5-3176090a0381967a6c2e4fdd22fd203bb335ed7a.tar.gz
hdf5-3176090a0381967a6c2e4fdd22fd203bb335ed7a.tar.bz2
[svn-r8907] Purpose:
To add collective chunk IO tests. Description: three tests are added. 1. Only one hyperslab for each process, and this hyperslab is fit in exactly one chunk. 2. non-contiguous hyperslabs in each process, these hyperslabs are fit in one chunk. 3. Single hyperslab for each process, smaller chunk is assigned. Number of chunks for every process is equal. Solution: the dataset size is set to be very small, will enlarge later. Platforms tested: AIX 5.1(copper) Misc. update:
Diffstat (limited to 'testpar/t_coll_chunk.c')
-rw-r--r--testpar/t_coll_chunk.c449
1 files changed, 449 insertions, 0 deletions
diff --git a/testpar/t_coll_chunk.c b/testpar/t_coll_chunk.c
new file mode 100644
index 0000000..0ed4cdc
--- /dev/null
+++ b/testpar/t_coll_chunk.c
@@ -0,0 +1,449 @@
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
+ * Copyright by the Board of Trustees of the University of Illinois. *
+ * All rights reserved. *
+ * *
+ * This file is part of HDF5. The full HDF5 copyright notice, including *
+ * terms governing use, modification, and redistribution, is contained in *
+ * the files COPYING and Copyright.html. COPYING can be found at the root *
+ * of the source code distribution tree; Copyright.html can be found at the *
+ * root level of an installed copy of the electronic HDF5 document set and *
+ * is linked from the top-level documents page. It can also be found at *
+ * http://hdf.ncsa.uiuc.edu/HDF5/doc/Copyright.html. If you do not have *
+ * access to either file, you may request a copy from hdfhelp@ncsa.uiuc.edu. *
+ * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+#include "testphdf5.h"
+
+/*#define SPACE_DIM1 256
+#define SPACE_DIM2 256
+#define BYROW_CONT 1
+#define BYROW_DISCONT 2
+#define DSET_COLLECTIVE_CHUNK_NAME "coll_chunk_name"
+*/
+
+
+void coll_chunktest(char* filename,int chunk_factor,int select_factor);
+/*-------------------------------------------------------------------------
+ * Function: coll_chunk1
+ *
+ * Purpose: Test the special case of the collective chunk io
+ *
+ * Return: Success: 0
+ *
+ * Failure: -1
+ *
+ * Programmer: Unknown
+ * July 12th, 2004
+ *
+ * Modifications:
+ *
+ *-------------------------------------------------------------------------
+ */
+void
+coll_chunk1(){
+
+ char *filename;
+ filename = (char *) GetTestParameters();
+ coll_chunktest(filename,1,BYROW_CONT);
+
+}
+
+void
+coll_chunk2(){
+
+ char *filename;
+ filename = (char *) GetTestParameters();
+ coll_chunktest(filename,1,BYROW_DISCONT);
+
+}
+
+
+void
+coll_chunk3(){
+
+ char *filename;
+ filename = (char *) GetTestParameters();
+ coll_chunktest(filename,4,BYROW_CONT);
+
+}
+
+void
+coll_chunktest(char* filename,int chunk_factor,int select_factor) {
+
+
+ hid_t file,dataset, file_dataspace;
+ hid_t acc_plist,xfer_plist,crp_plist;
+ hsize_t dims[RANK], chunk_dims[RANK];
+ int i,j;
+ int* data_array1 = NULL;
+ int* data_origin1 = NULL;
+ herr_t status;
+ hssize_t start[RANK];
+ hsize_t count[RANK],stride[RANK],block[RANK];
+
+ /* char * filename;*/
+
+ int mpi_size,mpi_rank;
+
+ MPI_Comm comm = MPI_COMM_WORLD;
+ MPI_Info info = MPI_INFO_NULL;
+
+ /* filename = (char *) GetTestParameters();*/
+
+ /* set up MPI parameters */
+ MPI_Comm_size(MPI_COMM_WORLD,&mpi_size);
+ MPI_Comm_rank(MPI_COMM_WORLD,&mpi_rank);
+
+ /* Create the data space */
+ acc_plist = H5Pcreate(H5P_FILE_ACCESS);
+ VRFY((acc_plist >= 0),"");
+
+
+ status = H5Pset_fapl_mpio(acc_plist,comm,info);
+ VRFY((acc_plist >= 0),"MPIO creation property list succeeded");
+
+ file = H5Fcreate(filename,H5F_ACC_TRUNC,H5P_DEFAULT,acc_plist);
+ VRFY((file >= 0),"H5Fcreate succeeded");
+
+ status = H5Pclose(acc_plist);
+ VRFY((status >= 0),"");
+
+ /* setup dimensionality object */
+
+ dims[0] = SPACE_DIM1;
+ dims[1] = SPACE_DIM2;
+
+ /* each process takes a slab of rows
+ stride[0] = 1;
+ stride[1] = 1;
+ count[0] = SPACE_DIM1/mpi_size;
+ count[1] = SPACE_DIM2;
+ start[0] = mpi_rank*count[0];
+ start[1] = 0;
+ block[0] = 1;
+ block[1] = 1;
+ */
+
+ /* allocate memory for data buffer */
+ data_array1 = (int *)malloc(SPACE_DIM1*SPACE_DIM2*sizeof(int));
+ VRFY((data_array1 != NULL), "data_array1 malloc succeeded");
+
+ /* set up dimensions of the slab this process accesses */
+ ccslab_set(mpi_rank, mpi_size, start, count, stride, block, select_factor);
+
+ file_dataspace = H5Screate_simple(2, dims, NULL);
+ VRFY((file_dataspace >= 0),"file dataspace created succeeded");
+
+ crp_plist = H5Pcreate(H5P_DATASET_CREATE);
+ VRFY((crp_plist >= 0),"");
+
+ /* test1: chunk size is equal to dataset size */
+ chunk_dims[0] = SPACE_DIM1/chunk_factor;
+ chunk_dims[1] = SPACE_DIM2/chunk_factor;
+ status = H5Pset_chunk(crp_plist, 2, chunk_dims);
+ VRFY((status >= 0),"chunk creation property list succeeded");
+
+ dataset = H5Dcreate(file,DSET_COLLECTIVE_CHUNK_NAME,H5T_NATIVE_INT,
+ file_dataspace,crp_plist);
+ VRFY((dataset >= 0),"dataset created succeeded");
+/* H5Sclose(file_dataspace); */
+
+ status = H5Pclose(crp_plist);
+ VRFY((status >= 0),"");
+
+ /*put some trivial data in the data array */
+ ccdataset_fill(start, stride,count,block, data_array1);
+ MESG("data_array initialized");
+
+/* file_dataspace = H5Dget_space(dataset); */
+ status=H5Sselect_hyperslab(file_dataspace, H5S_SELECT_SET, start, stride,
+ count, block);
+ VRFY((status >= 0),"hyperslab selection succeeded");
+
+ /* set up the collective transfer property list */
+ xfer_plist = H5Pcreate (H5P_DATASET_XFER);
+ VRFY((xfer_plist >= 0),"");
+
+ status = H5Pset_dxpl_mpio(xfer_plist, H5FD_MPIO_COLLECTIVE);
+ VRFY((status>= 0),"MPIO collective transfer property succeeded");
+
+ /* write data collectively */
+ status = H5Dwrite(dataset, H5T_NATIVE_INT, H5S_ALL, file_dataspace,
+ xfer_plist, data_array1);
+ VRFY((status >= 0),"dataset write succeeded");
+
+ status = H5Dclose(dataset);
+ VRFY((status >= 0),"");
+
+ /* check whether using collective IO */
+ /* Should use H5Pget and H5Pinsert to handle this test. */
+
+/* status = H5Pclose(xfer_plist);
+ VRFY((status >= 0),"");
+
+*/
+ status = H5Sclose(file_dataspace);
+ VRFY((status >= 0),"");
+
+ status = H5Fclose(file);
+ VRFY((status >= 0),"");
+
+ if (data_array1) free(data_array1);
+
+ /* Using read to verify the data inside the dataset is correct */
+
+ /* allocate memory for data buffer */
+ data_array1 = (int *)malloc(SPACE_DIM1*SPACE_DIM2*sizeof(int));
+ VRFY((data_array1 != NULL), "data_array1 malloc succeeded");
+
+ /* allocate memory for data buffer */
+ data_origin1 = (int *)malloc(SPACE_DIM1*SPACE_DIM2*sizeof(int));
+ VRFY((data_origin1 != NULL), "data_origin1 malloc succeeded");
+
+ /* Create the data space */
+ acc_plist = H5Pcreate(H5P_FILE_ACCESS);
+ VRFY((acc_plist >= 0),"");
+
+ status = H5Pset_fapl_mpio(acc_plist,comm,info);
+ VRFY((acc_plist >= 0),"MPIO creation property list succeeded");
+
+ file = H5Fopen(filename,H5F_ACC_RDONLY,acc_plist);
+ VRFY((file >= 0),"H5Fcreate succeeded");
+
+ status = H5Pclose(acc_plist);
+ VRFY((status >= 0),"");
+
+ /* open the dataset collectively */
+ dataset = H5Dopen(file, DSET_COLLECTIVE_CHUNK_NAME);
+ VRFY((dataset >= 0), "");
+
+ /* set up dimensions of the slab this process accesses */
+ ccslab_set(mpi_rank, mpi_size, start, count, stride, block, select_factor);
+
+ /* create a file dataspace independently */
+ file_dataspace = H5Dget_space (dataset);
+ VRFY((file_dataspace >= 0), "");
+ status=H5Sselect_hyperslab(file_dataspace, H5S_SELECT_SET, start, stride, count, block);
+ VRFY((status >= 0), "");
+
+ /* fill dataset with test data */
+ ccdataset_fill(start, stride,count,block, data_origin1);
+ /* read data collectively */
+ status = H5Dread(dataset, H5T_NATIVE_INT, H5S_ALL, file_dataspace,
+ xfer_plist, data_array1);
+ VRFY((status >= 0), "");
+/* printf("mpi rank %d\n",mpi_rank);
+ if(mpi_rank == 2) {
+ for (i = 0; i < SPACE_DIM1; i++) {
+ for (j = 0; j < SPACE_DIM2;j++) {
+ printf("i, j, data, %d, %d, %d \n",i,j,*(data_array1));
+ data_array1++;
+ }
+ }
+ }*/
+ /* verify the read data with original expected data */
+
+ status = ccdataset_vrfy(start, count, stride, block, data_array1, data_origin1);
+ if (status) nerrors++;
+
+
+ /* close dataset collectively */
+ status=H5Dclose(dataset);
+ VRFY((status >= 0), "");
+
+ status=H5Pclose(xfer_plist);
+ VRFY((status >= 0),"");
+
+ /* release all IDs created */
+ H5Sclose(file_dataspace);
+
+ /* close the file collectively */
+ H5Fclose(file);
+
+ /* release data buffers */
+ if (data_array1) free(data_array1);
+ if (data_origin1) free(data_origin1);
+
+}
+
+
+void
+ccslab_set(int mpi_rank, int mpi_size, hssize_t start[], hsize_t count[],
+ hsize_t stride[], hsize_t block[], int mode)
+{
+ switch (mode){
+ case BYROW_CONT:
+ /* Each process takes a slabs of rows. */
+ block[0] = 1;
+ block[1] = 1;
+ stride[0] = 1;
+ stride[1] = 1;
+ count[0] = SPACE_DIM1/mpi_size;
+ count[1] = SPACE_DIM2;
+ start[0] = mpi_rank*count[0];
+ start[1] = 0;
+
+ if (VERBOSE_MED) printf("slab_set BYROW_CONT\n");
+ break;
+ case BYROW_DISCONT:
+ /* Each process takes several disjoint blocks. */
+ block[0] = 2;
+ block[1] = 2;
+ stride[0] = 3;
+ stride[1] = 6;
+ count[0] = 2;
+ count[1] = 3;
+ start[0] = SPACE_DIM1/mpi_size*mpi_rank;
+ start[1] = 0;
+if (VERBOSE_MED) printf("slab_set BYROW_DISCONT\n");
+ break;
+ default:
+ /* Unknown mode. Set it to cover the whole dataset. */
+ printf("unknown slab_set mode (%d)\n", mode);
+ block[0] = SPACE_DIM1;
+ block[1] = SPACE_DIM2;
+ stride[0] = block[0];
+ stride[1] = block[1];
+ count[0] = 1;
+ count[1] = 1;
+ start[0] = 0;
+ start[1] = 0;
+if (VERBOSE_MED) printf("slab_set wholeset\n");
+ break;
+ }
+if (VERBOSE_MED){
+ printf("start[]=(%ld,%ld), count[]=(%lu,%lu), stride[]=(%lu,%lu), block[]=(%lu,%lu), total datapoints=%lu\n",
+ (long)start[0], (long)start[1], (unsigned long)count[0], (unsigned long)count[1],
+ (unsigned long)stride[0], (unsigned long)stride[1], (unsigned long)block[0], (unsigned long)block[1],
+ (unsigned long)(block[0]*block[1]*count[0]*count[1]));
+ }
+}
+
+
+/*
+ * Fill the dataset with trivial data for testing.
+ * Assume dimension rank is 2 and data is stored contiguous.
+ */
+void
+ccdataset_fill(hssize_t start[], hsize_t stride[], hsize_t count[],hsize_t block[], DATATYPE * dataset)
+{
+ DATATYPE *tmptr;
+ DATATYPE *dataptr = dataset;
+ DATATYPE temp;
+ int i, j,k1,k2;
+
+ /* put some trivial data in the data_array */
+
+ tmptr = dataptr;
+ /*for(i=0;i<SPACE_DIM1;i++){
+ for(j=0;j<SPACE_DIM2;j++){
+ *dataptr = 0;
+ dataptr++;
+ }
+ }*/
+
+ dataptr = tmptr;
+
+ /* assign the disjoint block (two-dimensional)data array value
+ through the pointer */
+ for (k1 = 0; k1 < count[0];k1++) {
+ for(i = 0;i < block[0];i++) {
+ for(k2 = 0; k2<count[1];k2++) {
+ for(j=0;j<block[1];j++) {
+
+ dataptr = tmptr + ((start[0]+k1*stride[0]+i)*SPACE_DIM2+
+ start[1]+k2*stride[1]+j);
+
+ /* printf("i,j,k1,k2 %lu %lu %lu %lu \n",i,j,k1,k2);
+ printf("Address of dataptr");
+ printf("= 0x%p\n",dataptr);
+ */
+ *dataptr = (DATATYPE)(k1+k2+i+j);
+ /*temp = *dataptr;
+ printf("data %03d\n",temp);
+ */
+ }
+ }
+ }
+ }
+
+}
+
+/*
+ * Print the first block of the content of the dataset.
+ */
+void
+ccdataset_print(hssize_t start[], hsize_t block[], DATATYPE * dataset)
+{
+ DATATYPE *dataptr = dataset;
+ hsize_t i, j;
+
+ /* print the column heading */
+ printf("Print only the first block of the dataset\n");
+ printf("%-8s", "Cols:");
+ for (j=0; j < block[1]; j++){
+ printf("%3ld ", (long)(start[1]+j));
+ }
+ printf("\n");
+
+ /* print the slab data */
+ for (i=0; i < block[0]; i++){
+ printf("Row %2ld: ", (long)(i+start[0]));
+ for (j=0; j < block[1]; j++){
+ printf("%03d ", *dataptr++);
+ }
+ printf("\n");
+ }
+}
+
+
+/*
+ * Print the content of the dataset.
+ */
+int ccdataset_vrfy(hssize_t start[], hsize_t count[], hsize_t stride[], hsize_t block[], DATATYPE *dataset, DATATYPE *original)
+{
+ hsize_t i, j,k1,k2;
+ int vrfyerrs;
+ DATATYPE *dataptr,*oriptr;
+
+ /* print it if VERBOSE_MED */
+ if (VERBOSE_MED) {
+ printf("dataset_vrfy dumping:::\n");
+ printf("start(%ld, %ld), count(%lu, %lu), stride(%lu, %lu), block(%lu, %lu)\n",
+ (long)start[0], (long)start[1], (unsigned long)count[0], (unsigned long)count[1],
+ (unsigned long)stride[0], (unsigned long)stride[1], (unsigned long)block[0], (unsigned long)block[1]);
+ printf("original values:\n");
+ ccdataset_print(start, block, original);
+ printf("compared values:\n");
+ ccdataset_print(start, block, dataset);
+ }
+
+ vrfyerrs = 0;
+
+ for (k1 = 0; k1 < count[0];k1++) {
+ for(i = 0;i < block[0];i++) {
+ for(k2 = 0; k2<count[1];k2++) {
+ for(j=0;j<block[1];j++) {
+
+ dataptr = dataset + ((start[0]+k1*stride[0]+i)*SPACE_DIM2+
+ start[1]+k2*stride[1]+j);
+ oriptr = original + ((start[0]+k1*stride[0]+i)*SPACE_DIM2+
+ start[1]+k2*stride[1]+j);
+
+ if (*dataptr != *oriptr){
+ if (vrfyerrs++ < MAX_ERR_REPORT || VERBOSE_MED){
+ printf("Dataset Verify failed at [%ld][%ld]: expect %d, got %d\n",
+ (long)i, (long)j,
+ *(original), *(dataset));
+ }
+ }
+ }
+ }
+ }
+ }
+ if (vrfyerrs > MAX_ERR_REPORT && !VERBOSE_MED)
+ printf("[more errors ...]\n");
+ if (vrfyerrs)
+ printf("%d errors found in dataset_vrfy\n", vrfyerrs);
+ return(vrfyerrs);
+}