Skip to content

Commit

Permalink
[Misc] Change RedundantReshapesPass and FusionPass logging from info …
Browse files Browse the repository at this point in the history
…to debug (#10308)

Signed-off-by: Tyler Michael Smith <[email protected]>
  • Loading branch information
tlrmchlsmth authored Nov 15, 2024
1 parent bf2ddc6 commit 2885ba0
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions vllm/compilation/fusion.py
Original file line number Diff line number Diff line change
Expand Up @@ -281,11 +281,11 @@ def __call__(self, graph: torch.fx.Graph):
self.dump_graph(graph, "before_fusion")

count = self.patterns.apply(graph)
logger.info("Replaced %s patterns", count)
logger.debug("Replaced %s patterns", count)
self.dump_graph(graph, "after_pattern_match")

# Manually process multi-output matches (and run DCE)
self.process_matches(graph)
logger.info("Post-processed %s matches", len(self.matches))
logger.debug("Post-processed %s matches", len(self.matches))
self.dump_graph(graph, "after_fusion")
self.matches.clear()
2 changes: 1 addition & 1 deletion vllm/compilation/reshapes.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,7 @@ def __call__(self, graph: torch.fx.Graph):
graph.erase_node(node)
count += 1

logger.info("Removed %s no-op reshapes", count)
logger.debug("Removed %s no-op reshapes", count)

self.dump_graph(graph, "after_reshapes")

Expand Down

0 comments on commit 2885ba0

Please sign in to comment.