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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
#!/usr/bin/env python3
#
# Copyright (C) 2018 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
import os
import statistics
import time
import adb
def lock_min(device):
device.shell_nocheck(["""
for x in /sys/devices/system/cpu/cpu?/cpufreq; do
echo userspace > $x/scaling_governor
cat $x/scaling_min_freq > $x/scaling_setspeed
done
"""])
def lock_max(device):
device.shell_nocheck(["""
for x in /sys/devices/system/cpu/cpu?/cpufreq; do
echo userspace > $x/scaling_governor
cat $x/scaling_max_freq > $x/scaling_setspeed
done
"""])
def unlock(device):
device.shell_nocheck(["""
for x in /sys/devices/system/cpu/cpu?/cpufreq; do
echo ondemand > $x/scaling_governor
echo sched > $x/scaling_governor
echo schedutil > $x/scaling_governor
done
"""])
def harmonic_mean(xs):
return 1.0 / statistics.mean([1.0 / x for x in xs])
def analyze(name, speeds):
median = statistics.median(speeds)
mean = harmonic_mean(speeds)
stddev = statistics.stdev(speeds)
msg = "%s: %d runs: median %.2f MiB/s, mean %.2f MiB/s, stddev: %.2f MiB/s"
print(msg % (name, len(speeds), median, mean, stddev))
def benchmark_push(device=None, file_size_mb=100):
if device == None:
device = adb.get_device()
remote_path = "/dev/null"
local_path = "/tmp/adb_benchmark_temp"
with open(local_path, "wb") as f:
f.truncate(file_size_mb * 1024 * 1024)
speeds = list()
for _ in range(0, 10):
begin = time.time()
device.push(local=local_path, remote=remote_path)
end = time.time()
speeds.append(file_size_mb / float(end - begin))
analyze("push %dMiB" % file_size_mb, speeds)
def benchmark_pull(device=None, file_size_mb=100):
if device == None:
device = adb.get_device()
remote_path = "/data/local/tmp/adb_benchmark_temp"
local_path = "/tmp/adb_benchmark_temp"
device.shell(["dd", "if=/dev/zero", "of=" + remote_path, "bs=1m",
"count=" + str(file_size_mb)])
speeds = list()
for _ in range(0, 10):
begin = time.time()
device.pull(remote=remote_path, local=local_path)
end = time.time()
speeds.append(file_size_mb / float(end - begin))
analyze("pull %dMiB" % file_size_mb, speeds)
def benchmark_shell(device=None, file_size_mb=100):
if device == None:
device = adb.get_device()
speeds = list()
for _ in range(0, 10):
begin = time.time()
device.shell(["dd", "if=/dev/zero", "bs=1m",
"count=" + str(file_size_mb)])
end = time.time()
speeds.append(file_size_mb / float(end - begin))
analyze("shell %dMiB" % file_size_mb, speeds)
def main():
device = adb.get_device()
unlock(device)
benchmark_push(device)
benchmark_pull(device)
if __name__ == "__main__":
main()
|
