[cherry-pick] examples/benchmark fix for resnet50 example (#1597) (#1598

)
neuralmagic · Feb 12, 2024 · a15543e · a15543e
1 parent 30ba1ca
commit a15543e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/benchmark/resnet50_benchmark.py b/examples/benchmark/resnet50_benchmark.py
@@ -47,7 +47,8 @@
 
 import numpy
 
-from deepsparse import benchmark_model, cpu
+from deepsparse import cpu
+from deepsparse.engine import benchmark_model
 
 
 CORES_PER_SOCKET, AVX_TYPE, VNNI = cpu.cpu_details()