mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-18 21:21:17 +00:00
492d9fd6cc
* switch to nonblocking threadpool in inference session and sessions state * switch to eigen threadpool - first draft * refine * refine * add a switch to easily revert back to windows thread pool * switch thread pool in test runner and turn on leak checker * remove unncessary files * fix build error * more build fixes * catch exceptions in parallel executor * fix mac build error * fix mac build error * more build fixes * more mac build fixes * fix cv issue * change macro to include cuda compiler for disabled compiler warning * try switching the macro to win32 only * test #error * move #disable warning to the top * Update onnxruntime_framework.cmake * move eigen include to public scope * turn off eigenthreadpool by default and add todo comment
21 lines
1 KiB
CMake
21 lines
1 KiB
CMake
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License.
|
|
|
|
file(GLOB onnxruntime_session_srcs
|
|
"${ONNXRUNTIME_INCLUDE_DIR}/core/session/*.h"
|
|
"${ONNXRUNTIME_ROOT}/core/session/*.h"
|
|
"${ONNXRUNTIME_ROOT}/core/session/*.cc"
|
|
)
|
|
|
|
source_group(TREE ${REPO_ROOT} FILES ${onnxruntime_session_srcs})
|
|
|
|
add_library(onnxruntime_session ${onnxruntime_session_srcs})
|
|
install(DIRECTORY ${PROJECT_SOURCE_DIR}/../include/onnxruntime/core/session DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/onnxruntime/core)
|
|
onnxruntime_add_include_to_target(onnxruntime_session onnxruntime_common onnxruntime_framework gsl onnx onnx_proto protobuf::libprotobuf)
|
|
target_include_directories(onnxruntime_session PRIVATE ${ONNXRUNTIME_ROOT} ${eigen_INCLUDE_DIRS})
|
|
add_dependencies(onnxruntime_session ${onnxruntime_EXTERNAL_DEPENDENCIES})
|
|
set_target_properties(onnxruntime_session PROPERTIES FOLDER "ONNXRuntime")
|
|
|
|
if(onnxruntime_USE_EIGEN_THREADPOOL)
|
|
target_compile_definitions(onnxruntime_session PUBLIC USE_EIGEN_THREADPOOL)
|
|
endif()
|