Changes in src/benchmark/bench.h [d67cdb7:b7170a64]

File:

• src/benchmark/bench.h (modified) (2 diffs)

src/benchmark/bench.h

@@ -1 +1 @@
 #pragma once
 
-#ifdef __CFORALL__
+#if defined(__CFORALL__)
 extern "C" {
 #endif
+        #include <stdlib.h>
         #include <unistd.h>                                     // sysconf
         #include <sys/times.h>                                  // times
         #include <time.h>
-#ifdef __CFORALL__
+#if defined(__CFORALL__)
 }
 #endif
@@ -27 +28 @@
 } // Time
 
-#ifndef N
-#define N 10000000
+#ifndef BENCH_N
+#define BENCH_N 500 //10000000
 #endif
+
+#define BENCH(statement, output)                \
+        size_t n = BENCH_N;                     \
+        if( argc > 2 ) return 1;                \
+        if( argc == 2 ) {                               \
+                n = atoi(argv[1]);              \
+        }                                               \
+        long long int StartTime, EndTime;       \
+        StartTime = Time();                     \
+        statement;                                      \
+        EndTime = Time();                               \
+        unsigned long long int output =         \
+        ( EndTime - StartTime ) / n;
 
 unsigned int default_preemption() {