| cmake_minimum_required(VERSION 3.6) | |
| project(rknn_benchmark) | |
| set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS}") | |
| set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++14") | |
| # rknn api | |
| set(RKNN_API_PATH ${CMAKE_SOURCE_DIR}/../../runtime//${CMAKE_SYSTEM_NAME}/librknn_api) | |
| if (CMAKE_SYSTEM_NAME STREQUAL "Android") | |
| set(RKNN_RT_LIB ${RKNN_API_PATH}/${CMAKE_ANDROID_ARCH_ABI}/librknnrt.so) | |
| else() | |
| if (CMAKE_C_COMPILER MATCHES "aarch64") | |
| set(LIB_ARCH aarch64) | |
| else() | |
| set(LIB_ARCH armhf) | |
| endif() | |
| set(RKNN_RT_LIB ${RKNN_API_PATH}/${LIB_ARCH}/librknnrt.so) | |
| endif() | |
| include_directories(${RKNN_API_PATH}/include) | |
| include_directories(${CMAKE_SOURCE_DIR}/../3rdparty) | |
| set(CMAKE_INSTALL_RPATH "lib") | |
| add_executable(rknn_benchmark | |
| src/rknn_benchmark.cpp | |
| src/cnpy/cnpy.cpp | |
| ) | |
| target_link_libraries(rknn_benchmark | |
| ${RKNN_RT_LIB} | |
| ${OpenCV_LIBS} | |
| ) | |
| # install target and libraries | |
| set(CMAKE_INSTALL_PREFIX ${CMAKE_SOURCE_DIR}/install/rknn_benchmark_${CMAKE_SYSTEM_NAME}) | |
| install(TARGETS rknn_benchmark DESTINATION ./) | |
| install(PROGRAMS ${RKNN_RT_LIB} DESTINATION lib) | |