[Core] Add dynamo_timed tracing for print_readable (#40834)

Signed-off-by: Frederik Gossen <frgossen@meta.com>
2026-08-21 09:07:21 -07:00
parent e00a034ad1
commit 47cd1c8885
1 changed files with 6 additions and 5 deletions
@@ -1264,11 +1264,12 @@ class VllmBackend:
            # code adapted from
            # https://github.com/thuml/depyf/blob/dab831108a752d1facc00acdd6d4243891845c37/depyf/explain/patched_lazy_format_graph_code.py#L30
            # use `print_readable` because it can include submodules
-            src = (
-                "from __future__ import annotations\nimport torch\n"
-                + self.split_gm.print_readable(print_output=False)
-            )
-            src = src.replace("<lambda>", "GraphModule")
+            with dynamo_timed("vllm_print_readable"):
+                src = (
+                    "from __future__ import annotations\nimport torch\n"
+                    + self.split_gm.print_readable(print_output=False)
+                )
+                src = src.replace("<lambda>", "GraphModule")
            with open(graph_path, "w") as f:
                f.write(src)