bitpolymul/benchmark.h at master · fast-crypto-lab/bitpolymul · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
/*
 * $Id: benchmark.h 1271 2008-06-08 08:06:13Z owenhsin $
 */

#ifndef BENCHMARK_H
#define BENCHMARK_H

#include <stdint.h>
#include <stdio.h>
#include <string.h>
#include <sys/types.h>

#define CONFIG_BENCH_SYSTIME

#if defined(CONFIG_BENCH_SYSTIME)
#include <sys/time.h>
#endif

/* Copied from http://en.wikipedia.org/wiki/RDTSC */
#ifdef __cplusplus
extern "C"
{
#endif
#if !defined(CONFIG_BENCH_SYSTIME)
#if defined(CONFIG_NEON)
static inline uint32_t
rdtsc32(void)
{
#if defined(__GNUC__) && defined(__ARM_ARCH_7A__)
	uint32_t r = 0;
	asm volatile("mrc p15, 0, %0, c9, c13, 0" : "=r"(r) );
	return r;
#else
#error Unsupported architecture/compiler!
#endif
}
#else
static inline uint64_t rdtsc() {
	uint32_t lo, hi;
	/* We cannot use "=A", since this would use %rax on x86_64 */
	__asm__ __volatile__ ("rdtsc" : "=a" (lo), "=d" (hi));
	return (uint64_t)hi << 32 | lo;
}
#endif
#endif
#ifdef __cplusplus
}
#endif

#define RECMAX 6

#define BENCHMARK(bm,call) do { \
		bm_start(&(bm)); \
		call; \
		bm_stop(&(bm)); \
	} while (0)

struct benchmark {
#if defined(CONFIG_BENCH_SYSTIME)
	struct timeval start;
	struct timeval stop;
#else
	uint64_t start;
	uint64_t stop;
#endif
	double record[RECMAX];
	double acc;
	int currec;
	int count;
};

static inline void
bm_init(struct benchmark *bm)
{
	memset(bm, 0, sizeof(*bm));
}

static inline void
bm_start(struct benchmark *bm)
{
#if defined(CONFIG_BENCH_SYSTIME)
	gettimeofday(& bm->start , NULL);
#else
#if defined(CONFIG_NEON)
	bm->start = rdtsc32();
#else
	bm->start = rdtsc();
#endif
#endif
}

static inline void
bm_stop(struct benchmark *bm)
{
#if defined(CONFIG_BENCH_SYSTIME)
	gettimeofday(& bm->stop , NULL);
	bm->record[bm->currec] = (bm->stop.tv_sec - bm->start.tv_sec)*1000000.0; /* sec to us */
	bm->record[bm->currec] += bm->stop.tv_usec - bm->start.tv_usec;
#else
#if defined(CONFIG_NEON)
	bm->stop = rdtsc32();
#else
	bm->stop = rdtsc();
#endif
	bm->record[bm->currec] = bm->stop - bm->start;
#endif
	bm->acc += bm->record[bm->currec];
	bm->currec = (bm->currec + 1) % RECMAX;
	++bm->count;
}

static inline void
bm_dump(char *buf, size_t bufsize, const struct benchmark *bm)
{
	int i;
	size_t len;

	len = snprintf(buf, bufsize, "%.0lf (%d):", bm->acc/bm->count, bm->count);
	buf += len;
	bufsize -= len;
	for (i = 0; i < RECMAX; ++i) {
		len = snprintf(buf, bufsize, " %.0lf", bm->record[i]);
		buf += len;
		bufsize -= len;
	}
}

#endif /* BENCHMARK_H */