mirror of
https://github.com/DrHo1y/ezrknn-llm.git
synced 2026-11-07 00:07:23 +07:00
22 lines
698 B
CMake
22 lines
698 B
CMake
cmake_minimum_required(VERSION 3.10)
|
|
project(llm_demo)
|
|
|
|
set(CMAKE_CXX_STANDARD 11)
|
|
set(CMAKE_CXX_STANDARD_REQUIRED ON)
|
|
|
|
set(SOURCE_FILES src/main.cpp)
|
|
|
|
add_executable(${PROJECT_NAME} ${SOURCE_FILES})
|
|
|
|
set(RKLLM_API_PATH "${CMAKE_SOURCE_DIR}/../../runtime/${CMAKE_SYSTEM_NAME}/librkllm_api")
|
|
include_directories(${RKLLM_API_PATH}/include)
|
|
if(CMAKE_SYSTEM_NAME STREQUAL "Android")
|
|
set(RKLLM_RT_LIB ${RKLLM_API_PATH}/${CMAKE_ANDROID_ARCH_ABI}/librkllmrt.so)
|
|
target_link_libraries(${PROJECT_NAME} ${RKLLM_RT_LIB} log)
|
|
elseif(CMAKE_SYSTEM_NAME STREQUAL "Linux")
|
|
set(RKLLM_RT_LIB ${RKLLM_API_PATH}/aarch64/librkllmrt.so)
|
|
target_link_libraries(${PROJECT_NAME} ${RKLLM_RT_LIB})
|
|
endif()
|
|
|
|
|