Merge pull request #18 from lisatn/hackbench_workload

Hackbench workload

wlauto/workloads/hackbench/__init__.py

@@ -0,0 +1,89 @@
+#    Copyright 2013-2015 ARM Limited
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# pylint: disable=W0201, C0103
+
+from wlauto import Workload, Parameter, Executable
+
+import os
+import re
+
+hackbench_results_txt = 'hackbench_results.txt'
+
+regex_map = {"total_groups": (re.compile(r'(\d+) groups'), "groups"),
+             "total_fd": (re.compile(r'(\d+) file descriptors'), "file_descriptors"),
+             "total_messages": (re.compile(r'(\d+) messages'), "messages"),
+             "total_bytes": (re.compile(r'(\d+) bytes'), "bytes"),
+             "test_time": (re.compile(r'Time: (\d+.*)'), "seconds")
+            }
+
+class Hackbench(Workload):
+
+    name = 'hackbench'
+    description = """
+    Hackbench runs a series of tests for the Linux scheduler.
+
+    For details, go to:
+    https://github.com/linux-test-project/ltp/
+
+    """
+
+    parameters = [
+        # Workload parameters go here e.g.
+        Parameter('datasize', kind=int, default=100, description='Message size in bytes.'),
+        Parameter('groups', kind=int, default=10, description='Number of groups.'),
+        Parameter('loops', kind=int, default=100, description='Number of loops.'),
+        Parameter('fds', kind=int, default=40, description='Number of file descriptors.'),
+        Parameter('extra_params', kind=str, default='',
+                  description='Extra parameters to pass in. See the hackbench man page'
+                              ' or type `hackbench --help` for list of options.'),
+        Parameter('duration', kind=int, default=30, description='Test duration in seconds.')
+    ]
+
+    def setup(self, context):
+        timeout_buf = 10
+        self.command = '{} -s {} -g {} -l {} {} > {}'
+        self.device_binary = None
+        self.hackbench_result = os.path.join(self.device.working_directory, hackbench_results_txt)
+        self.run_timeout = self.duration + timeout_buf
+
+        self.binary_name = 'hackbench'
+        if not self.device.is_installed(self.binary_name):
+            host_binary = context.resolver.get(Executable(self, self.device.abi, self.binary_name))
+            self.device_binary = self.device.install(host_binary)
+        else:
+            self.device_binary = self.binary_name
+
+        self.command = self.command.format(self.device_binary, self.datasize, self.groups,
+                                           self.loops, self.extra_params, self.hackbench_result)
+
+    def run(self, context):
+        self.device.execute(self.command, timeout=self.run_timeout)
+
+    def update_result(self, context):
+        self.device.pull_file(self.hackbench_result, context.output_directory)
+
+        with open(os.path.join(context.output_directory, hackbench_results_txt)) as hackbench_file:
+            for line in hackbench_file:
+                for label, (regex, units) in regex_map.iteritems():
+                    match = regex.search(line)
+                    if match:
+                        context.result.add_metric(label, float(match.group(1)), units)
+
+    def teardown(self, context):
+        self.device.uninstall_executable(self.binary_name)
+        self.device.execute('rm -f {}'.format(self.hackbench_result))
+
+    def validate(self):
+        pass

wlauto/workloads/hackbench/src/LICENSE

@@ -0,0 +1,3 @@
+Source for these binaries can be obtained here:
+
+http://git.kernel.org/cgit/linux/kernel/git/clrkwllms/rt-tests.git