diff --git a/CMakeLists.txt b/CMakeLists.txt index 3436311..c0b70d1 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -2,9 +2,12 @@ cmake_minimum_required(VERSION 3.14 FATAL_ERROR) project(wenet VERSION 0.1) # 测试模块参数设置 option(GRPC "build with gRPC" OFF) -option(ONNX "build with ONNX" OFF) option(GLOGS "build with GLOGS" OFF) +option(ONNX "build with ONNX" OFF) option(OPENFST "build with FST" ON) +option(CLEAN_CMAKE_CACHE "更换Build目录运行需要清空缓存" OFF) + + # set color string(ASCII 27 Esc) set(ColourReset "${Esc}[m") @@ -16,7 +19,14 @@ set(FETCHCONTENT_QUIET OFF) set(third_party_libraries) set(LIB_BASE_DIR /root/projects/temp_xiaoke/asr_runtime/lib_files) set(FETCHCONTENT_BASE_DIR ${LIB_BASE_DIR}) - +#清空缓存 +if(CLEAN_CMAKE_CACHE) + message(STATUS "Cleaning CMakeCache.txt") + execute_process( + COMMAND find . -name "CMake*" -exec rm -rf {} \; + WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR} + ) +endif () list(APPEND CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake-local) if(GLOGS) include(gflags) @@ -31,8 +41,8 @@ endif() if(ONNX) include(onnx) - add_executable(main "onnx.cc") - target_link_libraries(onnx_main ${third_party_libraries}) + add_executable(onnx_main "onnx.cc") + target_link_libraries(onnx_main PRIVATE ${third_party_libraries}) endif() if(OPENFST) include(openfst) diff --git a/bin_test/onnx.cc b/bin_test/onnx.cc index ab7ca35..cd6461f 100644 --- a/bin_test/onnx.cc +++ b/bin_test/onnx.cc @@ -8,6 +8,6 @@ int main() { Ort::SessionOptions opt = Ort::SessionOptions(); Ort::Env env = Ort::Env(ORT_LOGGING_LEVEL_WARNING, ""); opt.SetIntraOpNumThreads(1); - printf("Run ONNX Done..%d,%f\n"); + printf("Run ONNX Done..%d,%f\n",1,1.123); return 0; } \ No newline at end of file diff --git a/cmake-local/onnx.cmake b/cmake-local/onnx.cmake index 507a4c7..c63326c 100644 --- a/cmake-local/onnx.cmake +++ b/cmake-local/onnx.cmake @@ -1,7 +1,7 @@ FetchContent_Declare(onnxruntime SOURCE_DIR ${LIB_BASE_DIR}/onnxruntime-src) - FetchContent_MakeAvailable(onnxruntime) + include_directories(${onnxruntime_SOURCE_DIR}/include) link_directories(${onnxruntime_SOURCE_DIR}/lib)