wip

Author: chmjkb

common/rnexecutorch/bindings/ExecutorchModule.cpp

@@ -0,0 +1,43 @@
+#include "ExecutorchModule.h"
+
+#include <fmt/core.h>
+#include <rnexecutorch/Log.h>
+
+namespace rnexecutorch {
+
+using ::executorch::extension::Module;
+using ::executorch::runtime::Error;
+
+ExecutorchModule::ExecutorchModule(const std::string &modelSource,
+                                   facebook::jsi::Runtime *runtime)
+    : module(std::make_unique<Module>(
+          modelSource, Module::LoadMode::MmapUseMlockIgnoreErrors)),
+      runtime(runtime) {
+  Error loadError = module->load();
+  if (loadError != Error::Ok) {
+    throw std::runtime_error("Couldn't load the model, error: " +
+                             std::to_string(static_cast<uint32_t>(loadError)));
+  }
+}
+
+std::vector<int32_t> ExecutorchModule::getInputShape(std::string method_name,
+                                                     int index) {
+  auto method_meta = module->method_meta(method_name);
+  if (!method_meta.ok()) {
+    throw std::runtime_error(
+        fmt::format("Failed to load method with name {}", method_name));
+  }
+
+  std::vector<int32_t> input_shape;
+  auto input_meta = method_meta->input_tensor_meta(index);
+  if (!input_meta.ok()) {
+    throw std::runtime_error(
+        fmt::format("Failed to load forward input {}", index));
+  }
+
+  for (auto size : input_meta->sizes()) {
+    input_shape.push_back(size);
+  }
+  return input_shape;
+}
+} // namespace rnexecutorch

common/rnexecutorch/bindings/ExecutorchModule.h

@@ -0,0 +1,22 @@
+#pragma once
+
+#include <string>
+
+#include <executorch/extension/module/module.h>
+#include <fmt/core.h>
+#include <jsi/jsi.h>
+
+namespace rnexecutorch {
+
+class ExecutorchModule {
+public:
+  ExecutorchModule(const std::string &modelSource,
+                   facebook::jsi::Runtime *runtime);
+  std::vector<int32_t> getInputShape(std::string method_name, int index);
+
+protected:
+  std::unique_ptr<executorch::extension::Module> module;
+  facebook::jsi::Runtime *runtime;
+};
+
+} // namespace rnexecutorch