speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add cmakelist

Browse Source
This commit is contained in:
雾聪 2024-03-13 14:09:18 +08:00
parent c3192dffdd
commit c522408653
2 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
runtime/onnxruntime/CMakeLists.txt
View File

@ -4,6 +4,7 @@ project(FunASROnnx)
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.")
@ -37,6 +38,17 @@ include_directories(${PROJECT_SOURCE_DIR}/third_party/jieba/include)
include_directories(${PROJECT_SOURCE_DIR}/third_party/jieba/include/limonp/include)
include_directories(${PROJECT_SOURCE_DIR}/third_party/kaldi)
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)

1
runtime/onnxruntime/bin/CMakeLists.txt
View File

@ -10,6 +10,7 @@ SET(RELATION_SOURCE "../src/resample.cpp" "../src/util.cpp" "../src/alignedmem.c
endif()
add_executable(funasr-onnx-offline "funasr-onnx-offline.cpp" ${RELATION_SOURCE})
target_link_options(funasr-onnx-offline PRIVATE "-Wl,--no-as-needed")
target_link_libraries(funasr-onnx-offline PUBLIC funasr)
add_executable(funasr-onnx-offline-vad "funasr-onnx-offline-vad.cpp" ${RELATION_SOURCE})