mirror of
https://github.com/ggerganov/llama.cpp.git
synced 2024-12-27 03:44:35 +00:00
75 lines
2.3 KiB
CMake
75 lines
2.3 KiB
CMake
set(TEMPLATE_FILE "${CMAKE_CURRENT_SOURCE_DIR}/scripts/build-info.h.in")
|
|
set(HEADER_FILE "${CMAKE_CURRENT_SOURCE_DIR}/build-info.h")
|
|
set(BUILD_NUMBER 0)
|
|
set(BUILD_COMMIT "unknown")
|
|
set(BUILD_COMPILER "unknown")
|
|
set(BUILD_TARGET "unknown")
|
|
|
|
# Look for git
|
|
find_package(Git)
|
|
if(NOT Git_FOUND)
|
|
find_program(GIT_EXECUTABLE NAMES git git.exe)
|
|
if(GIT_EXECUTABLE)
|
|
set(Git_FOUND TRUE)
|
|
message(STATUS "Found Git: ${GIT_EXECUTABLE}")
|
|
else()
|
|
message(WARNING "Git not found. Build info will not be accurate.")
|
|
endif()
|
|
endif()
|
|
|
|
# Get the commit count and hash
|
|
if(Git_FOUND)
|
|
execute_process(
|
|
COMMAND ${GIT_EXECUTABLE} rev-parse --short HEAD
|
|
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
|
|
OUTPUT_VARIABLE HEAD
|
|
OUTPUT_STRIP_TRAILING_WHITESPACE
|
|
)
|
|
execute_process(
|
|
COMMAND ${GIT_EXECUTABLE} rev-list --count HEAD
|
|
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
|
|
OUTPUT_VARIABLE COUNT
|
|
OUTPUT_STRIP_TRAILING_WHITESPACE
|
|
)
|
|
set(BUILD_COMMIT ${HEAD})
|
|
set(BUILD_NUMBER ${COUNT})
|
|
endif()
|
|
|
|
if(MSVC)
|
|
set(BUILD_COMPILER "${CMAKE_C_COMPILER_ID} ${CMAKE_C_COMPILER_VERSION}")
|
|
set(BUILD_TARGET ${CMAKE_VS_PLATFORM_NAME})
|
|
else()
|
|
execute_process(
|
|
COMMAND sh -c "$@ --version | head -1" _ ${CMAKE_C_COMPILER}
|
|
OUTPUT_VARIABLE OUT
|
|
OUTPUT_STRIP_TRAILING_WHITESPACE
|
|
)
|
|
set(BUILD_COMPILER ${OUT})
|
|
execute_process(
|
|
COMMAND ${CMAKE_C_COMPILER} -dumpmachine
|
|
OUTPUT_VARIABLE OUT
|
|
OUTPUT_STRIP_TRAILING_WHITESPACE
|
|
)
|
|
set(BUILD_TARGET ${OUT})
|
|
endif()
|
|
|
|
# Only write the header if it's changed to prevent unnecessary recompilation
|
|
if(EXISTS ${HEADER_FILE})
|
|
file(READ ${HEADER_FILE} CONTENTS)
|
|
string(REGEX MATCH "BUILD_COMMIT \"([^\"]*)\"" _ ${CONTENTS})
|
|
set(OLD_COMMIT ${CMAKE_MATCH_1})
|
|
string(REGEX MATCH "BUILD_COMPILER \"([^\"]*)\"" _ ${CONTENTS})
|
|
set(OLD_COMPILER ${CMAKE_MATCH_1})
|
|
string(REGEX MATCH "BUILD_TARGET \"([^\"]*)\"" _ ${CONTENTS})
|
|
set(OLD_TARGET ${CMAKE_MATCH_1})
|
|
if (
|
|
NOT OLD_COMMIT STREQUAL BUILD_COMMIT OR
|
|
NOT OLD_COMPILER STREQUAL BUILD_COMPILER OR
|
|
NOT OLD_TARGET STREQUAL BUILD_TARGET
|
|
)
|
|
configure_file(${TEMPLATE_FILE} ${HEADER_FILE})
|
|
endif()
|
|
else()
|
|
configure_file(${TEMPLATE_FILE} ${HEADER_FILE})
|
|
endif()
|