Mike Frysinger | c7f1593 | 2013-03-20 13:43:35 -0400 | [diff] [blame] | 1 | #!/usr/bin/python |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 2 | # |
| 3 | # Copyright 2010 Google Inc. All Rights Reserved. |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 4 | """Script to run ChromeOS benchmarks |
| 5 | |
| 6 | Inputs: |
| 7 | chromeos_root |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 8 | board |
| 9 | [chromeos/cpu/<benchname>|chromeos/browser/[pagecycler|sunspider]|chromeos/startup] |
| 10 | hostname/IP of Chromeos machine |
| 11 | |
| 12 | chromeos/cpu/<benchname> |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 13 | - Read run script rules from bench.mk perflab-bin, copy benchmark to |
| 14 | host, run |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 15 | and return results. |
| 16 | |
| 17 | chromeos/startup |
| 18 | - Re-image host with image in perflab-bin |
| 19 | - Call run_tests to run startup test, gather results. |
| 20 | - Restore host back to what it was. |
| 21 | |
| 22 | chromeos/browser/* |
| 23 | - Call build_chromebrowser to build image with new browser |
| 24 | - Copy image to perflab-bin |
| 25 | |
| 26 | """ |
| 27 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 28 | __author__ = 'bjanakiraman@google.com (Bhaskar Janakiraman)' |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 29 | |
| 30 | import optparse |
asharif | 253e88b | 2013-02-15 05:15:34 +0000 | [diff] [blame] | 31 | import os |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 32 | import re |
| 33 | import sys |
kbaclawski | 20082a0 | 2013-02-16 02:12:57 +0000 | [diff] [blame] | 34 | |
asharif | 8de2c73 | 2013-02-15 05:15:25 +0000 | [diff] [blame] | 35 | import image_chromeos |
kbaclawski | 20082a0 | 2013-02-16 02:12:57 +0000 | [diff] [blame] | 36 | import run_tests |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 37 | from utils import command_executer |
asharif | 8de2c73 | 2013-02-15 05:15:25 +0000 | [diff] [blame] | 38 | from utils import logger |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 39 | |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 40 | KNOWN_BENCHMARKS = [ |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 41 | 'chromeos/startup', 'chromeos/browser/pagecycler', |
| 42 | 'chromeos/browser/sunspider', 'chromeos/browser/v8bench', |
| 43 | 'chromeos/cpu/bikjmp' |
| 44 | ] |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 45 | |
bjanakiraman | 81a30d0 | 2013-02-15 04:57:20 +0000 | [diff] [blame] | 46 | name_map = { |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 47 | 'pagecycler': 'Page', |
| 48 | 'sunspider': 'SunSpider', |
| 49 | 'v8bench': 'V8Bench', |
| 50 | 'startup': 'BootPerfServer' |
| 51 | } |
bjanakiraman | 81a30d0 | 2013-02-15 04:57:20 +0000 | [diff] [blame] | 52 | |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 53 | # Run command template |
| 54 | |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 55 | # Common initializations |
| 56 | cmd_executer = command_executer.GetCommandExecuter() |
| 57 | |
| 58 | |
| 59 | def Usage(parser, message): |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 60 | print 'ERROR: ' + message |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 61 | parser.print_help() |
| 62 | sys.exit(0) |
| 63 | |
| 64 | |
bjanakiraman | 81a30d0 | 2013-02-15 04:57:20 +0000 | [diff] [blame] | 65 | def RunBrowserBenchmark(chromeos_root, board, bench, workdir, machine): |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 66 | """Run browser benchmarks. |
| 67 | |
| 68 | Args: |
bjanakiraman | 81a30d0 | 2013-02-15 04:57:20 +0000 | [diff] [blame] | 69 | chromeos_root: ChromeOS src dir |
| 70 | board: Board being tested |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 71 | bench: Name of benchmark (chromeos/browser/*) |
| 72 | workdir: Directory containing benchmark directory |
| 73 | machine: name of chromeos machine |
| 74 | """ |
bjanakiraman | 81a30d0 | 2013-02-15 04:57:20 +0000 | [diff] [blame] | 75 | benchname = re.split('/', bench)[2] |
| 76 | benchdir = '%s/%s' % (workdir, benchname) |
| 77 | benchname = name_map[benchname] |
| 78 | retval = run_tests.RunRemoteTests(chromeos_root, machine, board, benchname) |
| 79 | return retval |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 80 | |
| 81 | |
bjanakiraman | 81a30d0 | 2013-02-15 04:57:20 +0000 | [diff] [blame] | 82 | def RunStartupBenchmark(chromeos_root, board, bench, workdir, machine): |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 83 | """Run browser benchmarks. |
| 84 | |
| 85 | Args: |
bjanakiraman | 81a30d0 | 2013-02-15 04:57:20 +0000 | [diff] [blame] | 86 | chromeos_root: ChromeOS src dir |
| 87 | board: Board being tested |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 88 | bench: Name of benchmark (chromeos/browser/*) |
| 89 | workdir: Directory containing benchmark directory |
| 90 | machine: name of chromeos machine |
| 91 | """ |
bjanakiraman | 81a30d0 | 2013-02-15 04:57:20 +0000 | [diff] [blame] | 92 | benchname = 'startup' |
| 93 | benchdir = '%s/%s' % (workdir, benchname) |
| 94 | benchname = name_map[benchname] |
| 95 | retval = run_tests.RunRemoteTests(chromeos_root, machine, board, benchname) |
| 96 | return retval |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 97 | |
| 98 | |
asharif | 4d2b716 | 2013-02-15 05:14:57 +0000 | [diff] [blame] | 99 | def RunCpuBenchmark(chromeos_root, bench, workdir, machine): |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 100 | """Run CPU benchmark. |
| 101 | |
| 102 | Args: |
| 103 | bench: Name of benchmark |
| 104 | workdir: directory containing benchmark directory |
| 105 | machine: name of chromeos machine |
| 106 | |
| 107 | Returns: |
| 108 | status: 0 on success |
| 109 | """ |
| 110 | |
| 111 | benchname = re.split('/', bench)[2] |
| 112 | benchdir = '%s/%s' % (workdir, benchname) |
| 113 | |
| 114 | # Delete any existing run directories on machine. |
| 115 | # Since this has exclusive access to the machine, |
| 116 | # we do not worry about duplicates. |
asharif | 4d2b716 | 2013-02-15 05:14:57 +0000 | [diff] [blame] | 117 | args = 'rm -rf /tmp/%s' % benchname |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 118 | retval = cmd_executer.CrosRunCommand(args, |
| 119 | chromeos_root=chromeos_root, |
asharif | 4d2b716 | 2013-02-15 05:14:57 +0000 | [diff] [blame] | 120 | machine=machine) |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 121 | if retval: |
| 122 | return retval |
| 123 | |
| 124 | # Copy benchmark directory. |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 125 | retval = cmd_executer.CopyFiles(benchdir, |
| 126 | '/tmp/' + benchname, |
| 127 | chromeos_root=chromeos_root, |
| 128 | dest_machine=machine, |
| 129 | dest_cros=True) |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 130 | if retval: |
| 131 | return retval |
| 132 | |
| 133 | # Parse bench.mk to extract run flags. |
| 134 | |
| 135 | benchmk_file = open('%s/bench.mk' % benchdir, 'r') |
| 136 | for line in benchmk_file: |
| 137 | line.rstrip() |
| 138 | if re.match('^run_cmd', line): |
bjanakiraman | 4e8bdf2 | 2013-02-15 04:57:09 +0000 | [diff] [blame] | 139 | line = re.sub('^run_cmd.*\${PERFLAB_PATH}', './out', line) |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 140 | line = re.sub('\${PERFLAB_INPUT}', './data', line) |
| 141 | run_cmd = line |
| 142 | break |
| 143 | |
| 144 | # Execute on remote machine |
| 145 | # Capture output and process it. |
asharif | 4d2b716 | 2013-02-15 05:14:57 +0000 | [diff] [blame] | 146 | sshargs = "'cd /tmp/%s;" % benchname |
| 147 | sshargs += "time -p %s'" % run_cmd |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 148 | cmd_executer.CrosRunCommand(sshargs, |
| 149 | chromeos_root=chromeos_root, |
asharif | 4d2b716 | 2013-02-15 05:14:57 +0000 | [diff] [blame] | 150 | machine=machine) |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 151 | |
| 152 | return retval |
| 153 | |
| 154 | |
| 155 | def Main(argv): |
| 156 | """Build ChromeOS.""" |
| 157 | # Common initializations |
| 158 | |
| 159 | parser = optparse.OptionParser() |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 160 | parser.add_option('-c', |
| 161 | '--chromeos_root', |
| 162 | dest='chromeos_root', |
| 163 | help='Target directory for ChromeOS installation.') |
| 164 | parser.add_option('-m', |
| 165 | '--machine', |
| 166 | dest='machine', |
| 167 | help='The chromeos host machine.') |
| 168 | parser.add_option('--workdir', |
| 169 | dest='workdir', |
| 170 | default='./perflab-bin', |
| 171 | help='Work directory for perflab outputs.') |
| 172 | parser.add_option('--board', |
| 173 | dest='board', |
| 174 | help='ChromeOS target board, e.g. x86-generic') |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 175 | |
| 176 | (options, args) = parser.parse_args(argv[1:]) |
| 177 | |
| 178 | # validate args |
| 179 | for arg in args: |
| 180 | if arg not in KNOWN_BENCHMARKS: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 181 | logger.GetLogger().LogFatal('Bad benchmark %s specified' % arg) |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 182 | |
| 183 | if options.chromeos_root is None: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 184 | Usage(parser, '--chromeos_root must be set') |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 185 | |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 186 | if options.board is None: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 187 | Usage(parser, '--board must be set') |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 188 | |
| 189 | if options.machine is None: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 190 | Usage(parser, '--machine must be set') |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 191 | |
| 192 | found_err = 0 |
| 193 | retval = 0 |
| 194 | for arg in args: |
| 195 | # CPU benchmarks |
asharif | 253e88b | 2013-02-15 05:15:34 +0000 | [diff] [blame] | 196 | comps = re.split('/', arg) |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 197 | if re.match('chromeos/cpu', arg): |
bjanakiraman | 1c19a7f | 2013-02-15 06:31:19 +0000 | [diff] [blame] | 198 | benchname = comps[2] |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 199 | print 'RUNNING %s' % benchname |
| 200 | retval = RunCpuBenchmark(options.chromeos_root, arg, options.workdir, |
| 201 | options.machine) |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 202 | if not found_err: |
| 203 | found_err = retval |
| 204 | elif re.match('chromeos/startup', arg): |
bjanakiraman | 1c19a7f | 2013-02-15 06:31:19 +0000 | [diff] [blame] | 205 | benchname = comps[1] |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 206 | image_args = [ |
| 207 | os.path.dirname(os.path.abspath(__file__)) + '/image_chromeos.py', |
| 208 | '--chromeos_root=' + options.chromeos_root, |
| 209 | '--remote=' + options.machine, '--image=' + options.workdir + '/' + |
| 210 | benchname + '/chromiumos_image.bin' |
| 211 | ] |
| 212 | logger.GetLogger().LogOutput('Reimaging machine %s' % options.machine) |
asharif | 8697d4e | 2013-02-15 09:18:09 +0000 | [diff] [blame] | 213 | image_chromeos.Main(image_args) |
asharif | 8de2c73 | 2013-02-15 05:15:25 +0000 | [diff] [blame] | 214 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 215 | logger.GetLogger().LogOutput('Running %s' % arg) |
| 216 | retval = RunStartupBenchmark(options.chromeos_root, options.board, arg, |
| 217 | options.workdir, options.machine) |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 218 | if not found_err: |
| 219 | found_err = retval |
| 220 | elif re.match('chromeos/browser', arg): |
bjanakiraman | 1c19a7f | 2013-02-15 06:31:19 +0000 | [diff] [blame] | 221 | benchname = comps[2] |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 222 | image_args = [ |
| 223 | os.path.dirname(os.path.abspath(__file__)) + '/image_chromeos.py', |
| 224 | '--chromeos_root=' + options.chromeos_root, |
| 225 | '--remote=' + options.machine, '--image=' + options.workdir + '/' + |
| 226 | benchname + '/chromiumos_image.bin' |
| 227 | ] |
| 228 | logger.GetLogger().LogOutput('Reimaging machine %s' % options.machine) |
asharif | 8697d4e | 2013-02-15 09:18:09 +0000 | [diff] [blame] | 229 | image_chromeos.Main(image_args) |
asharif | 8de2c73 | 2013-02-15 05:15:25 +0000 | [diff] [blame] | 230 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 231 | logger.GetLogger().LogOutput('Running %s' % arg) |
| 232 | retval = RunBrowserBenchmark(options.chromeos_root, options.board, arg, |
| 233 | options.workdir, options.machine) |
bjanakiraman | 229d626 | 2013-02-15 04:56:46 +0000 | [diff] [blame] | 234 | if not found_err: |
| 235 | found_err = retval |
| 236 | |
| 237 | return found_err |
| 238 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 239 | |
| 240 | if __name__ == '__main__': |
asharif | 2198c51 | 2013-02-15 09:21:35 +0000 | [diff] [blame] | 241 | retval = Main(sys.argv) |
| 242 | sys.exit(retval) |