diff --git a/CMakeLists.txt b/CMakeLists.txt index 67d006b46..02cbfa4bb 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -147,12 +147,6 @@ endif(ENABLE_PADDLE_FRONTEND) if(ENABLE_ORT_BACKEND) add_definitions(-DENABLE_ORT_BACKEND) list(APPEND ALL_DEPLOY_SRCS ${DEPLOY_ORT_SRCS}) - if (ORT_DIRECTORY) - message(STATUS "Use the onnxruntime lib specified by user. The ONNXRuntime path: ${ORT_DIRECTORY}") - STRING(REGEX REPLACE "\\\\" "/" ORT_DIRECTORY ${ORT_DIRECTORY}) - else() - message(STATUS "Use the default onnxruntime lib.") - endif() include(${PROJECT_SOURCE_DIR}/cmake/onnxruntime.cmake) list(APPEND DEPEND_LIBS external_onnxruntime) endif() diff --git a/cmake/onnxruntime.cmake b/cmake/onnxruntime.cmake index 7124f4ea8..5240366be 100644 --- a/cmake/onnxruntime.cmake +++ b/cmake/onnxruntime.cmake @@ -21,6 +21,8 @@ set(ONNXRUNTIME_SOURCE_DIR set(ONNXRUNTIME_INSTALL_DIR ${THIRD_PARTY_PATH}/install/onnxruntime) if (ORT_DIRECTORY) + message(STATUS "Use the onnxruntime lib specified by user. The ONNXRuntime path: ${ORT_DIRECTORY}") + STRING(REGEX REPLACE "\\\\" "/" ORT_DIRECTORY ${ORT_DIRECTORY}) set(ONNXRUNTIME_INC_DIR "${ORT_DIRECTORY}/include" CACHE PATH "onnxruntime include directory." FORCE) @@ -29,6 +31,7 @@ if (ORT_DIRECTORY) "${ORT_DIRECTORY}/lib" CACHE PATH "onnxruntime lib directory." FORCE) else() + message(STATUS "Use the default onnxruntime lib. The ONNXRuntime path: ${ONNXRUNTIME_INSTALL_DIR}") set(ONNXRUNTIME_INC_DIR "${ONNXRUNTIME_INSTALL_DIR}/include" CACHE PATH "onnxruntime include directory." FORCE)