File size: 1,105 Bytes
7c071a8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
cmake_minimum_required(VERSION 2.8) project(baichuan2) if (NOT DEFINED TARGET_ARCH) set(TARGET_ARCH pcie) endif() set(CMAKE_INSTALL_PREFIX install) if (${CMAKE_HOST_SYSTEM_PROCESSOR} STREQUAL "aarch64") add_definitions(-DSOC_TARGET) link_directories(${PROJECT_SOURCE_DIR}/../src/lib_soc) message("SoC mode, starting......") elseif (${TARGET_ARCH} STREQUAL "pcie") add_definitions(-DPCIE_TARGET) link_directories(${PROJECT_SOURCE_DIR}/../src/lib_pcie) message("Pcie mode, starting......") elseif (${TARGET_ARCH} STREQUAL "soc") add_definitions(-DSOC_TARGET) set(CMAKE_C_COMPILER aarch64-linux-gnu-gcc) set(CMAKE_ASM_COMPILER aarch64-linux-gnu-gcc) set(CMAKE_CXX_COMPILER aarch64-linux-gnu-g++) link_directories(${PROJECT_SOURCE_DIR}/lib_soc) message("SoC mode, starting......") endif() include_directories(${PROJECT_SOURCE_DIR}/../src/include) add_definitions(-DDEBUG --std=c++17 -fPIC -Wall -Werror) set(CMAKE_BUILD_TYPE "Debug") add_library(tpuchat SHARED chat.cpp) target_link_libraries(tpuchat bmrt bmlib sentencepiece) |