rknpu2.cmake 731 B

12345678910111213141516171819
  1. # get RKNPU2_URL
  2. set(RKNPU2_URL_BASE "https://bj.bcebos.com/fastdeploy/third_libs/")
  3. set(RKNPU2_VERSION "1.4.2b0")
  4. set(RKNPU2_FILE "rknpu2_runtime-linux-aarch64-${RKNPU2_VERSION}-${RKNN2_TARGET_SOC}.tgz")
  5. set(RKNPU2_URL "${RKNPU2_URL_BASE}${RKNPU2_FILE}")
  6. # download_and_decompress
  7. download_and_decompress(${RKNPU2_URL} ${CMAKE_CURRENT_BINARY_DIR}/${RKNPU2_FILE} ${THIRD_PARTY_PATH}/install/)
  8. # set path
  9. set(RKNPU_RUNTIME_PATH ${THIRD_PARTY_PATH}/install/rknpu2_runtime)
  10. # include lib
  11. if (EXISTS ${RKNPU_RUNTIME_PATH})
  12. set(RKNN_RT_LIB ${RKNPU_RUNTIME_PATH}/lib/librknnrt.so)
  13. include_directories(${RKNPU_RUNTIME_PATH}/include)
  14. else ()
  15. message(FATAL_ERROR "[rknpu2.cmake] RKNPU_RUNTIME_PATH does not exist.")
  16. endif ()