Print delegation info in export_llama in verbose

Author: Martin Yuan

examples/models/llama/export_llama_lib.py

@@ -23,6 +23,8 @@
 import torch
 
 from executorch.devtools.etrecord import generate_etrecord
+from executorch.devtools.backend_debug import get_delegation_info
+from tabulate import tabulate
 from executorch.exir.passes.init_mutable_pass import InitializedMutableBufferPass
 
 from executorch.extension.llm.export.builder import DType, LLMEdgeManager
@@ -816,6 +818,13 @@ def _export_llama(args) -> LLMEdgeManager:  # noqa: C901
 
         builder = builder.to_executorch(passes=additional_passes)
 
+    if args.verbose:
+        graph_module = builder.edge_manager.exported_program().graph_module
+        delegation_info = get_delegation_info(graph_module)
+        print(delegation_info.get_summary())
+        df = delegation_info.get_operator_delegation_dataframe()
+        print(tabulate(df, headers="keys", tablefmt="fancy_grid"))
+
     if args.profile_memory:
         generate_memory_trace(builder.export_program, "memory_profile.json")