w

Author: fs-eire

build_release_jsep.bat

@@ -0,0 +1 @@
+call build.bat --parallel --config Release --skip_submodule_sync --build_wasm --enable_wasm_simd --enable_wasm_threads --target onnxruntime_webassembly --skip_tests --enable_wasm_api_exception_catching --disable_rtti --build_dir C:\code\o2\build_jsep_rel\ --use_jsep --use_webnn --target onnxruntime_webassembly --skip_tests

build_release_webgpu.bat

@@ -0,0 +1 @@
+call build.bat --parallel --config Release --skip_submodule_sync --build_wasm --enable_wasm_simd --enable_wasm_threads --target onnxruntime_webassembly --skip_tests --enable_wasm_api_exception_catching --disable_rtti --build_dir C:\code\o2\build_webgpu_rel\ --use_webgpu --use_webnn --target onnxruntime_webassembly --skip_tests

cmake/onnxruntime_webassembly.cmake

@@ -406,6 +406,16 @@ jsepDownload:_pp_")
     list(APPEND onnxruntime_webassembly_script_deps "${ONNXRUNTIME_ROOT}/wasm/post-webgpu.js")
   endif()
 
+  if (onnxruntime_USE_WEBNN)
+    target_compile_definitions(onnxruntime_webassembly PRIVATE USE_WEBNN=1)
+    if (NOT onnxruntime_USE_JSEP)
+      target_link_options(onnxruntime_webassembly PRIVATE
+        "SHELL:--post-js \"${ONNXRUNTIME_ROOT}/wasm/post-webnn.js\""
+      )
+      list(APPEND onnxruntime_webassembly_script_deps "${ONNXRUNTIME_ROOT}/wasm/post-webnn.js")
+    endif()
+  endif()
+
   if (onnxruntime_USE_JSEP OR onnxruntime_USE_WEBGPU OR onnxruntime_USE_WEBNN)
     # if any of the above is enabled, we need to use the asyncify library
     target_link_options(onnxruntime_webassembly PRIVATE

js/web/lib/wasm/wasm-core-impl.ts

@@ -111,6 +111,10 @@ export const initEp = async (env: Env, epName: string): Promise<void> => {
     });
   }
 
+  if (epName === 'webnn' && BUILD_DEFS.USE_WEBGPU_EP) {
+    // This condition means that `--use_webgpu` and `--use_webnn` flags are both set, and `--use_jsep` is not set.
+  }
+
   if (!BUILD_DEFS.DISABLE_JSEP) {
     // eslint-disable-next-line @typescript-eslint/no-require-imports, @typescript-eslint/no-var-requires
     const initJsep = require('./jsep/init').init;

onnxruntime/wasm/post-webnn.js

@@ -0,0 +1,83 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+//
+// This file contains the post-run code for the ORT WebAssembly module. The code in this file will be injected into the
+// final module using Emscripten's `--post-js` option.
+//
+// This file will only be used in build with flag `--use_webnn`.
+
+/**
+ * This function is called only once when initializing the WebNN backend.
+ *
+ * @param params WebNN initialization parameters.
+ */
+Module["webnnInit"] = (params) => {
+  // Functions called from EM_ASM need to be assigned in a way that can be minified.
+  // Functions called via emscripten::val::module_property need to be assigned by name so that the minifier doesn't
+  // change the name.
+
+  const backend = params[0];
+  [
+    Module.webnnReserveTensorId,
+    Module.webnnReleaseTensorId,
+    Module["webnnEnsureTensor"],
+    Module.webnnUploadTensor,
+    Module["webnnDownloadTensor"],
+    Module["webnnEnableTraceEvent"],
+  ] = params.slice(1);
+
+  // This function is called from both JS and an EM_ASM block, it needs both a minifiable name and an explicit name.
+  Module["webnnReleaseTensorId"] = Module.webnnReleaseTensorId;
+  Module["webnnUploadTensor"] = Module.webnnUploadTensor;
+
+  // Functions called from JS also need to have explicit names.
+  Module["webnnOnRunStart"] = (sessionId) => {
+    return backend["onRunStart"](sessionId);
+  };
+  Module["webnnOnRunEnd"] = backend["onRunEnd"].bind(backend);
+  Module["webnnRegisterMLContext"] = (sessionId, mlContext) => {
+    backend["registerMLContext"](sessionId, mlContext);
+  };
+  Module["webnnOnReleaseSession"] = (sessionId) => {
+    backend["onReleaseSession"](sessionId);
+  };
+  Module["webnnCreateMLTensorDownloader"] = (tensorId, type) => {
+    return backend["createMLTensorDownloader"](tensorId, type);
+  };
+  Module["webnnRegisterMLTensor"] = (sessionId, tensor, dataType, shape) => {
+    return backend["registerMLTensor"](sessionId, tensor, dataType, shape);
+  };
+  Module["webnnCreateMLContext"] = (optionsOrGpuDevice) => {
+    return backend["createMLContext"](optionsOrGpuDevice);
+  };
+  Module["webnnRegisterMLConstant"] = (
+    externalFilePath,
+    dataOffset,
+    dataLength,
+    builder,
+    desc,
+    shouldConvertInt64ToInt32
+  ) => {
+    return backend["registerMLConstant"](
+      externalFilePath,
+      dataOffset,
+      dataLength,
+      builder,
+      desc,
+      Module.MountedFiles,
+      shouldConvertInt64ToInt32
+    );
+  };
+  Module["webnnRegisterGraphInput"] =
+    backend["registerGraphInput"].bind(backend);
+  Module["webnnIsGraphInput"] = backend["isGraphInput"].bind(backend);
+  Module["webnnRegisterGraphOutput"] =
+    backend["registerGraphOutput"].bind(backend);
+  Module["webnnIsGraphOutput"] = backend["isGraphOutput"].bind(backend);
+
+  Module["webnnCreateTemporaryTensor"] =
+    backend["createTemporaryTensor"].bind(backend);
+  Module["webnnIsGraphInputOutputTypeSupported"] =
+    backend["isGraphInputOutputTypeSupported"].bind(backend);
+};