From 12b97053570a5bb9c4b1115eae9509cafd975eb1 Mon Sep 17 00:00:00 2001
From: Mattias Andrée <maandree@kth.se>
Date: Sun, 24 Jul 2016 15:33:37 +0200
Subject: Rename bench/{benchmark.h => util.h}
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Mattias Andrée <maandree@kth.se>
---
 bench/util.h | 146 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 146 insertions(+)
 create mode 100644 bench/util.h

(limited to 'bench/util.h')

diff --git a/bench/util.h b/bench/util.h
new file mode 100644
index 0000000..63ca1ff
--- /dev/null
+++ b/bench/util.h
@@ -0,0 +1,146 @@
+#if defined(__linux__)
+# define _GNU_SOURCE
+# include <sched.h>
+#endif
+
+#include <limits.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <time.h>
+#include <unistd.h>
+
+
+#ifdef BENCHMARK_LIB
+# include BENCHMARK_LIB
+#else
+# include "../zahl.h"
+# define BIGINT_LIBRARY  "libzahl"
+#endif
+
+#ifndef LIBRARY_SUFFIX
+# define LIBRARY_SUFFIX  ""
+#endif
+
+#ifndef USE_CPU
+# define USE_CPU  0
+#endif
+
+
+#ifndef CLOCK_MONOTONIC_RAW
+# define CLOCK_MONOTONIC_RAW  CLOCK_MONOTONIC
+#endif
+
+
+#ifdef __x86_64__
+# define RDTSC_MAYBE_SUPPORTED
+#endif
+
+
+#if !defined(USE_RDTSC) && !defined(USE_CLOCK) && !defined(USE_GETTIME)
+# if 1 && defined(RDTSC_MAYBE_SUPPORTED) && defined(__linux__)
+#  define USE_RDTSC
+# elif 1
+#  define USE_CLOCK
+# else
+#  define USE_GETTIME
+# endif
+#endif
+
+
+
+static struct timespec dur;
+static char timebuf[512];
+
+
+#if defined(USE_RDTSC) && defined(__x86_64__)
+typedef unsigned long long int rdtsc_t;
+static unsigned int start_high, start_low, end_high, end_low;
+static unsigned long long int freq;
+
+# define TIC  (rdtsc(&start_low, &start_high))
+# define TOC\
+	do {\
+		rdtsc_t dur_cycles;\
+		double dur_seconds;\
+		rdtsc(&end_low, &end_high);\
+		dur_cycles = rdtsc_join(end_low, end_high);\
+		dur_cycles -= rdtsc_join(start_low, start_high);\
+		dur_seconds = (double)dur_cycles;\
+		dur_seconds /= 1000 * (double)freq;\
+		dur_seconds -= (double)(dur.tv_sec = (int)dur_seconds);\
+		dur.tv_nsec = (long int)(dur_seconds * 1000000000L);\
+	} while (0)
+
+static inline void
+rdtsc(unsigned int *low, unsigned int *high)
+{
+	__asm__ __volatile__ ("rdtsc" : "=a"(*low), "=d"(*high));
+}
+
+static inline rdtsc_t
+rdtsc_join(unsigned int low, unsigned int high)
+{
+	return (rdtsc_t)low | (((rdtsc_t)high) << 32);
+}
+
+
+#elif defined(USE_CLOCK)
+static clock_t start, end;
+
+# define TIC  (start = clock())
+# define TOC\
+	do {\
+		end = clock();\
+		dur.tv_sec = (end - start) / 1000000ULL;\
+		dur.tv_nsec = ((end - start) % 1000000ULL) * 1000;\
+	} while (0)
+
+
+#elif defined(USE_GETTIME)
+static struct timespec start;
+
+# define TIC  clock_gettime(CLOCK_MONOTONIC_RAW, &start)
+# define TOC\
+	do {\
+		clock_gettime(CLOCK_MONOTONIC_RAW, &dur);\
+		dur.tv_sec -= start.tv_sec;\
+		dur.tv_nsec -= start.tv_nsec;\
+		if (dur.tv_nsec < 0) {\
+			dur.tv_nsec += 1000000000L;\
+			dur.tv_sec -= 1;\
+		}\
+	} while (0)
+#endif
+
+
+#define TICKS  ((unsigned long long int)(dur.tv_sec) * 1000000000ULL + (unsigned long long int)(dur.tv_nsec))
+#define STIME  (sprintf(timebuf, "%lli.%09li", (long long)(dur.tv_sec), dur.tv_nsec), timebuf)
+
+
+static void
+benchmark_init(void)
+{
+#if defined(__linux__)
+	cpu_set_t cpuset;
+# if defined(USE_RDTSC)
+	FILE *f;
+	char *line = 0;
+	size_t size = 0;
+	char path[PATH_MAX];
+# endif
+	CPU_ZERO(&cpuset);
+	CPU_SET(USE_CPU, &cpuset);
+	sched_setaffinity(getpid(), sizeof(cpuset), &cpuset);
+# if defined(USE_RDTSC)
+	sprintf(path, "/sys/devices/system/cpu/cpu%i/cpufreq/cpuinfo_max_freq", USE_CPU);
+	f = fopen(path, "r");
+	if (getline(&line, &size, f) < 0)
+		abort();
+	fclose(f);
+	freq = strtoull(line, 0, 10);
+	free(line);
+# endif
+#endif
+	(void) timebuf;
+}
-- 
cgit v1.2.3-70-g09d2