Fix

anijain2305 · anijain2305 · commit 06e94c8739a9 · 2022-08-30T18:28:54.000Z
diff --git a/torchdynamo/debug_utils.py b/torchdynamo/debug_utils.py
@@ -462,4 +462,6 @@ def debug_wrapper(gm, example_inputs, **kwargs):
 
         return compiled_gm
 
+    debug_wrapper._torchdynamo_orig_callable = compiler_fn
+
     return debug_wrapper
