From eda3cf549a9648455e3f1f97d6698f1a373afe0c Mon Sep 17 00:00:00 2001 From: NotBad Date: Sun, 6 Sep 2020 23:55:12 +0800 Subject: [PATCH] support tflite. fix bench_case_sum calc --- core/engine.py | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) diff --git a/core/engine.py b/core/engine.py index 7ad8136..6bb6089 100644 --- a/core/engine.py +++ b/core/engine.py @@ -483,14 +483,24 @@ def run_bench_for_single_thread_func( * sum( [ len(self.config["cpu_thread_num"]) * cpu_backend_num - + len( - set(support_backend) - set("ARM") - set("ARM_XNNPACK") - ) # noqa + + 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"])