xref: /linux-6.15/kernel/rcu/refscale.c (revision e9b800db)
11fbeb3a8SPaul E. McKenney // SPDX-License-Identifier: GPL-2.0+
21fbeb3a8SPaul E. McKenney //
31fbeb3a8SPaul E. McKenney // Scalability test comparing RCU vs other mechanisms
41fbeb3a8SPaul E. McKenney // for acquiring references on objects.
51fbeb3a8SPaul E. McKenney //
61fbeb3a8SPaul E. McKenney // Copyright (C) Google, 2020.
71fbeb3a8SPaul E. McKenney //
81fbeb3a8SPaul E. McKenney // Author: Joel Fernandes <[email protected]>
91fbeb3a8SPaul E. McKenney 
101fbeb3a8SPaul E. McKenney #define pr_fmt(fmt) fmt
111fbeb3a8SPaul E. McKenney 
121fbeb3a8SPaul E. McKenney #include <linux/atomic.h>
131fbeb3a8SPaul E. McKenney #include <linux/bitops.h>
141fbeb3a8SPaul E. McKenney #include <linux/completion.h>
151fbeb3a8SPaul E. McKenney #include <linux/cpu.h>
161fbeb3a8SPaul E. McKenney #include <linux/delay.h>
171fbeb3a8SPaul E. McKenney #include <linux/err.h>
181fbeb3a8SPaul E. McKenney #include <linux/init.h>
191fbeb3a8SPaul E. McKenney #include <linux/interrupt.h>
201fbeb3a8SPaul E. McKenney #include <linux/kthread.h>
211fbeb3a8SPaul E. McKenney #include <linux/kernel.h>
221fbeb3a8SPaul E. McKenney #include <linux/mm.h>
231fbeb3a8SPaul E. McKenney #include <linux/module.h>
241fbeb3a8SPaul E. McKenney #include <linux/moduleparam.h>
251fbeb3a8SPaul E. McKenney #include <linux/notifier.h>
261fbeb3a8SPaul E. McKenney #include <linux/percpu.h>
271fbeb3a8SPaul E. McKenney #include <linux/rcupdate.h>
281fbeb3a8SPaul E. McKenney #include <linux/rcupdate_trace.h>
291fbeb3a8SPaul E. McKenney #include <linux/reboot.h>
301fbeb3a8SPaul E. McKenney #include <linux/sched.h>
311fbeb3a8SPaul E. McKenney #include <linux/spinlock.h>
321fbeb3a8SPaul E. McKenney #include <linux/smp.h>
331fbeb3a8SPaul E. McKenney #include <linux/stat.h>
341fbeb3a8SPaul E. McKenney #include <linux/srcu.h>
351fbeb3a8SPaul E. McKenney #include <linux/slab.h>
361fbeb3a8SPaul E. McKenney #include <linux/torture.h>
371fbeb3a8SPaul E. McKenney #include <linux/types.h>
381fbeb3a8SPaul E. McKenney 
391fbeb3a8SPaul E. McKenney #include "rcu.h"
401fbeb3a8SPaul E. McKenney 
411fbeb3a8SPaul E. McKenney #define SCALE_FLAG "-ref-scale: "
421fbeb3a8SPaul E. McKenney 
431fbeb3a8SPaul E. McKenney #define SCALEOUT(s, x...) \
441fbeb3a8SPaul E. McKenney 	pr_alert("%s" SCALE_FLAG s, scale_type, ## x)
451fbeb3a8SPaul E. McKenney 
461fbeb3a8SPaul E. McKenney #define VERBOSE_SCALEOUT(s, x...) \
471fbeb3a8SPaul E. McKenney 	do { if (verbose) pr_alert("%s" SCALE_FLAG s, scale_type, ## x); } while (0)
481fbeb3a8SPaul E. McKenney 
49e76506f0SPaul E. McKenney static atomic_t verbose_batch_ctr;
50e76506f0SPaul E. McKenney 
51e76506f0SPaul E. McKenney #define VERBOSE_SCALEOUT_BATCH(s, x...)							\
52e76506f0SPaul E. McKenney do {											\
53e76506f0SPaul E. McKenney 	if (verbose &&									\
54e76506f0SPaul E. McKenney 	    (verbose_batched <= 0 ||							\
55414c116eSPaul E. McKenney 	     !(atomic_inc_return(&verbose_batch_ctr) % verbose_batched))) {		\
56414c116eSPaul E. McKenney 		schedule_timeout_uninterruptible(1);					\
57e76506f0SPaul E. McKenney 		pr_alert("%s" SCALE_FLAG s, scale_type, ## x);				\
58414c116eSPaul E. McKenney 	}										\
59e76506f0SPaul E. McKenney } while (0)
60e76506f0SPaul E. McKenney 
611fbeb3a8SPaul E. McKenney #define VERBOSE_SCALEOUT_ERRSTRING(s, x...) \
621fbeb3a8SPaul E. McKenney 	do { if (verbose) pr_alert("%s" SCALE_FLAG "!!! " s, scale_type, ## x); } while (0)
631fbeb3a8SPaul E. McKenney 
641fbeb3a8SPaul E. McKenney MODULE_LICENSE("GPL");
651fbeb3a8SPaul E. McKenney MODULE_AUTHOR("Joel Fernandes (Google) <[email protected]>");
661fbeb3a8SPaul E. McKenney 
671fbeb3a8SPaul E. McKenney static char *scale_type = "rcu";
681fbeb3a8SPaul E. McKenney module_param(scale_type, charp, 0444);
691fbeb3a8SPaul E. McKenney MODULE_PARM_DESC(scale_type, "Type of test (rcu, srcu, refcnt, rwsem, rwlock.");
701fbeb3a8SPaul E. McKenney 
711fbeb3a8SPaul E. McKenney torture_param(int, verbose, 0, "Enable verbose debugging printk()s");
72e76506f0SPaul E. McKenney torture_param(int, verbose_batched, 0, "Batch verbose debugging printk()s");
731fbeb3a8SPaul E. McKenney 
741fbeb3a8SPaul E. McKenney // Wait until there are multiple CPUs before starting test.
751fbeb3a8SPaul E. McKenney torture_param(int, holdoff, IS_BUILTIN(CONFIG_RCU_REF_SCALE_TEST) ? 10 : 0,
761fbeb3a8SPaul E. McKenney 	      "Holdoff time before test start (s)");
771fbeb3a8SPaul E. McKenney // Number of loops per experiment, all readers execute operations concurrently.
781fbeb3a8SPaul E. McKenney torture_param(long, loops, 10000, "Number of loops per experiment.");
791fbeb3a8SPaul E. McKenney // Number of readers, with -1 defaulting to about 75% of the CPUs.
801fbeb3a8SPaul E. McKenney torture_param(int, nreaders, -1, "Number of readers, -1 for 75% of CPUs.");
811fbeb3a8SPaul E. McKenney // Number of runs.
821fbeb3a8SPaul E. McKenney torture_param(int, nruns, 30, "Number of experiments to run.");
831fbeb3a8SPaul E. McKenney // Reader delay in nanoseconds, 0 for no delay.
841fbeb3a8SPaul E. McKenney torture_param(int, readdelay, 0, "Read-side delay in nanoseconds.");
851fbeb3a8SPaul E. McKenney 
861fbeb3a8SPaul E. McKenney #ifdef MODULE
871fbeb3a8SPaul E. McKenney # define REFSCALE_SHUTDOWN 0
881fbeb3a8SPaul E. McKenney #else
891fbeb3a8SPaul E. McKenney # define REFSCALE_SHUTDOWN 1
901fbeb3a8SPaul E. McKenney #endif
911fbeb3a8SPaul E. McKenney 
921fbeb3a8SPaul E. McKenney torture_param(bool, shutdown, REFSCALE_SHUTDOWN,
931fbeb3a8SPaul E. McKenney 	      "Shutdown at end of scalability tests.");
941fbeb3a8SPaul E. McKenney 
951fbeb3a8SPaul E. McKenney struct reader_task {
961fbeb3a8SPaul E. McKenney 	struct task_struct *task;
971fbeb3a8SPaul E. McKenney 	int start_reader;
981fbeb3a8SPaul E. McKenney 	wait_queue_head_t wq;
991fbeb3a8SPaul E. McKenney 	u64 last_duration_ns;
1001fbeb3a8SPaul E. McKenney };
1011fbeb3a8SPaul E. McKenney 
1021fbeb3a8SPaul E. McKenney static struct task_struct *shutdown_task;
1031fbeb3a8SPaul E. McKenney static wait_queue_head_t shutdown_wq;
1041fbeb3a8SPaul E. McKenney 
1051fbeb3a8SPaul E. McKenney static struct task_struct *main_task;
1061fbeb3a8SPaul E. McKenney static wait_queue_head_t main_wq;
1071fbeb3a8SPaul E. McKenney static int shutdown_start;
1081fbeb3a8SPaul E. McKenney 
1091fbeb3a8SPaul E. McKenney static struct reader_task *reader_tasks;
1101fbeb3a8SPaul E. McKenney 
1111fbeb3a8SPaul E. McKenney // Number of readers that are part of the current experiment.
1121fbeb3a8SPaul E. McKenney static atomic_t nreaders_exp;
1131fbeb3a8SPaul E. McKenney 
1141fbeb3a8SPaul E. McKenney // Use to wait for all threads to start.
1151fbeb3a8SPaul E. McKenney static atomic_t n_init;
1161fbeb3a8SPaul E. McKenney static atomic_t n_started;
1171fbeb3a8SPaul E. McKenney static atomic_t n_warmedup;
1181fbeb3a8SPaul E. McKenney static atomic_t n_cooleddown;
1191fbeb3a8SPaul E. McKenney 
1201fbeb3a8SPaul E. McKenney // Track which experiment is currently running.
1211fbeb3a8SPaul E. McKenney static int exp_idx;
1221fbeb3a8SPaul E. McKenney 
1231fbeb3a8SPaul E. McKenney // Operations vector for selecting different types of tests.
1241fbeb3a8SPaul E. McKenney struct ref_scale_ops {
1251fbeb3a8SPaul E. McKenney 	void (*init)(void);
1261fbeb3a8SPaul E. McKenney 	void (*cleanup)(void);
1271fbeb3a8SPaul E. McKenney 	void (*readsection)(const int nloops);
1281fbeb3a8SPaul E. McKenney 	void (*delaysection)(const int nloops, const int udl, const int ndl);
1291fbeb3a8SPaul E. McKenney 	const char *name;
1301fbeb3a8SPaul E. McKenney };
1311fbeb3a8SPaul E. McKenney 
1321fbeb3a8SPaul E. McKenney static struct ref_scale_ops *cur_ops;
1331fbeb3a8SPaul E. McKenney 
1341fbeb3a8SPaul E. McKenney static void un_delay(const int udl, const int ndl)
1351fbeb3a8SPaul E. McKenney {
1361fbeb3a8SPaul E. McKenney 	if (udl)
1371fbeb3a8SPaul E. McKenney 		udelay(udl);
1381fbeb3a8SPaul E. McKenney 	if (ndl)
1391fbeb3a8SPaul E. McKenney 		ndelay(ndl);
1401fbeb3a8SPaul E. McKenney }
1411fbeb3a8SPaul E. McKenney 
1421fbeb3a8SPaul E. McKenney static void ref_rcu_read_section(const int nloops)
1431fbeb3a8SPaul E. McKenney {
1441fbeb3a8SPaul E. McKenney 	int i;
1451fbeb3a8SPaul E. McKenney 
1461fbeb3a8SPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
1471fbeb3a8SPaul E. McKenney 		rcu_read_lock();
1481fbeb3a8SPaul E. McKenney 		rcu_read_unlock();
1491fbeb3a8SPaul E. McKenney 	}
1501fbeb3a8SPaul E. McKenney }
1511fbeb3a8SPaul E. McKenney 
1521fbeb3a8SPaul E. McKenney static void ref_rcu_delay_section(const int nloops, const int udl, const int ndl)
1531fbeb3a8SPaul E. McKenney {
1541fbeb3a8SPaul E. McKenney 	int i;
1551fbeb3a8SPaul E. McKenney 
1561fbeb3a8SPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
1571fbeb3a8SPaul E. McKenney 		rcu_read_lock();
1581fbeb3a8SPaul E. McKenney 		un_delay(udl, ndl);
1591fbeb3a8SPaul E. McKenney 		rcu_read_unlock();
1601fbeb3a8SPaul E. McKenney 	}
1611fbeb3a8SPaul E. McKenney }
1621fbeb3a8SPaul E. McKenney 
1631fbeb3a8SPaul E. McKenney static void rcu_sync_scale_init(void)
1641fbeb3a8SPaul E. McKenney {
1651fbeb3a8SPaul E. McKenney }
1661fbeb3a8SPaul E. McKenney 
1671fbeb3a8SPaul E. McKenney static struct ref_scale_ops rcu_ops = {
1681fbeb3a8SPaul E. McKenney 	.init		= rcu_sync_scale_init,
1691fbeb3a8SPaul E. McKenney 	.readsection	= ref_rcu_read_section,
1701fbeb3a8SPaul E. McKenney 	.delaysection	= ref_rcu_delay_section,
1711fbeb3a8SPaul E. McKenney 	.name		= "rcu"
1721fbeb3a8SPaul E. McKenney };
1731fbeb3a8SPaul E. McKenney 
1741fbeb3a8SPaul E. McKenney // Definitions for SRCU ref scale testing.
1751fbeb3a8SPaul E. McKenney DEFINE_STATIC_SRCU(srcu_refctl_scale);
1761fbeb3a8SPaul E. McKenney static struct srcu_struct *srcu_ctlp = &srcu_refctl_scale;
1771fbeb3a8SPaul E. McKenney 
1781fbeb3a8SPaul E. McKenney static void srcu_ref_scale_read_section(const int nloops)
1791fbeb3a8SPaul E. McKenney {
1801fbeb3a8SPaul E. McKenney 	int i;
1811fbeb3a8SPaul E. McKenney 	int idx;
1821fbeb3a8SPaul E. McKenney 
1831fbeb3a8SPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
1841fbeb3a8SPaul E. McKenney 		idx = srcu_read_lock(srcu_ctlp);
1851fbeb3a8SPaul E. McKenney 		srcu_read_unlock(srcu_ctlp, idx);
1861fbeb3a8SPaul E. McKenney 	}
1871fbeb3a8SPaul E. McKenney }
1881fbeb3a8SPaul E. McKenney 
1891fbeb3a8SPaul E. McKenney static void srcu_ref_scale_delay_section(const int nloops, const int udl, const int ndl)
1901fbeb3a8SPaul E. McKenney {
1911fbeb3a8SPaul E. McKenney 	int i;
1921fbeb3a8SPaul E. McKenney 	int idx;
1931fbeb3a8SPaul E. McKenney 
1941fbeb3a8SPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
1951fbeb3a8SPaul E. McKenney 		idx = srcu_read_lock(srcu_ctlp);
1961fbeb3a8SPaul E. McKenney 		un_delay(udl, ndl);
1971fbeb3a8SPaul E. McKenney 		srcu_read_unlock(srcu_ctlp, idx);
1981fbeb3a8SPaul E. McKenney 	}
1991fbeb3a8SPaul E. McKenney }
2001fbeb3a8SPaul E. McKenney 
2011fbeb3a8SPaul E. McKenney static struct ref_scale_ops srcu_ops = {
2021fbeb3a8SPaul E. McKenney 	.init		= rcu_sync_scale_init,
2031fbeb3a8SPaul E. McKenney 	.readsection	= srcu_ref_scale_read_section,
2041fbeb3a8SPaul E. McKenney 	.delaysection	= srcu_ref_scale_delay_section,
2051fbeb3a8SPaul E. McKenney 	.name		= "srcu"
2061fbeb3a8SPaul E. McKenney };
2071fbeb3a8SPaul E. McKenney 
2081fbeb3a8SPaul E. McKenney // Definitions for RCU Tasks ref scale testing: Empty read markers.
2091fbeb3a8SPaul E. McKenney // These definitions also work for RCU Rude readers.
2101fbeb3a8SPaul E. McKenney static void rcu_tasks_ref_scale_read_section(const int nloops)
2111fbeb3a8SPaul E. McKenney {
2121fbeb3a8SPaul E. McKenney 	int i;
2131fbeb3a8SPaul E. McKenney 
2141fbeb3a8SPaul E. McKenney 	for (i = nloops; i >= 0; i--)
2151fbeb3a8SPaul E. McKenney 		continue;
2161fbeb3a8SPaul E. McKenney }
2171fbeb3a8SPaul E. McKenney 
2181fbeb3a8SPaul E. McKenney static void rcu_tasks_ref_scale_delay_section(const int nloops, const int udl, const int ndl)
2191fbeb3a8SPaul E. McKenney {
2201fbeb3a8SPaul E. McKenney 	int i;
2211fbeb3a8SPaul E. McKenney 
2221fbeb3a8SPaul E. McKenney 	for (i = nloops; i >= 0; i--)
2231fbeb3a8SPaul E. McKenney 		un_delay(udl, ndl);
2241fbeb3a8SPaul E. McKenney }
2251fbeb3a8SPaul E. McKenney 
2261fbeb3a8SPaul E. McKenney static struct ref_scale_ops rcu_tasks_ops = {
2271fbeb3a8SPaul E. McKenney 	.init		= rcu_sync_scale_init,
2281fbeb3a8SPaul E. McKenney 	.readsection	= rcu_tasks_ref_scale_read_section,
2291fbeb3a8SPaul E. McKenney 	.delaysection	= rcu_tasks_ref_scale_delay_section,
2301fbeb3a8SPaul E. McKenney 	.name		= "rcu-tasks"
2311fbeb3a8SPaul E. McKenney };
2321fbeb3a8SPaul E. McKenney 
2331fbeb3a8SPaul E. McKenney // Definitions for RCU Tasks Trace ref scale testing.
2341fbeb3a8SPaul E. McKenney static void rcu_trace_ref_scale_read_section(const int nloops)
2351fbeb3a8SPaul E. McKenney {
2361fbeb3a8SPaul E. McKenney 	int i;
2371fbeb3a8SPaul E. McKenney 
2381fbeb3a8SPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
2391fbeb3a8SPaul E. McKenney 		rcu_read_lock_trace();
2401fbeb3a8SPaul E. McKenney 		rcu_read_unlock_trace();
2411fbeb3a8SPaul E. McKenney 	}
2421fbeb3a8SPaul E. McKenney }
2431fbeb3a8SPaul E. McKenney 
2441fbeb3a8SPaul E. McKenney static void rcu_trace_ref_scale_delay_section(const int nloops, const int udl, const int ndl)
2451fbeb3a8SPaul E. McKenney {
2461fbeb3a8SPaul E. McKenney 	int i;
2471fbeb3a8SPaul E. McKenney 
2481fbeb3a8SPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
2491fbeb3a8SPaul E. McKenney 		rcu_read_lock_trace();
2501fbeb3a8SPaul E. McKenney 		un_delay(udl, ndl);
2511fbeb3a8SPaul E. McKenney 		rcu_read_unlock_trace();
2521fbeb3a8SPaul E. McKenney 	}
2531fbeb3a8SPaul E. McKenney }
2541fbeb3a8SPaul E. McKenney 
2551fbeb3a8SPaul E. McKenney static struct ref_scale_ops rcu_trace_ops = {
2561fbeb3a8SPaul E. McKenney 	.init		= rcu_sync_scale_init,
2571fbeb3a8SPaul E. McKenney 	.readsection	= rcu_trace_ref_scale_read_section,
2581fbeb3a8SPaul E. McKenney 	.delaysection	= rcu_trace_ref_scale_delay_section,
2591fbeb3a8SPaul E. McKenney 	.name		= "rcu-trace"
2601fbeb3a8SPaul E. McKenney };
2611fbeb3a8SPaul E. McKenney 
2621fbeb3a8SPaul E. McKenney // Definitions for reference count
2631fbeb3a8SPaul E. McKenney static atomic_t refcnt;
2641fbeb3a8SPaul E. McKenney 
2651fbeb3a8SPaul E. McKenney static void ref_refcnt_section(const int nloops)
2661fbeb3a8SPaul E. McKenney {
2671fbeb3a8SPaul E. McKenney 	int i;
2681fbeb3a8SPaul E. McKenney 
2691fbeb3a8SPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
2701fbeb3a8SPaul E. McKenney 		atomic_inc(&refcnt);
2711fbeb3a8SPaul E. McKenney 		atomic_dec(&refcnt);
2721fbeb3a8SPaul E. McKenney 	}
2731fbeb3a8SPaul E. McKenney }
2741fbeb3a8SPaul E. McKenney 
2751fbeb3a8SPaul E. McKenney static void ref_refcnt_delay_section(const int nloops, const int udl, const int ndl)
2761fbeb3a8SPaul E. McKenney {
2771fbeb3a8SPaul E. McKenney 	int i;
2781fbeb3a8SPaul E. McKenney 
2791fbeb3a8SPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
2801fbeb3a8SPaul E. McKenney 		atomic_inc(&refcnt);
2811fbeb3a8SPaul E. McKenney 		un_delay(udl, ndl);
2821fbeb3a8SPaul E. McKenney 		atomic_dec(&refcnt);
2831fbeb3a8SPaul E. McKenney 	}
2841fbeb3a8SPaul E. McKenney }
2851fbeb3a8SPaul E. McKenney 
2861fbeb3a8SPaul E. McKenney static struct ref_scale_ops refcnt_ops = {
2871fbeb3a8SPaul E. McKenney 	.init		= rcu_sync_scale_init,
2881fbeb3a8SPaul E. McKenney 	.readsection	= ref_refcnt_section,
2891fbeb3a8SPaul E. McKenney 	.delaysection	= ref_refcnt_delay_section,
2901fbeb3a8SPaul E. McKenney 	.name		= "refcnt"
2911fbeb3a8SPaul E. McKenney };
2921fbeb3a8SPaul E. McKenney 
2931fbeb3a8SPaul E. McKenney // Definitions for rwlock
2941fbeb3a8SPaul E. McKenney static rwlock_t test_rwlock;
2951fbeb3a8SPaul E. McKenney 
2961fbeb3a8SPaul E. McKenney static void ref_rwlock_init(void)
2971fbeb3a8SPaul E. McKenney {
2981fbeb3a8SPaul E. McKenney 	rwlock_init(&test_rwlock);
2991fbeb3a8SPaul E. McKenney }
3001fbeb3a8SPaul E. McKenney 
3011fbeb3a8SPaul E. McKenney static void ref_rwlock_section(const int nloops)
3021fbeb3a8SPaul E. McKenney {
3031fbeb3a8SPaul E. McKenney 	int i;
3041fbeb3a8SPaul E. McKenney 
3051fbeb3a8SPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
3061fbeb3a8SPaul E. McKenney 		read_lock(&test_rwlock);
3071fbeb3a8SPaul E. McKenney 		read_unlock(&test_rwlock);
3081fbeb3a8SPaul E. McKenney 	}
3091fbeb3a8SPaul E. McKenney }
3101fbeb3a8SPaul E. McKenney 
3111fbeb3a8SPaul E. McKenney static void ref_rwlock_delay_section(const int nloops, const int udl, const int ndl)
3121fbeb3a8SPaul E. McKenney {
3131fbeb3a8SPaul E. McKenney 	int i;
3141fbeb3a8SPaul E. McKenney 
3151fbeb3a8SPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
3161fbeb3a8SPaul E. McKenney 		read_lock(&test_rwlock);
3171fbeb3a8SPaul E. McKenney 		un_delay(udl, ndl);
3181fbeb3a8SPaul E. McKenney 		read_unlock(&test_rwlock);
3191fbeb3a8SPaul E. McKenney 	}
3201fbeb3a8SPaul E. McKenney }
3211fbeb3a8SPaul E. McKenney 
3221fbeb3a8SPaul E. McKenney static struct ref_scale_ops rwlock_ops = {
3231fbeb3a8SPaul E. McKenney 	.init		= ref_rwlock_init,
3241fbeb3a8SPaul E. McKenney 	.readsection	= ref_rwlock_section,
3251fbeb3a8SPaul E. McKenney 	.delaysection	= ref_rwlock_delay_section,
3261fbeb3a8SPaul E. McKenney 	.name		= "rwlock"
3271fbeb3a8SPaul E. McKenney };
3281fbeb3a8SPaul E. McKenney 
3291fbeb3a8SPaul E. McKenney // Definitions for rwsem
3301fbeb3a8SPaul E. McKenney static struct rw_semaphore test_rwsem;
3311fbeb3a8SPaul E. McKenney 
3321fbeb3a8SPaul E. McKenney static void ref_rwsem_init(void)
3331fbeb3a8SPaul E. McKenney {
3341fbeb3a8SPaul E. McKenney 	init_rwsem(&test_rwsem);
3351fbeb3a8SPaul E. McKenney }
3361fbeb3a8SPaul E. McKenney 
3371fbeb3a8SPaul E. McKenney static void ref_rwsem_section(const int nloops)
3381fbeb3a8SPaul E. McKenney {
3391fbeb3a8SPaul E. McKenney 	int i;
3401fbeb3a8SPaul E. McKenney 
3411fbeb3a8SPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
3421fbeb3a8SPaul E. McKenney 		down_read(&test_rwsem);
3431fbeb3a8SPaul E. McKenney 		up_read(&test_rwsem);
3441fbeb3a8SPaul E. McKenney 	}
3451fbeb3a8SPaul E. McKenney }
3461fbeb3a8SPaul E. McKenney 
3471fbeb3a8SPaul E. McKenney static void ref_rwsem_delay_section(const int nloops, const int udl, const int ndl)
3481fbeb3a8SPaul E. McKenney {
3491fbeb3a8SPaul E. McKenney 	int i;
3501fbeb3a8SPaul E. McKenney 
3511fbeb3a8SPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
3521fbeb3a8SPaul E. McKenney 		down_read(&test_rwsem);
3531fbeb3a8SPaul E. McKenney 		un_delay(udl, ndl);
3541fbeb3a8SPaul E. McKenney 		up_read(&test_rwsem);
3551fbeb3a8SPaul E. McKenney 	}
3561fbeb3a8SPaul E. McKenney }
3571fbeb3a8SPaul E. McKenney 
3581fbeb3a8SPaul E. McKenney static struct ref_scale_ops rwsem_ops = {
3591fbeb3a8SPaul E. McKenney 	.init		= ref_rwsem_init,
3601fbeb3a8SPaul E. McKenney 	.readsection	= ref_rwsem_section,
3611fbeb3a8SPaul E. McKenney 	.delaysection	= ref_rwsem_delay_section,
3621fbeb3a8SPaul E. McKenney 	.name		= "rwsem"
3631fbeb3a8SPaul E. McKenney };
3641fbeb3a8SPaul E. McKenney 
365*e9b800dbSPaul E. McKenney // Definitions for global spinlock
366*e9b800dbSPaul E. McKenney static DEFINE_SPINLOCK(test_lock);
367*e9b800dbSPaul E. McKenney 
368*e9b800dbSPaul E. McKenney static void ref_lock_section(const int nloops)
369*e9b800dbSPaul E. McKenney {
370*e9b800dbSPaul E. McKenney 	int i;
371*e9b800dbSPaul E. McKenney 
372*e9b800dbSPaul E. McKenney 	preempt_disable();
373*e9b800dbSPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
374*e9b800dbSPaul E. McKenney 		spin_lock(&test_lock);
375*e9b800dbSPaul E. McKenney 		spin_unlock(&test_lock);
376*e9b800dbSPaul E. McKenney 	}
377*e9b800dbSPaul E. McKenney 	preempt_enable();
378*e9b800dbSPaul E. McKenney }
379*e9b800dbSPaul E. McKenney 
380*e9b800dbSPaul E. McKenney static void ref_lock_delay_section(const int nloops, const int udl, const int ndl)
381*e9b800dbSPaul E. McKenney {
382*e9b800dbSPaul E. McKenney 	int i;
383*e9b800dbSPaul E. McKenney 
384*e9b800dbSPaul E. McKenney 	preempt_disable();
385*e9b800dbSPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
386*e9b800dbSPaul E. McKenney 		spin_lock(&test_lock);
387*e9b800dbSPaul E. McKenney 		un_delay(udl, ndl);
388*e9b800dbSPaul E. McKenney 		spin_unlock(&test_lock);
389*e9b800dbSPaul E. McKenney 	}
390*e9b800dbSPaul E. McKenney 	preempt_enable();
391*e9b800dbSPaul E. McKenney }
392*e9b800dbSPaul E. McKenney 
393*e9b800dbSPaul E. McKenney static struct ref_scale_ops lock_ops = {
394*e9b800dbSPaul E. McKenney 	.readsection	= ref_lock_section,
395*e9b800dbSPaul E. McKenney 	.delaysection	= ref_lock_delay_section,
396*e9b800dbSPaul E. McKenney 	.name		= "lock"
397*e9b800dbSPaul E. McKenney };
398*e9b800dbSPaul E. McKenney 
399*e9b800dbSPaul E. McKenney // Definitions for global irq-save spinlock
400*e9b800dbSPaul E. McKenney 
401*e9b800dbSPaul E. McKenney static void ref_lock_irq_section(const int nloops)
402*e9b800dbSPaul E. McKenney {
403*e9b800dbSPaul E. McKenney 	unsigned long flags;
404*e9b800dbSPaul E. McKenney 	int i;
405*e9b800dbSPaul E. McKenney 
406*e9b800dbSPaul E. McKenney 	preempt_disable();
407*e9b800dbSPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
408*e9b800dbSPaul E. McKenney 		spin_lock_irqsave(&test_lock, flags);
409*e9b800dbSPaul E. McKenney 		spin_unlock_irqrestore(&test_lock, flags);
410*e9b800dbSPaul E. McKenney 	}
411*e9b800dbSPaul E. McKenney 	preempt_enable();
412*e9b800dbSPaul E. McKenney }
413*e9b800dbSPaul E. McKenney 
414*e9b800dbSPaul E. McKenney static void ref_lock_irq_delay_section(const int nloops, const int udl, const int ndl)
415*e9b800dbSPaul E. McKenney {
416*e9b800dbSPaul E. McKenney 	unsigned long flags;
417*e9b800dbSPaul E. McKenney 	int i;
418*e9b800dbSPaul E. McKenney 
419*e9b800dbSPaul E. McKenney 	preempt_disable();
420*e9b800dbSPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
421*e9b800dbSPaul E. McKenney 		spin_lock_irqsave(&test_lock, flags);
422*e9b800dbSPaul E. McKenney 		un_delay(udl, ndl);
423*e9b800dbSPaul E. McKenney 		spin_unlock_irqrestore(&test_lock, flags);
424*e9b800dbSPaul E. McKenney 	}
425*e9b800dbSPaul E. McKenney 	preempt_enable();
426*e9b800dbSPaul E. McKenney }
427*e9b800dbSPaul E. McKenney 
428*e9b800dbSPaul E. McKenney static struct ref_scale_ops lock_irq_ops = {
429*e9b800dbSPaul E. McKenney 	.readsection	= ref_lock_irq_section,
430*e9b800dbSPaul E. McKenney 	.delaysection	= ref_lock_irq_delay_section,
431*e9b800dbSPaul E. McKenney 	.name		= "lock-irq"
432*e9b800dbSPaul E. McKenney };
433*e9b800dbSPaul E. McKenney 
434*e9b800dbSPaul E. McKenney // Definitions acquire-release.
435*e9b800dbSPaul E. McKenney static DEFINE_PER_CPU(unsigned long, test_acqrel);
436*e9b800dbSPaul E. McKenney 
437*e9b800dbSPaul E. McKenney static void ref_acqrel_section(const int nloops)
438*e9b800dbSPaul E. McKenney {
439*e9b800dbSPaul E. McKenney 	unsigned long x;
440*e9b800dbSPaul E. McKenney 	int i;
441*e9b800dbSPaul E. McKenney 
442*e9b800dbSPaul E. McKenney 	preempt_disable();
443*e9b800dbSPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
444*e9b800dbSPaul E. McKenney 		x = smp_load_acquire(this_cpu_ptr(&test_acqrel));
445*e9b800dbSPaul E. McKenney 		smp_store_release(this_cpu_ptr(&test_acqrel), x + 1);
446*e9b800dbSPaul E. McKenney 	}
447*e9b800dbSPaul E. McKenney 	preempt_enable();
448*e9b800dbSPaul E. McKenney }
449*e9b800dbSPaul E. McKenney 
450*e9b800dbSPaul E. McKenney static void ref_acqrel_delay_section(const int nloops, const int udl, const int ndl)
451*e9b800dbSPaul E. McKenney {
452*e9b800dbSPaul E. McKenney 	unsigned long x;
453*e9b800dbSPaul E. McKenney 	int i;
454*e9b800dbSPaul E. McKenney 
455*e9b800dbSPaul E. McKenney 	preempt_disable();
456*e9b800dbSPaul E. McKenney 	for (i = nloops; i >= 0; i--) {
457*e9b800dbSPaul E. McKenney 		x = smp_load_acquire(this_cpu_ptr(&test_acqrel));
458*e9b800dbSPaul E. McKenney 		un_delay(udl, ndl);
459*e9b800dbSPaul E. McKenney 		smp_store_release(this_cpu_ptr(&test_acqrel), x + 1);
460*e9b800dbSPaul E. McKenney 	}
461*e9b800dbSPaul E. McKenney 	preempt_enable();
462*e9b800dbSPaul E. McKenney }
463*e9b800dbSPaul E. McKenney 
464*e9b800dbSPaul E. McKenney static struct ref_scale_ops acqrel_ops = {
465*e9b800dbSPaul E. McKenney 	.readsection	= ref_acqrel_section,
466*e9b800dbSPaul E. McKenney 	.delaysection	= ref_acqrel_delay_section,
467*e9b800dbSPaul E. McKenney 	.name		= "acqrel"
468*e9b800dbSPaul E. McKenney };
469*e9b800dbSPaul E. McKenney 
4701fbeb3a8SPaul E. McKenney static void rcu_scale_one_reader(void)
4711fbeb3a8SPaul E. McKenney {
4721fbeb3a8SPaul E. McKenney 	if (readdelay <= 0)
4731fbeb3a8SPaul E. McKenney 		cur_ops->readsection(loops);
4741fbeb3a8SPaul E. McKenney 	else
4751fbeb3a8SPaul E. McKenney 		cur_ops->delaysection(loops, readdelay / 1000, readdelay % 1000);
4761fbeb3a8SPaul E. McKenney }
4771fbeb3a8SPaul E. McKenney 
4781fbeb3a8SPaul E. McKenney // Reader kthread.  Repeatedly does empty RCU read-side
4791fbeb3a8SPaul E. McKenney // critical section, minimizing update-side interference.
4801fbeb3a8SPaul E. McKenney static int
4811fbeb3a8SPaul E. McKenney ref_scale_reader(void *arg)
4821fbeb3a8SPaul E. McKenney {
4831fbeb3a8SPaul E. McKenney 	unsigned long flags;
4841fbeb3a8SPaul E. McKenney 	long me = (long)arg;
4851fbeb3a8SPaul E. McKenney 	struct reader_task *rt = &(reader_tasks[me]);
4861fbeb3a8SPaul E. McKenney 	u64 start;
4871fbeb3a8SPaul E. McKenney 	s64 duration;
4881fbeb3a8SPaul E. McKenney 
489e76506f0SPaul E. McKenney 	VERBOSE_SCALEOUT_BATCH("ref_scale_reader %ld: task started", me);
4901fbeb3a8SPaul E. McKenney 	set_cpus_allowed_ptr(current, cpumask_of(me % nr_cpu_ids));
4911fbeb3a8SPaul E. McKenney 	set_user_nice(current, MAX_NICE);
4921fbeb3a8SPaul E. McKenney 	atomic_inc(&n_init);
4931fbeb3a8SPaul E. McKenney 	if (holdoff)
4941fbeb3a8SPaul E. McKenney 		schedule_timeout_interruptible(holdoff * HZ);
4951fbeb3a8SPaul E. McKenney repeat:
496e76506f0SPaul E. McKenney 	VERBOSE_SCALEOUT_BATCH("ref_scale_reader %ld: waiting to start next experiment on cpu %d", me, smp_processor_id());
4971fbeb3a8SPaul E. McKenney 
4981fbeb3a8SPaul E. McKenney 	// Wait for signal that this reader can start.
4991fbeb3a8SPaul E. McKenney 	wait_event(rt->wq, (atomic_read(&nreaders_exp) && smp_load_acquire(&rt->start_reader)) ||
5001fbeb3a8SPaul E. McKenney 			   torture_must_stop());
5011fbeb3a8SPaul E. McKenney 
5021fbeb3a8SPaul E. McKenney 	if (torture_must_stop())
5031fbeb3a8SPaul E. McKenney 		goto end;
5041fbeb3a8SPaul E. McKenney 
5051fbeb3a8SPaul E. McKenney 	// Make sure that the CPU is affinitized appropriately during testing.
5061fbeb3a8SPaul E. McKenney 	WARN_ON_ONCE(smp_processor_id() != me);
5071fbeb3a8SPaul E. McKenney 
5081fbeb3a8SPaul E. McKenney 	WRITE_ONCE(rt->start_reader, 0);
5091fbeb3a8SPaul E. McKenney 	if (!atomic_dec_return(&n_started))
5101fbeb3a8SPaul E. McKenney 		while (atomic_read_acquire(&n_started))
5111fbeb3a8SPaul E. McKenney 			cpu_relax();
5121fbeb3a8SPaul E. McKenney 
513e76506f0SPaul E. McKenney 	VERBOSE_SCALEOUT_BATCH("ref_scale_reader %ld: experiment %d started", me, exp_idx);
5141fbeb3a8SPaul E. McKenney 
5151fbeb3a8SPaul E. McKenney 
5161fbeb3a8SPaul E. McKenney 	// To reduce noise, do an initial cache-warming invocation, check
5171fbeb3a8SPaul E. McKenney 	// in, and then keep warming until everyone has checked in.
5181fbeb3a8SPaul E. McKenney 	rcu_scale_one_reader();
5191fbeb3a8SPaul E. McKenney 	if (!atomic_dec_return(&n_warmedup))
5201fbeb3a8SPaul E. McKenney 		while (atomic_read_acquire(&n_warmedup))
5211fbeb3a8SPaul E. McKenney 			rcu_scale_one_reader();
5221fbeb3a8SPaul E. McKenney 	// Also keep interrupts disabled.  This also has the effect
5231fbeb3a8SPaul E. McKenney 	// of preventing entries into slow path for rcu_read_unlock().
5241fbeb3a8SPaul E. McKenney 	local_irq_save(flags);
5251fbeb3a8SPaul E. McKenney 	start = ktime_get_mono_fast_ns();
5261fbeb3a8SPaul E. McKenney 
5271fbeb3a8SPaul E. McKenney 	rcu_scale_one_reader();
5281fbeb3a8SPaul E. McKenney 
5291fbeb3a8SPaul E. McKenney 	duration = ktime_get_mono_fast_ns() - start;
5301fbeb3a8SPaul E. McKenney 	local_irq_restore(flags);
5311fbeb3a8SPaul E. McKenney 
5321fbeb3a8SPaul E. McKenney 	rt->last_duration_ns = WARN_ON_ONCE(duration < 0) ? 0 : duration;
5331fbeb3a8SPaul E. McKenney 	// To reduce runtime-skew noise, do maintain-load invocations until
5341fbeb3a8SPaul E. McKenney 	// everyone is done.
5351fbeb3a8SPaul E. McKenney 	if (!atomic_dec_return(&n_cooleddown))
5361fbeb3a8SPaul E. McKenney 		while (atomic_read_acquire(&n_cooleddown))
5371fbeb3a8SPaul E. McKenney 			rcu_scale_one_reader();
5381fbeb3a8SPaul E. McKenney 
5391fbeb3a8SPaul E. McKenney 	if (atomic_dec_and_test(&nreaders_exp))
5401fbeb3a8SPaul E. McKenney 		wake_up(&main_wq);
5411fbeb3a8SPaul E. McKenney 
542e76506f0SPaul E. McKenney 	VERBOSE_SCALEOUT_BATCH("ref_scale_reader %ld: experiment %d ended, (readers remaining=%d)",
5431fbeb3a8SPaul E. McKenney 				me, exp_idx, atomic_read(&nreaders_exp));
5441fbeb3a8SPaul E. McKenney 
5451fbeb3a8SPaul E. McKenney 	if (!torture_must_stop())
5461fbeb3a8SPaul E. McKenney 		goto repeat;
5471fbeb3a8SPaul E. McKenney end:
5481fbeb3a8SPaul E. McKenney 	torture_kthread_stopping("ref_scale_reader");
5491fbeb3a8SPaul E. McKenney 	return 0;
5501fbeb3a8SPaul E. McKenney }
5511fbeb3a8SPaul E. McKenney 
5521fbeb3a8SPaul E. McKenney static void reset_readers(void)
5531fbeb3a8SPaul E. McKenney {
5541fbeb3a8SPaul E. McKenney 	int i;
5551fbeb3a8SPaul E. McKenney 	struct reader_task *rt;
5561fbeb3a8SPaul E. McKenney 
5571fbeb3a8SPaul E. McKenney 	for (i = 0; i < nreaders; i++) {
5581fbeb3a8SPaul E. McKenney 		rt = &(reader_tasks[i]);
5591fbeb3a8SPaul E. McKenney 
5601fbeb3a8SPaul E. McKenney 		rt->last_duration_ns = 0;
5611fbeb3a8SPaul E. McKenney 	}
5621fbeb3a8SPaul E. McKenney }
5631fbeb3a8SPaul E. McKenney 
5641fbeb3a8SPaul E. McKenney // Print the results of each reader and return the sum of all their durations.
5651fbeb3a8SPaul E. McKenney static u64 process_durations(int n)
5661fbeb3a8SPaul E. McKenney {
5671fbeb3a8SPaul E. McKenney 	int i;
5681fbeb3a8SPaul E. McKenney 	struct reader_task *rt;
5691fbeb3a8SPaul E. McKenney 	char buf1[64];
5701fbeb3a8SPaul E. McKenney 	char *buf;
5711fbeb3a8SPaul E. McKenney 	u64 sum = 0;
5721fbeb3a8SPaul E. McKenney 
5731fbeb3a8SPaul E. McKenney 	buf = kmalloc(128 + nreaders * 32, GFP_KERNEL);
5741fbeb3a8SPaul E. McKenney 	if (!buf)
5751fbeb3a8SPaul E. McKenney 		return 0;
5761fbeb3a8SPaul E. McKenney 	buf[0] = 0;
5771fbeb3a8SPaul E. McKenney 	sprintf(buf, "Experiment #%d (Format: <THREAD-NUM>:<Total loop time in ns>)",
5781fbeb3a8SPaul E. McKenney 		exp_idx);
5791fbeb3a8SPaul E. McKenney 
5801fbeb3a8SPaul E. McKenney 	for (i = 0; i < n && !torture_must_stop(); i++) {
5811fbeb3a8SPaul E. McKenney 		rt = &(reader_tasks[i]);
5821fbeb3a8SPaul E. McKenney 		sprintf(buf1, "%d: %llu\t", i, rt->last_duration_ns);
5831fbeb3a8SPaul E. McKenney 
5841fbeb3a8SPaul E. McKenney 		if (i % 5 == 0)
5851fbeb3a8SPaul E. McKenney 			strcat(buf, "\n");
5861fbeb3a8SPaul E. McKenney 		strcat(buf, buf1);
5871fbeb3a8SPaul E. McKenney 
5881fbeb3a8SPaul E. McKenney 		sum += rt->last_duration_ns;
5891fbeb3a8SPaul E. McKenney 	}
5901fbeb3a8SPaul E. McKenney 	strcat(buf, "\n");
5911fbeb3a8SPaul E. McKenney 
5921fbeb3a8SPaul E. McKenney 	SCALEOUT("%s\n", buf);
5931fbeb3a8SPaul E. McKenney 
5941fbeb3a8SPaul E. McKenney 	kfree(buf);
5951fbeb3a8SPaul E. McKenney 	return sum;
5961fbeb3a8SPaul E. McKenney }
5971fbeb3a8SPaul E. McKenney 
5981fbeb3a8SPaul E. McKenney // The main_func is the main orchestrator, it performs a bunch of
5991fbeb3a8SPaul E. McKenney // experiments.  For every experiment, it orders all the readers
6001fbeb3a8SPaul E. McKenney // involved to start and waits for them to finish the experiment. It
6011fbeb3a8SPaul E. McKenney // then reads their timestamps and starts the next experiment. Each
6021fbeb3a8SPaul E. McKenney // experiment progresses from 1 concurrent reader to N of them at which
6031fbeb3a8SPaul E. McKenney // point all the timestamps are printed.
6041fbeb3a8SPaul E. McKenney static int main_func(void *arg)
6051fbeb3a8SPaul E. McKenney {
6061fbeb3a8SPaul E. McKenney 	bool errexit = false;
6071fbeb3a8SPaul E. McKenney 	int exp, r;
6081fbeb3a8SPaul E. McKenney 	char buf1[64];
6091fbeb3a8SPaul E. McKenney 	char *buf;
6101fbeb3a8SPaul E. McKenney 	u64 *result_avg;
6111fbeb3a8SPaul E. McKenney 
6121fbeb3a8SPaul E. McKenney 	set_cpus_allowed_ptr(current, cpumask_of(nreaders % nr_cpu_ids));
6131fbeb3a8SPaul E. McKenney 	set_user_nice(current, MAX_NICE);
6141fbeb3a8SPaul E. McKenney 
6151fbeb3a8SPaul E. McKenney 	VERBOSE_SCALEOUT("main_func task started");
6161fbeb3a8SPaul E. McKenney 	result_avg = kzalloc(nruns * sizeof(*result_avg), GFP_KERNEL);
6171fbeb3a8SPaul E. McKenney 	buf = kzalloc(64 + nruns * 32, GFP_KERNEL);
6181fbeb3a8SPaul E. McKenney 	if (!result_avg || !buf) {
6191fbeb3a8SPaul E. McKenney 		VERBOSE_SCALEOUT_ERRSTRING("out of memory");
6201fbeb3a8SPaul E. McKenney 		errexit = true;
6211fbeb3a8SPaul E. McKenney 	}
6221fbeb3a8SPaul E. McKenney 	if (holdoff)
6231fbeb3a8SPaul E. McKenney 		schedule_timeout_interruptible(holdoff * HZ);
6241fbeb3a8SPaul E. McKenney 
6251fbeb3a8SPaul E. McKenney 	// Wait for all threads to start.
6261fbeb3a8SPaul E. McKenney 	atomic_inc(&n_init);
6271fbeb3a8SPaul E. McKenney 	while (atomic_read(&n_init) < nreaders + 1)
6281fbeb3a8SPaul E. McKenney 		schedule_timeout_uninterruptible(1);
6291fbeb3a8SPaul E. McKenney 
6301fbeb3a8SPaul E. McKenney 	// Start exp readers up per experiment
6311fbeb3a8SPaul E. McKenney 	for (exp = 0; exp < nruns && !torture_must_stop(); exp++) {
6321fbeb3a8SPaul E. McKenney 		if (errexit)
6331fbeb3a8SPaul E. McKenney 			break;
6341fbeb3a8SPaul E. McKenney 		if (torture_must_stop())
6351fbeb3a8SPaul E. McKenney 			goto end;
6361fbeb3a8SPaul E. McKenney 
6371fbeb3a8SPaul E. McKenney 		reset_readers();
6381fbeb3a8SPaul E. McKenney 		atomic_set(&nreaders_exp, nreaders);
6391fbeb3a8SPaul E. McKenney 		atomic_set(&n_started, nreaders);
6401fbeb3a8SPaul E. McKenney 		atomic_set(&n_warmedup, nreaders);
6411fbeb3a8SPaul E. McKenney 		atomic_set(&n_cooleddown, nreaders);
6421fbeb3a8SPaul E. McKenney 
6431fbeb3a8SPaul E. McKenney 		exp_idx = exp;
6441fbeb3a8SPaul E. McKenney 
6451fbeb3a8SPaul E. McKenney 		for (r = 0; r < nreaders; r++) {
6461fbeb3a8SPaul E. McKenney 			smp_store_release(&reader_tasks[r].start_reader, 1);
6471fbeb3a8SPaul E. McKenney 			wake_up(&reader_tasks[r].wq);
6481fbeb3a8SPaul E. McKenney 		}
6491fbeb3a8SPaul E. McKenney 
6501fbeb3a8SPaul E. McKenney 		VERBOSE_SCALEOUT("main_func: experiment started, waiting for %d readers",
6511fbeb3a8SPaul E. McKenney 				nreaders);
6521fbeb3a8SPaul E. McKenney 
6531fbeb3a8SPaul E. McKenney 		wait_event(main_wq,
6541fbeb3a8SPaul E. McKenney 			   !atomic_read(&nreaders_exp) || torture_must_stop());
6551fbeb3a8SPaul E. McKenney 
6561fbeb3a8SPaul E. McKenney 		VERBOSE_SCALEOUT("main_func: experiment ended");
6571fbeb3a8SPaul E. McKenney 
6581fbeb3a8SPaul E. McKenney 		if (torture_must_stop())
6591fbeb3a8SPaul E. McKenney 			goto end;
6601fbeb3a8SPaul E. McKenney 
6611fbeb3a8SPaul E. McKenney 		result_avg[exp] = div_u64(1000 * process_durations(nreaders), nreaders * loops);
6621fbeb3a8SPaul E. McKenney 	}
6631fbeb3a8SPaul E. McKenney 
6641fbeb3a8SPaul E. McKenney 	// Print the average of all experiments
6651fbeb3a8SPaul E. McKenney 	SCALEOUT("END OF TEST. Calculating average duration per loop (nanoseconds)...\n");
6661fbeb3a8SPaul E. McKenney 
66758db5785SColin Ian King 	if (!errexit) {
6681fbeb3a8SPaul E. McKenney 		buf[0] = 0;
6691fbeb3a8SPaul E. McKenney 		strcat(buf, "\n");
6701fbeb3a8SPaul E. McKenney 		strcat(buf, "Runs\tTime(ns)\n");
67158db5785SColin Ian King 	}
6721fbeb3a8SPaul E. McKenney 
6731fbeb3a8SPaul E. McKenney 	for (exp = 0; exp < nruns; exp++) {
6741fbeb3a8SPaul E. McKenney 		u64 avg;
6751fbeb3a8SPaul E. McKenney 		u32 rem;
6761fbeb3a8SPaul E. McKenney 
6771fbeb3a8SPaul E. McKenney 		if (errexit)
6781fbeb3a8SPaul E. McKenney 			break;
6791fbeb3a8SPaul E. McKenney 		avg = div_u64_rem(result_avg[exp], 1000, &rem);
6801fbeb3a8SPaul E. McKenney 		sprintf(buf1, "%d\t%llu.%03u\n", exp + 1, avg, rem);
6811fbeb3a8SPaul E. McKenney 		strcat(buf, buf1);
6821fbeb3a8SPaul E. McKenney 	}
6831fbeb3a8SPaul E. McKenney 
6841fbeb3a8SPaul E. McKenney 	if (!errexit)
6851fbeb3a8SPaul E. McKenney 		SCALEOUT("%s", buf);
6861fbeb3a8SPaul E. McKenney 
6871fbeb3a8SPaul E. McKenney 	// This will shutdown everything including us.
6881fbeb3a8SPaul E. McKenney 	if (shutdown) {
6891fbeb3a8SPaul E. McKenney 		shutdown_start = 1;
6901fbeb3a8SPaul E. McKenney 		wake_up(&shutdown_wq);
6911fbeb3a8SPaul E. McKenney 	}
6921fbeb3a8SPaul E. McKenney 
6931fbeb3a8SPaul E. McKenney 	// Wait for torture to stop us
6941fbeb3a8SPaul E. McKenney 	while (!torture_must_stop())
6951fbeb3a8SPaul E. McKenney 		schedule_timeout_uninterruptible(1);
6961fbeb3a8SPaul E. McKenney 
6971fbeb3a8SPaul E. McKenney end:
6981fbeb3a8SPaul E. McKenney 	torture_kthread_stopping("main_func");
6991fbeb3a8SPaul E. McKenney 	kfree(result_avg);
7001fbeb3a8SPaul E. McKenney 	kfree(buf);
7011fbeb3a8SPaul E. McKenney 	return 0;
7021fbeb3a8SPaul E. McKenney }
7031fbeb3a8SPaul E. McKenney 
7041fbeb3a8SPaul E. McKenney static void
7051fbeb3a8SPaul E. McKenney ref_scale_print_module_parms(struct ref_scale_ops *cur_ops, const char *tag)
7061fbeb3a8SPaul E. McKenney {
7071fbeb3a8SPaul E. McKenney 	pr_alert("%s" SCALE_FLAG
7081fbeb3a8SPaul E. McKenney 		 "--- %s:  verbose=%d shutdown=%d holdoff=%d loops=%ld nreaders=%d nruns=%d readdelay=%d\n", scale_type, tag,
7091fbeb3a8SPaul E. McKenney 		 verbose, shutdown, holdoff, loops, nreaders, nruns, readdelay);
7101fbeb3a8SPaul E. McKenney }
7111fbeb3a8SPaul E. McKenney 
7121fbeb3a8SPaul E. McKenney static void
7131fbeb3a8SPaul E. McKenney ref_scale_cleanup(void)
7141fbeb3a8SPaul E. McKenney {
7151fbeb3a8SPaul E. McKenney 	int i;
7161fbeb3a8SPaul E. McKenney 
7171fbeb3a8SPaul E. McKenney 	if (torture_cleanup_begin())
7181fbeb3a8SPaul E. McKenney 		return;
7191fbeb3a8SPaul E. McKenney 
7201fbeb3a8SPaul E. McKenney 	if (!cur_ops) {
7211fbeb3a8SPaul E. McKenney 		torture_cleanup_end();
7221fbeb3a8SPaul E. McKenney 		return;
7231fbeb3a8SPaul E. McKenney 	}
7241fbeb3a8SPaul E. McKenney 
7251fbeb3a8SPaul E. McKenney 	if (reader_tasks) {
7261fbeb3a8SPaul E. McKenney 		for (i = 0; i < nreaders; i++)
7271fbeb3a8SPaul E. McKenney 			torture_stop_kthread("ref_scale_reader",
7281fbeb3a8SPaul E. McKenney 					     reader_tasks[i].task);
7291fbeb3a8SPaul E. McKenney 	}
7301fbeb3a8SPaul E. McKenney 	kfree(reader_tasks);
7311fbeb3a8SPaul E. McKenney 
7321fbeb3a8SPaul E. McKenney 	torture_stop_kthread("main_task", main_task);
7331fbeb3a8SPaul E. McKenney 	kfree(main_task);
7341fbeb3a8SPaul E. McKenney 
7351fbeb3a8SPaul E. McKenney 	// Do scale-type-specific cleanup operations.
7361fbeb3a8SPaul E. McKenney 	if (cur_ops->cleanup != NULL)
7371fbeb3a8SPaul E. McKenney 		cur_ops->cleanup();
7381fbeb3a8SPaul E. McKenney 
7391fbeb3a8SPaul E. McKenney 	torture_cleanup_end();
7401fbeb3a8SPaul E. McKenney }
7411fbeb3a8SPaul E. McKenney 
7421fbeb3a8SPaul E. McKenney // Shutdown kthread.  Just waits to be awakened, then shuts down system.
7431fbeb3a8SPaul E. McKenney static int
7441fbeb3a8SPaul E. McKenney ref_scale_shutdown(void *arg)
7451fbeb3a8SPaul E. McKenney {
7461fbeb3a8SPaul E. McKenney 	wait_event(shutdown_wq, shutdown_start);
7471fbeb3a8SPaul E. McKenney 
7481fbeb3a8SPaul E. McKenney 	smp_mb(); // Wake before output.
7491fbeb3a8SPaul E. McKenney 	ref_scale_cleanup();
7501fbeb3a8SPaul E. McKenney 	kernel_power_off();
7511fbeb3a8SPaul E. McKenney 
7521fbeb3a8SPaul E. McKenney 	return -EINVAL;
7531fbeb3a8SPaul E. McKenney }
7541fbeb3a8SPaul E. McKenney 
7551fbeb3a8SPaul E. McKenney static int __init
7561fbeb3a8SPaul E. McKenney ref_scale_init(void)
7571fbeb3a8SPaul E. McKenney {
7581fbeb3a8SPaul E. McKenney 	long i;
7591fbeb3a8SPaul E. McKenney 	int firsterr = 0;
7601fbeb3a8SPaul E. McKenney 	static struct ref_scale_ops *scale_ops[] = {
761*e9b800dbSPaul E. McKenney 		&rcu_ops, &srcu_ops, &rcu_trace_ops, &rcu_tasks_ops, &refcnt_ops, &rwlock_ops,
762*e9b800dbSPaul E. McKenney 		&rwsem_ops, &lock_ops, &lock_irq_ops, &acqrel_ops,
7631fbeb3a8SPaul E. McKenney 	};
7641fbeb3a8SPaul E. McKenney 
7651fbeb3a8SPaul E. McKenney 	if (!torture_init_begin(scale_type, verbose))
7661fbeb3a8SPaul E. McKenney 		return -EBUSY;
7671fbeb3a8SPaul E. McKenney 
7681fbeb3a8SPaul E. McKenney 	for (i = 0; i < ARRAY_SIZE(scale_ops); i++) {
7691fbeb3a8SPaul E. McKenney 		cur_ops = scale_ops[i];
7701fbeb3a8SPaul E. McKenney 		if (strcmp(scale_type, cur_ops->name) == 0)
7711fbeb3a8SPaul E. McKenney 			break;
7721fbeb3a8SPaul E. McKenney 	}
7731fbeb3a8SPaul E. McKenney 	if (i == ARRAY_SIZE(scale_ops)) {
7741fbeb3a8SPaul E. McKenney 		pr_alert("rcu-scale: invalid scale type: \"%s\"\n", scale_type);
7751fbeb3a8SPaul E. McKenney 		pr_alert("rcu-scale types:");
7761fbeb3a8SPaul E. McKenney 		for (i = 0; i < ARRAY_SIZE(scale_ops); i++)
7771fbeb3a8SPaul E. McKenney 			pr_cont(" %s", scale_ops[i]->name);
7781fbeb3a8SPaul E. McKenney 		pr_cont("\n");
7791fbeb3a8SPaul E. McKenney 		firsterr = -EINVAL;
7801fbeb3a8SPaul E. McKenney 		cur_ops = NULL;
7811fbeb3a8SPaul E. McKenney 		goto unwind;
7821fbeb3a8SPaul E. McKenney 	}
7831fbeb3a8SPaul E. McKenney 	if (cur_ops->init)
7841fbeb3a8SPaul E. McKenney 		cur_ops->init();
7851fbeb3a8SPaul E. McKenney 
7861fbeb3a8SPaul E. McKenney 	ref_scale_print_module_parms(cur_ops, "Start of test");
7871fbeb3a8SPaul E. McKenney 
7881fbeb3a8SPaul E. McKenney 	// Shutdown task
7891fbeb3a8SPaul E. McKenney 	if (shutdown) {
7901fbeb3a8SPaul E. McKenney 		init_waitqueue_head(&shutdown_wq);
7911fbeb3a8SPaul E. McKenney 		firsterr = torture_create_kthread(ref_scale_shutdown, NULL,
7921fbeb3a8SPaul E. McKenney 						  shutdown_task);
7931fbeb3a8SPaul E. McKenney 		if (firsterr)
7941fbeb3a8SPaul E. McKenney 			goto unwind;
7951fbeb3a8SPaul E. McKenney 		schedule_timeout_uninterruptible(1);
7961fbeb3a8SPaul E. McKenney 	}
7971fbeb3a8SPaul E. McKenney 
7981fbeb3a8SPaul E. McKenney 	// Reader tasks (default to ~75% of online CPUs).
7991fbeb3a8SPaul E. McKenney 	if (nreaders < 0)
8001fbeb3a8SPaul E. McKenney 		nreaders = (num_online_cpus() >> 1) + (num_online_cpus() >> 2);
8010c6d18d8SPaul E. McKenney 	if (WARN_ONCE(loops <= 0, "%s: loops = %ld, adjusted to 1\n", __func__, loops))
8020c6d18d8SPaul E. McKenney 		loops = 1;
8030c6d18d8SPaul E. McKenney 	if (WARN_ONCE(nreaders <= 0, "%s: nreaders = %d, adjusted to 1\n", __func__, nreaders))
8040c6d18d8SPaul E. McKenney 		nreaders = 1;
8050c6d18d8SPaul E. McKenney 	if (WARN_ONCE(nruns <= 0, "%s: nruns = %d, adjusted to 1\n", __func__, nruns))
8060c6d18d8SPaul E. McKenney 		nruns = 1;
8071fbeb3a8SPaul E. McKenney 	reader_tasks = kcalloc(nreaders, sizeof(reader_tasks[0]),
8081fbeb3a8SPaul E. McKenney 			       GFP_KERNEL);
8091fbeb3a8SPaul E. McKenney 	if (!reader_tasks) {
8101fbeb3a8SPaul E. McKenney 		VERBOSE_SCALEOUT_ERRSTRING("out of memory");
8111fbeb3a8SPaul E. McKenney 		firsterr = -ENOMEM;
8121fbeb3a8SPaul E. McKenney 		goto unwind;
8131fbeb3a8SPaul E. McKenney 	}
8141fbeb3a8SPaul E. McKenney 
8151fbeb3a8SPaul E. McKenney 	VERBOSE_SCALEOUT("Starting %d reader threads\n", nreaders);
8161fbeb3a8SPaul E. McKenney 
8171fbeb3a8SPaul E. McKenney 	for (i = 0; i < nreaders; i++) {
8181fbeb3a8SPaul E. McKenney 		firsterr = torture_create_kthread(ref_scale_reader, (void *)i,
8191fbeb3a8SPaul E. McKenney 						  reader_tasks[i].task);
8201fbeb3a8SPaul E. McKenney 		if (firsterr)
8211fbeb3a8SPaul E. McKenney 			goto unwind;
8221fbeb3a8SPaul E. McKenney 
8231fbeb3a8SPaul E. McKenney 		init_waitqueue_head(&(reader_tasks[i].wq));
8241fbeb3a8SPaul E. McKenney 	}
8251fbeb3a8SPaul E. McKenney 
8261fbeb3a8SPaul E. McKenney 	// Main Task
8271fbeb3a8SPaul E. McKenney 	init_waitqueue_head(&main_wq);
8281fbeb3a8SPaul E. McKenney 	firsterr = torture_create_kthread(main_func, NULL, main_task);
8291fbeb3a8SPaul E. McKenney 	if (firsterr)
8301fbeb3a8SPaul E. McKenney 		goto unwind;
8311fbeb3a8SPaul E. McKenney 
8321fbeb3a8SPaul E. McKenney 	torture_init_end();
8331fbeb3a8SPaul E. McKenney 	return 0;
8341fbeb3a8SPaul E. McKenney 
8351fbeb3a8SPaul E. McKenney unwind:
8361fbeb3a8SPaul E. McKenney 	torture_init_end();
8371fbeb3a8SPaul E. McKenney 	ref_scale_cleanup();
838bc80d353SPaul E. McKenney 	if (shutdown) {
839bc80d353SPaul E. McKenney 		WARN_ON(!IS_MODULE(CONFIG_RCU_REF_SCALE_TEST));
840bc80d353SPaul E. McKenney 		kernel_power_off();
841bc80d353SPaul E. McKenney 	}
8421fbeb3a8SPaul E. McKenney 	return firsterr;
8431fbeb3a8SPaul E. McKenney }
8441fbeb3a8SPaul E. McKenney 
8451fbeb3a8SPaul E. McKenney module_init(ref_scale_init);
8461fbeb3a8SPaul E. McKenney module_exit(ref_scale_cleanup);
847