Yunlian Jiang | 04dc5dc | 2013-04-23 15:05:05 -0700 | [diff] [blame] | 1 | #!/usr/bin/python |
| 2 | |
| 3 | # Copyright (c) 2013 The Chromium OS Authors. All rights reserved. |
| 4 | # Use of this source code is governed by a BSD-style license that can be |
| 5 | # found in the LICENSE file. |
| 6 | |
| 7 | import os |
| 8 | import time |
cmtice | e5a746f | 2013-11-25 14:57:10 -0500 | [diff] [blame] | 9 | import shlex |
Yunlian Jiang | 04dc5dc | 2013-04-23 15:05:05 -0700 | [diff] [blame] | 10 | |
| 11 | from utils import command_executer |
Caroline Tice | 7057cf6 | 2015-12-10 12:09:40 -0800 | [diff] [blame^] | 12 | import test_flag |
Yunlian Jiang | 04dc5dc | 2013-04-23 15:05:05 -0700 | [diff] [blame] | 13 | |
Caroline Tice | 9277419 | 2013-09-10 16:29:18 -0700 | [diff] [blame] | 14 | TEST_THAT_PATH = '/usr/bin/test_that' |
| 15 | CHROME_MOUNT_DIR = '/tmp/chrome_root' |
| 16 | |
cmtice | 4f0309d | 2014-06-15 13:36:05 -0700 | [diff] [blame] | 17 | def GetProfilerArgs (profiler_args): |
cmtice | e5a746f | 2013-11-25 14:57:10 -0500 | [diff] [blame] | 18 | # Remove "--" from in front of profiler args. |
| 19 | args_list = shlex.split(profiler_args) |
| 20 | new_list = [] |
| 21 | for arg in args_list: |
| 22 | if arg[0:2] == '--': |
| 23 | arg = arg[2:] |
| 24 | new_list.append(arg) |
| 25 | args_list = new_list |
| 26 | |
| 27 | # Remove "perf_options=" from middle of profiler args. |
| 28 | new_list = [] |
| 29 | for arg in args_list: |
| 30 | idx = arg.find("perf_options=") |
| 31 | if idx != -1: |
| 32 | prefix = arg[0:idx] |
| 33 | suffix = arg[idx + len("perf_options=") + 1 : -1] |
| 34 | new_arg = prefix + "'" + suffix + "'" |
| 35 | new_list.append(new_arg) |
| 36 | else: |
| 37 | new_list.append(arg) |
| 38 | args_list = new_list |
| 39 | |
| 40 | return " ".join(args_list) |
Caroline Tice | 9277419 | 2013-09-10 16:29:18 -0700 | [diff] [blame] | 41 | |
Yunlian Jiang | 04dc5dc | 2013-04-23 15:05:05 -0700 | [diff] [blame] | 42 | |
| 43 | class SuiteRunner(object): |
| 44 | """ This defines the interface from crosperf to test script. |
| 45 | """ |
Caroline Tice | 9277419 | 2013-09-10 16:29:18 -0700 | [diff] [blame] | 46 | |
cmtice | 4f0309d | 2014-06-15 13:36:05 -0700 | [diff] [blame] | 47 | def __init__(self, logger_to_use=None, log_level="verbose", cmd_exec=None, |
| 48 | cmd_term=None): |
Yunlian Jiang | 04dc5dc | 2013-04-23 15:05:05 -0700 | [diff] [blame] | 49 | self._logger = logger_to_use |
cmtice | 1390924 | 2014-03-11 13:38:07 -0700 | [diff] [blame] | 50 | self.log_level = log_level |
cmtice | 4f0309d | 2014-06-15 13:36:05 -0700 | [diff] [blame] | 51 | self._ce = cmd_exec or command_executer.GetCommandExecuter(self._logger, |
cmtice | 1390924 | 2014-03-11 13:38:07 -0700 | [diff] [blame] | 52 | log_level=self.log_level) |
cmtice | 4f0309d | 2014-06-15 13:36:05 -0700 | [diff] [blame] | 53 | self._ct = cmd_term or command_executer.CommandTerminator() |
Yunlian Jiang | 04dc5dc | 2013-04-23 15:05:05 -0700 | [diff] [blame] | 54 | |
cmtice | e5a746f | 2013-11-25 14:57:10 -0500 | [diff] [blame] | 55 | def Run(self, machine, label, benchmark, test_args, profiler_args): |
Luis Lozano | df76222 | 2015-05-19 12:22:37 -0700 | [diff] [blame] | 56 | for i in range(0, benchmark.retries + 1): |
| 57 | self.PinGovernorExecutionFrequencies(machine, label.chromeos_root) |
| 58 | if benchmark.suite == "telemetry": |
| 59 | ret_tup = self.Telemetry_Run(machine, label, benchmark, profiler_args) |
| 60 | elif benchmark.suite == "telemetry_Crosperf": |
| 61 | ret_tup = self.Telemetry_Crosperf_Run(machine, label, benchmark, |
| 62 | test_args, profiler_args) |
| 63 | else: |
| 64 | ret_tup = self.Test_That_Run(machine, label, benchmark, test_args, |
| 65 | profiler_args) |
| 66 | if ret_tup[0] != 0: |
| 67 | self._logger.LogOutput("benchmark %s failed. Retries left: %s" |
| 68 | % (benchmark.name, benchmark.retries - i)) |
| 69 | elif i > 0: |
| 70 | self._logger.LogOutput("benchmark %s succeded after %s retries" |
| 71 | % (benchmark.name, i)) |
| 72 | break |
| 73 | else: |
| 74 | self._logger.LogOutput("benchmark %s succeded on first try" |
| 75 | % benchmark.name) |
| 76 | break |
| 77 | return ret_tup |
Yunlian Jiang | 04dc5dc | 2013-04-23 15:05:05 -0700 | [diff] [blame] | 78 | |
Luis Lozano | 53c88e9 | 2013-10-08 15:15:48 -0700 | [diff] [blame] | 79 | def GetHighestStaticFrequency(self, machine_name, chromeos_root): |
| 80 | """ Gets the highest static frequency for the specified machine |
| 81 | """ |
Han Shen | fd0b178 | 2014-02-12 15:13:01 -0800 | [diff] [blame] | 82 | get_avail_freqs = ("cd /sys/devices/system/cpu/cpu0/cpufreq/; " |
| 83 | "if [[ -e scaling_available_frequencies ]]; then " |
| 84 | " cat scaling_available_frequencies; " |
| 85 | "else " |
| 86 | " cat scaling_max_freq ; " |
| 87 | "fi") |
Luis Lozano | 53c88e9 | 2013-10-08 15:15:48 -0700 | [diff] [blame] | 88 | ret, freqs_str, _ = self._ce.CrosRunCommand( |
| 89 | get_avail_freqs, return_output=True, machine=machine_name, |
| 90 | chromeos_root=chromeos_root) |
| 91 | self._logger.LogFatalIf(ret, "Could not get available frequencies " |
| 92 | "from machine: %s" % machine_name) |
| 93 | freqs = freqs_str.split() |
Luis Lozano | d0b4612 | 2015-12-07 15:09:33 -0800 | [diff] [blame] | 94 | # We need to make sure that the frequencies are sorted in decreasing |
| 95 | # order |
| 96 | freqs.sort(key=int, reverse=True) |
| 97 | |
Han Shen | fd0b178 | 2014-02-12 15:13:01 -0800 | [diff] [blame] | 98 | ## When there is no scaling_available_frequencies file, |
| 99 | ## we have only 1 choice. |
| 100 | if len(freqs) == 1: |
| 101 | return freqs[0] |
Luis Lozano | 53c88e9 | 2013-10-08 15:15:48 -0700 | [diff] [blame] | 102 | # The dynamic frequency ends with a "1000". So, ignore it if found. |
| 103 | if freqs[0].endswith("1000"): |
| 104 | return freqs[1] |
| 105 | else: |
| 106 | return freqs[0] |
| 107 | |
| 108 | def PinGovernorExecutionFrequencies(self, machine_name, chromeos_root): |
| 109 | """ Set min and max frequencies to max static frequency |
| 110 | """ |
| 111 | highest_freq = self.GetHighestStaticFrequency(machine_name, chromeos_root) |
| 112 | BASH_FOR = "for f in {list}; do {body}; done" |
| 113 | CPUFREQ_DIRS = "/sys/devices/system/cpu/cpu*/cpufreq/" |
| 114 | change_max_freq = BASH_FOR.format(list=CPUFREQ_DIRS + "scaling_max_freq", |
| 115 | body="echo %s > $f" % highest_freq) |
| 116 | change_min_freq = BASH_FOR.format(list=CPUFREQ_DIRS + "scaling_min_freq", |
| 117 | body="echo %s > $f" % highest_freq) |
| 118 | change_perf_gov = BASH_FOR.format(list=CPUFREQ_DIRS + "scaling_governor", |
| 119 | body="echo performance > $f") |
cmtice | 1390924 | 2014-03-11 13:38:07 -0700 | [diff] [blame] | 120 | if self.log_level == "average": |
| 121 | self._logger.LogOutput("Pinning governor execution frequencies for %s" |
| 122 | % machine_name) |
Luis Lozano | 53c88e9 | 2013-10-08 15:15:48 -0700 | [diff] [blame] | 123 | ret = self._ce.CrosRunCommand(" && ".join(("set -e ", |
| 124 | change_max_freq, |
| 125 | change_min_freq, |
| 126 | change_perf_gov)), |
| 127 | machine=machine_name, |
| 128 | chromeos_root=chromeos_root) |
| 129 | self._logger.LogFatalIf(ret, "Could not pin frequencies on machine: %s" |
| 130 | % machine_name) |
| 131 | |
Yunlian Jiang | 04dc5dc | 2013-04-23 15:05:05 -0700 | [diff] [blame] | 132 | def RebootMachine(self, machine_name, chromeos_root): |
Caroline Tice | 9277419 | 2013-09-10 16:29:18 -0700 | [diff] [blame] | 133 | command = "reboot && exit" |
Yunlian Jiang | 04dc5dc | 2013-04-23 15:05:05 -0700 | [diff] [blame] | 134 | self._ce.CrosRunCommand(command, machine=machine_name, |
| 135 | chromeos_root=chromeos_root) |
| 136 | time.sleep(60) |
Luis Lozano | 53c88e9 | 2013-10-08 15:15:48 -0700 | [diff] [blame] | 137 | # Whenever we reboot the machine, we need to restore the governor settings. |
| 138 | self.PinGovernorExecutionFrequencies(machine_name, chromeos_root) |
Yunlian Jiang | 04dc5dc | 2013-04-23 15:05:05 -0700 | [diff] [blame] | 139 | |
cmtice | e5a746f | 2013-11-25 14:57:10 -0500 | [diff] [blame] | 140 | def Test_That_Run(self, machine, label, benchmark, test_args, profiler_args): |
Caroline Tice | b47bff4 | 2013-08-19 15:59:02 -0700 | [diff] [blame] | 141 | """Run the test_that test..""" |
| 142 | options = "" |
| 143 | if label.board: |
| 144 | options += " --board=%s" % label.board |
| 145 | if test_args: |
| 146 | options += " %s" % test_args |
cmtice | e5a746f | 2013-11-25 14:57:10 -0500 | [diff] [blame] | 147 | if profiler_args: |
cmtice | 226e3e0 | 2014-04-27 22:28:42 -0700 | [diff] [blame] | 148 | self._logger.LogFatal("test_that does not support profiler.") |
Caroline Tice | b47bff4 | 2013-08-19 15:59:02 -0700 | [diff] [blame] | 149 | command = "rm -rf /usr/local/autotest/results/*" |
Luis Lozano | bd44704 | 2015-10-01 14:24:27 -0700 | [diff] [blame] | 150 | self._ce.CrosRunCommand(command, machine=machine, |
Caroline Tice | b47bff4 | 2013-08-19 15:59:02 -0700 | [diff] [blame] | 151 | chromeos_root=label.chromeos_root) |
| 152 | |
cmtice | 98a5369 | 2014-04-16 14:48:47 -0700 | [diff] [blame] | 153 | # We do this because some tests leave the machine in weird states. |
Luis Lozano | 53c88e9 | 2013-10-08 15:15:48 -0700 | [diff] [blame] | 154 | # Rebooting between iterations has proven to help with this. |
Caroline Tice | b47bff4 | 2013-08-19 15:59:02 -0700 | [diff] [blame] | 155 | self.RebootMachine(machine, label.chromeos_root) |
| 156 | |
Caroline Tice | 8df4493 | 2015-12-02 10:58:53 -0800 | [diff] [blame] | 157 | command = (("%s --autotest_dir ~/trunk/src/third_party/autotest/files --fast " |
| 158 | "%s %s %s") % |
Caroline Tice | 9277419 | 2013-09-10 16:29:18 -0700 | [diff] [blame] | 159 | (TEST_THAT_PATH, options, machine, benchmark.test_name)) |
cmtice | 1390924 | 2014-03-11 13:38:07 -0700 | [diff] [blame] | 160 | if self.log_level != "verbose": |
| 161 | self._logger.LogOutput("Running test.") |
| 162 | self._logger.LogOutput("CMD: %s" % command) |
Luis Lozano | 45b53c5 | 2015-09-30 11:36:27 -0700 | [diff] [blame] | 163 | # Use --no-ns-pid so that cros_sdk does not create a different |
| 164 | # process namespace and we can kill process created easily by |
| 165 | # their process group. |
Caroline Tice | b47bff4 | 2013-08-19 15:59:02 -0700 | [diff] [blame] | 166 | return self._ce.ChrootRunCommand(label.chromeos_root, |
| 167 | command, |
| 168 | True, |
Luis Lozano | 45b53c5 | 2015-09-30 11:36:27 -0700 | [diff] [blame] | 169 | self._ct, |
| 170 | cros_sdk_options="--no-ns-pid") |
Caroline Tice | b47bff4 | 2013-08-19 15:59:02 -0700 | [diff] [blame] | 171 | |
cmtice | 146058f | 2015-07-24 14:04:12 -0700 | [diff] [blame] | 172 | def RemoveTelemetryTempFile (self, machine, chromeos_root): |
| 173 | filename = "telemetry@%s" % machine |
| 174 | fullname = os.path.join (chromeos_root, |
| 175 | "chroot", |
| 176 | "tmp", |
| 177 | filename) |
| 178 | if os.path.exists(fullname): |
| 179 | os.remove(fullname) |
Caroline Tice | 9277419 | 2013-09-10 16:29:18 -0700 | [diff] [blame] | 180 | |
cmtice | e5a746f | 2013-11-25 14:57:10 -0500 | [diff] [blame] | 181 | def Telemetry_Crosperf_Run (self, machine, label, benchmark, test_args, |
| 182 | profiler_args): |
Caroline Tice | 9277419 | 2013-09-10 16:29:18 -0700 | [diff] [blame] | 183 | if not os.path.isdir(label.chrome_src): |
| 184 | self._logger.LogFatal("Cannot find chrome src dir to" |
cmtice | 5f3ccbd | 2013-11-20 16:36:04 -0800 | [diff] [blame] | 185 | " run telemetry: %s" % label.chrome_src) |
Caroline Tice | 9277419 | 2013-09-10 16:29:18 -0700 | [diff] [blame] | 186 | |
cmtice | 146058f | 2015-07-24 14:04:12 -0700 | [diff] [blame] | 187 | # Check for and remove temporary file that may have been left by |
| 188 | # previous telemetry runs (and which might prevent this run from |
| 189 | # working). |
| 190 | self.RemoveTelemetryTempFile (machine, label.chromeos_root) |
| 191 | |
Luis Lozano | 004bccb | 2015-03-20 18:19:58 -0700 | [diff] [blame] | 192 | # For telemetry runs, we can use the autotest copy from the source |
| 193 | # location. No need to have one under /build/<board>. |
| 194 | autotest_dir_arg = '--autotest_dir ~/trunk/src/third_party/autotest/files' |
| 195 | |
cmtice | 4f0309d | 2014-06-15 13:36:05 -0700 | [diff] [blame] | 196 | profiler_args = GetProfilerArgs (profiler_args) |
Luis Lozano | 5d7e21d | 2015-03-20 17:57:36 -0700 | [diff] [blame] | 197 | fast_arg = "" |
| 198 | if not profiler_args: |
| 199 | # --fast works unless we are doing profiling (autotest limitation). |
| 200 | # --fast avoids unnecessary copies of syslogs. |
| 201 | fast_arg = "--fast" |
cmtice | e5a746f | 2013-11-25 14:57:10 -0500 | [diff] [blame] | 202 | args_string = "" |
| 203 | if test_args: |
| 204 | # Strip double quotes off args (so we can wrap them in single |
| 205 | # quotes, to pass through to Telemetry). |
| 206 | if test_args[0] == '"' and test_args[-1] == '"': |
| 207 | test_args = test_args[1:-1] |
| 208 | args_string = "test_args='%s'" % test_args |
Luis Lozano | 004bccb | 2015-03-20 18:19:58 -0700 | [diff] [blame] | 209 | |
Ting-Yuan Huang | bc2d3d1 | 2015-07-02 13:09:03 +0800 | [diff] [blame] | 210 | cmd = ('{} {} {} --board={} --args="{} run_local={} test={} ' |
Luis Lozano | 5d7e21d | 2015-03-20 17:57:36 -0700 | [diff] [blame] | 211 | '{}" {} telemetry_Crosperf'.format(TEST_THAT_PATH, |
Luis Lozano | 004bccb | 2015-03-20 18:19:58 -0700 | [diff] [blame] | 212 | autotest_dir_arg, |
Luis Lozano | 5d7e21d | 2015-03-20 17:57:36 -0700 | [diff] [blame] | 213 | fast_arg, |
| 214 | label.board, |
| 215 | args_string, |
Ting-Yuan Huang | bc2d3d1 | 2015-07-02 13:09:03 +0800 | [diff] [blame] | 216 | benchmark.run_local, |
Luis Lozano | 5d7e21d | 2015-03-20 17:57:36 -0700 | [diff] [blame] | 217 | benchmark.test_name, |
| 218 | profiler_args, |
| 219 | machine)) |
| 220 | |
Luis Lozano | 45b53c5 | 2015-09-30 11:36:27 -0700 | [diff] [blame] | 221 | # Use --no-ns-pid so that cros_sdk does not create a different |
| 222 | # process namespace and we can kill process created easily by their |
| 223 | # process group. |
| 224 | chrome_root_options = ("--no-ns-pid " |
| 225 | "--chrome_root={} --chrome_root_mount={} " |
| 226 | "FEATURES=\"-usersandbox\" " |
Luis Lozano | 5d7e21d | 2015-03-20 17:57:36 -0700 | [diff] [blame] | 227 | "CHROME_ROOT={}".format(label.chrome_src, |
| 228 | CHROME_MOUNT_DIR, |
| 229 | CHROME_MOUNT_DIR)) |
cmtice | 1390924 | 2014-03-11 13:38:07 -0700 | [diff] [blame] | 230 | if self.log_level != "verbose": |
| 231 | self._logger.LogOutput("Running test.") |
| 232 | self._logger.LogOutput("CMD: %s" % cmd) |
Luis Lozano | 45b53c5 | 2015-09-30 11:36:27 -0700 | [diff] [blame] | 233 | return self._ce.ChrootRunCommand(label.chromeos_root, |
| 234 | cmd, |
| 235 | return_output=True, |
| 236 | command_terminator=self._ct, |
| 237 | cros_sdk_options=chrome_root_options) |
Caroline Tice | 9277419 | 2013-09-10 16:29:18 -0700 | [diff] [blame] | 238 | |
| 239 | |
cmtice | 226e3e0 | 2014-04-27 22:28:42 -0700 | [diff] [blame] | 240 | def Telemetry_Run(self, machine, label, benchmark, profiler_args): |
cmtice | 98a5369 | 2014-04-16 14:48:47 -0700 | [diff] [blame] | 241 | telemetry_run_path = "" |
Yunlian Jiang | 04dc5dc | 2013-04-23 15:05:05 -0700 | [diff] [blame] | 242 | if not os.path.isdir(label.chrome_src): |
Caroline Tice | 6aa8528 | 2013-08-06 16:02:04 -0700 | [diff] [blame] | 243 | self._logger.LogFatal("Cannot find chrome src dir to" |
cmtice | 98a5369 | 2014-04-16 14:48:47 -0700 | [diff] [blame] | 244 | " run telemetry.") |
| 245 | else: |
| 246 | telemetry_run_path = os.path.join(label.chrome_src, "src/tools/perf") |
| 247 | if not os.path.exists(telemetry_run_path): |
| 248 | self._logger.LogFatal("Cannot find %s directory." % telemetry_run_path) |
| 249 | |
cmtice | 226e3e0 | 2014-04-27 22:28:42 -0700 | [diff] [blame] | 250 | if profiler_args: |
| 251 | self._logger.LogFatal("Telemetry does not support the perf profiler.") |
| 252 | |
cmtice | 146058f | 2015-07-24 14:04:12 -0700 | [diff] [blame] | 253 | # Check for and remove temporary file that may have been left by |
| 254 | # previous telemetry runs (and which might prevent this run from |
| 255 | # working). |
Caroline Tice | 7057cf6 | 2015-12-10 12:09:40 -0800 | [diff] [blame^] | 256 | if not test_flag.GetTestMode(): |
| 257 | self.RemoveTelemetryTempFile (machine, label.chromeos_root) |
cmtice | 146058f | 2015-07-24 14:04:12 -0700 | [diff] [blame] | 258 | |
Yunlian Jiang | 04dc5dc | 2013-04-23 15:05:05 -0700 | [diff] [blame] | 259 | rsa_key = os.path.join(label.chromeos_root, |
| 260 | "src/scripts/mod_for_test_scripts/ssh_keys/testing_rsa") |
| 261 | |
| 262 | cmd = ("cd {0} && " |
cmtice | 98a5369 | 2014-04-16 14:48:47 -0700 | [diff] [blame] | 263 | "./run_measurement " |
Yunlian Jiang | 04dc5dc | 2013-04-23 15:05:05 -0700 | [diff] [blame] | 264 | "--browser=cros-chrome " |
| 265 | "--output-format=csv " |
| 266 | "--remote={1} " |
| 267 | "--identity {2} " |
cmtice | 98a5369 | 2014-04-16 14:48:47 -0700 | [diff] [blame] | 268 | "{3} {4}".format(telemetry_run_path, machine, |
Yunlian Jiang | 04dc5dc | 2013-04-23 15:05:05 -0700 | [diff] [blame] | 269 | rsa_key, |
| 270 | benchmark.test_name, |
| 271 | benchmark.test_args)) |
cmtice | 1390924 | 2014-03-11 13:38:07 -0700 | [diff] [blame] | 272 | if self.log_level != "verbose": |
| 273 | self._logger.LogOutput("Running test.") |
| 274 | self._logger.LogOutput("CMD: %s" % cmd) |
Yunlian Jiang | 04dc5dc | 2013-04-23 15:05:05 -0700 | [diff] [blame] | 275 | return self._ce.RunCommand(cmd, return_output=True, |
| 276 | print_to_console=False) |
| 277 | |
| 278 | def Terminate(self): |
| 279 | self._ct.Terminate() |
| 280 | |
| 281 | |
| 282 | class MockSuiteRunner(object): |
| 283 | def __init__(self): |
Caroline Tice | 9277419 | 2013-09-10 16:29:18 -0700 | [diff] [blame] | 284 | self._true = True |
Yunlian Jiang | 04dc5dc | 2013-04-23 15:05:05 -0700 | [diff] [blame] | 285 | |
Caroline Tice | 9277419 | 2013-09-10 16:29:18 -0700 | [diff] [blame] | 286 | def Run(self, *_args): |
| 287 | if self._true: |
Luis Lozano | df76222 | 2015-05-19 12:22:37 -0700 | [diff] [blame] | 288 | return [0, "", ""] |
Caroline Tice | 9277419 | 2013-09-10 16:29:18 -0700 | [diff] [blame] | 289 | else: |
Luis Lozano | df76222 | 2015-05-19 12:22:37 -0700 | [diff] [blame] | 290 | return [0, "", ""] |