CMakeLists.txt 2.7 KB

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