Skip to content

Commit ec6b88a

Browse files
Olivia-liufacebook-github-bot
authored andcommitted
Add a new line for lowered model in print_delegated_graph()
Summary: Add the missed \n because otherwise the first row of operator after "lowered_graph()" is at a weird position Reviewed By: mergennachin Differential Revision: D55213327 fbshipit-source-id: 9da3376dfb011528be145d06425d2fcef63b2352
1 parent 9f552af commit ec6b88a

File tree

1 file changed

+15
-13
lines changed

1 file changed

+15
-13
lines changed

exir/backend/utils.py

Lines changed: 15 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -220,23 +220,25 @@ def print_delegated_graph(graph_module: torch.fx.GraphModule) -> str:
220220
%arg2_1 : [num_users=2] = placeholder[target=arg2_1]
221221
%lowered_module_0 : [num_users=1] = get_attr[target=lowered_module_0]
222222
backend_id: BackendWithCompilerDemo
223-
lowered graph(): %arg0_1 : [num_users=1] = placeholder[target=arg0_1]
224-
%arg1_1 : [num_users=1] = placeholder[target=arg1_1]
225-
%arg2_1 : [num_users=1] = placeholder[target=arg2_1]
226-
%aten_mm_default : [num_users=1] = call_function[target=executorch.exir.dialects.edge._ops.aten.mm.default](args = (%arg0_1, %arg1_1), kwargs = {})
227-
%aten_add_tensor : [num_users=1] = call_function[target=executorch.exir.dialects.edge._ops.aten.add.Tensor](args = (%aten_mm_default, %arg2_1), kwargs = {})
228-
return [aten_add_tensor]
223+
lowered graph():
224+
%arg0_1 : [num_users=1] = placeholder[target=arg0_1]
225+
%arg1_1 : [num_users=1] = placeholder[target=arg1_1]
226+
%arg2_1 : [num_users=1] = placeholder[target=arg2_1]
227+
%aten_mm_default : [num_users=1] = call_function[target=executorch.exir.dialects.edge._ops.aten.mm.default](args = (%arg0_1, %arg1_1), kwargs = {})
228+
%aten_add_tensor : [num_users=1] = call_function[target=executorch.exir.dialects.edge._ops.aten.add.Tensor](args = (%aten_mm_default, %arg2_1), kwargs = {})
229+
return [aten_add_tensor]
229230
%executorch_call_delegate : [num_users=1] = call_function[target=torch.ops.higher_order.executorch_call_delegate](args = (%lowered_module_0, %arg0_1, %arg1_1, %arg2_1), kwargs = {})
230231
%getitem : [num_users=1] = call_function[target=operator.getitem](args = (%executorch_call_delegate, 0), kwargs = {})
231232
%aten_sub_tensor : [num_users=1] = call_function[target=executorch.exir.dialects.edge._ops.aten.sub.Tensor](args = (%getitem, %arg0_1), kwargs = {})
232233
%lowered_module_1 : [num_users=1] = get_attr[target=lowered_module_1]
233234
backend_id: BackendWithCompilerDemo
234-
lowered graph(): %aten_sub_tensor : [num_users=1] = placeholder[target=aten_sub_tensor]
235-
%arg1_1 : [num_users=1] = placeholder[target=arg1_1]
236-
%arg2_1 : [num_users=1] = placeholder[target=arg2_1]
237-
%aten_mm_default_1 : [num_users=1] = call_function[target=executorch.exir.dialects.edge._ops.aten.mm.default](args = (%aten_sub_tensor, %arg1_1), kwargs = {})
238-
%aten_add_tensor_1 : [num_users=1] = call_function[target=executorch.exir.dialects.edge._ops.aten.add.Tensor](args = (%aten_mm_default_1, %arg2_1), kwargs = {})
239-
return [aten_add_tensor_1]
235+
lowered graph():
236+
%aten_sub_tensor : [num_users=1] = placeholder[target=aten_sub_tensor]
237+
%arg1_1 : [num_users=1] = placeholder[target=arg1_1]
238+
%arg2_1 : [num_users=1] = placeholder[target=arg2_1]
239+
%aten_mm_default_1 : [num_users=1] = call_function[target=executorch.exir.dialects.edge._ops.aten.mm.default](args = (%aten_sub_tensor, %arg1_1), kwargs = {})
240+
%aten_add_tensor_1 : [num_users=1] = call_function[target=executorch.exir.dialects.edge._ops.aten.add.Tensor](args = (%aten_mm_default_1, %arg2_1), kwargs = {})
241+
return [aten_add_tensor_1]
240242
%executorch_call_delegate_1 : [num_users=1] = call_function[target=torch.ops.higher_order.executorch_call_delegate](args = (%lowered_module_1, %aten_sub_tensor, %arg1_1, %arg2_1), kwargs = {})
241243
%getitem_1 : [num_users=1] = call_function[target=operator.getitem](args = (%executorch_call_delegate_1, 0), kwargs = {})
242244
return [getitem_1]
@@ -253,7 +255,7 @@ def print_delegated_graph(graph_module: torch.fx.GraphModule) -> str:
253255
if node.op == "get_attr" and node.name.startswith("lowered_module_"):
254256
lowered_module = lowered_module_dict[node.name]
255257
graph_format_str += f"{indent * 2}backend_id: {lowered_module.backend_id}\n"
256-
graph_format_str += f"{indent * 2}lowered graph(): "
258+
graph_format_str += f"{indent * 2}lowered graph():\n"
257259
for node_in_lowered_module in lowered_module.original_module.graph.nodes:
258260
graph_format_str += (
259261
f"{indent * 3}{node_in_lowered_module.format_node()}\n"

0 commit comments

Comments
 (0)