Skip to content

Commit 0aa4e55

Browse files
committed
fixed the perf issue in the lowering pass
1 parent ed80070 commit 0aa4e55

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

py/torch_tensorrt/dynamo/lowering/passes/remove_num_users_is_0_nodes.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,8 @@ def remove_num_users_is_0_nodes(
2323
and len(node.all_input_nodes) > 0
2424
):
2525
gm.graph.erase_node(node)
26-
gm = clean_up_graph_after_modifications(gm)
26+
27+
gm = clean_up_graph_after_modifications(gm)
2728

2829
logger.debug(f"Removed ops that [num_users=0] nodes:\n{gm.graph}")
2930

0 commit comments

Comments
 (0)