forked from pz4kybsvg/Conception
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
96 lines
3.4 KiB
96 lines
3.4 KiB
2 years ago
|
load("@drake//tools/skylark:drake_cc.bzl", "drake_cc_binary", "drake_cc_test")
|
||
|
|
load("@drake//tools/skylark:py.bzl", "py_binary")
|
||
|
|
load("@drake//tools/workspace:generate_file.bzl", "generate_file")
|
||
|
|
|
||
|
|
# This file provides build system sugar for crafting benchmarking programs.
|
||
|
|
|
||
|
|
def drake_cc_googlebench_binary(
|
||
|
|
name,
|
||
|
|
*,
|
||
|
|
srcs,
|
||
|
|
deps,
|
||
|
|
data = None,
|
||
|
|
add_test_rule,
|
||
|
|
test_size = "small",
|
||
|
|
test_timeout = None,
|
||
|
|
test_args = None,
|
||
|
|
test_tags = None):
|
||
|
|
"""Declares a testonly binary that uses google benchmark. Automatically
|
||
|
|
adds appropriate deps and ensures it either has an automated smoke test
|
||
|
|
(via 'add_test_rule = True'), or else explicitly opts-out ('= False').
|
||
|
|
"""
|
||
|
|
if not srcs:
|
||
|
|
fail("Missing srcs")
|
||
|
|
if add_test_rule == None:
|
||
|
|
fail("Missing add_test_rule")
|
||
|
|
new_deps = (deps or []) + ["@googlebenchmark//:benchmark"]
|
||
|
|
|
||
|
|
# We need this to be a cc_binary (not a cc_test) so that Bazel's flag
|
||
|
|
# --trim_test_configuration will permit the py_experiment_binary target
|
||
|
|
# to depend on it. In general, non-tests cannot depend on tests, and the
|
||
|
|
# py_experiment_binary cannot be a test since it must not be sandboxed.
|
||
|
|
drake_cc_binary(
|
||
|
|
name = name,
|
||
|
|
srcs = srcs,
|
||
|
|
testonly = True,
|
||
|
|
linkstatic = True,
|
||
|
|
data = data,
|
||
|
|
deps = new_deps,
|
||
|
|
)
|
||
|
|
|
||
|
|
if add_test_rule:
|
||
|
|
# We can't simply use a sh_test wrapper around the cc_binary to run it
|
||
|
|
# as a test, because that would exclude it from ASan, Memcheck, etc.
|
||
|
|
# We'll need to compile it a second time from scratch, unfortunately.
|
||
|
|
drake_cc_test(
|
||
|
|
name = name + "_test",
|
||
|
|
srcs = srcs,
|
||
|
|
data = data,
|
||
|
|
deps = new_deps,
|
||
|
|
size = test_size,
|
||
|
|
timeout = test_timeout,
|
||
|
|
args = [
|
||
|
|
# When running as a unit test, run each function only once to
|
||
|
|
# save time. (Once should be sufficient to prove the lack of
|
||
|
|
# runtime errors.)
|
||
|
|
"--benchmark_min_time=0",
|
||
|
|
] + (test_args or []),
|
||
|
|
tags = test_tags,
|
||
|
|
)
|
||
|
|
|
||
|
|
def drake_py_experiment_binary(name, *, googlebench_binary, **kwargs):
|
||
|
|
"""Declares a testonly binary that wraps google benchmark binary with
|
||
|
|
machinery to run it under controlled conditions and summarize results.
|
||
|
|
"""
|
||
|
|
if not googlebench_binary.startswith(":"):
|
||
|
|
fail("googlebench_binary must be local to this package")
|
||
|
|
dut = "drake/{}/{}".format(native.package_name(), googlebench_binary[1:])
|
||
|
|
template = """
|
||
|
|
import os, sys
|
||
|
|
from bazel_tools.tools.python.runfiles.runfiles import Create
|
||
|
|
runfiles = Create()
|
||
|
|
tool = runfiles.Rlocation("drake/tools/performance/benchmark_tool")
|
||
|
|
dut = runfiles.Rlocation({dut})
|
||
|
|
assert tool and dut
|
||
|
|
env = dict(os.environ)
|
||
|
|
env.update(runfiles.EnvVars())
|
||
|
|
os.execve(tool, [tool, "--binary", dut] + sys.argv[1:], env=env)
|
||
|
|
""".replace("\n ", "\n")
|
||
|
|
generate_file(
|
||
|
|
name = "{}.py".format(name),
|
||
|
|
content = template.format(dut = repr(dut)),
|
||
|
|
)
|
||
|
|
py_binary(
|
||
|
|
name = name,
|
||
|
|
testonly = True,
|
||
|
|
srcs = [":{}.py".format(name)],
|
||
|
|
tags = ["nolint"],
|
||
|
|
data = [
|
||
|
|
googlebench_binary,
|
||
|
|
"//tools/performance:benchmark_tool",
|
||
|
|
],
|
||
|
|
deps = [
|
||
|
|
"@bazel_tools//tools/python/runfiles",
|
||
|
|
],
|
||
|
|
)
|