More descriptive graph diagram names in svg (pytorch#106146)

Pull Request resolved: pytorch#106146
Approved by: https://github.com/jansel, https://github.com/Chillee

torch/_inductor/config.py

@@ -413,7 +413,7 @@ class trace:
     output_code = True
 
     # SVG figure showing post-fusion graph
-    graph_diagram = False
+    graph_diagram = os.environ.get("INDUCTOR_POST_FUSION_SVG", "0") == "1"
 
     # Store cProfile (see snakeviz to view)
     compile_profile = False

torch/_inductor/debug.py

@@ -96,7 +96,6 @@ def func1(*args):
 
     FusionMeta = collections.namedtuple("FusionMeta", ["group", "snode", "type"])
 
-    func_dict = {s: get_fake_func(s) for s in ["extern", "nop", "compute", "fused"]}
     buf_to_fx_node = {}
     graph = torch.fx.Graph()
     first_node = None
@@ -122,7 +121,12 @@ def func1(*args):
             group = snode.group
         else:
             raise RuntimeError("Unknown node type")
-        node_func = func_dict[node_type]
+
+        fused_name = torch._inductor.utils.get_fused_kernel_name(
+            snode.get_nodes(), "original_aten"
+        )
+        func_name = f"{node_type}: {fused_name}"
+        node_func = get_fake_func(func_name)
         fx_node = graph.call_function(node_func, args=(), kwargs=None)
 
         def in_output(snode):