Merge branch 'main' into graph_profiling

Author: qjia7

cmake/CMakeLists.txt

@@ -1801,8 +1801,11 @@ if (onnxruntime_ENABLE_EXTERNAL_CUSTOM_OP_SCHEMAS)
   )
 endif()
 
-if(NOT onnxruntime_BUILD_SHARED_LIB AND onnxruntime_USE_WEBGPU)
-  message(WARNING "CMake target files will not be generated for static onnxruntime builds with webgpu support")
+if (NOT onnxruntime_BUILD_SHARED_LIB AND
+    (onnxruntime_USE_WEBGPU OR (CMAKE_SYSTEM_NAME STREQUAL "Emscripten" AND onnxruntime_USE_XNNPACK)))
+  message(WARNING
+    "CMake target files will not be generated for static onnxruntime builds "
+    "with WebGPU or Emscripten+XNNPACK support")
 else()
   # Install
   include(CMakePackageConfigHelpers)

cmake/external/abseil-cpp.cmake

@@ -12,12 +12,8 @@ set(ABSL_USE_EXTERNAL_GOOGLETEST ON)
 
 # Both abseil and xnnpack create a target called memory, which
 # results in a duplicate target if ABSL_ENABLE_INSTALL is on.
-if (onnxruntime_USE_XNNPACK)
-  set(ABSL_ENABLE_INSTALL OFF)
-else()
-  if (NOT CMAKE_SYSTEM_NAME MATCHES "AIX")
+if (NOT CMAKE_SYSTEM_NAME MATCHES "AIX")
     set(ABSL_ENABLE_INSTALL ON)
-  endif()
 endif()
 
 if(Patch_FOUND)

cmake/external/xnnpack.cmake

@@ -62,6 +62,8 @@ ELSEIF(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64")
   SET(ORT_TARGET_PROCESSOR "arm64")
 ELSEIF(CMAKE_SYSTEM_PROCESSOR STREQUAL "ppc64le")
   SET(ORT_TARGET_PROCESSOR "ppc64")
+ELSEIF(CMAKE_SYSTEM_NAME STREQUAL "Emscripten")
+  SET(ORT_TARGET_PROCESSOR "wasm")
 ELSEIF(NOT ORT_TARGET_PROCESSOR MATCHES "^(x86(_64)?|arm64|riscv(32|64|128)|Hexagon|ppc64)$")
   SET(ORT_TARGET_PROCESSOR "${CMAKE_SYSTEM_PROCESSOR}")
 ELSE()
@@ -90,18 +92,21 @@ onnxruntime_fetchcontent_makeavailable(googlexnnpack)
 set(XNNPACK_DIR ${googlexnnpack_SOURCE_DIR})
 set(XNNPACK_INCLUDE_DIR ${XNNPACK_DIR}/include)
 
-set(onnxruntime_EXTERNAL_LIBRARIES_XNNPACK XNNPACK xnnpack-microkernels-prod pthreadpool)
+set(onnxruntime_EXTERNAL_LIBRARIES_XNNPACK XNNPACK pthreadpool)
 if(ORT_TARGET_PROCESSOR MATCHES "^arm64.*" AND NOT CMAKE_C_COMPILER_ID STREQUAL "MSVC")
   list(APPEND onnxruntime_EXTERNAL_LIBRARIES_XNNPACK kleidiai)
 endif()
+if(NOT CMAKE_SYSTEM_NAME STREQUAL "Emscripten")
+  list(APPEND onnxruntime_EXTERNAL_LIBRARIES_XNNPACK xnnpack-microkernels-prod)
+endif()
 
 # the XNNPACK CMake setup doesn't include the WASM kernels so we have to manually set those up
 if(CMAKE_SYSTEM_NAME STREQUAL "Emscripten")
   # See source lists in _deps/googlexnnpack-src/BUILD.bazel for wasm_prod_microkernels
   message("Adding WebAssembly Source Files to XNNPACK")
   set(wasm_srcs "")
 
-  file(READ "${XNNPACK_DIR}/BUILD.bazel" xnnpack_bazel_config)
+  file(READ "${XNNPACK_DIR}/build_srcs.bzl" xnnpack_bazel_config)
 
   # Replace newlines with semicolon so that it is treated as a list by CMake
   # Also replace '[' and ']' so the bazel source lists don't get parsed as a nested list by cmake
@@ -139,19 +144,26 @@ if(CMAKE_SYSTEM_NAME STREQUAL "Emscripten")
   GetSrcListFromBazel("TABLE_SRCS" table_srcs)
   list(APPEND wasm_srcs ${operator_srcs} ${table_srcs})
 
-  # kernels
-  list(APPEND wasm_srcs ${XNNPACK_DIR}/src/amalgam/gen/scalar.c)
-  list(APPEND wasm_srcs ${XNNPACK_DIR}/src/amalgam/gen/wasm.c)
+  set(microkernel_src "")
+
+  include(${XNNPACK_DIR}/cmake/gen/scalar_microkernels.cmake)
+  list(APPEND microkernel_src ${PROD_SCALAR_MICROKERNEL_SRCS})
+  list(APPEND microkernel_src ${PROD_WASM_MICROKERNEL_SRCS})
 
   if(onnxruntime_ENABLE_WEBASSEMBLY_RELAXED_SIMD)
-    list(APPEND wasm_srcs ${XNNPACK_DIR}/src/amalgam/gen/wasmsimd.c)
-    list(APPEND wasm_srcs ${XNNPACK_DIR}/src/amalgam/gen/wasmrelaxedsimd.c)
+    include(${XNNPACK_DIR}/cmake/gen/wasmsimd_microkernels.cmake)
+    include(${XNNPACK_DIR}/cmake/gen/wasmrelaxedsimd_microkernels.cmake)
+    list(APPEND microkernel_src ${PROD_WASMSIMD_MICROKERNEL_SRCS})
+    list(APPEND microkernel_src ${PROD_WASMRELAXEDSIMD_MICROKERNEL_SRCS})
     target_compile_options(XNNPACK PRIVATE "-msimd128")
     target_compile_options(XNNPACK PRIVATE "-mrelaxed-simd")
   elseif(onnxruntime_ENABLE_WEBASSEMBLY_SIMD)
-    list(APPEND wasm_srcs ${XNNPACK_DIR}/src/amalgam/gen/wasmsimd.c)
+    include(${XNNPACK_DIR}/cmake/gen/wasmsimd_microkernels.cmake)
+    list(APPEND microkernel_src ${PROD_WASMSIMD_MICROKERNEL_SRCS})
     target_compile_options(XNNPACK PRIVATE "-msimd128")
   endif()
+  list(TRANSFORM microkernel_src PREPEND "${XNNPACK_DIR}/")
+  list(APPEND wasm_srcs ${microkernel_src})
 
   message(DEBUG "wasm_srcs: ${wasm_srcs}\n")
   target_sources(XNNPACK PRIVATE ${wasm_srcs})