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 ##############################################################################
13 import yardstick.ssh as ssh
14 from yardstick.benchmark.scenarios import base
16 LOG = logging.getLogger(__name__)
19 class Ramspeed(base.Scenario):
20 """Execute ramspeed benchmark in a host
21 The ramspeed script takes a number of options which you can use to
22 customise a test. The full run time usage
25 ramspeed -b ID [-g size] [-m size] [-l runs] [-r speed-format]
27 -b runs a specified benchmark (by an ID number):
28 1 -- INTmark [writing] 4 -- FLOATmark [writing]
29 2 -- INTmark [reading] 5 -- FLOATmark [reading]
30 3 -- INTmem 6 -- FLOATmem
31 7 -- MMXmark [writing] 10 -- SSEmark [writing]
32 8 -- MMXmark [reading] 11 -- SSEmark [reading]
33 9 -- MMXmem 12 -- SSEmem
34 13 -- MMXmark (nt) [writing] 16 -- SSEmark (nt) [writing]
35 14 -- MMXmark (nt) [reading] 17 -- SSEmark (nt) [reading]
36 15 -- MMXmem (nt) 18 -- SSEmem (nt)
37 In this scenario, only the first 6 test type will be used for testing.
39 -g specifies a # of Gbytes per pass (default is 8)
40 -m specifies a # of Mbytes per array (default is 32)
41 -l enables the BatchRun mode (for *mem benchmarks only),
42 and specifies a # of runs (suggested is 5)
43 -r displays speeds in real megabytes per second (default: decimal)
45 The -b option is required, others are recommended.
48 type_id - specifies whether to run *mark benchmark or *mem benchmark
49 the type_id can be any number from 1 to 19
53 load - specifies a # of Gbytes per pass
57 block_size - specifies a # of Mbytes per array
62 Parameters for *mem benchmark
63 iteration - specifies a # of runs for each test
67 more info http://alasir.com/software/ramspeed
69 __scenario_type__ = "Ramspeed"
71 RAMSPEED_MARK_SCRIPT = "ramspeed_mark_benchmark.bash"
72 RAMSPEED_MEM_SCRIPT = "ramspeed_mem_benchmark.bash"
74 def __init__(self, scenario_cfg, context_cfg):
75 self.scenario_cfg = scenario_cfg
76 self.context_cfg = context_cfg
77 self.setup_done = False
81 self.mark_target_script = pkg_resources.resource_filename(
82 "yardstick.benchmark.scenarios.compute",
83 Ramspeed.RAMSPEED_MARK_SCRIPT)
84 self.mem_target_script = pkg_resources.resource_filename(
85 "yardstick.benchmark.scenarios.compute",
86 Ramspeed.RAMSPEED_MEM_SCRIPT)
88 host = self.context_cfg["host"]
89 user = host.get("user", "ubuntu")
90 ssh_port = host.get("ssh_port", ssh.DEFAULT_PORT)
91 ip = host.get("ip", None)
92 key_filename = host.get('key_filename', "~/.ssh/id_rsa")
94 LOG.info("user:%s, host:%s", user, ip)
95 self.client = ssh.SSH(user, ip, key_filename=key_filename,
97 self.client.wait(timeout=600)
99 # copy scripts to host
100 self.client._put_file_shell(
101 self.mark_target_script, '~/ramspeed_mark_benchmark.sh')
102 self.client._put_file_shell(
103 self.mem_target_script, '~/ramspeed_mem_benchmark.sh')
104 self.setup_done = True
106 def run(self, result):
107 """execute the benchmark"""
109 if not self.setup_done:
112 options = self.scenario_cfg['options']
113 test_id = options.get('type_id', 1)
114 load = options.get('load', 8)
115 block_size = options.get('block_size', 32)
117 if test_id == 3 or test_id == 6:
118 iteration = options.get('iteration', 1)
119 cmd = "sudo bash ramspeed_mem_benchmark.sh %d %d %d %d" % \
120 (test_id, load, block_size, iteration)
121 elif 0 < test_id <= 5:
122 cmd = "sudo bash ramspeed_mark_benchmark.sh %d %d %d" % \
123 (test_id, load, block_size)
124 # only the test_id 1-6 will be used in this scenario
126 raise RuntimeError("No such type_id: %s for Ramspeed scenario",
129 LOG.debug("Executing command: %s", cmd)
130 status, stdout, stderr = self.client.execute(cmd)
132 raise RuntimeError(stderr)
134 result.update(json.loads(stdout))
136 if "sla" in self.scenario_cfg:
138 sla_min_bw = int(self.scenario_cfg['sla']['min_bandwidth'])
139 for i in result["Result"]:
140 bw = i["Bandwidth(MBps)"]
142 sla_error += "Bandwidth %f < " \
143 "sla:min_bandwidth(%f)" % (bw, sla_min_bw)
144 assert sla_error == "", sla_error