Skip to content

GitLab

Unverified Commit 66aaa772 authored by youkaichao's avatar youkaichao Committed by GitHub
Browse files

[torch.compile] remove graph logging in ci (#11110) 


Signed-off-by: default avataryoukaichao <youkaichao@gmail.com>
parent d643c2ab
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
vllm/compilation/backends.py
View file @ 66aaa772
...@@ -287,9 +287,11 @@ class VllmBackend: ...@@ -287,9 +287,11 @@ class VllmBackend:
graph, self.compilation_config.splitting_ops) graph, self.compilation_config.splitting_ops)
from torch._dynamo.utils import lazy_format_graph_code from torch._dynamo.utils import lazy_format_graph_code
logger.debug("%s", lazy_format_graph_code("before split", self.graph))
logger.debug("%s", lazy_format_graph_code("after split", # depyf will hook lazy_format_graph_code and dump the graph
self.split_gm)) # for debugging, no need to print the graph here
lazy_format_graph_code("before split", self.graph)
lazy_format_graph_code("after split", self.split_gm)
compilation_counter.num_piecewise_graphs_seen += len( compilation_counter.num_piecewise_graphs_seen += len(
self.piecewise_graphs) self.piecewise_graphs)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment