mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-18 21:21:17 +00:00
15219e2e71
### Description <!-- Describe your changes. --> the crash caused by the neural_speed turns out to be a very corn case. Turn it on by default. ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. -->
16 lines
636 B
CMake
16 lines
636 B
CMake
if ("${CMAKE_CXX_COMPILER_ID}" STREQUAL "GNU" AND onnxruntime_target_platform STREQUAL "x86_64")
|
|
set(USE_NEURAL_SPEED TRUE)
|
|
elseif ("${CMAKE_CXX_COMPILER_ID}" STREQUAL "MSVC" AND onnxruntime_target_platform STREQUAL "x64")
|
|
set(USE_NEURAL_SPEED TRUE)
|
|
endif()
|
|
|
|
if(USE_NEURAL_SPEED)
|
|
FetchContent_Declare(
|
|
neural_speed
|
|
URL ${DEP_URL_neural_speed}
|
|
URL_HASH SHA1=${DEP_SHA1_neural_speed}
|
|
PATCH_COMMAND ${Patch_EXECUTABLE} -p1 < ${PROJECT_SOURCE_DIR}/patches/neural_speed/150e7527d5286ddd3a995c228dedf8d76a7a86bc.patch
|
|
)
|
|
set(BTLA_USE_OPENMP OFF)
|
|
onnxruntime_fetchcontent_makeavailable(neural_speed)
|
|
endif()
|