Print AOT Autograd graph name when accuracy failed

Signed-off-by: Edward Z. Yang <ezyangmeta.com>

ghstack-source-id: bc335ef7657ef28042665fc04f3c49a67e7ae321
Pull Request resolved: #99366

torch/_dynamo/repro/after_aot.py

@@ -51,6 +51,10 @@ def debug_wrapper(gm, example_inputs, **kwargs):
 
         compiler_fn = functools.partial(unconfigured_compiler_fn, **kwargs)
 
+        from torch._functorch.aot_autograd import get_aot_graph_name
+
+        graph_name = get_aot_graph_name()
+
         # TODO: why do we need to deepcopy the original graph?
         orig_graph = copy.deepcopy(gm.graph)
         assert config.repro_after in ("dynamo", "aot", None)
@@ -104,7 +108,9 @@ def deferred_for_real_inputs(real_inputs):
                         "Accuracy minification is supported for inductor only"
                     )
                 if backend_aot_accuracy_fails(gm, real_inputs, compiler_fn):
-                    log.warning("Accuracy failed for the AOT Autograd graph")
+                    log.warning(
+                        "Accuracy failed for the AOT Autograd graph %s", graph_name
+                    )
                     dump_compiler_graph_state(
                         fx.GraphModule(gm, orig_graph),
                         copy_tensor_attrs,