Commit | Line | Data |
---|---|---|
8a953620 MD |
1 | /* |
2 | * rcutorture.h: simple user-level performance/stress test of RCU. | |
3 | * | |
4 | * Usage: | |
5 | * ./rcu <nreaders> rperf [ <cpustride> ] | |
6 | * Run a read-side performance test with the specified | |
7 | * number of readers spaced by <cpustride>. | |
8 | * Thus "./rcu 16 rperf 2" would run 16 readers on even-numbered | |
9 | * CPUs from 0 to 30. | |
10 | * ./rcu <nupdaters> uperf [ <cpustride> ] | |
11 | * Run an update-side performance test with the specified | |
12 | * number of updaters and specified CPU spacing. | |
13 | * ./rcu <nreaders> perf [ <cpustride> ] | |
14 | * Run a combined read/update performance test with the specified | |
15 | * number of readers and one updater and specified CPU spacing. | |
16 | * The readers run on the low-numbered CPUs and the updater | |
17 | * of the highest-numbered CPU. | |
18 | * | |
19 | * The above tests produce output as follows: | |
20 | * | |
21 | * n_reads: 46008000 n_updates: 146026 nreaders: 2 nupdaters: 1 duration: 1 | |
22 | * ns/read: 43.4707 ns/update: 6848.1 | |
23 | * | |
24 | * The first line lists the total number of RCU reads and updates executed | |
25 | * during the test, the number of reader threads, the number of updater | |
26 | * threads, and the duration of the test in seconds. The second line | |
27 | * lists the average duration of each type of operation in nanoseconds, | |
28 | * or "nan" if the corresponding type of operation was not performed. | |
29 | * | |
30 | * ./rcu <nreaders> stress | |
31 | * Run a stress test with the specified number of readers and | |
32 | * one updater. None of the threads are affinitied to any | |
33 | * particular CPU. | |
34 | * | |
35 | * This test produces output as follows: | |
36 | * | |
37 | * n_reads: 114633217 n_updates: 3903415 n_mberror: 0 | |
38 | * rcu_stress_count: 114618391 14826 0 0 0 0 0 0 0 0 0 | |
39 | * | |
40 | * The first line lists the number of RCU read and update operations | |
41 | * executed, followed by the number of memory-ordering violations | |
42 | * (which will be zero in a correct RCU implementation). The second | |
43 | * line lists the number of readers observing progressively more stale | |
44 | * data. A correct RCU implementation will have all but the first two | |
45 | * numbers non-zero. | |
46 | * | |
47 | * This program is free software; you can redistribute it and/or modify | |
48 | * it under the terms of the GNU General Public License as published by | |
49 | * the Free Software Foundation; either version 2 of the License, or | |
50 | * (at your option) any later version. | |
51 | * | |
52 | * This program is distributed in the hope that it will be useful, | |
53 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
54 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
55 | * GNU General Public License for more details. | |
56 | * | |
57 | * You should have received a copy of the GNU General Public License | |
58 | * along with this program; if not, write to the Free Software | |
3282a76b | 59 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
8a953620 MD |
60 | * |
61 | * Copyright (c) 2008 Paul E. McKenney, IBM Corporation. | |
62 | */ | |
63 | ||
64 | /* | |
65 | * Test variables. | |
66 | */ | |
67 | ||
b57aee66 | 68 | #include <stdlib.h> |
ad460058 MD |
69 | #include "tap.h" |
70 | ||
71 | #define NR_TESTS 1 | |
b57aee66 | 72 | |
8a953620 MD |
73 | DEFINE_PER_THREAD(long long, n_reads_pt); |
74 | DEFINE_PER_THREAD(long long, n_updates_pt); | |
75 | ||
26b5a74b MD |
76 | enum callrcu_type { |
77 | CALLRCU_GLOBAL, | |
78 | CALLRCU_PERCPU, | |
79 | CALLRCU_PERTHREAD, | |
80 | }; | |
81 | ||
82 | static enum callrcu_type callrcu_type = CALLRCU_GLOBAL; | |
83 | ||
8a953620 MD |
84 | long long n_reads = 0LL; |
85 | long n_updates = 0L; | |
6ee91d83 | 86 | int nthreadsrunning; |
8a953620 MD |
87 | char argsbuf[64]; |
88 | ||
89 | #define GOFLAG_INIT 0 | |
90 | #define GOFLAG_RUN 1 | |
91 | #define GOFLAG_STOP 2 | |
92 | ||
4967f005 PB |
93 | volatile int goflag __attribute__((__aligned__(CAA_CACHE_LINE_SIZE))) |
94 | = GOFLAG_INIT; | |
8a953620 MD |
95 | |
96 | #define RCU_READ_RUN 1000 | |
97 | ||
98 | //MD | |
99 | #define RCU_READ_NESTABLE | |
100 | ||
101 | #ifdef RCU_READ_NESTABLE | |
102 | #define rcu_read_lock_nest() rcu_read_lock() | |
103 | #define rcu_read_unlock_nest() rcu_read_unlock() | |
104 | #else /* #ifdef RCU_READ_NESTABLE */ | |
105 | #define rcu_read_lock_nest() | |
106 | #define rcu_read_unlock_nest() | |
107 | #endif /* #else #ifdef RCU_READ_NESTABLE */ | |
108 | ||
1a43bbd8 MD |
109 | #ifdef TORTURE_QSBR |
110 | #define mark_rcu_quiescent_state rcu_quiescent_state | |
111 | #define put_thread_offline rcu_thread_offline | |
112 | #define put_thread_online rcu_thread_online | |
113 | #endif | |
114 | ||
8a953620 MD |
115 | #ifndef mark_rcu_quiescent_state |
116 | #define mark_rcu_quiescent_state() do ; while (0) | |
117 | #endif /* #ifdef mark_rcu_quiescent_state */ | |
118 | ||
119 | #ifndef put_thread_offline | |
120 | #define put_thread_offline() do ; while (0) | |
121 | #define put_thread_online() do ; while (0) | |
122 | #define put_thread_online_delay() do ; while (0) | |
123 | #else /* #ifndef put_thread_offline */ | |
124 | #define put_thread_online_delay() synchronize_rcu() | |
125 | #endif /* #else #ifndef put_thread_offline */ | |
126 | ||
127 | /* | |
128 | * Performance test. | |
129 | */ | |
130 | ||
131 | void *rcu_read_perf_test(void *arg) | |
132 | { | |
133 | int i; | |
134 | int me = (long)arg; | |
8a953620 MD |
135 | long long n_reads_local = 0; |
136 | ||
121a5d44 | 137 | rcu_register_thread(); |
8a953620 | 138 | run_on(me); |
ec4e58a3 | 139 | uatomic_inc(&nthreadsrunning); |
3bf02b5b | 140 | put_thread_offline(); |
8a953620 | 141 | while (goflag == GOFLAG_INIT) |
775aff2e | 142 | (void) poll(NULL, 0, 1); |
3bf02b5b | 143 | put_thread_online(); |
8a953620 MD |
144 | while (goflag == GOFLAG_RUN) { |
145 | for (i = 0; i < RCU_READ_RUN; i++) { | |
146 | rcu_read_lock(); | |
147 | /* rcu_read_lock_nest(); */ | |
148 | /* rcu_read_unlock_nest(); */ | |
149 | rcu_read_unlock(); | |
150 | } | |
151 | n_reads_local += RCU_READ_RUN; | |
152 | mark_rcu_quiescent_state(); | |
153 | } | |
154 | __get_thread_var(n_reads_pt) += n_reads_local; | |
155 | put_thread_offline(); | |
121a5d44 | 156 | rcu_unregister_thread(); |
8a953620 MD |
157 | |
158 | return (NULL); | |
159 | } | |
160 | ||
161 | void *rcu_update_perf_test(void *arg) | |
162 | { | |
163 | long long n_updates_local = 0; | |
164 | ||
26b5a74b | 165 | if (callrcu_type == CALLRCU_PERTHREAD) { |
b57aee66 PM |
166 | struct call_rcu_data *crdp; |
167 | ||
c1d2c60b | 168 | crdp = create_call_rcu_data(0, -1); |
b57aee66 | 169 | if (crdp != NULL) { |
26b5a74b | 170 | diag("Successfully using per-thread call_rcu() worker."); |
b57aee66 PM |
171 | set_thread_call_rcu_data(crdp); |
172 | } | |
173 | } | |
ec4e58a3 | 174 | uatomic_inc(&nthreadsrunning); |
8a953620 | 175 | while (goflag == GOFLAG_INIT) |
775aff2e | 176 | (void) poll(NULL, 0, 1); |
8a953620 MD |
177 | while (goflag == GOFLAG_RUN) { |
178 | synchronize_rcu(); | |
179 | n_updates_local++; | |
180 | } | |
181 | __get_thread_var(n_updates_pt) += n_updates_local; | |
26b5a74b MD |
182 | if (callrcu_type == CALLRCU_PERTHREAD) { |
183 | struct call_rcu_data *crdp; | |
184 | ||
185 | crdp = get_thread_call_rcu_data(); | |
186 | set_thread_call_rcu_data(NULL); | |
187 | call_rcu_data_free(crdp); | |
188 | } | |
b0b31506 | 189 | return NULL; |
8a953620 MD |
190 | } |
191 | ||
192 | void perftestinit(void) | |
193 | { | |
194 | init_per_thread(n_reads_pt, 0LL); | |
195 | init_per_thread(n_updates_pt, 0LL); | |
ec4e58a3 | 196 | uatomic_set(&nthreadsrunning, 0); |
8a953620 MD |
197 | } |
198 | ||
ad460058 | 199 | int perftestrun(int nthreads, int nreaders, int nupdaters) |
8a953620 MD |
200 | { |
201 | int t; | |
202 | int duration = 1; | |
203 | ||
5481ddb3 | 204 | cmm_smp_mb(); |
ec4e58a3 | 205 | while (uatomic_read(&nthreadsrunning) < nthreads) |
775aff2e | 206 | (void) poll(NULL, 0, 1); |
8a953620 | 207 | goflag = GOFLAG_RUN; |
5481ddb3 | 208 | cmm_smp_mb(); |
8a953620 | 209 | sleep(duration); |
5481ddb3 | 210 | cmm_smp_mb(); |
8a953620 | 211 | goflag = GOFLAG_STOP; |
5481ddb3 | 212 | cmm_smp_mb(); |
8a953620 MD |
213 | wait_all_threads(); |
214 | for_each_thread(t) { | |
215 | n_reads += per_thread(n_reads_pt, t); | |
216 | n_updates += per_thread(n_updates_pt, t); | |
217 | } | |
ad460058 | 218 | diag("n_reads: %lld n_updates: %ld nreaders: %d nupdaters: %d duration: %d", |
8a953620 | 219 | n_reads, n_updates, nreaders, nupdaters, duration); |
ad460058 | 220 | diag("ns/read: %g ns/update: %g", |
8a953620 MD |
221 | ((duration * 1000*1000*1000.*(double)nreaders) / |
222 | (double)n_reads), | |
223 | ((duration * 1000*1000*1000.*(double)nupdaters) / | |
224 | (double)n_updates)); | |
7106ddf8 | 225 | if (get_cpu_call_rcu_data(0)) { |
ad460058 | 226 | diag("Deallocating per-CPU call_rcu threads.\n"); |
7106ddf8 PM |
227 | free_all_cpu_call_rcu_data(); |
228 | } | |
ad460058 | 229 | return 0; |
8a953620 MD |
230 | } |
231 | ||
ad460058 | 232 | int perftest(int nreaders, int cpustride) |
8a953620 MD |
233 | { |
234 | int i; | |
235 | long arg; | |
236 | ||
237 | perftestinit(); | |
238 | for (i = 0; i < nreaders; i++) { | |
239 | arg = (long)(i * cpustride); | |
240 | create_thread(rcu_read_perf_test, (void *)arg); | |
241 | } | |
242 | arg = (long)(i * cpustride); | |
243 | create_thread(rcu_update_perf_test, (void *)arg); | |
ad460058 | 244 | return perftestrun(i + 1, nreaders, 1); |
8a953620 MD |
245 | } |
246 | ||
ad460058 | 247 | int rperftest(int nreaders, int cpustride) |
8a953620 MD |
248 | { |
249 | int i; | |
250 | long arg; | |
251 | ||
252 | perftestinit(); | |
253 | init_per_thread(n_reads_pt, 0LL); | |
254 | for (i = 0; i < nreaders; i++) { | |
255 | arg = (long)(i * cpustride); | |
256 | create_thread(rcu_read_perf_test, (void *)arg); | |
257 | } | |
ad460058 | 258 | return perftestrun(i, nreaders, 0); |
8a953620 MD |
259 | } |
260 | ||
ad460058 | 261 | int uperftest(int nupdaters, int cpustride) |
8a953620 MD |
262 | { |
263 | int i; | |
264 | long arg; | |
265 | ||
266 | perftestinit(); | |
267 | init_per_thread(n_reads_pt, 0LL); | |
268 | for (i = 0; i < nupdaters; i++) { | |
269 | arg = (long)(i * cpustride); | |
270 | create_thread(rcu_update_perf_test, (void *)arg); | |
271 | } | |
ad460058 | 272 | return perftestrun(i, 0, nupdaters); |
8a953620 MD |
273 | } |
274 | ||
275 | /* | |
276 | * Stress test. | |
277 | */ | |
278 | ||
279 | #define RCU_STRESS_PIPE_LEN 10 | |
280 | ||
281 | struct rcu_stress { | |
282 | int pipe_count; | |
283 | int mbtest; | |
284 | }; | |
285 | ||
b0b31506 | 286 | struct rcu_stress rcu_stress_array[RCU_STRESS_PIPE_LEN] = { { 0 } }; |
8a953620 MD |
287 | struct rcu_stress *rcu_stress_current; |
288 | int rcu_stress_idx = 0; | |
289 | ||
290 | int n_mberror = 0; | |
291 | DEFINE_PER_THREAD(long long [RCU_STRESS_PIPE_LEN + 1], rcu_stress_count); | |
292 | ||
293 | int garbage = 0; | |
294 | ||
295 | void *rcu_read_stress_test(void *arg) | |
296 | { | |
297 | int i; | |
298 | int itercnt = 0; | |
299 | struct rcu_stress *p; | |
300 | int pc; | |
301 | ||
121a5d44 | 302 | rcu_register_thread(); |
3bf02b5b | 303 | put_thread_offline(); |
8a953620 | 304 | while (goflag == GOFLAG_INIT) |
775aff2e | 305 | (void) poll(NULL, 0, 1); |
3bf02b5b | 306 | put_thread_online(); |
8a953620 MD |
307 | while (goflag == GOFLAG_RUN) { |
308 | rcu_read_lock(); | |
309 | p = rcu_dereference(rcu_stress_current); | |
310 | if (p->mbtest == 0) | |
311 | n_mberror++; | |
312 | rcu_read_lock_nest(); | |
313 | for (i = 0; i < 100; i++) | |
314 | garbage++; | |
315 | rcu_read_unlock_nest(); | |
316 | pc = p->pipe_count; | |
317 | rcu_read_unlock(); | |
318 | if ((pc > RCU_STRESS_PIPE_LEN) || (pc < 0)) | |
319 | pc = RCU_STRESS_PIPE_LEN; | |
320 | __get_thread_var(rcu_stress_count)[pc]++; | |
321 | __get_thread_var(n_reads_pt)++; | |
322 | mark_rcu_quiescent_state(); | |
323 | if ((++itercnt % 0x1000) == 0) { | |
324 | put_thread_offline(); | |
325 | put_thread_online_delay(); | |
326 | put_thread_online(); | |
327 | } | |
328 | } | |
329 | put_thread_offline(); | |
121a5d44 | 330 | rcu_unregister_thread(); |
8a953620 MD |
331 | |
332 | return (NULL); | |
333 | } | |
334 | ||
b57aee66 PM |
335 | static pthread_mutex_t call_rcu_test_mutex = PTHREAD_MUTEX_INITIALIZER; |
336 | static pthread_cond_t call_rcu_test_cond = PTHREAD_COND_INITIALIZER; | |
337 | ||
338 | void rcu_update_stress_test_rcu(struct rcu_head *head) | |
339 | { | |
ad460058 MD |
340 | int ret; |
341 | ||
342 | ret = pthread_mutex_lock(&call_rcu_test_mutex); | |
343 | if (ret) { | |
344 | errno = ret; | |
345 | diag("pthread_mutex_lock: %s", | |
346 | strerror(errno)); | |
347 | abort(); | |
b57aee66 | 348 | } |
ad460058 MD |
349 | ret = pthread_cond_signal(&call_rcu_test_cond); |
350 | if (ret) { | |
351 | errno = ret; | |
352 | diag("pthread_cond_signal: %s", | |
353 | strerror(errno)); | |
354 | abort(); | |
b57aee66 | 355 | } |
ad460058 MD |
356 | ret = pthread_mutex_unlock(&call_rcu_test_mutex); |
357 | if (ret) { | |
358 | errno = ret; | |
359 | diag("pthread_mutex_unlock: %s", | |
360 | strerror(errno)); | |
361 | abort(); | |
b57aee66 PM |
362 | } |
363 | } | |
364 | ||
8a953620 MD |
365 | void *rcu_update_stress_test(void *arg) |
366 | { | |
367 | int i; | |
368 | struct rcu_stress *p; | |
b57aee66 | 369 | struct rcu_head rh; |
8a953620 | 370 | |
fcb81146 MD |
371 | rcu_register_thread(); |
372 | rcu_thread_offline(); | |
373 | ||
8a953620 | 374 | while (goflag == GOFLAG_INIT) |
775aff2e | 375 | (void) poll(NULL, 0, 1); |
8a953620 MD |
376 | while (goflag == GOFLAG_RUN) { |
377 | i = rcu_stress_idx + 1; | |
378 | if (i >= RCU_STRESS_PIPE_LEN) | |
379 | i = 0; | |
380 | p = &rcu_stress_array[i]; | |
381 | p->mbtest = 0; | |
5481ddb3 | 382 | cmm_smp_mb(); |
8a953620 MD |
383 | p->pipe_count = 0; |
384 | p->mbtest = 1; | |
385 | rcu_assign_pointer(rcu_stress_current, p); | |
386 | rcu_stress_idx = i; | |
387 | for (i = 0; i < RCU_STRESS_PIPE_LEN; i++) | |
388 | if (i != rcu_stress_idx) | |
389 | rcu_stress_array[i].pipe_count++; | |
b57aee66 PM |
390 | if (n_updates & 0x1) |
391 | synchronize_rcu(); | |
392 | else { | |
ad460058 MD |
393 | int ret; |
394 | ||
395 | ret = pthread_mutex_lock(&call_rcu_test_mutex); | |
396 | if (ret) { | |
397 | errno = ret; | |
398 | diag("pthread_mutex_lock: %s", | |
399 | strerror(errno)); | |
400 | abort(); | |
b57aee66 PM |
401 | } |
402 | call_rcu(&rh, rcu_update_stress_test_rcu); | |
ad460058 MD |
403 | ret = pthread_cond_wait(&call_rcu_test_cond, |
404 | &call_rcu_test_mutex); | |
405 | if (ret) { | |
406 | errno = ret; | |
407 | diag("pthread_cond_signal: %s", | |
408 | strerror(errno)); | |
409 | abort(); | |
b57aee66 | 410 | } |
ad460058 MD |
411 | ret = pthread_mutex_unlock(&call_rcu_test_mutex); |
412 | if (ret) { | |
413 | errno = ret; | |
414 | diag("pthread_mutex_unlock: %s", | |
415 | strerror(errno)); | |
416 | abort(); | |
b57aee66 PM |
417 | } |
418 | } | |
8a953620 MD |
419 | n_updates++; |
420 | } | |
fcb81146 MD |
421 | |
422 | rcu_thread_online(); | |
423 | rcu_unregister_thread(); | |
b0b31506 | 424 | return NULL; |
8a953620 MD |
425 | } |
426 | ||
427 | void *rcu_fake_update_stress_test(void *arg) | |
428 | { | |
26b5a74b | 429 | if (callrcu_type == CALLRCU_PERTHREAD) { |
b57aee66 PM |
430 | struct call_rcu_data *crdp; |
431 | ||
c1d2c60b | 432 | crdp = create_call_rcu_data(0, -1); |
b57aee66 | 433 | if (crdp != NULL) { |
26b5a74b | 434 | diag("Successfully using per-thread call_rcu() worker."); |
b57aee66 PM |
435 | set_thread_call_rcu_data(crdp); |
436 | } | |
437 | } | |
8a953620 | 438 | while (goflag == GOFLAG_INIT) |
775aff2e | 439 | (void) poll(NULL, 0, 1); |
8a953620 MD |
440 | while (goflag == GOFLAG_RUN) { |
441 | synchronize_rcu(); | |
775aff2e | 442 | (void) poll(NULL, 0, 1); |
8a953620 | 443 | } |
26b5a74b MD |
444 | if (callrcu_type == CALLRCU_PERTHREAD) { |
445 | struct call_rcu_data *crdp; | |
446 | ||
447 | crdp = get_thread_call_rcu_data(); | |
448 | set_thread_call_rcu_data(NULL); | |
449 | call_rcu_data_free(crdp); | |
450 | } | |
b0b31506 | 451 | return NULL; |
8a953620 MD |
452 | } |
453 | ||
ad460058 | 454 | int stresstest(int nreaders) |
8a953620 MD |
455 | { |
456 | int i; | |
457 | int t; | |
458 | long long *p; | |
459 | long long sum; | |
460 | ||
461 | init_per_thread(n_reads_pt, 0LL); | |
462 | for_each_thread(t) { | |
463 | p = &per_thread(rcu_stress_count,t)[0]; | |
464 | for (i = 0; i <= RCU_STRESS_PIPE_LEN; i++) | |
465 | p[i] = 0LL; | |
466 | } | |
467 | rcu_stress_current = &rcu_stress_array[0]; | |
468 | rcu_stress_current->pipe_count = 0; | |
469 | rcu_stress_current->mbtest = 1; | |
470 | for (i = 0; i < nreaders; i++) | |
471 | create_thread(rcu_read_stress_test, NULL); | |
472 | create_thread(rcu_update_stress_test, NULL); | |
473 | for (i = 0; i < 5; i++) | |
474 | create_thread(rcu_fake_update_stress_test, NULL); | |
5481ddb3 | 475 | cmm_smp_mb(); |
8a953620 | 476 | goflag = GOFLAG_RUN; |
5481ddb3 | 477 | cmm_smp_mb(); |
8a953620 | 478 | sleep(10); |
5481ddb3 | 479 | cmm_smp_mb(); |
8a953620 | 480 | goflag = GOFLAG_STOP; |
5481ddb3 | 481 | cmm_smp_mb(); |
8a953620 MD |
482 | wait_all_threads(); |
483 | for_each_thread(t) | |
484 | n_reads += per_thread(n_reads_pt, t); | |
ad460058 | 485 | diag("n_reads: %lld n_updates: %ld n_mberror: %d", |
8a953620 | 486 | n_reads, n_updates, n_mberror); |
ad460058 MD |
487 | rdiag_start(); |
488 | rdiag("rcu_stress_count:"); | |
8a953620 MD |
489 | for (i = 0; i <= RCU_STRESS_PIPE_LEN; i++) { |
490 | sum = 0LL; | |
491 | for_each_thread(t) { | |
492 | sum += per_thread(rcu_stress_count, t)[i]; | |
493 | } | |
ad460058 | 494 | rdiag(" %lld", sum); |
8a953620 | 495 | } |
ad460058 | 496 | rdiag_end(); |
7106ddf8 | 497 | if (get_cpu_call_rcu_data(0)) { |
ad460058 | 498 | diag("Deallocating per-CPU call_rcu threads."); |
7106ddf8 PM |
499 | free_all_cpu_call_rcu_data(); |
500 | } | |
ad460058 MD |
501 | if (!n_mberror) |
502 | return 0; | |
503 | else | |
504 | return -1; | |
8a953620 MD |
505 | } |
506 | ||
507 | /* | |
508 | * Mainprogram. | |
509 | */ | |
510 | ||
511 | void usage(int argc, char *argv[]) | |
512 | { | |
26b5a74b | 513 | diag("Usage: %s nreaders [ perf | rperf | uperf | stress ] [ stride ] [ callrcu_global | callrcu_percpu | callrcu_perthread ]\n", argv[0]); |
8a953620 MD |
514 | exit(-1); |
515 | } | |
516 | ||
517 | int main(int argc, char *argv[]) | |
518 | { | |
519 | int nreaders = 1; | |
520 | int cpustride = 1; | |
521 | ||
ad460058 MD |
522 | plan_tests(NR_TESTS); |
523 | ||
8a953620 MD |
524 | smp_init(); |
525 | //rcu_init(); | |
26b5a74b MD |
526 | if (argc > 4) { |
527 | const char *callrcu_str = argv[4];; | |
528 | ||
529 | if (strcmp(callrcu_str, "callrcu_global") == 0) { | |
530 | callrcu_type = CALLRCU_GLOBAL; | |
531 | } else if (strcmp(callrcu_str, "callrcu_percpu") == 0) { | |
532 | callrcu_type = CALLRCU_PERCPU; | |
533 | } else if (strcmp(callrcu_str, "callrcu_perthread") == 0) { | |
534 | callrcu_type = CALLRCU_PERTHREAD; | |
535 | } else { | |
536 | usage(argc, argv); | |
537 | goto end; | |
538 | } | |
539 | } | |
929cfaff | 540 | |
26b5a74b MD |
541 | switch (callrcu_type) { |
542 | case CALLRCU_GLOBAL: | |
543 | diag("Using global per-process call_rcu thread."); | |
544 | break; | |
545 | case CALLRCU_PERCPU: | |
546 | diag("Using per-CPU call_rcu threads."); | |
b57aee66 | 547 | if (create_all_cpu_call_rcu_data(0)) |
ad460058 MD |
548 | diag("create_all_cpu_call_rcu_data: %s", |
549 | strerror(errno)); | |
26b5a74b MD |
550 | break; |
551 | case CALLRCU_PERTHREAD: | |
552 | diag("Using per-thread call_rcu() worker."); | |
553 | break; | |
554 | default: | |
555 | abort(); | |
b57aee66 | 556 | } |
8a953620 | 557 | |
9b171f46 MD |
558 | #ifdef DEBUG_YIELD |
559 | yield_active |= YIELD_READ; | |
560 | yield_active |= YIELD_WRITE; | |
561 | #endif | |
562 | ||
8a953620 | 563 | if (argc > 1) { |
26b5a74b MD |
564 | if (strcmp(argv[1], "-h") == 0 |
565 | || strcmp(argv[1], "--help") == 0) { | |
566 | usage(argc, argv); | |
567 | goto end; | |
568 | } | |
8a953620 | 569 | nreaders = strtoul(argv[1], NULL, 0); |
ad460058 MD |
570 | if (argc == 2) { |
571 | ok(!perftest(nreaders, cpustride), | |
572 | "perftest readers: %d, stride: %d", | |
573 | nreaders, cpustride); | |
574 | goto end; | |
575 | } | |
8a953620 MD |
576 | if (argc > 3) |
577 | cpustride = strtoul(argv[3], NULL, 0); | |
578 | if (strcmp(argv[2], "perf") == 0) | |
ad460058 MD |
579 | ok(!perftest(nreaders, cpustride), |
580 | "perftest readers: %d, stride: %d", | |
581 | nreaders, cpustride); | |
8a953620 | 582 | else if (strcmp(argv[2], "rperf") == 0) |
ad460058 MD |
583 | ok(!rperftest(nreaders, cpustride), |
584 | "rperftest readers: %d, stride: %d", | |
585 | nreaders, cpustride); | |
8a953620 | 586 | else if (strcmp(argv[2], "uperf") == 0) |
ad460058 MD |
587 | ok(!uperftest(nreaders, cpustride), |
588 | "uperftest readers: %d, stride: %d", | |
589 | nreaders, cpustride); | |
8a953620 | 590 | else if (strcmp(argv[2], "stress") == 0) |
ad460058 MD |
591 | ok(!stresstest(nreaders), |
592 | "stresstest readers: %d, stride: %d", | |
593 | nreaders, cpustride); | |
594 | else | |
595 | usage(argc, argv); | |
596 | } else { | |
26b5a74b | 597 | usage(argc, argv); |
8a953620 | 598 | } |
ad460058 MD |
599 | end: |
600 | return exit_status(); | |
8a953620 | 601 | } |