live debugging

Signed-off-by: youkaichao <youkaichao@gmail.com>
thuml · Dec 6, 2024 · d1c71d6 · d1c71d6
1 parent 810a8fb
commit d1c71d6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/depyf/explain/patched_lazy_format_graph_code.py b/depyf/explain/patched_lazy_format_graph_code.py
@@ -30,6 +30,7 @@ def patched_lazy_format_graph_code(name, gm, maybe_id=None, **kwargs):
     src = "from __future__ import annotations\nimport torch\n" + \
         gm.print_readable(print_output=False)
     src = src.replace("<lambda>", "GraphModule")
+    print(src)
     try:
         compile(src, "noname", "exec")
     except Exception as e: