Move gen_test_files from AIBench to ExecuTorch examples (#1107)

Summary:
Pull Request resolved: #1107

This script is useful to generate a sample ETRecord from the example models provided in the repo. This should help users explore the features of the ETrecord without needing to first export a .pte model.

Reviewed By: Olivia-liu

Differential Revision: D50521582

fbshipit-source-id: bad49adfd3dc106c1dcdc58bea0a26ec98022319

Author: Varun Puri

examples/sdk/scripts/gen_sample_etrecord.py

@@ -0,0 +1,43 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+# Generate fixture files
+from pathlib import Path
+
+import executorch.exir as exir
+from executorch.exir import ExecutorchBackendConfig
+
+from executorch.exir.tests.models import BasicSinMax
+from executorch.sdk.etrecord import generate_etrecord
+from torch.export import export
+
+
+def get_module_path() -> Path:
+    return Path(__file__).resolve().parents[0]
+
+
+def gen_etrecord():
+    f = BasicSinMax()
+    aten_dialect = export(
+        f,
+        f.get_random_inputs(),
+    )
+    edge_program = exir.to_edge(
+        aten_dialect, compile_config=exir.EdgeCompileConfig(_check_ir_validity=False)
+    )
+    et_program = edge_program.to_executorch(ExecutorchBackendConfig(passes=[]))
+    generate_etrecord(
+        str(get_module_path()) + "/etrecord.bin",
+        edge_dialect_program=edge_program,
+        executorch_program=et_program,
+        export_modules={
+            "aten_dialect_output": aten_dialect,
+        },
+    )
+
+
+if __name__ == "__main__":
+    gen_etrecord()