print graph break in llama -- a different if-else branch (#2493)

Summary:
Pull Request resolved: #2493

D54969119 added the print to only one of the if branches (line 315-319), but the other branch is also used.

Reviewed By: cccclai

Differential Revision: D55033430

fbshipit-source-id: 4b42fa750245555e353838d2bcca86b689b16c26

Author: Olivia-liu

examples/models/llama2/builder.py

@@ -302,6 +302,11 @@ def to_backend(
                 assert self.edge_manager is not None
                 self.edge_manager = self.edge_manager.to_backend(p)
                 if self.verbose:
+                    logging.info(
+                        print_delegated_graph(
+                            self.edge_manager.exported_program().graph_module
+                        )
+                    )
                     logging.info(f"Applied partitioners: {key}")
         elif isinstance(partitioner, Partitioner):
             assert self.edge_manager is not None

exir/backend/utils.py

@@ -258,7 +258,6 @@ def print_delegated_graph(graph_module: torch.fx.GraphModule) -> str:
                 graph_format_str += (
                     f"{indent * 3}{node_in_lowered_module.format_node()}\n"
                 )
-    print(graph_format_str)
     return graph_format_str