CMakeLists.txt 2.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  1. # common
  2. find_package(Threads REQUIRED)
  3. # Build info header
  4. #
  5. if(EXISTS "${CMAKE_CURRENT_SOURCE_DIR}/../.git")
  6. set(GIT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/../.git")
  7. # Is git submodule
  8. if(NOT IS_DIRECTORY "${GIT_DIR}")
  9. file(READ ${GIT_DIR} REAL_GIT_DIR_LINK)
  10. string(REGEX REPLACE "gitdir: (.*)\n$" "\\1" REAL_GIT_DIR ${REAL_GIT_DIR_LINK})
  11. string(FIND "${REAL_GIT_DIR}" "/" SLASH_POS)
  12. if (SLASH_POS EQUAL 0)
  13. set(GIT_DIR "${REAL_GIT_DIR}")
  14. else()
  15. set(GIT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/../${REAL_GIT_DIR}")
  16. endif()
  17. endif()
  18. if(EXISTS "${GIT_DIR}/index")
  19. set(GIT_INDEX "${GIT_DIR}/index")
  20. else()
  21. message(WARNING "Git index not found in git repository.")
  22. set(GIT_INDEX "")
  23. endif()
  24. else()
  25. message(WARNING "Git repository not found; to enable automatic generation of build info, make sure Git is installed and the project is a Git repository.")
  26. set(GIT_INDEX "")
  27. endif()
  28. # Add a custom command to rebuild build-info.cpp when .git/index changes
  29. add_custom_command(
  30. OUTPUT "${CMAKE_CURRENT_SOURCE_DIR}/build-info.cpp"
  31. COMMENT "Generating build details from Git"
  32. COMMAND ${CMAKE_COMMAND} -DMSVC=${MSVC} -DCMAKE_C_COMPILER_VERSION=${CMAKE_C_COMPILER_VERSION}
  33. -DCMAKE_C_COMPILER_ID=${CMAKE_C_COMPILER_ID} -DCMAKE_VS_PLATFORM_NAME=${CMAKE_VS_PLATFORM_NAME}
  34. -DCMAKE_C_COMPILER=${CMAKE_C_COMPILER} -P "${CMAKE_CURRENT_SOURCE_DIR}/cmake/build-info-gen-cpp.cmake"
  35. WORKING_DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}/.."
  36. DEPENDS "${CMAKE_CURRENT_SOURCE_DIR}/build-info.cpp.in" ${GIT_INDEX}
  37. VERBATIM
  38. )
  39. set(TARGET build_info)
  40. add_library(${TARGET} OBJECT build-info.cpp)
  41. if (BUILD_SHARED_LIBS)
  42. set_target_properties(${TARGET} PROPERTIES POSITION_INDEPENDENT_CODE ON)
  43. endif()
  44. set(TARGET common)
  45. add_library(${TARGET} STATIC
  46. base64.hpp
  47. common.h
  48. common.cpp
  49. sampling.h
  50. sampling.cpp
  51. console.h
  52. console.cpp
  53. grammar-parser.h
  54. grammar-parser.cpp
  55. json.hpp
  56. json-schema-to-grammar.cpp
  57. train.h
  58. train.cpp
  59. ngram-cache.h
  60. ngram-cache.cpp
  61. )
  62. if (BUILD_SHARED_LIBS)
  63. set_target_properties(${TARGET} PROPERTIES POSITION_INDEPENDENT_CODE ON)
  64. endif()
  65. set(LLAMA_COMMON_EXTRA_LIBS build_info)
  66. # Use curl to download model url
  67. if (LLAMA_CURL)
  68. find_package(CURL REQUIRED)
  69. add_definitions(-DLLAMA_USE_CURL)
  70. include_directories(${CURL_INCLUDE_DIRS})
  71. find_library(CURL_LIBRARY curl REQUIRED)
  72. set(LLAMA_COMMON_EXTRA_LIBS ${LLAMA_COMMON_EXTRA_LIBS} ${CURL_LIBRARY})
  73. endif ()
  74. target_include_directories(${TARGET} PUBLIC .)
  75. target_compile_features (${TARGET} PUBLIC cxx_std_11)
  76. target_link_libraries (${TARGET} PRIVATE ${LLAMA_COMMON_EXTRA_LIBS} PUBLIC llama Threads::Threads)