fix subgraph unintended input elimination

inisis · Oct 16, 2024 · f6c77ac · f6c77ac
1 parent d8061d0
commit f6c77ac
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/onnxslim/third_party/onnx_graphsurgeon/ir/graph.py b/onnxslim/third_party/onnx_graphsurgeon/ir/graph.py
@@ -405,7 +405,7 @@ def cleanup_subgraphs():
                 subgraph.cleanup(
                     remove_unused_node_outputs=remove_unused_node_outputs,
                     recurse_subgraphs=recurse_subgraphs,
-                    remove_unused_graph_inputs=remove_unused_graph_inputs,
+                    remove_unused_graph_inputs=False,
                     recurse_functions=False,  # Only cleanup functions once
                 )
 
@@ -435,7 +435,7 @@ def cleanup_subgraphs():
                 if inp in used_tensors or not remove_unused_graph_inputs:
                     inputs.append(inp)
                 else:
-                    G_LOGGER.ultra_verbose(f"Removing unused input: {inp}")
+                    G_LOGGER.debug(f"Removing unused input: {inp}")
             self.inputs = inputs
 
             nodes = []
@@ -645,7 +645,7 @@ def add_to_tensor_map(tensor):
 
                     if check_duplicates:
                         G_LOGGER.critical(msg)
-                    G_LOGGER.warning(msg)
+                    # G_LOGGER.warning(msg)
 
                 tensor_map[tensor.name] = tensor