@@ -220,23 +220,25 @@ def print_delegated_graph(graph_module: torch.fx.GraphModule) -> str:
220
220
%arg2_1 : [num_users=2] = placeholder[target=arg2_1]
221
221
%lowered_module_0 : [num_users=1] = get_attr[target=lowered_module_0]
222
222
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]
229
230
%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 = {})
230
231
%getitem : [num_users=1] = call_function[target=operator.getitem](args = (%executorch_call_delegate, 0), kwargs = {})
231
232
%aten_sub_tensor : [num_users=1] = call_function[target=executorch.exir.dialects.edge._ops.aten.sub.Tensor](args = (%getitem, %arg0_1), kwargs = {})
232
233
%lowered_module_1 : [num_users=1] = get_attr[target=lowered_module_1]
233
234
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]
240
242
%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 = {})
241
243
%getitem_1 : [num_users=1] = call_function[target=operator.getitem](args = (%executorch_call_delegate_1, 0), kwargs = {})
242
244
return [getitem_1]
@@ -253,7 +255,7 @@ def print_delegated_graph(graph_module: torch.fx.GraphModule) -> str:
253
255
if node .op == "get_attr" and node .name .startswith ("lowered_module_" ):
254
256
lowered_module = lowered_module_dict [node .name ]
255
257
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 "
257
259
for node_in_lowered_module in lowered_module .original_module .graph .nodes :
258
260
graph_format_str += (
259
261
f"{ indent * 3 } { node_in_lowered_module .format_node ()} \n "
0 commit comments