onnx-simplifier

Форк
0
/
build_ort.cmake 
25 строк · 1.1 Кб
1
# For MessageDifferencer::Equals
2
option(onnxruntime_USE_FULL_PROTOBUF "" ON)
3
if (EMSCRIPTEN)
4
  if (NOT DEFINED ONNX_CUSTOM_PROTOC_EXECUTABLE)
5
    message(FATAL_ERROR "ONNX_CUSTOM_PROTOC_EXECUTABLE must be set for emscripten")
6
  endif()
7

8
  option(onnxruntime_BUILD_WEBASSEMBLY "" ON)
9
  option(onnxruntime_BUILD_WEBASSEMBLY_STATIC_LIB "" ON)
10
  option(onnxruntime_ENABLE_WEBASSEMBLY_SIMD "" OFF)
11
  option(onnxruntime_ENABLE_WEBASSEMBLY_EXCEPTION_CATCHING "" ON)
12
  option(onnxruntime_ENABLE_WEBASSEMBLY_THREADS "" OFF)
13
  option(onnxruntime_BUILD_UNIT_TESTS "" OFF)
14
  set(onnxruntime_EMSCRIPTEN_SETTINGS "MALLOC=dlmalloc")
15

16
  # For custom onnx target in onnx optimizer
17
  set(ONNX_TARGET_NAME onnxruntime_webassembly)
18
else()
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
22
  set(BUILD_SHARED_LIBS ON)
23
  option(onnxruntime_BUILD_SHARED_LIB "" ON)
24
endif()
25
add_subdirectory(third_party/onnxruntime/cmake)
26

27

Использование cookies

Мы используем файлы cookie в соответствии с Политикой конфиденциальности и Политикой использования cookies.

Нажимая кнопку «Принимаю», Вы даете АО «СберТех» согласие на обработку Ваших персональных данных в целях совершенствования нашего веб-сайта и Сервиса GitVerse, а также повышения удобства их использования.

Запретить использование cookies Вы можете самостоятельно в настройках Вашего браузера.