add graph_code_verbose_log artifact for fx passes

deyshift · deyshift · commit c3c13e262220 · 2025-05-17T06:35:47.000Z
diff --git a/torch/_dynamo/output_graph.py b/torch/_dynamo/output_graph.py
@@ -154,6 +154,7 @@
 log = logging.getLogger(__name__)
 graph_tabular_log = torch._logging.getArtifactLogger(__name__, "graph")
 graph_code_log = torch._logging.getArtifactLogger(__name__, "graph_code")
+graph_code_verbose_log = torch._logging.getArtifactLogger(__name__, "graph_code_verbose")
 graph_sizes_log = torch._logging.getArtifactLogger(__name__, "graph_sizes")
 trace_call_log = torch._logging.getArtifactLogger(__name__, "trace_call")
 
diff --git a/torch/_logging/_registrations.py b/torch/_logging/_registrations.py
@@ -70,6 +70,10 @@
     "Prints the dynamo traced graph (prior to AOTDispatch) in a table. If you prefer python code use `graph_code` instead. ",
 )
 register_artifact("graph_code", "Like `graph`, but gives you the Python code instead.")
+register_artifact(
+    "graph_code_verbose",
+    "Verbose FX pass logs, e.g. from tensorify_python_scalars and runtime_assert.",
+)
 register_artifact(
     "graph_sizes", "Prints the sizes of all FX nodes in the dynamo graph."
 )
diff --git a/torch/fx/passes/_tensorify_python_scalars.py b/torch/fx/passes/_tensorify_python_scalars.py
@@ -35,7 +35,7 @@
 __all__: list[str] = []
 
 log = logging.getLogger(__name__)
-graph_code_log = torch._logging.getArtifactLogger(__name__, "graph_code")
+graph_code_log = torch._logging.getArtifactLogger(__name__, "graph_code_verbose")
 
 # The general shape of this transformation is to look for Tensor operations
 # that take a backed SymFloat as an argument, and then redo them as tensor
diff --git a/torch/fx/passes/runtime_assert.py b/torch/fx/passes/runtime_assert.py
@@ -28,7 +28,7 @@
 __all__ = ["insert_deferred_runtime_asserts"]
 
 log = logging.getLogger(__name__)
-graph_code_log = torch._logging.getArtifactLogger(__name__, "graph_code")
+graph_code_log = torch._logging.getArtifactLogger(__name__, "graph_code_verbose")
 
 
 def _get_example_value(node: fx.Node) -> Optional[str]:

Original file line number	Diff line number	Diff line change
`@@ -70,6 +70,10 @@`
`70`	`70`	"Prints the dynamo traced graph (prior to AOTDispatch) in a table. If you prefer python code use `graph_code` instead. ",
`71`	`71`	`)`
`72`	`72`	register_artifact("graph_code", "Like `graph`, but gives you the Python code instead.")
	`73`	`+register_artifact(`
	`74`	`+ "graph_code_verbose",`
	`75`	`+ "Verbose FX pass logs, e.g. from tensorify_python_scalars and runtime_assert.",`
	`76`	`+)`
`73`	`77`	`register_artifact(`
`74`	`78`	`"graph_sizes", "Prints the sizes of all FX nodes in the dynamo graph."`
`75`	`79`	`)`