| | |
| | | |
| | | option(ENABLE_GLOG "Whether to build glog" ON) |
| | | option(ENABLE_FST "Whether to build openfst" ON) # ITN need openfst compiled |
| | | option(GPU "Whether to build with GPU" OFF) |
| | | |
| | | # set(CMAKE_CXX_STANDARD 11) |
| | | set(CMAKE_CXX_STANDARD 14 CACHE STRING "The C++ version to be used.") |
| | |
| | | include_directories(${PROJECT_SOURCE_DIR}/third_party/kaldi) |
| | | include_directories(${PROJECT_SOURCE_DIR}/third_party/json/include) |
| | | |
| | | if(GPU) |
| | | add_definitions(-DUSE_GPU) |
| | | set(TORCH_DIR "/usr/local/lib/python3.8/dist-packages/torch") |
| | | set(TORCH_BLADE_DIR "/usr/local/lib/python3.8/dist-packages/torch_blade") |
| | | include_directories(${TORCH_DIR}/include) |
| | | include_directories(${TORCH_DIR}/include/torch/csrc/api/include) |
| | | link_directories(${TORCH_DIR}/lib) |
| | | link_directories(${TORCH_BLADE_DIR}) |
| | | set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fPIC -D_GLIBCXX_USE_CXX11_ABI=0") |
| | | endif() |
| | | |
| | | if(ENABLE_GLOG) |
| | | include_directories(${PROJECT_SOURCE_DIR}/third_party/glog/src) |
| | | set(BUILD_TESTING OFF) |