blob: 8a6faad6ece34f2bf162bbd31d12be3b17a472a2 [file] [log] [blame]
Emilio G. Cota070e3ed2016-06-27 15:02:05 -04001#include "qemu/osdep.h"
2#include "qemu/thread.h"
3#include "qemu/host-utils.h"
4#include "qemu/processor.h"
Peter Maydell5df022c2022-02-26 18:07:23 +00005#include "qemu/memalign.h"
Emilio G. Cota070e3ed2016-06-27 15:02:05 -04006
7struct thread_info {
8 uint64_t r;
9} QEMU_ALIGNED(64);
10
11struct count {
Emilio G. Cota70c31262018-04-25 10:54:56 +080012 QemuMutex lock;
Emilio G. Cota070e3ed2016-06-27 15:02:05 -040013 unsigned long val;
14} QEMU_ALIGNED(64);
15
16static QemuThread *threads;
17static struct thread_info *th_info;
18static unsigned int n_threads = 1;
19static unsigned int n_ready_threads;
20static struct count *counts;
21static unsigned int duration = 1;
22static unsigned int range = 1024;
Emilio G. Cota70c31262018-04-25 10:54:56 +080023static bool use_mutex;
Emilio G. Cota070e3ed2016-06-27 15:02:05 -040024static bool test_start;
25static bool test_stop;
26
27static const char commands_string[] =
28 " -n = number of threads\n"
Emilio G. Cota70c31262018-04-25 10:54:56 +080029 " -m = use mutexes instead of atomic increments\n"
Emilio G. Cota9d5cff32018-08-15 11:43:44 -040030 " -p = enable sync profiler\n"
Emilio G. Cota070e3ed2016-06-27 15:02:05 -040031 " -d = duration in seconds\n"
32 " -r = range (will be rounded up to pow2)";
33
34static void usage_complete(char *argv[])
35{
36 fprintf(stderr, "Usage: %s [options]\n", argv[0]);
37 fprintf(stderr, "options:\n%s\n", commands_string);
38}
39
40/*
41 * From: https://en.wikipedia.org/wiki/Xorshift
42 * This is faster than rand_r(), and gives us a wider range (RAND_MAX is only
43 * guaranteed to be >= INT_MAX).
44 */
45static uint64_t xorshift64star(uint64_t x)
46{
47 x ^= x >> 12; /* a */
48 x ^= x << 25; /* b */
49 x ^= x >> 27; /* c */
50 return x * UINT64_C(2685821657736338717);
51}
52
53static void *thread_func(void *arg)
54{
55 struct thread_info *info = arg;
56
Stefan Hajnoczid73415a2020-09-23 11:56:46 +010057 qatomic_inc(&n_ready_threads);
58 while (!qatomic_read(&test_start)) {
Emilio G. Cota070e3ed2016-06-27 15:02:05 -040059 cpu_relax();
60 }
61
Stefan Hajnoczid73415a2020-09-23 11:56:46 +010062 while (!qatomic_read(&test_stop)) {
Emilio G. Cota070e3ed2016-06-27 15:02:05 -040063 unsigned int index;
64
65 info->r = xorshift64star(info->r);
66 index = info->r & (range - 1);
Emilio G. Cota70c31262018-04-25 10:54:56 +080067 if (use_mutex) {
68 qemu_mutex_lock(&counts[index].lock);
69 counts[index].val += 1;
70 qemu_mutex_unlock(&counts[index].lock);
71 } else {
Stefan Hajnoczid73415a2020-09-23 11:56:46 +010072 qatomic_inc(&counts[index].val);
Emilio G. Cota70c31262018-04-25 10:54:56 +080073 }
Emilio G. Cota070e3ed2016-06-27 15:02:05 -040074 }
75 return NULL;
76}
77
78static void run_test(void)
79{
Emilio G. Cota070e3ed2016-06-27 15:02:05 -040080 unsigned int i;
81
Stefan Hajnoczid73415a2020-09-23 11:56:46 +010082 while (qatomic_read(&n_ready_threads) != n_threads) {
Emilio G. Cota070e3ed2016-06-27 15:02:05 -040083 cpu_relax();
84 }
Alex Bennéeeb4f8e12019-01-11 13:50:02 +000085
Stefan Hajnoczid73415a2020-09-23 11:56:46 +010086 qatomic_set(&test_start, true);
Alex Bennéeeb4f8e12019-01-11 13:50:02 +000087 g_usleep(duration * G_USEC_PER_SEC);
Stefan Hajnoczid73415a2020-09-23 11:56:46 +010088 qatomic_set(&test_stop, true);
Emilio G. Cota070e3ed2016-06-27 15:02:05 -040089
90 for (i = 0; i < n_threads; i++) {
91 qemu_thread_join(&threads[i]);
92 }
93}
94
95static void create_threads(void)
96{
97 unsigned int i;
98
99 threads = g_new(QemuThread, n_threads);
100 th_info = g_new(struct thread_info, n_threads);
101 counts = qemu_memalign(64, sizeof(*counts) * range);
102 memset(counts, 0, sizeof(*counts) * range);
Emilio G. Cota70c31262018-04-25 10:54:56 +0800103 for (i = 0; i < range; i++) {
104 qemu_mutex_init(&counts[i].lock);
105 }
Emilio G. Cota070e3ed2016-06-27 15:02:05 -0400106
107 for (i = 0; i < n_threads; i++) {
108 struct thread_info *info = &th_info[i];
109
110 info->r = (i + 1) ^ time(NULL);
111 qemu_thread_create(&threads[i], NULL, thread_func, info,
112 QEMU_THREAD_JOINABLE);
113 }
114}
115
116static void pr_params(void)
117{
118 printf("Parameters:\n");
119 printf(" # of threads: %u\n", n_threads);
120 printf(" duration: %u\n", duration);
121 printf(" ops' range: %u\n", range);
122}
123
124static void pr_stats(void)
125{
126 unsigned long long val = 0;
127 unsigned int i;
128 double tx;
129
130 for (i = 0; i < range; i++) {
131 val += counts[i].val;
132 }
133 tx = val / duration / 1e6;
134
135 printf("Results:\n");
136 printf("Duration: %u s\n", duration);
137 printf(" Throughput: %.2f Mops/s\n", tx);
138 printf(" Throughput/thread: %.2f Mops/s/thread\n", tx / n_threads);
139}
140
141static void parse_args(int argc, char *argv[])
142{
143 int c;
144
145 for (;;) {
Emilio G. Cota9d5cff32018-08-15 11:43:44 -0400146 c = getopt(argc, argv, "hd:n:mpr:");
Emilio G. Cota070e3ed2016-06-27 15:02:05 -0400147 if (c < 0) {
148 break;
149 }
150 switch (c) {
151 case 'h':
152 usage_complete(argv);
153 exit(0);
154 case 'd':
155 duration = atoi(optarg);
156 break;
157 case 'n':
158 n_threads = atoi(optarg);
159 break;
Emilio G. Cota70c31262018-04-25 10:54:56 +0800160 case 'm':
161 use_mutex = true;
162 break;
Emilio G. Cota9d5cff32018-08-15 11:43:44 -0400163 case 'p':
164 qsp_enable();
165 break;
Emilio G. Cota070e3ed2016-06-27 15:02:05 -0400166 case 'r':
167 range = pow2ceil(atoi(optarg));
168 break;
169 }
170 }
171}
172
173int main(int argc, char *argv[])
174{
175 parse_args(argc, argv);
176 pr_params();
177 create_threads();
178 run_test();
179 pr_stats();
180 return 0;
181}