CMakeLists.txt 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587
  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. option(LLAMA_CUDA_FORCE_DMMV "llama: use dmmv instead of mmvq CUDA kernels" OFF)
  59. set(LLAMA_CUDA_DMMV_X "32" CACHE STRING "llama: x stride for dmmv CUDA kernels")
  60. set(LLAMA_CUDA_MMV_Y "1" CACHE STRING "llama: y block size for mmv CUDA kernels")
  61. option(LLAMA_CUDA_F16 "llama: use 16 bit floats for some calculations" OFF)
  62. set(LLAMA_CUDA_KQUANTS_ITER "2" CACHE STRING "llama: iters./thread per block for Q2_K/Q6_K")
  63. option(LLAMA_CLBLAST "llama: use CLBlast" OFF)
  64. option(LLAMA_METAL "llama: use Metal" OFF)
  65. option(LLAMA_MPI "llama: use MPI" OFF)
  66. option(LLAMA_K_QUANTS "llama: use k-quants" ON)
  67. option(LLAMA_QKK_64 "llama: use super-block size of 64 for k-quants" OFF)
  68. option(LLAMA_BUILD_TESTS "llama: build tests" ${LLAMA_STANDALONE})
  69. option(LLAMA_BUILD_EXAMPLES "llama: build examples" ${LLAMA_STANDALONE})
  70. option(LLAMA_BUILD_SERVER "llama: build server example" ON)
  71. #
  72. # Build info header
  73. #
  74. # Generate initial build-info.h
  75. include(${CMAKE_CURRENT_SOURCE_DIR}/scripts/build-info.cmake)
  76. if(EXISTS "${CMAKE_CURRENT_SOURCE_DIR}/.git")
  77. set(GIT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/.git")
  78. # Is git submodule
  79. if(NOT IS_DIRECTORY "${GIT_DIR}")
  80. file(READ ${GIT_DIR} REAL_GIT_DIR_LINK)
  81. string(REGEX REPLACE "gitdir: (.*)\n$" "\\1" REAL_GIT_DIR ${REAL_GIT_DIR_LINK})
  82. set(GIT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/${REAL_GIT_DIR}")
  83. endif()
  84. # Add a custom target for build-info.h
  85. add_custom_target(BUILD_INFO ALL DEPENDS "${CMAKE_CURRENT_SOURCE_DIR}/build-info.h")
  86. # Add a custom command to rebuild build-info.h when .git/index changes
  87. add_custom_command(
  88. OUTPUT "${CMAKE_CURRENT_SOURCE_DIR}/build-info.h"
  89. COMMENT "Generating build details from Git"
  90. COMMAND ${CMAKE_COMMAND} -P "${CMAKE_CURRENT_SOURCE_DIR}/scripts/build-info.cmake"
  91. WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
  92. DEPENDS "${GIT_DIR}/index"
  93. VERBATIM
  94. )
  95. else()
  96. 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.")
  97. endif()
  98. #
  99. # Compile flags
  100. #
  101. set(CMAKE_CXX_STANDARD 11)
  102. set(CMAKE_CXX_STANDARD_REQUIRED true)
  103. set(CMAKE_C_STANDARD 11)
  104. set(CMAKE_C_STANDARD_REQUIRED true)
  105. set(THREADS_PREFER_PTHREAD_FLAG ON)
  106. find_package(Threads REQUIRED)
  107. if (NOT MSVC)
  108. if (LLAMA_SANITIZE_THREAD)
  109. add_compile_options(-fsanitize=thread)
  110. link_libraries(-fsanitize=thread)
  111. endif()
  112. if (LLAMA_SANITIZE_ADDRESS)
  113. add_compile_options(-fsanitize=address -fno-omit-frame-pointer)
  114. link_libraries(-fsanitize=address)
  115. endif()
  116. if (LLAMA_SANITIZE_UNDEFINED)
  117. add_compile_options(-fsanitize=undefined)
  118. link_libraries(-fsanitize=undefined)
  119. endif()
  120. endif()
  121. if (APPLE AND LLAMA_ACCELERATE)
  122. find_library(ACCELERATE_FRAMEWORK Accelerate)
  123. if (ACCELERATE_FRAMEWORK)
  124. message(STATUS "Accelerate framework found")
  125. add_compile_definitions(GGML_USE_ACCELERATE)
  126. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} ${ACCELERATE_FRAMEWORK})
  127. else()
  128. message(WARNING "Accelerate framework not found")
  129. endif()
  130. endif()
  131. if (LLAMA_BLAS)
  132. if (LLAMA_STATIC)
  133. set(BLA_STATIC ON)
  134. endif()
  135. if ($(CMAKE_VERSION) VERSION_GREATER_EQUAL 3.22)
  136. set(BLA_SIZEOF_INTEGER 8)
  137. endif()
  138. set(BLA_VENDOR ${LLAMA_BLAS_VENDOR})
  139. find_package(BLAS)
  140. if (BLAS_FOUND)
  141. message(STATUS "BLAS found, Libraries: ${BLAS_LIBRARIES}")
  142. if ("${BLAS_INCLUDE_DIRS}" STREQUAL "")
  143. # BLAS_INCLUDE_DIRS is missing in FindBLAS.cmake.
  144. # see https://gitlab.kitware.com/cmake/cmake/-/issues/20268
  145. find_package(PkgConfig REQUIRED)
  146. if (${LLAMA_BLAS_VENDOR} MATCHES "Generic")
  147. pkg_check_modules(DepBLAS REQUIRED blas)
  148. elseif (${LLAMA_BLAS_VENDOR} MATCHES "OpenBLAS")
  149. pkg_check_modules(DepBLAS REQUIRED openblas)
  150. elseif (${LLAMA_BLAS_VENDOR} MATCHES "FLAME")
  151. pkg_check_modules(DepBLAS REQUIRED blis)
  152. elseif (${LLAMA_BLAS_VENDOR} MATCHES "ATLAS")
  153. pkg_check_modules(DepBLAS REQUIRED blas-atlas)
  154. elseif (${LLAMA_BLAS_VENDOR} MATCHES "FlexiBLAS")
  155. pkg_check_modules(DepBLAS REQUIRED flexiblas_api)
  156. elseif (${LLAMA_BLAS_VENDOR} MATCHES "Intel")
  157. # all Intel* libraries share the same include path
  158. pkg_check_modules(DepBLAS REQUIRED mkl-sdl)
  159. elseif (${LLAMA_BLAS_VENDOR} MATCHES "NVHPC")
  160. # this doesn't provide pkg-config
  161. # suggest to assign BLAS_INCLUDE_DIRS on your own
  162. if ("${NVHPC_VERSION}" STREQUAL "")
  163. message(WARNING "Better to set NVHPC_VERSION")
  164. else()
  165. set(DepBLAS_FOUND ON)
  166. set(DepBLAS_INCLUDE_DIRS "/opt/nvidia/hpc_sdk/${CMAKE_SYSTEM_NAME}_${CMAKE_SYSTEM_PROCESSOR}/${NVHPC_VERSION}/math_libs/include")
  167. endif()
  168. endif()
  169. if (DepBLAS_FOUND)
  170. set(BLAS_INCLUDE_DIRS ${DepBLAS_INCLUDE_DIRS})
  171. else()
  172. message(WARNING "BLAS_INCLUDE_DIRS neither been provided nor been automatically"
  173. " detected by pkgconfig, trying to find cblas.h from possible paths...")
  174. find_path(BLAS_INCLUDE_DIRS
  175. NAMES cblas.h
  176. HINTS
  177. /usr/include
  178. /usr/local/include
  179. /usr/include/openblas
  180. /opt/homebrew/opt/openblas/include
  181. /usr/local/opt/openblas/include
  182. /usr/include/x86_64-linux-gnu/openblas/include
  183. )
  184. endif()
  185. endif()
  186. message(STATUS "BLAS found, Includes: ${BLAS_INCLUDE_DIRS}")
  187. add_compile_options(${BLAS_LINKER_FLAGS})
  188. add_compile_definitions(GGML_USE_OPENBLAS)
  189. if (${BLAS_INCLUDE_DIRS} MATCHES "mkl" AND (${LLAMA_BLAS_VENDOR} MATCHES "Generic" OR ${LLAMA_BLAS_VENDOR} MATCHES "Intel"))
  190. add_compile_definitions(GGML_BLAS_USE_MKL)
  191. endif()
  192. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} ${BLAS_LIBRARIES})
  193. set(LLAMA_EXTRA_INCLUDES ${LLAMA_EXTRA_INCLUDES} ${BLAS_INCLUDE_DIRS})
  194. else()
  195. message(WARNING "BLAS not found, please refer to "
  196. "https://cmake.org/cmake/help/latest/module/FindBLAS.html#blas-lapack-vendors"
  197. " to set correct LLAMA_BLAS_VENDOR")
  198. endif()
  199. endif()
  200. if (LLAMA_K_QUANTS)
  201. set(GGML_SOURCES_EXTRA ${GGML_SOURCES_EXTRA} k_quants.c k_quants.h)
  202. add_compile_definitions(GGML_USE_K_QUANTS)
  203. if (LLAMA_QKK_64)
  204. add_compile_definitions(GGML_QKK_64)
  205. endif()
  206. endif()
  207. if (LLAMA_CUBLAS)
  208. cmake_minimum_required(VERSION 3.17)
  209. find_package(CUDAToolkit)
  210. if (CUDAToolkit_FOUND)
  211. message(STATUS "cuBLAS found")
  212. enable_language(CUDA)
  213. set(GGML_SOURCES_CUDA ggml-cuda.cu ggml-cuda.h)
  214. add_compile_definitions(GGML_USE_CUBLAS)
  215. # if (LLAMA_CUDA_CUBLAS)
  216. # add_compile_definitions(GGML_CUDA_CUBLAS)
  217. # endif()
  218. if (LLAMA_CUDA_FORCE_DMMV)
  219. add_compile_definitions(GGML_CUDA_FORCE_DMMV)
  220. endif()
  221. add_compile_definitions(GGML_CUDA_DMMV_X=${LLAMA_CUDA_DMMV_X})
  222. add_compile_definitions(GGML_CUDA_MMV_Y=${LLAMA_CUDA_MMV_Y})
  223. if (DEFINED LLAMA_CUDA_DMMV_Y)
  224. add_compile_definitions(GGML_CUDA_MMV_Y=${LLAMA_CUDA_DMMV_Y}) # for backwards compatibility
  225. endif()
  226. if (LLAMA_CUDA_F16 OR LLAMA_CUDA_DMMV_F16)
  227. add_compile_definitions(GGML_CUDA_F16)
  228. endif()
  229. add_compile_definitions(K_QUANTS_PER_ITERATION=${LLAMA_CUDA_KQUANTS_ITER})
  230. if (LLAMA_STATIC)
  231. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} CUDA::cudart_static CUDA::cublas_static CUDA::cublasLt_static)
  232. else()
  233. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} CUDA::cudart CUDA::cublas CUDA::cublasLt)
  234. endif()
  235. if (NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
  236. # 52 == lowest CUDA 12 standard
  237. # 60 == f16 CUDA intrinsics
  238. # 61 == integer CUDA intrinsics
  239. # 70 == compute capability at which unrolling a loop in mul_mat_q kernels is faster
  240. if (LLAMA_CUDA_F16 OR LLAMA_CUDA_DMMV_F16)
  241. set(CMAKE_CUDA_ARCHITECTURES "60;61;70") # needed for f16 CUDA intrinsics
  242. else()
  243. set(CMAKE_CUDA_ARCHITECTURES "52;61;70") # lowest CUDA 12 standard + lowest for integer intrinsics
  244. endif()
  245. endif()
  246. message(STATUS "Using CUDA architectures: ${CMAKE_CUDA_ARCHITECTURES}")
  247. else()
  248. message(WARNING "cuBLAS not found")
  249. endif()
  250. endif()
  251. if (LLAMA_METAL)
  252. find_library(FOUNDATION_LIBRARY Foundation REQUIRED)
  253. find_library(METAL_FRAMEWORK Metal REQUIRED)
  254. find_library(METALKIT_FRAMEWORK MetalKit REQUIRED)
  255. find_library(METALPERFORMANCE_FRAMEWORK MetalPerformanceShaders REQUIRED)
  256. set(GGML_SOURCES_METAL ggml-metal.m ggml-metal.h)
  257. add_compile_definitions(GGML_USE_METAL)
  258. add_compile_definitions(GGML_METAL_NDEBUG)
  259. # get full path to the file
  260. #add_compile_definitions(GGML_METAL_DIR_KERNELS="${CMAKE_CURRENT_SOURCE_DIR}/")
  261. # copy ggml-metal.metal to bin directory
  262. configure_file(ggml-metal.metal bin/ggml-metal.metal COPYONLY)
  263. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS}
  264. ${FOUNDATION_LIBRARY}
  265. ${METAL_FRAMEWORK}
  266. ${METALKIT_FRAMEWORK}
  267. ${METALPERFORMANCE_FRAMEWORK}
  268. )
  269. endif()
  270. if (LLAMA_MPI)
  271. cmake_minimum_required(VERSION 3.10)
  272. find_package(MPI)
  273. if (MPI_C_FOUND)
  274. message(STATUS "MPI found")
  275. set(GGML_SOURCES_MPI ggml-mpi.c ggml-mpi.h)
  276. add_compile_definitions(GGML_USE_MPI)
  277. add_compile_definitions(${MPI_C_COMPILE_DEFINITIONS})
  278. set(cxx_flags ${cxx_flags} -Wno-cast-qual)
  279. set(c_flags ${c_flags} -Wno-cast-qual)
  280. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} ${MPI_C_LIBRARIES})
  281. set(LLAMA_EXTRA_INCLUDES ${LLAMA_EXTRA_INCLUDES} ${MPI_C_INCLUDE_DIRS})
  282. # Even if you're only using the C header, C++ programs may bring in MPI
  283. # C++ functions, so more linkage is needed
  284. if (MPI_CXX_FOUND)
  285. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} ${MPI_CXX_LIBRARIES})
  286. endif()
  287. else()
  288. message(WARNING "MPI not found")
  289. endif()
  290. endif()
  291. if (LLAMA_CLBLAST)
  292. find_package(CLBlast)
  293. if (CLBlast_FOUND)
  294. message(STATUS "CLBlast found")
  295. set(GGML_SOURCES_OPENCL ggml-opencl.cpp ggml-opencl.h)
  296. add_compile_definitions(GGML_USE_CLBLAST)
  297. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} clblast)
  298. else()
  299. message(WARNING "CLBlast not found")
  300. endif()
  301. endif()
  302. if (LLAMA_ALL_WARNINGS)
  303. if (NOT MSVC)
  304. set(c_flags
  305. -Wall
  306. -Wextra
  307. -Wpedantic
  308. -Wcast-qual
  309. -Wdouble-promotion
  310. -Wshadow
  311. -Wstrict-prototypes
  312. -Wpointer-arith
  313. -Wmissing-prototypes
  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-alloc.c
  445. ggml-alloc.h
  446. ${GGML_SOURCES_CUDA}
  447. ${GGML_SOURCES_OPENCL}
  448. ${GGML_SOURCES_METAL}
  449. ${GGML_SOURCES_MPI}
  450. ${GGML_SOURCES_EXTRA}
  451. )
  452. target_include_directories(ggml PUBLIC . ${LLAMA_EXTRA_INCLUDES})
  453. target_compile_features(ggml PUBLIC c_std_11) # don't bump
  454. target_link_libraries(ggml PUBLIC Threads::Threads ${LLAMA_EXTRA_LIBS})
  455. add_library(ggml_static STATIC $<TARGET_OBJECTS:ggml>)
  456. if (BUILD_SHARED_LIBS)
  457. set_target_properties(ggml PROPERTIES POSITION_INDEPENDENT_CODE ON)
  458. add_library(ggml_shared SHARED $<TARGET_OBJECTS:ggml>)
  459. target_link_libraries(ggml_shared PUBLIC Threads::Threads ${LLAMA_EXTRA_LIBS})
  460. install(TARGETS ggml_shared LIBRARY)
  461. endif()
  462. add_library(llama
  463. llama.cpp
  464. llama.h
  465. llama-util.h
  466. )
  467. target_include_directories(llama PUBLIC .)
  468. target_compile_features(llama PUBLIC cxx_std_11) # don't bump
  469. target_link_libraries(llama PRIVATE
  470. ggml
  471. ${LLAMA_EXTRA_LIBS}
  472. )
  473. if (BUILD_SHARED_LIBS)
  474. set_target_properties(llama PROPERTIES POSITION_INDEPENDENT_CODE ON)
  475. target_compile_definitions(llama PRIVATE LLAMA_SHARED LLAMA_BUILD)
  476. if (LLAMA_METAL)
  477. set_target_properties(llama PROPERTIES RESOURCE "${CMAKE_CURRENT_SOURCE_DIR}/ggml-metal.metal")
  478. endif()
  479. install(TARGETS llama LIBRARY)
  480. endif()
  481. include(GNUInstallDirs)
  482. install(
  483. FILES convert.py
  484. PERMISSIONS
  485. OWNER_READ
  486. OWNER_WRITE
  487. OWNER_EXECUTE
  488. GROUP_READ
  489. GROUP_EXECUTE
  490. WORLD_READ
  491. WORLD_EXECUTE
  492. DESTINATION ${CMAKE_INSTALL_BINDIR})
  493. install(
  494. FILES convert-lora-to-ggml.py
  495. PERMISSIONS
  496. OWNER_READ
  497. OWNER_WRITE
  498. OWNER_EXECUTE
  499. GROUP_READ
  500. GROUP_EXECUTE
  501. WORLD_READ
  502. WORLD_EXECUTE
  503. DESTINATION ${CMAKE_INSTALL_BINDIR})
  504. #
  505. # programs, examples and tests
  506. #
  507. if (LLAMA_BUILD_TESTS AND NOT CMAKE_JS_VERSION)
  508. include(CTest)
  509. add_subdirectory(tests)
  510. endif ()
  511. if (LLAMA_BUILD_EXAMPLES)
  512. add_subdirectory(examples)
  513. add_subdirectory(pocs)
  514. endif()