frozenleaves/vllm
1
0
Fork 0
mirror of https://github.com/vllm-project/vllm.git synced 2025-10-20 14:53:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Misc] Change RedundantReshapesPass and FusionPass logging from info to debug (#10308)

Browse Source 
Signed-off-by: Tyler Michael Smith <tyler@neuralmagic.com>
This commit is contained in:
Tyler Michael Smith
2024-11-14 21:44:26 -05:00
committed by GitHub
parent bf2ddc6610
commit 2885ba0e24
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
vllm/compilation/fusion.py
View File

@ -281,11 +281,11 @@ class FusionPass(InductorPass):
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
vllm/compilation/reshapes.py
View File

@ -53,7 +53,7 @@ class RedundantReshapesPass(InductorPass):
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")