Re-enable generate etrecord (#2448)

Summary:
Pull Request resolved: #2448

So this was disable because Martin had an import issue, but it wasn't reproducible. Checked with Martin offline and confirmed that he's able to run and no longer see the issue.

bypass-github-export-checks
bypass-github-pytorch-ci-checks
bypass-github-executorch-ci-checks

Reviewed By: Jack-Khuu

Differential Revision: D54929961

fbshipit-source-id: 6041c3a95b01de58ad4bac4337b0c81cd4fb0317

Author: Olivia-liu

examples/models/llama2/export_llama_lib.py

@@ -23,7 +23,7 @@
     XnnpackDynamicallyQuantizedPartitioner,
 )
 
-# from executorch.sdk.etrecord import generate_etrecord
+from executorch.sdk.etrecord import generate_etrecord
 from executorch.util.activation_memory_profiler import generate_memory_trace
 from sentencepiece import SentencePieceProcessor
 from torch.ao.quantization.quantizer import Quantizer
@@ -496,20 +496,19 @@ def _export_llama(modelname, args) -> str:  # noqa: C901
         if not builder_exported_to_edge.edge_manager:
             raise ValueError("Unable to generate etrecord due to missing edge manager.")
 
-        # logging.info("Generating etrecord")
+        logging.info("Generating etrecord")
         # Copy the edge manager which will be serialized into etrecord. This is memory-wise expensive.
         edge_manager_copy = copy.deepcopy(builder_exported_to_edge.edge_manager)
         builder = builder_exported_to_edge.to_backend(partitioners).to_executorch()
 
         # Generate ETRecord
         if edge_manager_copy:
-            # generate_etrecord(
-            #     etrecord_path="etrecord.bin",
-            #     edge_dialect_program=edge_manager_copy,
-            #     executorch_program=builder.export_program,
-            # )
-            # logging.info("Generated etrecord.bin")
-            pass
+            generate_etrecord(
+                etrecord_path="etrecord.bin",
+                edge_dialect_program=edge_manager_copy,
+                executorch_program=builder.export_program,
+            )
+            logging.info("Generated etrecord.bin")
     else:
         builder = builder_exported_to_edge.to_backend(partitioners).to_executorch()