CMakeLists.txt 21 KB

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