Add support for using ETDumpGen (flatcc) in sdk/executor_runner

Differential Revision: D48975977

fbshipit-source-id: a515e9cdc7ef936ea872d375d65120f2da736385

Author: tarun292

sdk/etdump/etdump_flatcc.cpp

@@ -15,7 +15,7 @@ namespace torch {
 namespace executor {
 
 // Constructor implementation
-ETDumpGen::ETDumpGen(void* buffer, size_t buf_size) {
+ETDumpGen::ETDumpGen() {
   // Initialize the flatcc builder using the buffer and buffer size
   flatcc_builder_init(&builder);
   flatbuffers_buffer_start(&builder, etdump_ETDump_file_identifier);

sdk/etdump/etdump_flatcc.h

@@ -32,7 +32,7 @@ struct etdump_result {
 
 class ETDumpGen : public EventTracer {
  public:
-  ETDumpGen(void* buffer, size_t buf_size);
+  ETDumpGen();
 
   ~ETDumpGen() override;
   void clear_builder();

sdk/runners/executor_runner.cpp

@@ -16,6 +16,7 @@
 #include <executorch/runtime/platform/profiler.h>
 #include <executorch/runtime/platform/runtime.h>
 #include <executorch/sdk/etdump/etdump.h>
+#include <executorch/sdk/etdump/etdump_flatcc.h>
 #include <executorch/util/bundled_program_verification.h>
 #include <executorch/util/util.h>
 #ifdef USE_ATEN_LIB
@@ -304,7 +305,9 @@ int main(int argc, char** argv) {
   //
 
   prof_tok = EXECUTORCH_BEGIN_PROF("load model");
-  Result<Method> method = program->load_method(method_name, &memory_manager);
+  torch::executor::ETDumpGen etdump_gen = torch::executor::ETDumpGen();
+  Result<Method> method =
+      program->load_method(method_name, &memory_manager, &etdump_gen);
   EXECUTORCH_END_PROF(prof_tok);
   ET_CHECK_MSG(
       method.ok(),
@@ -433,5 +436,13 @@ int main(int argc, char** argv) {
     }
   }
 
+  etdump_result result = etdump_gen.get_etdump_data();
+  if (result.buf != nullptr && result.size > 0) {
+    FILE* f = fopen("/home/tkaruturi/temp/etdump_flatcc.etdp", "w+");
+    fwrite((uint8_t*)result.buf, 1, result.size, f);
+    fclose(f);
+    free(result.buf);
+  }
+
   return 0;
 }

sdk/runners/targets.bzl

@@ -48,6 +48,7 @@ def define_common_targets():
                 "//executorch/runtime/executor/test:test_backend_with_delegate_mapping" + aten_suffix,
                 "//executorch/runtime/executor:program" + aten_suffix,
                 "//executorch/sdk/etdump:etdump",
+                "//executorch/sdk/etdump:etdump_flatcc",
                 "//executorch/util:bundled_program_verification" + aten_suffix,
                 "//executorch/extension/data_loader:buffer_data_loader",
                 "//executorch/extension/data_loader:file_data_loader",