CMakeLists.txt 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585
  1. cmake_minimum_required(VERSION 3.12) # Don't bump this version for no reason
  2. project("llama.cpp" C CXX)
  3. set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
  4. if (NOT XCODE AND NOT MSVC AND NOT CMAKE_BUILD_TYPE)
  5. set(CMAKE_BUILD_TYPE Release CACHE STRING "Build type" FORCE)
  6. set_property(CACHE CMAKE_BUILD_TYPE PROPERTY STRINGS "Debug" "Release" "MinSizeRel" "RelWithDebInfo")
  7. endif()
  8. set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/bin)
  9. if(CMAKE_SOURCE_DIR STREQUAL CMAKE_CURRENT_SOURCE_DIR)
  10. set(LLAMA_STANDALONE ON)
  11. # configure project version
  12. # TODO
  13. else()
  14. set(LLAMA_STANDALONE OFF)
  15. endif()
  16. if (EMSCRIPTEN)
  17. set(BUILD_SHARED_LIBS_DEFAULT OFF)
  18. option(LLAMA_WASM_SINGLE_FILE "llama: embed WASM inside the generated llama.js" ON)
  19. else()
  20. if (MINGW)
  21. set(BUILD_SHARED_LIBS_DEFAULT OFF)
  22. else()
  23. set(BUILD_SHARED_LIBS_DEFAULT ON)
  24. endif()
  25. endif()
  26. #
  27. # Option list
  28. #
  29. # general
  30. option(LLAMA_STATIC "llama: static link libraries" OFF)
  31. option(LLAMA_NATIVE "llama: enable -march=native flag" OFF)
  32. option(LLAMA_LTO "llama: enable link time optimization" OFF)
  33. # debug
  34. option(LLAMA_ALL_WARNINGS "llama: enable all compiler warnings" ON)
  35. option(LLAMA_ALL_WARNINGS_3RD_PARTY "llama: enable all compiler warnings in 3rd party libs" OFF)
  36. option(LLAMA_GPROF "llama: enable gprof" OFF)
  37. # sanitizers
  38. option(LLAMA_SANITIZE_THREAD "llama: enable thread sanitizer" OFF)
  39. option(LLAMA_SANITIZE_ADDRESS "llama: enable address sanitizer" OFF)
  40. option(LLAMA_SANITIZE_UNDEFINED "llama: enable undefined sanitizer" OFF)
  41. # instruction set specific
  42. option(LLAMA_AVX "llama: enable AVX" ON)
  43. option(LLAMA_AVX2 "llama: enable AVX2" ON)
  44. option(LLAMA_AVX512 "llama: enable AVX512" OFF)
  45. option(LLAMA_AVX512_VBMI "llama: enable AVX512-VBMI" OFF)
  46. option(LLAMA_AVX512_VNNI "llama: enable AVX512-VNNI" OFF)
  47. option(LLAMA_FMA "llama: enable FMA" ON)
  48. # in MSVC F16C is implied with AVX2/AVX512
  49. if (NOT MSVC)
  50. option(LLAMA_F16C "llama: enable F16C" ON)
  51. endif()
  52. # 3rd party libs
  53. option(LLAMA_ACCELERATE "llama: enable Accelerate framework" ON)
  54. option(LLAMA_BLAS "llama: use BLAS" OFF)
  55. set(LLAMA_BLAS_VENDOR "Generic" CACHE STRING "llama: BLAS library vendor")
  56. option(LLAMA_CUBLAS "llama: use cuBLAS" OFF)
  57. option(LLAMA_CUDA_FORCE_DMMV "llama: use dmmv instead of mmvq CUDA kernels" OFF)
  58. set(LLAMA_CUDA_DMMV_X "32" CACHE STRING "llama: x stride for dmmv CUDA kernels")
  59. set(LLAMA_CUDA_MMV_Y "1" CACHE STRING "llama: y block size for mmv CUDA kernels")
  60. option(LLAMA_CUDA_DMMV_F16 "llama: use 16 bit floats for dmmv CUDA kernels" OFF)
  61. set(LLAMA_CUDA_KQUANTS_ITER "2" CACHE STRING "llama: iters./thread per block for Q2_K/Q6_K")
  62. option(LLAMA_CLBLAST "llama: use CLBlast" OFF)
  63. option(LLAMA_METAL "llama: use Metal" OFF)
  64. option(LLAMA_MPI "llama: use MPI" OFF)
  65. option(LLAMA_K_QUANTS "llama: use k-quants" ON)
  66. option(LLAMA_QKK_64 "llama: use super-block size of 64 for k-quants" OFF)
  67. option(LLAMA_BUILD_TESTS "llama: build tests" ${LLAMA_STANDALONE})
  68. option(LLAMA_BUILD_EXAMPLES "llama: build examples" ${LLAMA_STANDALONE})
  69. option(LLAMA_BUILD_SERVER "llama: build server example" ON)
  70. #
  71. # Build info header
  72. #
  73. # Generate initial build-info.h
  74. include(${CMAKE_CURRENT_SOURCE_DIR}/scripts/build-info.cmake)
  75. if(EXISTS "${CMAKE_CURRENT_SOURCE_DIR}/.git")
  76. set(GIT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/.git")
  77. # Is git submodule
  78. if(NOT IS_DIRECTORY "${GIT_DIR}")
  79. file(READ ${GIT_DIR} REAL_GIT_DIR_LINK)
  80. string(REGEX REPLACE "gitdir: (.*)\n$" "\\1" REAL_GIT_DIR ${REAL_GIT_DIR_LINK})
  81. set(GIT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/${REAL_GIT_DIR}")
  82. endif()
  83. # Add a custom target for build-info.h
  84. add_custom_target(BUILD_INFO ALL DEPENDS "${CMAKE_CURRENT_SOURCE_DIR}/build-info.h")
  85. # Add a custom command to rebuild build-info.h when .git/index changes
  86. add_custom_command(
  87. OUTPUT "${CMAKE_CURRENT_SOURCE_DIR}/build-info.h"
  88. COMMENT "Generating build details from Git"
  89. COMMAND ${CMAKE_COMMAND} -P "${CMAKE_CURRENT_SOURCE_DIR}/scripts/build-info.cmake"
  90. WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
  91. DEPENDS "${GIT_DIR}/index"
  92. VERBATIM
  93. )
  94. else()
  95. 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.")
  96. endif()
  97. #
  98. # Compile flags
  99. #
  100. set(CMAKE_CXX_STANDARD 11)
  101. set(CMAKE_CXX_STANDARD_REQUIRED true)
  102. set(CMAKE_C_STANDARD 11)
  103. set(CMAKE_C_STANDARD_REQUIRED true)
  104. set(THREADS_PREFER_PTHREAD_FLAG ON)
  105. find_package(Threads REQUIRED)
  106. if (NOT MSVC)
  107. if (LLAMA_SANITIZE_THREAD)
  108. add_compile_options(-fsanitize=thread)
  109. link_libraries(-fsanitize=thread)
  110. endif()
  111. if (LLAMA_SANITIZE_ADDRESS)
  112. add_compile_options(-fsanitize=address -fno-omit-frame-pointer)
  113. link_libraries(-fsanitize=address)
  114. endif()
  115. if (LLAMA_SANITIZE_UNDEFINED)
  116. add_compile_options(-fsanitize=undefined)
  117. link_libraries(-fsanitize=undefined)
  118. endif()
  119. endif()
  120. if (APPLE AND LLAMA_ACCELERATE)
  121. find_library(ACCELERATE_FRAMEWORK Accelerate)
  122. if (ACCELERATE_FRAMEWORK)
  123. message(STATUS "Accelerate framework found")
  124. add_compile_definitions(GGML_USE_ACCELERATE)
  125. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} ${ACCELERATE_FRAMEWORK})
  126. else()
  127. message(WARNING "Accelerate framework not found")
  128. endif()
  129. endif()
  130. if (LLAMA_BLAS)
  131. if (LLAMA_STATIC)
  132. set(BLA_STATIC ON)
  133. endif()
  134. if ($(CMAKE_VERSION) VERSION_GREATER_EQUAL 3.22)
  135. set(BLA_SIZEOF_INTEGER 8)
  136. endif()
  137. set(BLA_VENDOR ${LLAMA_BLAS_VENDOR})
  138. find_package(BLAS)
  139. if (BLAS_FOUND)
  140. message(STATUS "BLAS found, Libraries: ${BLAS_LIBRARIES}")
  141. if ("${BLAS_INCLUDE_DIRS}" STREQUAL "")
  142. # BLAS_INCLUDE_DIRS is missing in FindBLAS.cmake.
  143. # see https://gitlab.kitware.com/cmake/cmake/-/issues/20268
  144. find_package(PkgConfig REQUIRED)
  145. if (${LLAMA_BLAS_VENDOR} MATCHES "Generic")
  146. pkg_check_modules(DepBLAS REQUIRED blas)
  147. elseif (${LLAMA_BLAS_VENDOR} MATCHES "OpenBLAS")
  148. pkg_check_modules(DepBLAS REQUIRED openblas)
  149. elseif (${LLAMA_BLAS_VENDOR} MATCHES "FLAME")
  150. pkg_check_modules(DepBLAS REQUIRED blis)
  151. elseif (${LLAMA_BLAS_VENDOR} MATCHES "ATLAS")
  152. pkg_check_modules(DepBLAS REQUIRED blas-atlas)
  153. elseif (${LLAMA_BLAS_VENDOR} MATCHES "FlexiBLAS")
  154. pkg_check_modules(DepBLAS REQUIRED flexiblas_api)
  155. elseif (${LLAMA_BLAS_VENDOR} MATCHES "Intel")
  156. # all Intel* libraries share the same include path
  157. pkg_check_modules(DepBLAS mkl-sdl)
  158. if (NOT DepBLAS)
  159. if (BUILD_SHARED_LIBS)
  160. set(LINK_METHOD dynamic)
  161. else()
  162. set(LINK_METHOD static)
  163. endif()
  164. string(REGEX REPLACE ".*_" "" DATA_TYPE_MODEL ${LLAMA_BLAS_VENDOR})
  165. pkg_check_modules(DepBLAS REQUIRED mkl-${LINK_METHOD}-${DATA_TYPE_MODEL}-iomp)
  166. endif()
  167. elseif (${LLAMA_BLAS_VENDOR} MATCHES "NVHPC")
  168. # this doesn't provide pkg-config
  169. # suggest to assign BLAS_INCLUDE_DIRS on your own
  170. if ("${NVHPC_VERSION}" STREQUAL "")
  171. message(WARNING "Better to set NVHPC_VERSION")
  172. else()
  173. set(DepBLAS_FOUND ON)
  174. set(DepBLAS_INCLUDE_DIRS "/opt/nvidia/hpc_sdk/${CMAKE_SYSTEM_NAME}_${CMAKE_SYSTEM_PROCESSOR}/${NVHPC_VERSION}/math_libs/include")
  175. endif()
  176. endif()
  177. if (DepBLAS_FOUND)
  178. set(BLAS_INCLUDE_DIRS ${DepBLAS_INCLUDE_DIRS})
  179. else()
  180. message(WARNING "BLAS_INCLUDE_DIRS neither been provided nor been automatically"
  181. " detected by pkgconfig, trying to find cblas.h from possible paths...")
  182. find_path(BLAS_INCLUDE_DIRS
  183. NAMES cblas.h
  184. HINTS
  185. /usr/include
  186. /usr/local/include
  187. /usr/include/openblas
  188. /opt/homebrew/opt/openblas/include
  189. /usr/local/opt/openblas/include
  190. /usr/include/x86_64-linux-gnu/openblas/include
  191. )
  192. endif()
  193. endif()
  194. message(STATUS "BLAS found, Includes: ${BLAS_INCLUDE_DIRS}")
  195. add_compile_options(${BLAS_LINKER_FLAGS})
  196. add_compile_definitions(GGML_USE_OPENBLAS)
  197. if (${BLAS_INCLUDE_DIRS} MATCHES "mkl" AND (${LLAMA_BLAS_VENDOR} MATCHES "Generic" OR ${LLAMA_BLAS_VENDOR} MATCHES "Intel"))
  198. add_compile_definitions(GGML_BLAS_USE_MKL)
  199. endif()
  200. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} ${BLAS_LIBRARIES})
  201. set(LLAMA_EXTRA_INCLUDES ${LLAMA_EXTRA_INCLUDES} ${BLAS_INCLUDE_DIRS})
  202. else()
  203. message(WARNING "BLAS not found, please refer to "
  204. "https://cmake.org/cmake/help/latest/module/FindBLAS.html#blas-lapack-vendors"
  205. " to set correct LLAMA_BLAS_VENDOR")
  206. endif()
  207. endif()
  208. if (LLAMA_K_QUANTS)
  209. set(GGML_SOURCES_EXTRA ${GGML_SOURCES_EXTRA} k_quants.c k_quants.h)
  210. add_compile_definitions(GGML_USE_K_QUANTS)
  211. if (LLAMA_QKK_64)
  212. add_compile_definitions(GGML_QKK_64)
  213. endif()
  214. endif()
  215. if (LLAMA_CUBLAS)
  216. cmake_minimum_required(VERSION 3.17)
  217. find_package(CUDAToolkit)
  218. if (CUDAToolkit_FOUND)
  219. message(STATUS "cuBLAS found")
  220. enable_language(CUDA)
  221. set(GGML_SOURCES_CUDA ggml-cuda.cu ggml-cuda.h)
  222. add_compile_definitions(GGML_USE_CUBLAS)
  223. if (LLAMA_CUDA_FORCE_DMMV)
  224. add_compile_definitions(GGML_CUDA_FORCE_DMMV)
  225. endif()
  226. add_compile_definitions(GGML_CUDA_DMMV_X=${LLAMA_CUDA_DMMV_X})
  227. add_compile_definitions(GGML_CUDA_MMV_Y=${LLAMA_CUDA_MMV_Y})
  228. if (DEFINED LLAMA_CUDA_DMMV_Y)
  229. add_compile_definitions(GGML_CUDA_MMV_Y=${LLAMA_CUDA_DMMV_Y}) # for backwards compatibility
  230. endif()
  231. if (LLAMA_CUDA_DMMV_F16)
  232. add_compile_definitions(GGML_CUDA_DMMV_F16)
  233. endif()
  234. add_compile_definitions(K_QUANTS_PER_ITERATION=${LLAMA_CUDA_KQUANTS_ITER})
  235. if (LLAMA_STATIC)
  236. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} CUDA::cudart_static CUDA::cublas_static CUDA::cublasLt_static)
  237. else()
  238. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} CUDA::cudart CUDA::cublas CUDA::cublasLt)
  239. endif()
  240. if (NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
  241. if (LLAMA_CUDA_DMMV_F16)
  242. set(CMAKE_CUDA_ARCHITECTURES "60;61") # needed for f16 CUDA intrinsics
  243. else()
  244. set(CMAKE_CUDA_ARCHITECTURES "52;61") # lowest CUDA 12 standard + lowest for integer intrinsics
  245. endif()
  246. endif()
  247. message(STATUS "Using CUDA architectures: ${CMAKE_CUDA_ARCHITECTURES}")
  248. else()
  249. message(WARNING "cuBLAS not found")
  250. endif()
  251. endif()
  252. if (LLAMA_METAL)
  253. find_library(FOUNDATION_LIBRARY Foundation REQUIRED)
  254. find_library(METAL_FRAMEWORK Metal REQUIRED)
  255. find_library(METALKIT_FRAMEWORK MetalKit REQUIRED)
  256. find_library(METALPERFORMANCE_FRAMEWORK MetalPerformanceShaders REQUIRED)
  257. set(GGML_SOURCES_METAL ggml-metal.m ggml-metal.h)
  258. add_compile_definitions(GGML_USE_METAL)
  259. add_compile_definitions(GGML_METAL_NDEBUG)
  260. # get full path to the file
  261. #add_compile_definitions(GGML_METAL_DIR_KERNELS="${CMAKE_CURRENT_SOURCE_DIR}/")
  262. # copy ggml-metal.metal to bin directory
  263. configure_file(ggml-metal.metal bin/ggml-metal.metal COPYONLY)
  264. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS}
  265. ${FOUNDATION_LIBRARY}
  266. ${METAL_FRAMEWORK}
  267. ${METALKIT_FRAMEWORK}
  268. ${METALPERFORMANCE_FRAMEWORK}
  269. )
  270. endif()
  271. if (LLAMA_MPI)
  272. cmake_minimum_required(VERSION 3.10)
  273. find_package(MPI)
  274. if (MPI_C_FOUND)
  275. message(STATUS "MPI found")
  276. set(GGML_SOURCES_MPI ggml-mpi.c ggml-mpi.h)
  277. add_compile_definitions(GGML_USE_MPI)
  278. add_compile_definitions(${MPI_C_COMPILE_DEFINITIONS})
  279. set(cxx_flags ${cxx_flags} -Wno-cast-qual)
  280. set(c_flags ${c_flags} -Wno-cast-qual)
  281. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} ${MPI_C_LIBRARIES})
  282. set(LLAMA_EXTRA_INCLUDES ${LLAMA_EXTRA_INCLUDES} ${MPI_C_INCLUDE_DIRS})
  283. # Even if you're only using the C header, C++ programs may bring in MPI
  284. # C++ functions, so more linkage is needed
  285. if (MPI_CXX_FOUND)
  286. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} ${MPI_CXX_LIBRARIES})
  287. endif()
  288. else()
  289. message(WARNING "MPI not found")
  290. endif()
  291. endif()
  292. if (LLAMA_CLBLAST)
  293. find_package(CLBlast)
  294. if (CLBlast_FOUND)
  295. message(STATUS "CLBlast found")
  296. set(GGML_SOURCES_OPENCL ggml-opencl.cpp ggml-opencl.h)
  297. add_compile_definitions(GGML_USE_CLBLAST)
  298. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} clblast)
  299. else()
  300. message(WARNING "CLBlast not found")
  301. endif()
  302. endif()
  303. if (LLAMA_ALL_WARNINGS)
  304. if (NOT MSVC)
  305. set(c_flags
  306. -Wall
  307. -Wextra
  308. -Wpedantic
  309. -Wcast-qual
  310. -Wdouble-promotion
  311. -Wshadow
  312. -Wstrict-prototypes
  313. -Wpointer-arith
  314. )
  315. set(cxx_flags
  316. -Wall
  317. -Wextra
  318. -Wpedantic
  319. -Wcast-qual
  320. -Wno-unused-function
  321. -Wno-multichar
  322. )
  323. else()
  324. # todo : msvc
  325. endif()
  326. add_compile_options(
  327. "$<$<COMPILE_LANGUAGE:C>:${c_flags}>"
  328. "$<$<COMPILE_LANGUAGE:CXX>:${cxx_flags}>"
  329. )
  330. endif()
  331. if (MSVC)
  332. add_compile_definitions(_CRT_SECURE_NO_WARNINGS)
  333. if (BUILD_SHARED_LIBS)
  334. set(CMAKE_WINDOWS_EXPORT_ALL_SYMBOLS ON)
  335. endif()
  336. endif()
  337. if (LLAMA_LTO)
  338. include(CheckIPOSupported)
  339. check_ipo_supported(RESULT result OUTPUT output)
  340. if (result)
  341. set(CMAKE_INTERPROCEDURAL_OPTIMIZATION TRUE)
  342. else()
  343. message(WARNING "IPO is not supported: ${output}")
  344. endif()
  345. endif()
  346. # Architecture specific
  347. # TODO: probably these flags need to be tweaked on some architectures
  348. # feel free to update the Makefile for your architecture and send a pull request or issue
  349. message(STATUS "CMAKE_SYSTEM_PROCESSOR: ${CMAKE_SYSTEM_PROCESSOR}")
  350. if (NOT MSVC)
  351. if (LLAMA_STATIC)
  352. add_link_options(-static)
  353. if (MINGW)
  354. add_link_options(-static-libgcc -static-libstdc++)
  355. endif()
  356. endif()
  357. if (LLAMA_GPROF)
  358. add_compile_options(-pg)
  359. endif()
  360. if (LLAMA_NATIVE)
  361. add_compile_options(-march=native)
  362. endif()
  363. endif()
  364. if (${CMAKE_SYSTEM_PROCESSOR} MATCHES "arm" OR ${CMAKE_SYSTEM_PROCESSOR} MATCHES "aarch64")
  365. message(STATUS "ARM detected")
  366. if (MSVC)
  367. # TODO: arm msvc?
  368. else()
  369. if (${CMAKE_SYSTEM_PROCESSOR} MATCHES "armv6")
  370. # Raspberry Pi 1, Zero
  371. add_compile_options(-mfpu=neon-fp-armv8 -mfp16-format=ieee -mno-unaligned-access)
  372. endif()
  373. if (${CMAKE_SYSTEM_PROCESSOR} MATCHES "armv7")
  374. # Raspberry Pi 2
  375. add_compile_options(-mfpu=neon-fp-armv8 -mfp16-format=ieee -mno-unaligned-access -funsafe-math-optimizations)
  376. endif()
  377. if (${CMAKE_SYSTEM_PROCESSOR} MATCHES "armv8")
  378. # Raspberry Pi 3, 4, Zero 2 (32-bit)
  379. add_compile_options(-mfp16-format=ieee -mno-unaligned-access)
  380. endif()
  381. endif()
  382. elseif (${CMAKE_SYSTEM_PROCESSOR} MATCHES "^(x86_64|i686|AMD64)$")
  383. message(STATUS "x86 detected")
  384. if (MSVC)
  385. if (LLAMA_AVX512)
  386. add_compile_options($<$<COMPILE_LANGUAGE:C>:/arch:AVX512>)
  387. add_compile_options($<$<COMPILE_LANGUAGE:CXX>:/arch:AVX512>)
  388. # MSVC has no compile-time flags enabling specific
  389. # AVX512 extensions, neither it defines the
  390. # macros corresponding to the extensions.
  391. # Do it manually.
  392. if (LLAMA_AVX512_VBMI)
  393. add_compile_definitions($<$<COMPILE_LANGUAGE:C>:__AVX512VBMI__>)
  394. add_compile_definitions($<$<COMPILE_LANGUAGE:CXX>:__AVX512VBMI__>)
  395. endif()
  396. if (LLAMA_AVX512_VNNI)
  397. add_compile_definitions($<$<COMPILE_LANGUAGE:C>:__AVX512VNNI__>)
  398. add_compile_definitions($<$<COMPILE_LANGUAGE:CXX>:__AVX512VNNI__>)
  399. endif()
  400. elseif (LLAMA_AVX2)
  401. add_compile_options($<$<COMPILE_LANGUAGE:C>:/arch:AVX2>)
  402. add_compile_options($<$<COMPILE_LANGUAGE:CXX>:/arch:AVX2>)
  403. elseif (LLAMA_AVX)
  404. add_compile_options($<$<COMPILE_LANGUAGE:C>:/arch:AVX>)
  405. add_compile_options($<$<COMPILE_LANGUAGE:CXX>:/arch:AVX>)
  406. endif()
  407. else()
  408. if (LLAMA_F16C)
  409. add_compile_options(-mf16c)
  410. endif()
  411. if (LLAMA_FMA)
  412. add_compile_options(-mfma)
  413. endif()
  414. if (LLAMA_AVX)
  415. add_compile_options(-mavx)
  416. endif()
  417. if (LLAMA_AVX2)
  418. add_compile_options(-mavx2)
  419. endif()
  420. if (LLAMA_AVX512)
  421. add_compile_options(-mavx512f)
  422. add_compile_options(-mavx512bw)
  423. endif()
  424. if (LLAMA_AVX512_VBMI)
  425. add_compile_options(-mavx512vbmi)
  426. endif()
  427. if (LLAMA_AVX512_VNNI)
  428. add_compile_options(-mavx512vnni)
  429. endif()
  430. endif()
  431. elseif (${CMAKE_SYSTEM_PROCESSOR} MATCHES "ppc64")
  432. message(STATUS "PowerPC detected")
  433. add_compile_options(-mcpu=native -mtune=native)
  434. #TODO: Add targets for Power8/Power9 (Altivec/VSX) and Power10(MMA) and query for big endian systems (ppc64/le/be)
  435. else()
  436. message(STATUS "Unknown architecture")
  437. endif()
  438. #
  439. # Build libraries
  440. #
  441. add_library(ggml OBJECT
  442. ggml.c
  443. ggml.h
  444. ${GGML_SOURCES_CUDA}
  445. ${GGML_SOURCES_OPENCL}
  446. ${GGML_SOURCES_METAL}
  447. ${GGML_SOURCES_MPI}
  448. ${GGML_SOURCES_EXTRA}
  449. )
  450. target_include_directories(ggml PUBLIC . ${LLAMA_EXTRA_INCLUDES})
  451. target_compile_features(ggml PUBLIC c_std_11) # don't bump
  452. target_link_libraries(ggml PUBLIC Threads::Threads ${LLAMA_EXTRA_LIBS})
  453. add_library(ggml_static STATIC $<TARGET_OBJECTS:ggml>)
  454. if (BUILD_SHARED_LIBS)
  455. set_target_properties(ggml PROPERTIES POSITION_INDEPENDENT_CODE ON)
  456. add_library(ggml_shared SHARED $<TARGET_OBJECTS:ggml>)
  457. target_link_libraries(ggml_shared PUBLIC Threads::Threads ${LLAMA_EXTRA_LIBS})
  458. install(TARGETS ggml_shared LIBRARY)
  459. endif()
  460. add_library(llama
  461. llama.cpp
  462. llama.h
  463. llama-util.h
  464. )
  465. target_include_directories(llama PUBLIC .)
  466. target_compile_features(llama PUBLIC cxx_std_11) # don't bump
  467. target_link_libraries(llama PRIVATE
  468. ggml
  469. ${LLAMA_EXTRA_LIBS}
  470. )
  471. if (BUILD_SHARED_LIBS)
  472. set_target_properties(llama PROPERTIES POSITION_INDEPENDENT_CODE ON)
  473. target_compile_definitions(llama PRIVATE LLAMA_SHARED LLAMA_BUILD)
  474. if (LLAMA_METAL)
  475. set_target_properties(llama PROPERTIES RESOURCE "${CMAKE_CURRENT_SOURCE_DIR}/ggml-metal.metal")
  476. endif()
  477. install(TARGETS llama LIBRARY)
  478. endif()
  479. include(GNUInstallDirs)
  480. install(
  481. FILES convert.py
  482. PERMISSIONS
  483. OWNER_READ
  484. OWNER_WRITE
  485. OWNER_EXECUTE
  486. GROUP_READ
  487. GROUP_EXECUTE
  488. WORLD_READ
  489. WORLD_EXECUTE
  490. DESTINATION ${CMAKE_INSTALL_BINDIR})
  491. install(
  492. FILES convert-lora-to-ggml.py
  493. PERMISSIONS
  494. OWNER_READ
  495. OWNER_WRITE
  496. OWNER_EXECUTE
  497. GROUP_READ
  498. GROUP_EXECUTE
  499. WORLD_READ
  500. WORLD_EXECUTE
  501. DESTINATION ${CMAKE_INSTALL_BINDIR})
  502. #
  503. # programs, examples and tests
  504. #
  505. if (LLAMA_BUILD_TESTS AND NOT CMAKE_JS_VERSION)
  506. include(CTest)
  507. add_subdirectory(tests)
  508. endif ()
  509. if (LLAMA_BUILD_EXAMPLES)
  510. add_subdirectory(examples)
  511. add_subdirectory(pocs)
  512. endif()