Add support for event tracer in module constructor (#2360)

Summary:
Pull Request resolved: #2360

As titled, default to nullptr

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

Reviewed By: Jack-Khuu

Differential Revision: D54764501

fbshipit-source-id: 1a9475507826a94bc477e01bebf27e2e7979f2f9

Author: Varun Puri

extension/module/module.cpp

@@ -36,10 +36,12 @@ namespace torch::executor {
 
 Module::Module(
     const std::string& file_path,
-    const Module::MlockConfig mlock_config)
+    const Module::MlockConfig mlock_config,
+    std::unique_ptr<EventTracer> event_tracer)
     : file_path_(file_path),
       mlock_config_(mlock_config),
-      memory_allocator_(std::make_unique<util::MallocMemoryAllocator>()) {
+      memory_allocator_(std::make_unique<util::MallocMemoryAllocator>()),
+      event_tracer_(std::move(event_tracer)) {
   runtime_init();
 }
 

extension/module/module.h

@@ -44,7 +44,8 @@ class Module final {
    */
   explicit Module(
       const std::string& file_path,
-      const MlockConfig mlock_config = MlockConfig::UseMlock);
+      const MlockConfig mlock_config = MlockConfig::UseMlock,
+      std::unique_ptr<EventTracer> event_tracer = nullptr);
 
   /**
    * Constructs an instance with the provided data loader and memory allocator.