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

Signed-off-by: youkaichao <[email protected]>
ruisearch42 · Dec 11, 2024 · 66aaa77 · 66aaa77
1 parent d643c2a
commit 66aaa77
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/vllm/compilation/backends.py b/vllm/compilation/backends.py
@@ -287,9 +287,11 @@ def __call__(self, graph: fx.GraphModule, example_inputs) -> Callable:
             graph, self.compilation_config.splitting_ops)
 
         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",
-                                                  self.split_gm))
+
+        # depyf will hook lazy_format_graph_code and dump the graph
+        # 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(
             self.piecewise_graphs)