def get_system_version(serial_num):
lookup_sys_ver_cmd = "adb -s {} shell getprop ro.build.version.release".format( # noqa
serial_num
)
sys_ver = run_cmd(lookup_sys_ver_cmd)[0]
logger.debug("system_version:{}".format(sys_ver))
return sys_ver
def cpu_idx_str_to_mask(cpu_idx_str_raw, power_mode):
if power_mode == "no_bind":
return " "
# TODO(ysh329):
str_cpu_idx_list = cpu_idx_str_raw.split(",")
str_cpu_idx_list = filter(lambda s: s != "", str_cpu_idx_list)
int_cpu_idx_list = map(int, str_cpu_idx_list)
cpu_mask_10 = 0
for cpu_idx in int_cpu_idx_list:
cpu_mask_10 += 2**cpu_idx
cpu_mask_hex_raw = hex(cpu_mask_10)
cpu_mask_hex_std_list = cpu_mask_hex_raw.split("x")
assert len(cpu_mask_hex_std_list) == 2
zero_num = 8 - len(cpu_mask_hex_std_list[1])
cpu_mask_hex_std = "0x" + zero_num * "0" + cpu_mask_hex_std_list[1]
logger.debug(
"cpu_idx_str_raw:{}, cpu_mask_10:{}, cpu_mask_hex_raw:{}," # noqa
" cpu_mask_hex_std:{}".format(
cpu_idx_str_raw,
cpu_mask_10,
cpu_mask_hex_raw,
cpu_mask_hex_std # noqa
)) # noqa
# TODO(ysh329): can improve
cpu_mask_hex_std = " taskset " + cpu_mask_hex_std
return cpu_mask_hex_std
def prepare_models_for_devices(self):
logger.info(
"==== {} ====".format(self.prepare_models_for_devices.__name__)
) # noqa
device_work_dir = self.config["device_work_dir"]
device_dict = self.config["device_dict"]
model_dict = self.config["model_dict"]
device_serials = list(device_dict.keys())
model_names = list(model_dict.keys())
cmds = list()
for didx in range(len(device_serials)):
device_serial = device_serials[didx]
mkdir_cmd = "adb -s {} shell mkdir -p {}".format(
device_serial, device_work_dir
)
cmds.append(mkdir_cmd)
for midx in range(len(model_names)):
model_name = model_names[midx]
model_proto = model_dict[model_name]
model_param = model_proto.replace("tnnproto", "tnnmodel")
push_proto_cmd = "adb -s {} push {} {}".format(
device_serial, model_proto, device_work_dir
)
push_param_cmd = "adb -s {} push {} {}".format(
device_serial, model_param, device_work_dir
)
cmds.extend([push_proto_cmd, push_param_cmd])
logger.debug([push_proto_cmd, push_param_cmd])
run_cmds(cmds)
return 0
def get_product(serial_num):
lookup_product_cmd = "adb -s {} shell getprop | grep 'ro.product.model'".format( # noqa
serial_num # noqa
) # noqa
product = run_cmd(lookup_product_cmd)[0]
product = product.split(": ")[1].strip().replace("[", "").replace("]", "") # noqa
logger.debug(product)
return product
def get_soc_code(serial_num):
lookup_soc_code_cmd = (
"adb -s {} shell getprop | " # noqa
"grep 'ro.board.platform'".format(serial_num) # noqa
)
soc_code = run_cmd(lookup_soc_code_cmd)[0]
soc_code = soc_code.split(": ")[1].strip().replace("[", "").replace("]", "") # noqa
logger.debug(soc_code)
return soc_code
def parse_benchmark(self, cmd_res):
logger.info("==== {} ====".format(self.parse_benchmark.__name__))
output_lines = cmd_res
logger.debug(output_lines)
framework_name = self.config["framework_name"]
benchmark = dict()
if framework_name == "tnn" or framework_name == "mnn":
if framework_name == "tnn":
bench_res_keyword = "time cost"
elif framework_name == "mnn":
bench_res_keyword = "max ="
output_lines = filter( # noqa
lambda line: bench_res_keyword in line, output_lines)
output_lines = list(output_lines)
assert len(output_lines) == 1
line = output_lines[0].split()
logger.debug(line)
line = "".join(line)
logger.debug(line)
benchmark["min"] = pattern_match(line, "min=", "ms", False)
benchmark["max"] = pattern_match(line, "max=", "ms", False)
benchmark["avg"] = pattern_match(line, "avg=", "ms", False)
benchmark["std_dev"] = pattern_match(line, "std_dev=", "ms", False)
elif framework_name == "ncnn":
is_no_vulkan = list(
filter(lambda line: "no vulkan device" in line, output_lines))
is_no_vulkan = len(is_no_vulkan) > 0
if is_no_vulkan:
benchmark["min"] = 0.0
benchmark["max"] = 0.0
benchmark["avg"] = 0.0
benchmark["std_dev"] = 0.0
else:
output_lines = filter(lambda line: "min = " in line,
output_lines) # noqa
output_lines = list(output_lines)
logger.debug("output_lines:\n{}".format(output_lines))
assert len(output_lines) == 1
line = output_lines[0]
line = line.split()
logger.debug(line)
line = "".join(line) + "END"
benchmark["min"] = pattern_match(line, "min=", "max", False)
benchmark["max"] = pattern_match(line, "max=", "avg", False)
benchmark["avg"] = pattern_match(line, "avg=", "std_dev",
False) # noqa
benchmark["std_dev"] = pattern_match(line, "std_dev=", "END",
False) # noqa
else:
logger.fatal("Unsupported framework {}".format( # noqa
self.config["framework_name"] # noqa
) # noqa
)
exit(1)
logger.info("benchmark:{}".format(benchmark))
assert len(benchmark) != 0
return benchmark
def get_cpu_max_freqs(serial_num):
check_cpu_num_cmd = "adb -s {} shell cat /proc/cpuinfo | grep processor".format( # noqa
serial_num
)
cmd_res = run_cmd(check_cpu_num_cmd)
cpu_num = len(cmd_res)
check_cpu_max_freq_cmd_pattern = (
"adb -s {} shell cat "
"/sys/devices/system/cpu/cpu{}/cpufreq/cpuinfo_max_freq" # noqa
)
cmds = map(
lambda cpu_idx: check_cpu_max_freq_cmd_pattern.format( # noqa
serial_num, cpu_idx
),
range(cpu_num),
)
cmds = list(cmds)
try:
cmd_res_list = run_cmds(cmds)
logger.info(cmd_res_list[cmds[0]])
cpu_max_freqs = map(lambda cmd_key: cmd_res_list[cmd_key], cmds)
except IndexError:
logger.warn(
"cat: /sys/devices/system/cpu/cpu0/cpufreq/scaling_max_freq:" # noqa
" Permission denied" # noqa
)
logger.warn("replacing scaling_max_freq with cpuinfo_max_freq")
cmds = map(lambda c: c.replace("cpuinfo", "scaling"), cmds)
cmd_res_list = run_cmds(cmds)
# logger.warn(cmd_res_list[cmds[0]].strip())
cpu_max_freqs = map(
lambda cmd_key: cmd_res_list[cmd_key].strip(), cmds # noqa
) # noqa
cpu_max_freqs = list(cpu_max_freqs)
cpu_max_freqs = cpu_max_freqs[:cpu_num]
# get str from list
cpu_max_freqs = list(map(lambda l: l[0], cpu_max_freqs))
logger.debug(
"cpu_max_freqs:{}, cpu_num:{}".format(cpu_max_freqs, cpu_num) # noqa
) # noqa
is_valid_freq = (
lambda freq_str: True
if "No such file or director" not in freq_str
else False # noqa
)
cpu_max_freqs_ghz = map(
lambda freq: float(freq) / 1e6 if is_valid_freq(freq) else None, # noqa
cpu_max_freqs, # noqa
)
logger.debug(cpu_max_freqs_ghz)
cpu_max_freqs_ghz = list(cpu_max_freqs_ghz)
return cpu_max_freqs_ghz
def run(self): # called by start()
self.start_time = ctime()
logger.debug("start {} on device {} at {}".format(
self.thread_idx, self.device_serial, self.start_time))
self.res = self.func(*self.args)
self.end_time = ctime()
logger.debug(
"finish thread(from0) {} on device {}, framework {} at {}".format(
self.thread_idx,
self.device_serial,
self.framework_name,
self.end_time, # noqa
))
def get_imei(serial_num):
lookup_imei_cmd = "adb -s {} shell service call iphonesubinfo 1".format( # noqa
serial_num)
imei_list = run_cmd(lookup_imei_cmd)
imei_list = list(filter(lambda l: "." in l, imei_list))
assert 0 < len(imei_list)
imei_list = list(
map(lambda l: pattern_match(l, "'", "'", False), imei_list)) # noqa
logger.debug("imei_list:{}".format(imei_list))
imei_list = map(lambda l: l.replace(".", ""), imei_list)
imei = "".join(imei_list)
imei = imei.replace(" ", "")
logger.debug("imei:{}".format(imei))
return imei
def prepare_models_for_devices(self):
logger.info("==== {} ====".format(
self.prepare_models_for_devices.__name__)) # noqa
device_work_dir = self.config["device_work_dir"]
device_dict = self.config["device_dict"]
model_dict = self.config["model_dict"]
device_serials = list(device_dict.keys())
model_names = list(model_dict.keys())
logger.debug(model_dict)
cmds = list()
for didx in range(len(device_serials)):
device_serial = device_serials[didx]
mkdir_cmd = "adb -s {} shell mkdir -p {}".format(
device_serial, device_work_dir)
cmds.append(mkdir_cmd)
for midx in range(len(model_names)):
model_name = model_names[midx]
model_proto = model_dict[model_name]
if (self.config["framework_name"] == "ncnn"
or self.config["framework_name"] == "tnn"):
model_param = model_proto.replace("tnnproto", "tnnmodel")
elif (self.config["framework_name"] == "mnn"
or self.config["framework_name"] == "tflite"):
model_param = None
else:
logger.fatal("Unsupported framework {}".format( # noqa
self.config["framework_name"] # noqa
) # noqa
)
exit(1)
push_proto_cmd = "adb -s {} push {} {}".format(
device_serial,
model_proto,
"/".join([device_work_dir,
os.path.basename(model_proto)]), # noqa
)
push_param_cmd = "adb -s {} push {} {}".format(
device_serial, model_param, device_work_dir)
push_param_cmd = (
"echo" if model_param is None else push_param_cmd) # noqa
cmds.extend([push_proto_cmd, push_param_cmd])
run_cmds(cmds)
return 0
def run_bench_single_thread(self):
logger.info("==== {} ====".format(
self.run_bench_single_thread.__name__) # noqa
)
device_dict = self.config["device_dict"]
device_serials = list(device_dict.keys())
bench_dict = dict()
for didx in range(len(device_serials)):
device_serial_num = device_serials[didx]
logger.debug("didx:{}, serial_num:{}".format(
didx, device_serial_num) # noqa
)
bench_dict[
device_serial_num] = self.run_bench_for_single_thread_func( # noqa
device_serial_num, device_idx=didx)[device_serial_num]
return bench_dict
def run_cmd(cmd, wait_interval_sec=5, max_timeout_sec=100):
cmd_type = "CMD"
logger.info("{}> {}".format(cmd_type, cmd))
subp = subprocess.Popen(
cmd,
shell=True,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
encoding="utf-8",
)
subp.wait(max_timeout_sec)
subp_status = -1
duration_sec = 0
while 1:
if subp.poll() == 0:
logger.debug("{} finished".format(cmd_type))
subp_status = int(subp.poll())
break
elif subp.poll() is None:
logger.debug("{} duration += {} second".format(
cmd_type, wait_interval_sec) # noqa
) # noqa
time.sleep(wait_interval_sec)
duration_sec += wait_interval_sec
if duration_sec > max_timeout_sec:
logger.error(
"{} duration {} second timeout with max_timeout_sec {}".
format( # noqa
cmd_type, duration_sec, max_timeout_sec))
subp.kill()
break
else:
subp_status = subp.poll()
logger.fatal(
"exited with abnormal subprocess status: {}".format( # noqa
subp_status) # noqa
)
if subp_status == 139:
break
else:
exit(1)
logger.debug("{} consume {} seconds to finish".format(
cmd_type, duration_sec) # noqa
) # noqa
cmd_res = None
if subp_status == 0 or subp_status == 139:
cmd_res = "".join(subp.communicate())
logger.debug("cmd_res:{}".format(subp.communicate()))
cmd_res = cmd_res.split("\n")
cmd_res = filter(lambda r: r != "", cmd_res)
cmd_res = list(cmd_res)
return cmd_res
def prepare_models(self):
logger.info("==== {} ====".format(self.prepare_models.__name__))
cmds = list()
framework_name = self.config["framework_name"]
model_repo = self.config["model_repo"]
model_type_keyword = self.config["model_type_keyword"]
repo_name = get_file_name(model_repo, False)
logger.info("repo_name:{}".format(repo_name))
logger.debug("os.path.exists:{}".format(os.path.exists(repo_name)))
if os.path.exists(repo_name):
clone_models_cmd = "ls {}".format(repo_name)
else:
clone_models_cmd = "git clone {}".format(model_repo)
model_repo_version_cmd = (
'cd ./{}-models/; git log --pretty=format:"SHA-1:%h date:%ad" '
'--date=format:"%y-%m-%d" -n1 #--shortstat -n1'.format(
framework_name) # noqa
)
model_repo_version_extra_cmd = (
"cd ./{}-models/; "
'git log --pretty=format:"SHA-1:%h - author:%an date:%ad '
'note:%s" --date=format:"%y-%m-%d %H:%M:%S" -n1'.format(
framework_name) # noqa
)
model_repo_branch_cmd = "cd ./{}-models/; git branch | sed 's/\* //g'".format( # noqa
framework_name)
lookup_models_path_cmd = "realpath ./{}-models/*{}*".format(
framework_name, model_type_keyword)
cmds.extend([
clone_models_cmd,
model_repo_version_cmd,
model_repo_version_extra_cmd,
model_repo_branch_cmd,
lookup_models_path_cmd,
])
cmds_res = run_cmds(cmds)
# TODO(ysh329): add model_repo_version to config
# self.config["model_repo_branch"] =
# cmds_res[model_repo_branch_cmd][0]
# self.config["model_repo_version"] =
# cmds_res[model_repo_version_cmd].readalines()[0]
# self.config["model_repo_version_extra"] =
# cmds_res[model_repo_version_extra_cmd].readalines()[0]
models_dir = list(
map(lambda path: path.strip(),
cmds_res[lookup_models_path_cmd]) # noqa
)
model_dict = dict()
for midx in range(len(models_dir)):
model_dir = models_dir[midx]
logger.debug("{} {}".format(midx, model_dir))
file_type = model_dir.split(".")[-1]
model_name = (model_dir.split("/")[-1].replace(
"." + file_type, "").replace(file_type, ""))
logger.debug("model_name:{}, file_type:{}".format(
model_name, file_type)) # noqa
model_dict[model_name] = model_dir
logger.debug(models_dir)
logger.debug(model_dict)
return model_dict
def run_bench_for_single_thread_func(
self,
device_serial,
device_idx=-1,
thread_idx=-1,
thread_num=-1,
framework_name="",
):
logger.info( # noqa
"==== {}, thread_idx(from0):{}/{} ====".format(
self.run_bench_for_single_thread_func.__name__,
thread_idx,
thread_num, # noqa
))
device_work_dir = self.config["device_work_dir"]
device_dict = self.config["device_dict"]
cur_device_dict = device_dict[device_serial]
logger.info("cur_device_dict:{}".format(cur_device_dict))
model_dict = self.config["model_dict"]
model_names = list(model_dict.keys())
platforms = self.config["benchmark_platform"]
support_backend = self.config["support_backend"]
bench_cmd_pattern = self.config["bench_cmd_pattern"]
# note(ysh329): this bench_dict is for single thread about device
bench_dict = dict()
bench_dict[device_serial] = dict()
cpu_backend_num = list(
map(self.config["is_cpu_backend"], support_backend)).count(True)
bench_case_num = (
len(platforms) * len(model_names) * sum([
len(self.config["cpu_thread_num"]) * cpu_backend_num +
len(set(support_backend) - set(["ARM", "ARM_XNNPACK"])) # noqa
]))
logger.info("len(platform):{}".format(len(platforms)))
logger.info("len(model_names):{}".format(len(model_names)))
logger.info(
'len(self.config["cpu_thread_num"]) * cpu_backend_num:{}'.format(
len(self.config["cpu_thread_num"]) * cpu_backend_num))
logger.info("support_backend:{}".format(support_backend))
logger.info(
'len(set(support_backend) - set("ARM") - set("ARM_XNNPACK")):{}'.
format( # noqa
len(set(support_backend) - set("ARM") - set("ARM_XNNPACK"))))
logger.info("bench_case_num:{}".format(bench_case_num))
logger.info(
'len(self.config["cpu_thread_num"]) if "CPU" in support_backend else 0:{}'
.format( # noqa
len(self.config["cpu_thread_num"])
if "CPU" in support_backend else 0 # noqa
))
logger.info('len(set(support_backend) - set("ARM")):{}'.format(
len(set(support_backend) - set("ARM"))))
bench_case_idx = 0
# platform: armv7/armv8/...
for pidx in range(len(platforms)):
platform = platforms[pidx]
device_work_dir_platform = device_work_dir + "/" + platform # noqa
device_benchmark_bin = "/".join([
device_work_dir_platform,
os.path.basename(
self.config[platform]["benchmark_bin"]), # noqa
])
bench_dict[device_serial][platform] = dict()
logger.debug("pidx:{}, platform:{}".format(pidx, platform))
# model: mobilenetv1/v2/...
for midx in range(len(model_names)):
model_name = model_names[midx]
model_dir = "/".join([
device_work_dir,
os.path.basename(model_dict[model_name]),
] # noqa
)
logger.debug("midx:{}, model_name:{}, model_dir:{}".format(
midx, model_name, model_dir))
bench_dict[device_serial][platform][model_name] = dict(
) # noqa
# backend: cpu/gpu/xpu/...
for bidx in range(len(support_backend)):
backend = support_backend[bidx]
logger.debug("bidx:{}, backend:{}".format(bidx,
backend)) # noqa
bench_dict[device_serial][platform][model_name][
backend] = dict() # noqa
# thread: 1/2/4/...
for tidx in range(
len(self.config["cpu_thread_num"]) if self.
config["is_cpu_backend"](backend) else 1 # noqa
):
bench_case_idx += 1
logger.info(
"\n\nframework_name:{}, device_idx(from1):{}/{}, bench_case_idx(from1):{}/{}," # noqa
" enable_multi_threads:{}, thread_idx(from0):{}/{}"
.format( # noqa
# noqa
self.engine_name(), # noqa
device_idx + 1,
len(device_dict), # noqa
bench_case_idx,
bench_case_num, # noqa
self.config["enable_multi_threads"], # noqa
thread_idx,
thread_num, # noqa
) # noqa
)
cpu_thread_num = (
self.config["cpu_thread_num"][tidx]
if self.config["is_cpu_backend"](backend) else 1
) # noqa
bench_dict[device_serial][platform][
model_name][ # noqa
backend][cpu_thread_num] = dict() # noqa
#######################
# bench case start
#######################
if self.config["framework_name"] == "tnn":
bench_cmd = bench_cmd_pattern.format(
**{
"serial_num":
device_serial,
"device_work_dir":
device_work_dir_platform, # noqa
"device_benchmark_bin":
device_benchmark_bin, # noqa
"model_type":
self.config["framework_name"], # noqa
"model_dir":
model_dir,
"backend":
backend,
"repeats":
self.config["repeats"](backend), # noqa
"warmup":
self.config["warmup"],
"thread_num":
cpu_thread_num,
"bind_cpu_idx":
device_dict[device_serial][ # noqa
"bind_cpu_idx"],
})
elif self.config["framework_name"] == "ncnn":
bench_cmd = bench_cmd_pattern.format(
**{
"serial_num": device_serial,
"device_benchmark_bin":
device_benchmark_bin, # noqa
"model_dir": model_dir,
"repeats": self.config["repeats"](
backend), # noqa
"warmup": self.config["warmup"],
"thread_num": cpu_thread_num,
"power_mode":
self.config["power_mode_id"], # noqa
"gpu_device": backend,
})
elif self.config["framework_name"] == "mnn":
# '{device_benchmark_bin} {model_dir} {repeats} {warmup}' # noqa
# '{forwardtype} {thread_num} {precision}"'
bench_cmd = bench_cmd_pattern.format(
**{
"serial_num": device_serial,
"device_work_dir":
device_work_dir_platform, # noqa
"device_benchmark_bin":
device_benchmark_bin, # noqa
"model_dir": model_dir,
"repeats": self.config["repeats"](
backend), # noqa
"warmup": self.config["warmup"],
"thread_num": cpu_thread_num,
"forwardtype": backend,
# power_mode: big_core default
})
elif self.config["framework_name"] == "tflite":
# {device_benchmark_bin} --graph={model_dir}
# --output_prefix={model_name}
# --num_runs={repeats} --warmup_runs={warmup}
# --num_threads={thread_num} {backend}
bench_cmd = bench_cmd_pattern.format(
**{
"serial_num":
device_serial,
"device_work_dir":
device_work_dir_platform, # noqa
"device_benchmark_bin":
device_benchmark_bin, # noqa
"model_dir":
model_dir,
"model_name":
model_name,
"repeats":
self.config["repeats"](backend), # noqa
"warmup":
self.config["warmup"],
"thread_num":
cpu_thread_num,
"backend":
self.config["support_backend_cmd_id"]
( # noqa
backend),
# power_mode(TODO): no bind deafault? need explore deeply # noqa
})
print(bench_cmd)
else:
logger.fatal("Unsupported framework {}".format(
self.config["framework_name"]))
exit(1)
#################################
# run benchmark
#################################
run_times_sum = (
self.config["repeats"](backend) +
self.config["warmup"] # noqa
) # noqa
max_wait_sec = (MAX_TIMEOUT_SECOND_ONCE_INFER *
run_times_sum) # noqa
cmd_res = run_cmd(
bench_cmd,
wait_interval_sec=3,
max_timeout_sec=max_wait_sec,
) # noqa
perf_dict = self.parse_benchmark(cmd_res)
#################################
# summarize benchmark info
#################################
bench_record = {
"soc_code":
device_dict[device_serial]["soc_code"], ## noqa
"product":
device_dict[device_serial][ # noqa
"product"], # noqa
"serial_num":
device_serial,
"platform":
platform,
"model_name":
model_name,
"repeats":
self.config["repeats"](backend),
"warmup":
self.config["warmup"],
"avg":
perf_dict["avg"],
"max":
perf_dict["max"],
"min":
perf_dict["min"],
"std_dev":
perf_dict["std_dev"],
"backend":
backend,
"cpu_thread_num":
cpu_thread_num,
"power_mode":
self.config["power_mode"],
"bind_cpu_idx":
device_dict[device_serial]["bind_cpu_idx"], # noqa
"cpu_max_freqs":
device_dict[device_serial][ # noqa
"cpu_max_freqs"],
"battery_level":
device_dict[device_serial][ # noqa
"battery_level"],
"system_version":
device_dict[device_serial][ # noqa
"system_version"],
"cmd":
bench_cmd,
"imei":
device_dict[device_serial]["imei"],
}
bench_dict[device_serial][platform][
model_name][ # noqa
backend][cpu_thread_num] = bench_record # noqa
logger.info(bench_record)
return bench_dict
def prepare_benchmark_assets_for_devices(self):
logger.info("==== {} ====".format(
self.prepare_benchmark_assets_for_devices.__name__) # noqa
)
benchmark_platform = self.config["benchmark_platform"]
device_work_dir = self.config["device_work_dir"]
device_dict = self.config["device_dict"]
device_serials = list(device_dict.keys())
cmds = list()
for didx in range(len(device_serials)):
device_serial = device_serials[didx]
for pidx in range(len(benchmark_platform)):
platform = benchmark_platform[pidx]
device_work_dir_platform = device_work_dir + "/" + platform
# benchmark assets
benchmark_bin = self.config[platform]["benchmark_bin"]
benchmark_lib = self.config[platform]["shared_lib"]
benchmark_libs = (benchmark_lib if isinstance(
benchmark_lib, list) else [benchmark_lib])
# create cmds
rmdir_cmd = "adb -s {} shell rm -rf {}".format(
device_serial, device_work_dir_platform)
mkdir_cmd = "adb -s {} shell mkdir -p {}".format(
device_serial, device_work_dir_platform)
# lib
benchmark_lib_device_paths = map(
lambda lib: device_work_dir_platform + "/" + os.path.
basename(lib), # noqa
benchmark_libs,
)
benchmark_lib_device_paths = list(benchmark_lib_device_paths)
logger.debug("benchmark_lib_device_paths:{}".format( # noqa
benchmark_lib_device_paths # noqa
) # noqa
)
push_shared_lib_cmds = map(
lambda lib, lib_device: "adb -s {} push {} {}".
format( # noqa
device_serial, lib, lib_device), # noqa
benchmark_libs,
benchmark_lib_device_paths,
)
push_shared_lib_cmds = list(push_shared_lib_cmds)
push_shared_lib_cmds = (["echo"] if benchmark_lib is None else
push_shared_lib_cmds)
logger.debug("push_shared_lib_cmds:{}".format( # noqa
push_shared_lib_cmds) # noqa
)
logger.debug("len(push_shared_lib_cmds):{}".format( # noqa
len(push_shared_lib_cmds) # noqa
) # noqa
)
# bin
benchmark_bin_device_path = (
device_work_dir_platform + "/" +
os.path.basename(benchmark_bin) # noqa
)
push_benchmark_bin_cmd = "adb -s {} push {} {}".format(
device_serial, benchmark_bin, benchmark_bin_device_path)
chmod_x_bin_cmd = "adb -s {} shell chmod +x {}".format(
device_serial, benchmark_bin_device_path)
cmds.extend([rmdir_cmd, mkdir_cmd])
cmds.extend(push_shared_lib_cmds)
cmds.extend([push_benchmark_bin_cmd, chmod_x_bin_cmd])
logger.info(cmds)
run_cmds(cmds)
return 0
def prepare_devices(self):
logger.info("==== {} ====".format(self.prepare_devices.__name__))
device_status_dict = get_adb_devices(True)
serial_num_list = list(device_status_dict.keys())
logger.debug(serial_num_list)
device_dict = dict()
for sidx in range(len(serial_num_list)):
ser = serial_num_list[sidx]
device_status = device_status_dict[ser]
if device_status != "device":
logger.info("device {} status is {}, skipped".format( # noqa
ser, device_status) # noqa
)
continue
device_dict[ser] = dict()
device_dict[ser]["status"] = device_status
device_dict[ser]["cpu_max_freqs"] = get_cpu_max_freqs(ser) # noqa
cpu_max_freqs = get_cpu_max_freqs(ser)
cpu_valid_freqs = list(
filter(lambda freq: freq is not None, cpu_max_freqs)) # noqa
big_cores_idx = get_target_freq_idx(max(cpu_valid_freqs), ser,
cpu_max_freqs)
big_cores_idx_str = ",".join(big_cores_idx)
little_cores_idx = get_target_freq_idx(min(cpu_valid_freqs), ser,
cpu_max_freqs)
little_cores_idx_str = ",".join(little_cores_idx)
device_dict[ser]["big_cores_idx"] = big_cores_idx_str
device_dict[ser]["little_cores_idx"] = little_cores_idx_str # noqa
if self.config["power_mode"] == "big_cores":
device_dict[ser]["bind_cpu_idx"] = big_cores_idx_str # noqa
elif self.config["power_mode"] == "little_cores":
device_dict[ser]["bind_cpu_idx"] = little_cores_idx_str # noqa
elif self.config["power_mode"] == "no_bind":
device_dict[ser]["bind_cpu_idx"] = ",".join(
map(str, range(len(cpu_max_freqs))))
else:
logger.info("Unsupported power_mode:{}".format(
self.config["power_mode"]) # noqa
)
exit(1)
# battery level
device_dict[ser]["battery_level"] = get_battery_level(ser) # noqa
# system version
device_dict[ser]["system_version"] = get_system_version(
ser) # noqa
# imie
device_dict[ser]["imei"] = get_imei(ser) # noqa
# ro.board.platform, ro.board.chiptype, ro.board.hardware
device_dict[ser]["soc_code"] = get_soc_code(ser) # noqa
# soc info
device_dict[ser]["soc_info"] = get_soc_info_from_soc_code(
device_dict[ser]["soc_code"])
# product
device_dict[ser]["product"] = get_product(ser) # noqa
logger.debug(device_dict)
logger.info("len(device_dict):{}".format(len(device_dict)))
return device_dict
def gen_bench_cmds(self):
logger.info("==== {} ====".format(self.gen_bench_cmds.__name__))
device_work_dir = self.config["device_work_dir"]
device_dict = self.config["device_dict"]
model_dict = self.config["model_dict"]
device_serials = list(device_dict.keys())
model_names = list(model_dict.keys())
benchmark_platform = self.config["benchmark_platform"]
support_backend = self.config["support_backend"]
bench_cmd_pattern = self.config["bench_cmd_pattern"]
bench_dict = dict()
bench_case_idx = 0
for didx in range(len(device_serials)):
device_serial_num = device_serials[didx]
logger.debug("didx:{}, serial_num:{}".format(
didx, device_serial_num) # noqa
)
bench_dict[device_serial_num] = dict()
bench_cmds = []
for pidx in range(len(benchmark_platform)):
platform = benchmark_platform[pidx]
device_work_dir_platform = device_work_dir + "/" + platform # noqa
device_benchmark_bin = "/".join([
device_work_dir_platform,
os.path.basename(
self.config[platform]["benchmark_bin"]), # noqa
])
bench_dict[device_serial_num][platform] = dict()
logger.debug("pidx:{}, platform:{}".format(pidx, platform))
for midx in range(len(model_names)):
model_name = model_names[midx]
model_dir = "/".join([
device_work_dir,
os.path.basename(model_dict[model_name]),
] # noqa
)
logger.debug("midx:{}, model_name:{}, model_dir:{}".format(
midx, model_name, model_dir))
bench_dict[device_serial_num][platform][model_name] = dict(
) # noqa
for bidx in range(len(support_backend)):
backend = support_backend[bidx]
is_cpu = self.config["is_cpu_backend"]
logger.debug("bidx:{}, backend:{}".format(
bidx, backend)) # noqa
bench_dict[device_serial_num][platform][model_name][
backend] = dict() # noqa
for tidx in range(
len(self.config["cpu_thread_num"]
) if is_cpu(backend) else 1 # noqa
):
bench_case_idx += 1
logger.info("\n\nbench_case_idx(from 1):{}".format(
bench_case_idx) # noqa
)
cpu_thread_num = self.config["cpu_thread_num"][
tidx] # noqa
bench_dict[device_serial_num][platform][
model_name][ # noqa
backend][cpu_thread_num] = dict() # noqa
#######################
# bench start
#######################
if self.config["framework_name"] == "tnn":
bench_cmd = bench_cmd_pattern.format(
**{
"serial_num":
device_serial_num,
"device_work_dir":
device_work_dir_platform, # noqa
"device_benchmark_bin":
device_benchmark_bin, # noqa
"model_type":
self.config["framework_name"], # noqa
"model_dir":
model_dir,
"backend":
backend,
"repeats":
self.config["repeats"](
backend), # noqa
"warmup":
self.config["warmup"],
"thread_num":
cpu_thread_num,
"bind_cpu_idx":
device_dict[device_serial_num][ # noqa
"bind_cpu_idx"],
})
elif self.config["framework_name"] == "ncnn":
bench_cmd = bench_cmd_pattern.format(
**{
"serial_num": device_serial_num,
"device_benchmark_bin":
device_benchmark_bin, # noqa
"model_dir": model_dir,
"repeats": self.config["repeats"](
backend), # noqa
"warmup": self.config["warmup"],
"thread_num": cpu_thread_num,
"power_mode":
self.config["power_mode_id"], # noqa
"gpu_device": backend,
})
elif self.config["framework_name"] == "mnn":
# '{device_benchmark_bin} {model_dir} {repeats} {warmup}' # noqa
# '{forwardtype} {thread_num} {precision}"'
bench_cmd = bench_cmd_pattern.format(
**{
"serial_num": device_serial_num,
"device_work_dir":
device_work_dir_platform, # noqa
"device_benchmark_bin":
device_benchmark_bin, # noqa
"model_dir": model_dir,
"repeats": self.config["repeats"](
backend), # noqa
"warmup": self.config["warmup"],
"thread_num": cpu_thread_num,
"forwardtype": backend,
# power_mode: big_core default
})
else:
logger.fatal("Unsupported framework {}".format(
self.config["framework_name"]))
continue
bench_cmds.append(bench_cmd)
bench_dict[device_serial_num]["bench_cmds"] = bench_cmds
return bench_dict
def parse_benchmark(self, cmd_res):
logger.info("==== {} ====".format(self.parse_benchmark.__name__))
output_lines = cmd_res
logger.debug(output_lines)
framework_name = self.config["framework_name"]
benchmark = dict()
benchmark["min"] = 0.0
benchmark["max"] = 0.0
benchmark["avg"] = 0.0
benchmark["std_dev"] = 0.0
if cmd_res is None:
return benchmark
elif (framework_name == "tnn" or framework_name == "mnn"
or framework_name == "tflite"):
if framework_name == "tnn" or framework_name == "tflite":
bench_res_keyword = "time cost"
elif framework_name == "mnn":
bench_res_keyword = "max ="
output_lines_str = "".join(output_lines)
output_lines = filter( # noqa
lambda line: bench_res_keyword in line, output_lines)
if (framework_name == "mnn"
and "Floating point exception" in output_lines_str):
return benchmark
elif (framework_name == "tflite" and
'Error: dlopen failed: library "libhexagon_interface.so" not found' # noqa
in output_lines_str):
return benchmark
output_lines = list(output_lines)
assert len(output_lines) == 1
line = output_lines[0].split()
logger.debug(line)
line = "".join(line)
logger.debug(line)
benchmark["min"] = pattern_match(line, "min=", "ms", False)
benchmark["max"] = pattern_match(line, "max=", "ms", False)
benchmark["avg"] = pattern_match(line, "avg=", "ms", False)
benchmark["std_dev"] = pattern_match(line, "std_dev=", "ms", False)
elif framework_name == "ncnn":
is_no_vulkan = list(
filter(
lambda line: "no vulkan device" in line or
'"libvulkan.so" not found' in line,
output_lines,
))
is_no_vulkan = len(is_no_vulkan) > 0
if is_no_vulkan:
return benchmark
else:
output_lines = filter(lambda line: "min = " in line,
output_lines) # noqa
output_lines = list(output_lines)
logger.debug("output_lines:\n{}".format(output_lines))
assert len(output_lines) == 1
line = output_lines[0]
line = line.split()
logger.debug(line)
line = "".join(line) + "END"
benchmark["min"] = pattern_match(line, "min=", "max", False)
benchmark["max"] = pattern_match(line, "max=", "avg", False)
benchmark["avg"] = pattern_match(line, "avg=", "std_dev",
False) # noqa
benchmark["std_dev"] = pattern_match(line, "std_dev=", "END",
False) # noqa
else:
logger.fatal("Unsupported framework {}".format( # noqa
self.config["framework_name"] # noqa
) # noqa
)
exit(1)
logger.info("benchmark:{}".format(benchmark))
assert len(benchmark) != 0
return benchmark
def prepare_devices(self):
logger.info("==== {} ====".format(self.prepare_devices.__name__))
device_status_dict = get_adb_devices(True)
serial_num_list = list(device_status_dict.keys())
logger.debug(serial_num_list)
device_dict = dict()
for sidx in range(len(serial_num_list)):
device_serial_num = serial_num_list[sidx]
device_status = device_status_dict[device_serial_num]
if device_status != "device":
logger.info(
"device {} status is {}, skipped".format(
device_serial_num, device_status
)
)
continue
device_dict[device_serial_num] = dict()
device_dict[device_serial_num]["status"] = device_status
device_dict[device_serial_num]["cpu_max_freqs"] = get_cpu_max_freqs( # noqa
device_serial_num
)
cpu_max_freqs = get_cpu_max_freqs(device_serial_num)
cpu_valid_freqs = list(
filter(lambda freq: freq is not None, cpu_max_freqs)
) # noqa
big_cores_idx = get_target_freq_idx(
max(cpu_valid_freqs), device_serial_num, cpu_max_freqs
)
big_cores_idx_str = ",".join(big_cores_idx)
little_cores_idx = get_target_freq_idx(
min(cpu_valid_freqs), device_serial_num, cpu_max_freqs
)
little_cores_idx_str = ",".join(little_cores_idx)
device_dict[device_serial_num]["big_cores_idx"] = big_cores_idx_str
device_dict[device_serial_num][
"little_cores_idx"
] = little_cores_idx_str # noqa
if self.config["power_mode"] == "big_cores":
device_dict[device_serial_num][
"bind_cpu_idx"
] = big_cores_idx_str # noqa
elif self.config["power_mode"] == "little_cores":
device_dict[device_serial_num][
"bind_cpu_idx"
] = little_cores_idx_str # noqa
elif self.config["power_mode"] == "no_bind":
device_dict[device_serial_num]["bind_cpu_idx"] = ",".join(
map(str, range(len(cpu_max_freqs)))
)
else:
logger.info(
"Unsupported power_mode:{}".format(
self.config["power_mode"]
) # noqa
)
exit(1)
# battery level
device_dict[device_serial_num]["battery_level"] = get_battery_level( # noqa
device_serial_num
)
# system version
device_dict[device_serial_num][
"system_version"
] = get_system_version( # noqa
device_serial_num
)
# imie
device_dict[device_serial_num]["imei"] = get_imei(device_serial_num) # noqa
# ro.board.platform, ro.board.chiptype, ro.board.hardware
device_soc_cmd = (
"adb -s {} shell getprop |"
" grep 'ro.board.platform'".format(device_serial_num)
)
cmd_handls = run_cmds([device_soc_cmd])
soc = cmd_handls[device_soc_cmd][0]
soc = soc.split(": ")[1].strip().replace("[", "").replace("]", "") # noqa
device_dict[device_serial_num]["soc"] = soc
logger.debug(soc)
# product
device_product_cmd = (
"adb -s {} shell getprop | "
"grep 'ro.product.model'".format(device_serial_num)
)
cmd_handle = run_cmd(device_product_cmd)
product = cmd_handle[0]
product = (
product.split(": ")[1].strip().replace("[", "").replace("]", "") # noqa
) # noqa
device_dict[device_serial_num]["product"] = product
logger.debug(product)
logger.debug(device_dict)
logger.info("len(device_dict):{}".format(len(device_dict)))
return device_dict
def run_bench(self):
logger.info("==== {} ====".format(self.run_bench.__name__))
device_work_dir = self.config["device_work_dir"]
device_dict = self.config["device_dict"]
model_dict = self.config["model_dict"]
device_serials = list(device_dict.keys())
model_names = list(model_dict.keys())
benchmark_platform = self.config["benchmark_platform"]
support_backend = self.config["support_backend"]
bench_cmd_pattern = self.config["bench_cmd_pattern"]
bench_dict = dict()
bench_case_idx = 0
for didx in range(len(device_serials)):
device_serial_num = device_serials[didx]
logger.debug("didx:{}, serial_num:{}".format(
didx, device_serial_num) # noqa
)
bench_dict[device_serial_num] = dict()
for pidx in range(len(benchmark_platform)):
platform = benchmark_platform[pidx]
device_work_dir_platform = device_work_dir + "/" + platform # noqa
device_benchmark_bin = "/".join([
device_work_dir_platform,
os.path.basename(
self.config[platform]["benchmark_bin"]), # noqa
])
bench_dict[device_serial_num][platform] = dict()
logger.debug("pidx:{}, platform:{}".format(pidx, platform))
for midx in range(len(model_names)):
model_name = model_names[midx]
model_dir = "/".join([
device_work_dir,
os.path.basename(model_dict[model_name]),
] # noqa
)
logger.debug("midx:{}, model_name:{}, model_dir:{}".format(
midx, model_name, model_dir))
bench_dict[device_serial_num][platform][model_name] = dict(
) # noqa
for bidx in range(len(support_backend)):
backend = support_backend[bidx]
is_cpu = self.config["is_cpu_backend"]
logger.debug("bidx:{}, backend:{}".format(
bidx, backend)) # noqa
bench_dict[device_serial_num][platform][model_name][
backend] = dict() # noqa
for tidx in range(
len(self.config["cpu_thread_num"]
) if is_cpu(backend) else 1 # noqa
):
bench_case_idx += 1
logger.info("\n\nbench_case_idx(from 1):{}".format(
bench_case_idx) # noqa
)
cpu_thread_num = self.config["cpu_thread_num"][
tidx] # noqa
bench_dict[device_serial_num][platform][
model_name][ # noqa
backend][cpu_thread_num] = dict() # noqa
#######################
# bench start
#######################
if self.config["framework_name"] == "tnn":
bench_cmd = bench_cmd_pattern.format(
**{
"serial_num":
device_serial_num,
"device_work_dir":
device_work_dir_platform, # noqa
"device_benchmark_bin":
device_benchmark_bin, # noqa
"model_type":
self.config["framework_name"], # noqa
"model_dir":
model_dir,
"backend":
backend,
"repeats":
self.config["repeats"](
backend), # noqa
"warmup":
self.config["warmup"],
"thread_num":
cpu_thread_num,
"bind_cpu_idx":
device_dict[device_serial_num][ # noqa
"bind_cpu_idx"],
})
elif self.config["framework_name"] == "ncnn":
bench_cmd = bench_cmd_pattern.format(
**{
"serial_num": device_serial_num,
"device_benchmark_bin":
device_benchmark_bin, # noqa
"model_dir": model_dir,
"repeats": self.config["repeats"](
backend), # noqa
"warmup": self.config["warmup"],
"thread_num": cpu_thread_num,
"power_mode":
self.config["power_mode_id"], # noqa
"gpu_device": backend,
})
elif self.config["framework_name"] == "mnn":
# '{device_benchmark_bin} {model_dir} {repeats} {warmup}' # noqa
# '{forwardtype} {thread_num} {precision}"'
bench_cmd = bench_cmd_pattern.format(
**{
"serial_num": device_serial_num,
"device_work_dir":
device_work_dir_platform, # noqa
"device_benchmark_bin":
device_benchmark_bin, # noqa
"model_dir": model_dir,
"repeats": self.config["repeats"](
backend), # noqa
"warmup": self.config["warmup"],
"thread_num": cpu_thread_num,
"forwardtype": backend,
# power_mode: big_core default
})
else:
logger.fatal("Unsupported framework {}".format(
self.config["framework_name"]))
exit(1)
#################################
# run benchmark
#################################
run_times_sum = (
self.config["repeats"](backend) +
self.config["warmup"] # noqa
) # noqa
max_wait_sec = (MAX_TIMEOUT_SECOND_ONCE_INFER *
run_times_sum) # noqa
cmd_res = run_cmd(
bench_cmd,
wait_interval_sec=3,
max_timeout_sec=max_wait_sec,
) # noqa
perf_dict = self.parse_benchmark(cmd_res)
#################################
# summarize benchmark info
#################################
bench_record = {
"soc_code":
device_dict[device_serial_num]
["soc_code"], ## noqa
"product":
device_dict[device_serial_num][ # noqa
"product"], # noqa
"serial_num":
device_serial_num,
"platform":
platform,
"model_name":
model_name,
"repeats":
self.config["repeats"](backend),
"warmup":
self.config["warmup"],
"avg":
perf_dict["avg"],
"max":
perf_dict["max"],
"min":
perf_dict["min"],
"std_dev":
perf_dict["std_dev"],
"backend":
backend,
"cpu_thread_num":
cpu_thread_num,
"power_mode":
self.config["power_mode"],
"bind_cpu_idx":
device_dict[device_serial_num]["bind_cpu_idx"],
"cpu_max_freqs":
device_dict[device_serial_num][ # noqa
"cpu_max_freqs"],
"battery_level":
device_dict[device_serial_num][ # noqa
"battery_level"],
"system_version":
device_dict[device_serial_num][ # noqa
"system_version"],
"cmd":
bench_cmd,
"imei":
device_dict[device_serial_num]["imei"],
}
bench_dict[device_serial_num][platform][
model_name][ # noqa
backend][
cpu_thread_num] = bench_record # noqa
# bench_dict[model_name].append(bench_record)
logger.info(bench_record)
return bench_dict
def generate_benchmark_summary(self, bench_dict, is_print_summary=True):
logger.info("==== {} ====".format(
self.generate_benchmark_summary.__name__)) # noqa
summary_header = [
"framework",
"branch",
"commit_id",
"model_name",
"platform",
"soc_code",
"soc_name",
"cpu",
"gpu",
"npu",
"product",
"power_mode",
"backend",
"cpu_thread_num",
"avg",
"max",
"min",
"std_dev",
"battery_level",
"system_version",
"repeats",
"warmup",
"imei",
]
summary_header_str = ",".join(summary_header)
summary = [summary_header_str]
# bench_dict[device_serial_num][platform][model_name][support_backend][cpu_thread_num] = bench_record # noqa
device_serials = list(bench_dict.keys())
for didx in range(len(device_serials)):
serial = device_serials[didx]
soc_info_dict = self.config["device_dict"][serial]["soc_info"]
platforms = list(bench_dict[serial].keys())
for pidx in range(len(platforms)):
platform = platforms[pidx]
model_names = list(bench_dict[serial][platform].keys())
for midx in range(len(model_names)):
model_name = model_names[midx]
backends = list(bench_dict[serial][platform]
[model_name].keys()) # noqa
for bidx in range(len(backends)):
backend = backends[bidx]
cpu_thread_nums = list(
bench_dict[serial][platform][model_name]
[backend].keys() # noqa
) # noqa
for tidx in range(len(cpu_thread_nums)):
cpu_thread_num = cpu_thread_nums[tidx] # noqa
# get record
record_dict = bench_dict[serial][platform][
model_name][ # noqa
backend][cpu_thread_num] # noqa
logger.debug("record_dict:{}".format(record_dict))
record = [
self.config["framework_name"],
self.config["framework_repo_branch"],
self.config["framework_repo_commit_id"],
record_dict["model_name"],
record_dict["platform"],
record_dict["soc_code"],
soc_info_dict["name"],
soc_info_dict["cpu"],
soc_info_dict["gpu"],
soc_info_dict["npu/apu/xpu/dsp"],
record_dict["product"],
record_dict["power_mode"],
self.config["support_backend_id"](
record_dict["backend"]), # noqa
record_dict["cpu_thread_num"],
record_dict["avg"],
record_dict["max"],
record_dict["min"],
record_dict["std_dev"],
record_dict["battery_level"],
record_dict["system_version"],
record_dict["repeats"],
record_dict["warmup"],
record_dict["imei"],
]
record_str = ",".join(map(str, record))
if True:
logger.info(record_str)
summary.append(record_str)
if is_print_summary:
summary_str = "\n".join(summary)
logger.info("\n" + summary_str)
return summary
def run_bench(self):
logger.info("==== {} ====".format(self.run_bench.__name__))
device_work_dir = self.config["device_work_dir"]
device_dict = self.config["device_dict"]
model_dict = self.config["model_dict"]
device_serials = list(device_dict.keys())
model_names = list(model_dict.keys())
benchmark_platform = self.config["benchmark_platform"]
support_backend = self.config["support_backend"]
bench_cmd_pattern = self.config["bench_cmd_pattern"]
bench_dict = dict()
bench_case_idx = 0
for didx in range(len(device_serials)):
device_serial_num = device_serials[didx]
logger.debug(
"didx:{}, serial_num:{}".format(didx, device_serial_num) # noqa
)
for midx in range(len(model_names)):
model_name = model_names[midx]
model_dir = "/".join(
[device_work_dir, os.path.basename(model_dict[model_name])] # noqa
)
logger.debug(
"midx:{}, model_name:{}, model_dir:{}".format(
midx, model_name, model_dir
)
)
bench_dict[model_name] = []
for pidx in range(len(benchmark_platform)):
platform = benchmark_platform[pidx]
device_work_dir_platform = device_work_dir + "/" + platform # noqa
device_benchmark_bin = "/".join(
[
device_work_dir_platform,
os.path.basename(
self.config[platform]["benchmark_bin"]
), # noqa
]
)
logger.debug("pidx:{}, platform:{}".format(pidx, platform))
for bidx in range(len(support_backend)):
backend = support_backend[bidx]
is_cpu = self.config["is_cpu_backend"]
logger.debug(
"bidx:{}, backend:{}".format(bidx, backend)
) # noqa
for tidx in range(
len(self.config["cpu_thread_num"])
if is_cpu(backend)
else 1 # noqa
):
bench_case_idx += 1
logger.info(
"\n\nbench_case_idx(from 1):{}".format(
bench_case_idx
) # noqa
)
cpu_thread_num = self.config["cpu_thread_num"][tidx] # noqa
if self.config["framework_name"] == "tnn":
bench_cmd = bench_cmd_pattern.format(
**{
"serial_num": device_serial_num,
"device_work_dir": device_work_dir_platform, # noqa
"device_benchmark_bin": device_benchmark_bin, # noqa
"model_type": self.config[
"framework_name"
], # noqa
"model_dir": model_dir,
"backend": backend,
"repeats": self.config["repeats"](
backend
), # noqa
"warmup": self.config["warmup"],
"thread_num": cpu_thread_num,
"bind_cpu_idx": device_dict[
device_serial_num
][ # noqa
"bind_cpu_idx"
],
}
)
elif self.config["framework_name"] == "ncnn":
bench_cmd = bench_cmd_pattern.format(
**{
"serial_num": device_serial_num,
"device_benchmark_bin": device_benchmark_bin, # noqa
"model_dir": model_dir,
"repeats": self.config["repeats"](
backend
), # noqa
"warmup": self.config["warmup"],
"thread_num": cpu_thread_num,
"power_mode": self.config[
"power_mode_id"
], # noqa
"gpu_device": backend,
}
)
cmd_handle = run_cmd(
bench_cmd, wait_interval_sec=3 # noqa
) # noqa
perf_dict = self.parse_benchmark(cmd_handle)
# summarize benchmark info
bench_record = {
"soc": device_dict[device_serial_num]["soc"],
"product": device_dict[device_serial_num][ # noqa
"product"
], # noqa
"serial_num": device_serial_num,
"platform": platform,
"model_name": model_name,
"repeats": self.config["repeats"](backend),
"warmup": self.config["warmup"],
"avg": perf_dict["avg"],
"max": perf_dict["max"],
"min": perf_dict["min"],
"backend": backend,
"cpu_thread_num": cpu_thread_num,
"power_mode": self.config["power_mode"],
"bind_cpu_idx": device_dict[device_serial_num][
"bind_cpu_idx"
],
"cpu_max_freqs": device_dict[device_serial_num][ # noqa
"cpu_max_freqs"
],
"battery_level": device_dict[device_serial_num][ # noqa
"battery_level"
],
"system_version": device_dict[
device_serial_num
][ # noqa
"system_version"
],
"cmd": bench_cmd,
"imei": device_dict[device_serial_num]["imei"],
}
bench_dict[model_name].append(bench_record)
logger.info(bench_record)
return bench_dict
