[linux.git] / lib / dhry_run.c

// SPDX-License-Identifier: GPL-2.0-only
/*
 * Dhrystone benchmark test module
 *
 * Copyright (C) 2022 Glider bv
 */

#include "dhry.h"

#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/moduleparam.h>
#include <linux/smp.h>

#define DHRY_VAX	1757

static int dhry_run_set(const char *val, const struct kernel_param *kp);
static const struct kernel_param_ops run_ops = {
	.flags = KERNEL_PARAM_OPS_FL_NOARG,
	.set = dhry_run_set,
};
static bool dhry_run;
module_param_cb(run, &run_ops, &dhry_run, 0200);
MODULE_PARM_DESC(run, "Run the test (default: false)");

static int iterations = -1;
module_param(iterations, int, 0644);
MODULE_PARM_DESC(iterations,
		"Number of iterations through the benchmark (default: auto)");

static void dhry_benchmark(void)
{
	unsigned int cpu = get_cpu();
	int i, n;

	if (iterations > 0) {
		n = dhry(iterations);
		goto report;
	}

	for (i = DHRY_VAX; i > 0; i <<= 1) {
		n = dhry(i);
		if (n != -EAGAIN)
			break;
	}

report:
	put_cpu();
	if (n >= 0)
		pr_info("CPU%u: Dhrystones per Second: %d (%d DMIPS)\n", cpu,
			n, n / DHRY_VAX);
	else if (n == -EAGAIN)
		pr_err("Please increase the number of iterations\n");
	else
		pr_err("Dhrystone benchmark failed error %pe\n", ERR_PTR(n));
}

static int dhry_run_set(const char *val, const struct kernel_param *kp)
{
	int ret;

	if (val) {
		ret = param_set_bool(val, kp);
		if (ret)
			return ret;
	} else {
		dhry_run = true;
	}

	if (dhry_run && system_state == SYSTEM_RUNNING)
		dhry_benchmark();

	return 0;
}

static int __init dhry_init(void)
{
	if (dhry_run)
		dhry_benchmark();

	return 0;
}
module_init(dhry_init);

MODULE_AUTHOR("Geert Uytterhoeven <[email protected]>");
MODULE_DESCRIPTION("Dhrystone benchmark test module");
MODULE_LICENSE("GPL");
Commit	Line	Data
d5528cc1 GU	1	// SPDX-License-Identifier: GPL-2.0-only
	2	/*
	3	* Dhrystone benchmark test module
	4	*
	5	* Copyright (C) 2022 Glider bv
	6	*/
	7
	8	#include "dhry.h"
	9
	10	#include <linux/kernel.h>
	11	#include <linux/module.h>
	12	#include <linux/moduleparam.h>
d5528cc1 GU	13	#include <linux/smp.h>
	14
	15	#define DHRY_VAX 1757
	16
	17	static int dhry_run_set(const char val, const struct kernel_param kp);
	18	static const struct kernel_param_ops run_ops = {
	19	.flags = KERNEL_PARAM_OPS_FL_NOARG,
	20	.set = dhry_run_set,
	21	};
	22	static bool dhry_run;
	23	module_param_cb(run, &run_ops, &dhry_run, 0200);
	24	MODULE_PARM_DESC(run, "Run the test (default: false)");
	25
	26	static int iterations = -1;
	27	module_param(iterations, int, 0644);
	28	MODULE_PARM_DESC(iterations,
	29	"Number of iterations through the benchmark (default: auto)");
	30
	31	static void dhry_benchmark(void)
	32	{
13684e96	33	unsigned int cpu = get_cpu();
d5528cc1 GU	34	int i, n;
	35
	36	if (iterations > 0) {
	37	n = dhry(iterations);
	38	goto report;
	39	}
	40
	41	for (i = DHRY_VAX; i > 0; i <<= 1) {
	42	n = dhry(i);
	43	if (n != -EAGAIN)
	44	break;
	45	}
	46
	47	report:
13684e96	48	put_cpu();
d5528cc1	49	if (n >= 0)
13684e96 GU	50	pr_info("CPU%u: Dhrystones per Second: %d (%d DMIPS)\n", cpu,
13684e96 GU	51	n, n / DHRY_VAX);
d5528cc1 GU	52	else if (n == -EAGAIN)
	53	pr_err("Please increase the number of iterations\n");
	54	else
	55	pr_err("Dhrystone benchmark failed error %pe\n", ERR_PTR(n));
	56	}
	57
	58	static int dhry_run_set(const char val, const struct kernel_param kp)
	59	{
	60	int ret;
	61
	62	if (val) {
	63	ret = param_set_bool(val, kp);
	64	if (ret)
	65	return ret;
	66	} else {
	67	dhry_run = true;
	68	}
	69
	70	if (dhry_run && system_state == SYSTEM_RUNNING)
	71	dhry_benchmark();
	72
	73	return 0;
	74	}
	75
	76	static int __init dhry_init(void)
	77	{
	78	if (dhry_run)
	79	dhry_benchmark();
	80
	81	return 0;
	82	}
	83	module_init(dhry_init);
	84
	85	MODULE_AUTHOR("Geert Uytterhoeven <[email protected]>");
30347491	86	MODULE_DESCRIPTION("Dhrystone benchmark test module");
d5528cc1	87	MODULE_LICENSE("GPL");