onnx-simplifier
25 строк · 1.1 Кб
1# For MessageDifferencer::Equals
2option(onnxruntime_USE_FULL_PROTOBUF "" ON)
3if (EMSCRIPTEN)
4if (NOT DEFINED ONNX_CUSTOM_PROTOC_EXECUTABLE)
5message(FATAL_ERROR "ONNX_CUSTOM_PROTOC_EXECUTABLE must be set for emscripten")
6endif()
7
8option(onnxruntime_BUILD_WEBASSEMBLY "" ON)
9option(onnxruntime_BUILD_WEBASSEMBLY_STATIC_LIB "" ON)
10option(onnxruntime_ENABLE_WEBASSEMBLY_SIMD "" OFF)
11option(onnxruntime_ENABLE_WEBASSEMBLY_EXCEPTION_CATCHING "" ON)
12option(onnxruntime_ENABLE_WEBASSEMBLY_THREADS "" OFF)
13option(onnxruntime_BUILD_UNIT_TESTS "" OFF)
14set(onnxruntime_EMSCRIPTEN_SETTINGS "MALLOC=dlmalloc")
15
16# For custom onnx target in onnx optimizer
17set(ONNX_TARGET_NAME onnxruntime_webassembly)
18else()
19# For native build, only shared libs is ok. Otherwise libonnx.a will be linked twice (in onnxruntime and in onnxsim)
20# For emscripten build, since the libonnxruntime_webassembly.a is bundled by `bundle_static_library`, onnxsim can link
21# to the single libonnxruntime_webassembly.a
22set(BUILD_SHARED_LIBS ON)
23option(onnxruntime_BUILD_SHARED_LIB "" ON)
24endif()
25add_subdirectory(third_party/onnxruntime/cmake)
26
27