Make configuration configurable ...

benchmark.c

@@ -30,6 +30,9 @@
 
 const int N_RUNS = 4;
 const int N_ARRAYS = 6;
+const int INITIAL_SIZE = 8;
+
+#define UPDATE_SIZE(size) size *= 8;
 
 
 #ifdef HAVE_OPENCL
@@ -61,7 +64,7 @@ loop_data_opencl (const char *vendor,
 {
     Timer *timer;
     cl_int err;
-    size_t size = 8;
+    size_t size = INITIAL_SIZE;
 
     timer = timer_new ();
 
@@ -76,7 +79,8 @@ loop_data_opencl (const char *vendor,
         cl_mem dev_mem;
         cl_mem dev_out_mem;
 
-        size *= 8;
+        UPDATE_SIZE (size);
+
         size_bytes = size * 2 * sizeof (float);
         host_orig_mem = malloc (size_bytes);
         host_result_mem = malloc (size_bytes);
@@ -136,7 +140,7 @@ static void
 loop_data_fftw (double *times, double *errors, FILE *fp)
 {
     Timer *timer;
-    size_t size = 8;
+    size_t size = INITIAL_SIZE;
 
     timer = timer_new ();
     fprintf (fp, "FFTW_bw FFTW_err ");
@@ -151,7 +155,8 @@ loop_data_fftw (double *times, double *errors, FILE *fp)
         double mflops;
         double sum = 0.0;
 
-        size *= 8;
+        UPDATE_SIZE (size);
+
         host_orig_mem = fftw_malloc (sizeof (fftw_complex) * size);
         host_immediate_mem = fftw_malloc (sizeof (fftw_complex) * size);
         host_result_mem = fftw_malloc (sizeof (fftw_complex) * size);
@@ -386,7 +391,7 @@ main (int argc, char **argv)
         {0, 0, 0, 0}
     };
 
-    size_t size = 8;
+    size_t size = INITIAL_SIZE;
     FILE *fp;
 
     /* Parse options */
@@ -455,7 +460,8 @@ main (int argc, char **argv)
 #endif
 
     for (int i = 0; i < N_ARRAYS; i++) {
-        size *= 8;
+        UPDATE_SIZE (size);
+
         fprintf (fp, "\n%zu ", size);
 
         for (int j = 0; j < n_devices; j++) {