You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
1.8 KiB
73 lines
1.8 KiB
cmake_minimum_required(VERSION 3.14 FATAL_ERROR)
|
|
|
|
project(wenet VERSION 0.1)
|
|
|
|
option(CXX11_ABI "whether to use CXX11_ABI libtorch" OFF)
|
|
option(GRAPH_TOOLS "whether to build TLG graph tools" OFF)
|
|
option(BUILD_TESTING "whether to build unit test" OFF)
|
|
option(GRPC "whether to build with gRPC" ON)
|
|
# TODO(Binbin Zhang): Change websocket to OFF since it depends on boost which is a very big library
|
|
#
|
|
option(WEBSOCKET "whether to build with websocket" OFF)
|
|
option(HTTP "whether to build with http" OFF)
|
|
option(TORCH "whether to build with Torch" OFF)
|
|
option(ONNX "whether to build with ONNX" ON)
|
|
option(GPU "whether to build with GPU" OFF)
|
|
option(ITN "whether to use WeTextProcessing" OFF)
|
|
|
|
set(CMAKE_CXX_FLAGS_DEBUG "$ENV{CXXFLAGS} -g")
|
|
set(CMAKE_VERBOSE_MAKEFILE OFF)
|
|
|
|
include(FetchContent)
|
|
set(FETCHCONTENT_QUIET OFF)
|
|
get_filename_component(fc_base "fc_base" REALPATH BASE_DIR "${CMAKE_CURRENT_SOURCE_DIR}")
|
|
set(FETCHCONTENT_BASE_DIR ${fc_base})
|
|
|
|
list(APPEND CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake)
|
|
|
|
if(NOT MSVC)
|
|
# Keep the same with openfst, -fPIC or -fpic
|
|
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++17 -pthread -fPIC")
|
|
else()
|
|
set(CMAKE_WINDOWS_EXPORT_ALL_SYMBOLS ON)
|
|
add_compile_options("$<$<CXX_COMPILER_ID:MSVC>:/utf-8>")
|
|
endif()
|
|
|
|
|
|
if(ONNX)
|
|
include(onnx)
|
|
endif()
|
|
#
|
|
include(openfst)
|
|
include_directories(
|
|
${CMAKE_CURRENT_SOURCE_DIR}
|
|
${CMAKE_CURRENT_SOURCE_DIR}/kaldi
|
|
)
|
|
|
|
if(ITN)
|
|
include(wetextprocessing)
|
|
endif()
|
|
|
|
# Build all libraries
|
|
add_subdirectory(utils)
|
|
add_subdirectory(frontend)
|
|
add_subdirectory(post_processor)
|
|
add_subdirectory(kaldi) # kaldi:
|
|
add_subdirectory(decoder)
|
|
|
|
# Optionally, you can build with websocket
|
|
if(WEBSOCKET)
|
|
include(boost)
|
|
add_subdirectory(websocket)
|
|
endif()
|
|
|
|
# Optionally, you can build with gRPC
|
|
if(GRPC)
|
|
include(grpc)
|
|
add_subdirectory(grpc)
|
|
endif()
|
|
|
|
|
|
|
|
# Build all bins
|
|
add_subdirectory(bin)
|