--track-energy feature

pyperf/_manager.py

@@ -65,6 +65,8 @@ def worker_cmd(self, calibrate_loops, calibrate_warmups, wpipe):
             cmd.append('--tracemalloc')
         if args.track_memory:
             cmd.append('--track-memory')
+        if args.track_energy:
+            cmd.append('--track-energy')
 
         if self.runner._add_cmdline_args:
             self.runner._add_cmdline_args(cmd, args)

pyperf/_runner.py

@@ -104,9 +104,6 @@ def __init__(self, values=None, warmups=None, processes=None,
         # Set used to check that benchmark names are unique
         self._bench_names = set()
 
-        # result of argparser.parse_args()
-        self.args = None
-
         # callback used to prepare command line arguments to spawn a worker
         # child process. The callback is called with prepare(runner.args, cmd).
         # args must be modified in-place.
@@ -221,6 +218,9 @@ def __init__(self, values=None, warmups=None, processes=None,
                             help='option used with --compare-to to name '
                                  'PYTHON as CHANGED_NAME '
                                  'and REF_PYTHON as REF_NAME in results')
+        parser.add_argument("--track-energy",
+                            action="store_true",
+                            help="Measure energy instead of wall clock time.")
 
         memory = parser.add_mutually_exclusive_group()
         memory.add_argument('--tracemalloc', action="store_true",
@@ -230,6 +230,9 @@ def __init__(self, values=None, warmups=None, processes=None,
 
         self.argparser = parser
 
+        # result of argparser.parse_args()
+        self.args = None
+
     def _multiline_output(self):
         return self.args.verbose or multiline_output(self.args)
 
@@ -420,7 +423,7 @@ def _main(self, task):
         if task.name in self._bench_names:
             raise ValueError("duplicated benchmark name: %r" % task.name)
         self._bench_names.add(task.name)
-
+        
         args = self.parse_args()
         try:
             if args.worker:
@@ -491,6 +494,7 @@ def task_func(task, loops):
                 dt = local_timer() - t0
 
             return dt
+
 
         task = WorkerProcessTask(self, name, task_func, metadata)
         task.inner_loops = inner_loops

pyperf/_worker.py

@@ -15,6 +15,10 @@
 MAX_WARMUP_VALUES = 300
 WARMUP_SAMPLE_SIZE = 20
 
+# To invoke C in the context of --track-energy.
+import ctypes
+import os
+
 
 class WorkerTask:
     def __init__(self, runner, name, task_func, func_metadata):
@@ -63,9 +67,17 @@ def _compute_values(self, values, nvalue,
         while True:
             if index > nvalue:
                 break
-
-            raw_value = self.task_func(self, self.loops)
-            raw_value = float(raw_value)
+            if self.args.track_energy:
+              # Use environment variable for where the readings are stored.
+              c_lib = ctypes.CDLL(os.environ.get("READEN"))
+              # Energy value is the difference between recorded energies
+              # before and after executing task function.
+              e_0 = ctypes.c_ulonglong(c_lib.readen(os.environ.get("ENFILE").encode('utf-8')))
+              self.task_func(self, self.loops)
+              e_1 = ctypes.c_ulonglong(c_lib.readen(os.environ.get("ENFILE").encode('utf-8')))
+              raw_value = float(e_1.value) - float(e_0.value)
+            else:
+              raw_value = float(self.task_func(self, self.loops))
             value = raw_value / (self.loops * inner_loops)
 
             if not value and not calibrate_loops:

pyperf/energy.txt

@@ -0,0 +1 @@
+62

pyperf/read_file.c

@@ -0,0 +1,29 @@
+#define _GNU_SOURCE
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+unsigned long long int readen(char *path) {
+    char *line = NULL;
+    size_t len = 0;
+    ssize_t read;
+    unsigned long long int data;
+
+    FILE *fd = fopen(path, "r");
+
+    if (fd == NULL)
+      exit(EXIT_FAILURE);
+
+    while ((read = getline(&line, &len, fd)) != -1) {
+      //Do nothing.
+    }
+
+    data = strtoull(line, NULL, 10);
+
+    if (line)
+      free(line);
+
+    fclose(fd);
+
+    return data;  
+}