2024.5.9

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)

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;
 }

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)