X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=tests%2Frcutorture.h;h=181547a0097dbaab406eceba1d19177da6fb8373;hb=1a43bbd8f98d5dcf1404b16ab4dc4bfddd65c3b0;hp=7d4948e59626b486524af30fa4b0dcc5f9bfc02b;hpb=6ee91d83a7a5b33160158f57f94dd75e6d0a67ed;p=urcu.git

diff --git a/tests/rcutorture.h b/tests/rcutorture.h
index 7d4948e..181547a 100644
--- a/tests/rcutorture.h
+++ b/tests/rcutorture.h
@@ -92,6 +92,12 @@ int goflag __attribute__((__aligned__(CACHE_LINE_SIZE))) = GOFLAG_INIT;
 #define rcu_read_unlock_nest()
 #endif /* #else #ifdef RCU_READ_NESTABLE */
 
+#ifdef TORTURE_QSBR
+#define mark_rcu_quiescent_state	rcu_quiescent_state
+#define put_thread_offline		rcu_thread_offline
+#define put_thread_online		rcu_thread_online
+#endif
+
 #ifndef mark_rcu_quiescent_state
 #define mark_rcu_quiescent_state() do ; while (0)
 #endif /* #ifdef mark_rcu_quiescent_state */
@@ -116,7 +122,7 @@ void *rcu_read_perf_test(void *arg)
 
 	rcu_register_thread();
 	run_on(me);
-	atomic_inc(&nthreadsrunning);
+	uatomic_inc(&nthreadsrunning);
 	while (goflag == GOFLAG_INIT)
 		poll(NULL, 0, 1);
 	mark_rcu_quiescent_state();
@@ -141,7 +147,7 @@ void *rcu_update_perf_test(void *arg)
 {
 	long long n_updates_local = 0;
 
-	atomic_inc(&nthreadsrunning);
+	uatomic_inc(&nthreadsrunning);
 	while (goflag == GOFLAG_INIT)
 		poll(NULL, 0, 1);
 	while (goflag == GOFLAG_RUN) {
@@ -156,7 +162,7 @@ void perftestinit(void)
 {
 	init_per_thread(n_reads_pt, 0LL);
 	init_per_thread(n_updates_pt, 0LL);
-	atomic_set(&nthreadsrunning, 0);
+	uatomic_set(&nthreadsrunning, 0);
 }
 
 void perftestrun(int nthreads, int nreaders, int nupdaters)
@@ -165,7 +171,7 @@ void perftestrun(int nthreads, int nreaders, int nupdaters)
 	int duration = 1;
 
 	smp_mb();
-	while (atomic_read(&nthreadsrunning) < nthreads)
+	while (uatomic_read(&nthreadsrunning) < nthreads)
 		poll(NULL, 0, 1);
 	goflag = GOFLAG_RUN;
 	smp_mb();