print sizes of file datatype and buffer datatype

Author: wkliao

tests/large_dtype.c

@@ -17,6 +17,7 @@
 
 #include <stdio.h>
 #include <stdlib.h>
+#include <string.h> /* strcpy() */
 #include <unistd.h> /* getopt() */
 #include <assert.h>
 
@@ -53,15 +54,15 @@ static void
 usage(char *argv0)
 {
     char *help =
-    "Usage: %s [-hvwr | -n num | -l num | -g num | file_name]\n"
+    "Usage: %s [-hvwr | -n num | -l num | -g num ] -f file_name\n"
     "       [-h] Print this help\n"
     "       [-v] verbose mode\n"
     "       [-w] performs write only (default: both write and read)\n"
     "       [-r] performs read  only (default: both write and read)\n"
     "       [-n num] number of global variables (default: %d)\n"
     "       [-l num] length of dimensions X and Y each local variable (default: %d)\n"
     "       [-g num] gap at the end of each dimension (default: %d)\n"
-    "       [file_name] output file name\n";
+    "        -f file_name: output file name\n";
     fprintf(stderr, help, argv0, NVARS, LEN, GAP);
 }
 
@@ -70,15 +71,15 @@ int main(int argc, char **argv)
 {
     char filename[512];
     size_t i, buf_len;
-    int ret, err, rank, verbose, omode, nprocs, do_read, do_write;
+    int ret, err, rank, verbose, omode, nprocs, do_read, do_write, type_size;
     int nvars, len, gap, psize[2], gsize[2], count[2], start[2];
     char *buf;
     MPI_File     fh;
     MPI_Datatype subType, filetype, buftype;
     MPI_Status   status;
     MPI_Offset fsize;
     int *array_of_blocklengths;
-    MPI_Aint *array_of_displacements;
+    MPI_Aint lb, extent, *array_of_displacements;
     MPI_Datatype *array_of_types;
     MPI_Request req[2];
 
@@ -93,9 +94,10 @@ int main(int argc, char **argv)
     do_write = 1;
     do_read  = 1;
     verbose = 0;
+    filename[0] = '\0';
 
     /* get command-line arguments */
-    while ((ret = getopt(argc, argv, "hvwrn:l:g:")) != EOF)
+    while ((ret = getopt(argc, argv, "hvwrn:l:g:f:")) != EOF)
         switch(ret) {
             case 'v': verbose = 1;
                       break;
@@ -109,15 +111,19 @@ int main(int argc, char **argv)
                       break;
             case 'g': gap = atoi(optarg);
                       break;
+            case 'f': strcpy(filename, optarg);
+                      break;
             case 'h':
             default:  if (rank==0) usage(argv[0]);
                       MPI_Finalize();
                       return 1;
         }
-    if (argv[optind] == NULL)
-        sprintf(filename, "%s.out", argv[0]);
-    else
-        snprintf(filename, 256, "%s", argv[optind]);
+
+    if (filename[0] == '\0') {
+        if (rank==0) usage(argv[0]);
+        MPI_Finalize();
+        return 1;
+    }
 
     array_of_blocklengths = (int*) malloc(sizeof(int) * nvars);
     array_of_displacements = (MPI_Aint*) malloc(sizeof(MPI_Aint) * nvars);
@@ -179,6 +185,12 @@ int main(int argc, char **argv)
     err = MPI_Type_free(&subType);
     CHECK_MPI_ERROR("MPI_Type_free");
 
+    MPI_Type_size(filetype, &type_size);
+    lb = 0;
+    MPI_Type_get_extent(filetype, &lb, &extent);
+    if (verbose && rank == 0)
+        printf("file   type size = %d extent = %ld\n", type_size, extent);
+
     /* Create local buffer datatype: each 2D variable is of size len x len */
     gsize[0] = len;
     gsize[1] = len;
@@ -214,6 +226,12 @@ int main(int argc, char **argv)
     free(array_of_displacements);
     free(array_of_types);
 
+    MPI_Type_size(buftype, &type_size);
+    lb = 0;
+    MPI_Type_get_extent(buftype, &lb, &extent);
+    if (verbose && rank == 0)
+        printf("buffer type size = %d extent = %ld\n", type_size, extent);
+
     /* allocate a local buffer */
     buf_len = (size_t)nvars * len * len;
     buf = (char*) malloc(buf_len);