From de64fe5f0d93b1d036a37c6cee22dcaff13663d9 Mon Sep 17 00:00:00 2001 From: Benjamin Fineran Date: Thu, 8 Feb 2024 15:31:42 -0500 Subject: [PATCH] make sure benchmark_* imports are the functions, not modules (#1593) --- src/deepsparse/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/deepsparse/__init__.py b/src/deepsparse/__init__.py index fab294bfaa..46c49b236d 100644 --- a/src/deepsparse/__init__.py +++ b/src/deepsparse/__init__.py @@ -40,6 +40,7 @@ from .subgraph_execute import * from .analyze import analyze from .evaluation.evaluator import evaluate -from .benchmark import benchmark_model, benchmark_pipeline +from .benchmark.benchmark_model import benchmark_model +from .benchmark.benchmark_pipeline import benchmark_pipeline _analytics.send_event("python__init")