Skip to content
This repository was archived by the owner on Nov 25, 2023. It is now read-only.

Commit eff4473

Browse files
committed
2 parents b446f88 + 597793a commit eff4473

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

core/engine.py

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -483,14 +483,24 @@ def run_bench_for_single_thread_func(
483483
* sum(
484484
[
485485
len(self.config["cpu_thread_num"]) * cpu_backend_num
486-
+ len(
487-
set(support_backend) - set("ARM") - set("ARM_XNNPACK")
488-
) # noqa
486+
+ len(set(support_backend) - set(["ARM", "ARM_XNNPACK"])) # noqa
489487
]
490488
)
491489
)
492490
logger.info("len(platform):{}".format(len(platforms)))
493491
logger.info("len(model_names):{}".format(len(model_names)))
492+
logger.info(
493+
'len(self.config["cpu_thread_num"]) * cpu_backend_num:{}'.format(
494+
len(self.config["cpu_thread_num"]) * cpu_backend_num
495+
)
496+
)
497+
logger.info("support_backend:{}".format(support_backend))
498+
logger.info(
499+
'len(set(support_backend) - set("ARM") - set("ARM_XNNPACK")):{}'.format( # noqa
500+
len(set(support_backend) - set("ARM") - set("ARM_XNNPACK"))
501+
)
502+
)
503+
logger.info("bench_case_num:{}".format(bench_case_num))
494504
logger.info(
495505
'len(self.config["cpu_thread_num"]) if "CPU" in support_backend else 0:{}'.format( # noqa
496506
len(self.config["cpu_thread_num"])

0 commit comments

Comments
 (0)