ollama/llama/compat/apply-patch.cmake
2026-06-14 20:05:08 -07:00

60 lines
2.2 KiB
CMake

# Idempotent patch applier used by compat.cmake.
#
# Invocation (from a CMake PATCH_COMMAND):
# cmake -DPATCH_DIR=<dir of *.patch> -P apply-patch.cmake
#
# Every *.patch under PATCH_DIR is applied in numeric filename order in the
# current working directory (which ExternalProject / FetchContent sets to the
# fetched source's SOURCE_DIR). A patch already applied — detected via
# `git apply --reverse --check` — is skipped. This makes re-configuring and
# re-building safe.
if(NOT DEFINED PATCH_DIR)
message(FATAL_ERROR "apply-patch.cmake: PATCH_DIR not set")
endif()
find_package(Git QUIET REQUIRED)
get_filename_component(_patch_workdir "." ABSOLUTE)
get_filename_component(_git_ceiling "${_patch_workdir}" DIRECTORY)
set(_git_apply_env GIT_CEILING_DIRECTORIES=${_git_ceiling})
file(GLOB_RECURSE _patches "${PATCH_DIR}/*.patch")
set(_patch_entries)
foreach(PATCH_FILE IN LISTS _patches)
get_filename_component(_patch_name "${PATCH_FILE}" NAME)
list(APPEND _patch_entries "${_patch_name}|${PATCH_FILE}")
endforeach()
list(SORT _patch_entries)
foreach(_patch_entry IN LISTS _patch_entries)
string(REGEX REPLACE "^[^|]*\\|" "" PATCH_FILE "${_patch_entry}")
file(RELATIVE_PATH _patch_rel "${PATCH_DIR}" "${PATCH_FILE}")
# If the patch can be REVERSED cleanly, it's already applied. Skip.
execute_process(
COMMAND ${CMAKE_COMMAND} -E env ${_git_apply_env}
${GIT_EXECUTABLE} apply --reverse --check "${PATCH_FILE}"
RESULT_VARIABLE _reverse_check
OUTPUT_QUIET ERROR_QUIET
)
if(_reverse_check EQUAL 0)
message(STATUS "llama/compat: ${_patch_rel} already applied, skipping")
continue()
endif()
# Otherwise, apply forward.
execute_process(
COMMAND ${CMAKE_COMMAND} -E env ${_git_apply_env}
${GIT_EXECUTABLE} apply --whitespace=nowarn "${PATCH_FILE}"
RESULT_VARIABLE _apply_result
)
if(NOT _apply_result EQUAL 0)
message(FATAL_ERROR
"llama/compat: failed to apply ${_patch_rel}\n"
"This usually means the pinned llama.cpp source has changed. "
"Regenerate the patch against the pinned LLAMA_CPP_VERSION and retry.")
endif()
message(STATUS "llama/compat: applied ${_patch_rel}")
endforeach()