|
|
@ -1,49 +1,33 @@ |
|
|
|
cmake_minimum_required(VERSION 3.14 FATAL_ERROR) |
|
|
|
|
|
|
|
project(wenet VERSION 0.1) |
|
|
|
# 测试模块参数设置 |
|
|
|
option(GRPC "build with gRPC" 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) |
|
|
|
project(wenet VERSION 0.2) |
|
|
|
#运行参数 |
|
|
|
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++17 -pthread -fPIC") |
|
|
|
option(GRPC "whether to build with gRPC" ON) |
|
|
|
option(ONNX "whether to build with ONNX" ON) |
|
|
|
|
|
|
|
|
|
|
|
# set color |
|
|
|
string(ASCII 27 Esc) |
|
|
|
set(ColourReset "${Esc}[m") |
|
|
|
set(BoldGreen "${Esc}[1;32m") |
|
|
|
set(BoldRed "${Esc}[31m") |
|
|
|
|
|
|
|
set(third_party_libraries) |
|
|
|
set(LIB_BASE_DIR /root/projects/temp_xiaoke/asr_runtime/lib_files) |
|
|
|
#清空缓存 |
|
|
|
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-linux) |
|
|
|
if(GLOGS) |
|
|
|
include(gflags) |
|
|
|
include(glog) |
|
|
|
add_executable(glog_main "glogs.cc") |
|
|
|
target_link_libraries(glog_main ${third_party_libraries}) |
|
|
|
# Model Option |
|
|
|
if(ONNX) |
|
|
|
include(onnx) |
|
|
|
endif() |
|
|
|
include(openfst) |
|
|
|
include_directories( |
|
|
|
${CMAKE_CURRENT_SOURCE_DIR} |
|
|
|
${CMAKE_CURRENT_SOURCE_DIR}/kaldi |
|
|
|
) |
|
|
|
|
|
|
|
# Build all libraries |
|
|
|
add_subdirectory(utils) |
|
|
|
add_subdirectory(frontend) |
|
|
|
add_subdirectory(post_processor) |
|
|
|
add_subdirectory(kaldi) # kaldi: |
|
|
|
add_subdirectory(decoder) |
|
|
|
#Connection Option |
|
|
|
if(GRPC) |
|
|
|
include(grpc) |
|
|
|
add_subdirectory(bin_grpc) |
|
|
|
add_subdirectory(grpc) |
|
|
|
endif() |
|
|
|
|
|
|
|
if(ONNX) |
|
|
|
include(onnx) |
|
|
|
add_executable(onnx_main "onnx.cc") |
|
|
|
target_link_libraries(onnx_main PRIVATE ${third_party_libraries}) |
|
|
|
endif() |
|
|
|
if(OPENFST) |
|
|
|
include(openfst) |
|
|
|
add_executable(fst_main "fst.cc") |
|
|
|
target_link_libraries(fst_main PRIVATE ${third_party_libraries}) |
|
|
|
endif() |
|
|
|
# Build all bins |
|
|
|
add_subdirectory(bin) |