Comments

openvinotoolkit · Oct 17, 2024 · fa1f8f6 · fa1f8f6
1 parent ccbe0f6
commit fa1f8f6
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/nncf/experimental/torch/fx/constant_folding.py b/nncf/experimental/torch/fx/constant_folding.py
@@ -240,15 +240,17 @@ def insert_placerholder_values(self, env: Dict[torch.fx.Node, Any]) -> None:
 
 def constant_fold(
     gm: torch.fx.GraphModule,
-    constraint_fn: Optional[Callable[[torch.fx.Node], bool]] = None,
 ) -> None:
+    """
+    Calcualtes constant subgraphs values and replaces them with a constan node inplace.
+
+    :param gm: Given graph model.
+    """
     with torch.utils._python_dispatch._disable_current_modes():
         cf = ConstantFolder(gm, skip_constructors=True)
         cf.run()
 
         for node, constant in cf.node_replacements.items():
-            if constraint_fn is not None and not constraint_fn(node):
-                continue
             replace_node_with_constant(gm, node, constant)
 
         erased_params = []