1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
#!/usr/bin/env python3
# Copyright 2023 the V8 project authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can
# be found in the LICENSE file.
from pathlib import Path
import argparse
import subprocess
import sys
def main():
args = parse_arguments()
run_benchmark(args.benchmark_path, args.d8_path, args.output_dir)
run_get_hints(args.output_dir, args.profile_name)
def parse_arguments():
parser = argparse.ArgumentParser(
description=('Generate builtin PGO profiles. '
'The script is designed to run in swarming context where '
'the isolate aready contains the instrumented binary.'))
parser.add_argument(
'--profile-name',
default='pgo',
help='target cpu to build the profile for: x64 or arm64')
parser.add_argument(
'--benchmark_path',
default=Path('./JetStream2/cli.js'),
help='path to benchmark runner .js file, usually JetStream2\'s `cli.js`',
type=Path)
parser.add_argument(
'--d8-path',
default=Path('./out/build/d8'),
help=('path to the d8 executable, by default `./out/build/d8` in '
'swarming context'),
type=Path)
parser.add_argument('--output-dir', type=Path)
return parser.parse_args()
def run_benchmark(benchmark_path, d8_path, output_dir):
root_dir = tools_pgo_dir().parent.parent
benchmark_dir = (root_dir / benchmark_path).parent.absolute()
assert benchmark_dir.exists(), "Could not find benchmark path!"
benchmark_file = benchmark_path.name
d8_path_abs = (root_dir / d8_path).absolute()
assert d8_path_abs.exists(), "Could not find d8 path!"
log_path = benchmark_log_path(output_dir)
cmd = [d8_path_abs, f"--turbo-profiling-output={log_path}", benchmark_file]
run(cmd, cwd=benchmark_dir)
assert log_path.exists(), "Could not find benchmark logs path!"
def tools_pgo_dir():
return Path(__file__).parent
def benchmark_log_path(output_dir):
return (output_dir / "v8.builtins.pgo").absolute()
def run_get_hints(output_dir, profile_name):
get_hints_path = (tools_pgo_dir() / "get_hints.py").absolute()
assert get_hints_path.exists(), "Could not find get_hints.py script path!"
profile_path = (output_dir / f"{profile_name}.profile").absolute()
run([
sys.executable, '-u', get_hints_path,
benchmark_log_path(output_dir), profile_path
])
assert profile_path.exists(), "Could not find profile path!"
def run(cmd, **kwargs):
print(f"# CMD: {cmd} {kwargs}")
subprocess.run(cmd, **kwargs, check=True)
if __name__ == "__main__": # pragma: no cover
sys.exit(main())
|
