CMakeLists.txt 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144
  1. # common
  2. find_package(Threads REQUIRED)
  3. llama_add_compile_flags()
  4. # Build info header
  5. #
  6. if(EXISTS "${CMAKE_CURRENT_SOURCE_DIR}/../.git")
  7. set(GIT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/../.git")
  8. # Is git submodule
  9. if(NOT IS_DIRECTORY "${GIT_DIR}")
  10. file(READ ${GIT_DIR} REAL_GIT_DIR_LINK)
  11. string(REGEX REPLACE "gitdir: (.*)\n$" "\\1" REAL_GIT_DIR ${REAL_GIT_DIR_LINK})
  12. string(FIND "${REAL_GIT_DIR}" "/" SLASH_POS)
  13. if (SLASH_POS EQUAL 0)
  14. set(GIT_DIR "${REAL_GIT_DIR}")
  15. else()
  16. set(GIT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/../${REAL_GIT_DIR}")
  17. endif()
  18. endif()
  19. if(EXISTS "${GIT_DIR}/index")
  20. set(GIT_INDEX "${GIT_DIR}/index")
  21. else()
  22. message(WARNING "Git index not found in git repository.")
  23. set(GIT_INDEX "")
  24. endif()
  25. else()
  26. 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.")
  27. set(GIT_INDEX "")
  28. endif()
  29. # Add a custom command to rebuild build-info.cpp when .git/index changes
  30. add_custom_command(
  31. OUTPUT "${CMAKE_CURRENT_SOURCE_DIR}/build-info.cpp"
  32. COMMENT "Generating build details from Git"
  33. COMMAND ${CMAKE_COMMAND} -DMSVC=${MSVC} -DCMAKE_C_COMPILER_VERSION=${CMAKE_C_COMPILER_VERSION}
  34. -DCMAKE_C_COMPILER_ID=${CMAKE_C_COMPILER_ID} -DCMAKE_VS_PLATFORM_NAME=${CMAKE_VS_PLATFORM_NAME}
  35. -DCMAKE_C_COMPILER=${CMAKE_C_COMPILER} -P "${CMAKE_CURRENT_SOURCE_DIR}/cmake/build-info-gen-cpp.cmake"
  36. WORKING_DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}/.."
  37. DEPENDS "${CMAKE_CURRENT_SOURCE_DIR}/build-info.cpp.in" ${GIT_INDEX}
  38. VERBATIM
  39. )
  40. set(TARGET build_info)
  41. add_library(${TARGET} OBJECT build-info.cpp)
  42. if (BUILD_SHARED_LIBS)
  43. set_target_properties(${TARGET} PROPERTIES POSITION_INDEPENDENT_CODE ON)
  44. endif()
  45. set(TARGET common)
  46. add_library(${TARGET} STATIC
  47. arg.cpp
  48. arg.h
  49. base64.hpp
  50. chat.cpp
  51. chat.h
  52. common.cpp
  53. common.h
  54. console.cpp
  55. console.h
  56. json-schema-to-grammar.cpp
  57. json.hpp
  58. llguidance.cpp
  59. log.cpp
  60. log.h
  61. minja/chat-template.hpp
  62. minja/minja.hpp
  63. ngram-cache.cpp
  64. ngram-cache.h
  65. sampling.cpp
  66. sampling.h
  67. speculative.cpp
  68. speculative.h
  69. )
  70. if (BUILD_SHARED_LIBS)
  71. set_target_properties(${TARGET} PROPERTIES POSITION_INDEPENDENT_CODE ON)
  72. endif()
  73. set(LLAMA_COMMON_EXTRA_LIBS build_info)
  74. # Use curl to download model url
  75. if (LLAMA_CURL)
  76. find_package(CURL)
  77. if (NOT CURL_FOUND)
  78. message(FATAL_ERROR "Could NOT find CURL. Hint: to disable this feature, set -DLLAMA_CURL=OFF")
  79. endif()
  80. target_compile_definitions(${TARGET} PUBLIC LLAMA_USE_CURL)
  81. include_directories(${CURL_INCLUDE_DIRS})
  82. find_library(CURL_LIBRARY curl REQUIRED)
  83. set(LLAMA_COMMON_EXTRA_LIBS ${LLAMA_COMMON_EXTRA_LIBS} ${CURL_LIBRARY})
  84. endif ()
  85. if (LLAMA_LLGUIDANCE)
  86. include(ExternalProject)
  87. set(LLGUIDANCE_SRC ${CMAKE_BINARY_DIR}/llguidance/source)
  88. set(LLGUIDANCE_PATH ${LLGUIDANCE_SRC}/target/release)
  89. # Set the correct library file extension based on platform
  90. if (WIN32)
  91. set(LLGUIDANCE_LIB_NAME "llguidance.lib")
  92. # Add Windows-specific libraries
  93. set(LLGUIDANCE_PLATFORM_LIBS
  94. ws2_32 # Windows Sockets API
  95. userenv # For GetUserProfileDirectoryW
  96. ntdll # For NT functions
  97. bcrypt # For BCryptGenRandom
  98. )
  99. else()
  100. set(LLGUIDANCE_LIB_NAME "libllguidance.a")
  101. set(LLGUIDANCE_PLATFORM_LIBS "")
  102. endif()
  103. ExternalProject_Add(llguidance_ext
  104. GIT_REPOSITORY https://github.com/guidance-ai/llguidance
  105. # v0.7.10:
  106. GIT_TAG 0309d2a6bf40abda35344a362edc71e06d5009f8
  107. PREFIX ${CMAKE_BINARY_DIR}/llguidance
  108. SOURCE_DIR ${LLGUIDANCE_SRC}
  109. BUILD_IN_SOURCE TRUE
  110. CONFIGURE_COMMAND ""
  111. BUILD_COMMAND cargo build --release
  112. INSTALL_COMMAND ""
  113. BUILD_BYPRODUCTS ${LLGUIDANCE_PATH}/${LLGUIDANCE_LIB_NAME} ${LLGUIDANCE_PATH}/llguidance.h
  114. UPDATE_COMMAND ""
  115. )
  116. target_compile_definitions(${TARGET} PUBLIC LLAMA_USE_LLGUIDANCE)
  117. add_library(llguidance STATIC IMPORTED)
  118. set_target_properties(llguidance PROPERTIES IMPORTED_LOCATION ${LLGUIDANCE_PATH}/${LLGUIDANCE_LIB_NAME})
  119. add_dependencies(llguidance llguidance_ext)
  120. target_include_directories(${TARGET} PRIVATE ${LLGUIDANCE_PATH})
  121. # Add platform libraries to the main target
  122. set(LLAMA_COMMON_EXTRA_LIBS ${LLAMA_COMMON_EXTRA_LIBS} llguidance ${LLGUIDANCE_PLATFORM_LIBS})
  123. endif ()
  124. target_include_directories(${TARGET} PUBLIC .)
  125. target_compile_features (${TARGET} PUBLIC cxx_std_17)
  126. target_link_libraries (${TARGET} PRIVATE ${LLAMA_COMMON_EXTRA_LIBS} PUBLIC llama Threads::Threads)