fix passes.ConstantFolding (#110)

openvinotoolkit · Jun 29, 2023 · 2fb4b0e · 2fb4b0e
1 parent 215b3c6
commit 2fb4b0e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/model_analyzer/openvino_core_service.py b/model_analyzer/openvino_core_service.py
@@ -5,6 +5,7 @@
 
 # pylint: disable=import-error
 from openvino.runtime import CompiledModel, Core, Model
+from openvino.runtime.passes import ConstantFolding
 from openvino.runtime.passes import Manager
 
 
@@ -35,7 +36,7 @@ def compile_model(self, model: Model, device: str) -> CompiledModel:
     def pass_constant_folding(model: Model):
         try:
             pass_manager = Manager()
-            pass_manager.register_pass('ConstantFolding')
+            pass_manager.register_pass(ConstantFolding())
             pass_manager.set_per_pass_validation(False)
             pass_manager.run_passes(model)
         # pylint: disable=broad-except