Module test working

Author: Conarnar

CMakeLists.txt

@@ -673,6 +673,10 @@ if(EXECUTORCH_BUILD_PYBIND)
   )
 endif()
 
+if(EXECUTORCH_BUILD_WASM)
+  add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/extension/wasm)
+endif()
+
 if(EXECUTORCH_BUILD_EXTENSION_TRAINING)
   add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/extension/training)
 endif()

extension/wasm/CMakeLists.txt

@@ -0,0 +1,47 @@
+
+cmake_minimum_required(VERSION 3.24) # 3.24 is required for WHOLE_ARCHIVE
+
+project(executorch_wasm)
+
+if(NOT CMAKE_CXX_STANDARD)
+  set(CMAKE_CXX_STANDARD 17)
+endif()
+
+if(NOT EMSCRIPTEN)
+  message(FATAL_ERROR "Emscripten is required to build this target")
+endif()
+
+# Source root directory for executorch.
+if(NOT EXECUTORCH_ROOT)
+  set(EXECUTORCH_ROOT ${CMAKE_CURRENT_SOURCE_DIR}/../..)
+endif()
+
+include(${EXECUTORCH_ROOT}/tools/cmake/Utils.cmake)
+set(_common_compile_options -Wno-deprecated-declarations -fPIC)
+set(_common_include_directories ${EXECUTORCH_ROOT}/..)
+
+set(link_libraries)
+list(
+  APPEND
+  link_libraries
+  embind
+  executorch_core
+  extension_data_loader
+  portable_ops_lib
+  extension_module_static
+  extension_tensor
+  extension_runner_util
+)
+
+add_executable(executorch_wasm wasm_bindings.cpp)
+
+target_compile_options(executorch_wasm PUBLIC ${_common_compile_options})
+target_include_directories(executorch_wasm PUBLIC ${_common_include_directories})
+target_link_libraries(executorch_wasm PUBLIC ${link_libraries})
+target_link_options(executorch_wasm PUBLIC -sALLOW_MEMORY_GROWTH=1 --embed-file=${CMAKE_CURRENT_SOURCE_DIR}/[email protected])
+
+add_custom_target(executorch_wasm_test ALL
+  COMMAND cp ${CMAKE_CURRENT_SOURCE_DIR}/test_module.js ${CMAKE_CURRENT_BINARY_DIR}/test_module.js
+  DEPENDS executorch_wasm
+  COMMENT "Copying test_module.js to build output directory"
+)

extension/wasm/build_wasm.sh

@@ -0,0 +1,12 @@
+
+cd "$(dirname "${BASH_SOURCE[0]}")/../../"
+mkdir -p cmake-out-wasm
+cd cmake-out-wasm
+emcmake cmake -DEXECUTORCH_BUILD_WASM=ON \
+    -DEXECUTORCH_BUILD_EXTENSION_DATA_LOADER=ON \
+    -DEXECUTORCH_BUILD_EXTENSION_FLAT_TENSOR=ON \
+    -DEXECUTORCH_BUILD_DEVTOOLS=ON \
+    -DEXECUTORCH_BUILD_EXTENSION_MODULE=ON \
+    -DEXECUTORCH_BUILD_EXTENSION_TENSOR=ON \
+    -DEXECUTORCH_BUILD_EXTENSION_RUNNER_UTIL=ON ..
+make executorch_wasm_test -j32

extension/wasm/test_module.js

@@ -0,0 +1,27 @@
+
+const et = require("./executorch_wasm");
+et.onRuntimeInitialized = () => {
+    module = et.Module.load("model.pte");
+    var methods = module.getMethods();
+    console.log(methods);
+    var method = methods[0];
+    var methodMeta = module.getMethodMeta(method);
+    var inputs = [];
+    for (var i = 0; i < methodMeta.numInputs; i++) {
+        var tensor = et.FloatTensor.ones(methodMeta.inputTensorMeta(i).sizes);
+        console.log("input", i, tensor.getData(), tensor.getSizes());
+        inputs.push(tensor);
+    }
+    var output = module.execute(method, inputs);
+
+    for (var i = 0; i < inputs.length; i++) {
+        inputs[i].delete();
+    }
+
+    for (var i = 0; i < output.length; i++) {
+        console.log("output", i, output[i].getData(), output[i].getSizes());
+        output[i].delete();
+    }
+
+    module.delete();
+}