engines/cpu.c

   1 /*
   2  * CPU engine
   3  *
   4  * Doesn't transfer any data, merely burns CPU cycles according to
   5  * the settings.
   6  *
   7  */
   8 #include "../fio.h"
   9 #include "../os.h"
  10
  11 static int fio_cpuio_queue(struct thread_data *td, struct io_u fio_unused *io_u)
  12 {
  13         __usec_sleep(td->cpucycle);
  14         return FIO_Q_COMPLETED;
  15 }
  16
  17 static int fio_cpuio_setup(struct thread_data fio_unused *td)
  18 {
  19         struct fio_file *f;
  20         unsigned int i;
  21
  22         td->total_file_size = -1;
  23         td->io_size = td->total_file_size;
  24         td->total_io_size = td->io_size;
  25
  26         for_each_file(td, f, i) {
  27                 f->real_file_size = -1;
  28                 f->file_size = -1;
  29         }
  30
  31         return 0;
  32 }
  33
  34 static int fio_cpuio_init(struct thread_data *td)
  35 {
  36         if (!td->cpuload) {
  37                 td_vmsg(td, EINVAL, "cpu thread needs rate (cpuload=)","cpuio");
  38                 return 1;
  39         }
  40
  41         if (td->cpuload > 100)
  42                 td->cpuload = 100;
  43
  44         /*
  45          * set thinktime_sleep and thinktime_spin appropriately
  46          */
  47         td->thinktime_blocks = 1;
  48         td->thinktime_spin = 0;
  49         td->thinktime = (td->cpucycle * (100 - td->cpuload)) / td->cpuload;
  50
  51         td->nr_files = td->open_files = 1;
  52         return 0;
  53 }
  54
  55 static int fio_cpuio_open(struct thread_data fio_unused *td, struct fio_file *f)
  56 {
  57         f->fd = 0;
  58         return 0;
  59 }
  60
  61 static struct ioengine_ops ioengine = {
  62         .name           = "cpuio",
  63         .version        = FIO_IOOPS_VERSION,
  64         .queue          = fio_cpuio_queue,
  65         .init           = fio_cpuio_init,
  66         .setup          = fio_cpuio_setup,
  67         .open_file      = fio_cpuio_open,
  68         .flags          = FIO_SYNCIO | FIO_DISKLESSIO,
  69 };
  70
  71 static void fio_init fio_cpuio_register(void)
  72 {
  73         register_ioengine(&ioengine);
  74 }
  75
  76 static void fio_exit fio_cpuio_unregister(void)
  77 {
  78         unregister_ioengine(&ioengine);
  79 }