-
Notifications
You must be signed in to change notification settings - Fork 0
/
thr_ping.c
102 lines (83 loc) · 1.93 KB
/
thr_ping.c
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
#define _GNU_SOURCE
#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include "bench.h"
#define N (1*1000*1000)
#define MAXTHREADS 16
struct pingpong {
long ping;
long *cache[0] __attribute__((aligned(128)));
long pong;
} *pingpong;
void ping(struct thrarg *arg)
{
(void)arg;
long ping_val, pong_val;
long *ping = &pingpong->ping;
long *pong = &pingpong->pong;
long n = arg->params.iters;
ping_val = __atomic_load_n(ping, __ATOMIC_RELAXED);
assert(ping_val == 0);
do {
ping_val += 1;
__atomic_store_n(ping, ping_val, __ATOMIC_RELEASE);
do {
pong_val = __atomic_load_n(pong, __ATOMIC_ACQUIRE);
} while (pong_val != ping_val);
} while (pong_val != n);
}
void pong(struct thrarg *arg)
{
(void)arg;
long ping_val, pong_val;
long *ping = &pingpong->ping;
long *pong = &pingpong->pong;
long n = arg->params.iters;
pong_val = __atomic_load_n(pong, __ATOMIC_RELAXED);
assert(pong_val == 0);
do {
do {
ping_val = __atomic_load_n(ping, __ATOMIC_ACQUIRE);
} while (ping_val == pong_val);
pong_val += 1;
__atomic_store_n(pong, pong_val, __ATOMIC_RELEASE);
} while (ping_val != n);
}
void benchmark_ping(struct thrarg *arg)
{
if (arg->params.id)
pong(arg);
else
ping(arg);
}
void init(struct thrarg *arg)
{
(void)arg;
memset(pingpong, 0, sizeof(struct pingpong));
}
int main(int argc, char **argv)
{
unsigned nthreads = 2;
(void)argc; (void)argv;
if (posix_memalign((void **)&pingpong, 128, sizeof(struct pingpong))) {
perror("posix_memalign");
return 1;
}
memset(pingpong, 0, sizeof(struct pingpong));
struct thrarg thrarg = { .params = {
.threads = nthreads,
.benchmark = benchmark_ping,
.init = init,
.min_time = 1*1000*1000,
}};
int err = benchmark_auto(&thrarg);
if (err) {
fprintf(stderr, "Bench error %s\n", strerror(err));
return 1;
}
printf("%.2f %.2f %.2f\n", thrarg.result.avg, thrarg.result.err, thrarg.result.u);
return 0;
}