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.

155 lines
4.5 KiB

#!/bin/bash -e
#
# Copyright (C) 2019 The Android Open Source Project
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# * Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in
# the documentation and/or other materials provided with the
# distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
# FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
# COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
# BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
# OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
# OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
# SUCH DAMAGE.
top=$(cd $(dirname $0) && pwd)
usage() {
cat <<EOF
Usage: $0 [options]
Toolchain:
--cc CC C compiler driver to compiler the benchmark with
Run config:
--cpu-mask MASK MASK is a hex mask of CPU affinity passed to taskset
--linker LINKER Run the benchmark using a specific dynamic linker
--adb Run the test using adb
Timing options:
--multitime N Use multitime to run the benchmark N iterations
--time CMD Use the given time command
Misc:
--keep-tmp-dir Don't delete the temporary directory on exit
Examples:
--cc "\$NDK/toolchains/llvm/prebuilt/linux-x86_64/bin/aarch64-linux-android29-clang
-fuse-ld=lld -Wl,--pack-dyn-relocs=android+relr -Wl,--use-android-relr-tags"
EOF
exit 0
}
cc=
cpu_mask=0
linker=
use_adb=0
time_cmd=
keep_tmp_dir=0
need_arg() {
if [ $2 -eq 0 ]; then
echo "error: $1 needs an argument"
exit 1
fi
}
while [ $# -gt 0 ]; do
arg=$1
shift
case "$arg" in
--cc) need_arg "$arg" "$#"; cc=$1; shift;;
--cpu-mask) need_arg "$arg" "$#"; cpu_mask=$1; shift;;
--linker) need_arg "#arg" "$#"; linker=$1; shift;;
--adb) use_adb=1;;
--multitime) need_arg "$arg" "$#"; time_cmd=--multitime=$1; shift;;
--time) need_arg "$arg" "$#"; time_cmd=--time=$1; shift;;
--keep-tmp-dir) keep_tmp_dir=1;;
--help) usage;;
*)
echo "error: unrecognized option: $arg"
exit 1
;;
esac
done
# If no --cc is specified, use gcc because that's likely available.
if [ "$cc" == "" ]; then
cc='gcc -ldl -pthread'
fi
case "$time_cmd" in
"") time_cmd=time;;
--time=*) time_cmd="${time_cmd#--time=}";;
--multitime=*)
iter="${time_cmd#--multitime=}"
if [ $use_adb -eq 1 ]; then
time_cmd="/data/local/tmp/multitime"
else
time_cmd="multitime"
fi
time_cmd="$time_cmd -n$iter -s0"
;;
esac
# Generate the JSON file as a side effect of regenerating the benchmark.
json_file=$top/tmp/libandroid_servers_arm64.json
if [ ! -e $json_file ]; then
$top/gen_bench.sh --keep-tmp-dir --skip-gen-bench
fi
tmp_dir=$(mktemp -d)
work_dir=$tmp_dir/linker-reloc-bench
python3 -B $top/regen/gen_bench.py --ninja --cc "$cc" $json_file $work_dir
env_setup=()
if [ $use_adb -eq 1 ]; then
target_dir=/data/local/tmp/linker-reloc-bench
adb shell rm -rf $target_dir
(cd $tmp_dir; tar -c linker-reloc-bench/linker_reloc_bench_main linker-reloc-bench/*.so \
| gzip -1 -c | adb shell 'cd /data/local/tmp && tar xz')
env_setup+=( adb shell LD_LIBRARY_PATH=$target_dir )
else
target_dir=$work_dir
env_setup+=( env LD_LIBRARY_PATH=$target_dir )
fi
main_cmd=()
if [ "$linker" != "" ]; then
main_cmd+=( $linker )
fi
main_cmd+=( $target_dir/linker_reloc_bench_main )
# Run the program once to warm the cache.
echo "+ ${env_setup[@]}" "${main_cmd[@]}"
"${env_setup[@]}" "${main_cmd[@]}"
# Run the benchmark proper.
run_cmd=()
if [ $cpu_mask -ne 0 ]; then
run_cmd+=( taskset $cpu_mask )
fi
run_cmd+=( $time_cmd )
echo "+ ${env_setup[@]}" "${run_cmd[@]}" "${main_cmd[@]}"
"${env_setup[@]}" "${run_cmd[@]}" "${main_cmd[@]}"
if [ $keep_tmp_dir -eq 0 ]; then
rm -fr $tmp_dir
fi