Add a script to run the PGO benchmarks.

build/win/run_pgo_profiling_benchmarks.py

Index: build/win/run_pgo_profiling_benchmarks.py
diff --git a/build/win/run_pgo_profiling_benchmarks.py b/build/win/run_pgo_profiling_benchmarks.py
new file mode 100644
index 0000000000000000000000000000000000000000..49df03567657b7313094d0eb753cef63375679d1
--- /dev/null
+++ b/build/win/run_pgo_profiling_benchmarks.py
@@ -0,0 +1,149 @@
+# Copyright 2016 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+"""Utility script to run the benchmarks during the profiling step of a PGO

[scottmg, 2016/09/27 17:39:30]

nit; blank line after copyright before block comment.

[Sébastien Marchand, 2016/09/27 17:59:56]

Done.

+build.
+"""
+
+import json
+import optparse
+import os
+import subprocess
+import sys
+
+# Make sure that we're running as admin, this is required to run the Telemetry
+# benchmarks.
+from win32com.shell import shell
+if not shell.IsUserAnAdmin():
+  raise Exception('This script has to be run as admin.')
+
+
+_SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__))
+_CHROME_BUILD_DIR = os.path.dirname(_SCRIPT_DIR)
+_CHROME_SRC_DIR = os.path.dirname(_CHROME_BUILD_DIR)
+
+
+# List of the benchmark that we run during the profiling step.
+_BENCHMARKS_TO_RUN = {
+  'blink_perf.bindings',
+  'blink_perf.canvas',
+  'blink_perf.css',
+  'blink_perf.dom',
+  'blink_perf.paint',
+  'blink_perf.svg',
+  'blink_style.top_25',
+  'dromaeo.cssqueryjquery',
+  'dromaeo.domcoreattr',
+  'dromaeo.domcoremodify',
+  'dromaeo.domcorequery',
+  'dromaeo.domcoretraverse',
+  'dromaeo.jslibattrprototype',
+  'dromaeo.jslibeventprototype',
+  'dromaeo.jslibmodifyprototype',
+  'dromaeo.jslibstyleprototype',
+  'dromaeo.jslibtraversejquery',
+  'dromaeo.jslibtraverseprototype',
+  'indexeddb_perf',
+  'media.tough_video_cases',
+  'octane',
+  'smoothness.top_25_smooth',
+  'speedometer',
+  'sunspider',
+}
+
+
+def FindPgosweep(target_bits):
+  """Find the directory containing pgosweep.exe."""

[scottmg, 2016/09/27 17:39:30]

nit; " target_bits should be 32 or 64." since I probably would have passed "x86"
or "x64". :)

[Sébastien Marchand, 2016/09/27 17:59:55]

Done, added a check as well.

+  win_toolchain_json_file = os.path.join(_CHROME_BUILD_DIR,
+                                         'win_toolchain.json')
+  if not os.path.exists(win_toolchain_json_file):
+    raise Exception('The toolchain JSON file (%s) is missing.' %
+                    win_toolchain_json_file)
+  with open(win_toolchain_json_file) as temp_f:
+    toolchain_data = json.load(temp_f)
+  if not os.path.isdir(toolchain_data['path']):
+    raise Exception('The toolchain JSON file is invalid.')

[scottmg, 2016/09/27 17:39:30]

nit; 'The toolchain JSON file's "path" entry (%s) does not refer to a path' or
something.

[Sébastien Marchand, 2016/09/27 17:59:56]

Done.

+
+  pgo_sweep_dir = os.path.join(toolchain_data['path'], 'VC', 'bin')
+  if target_bits == 64:
+    pgo_sweep_dir = os.path.join(pgo_sweep_dir, 'amd64')
+
+  if not os.path.exists(os.path.join(pgo_sweep_dir, 'pgosweep.exe')):
+    raise Exception('pgosweep.exe is missing from %s.' % pgo_sweep_dir)
+
+  return pgo_sweep_dir
+
+
+def RunBenchmarks(options):
+  """Run the benchmarks."""
+  # Starts by finding the directory containing pgosweep.exe
+  pgo_sweep_dir = FindPgosweep(options.target_bits)
+  if not pgo_sweep_dir:

[scottmg, 2016/09/27 17:39:30]

I don't think this can return empty/None because the error conditions all raise,
so you can just remove this.

[Sébastien Marchand, 2016/09/27 17:59:56]

Ha, good point!

+    return 1
+
+  # Find the run_benchmark script.
+  chrome_run_benchmark_script = os.path.join(_CHROME_SRC_DIR, 'tools',
+                                             'perf', 'run_benchmark')
+  if not os.path.exists(chrome_run_benchmark_script):
+    raise Exception('Unable to find the run_benchmark script '
+                    '(%s doesn\'t exist) ' % chrome_run_benchmark_script)
+
+  # Augment the PATH to make sure that the benchmarking script can find
+  # pgosweep.exe and its runtime libraries.
+  env = os.environ.copy()
+  env['PATH'] = str(os.pathsep.join([pgo_sweep_dir, options.build_dir,
+                                     os.environ['PATH']]))
+  env['PogoSafeMode'] = '1'
+  # Apply a scaling factor of 0.5 to the PGO profiling buffers for the 32-bit
+  # builds, without this the buffers will be too large and the process will
+  # fail to start. See crbug.com/632864#c22.
+  if options.target_bits == 32:
+    env['VCPROFILE_ALLOC_SCALE'] = '0.5'
+
+  # Run all the benchmarks.
+  # TODO(sebmarchand): Make this run in parallel.

[scottmg, 2016/09/27 17:39:30]

[Seems like you wouldn't want to run benchmarks in parallel because it'd thrash
things/change timings? Dunno.]

[Sébastien Marchand, 2016/09/27 17:59:56]

I don't think that it could cause any issue? When I use pgosweep I always
specify the process PID, so it shouldn't be an issue. Making the benchmarks run
in parallel will really help to shorten the profiling step. The 'benchmarking'
part of the benchmarks doesn't matter, I could probably rename this to
run_pgo_profiling_scenarios?

+  for benchmark in _BENCHMARKS_TO_RUN:
+    try:
+      benchmark_command = [
+          sys.executable,
+          chrome_run_benchmark_script,
+          '--browser', options.browser_type,
+        ]
+      # Automatically set the arguments to run this script on a local build.
+      if options.browser_type == 'exact':
+        benchmark_command += [
+          '--browser-executable', os.path.join(options.build_dir, 'chrome.exe')
+        ]
+      benchmark_command += [
+          '--profiler', 'win_pgo_profiler',
+          benchmark
+        ]
+      subprocess.call(benchmark_command, env=env)

[scottmg, 2016/09/27 17:39:30]

check_call maybe?

[Sébastien Marchand, 2016/09/27 17:59:56]

Done.

+    except:
+      continue

[scottmg, 2016/09/27 17:39:30]

log something about exception maybe?

[Sébastien Marchand, 2016/09/27 17:59:55]

Done.

+
+  return 0
+
+
+def main():
+  parser = optparse.OptionParser(usage='%prog [options]')
+  parser.add_option(
+      '--browser-type', help='The browser type (to be passed to Telemetry\'s '
+                              'benchmark runner).')
+  # TODO(sebmarchand): Parse the args.gn file to automatically set this value.
+  parser.add_option('--target-bits', help='The target\'s bitness.', type=int)

[scottmg, 2016/09/27 17:39:30]

Maybe "target_cpu" and make it a string that's "x86" or "x64" to make it match
GN if you plan to do this.

[Sébastien Marchand, 2016/09/27 17:59:56]

Done.

+  parser.add_option('--build-dir', help='Chrome build directory.')
+  options, _ = parser.parse_args()
+
+  if not options.target_bits:
+    parser.error('--target-bits is required')
+  if not options.build_dir:
+    parser.error('--build-dir is required')
+  if not options.browser_type:
+    options.browser_type = 'exact'
+
+  return RunBenchmarks(options)
+
+
+if __name__ == '__main__':
+  sys.exit(main())