build with onnxruntime

This commit is contained in:
znsoft 2023-04-22 05:12:09 +08:00
parent 5d59006c90
commit 6bb021d2b1
2 changed files with 6 additions and 6 deletions

View File

@ -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})

View File

@ -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"