Executorch Wasm Internal Build

Summary:
Added internal build scripts for Executorch Wasm and unit tests.

Due to some intricacies the following things were done:
- Emscripten build options don't work properly on `fbcode`, so only `xplat` targets were made.
- Export scripts are only supported on `fbcode`, but do not work properly with the Emscripten build mode. Because of this, the test models for unit testing need to be built separately and copied into `xplat`.
- Jest is configured to not search for unit tests within `buck-out`, so they must also be copied into `xplat`.
- These are handled by `xplat_build.sh`, which can be run with `yarn build`. These files can also be deleted with `yarn clean`.
- Unit tests won't build without `test_models` directory so dummy file is necessary to pass build rule CI.

Differential Revision: D81276482

Author: Conarnar

extension/wasm/test/.gitignore

@@ -0,0 +1,3 @@
+# Internal build generated files
+__tests__/
+test_models/*.pte

extension/wasm/test/TARGETS

@@ -0,0 +1,69 @@
+load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
+
+oncall("executorch")
+
+export_genrule_cmd = [
+    "$(exe //executorch/examples/portable/scripts:export)",
+    "--model_name={model_name}",
+    "--output_dir=$OUT",
+]
+
+runtime.genrule(
+    name = "add_mul.pte",
+    outs = {
+        "add_mul.pte": ["add_mul.pte"],
+    },
+    cmd = " ".join(export_genrule_cmd).format(model_name="add_mul"),
+    default_outs = ["."],
+)
+
+runtime.genrule(
+    name = "add.pte",
+    outs = {
+        "add.pte": ["add.pte"],
+    },
+    cmd = " ".join(export_genrule_cmd).format(model_name="add"),
+    default_outs = ["."],
+)
+
+runtime.python_binary(
+    name = "test_model",
+    srcs = [
+        "test_model.py",
+    ],
+    main_module = "executorch.extension.wasm.test.test_model",
+    deps = [
+        "//executorch/extension/export_util:export_util",
+    ],
+)
+
+test_pte_genrule_cmd = [
+    "$(exe :test_model)",
+    "$OUT/test.pte",
+]
+
+runtime.genrule(
+    name = "test.pte",
+    outs = {
+        "test.pte": ["test.pte"],
+    },
+    cmd = " ".join(test_pte_genrule_cmd),
+    default_outs = ["."],
+)
+
+models_genrule_cmd = [
+    "mkdir -p $OUT &&",
+    "cp $(location :add.pte)/add.pte $OUT/add.pte &&",
+    "cp $(location :add_mul.pte)/add_mul.pte $OUT/add_mul.pte &&",
+    "cp $(location :test.pte)/test.pte $OUT/test.pte",
+]
+
+runtime.genrule(
+    name = "models",
+    outs = {
+        "add.pte": ["add.pte"],
+        "add_mul.pte": ["add_mul.pte"],
+        "test.pte": ["test.pte"],
+    },
+    cmd = " ".join(models_genrule_cmd),
+)

extension/wasm/test/package.json

@@ -1,5 +1,12 @@
 {
+  "name": "wasm-test",
+  "private": true,
   "scripts": {
-    "test": "jest"
+    "build": "bash xplat_build.sh",
+    "test": "jest --verbose",
+    "clean": "rm -rf __tests__ node_modules test_models/*.pte"
+  },
+  "devDependencies": {
+    "jest": "^30.1.3"
   }
 }

extension/wasm/test/test_models/dummy

@@ -0,0 +1,53 @@
+#!/usr/bin/env bash
+# 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.
+
+# This script is used to build unit tests internally. It is not intended to be used in OSS.
+
+set -xueo pipefail
+
+THIS_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+OUTPUT_DIR="${THIS_DIR}/__tests__"
+TEST_MODEL_DIR="${THIS_DIR}/test_models/"
+
+if [[ "${THIS_DIR}" != *"xplat"* ]]; then
+    echo "This script must be run from xplat, not fbcode or oss"
+    exit 1
+fi
+
+ENABLE_ETDUMP=0
+for arg in "$@"; do
+    if [[ "$arg" == "etdump" ]]; then
+        ENABLE_ETDUMP=1
+    else
+        echo "Unknown argument: $arg"
+        exit 1
+    fi
+done
+
+# Build the models
+# Using fbcode because the Python scripts are only supported in fbcode
+MODEL_TARGET_DIR=$(buck2 build fbcode//executorch/extension/wasm/test:models --show-full-output | awk '{print $2}')
+
+mkdir -p "${TEST_MODEL_DIR}"
+cp ${MODEL_TARGET_DIR}/*.pte ${TEST_MODEL_DIR}
+
+if (( ENABLE_ETDUMP != 0 )); then
+    ETDUMP_OPTIONS="-DET_EVENT_TRACER_ENABLED=1"
+    WASM_TARGET_NAME="wasm_etdump.test"
+else
+    ETDUMP_OPTIONS=""
+    WASM_TARGET_NAME="wasm.test"
+fi
+
+# Emscripten build options don't work properly on fbcode
+BUILD_TARGET_DIR=$(dirname $(buck2 build :${WASM_TARGET_NAME}.js --show-full-output -c "cxx.extra_cxxflags=-DET_LOG_ENABLED=0 $ETDUMP_OPTIONS" | awk '{print $2}'))
+
+mkdir -p "${OUTPUT_DIR}"
+cp ${BUILD_TARGET_DIR}/${WASM_TARGET_NAME}.js ${OUTPUT_DIR}
+cp ${BUILD_TARGET_DIR}/${WASM_TARGET_NAME}.wasm ${OUTPUT_DIR}
+
+yarn install