Update on "[wip][compiled autograd] runtime wrapper for verbose debug…

…ging" cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx chenyang78 kadeng chauhang [ghstack-poisoned]
pytorch · May 2, 2024 · c39d929 · c39d929
1 parent d83cdbd
commit c39d929
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch/_dynamo/compiled_autograd.py b/torch/_dynamo/compiled_autograd.py
@@ -222,8 +222,8 @@ def end_capture(self, outputs):
         )
 
         def runtime_wrapper(compiled_fn, inputs, sizes, hooks):
-            breakpoint()
-            # insert inspecting code here
+            # insert debug code here
+            verbose_log.debug(f"lifted hooks={hooks}")
             return compiled_fn(inputs, sizes, hooks)
 
         return runtime_wrapper, self.compiler_fn(graph)