[fio.git] / engines / cpu.c

/*
 * CPU engine
 *
 * Doesn't transfer any data, merely burns CPU cycles according to
 * the settings.
 *
 */
#include "../fio.h"

struct cpu_options {
	struct thread_data *td;
	unsigned int cpuload;
	unsigned int cpucycle;
	unsigned int exit_io_done;
};

static struct fio_option options[] = {
	{
		.name	= "cpuload",
		.lname	= "CPU load",
		.type	= FIO_OPT_INT,
		.off1	= offsetof(struct cpu_options, cpuload),
		.help	= "Use this percentage of CPU",
		.category = FIO_OPT_C_GENERAL,
		.group	= FIO_OPT_G_INVALID,
	},
	{
		.name	= "cpuchunks",
		.lname	= "CPU chunk",
		.type	= FIO_OPT_INT,
		.off1	= offsetof(struct cpu_options, cpucycle),
		.help	= "Length of the CPU burn cycles (usecs)",
		.def	= "50000",
		.parent = "cpuload",
		.hide	= 1,
		.category = FIO_OPT_C_GENERAL,
		.group	= FIO_OPT_G_INVALID,
	},
	{
		.name	= "exit_on_io_done",
		.lname	= "Exit when IO threads are done",
		.type	= FIO_OPT_BOOL,
		.off1	= offsetof(struct cpu_options, exit_io_done),
		.help	= "Exit when IO threads finish",
		.def	= "0",
		.category = FIO_OPT_C_GENERAL,
		.group	= FIO_OPT_G_INVALID,
	},
	{
		.name	= NULL,
	},
};


static int fio_cpuio_queue(struct thread_data *td, struct io_u fio_unused *io_u)
{
	struct cpu_options *co = td->eo;

	if (co->exit_io_done && !fio_running_or_pending_io_threads()) {
		td->done = 1;
		return FIO_Q_BUSY;
	}

	usec_spin(co->cpucycle);
	return FIO_Q_COMPLETED;
}

static int fio_cpuio_init(struct thread_data *td)
{
	struct thread_options *o = &td->o;
	struct cpu_options *co = td->eo;

	if (!co->cpuload) {
		td_vmsg(td, EINVAL, "cpu thread needs rate (cpuload=)","cpuio");
		return 1;
	}

	if (co->cpuload > 100)
		co->cpuload = 100;

	/*
	 * set thinktime_sleep and thinktime_spin appropriately
	 */
	o->thinktime_blocks = 1;
	o->thinktime_spin = 0;
	o->thinktime = (co->cpucycle * (100 - co->cpuload)) / co->cpuload;

	o->nr_files = o->open_files = 1;

	log_info("%s: ioengine=cpu, cpuload=%u, cpucycle=%u\n", td->o.name,
						co->cpuload, co->cpucycle);

	return 0;
}

static int fio_cpuio_open(struct thread_data fio_unused *td,
			  struct fio_file fio_unused *f)
{
	return 0;
}

static struct ioengine_ops ioengine = {
	.name		= "cpuio",
	.version	= FIO_IOOPS_VERSION,
	.queue		= fio_cpuio_queue,
	.init		= fio_cpuio_init,
	.open_file	= fio_cpuio_open,
	.flags		= FIO_SYNCIO | FIO_DISKLESSIO | FIO_NOIO,
	.options		= options,
	.option_struct_size	= sizeof(struct cpu_options),
};

static void fio_init fio_cpuio_register(void)
{
	register_ioengine(&ioengine);
}

static void fio_exit fio_cpuio_unregister(void)
{
	unregister_ioengine(&ioengine);
}
Commit	Line	Data
da751ca9 JA	1	/*
	2	* CPU engine
	3	*
	4	* Doesn't transfer any data, merely burns CPU cycles according to
	5	* the settings.
	6	*
	7	*/
