biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

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

Browse Source 
Signed-off-by: youkaichao <youkaichao@gmail.com>
This commit is contained in:
youkaichao
2024-12-11 10:59:50 -08:00
committed by GitHub
parent d643c2aba1
commit 66aaa7722d
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
vllm/compilation/backends.py
View File

@@ -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)