1 ##############################################################################
2 # Copyright (c) 2016 Huawei Technologies Co.,Ltd and others.
4 # All rights reserved. This program and the accompanying materials
5 # are made available under the terms of the Apache License, Version 2.0
6 # which accompanies this distribution, and is available at
7 # http://www.apache.org/licenses/LICENSE-2.0
8 ##############################################################################
9 from __future__ import absolute_import
14 from oslo_serialization import jsonutils
16 import yardstick.ssh as ssh
17 from yardstick.benchmark.scenarios import base
19 LOG = logging.getLogger(__name__)
22 class Ramspeed(base.Scenario):
23 """Execute ramspeed benchmark in a host
24 The ramspeed script takes a number of options which you can use to
25 customise a test. The full run time usage
28 ramspeed -b ID [-g size] [-m size] [-l runs] [-r speed-format]
30 -b runs a specified benchmark (by an ID number):
31 1 -- INTmark [writing] 4 -- FLOATmark [writing]
32 2 -- INTmark [reading] 5 -- FLOATmark [reading]
33 3 -- INTmem 6 -- FLOATmem
34 7 -- MMXmark [writing] 10 -- SSEmark [writing]
35 8 -- MMXmark [reading] 11 -- SSEmark [reading]
36 9 -- MMXmem 12 -- SSEmem
37 13 -- MMXmark (nt) [writing] 16 -- SSEmark (nt) [writing]
38 14 -- MMXmark (nt) [reading] 17 -- SSEmark (nt) [reading]
39 15 -- MMXmem (nt) 18 -- SSEmem (nt)
40 In this scenario, only the first 6 test type will be used for testing.
42 -g specifies a # of Gbytes per pass (default is 8)
43 -m specifies a # of Mbytes per array (default is 32)
44 -l enables the BatchRun mode (for *mem benchmarks only),
45 and specifies a # of runs (suggested is 5)
46 -r displays speeds in real megabytes per second (default: decimal)
48 The -b option is required, others are recommended.
51 type_id - specifies whether to run *mark benchmark or *mem benchmark
52 the type_id can be any number from 1 to 19
56 load - specifies a # of Gbytes per pass
60 block_size - specifies a # of Mbytes per array
65 Parameters for *mem benchmark
66 iteration - specifies a # of runs for each test
70 more info http://alasir.com/software/ramspeed
72 __scenario_type__ = "Ramspeed"
74 RAMSPEED_MARK_SCRIPT = "ramspeed_mark_benchmark.bash"
75 RAMSPEED_MEM_SCRIPT = "ramspeed_mem_benchmark.bash"
77 def __init__(self, scenario_cfg, context_cfg):
78 self.scenario_cfg = scenario_cfg
79 self.context_cfg = context_cfg
80 self.setup_done = False
84 self.mark_target_script = pkg_resources.resource_filename(
85 "yardstick.benchmark.scenarios.compute",
86 Ramspeed.RAMSPEED_MARK_SCRIPT)
87 self.mem_target_script = pkg_resources.resource_filename(
88 "yardstick.benchmark.scenarios.compute",
89 Ramspeed.RAMSPEED_MEM_SCRIPT)
91 host = self.context_cfg["host"]
93 self.client = ssh.SSH.from_node(host, defaults={"user": "ubuntu"})
94 self.client.wait(timeout=600)
96 # copy scripts to host
97 self.client._put_file_shell(
98 self.mark_target_script, '~/ramspeed_mark_benchmark.sh')
99 self.client._put_file_shell(
100 self.mem_target_script, '~/ramspeed_mem_benchmark.sh')
101 self.setup_done = True
103 def run(self, result):
104 """execute the benchmark"""
106 if not self.setup_done:
109 options = self.scenario_cfg['options']
110 test_id = options.get('type_id', 1)
111 load = options.get('load', 8)
112 block_size = options.get('block_size', 32)
114 if test_id == 3 or test_id == 6:
115 iteration = options.get('iteration', 1)
116 cmd = "sudo bash ramspeed_mem_benchmark.sh %d %d %d %d" % \
117 (test_id, load, block_size, iteration)
118 elif 0 < test_id <= 5:
119 cmd = "sudo bash ramspeed_mark_benchmark.sh %d %d %d" % \
120 (test_id, load, block_size)
121 # only the test_id 1-6 will be used in this scenario
123 raise RuntimeError("No such type_id: %s for Ramspeed scenario",
126 LOG.debug("Executing command: %s", cmd)
127 status, stdout, stderr = self.client.execute(cmd)
129 raise RuntimeError(stderr)
131 result.update(jsonutils.loads(stdout))
133 if "sla" in self.scenario_cfg:
135 sla_min_bw = int(self.scenario_cfg['sla']['min_bandwidth'])
136 for i in result["Result"]:
137 bw = i["Bandwidth(MBps)"]
139 sla_error += "Bandwidth %f < " \
140 "sla:min_bandwidth(%f)" % (bw, sla_min_bw)
141 assert sla_error == "", sla_error