znsoft пре 2 година
родитељ
комит
6bb021d2b1

+ 3 - 3
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})
+target_link_libraries( ${APP_NAME} PRIVATE  ${LIB_NAME})

+ 3 - 3
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"
+#include "libpuncapi.h"