Stats C0 leverage

expetator/leverages/nodeepsleep.py

+import os
+
+source_script = """#!/usr/bin/python3
+
+import signal
+import os
+
+def signal_handler(signal_number, frame):
+    os.close(fd)
+
+fd = os.open("/dev/cpu_dma_latency", os.O_RDWR)
+os.write(fd, b'\0\0\0\0')
+
+signal.signal(signal.SIGINT, signal_handler)
+signal.pause()
+"""
+
+class Nodeepsleep:
+    'Activates or deactivates CX'
+    def __init__(self, modes = {True, False}):
+        self.modes = modes
+        self.binary = '/tmp/bin/script_nodeepsleep.py'
+        self.cur_mode = False
+        
+    def build(self, executor):
+        'Builds the software'
+        self.executor = executor
+        with open(self.binary, 'w') as out_file:
+            out_file.write(script)
+        os.chmod(self.binary, 0o777)
+
+    def available_states(self):
+        'Returns all the modes'
+        return self.modes
+
+    def start(self, mode):
+        'Sets the right mode on all hosts'
+        self.curr_mode = mode
+        if mode:
+            self.executor.hosts(self.binary+' &', root=True)
+
+    def stop(self, output_file=None):
+        'Reverts to the maximum frequency'
+        self.executor.hosts('killall script_nodeepsleep.py', root=True)
+        self.curr_mode = False
+            
+    def get_state(self):
+        'Returns the current state'
+        return (self.curr_mode)
+
+    def state_to_str(self):
+        'Returns the current min and max frequencies as a string'
+        return '%s' % self.curr_mode
+
+    def get_labels(self):
+        'Returns the label for frequenciess'
+        return 'C0only'

setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setuptools.setup(
     name="expetator",
-    version="0.2.2",
+    version="0.2.3",
     author="Georges Da Costa",
     author_email="georges.da-costa@irit.fr",
     description="A framework for monitoring HPC applications using DVFS",