Changeset 83 for trunk/src/bench.c

Timestamp:

05/04/10 03:38:19 (3 years ago)

Author:

faltet

Message:

New version of bench.c that accepts command line params.

Version bumped to 0.9.0.

File:

• trunk/src/bench.c (modified) (6 diffs)

trunk/src/bench.c

@@ -37 +37 @@
 /* #define NCHUNKS (100) */
 /* #define NITER  (10)               /\* Number of iterations *\/ */
-#define NCHUNKS (100)
+#define NCHUNKS (128)
 #define NITER  (10)               /* Number of iterations */
 
@@ -106 +106 @@
 
   v = (i<<26)^(i<<18)^(i<<11)^(i<<3)^i;
-  v &= (1 << (32-rshift)) - 1;
+  v &= (1 << rshift) - 1;
   return v;
 }
@@ -126 +126 @@
     //_src[i] = i * 1/.3;
     //_src[i] = i;
-    //_src[i] = rand() >> rshift;
+    //_src[i] = rand() >> (32-rshift);
     _src[i] = get_value(i, rshift);
   }
@@ -132 +132 @@
 
 
-int main(void) {
+int main(int argc, char *argv[]) {
   int nbytes, cbytes;
   void *src, *srccpy;
@@ -140 +140 @@
   float tmemcpy, tshuf, tunshuf;
   int clevel;
-  unsigned int size = 128*1024;   /* Buffer size */
-  unsigned int elsize = 4;        /* Datatype size */
-  int rshift = 12;                /* For random data */
-  int nthreads = 1;               /* The number of threads */
-  int doshuffle = 1;              /* Shuffle? */
+  unsigned int size = 1024*1024;   /* Buffer size */
+  unsigned int elsize = 4;         /* Datatype size */
+  int rshift = 20;                 /* Significant bits */
+  int nthreads = 1;                /* The number of threads */
+  int doshuffle = 1;               /* Shuffle? */
   unsigned char *orig, *round;
+
+  if (argc >= 2) {
+    nthreads = atoi(argv[1]);
+  }
+  if (argc >= 3) {
+    size = atoi(argv[2])*1024;
+    if (size > 2*1024*1024) {
+      printf("The test is going to require more than 256 MB of RAM!\n");
+    }
+  }
+  if (argc >= 4) {
+    elsize = atoi(argv[3]);
+  }
+  if (argc >= 5) {
+    rshift = atoi(argv[4]);
+  }
+  if (argc >= 6) {
+    doshuffle = atoi(argv[5]);
+  }
+  if (argc >= 7) {
+    printf("Usage: bench [nthreads [bufsize(KB) [typesize [sbits [shuf]]]]]\n");
+    exit(1);
+  }
 
   blosc_set_nthreads(nthreads);
@@ -163 +186 @@
   printf("********************** Setup info *****************************\n");
   printf("Blosc version: %s (%s)\n", BLOSC_VERSION_STRING, BLOSC_VERSION_DATE);
-  printf("Using random data with %d significant bits (out of 32)\n", 32-rshift);
-  printf("Dataset size: %d bytes\t Type size: %d bytes\n", size, elsize);
-  printf("Shuffle active?  %s\n", doshuffle ? "Yes" : "No");
+  printf("Using random data with %d significant bits (out of 32)\n", rshift);
+  printf("Dataset size: %d bytes\tType size: %d bytes\n", size, elsize);
+  printf("Shuffle active?  %s\t\t", doshuffle ? "Yes" : "No");
+  printf("Number of threads: %d\n", nthreads);
   printf("********************** Running benchmarks *********************\n");