From 6bb021d2b143a58ece3a005af3a1135b0cdac4f2 Mon Sep 17 00:00:00 2001 From: znsoft Date: Sat, 22 Apr 2023 05:12:09 +0800 Subject: [PATCH] build with onnxruntime --- funasr/runtime/rapidpunc/CMakeLists.txt | 6 +++--- funasr/runtime/rapidpunc/include/precomp.h | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/funasr/runtime/rapidpunc/CMakeLists.txt b/funasr/runtime/rapidpunc/CMakeLists.txt index 763ec217b..9b2b4977c 100644 --- a/funasr/runtime/rapidpunc/CMakeLists.txt +++ b/funasr/runtime/rapidpunc/CMakeLists.txt @@ -5,9 +5,9 @@ add_subdirectory(yaml-cpp) -include_directories(${CMAKE_CURRENT_SOURCE_DIR}/include yaml-cpp/include ${CMAKE_INSTALL_PREFIX}/include) +include_directories(${CMAKE_CURRENT_SOURCE_DIR}/include yaml-cpp/include ${ONNXRUNTIME_DIR}/include) -link_directories(yaml-cpp ${CMAKE_INSTALL_PREFIX}/lib ${CMAKE_INSTALL_PREFIX}/lib64) +link_directories(yaml-cpp ${ONNXRUNTIME_DIR}/lib ) set(APP_NAME rapidpunc_tester) set(LIB_NAME rapidpunc) @@ -32,4 +32,4 @@ target_link_libraries( ${LIB_NAME} PUBLIC yaml-cpp onnxruntime ) add_executable(${APP_NAME} sources/tester.cpp ) -target_link_libraries( ${APP_NAME} PRIVATE ${LIB_NAME}) \ No newline at end of file +target_link_libraries( ${APP_NAME} PRIVATE ${LIB_NAME}) diff --git a/funasr/runtime/rapidpunc/include/precomp.h b/funasr/runtime/rapidpunc/include/precomp.h index b56046082..68941c06b 100644 --- a/funasr/runtime/rapidpunc/include/precomp.h +++ b/funasr/runtime/rapidpunc/include/precomp.h @@ -19,8 +19,8 @@ using namespace std; -#include "onnxruntime/onnxruntime_run_options_config_keys.h" -#include "onnxruntime/onnxruntime_cxx_api.h" +#include "onnxruntime_run_options_config_keys.h" +#include "onnxruntime_cxx_api.h" @@ -34,4 +34,4 @@ using namespace std; #include "punc_infer.h" -#include "libpuncapi.h" \ No newline at end of file +#include "libpuncapi.h"