print-graph

Author: lucylq

examples/models/llama/export_llama_lib.py

@@ -800,6 +800,9 @@ def _export_llama(args) -> LLMEdgeManager:  # noqa: C901
             # pyre-fixme[16]: Module `backends` has no attribute `qualcomm`.
             canonicalize_program(builder.edge_manager.exported_program())
 
+        print(builder.edge_manager.exported_program().graph_module)
+        print(builder.edge_manager.exported_program().graph_module.print_readable())
+
         builder = builder.to_executorch()
 
     if args.profile_memory: