saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

make xnnpack build for ort-web (#11745)

Browse source 
* make xnnpack build for ort-web

* make ci happy
This commit is contained in:
Guenther Schmuelling 2022-06-10 08:47:57 -07:00 committed by GitHub
parent f745eb1d3f
commit d4ea59654c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cmake/CMakeLists.txt
View file

@ -1590,7 +1590,8 @@ if (onnxruntime_USE_XNNPACK)
set(wasmsimd_src_patterns "${XNNPACK_DIR}/src/wasmsimd-*.c"
"${XNNPACK_DIR}/src/*-wasmsimd-*.c"
"${XNNPACK_DIR}/src/*-wasmsimd.c")
"${XNNPACK_DIR}/src/*-wasmsimd.c"
"${XNNPACK_DIR}/src/*/wasmsimd.c")
file(GLOB_RECURSE XNNPACK_WASMSIMD_MICROKERNEL_SRCS CONFIGURE_DEPENDS ${wasmsimd_src_patterns})
message(DEBUG "XNNPACK_WASMSIMD_MICROKERNEL_SRCS:${XNNPACK_WASMSIMD_MICROKERNEL_SRCS}")

3
cmake/onnxruntime_webassembly.cmake
View file

@ -174,6 +174,9 @@ else()
onnxruntime_util
re2::re2
)
if (onnxruntime_USE_XNNPACK)
target_link_libraries(onnxruntime_webassembly PRIVATE XNNPACK)
endif()
if (onnxruntime_ENABLE_TRAINING OR onnxruntime_ENABLE_TRAINING_OPS)
target_link_libraries(onnxruntime_webassembly PRIVATE tensorboard)

9
onnxruntime/wasm/api.cc
View file

@ -4,6 +4,9 @@
#include "api.h"
#include "core/session/onnxruntime_cxx_api.h"
#ifdef USE_XNNPACK
#include "core/providers/xnnpack/xnnpack_provider_factory.h"
#endif
#include <iostream>
#include <vector>
@ -112,7 +115,9 @@ OrtSessionOptions* OrtCreateSessionOptions(size_t graph_optimization_level,
// Enable ORT CustomOps in onnxruntime-extensions
RETURN_NULLPTR_IF_ERROR(EnableOrtCustomOps, session_options);
#endif
#if defined(USE_XNNPACK)
(void)OrtSessionOptionsAppendExecutionProvider_Xnnpack(session_options);
#endif
return session_options;
}
@ -359,4 +364,4 @@ char* OrtEndProfiling(ort_session_handle_t session) {
return (CHECK_STATUS(SessionEndProfiling, session, allocator, &file_name) == ORT_OK)
? file_name
: nullptr;
}
}