5f350952	8	#include "../fio.h"
2866c82d	9
0353050f JA	10	struct cpu_options {
	11	struct thread_data *td;
	12	unsigned int cpuload;
	13	unsigned int cpucycle;
046395d7	14	unsigned int exit_io_done;
0353050f JA	15	};
	16
	17	static struct fio_option options[] = {
	18	{
	19	.name = "cpuload",
	20	.lname = "CPU load",
	21	.type = FIO_OPT_INT,
	22	.off1 = offsetof(struct cpu_options, cpuload),
	23	.help = "Use this percentage of CPU",
	24	.category = FIO_OPT_C_GENERAL,
	25	.group = FIO_OPT_G_INVALID,
	26	},
	27	{
	28	.name = "cpuchunks",
	29	.lname = "CPU chunk",
	30	.type = FIO_OPT_INT,
	31	.off1 = offsetof(struct cpu_options, cpucycle),
	32	.help = "Length of the CPU burn cycles (usecs)",
	33	.def = "50000",
	34	.parent = "cpuload",
	35	.hide = 1,
	36	.category = FIO_OPT_C_GENERAL,
	37	.group = FIO_OPT_G_INVALID,
	38	},
046395d7 JA	39	{
	40	.name = "exit_on_io_done",
	41	.lname = "Exit when IO threads are done",
	42	.type = FIO_OPT_BOOL,
	43	.off1 = offsetof(struct cpu_options, exit_io_done),
	44	.help = "Exit when IO threads finish",
	45	.def = "0",
	46	.category = FIO_OPT_C_GENERAL,
	47	.group = FIO_OPT_G_INVALID,
	48	},
0353050f JA	49	{
	50	.name = NULL,
	51	},
	52	};
	53
	54
ba0fbe10 JA	55	static int fio_cpuio_queue(struct thread_data td, struct io_u fio_unused io_u)
ba0fbe10 JA	56	{
0353050f JA	57	struct cpu_options *co = td->eo;
0353050f JA	58
046395d7 JA	59	if (co->exit_io_done && !fio_running_or_pending_io_threads()) {
	60	td->done = 1;
	61	return FIO_Q_BUSY;
	62	}
	63
0353050f	64	usec_spin(co->cpucycle);
ba0fbe10 JA	65	return FIO_Q_COMPLETED;
	66	}
	67
2866c82d JA	68	static int fio_cpuio_init(struct thread_data *td)
2866c82d JA	69	{
2dc1bbeb	70	struct thread_options *o = &td->o;
0353050f	71	struct cpu_options *co = td->eo;
2dc1bbeb	72
0353050f	73	if (!co->cpuload) {
ba0fbe10	74	td_vmsg(td, EINVAL, "cpu thread needs rate (cpuload=)","cpuio");
2866c82d	75	return 1;
ba0fbe10 JA	76	}
ba0fbe10 JA	77
0353050f JA	78	if (co->cpuload > 100)
0353050f JA	79	co->cpuload = 100;
2866c82d	80
ba0fbe10 JA	81	/*
	82	* set thinktime_sleep and thinktime_spin appropriately
	83	*/
2dc1bbeb JA	84	o->thinktime_blocks = 1;
2dc1bbeb JA	85	o->thinktime_spin = 0;
0353050f	86	o->thinktime = (co->cpucycle * (100 - co->cpuload)) / co->cpuload;
2866c82d	87
2dc1bbeb	88	o->nr_files = o->open_files = 1;
0353050f JA	89
	90	log_info("%s: ioengine=cpu, cpuload=%u, cpucycle=%u\n", td->o.name,
	91	co->cpuload, co->cpucycle);
	92
ba0fbe10 JA	93	return 0;
	94	}
	95
f4e62a5f JA	96	static int fio_cpuio_open(struct thread_data fio_unused *td,
f4e62a5f JA	97	struct fio_file fio_unused *f)
ba0fbe10	98	{
2866c82d JA	99	return 0;
	100	}
	101
5f350952	102	static struct ioengine_ops ioengine = {
2866c82d JA	103	.name = "cpuio",
2866c82d JA	104	.version = FIO_IOOPS_VERSION,
ba0fbe10	105	.queue = fio_cpuio_queue,
2866c82d	106	.init = fio_cpuio_init,
ba0fbe10	107	.open_file = fio_cpuio_open,
1f809d15	108	.flags = FIO_SYNCIO \| FIO_DISKLESSIO \| FIO_NOIO,
0353050f JA	109	.options = options,
0353050f JA	110	.option_struct_size = sizeof(struct cpu_options),
2866c82d	111	};
5f350952 JA	112
	113	static void fio_init fio_cpuio_register(void)
	114	{
	115	register_ioengine(&ioengine);
	116	}
	117
	118	static void fio_exit fio_cpuio_unregister(void)
	119	{
	120	unregister_ioengine(&ioengine);
	121	}