File tree 1 file changed +4
-1
lines changed
1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -101,6 +101,7 @@ def debugprint(
101
101
print_view_map : bool = False ,
102
102
print_memory_map : bool = False ,
103
103
print_fgraph_inputs : bool = False ,
104
+ print_inner_graphs : bool = True ,
104
105
) -> str | TextIO :
105
106
r"""Print a graph as text.
106
107
@@ -161,6 +162,8 @@ def debugprint(
161
162
Whether to set both `print_destroy_map` and `print_view_map` to ``True``.
162
163
print_fgraph_inputs
163
164
Print the inputs of `FunctionGraph`\s.
165
+ print_inner_graphs
166
+ Whether to print the inner graphs of `Op`\s
164
167
165
168
Returns
166
169
-------
@@ -322,7 +325,7 @@ def debugprint(
322
325
print_view_map = print_view_map ,
323
326
)
324
327
325
- if len (inner_graph_vars ) > 0 :
328
+ if len (inner_graph_vars ) > 0 and print_inner_graphs :
326
329
print ("" , file = _file )
327
330
prefix = ""
328
331
new_prefix = prefix + " ← "
You can’t perform that action at this time.
0 commit comments