CMakeLists.txt 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604
  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. set(GGML_SOURCES_METAL ggml-metal.m ggml-metal.h)
  256. add_compile_definitions(GGML_USE_METAL)
  257. add_compile_definitions(GGML_METAL_NDEBUG)
  258. # get full path to the file
  259. #add_compile_definitions(GGML_METAL_DIR_KERNELS="${CMAKE_CURRENT_SOURCE_DIR}/")
  260. # copy ggml-metal.metal to bin directory
  261. configure_file(ggml-metal.metal bin/ggml-metal.metal COPYONLY)
  262. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS}
  263. ${FOUNDATION_LIBRARY}
  264. ${METAL_FRAMEWORK}
  265. ${METALKIT_FRAMEWORK}
  266. )
  267. endif()
  268. if (LLAMA_MPI)
  269. cmake_minimum_required(VERSION 3.10)
  270. find_package(MPI)
  271. if (MPI_C_FOUND)
  272. message(STATUS "MPI found")
  273. set(GGML_SOURCES_MPI ggml-mpi.c ggml-mpi.h)
  274. add_compile_definitions(GGML_USE_MPI)
  275. add_compile_definitions(${MPI_C_COMPILE_DEFINITIONS})
  276. set(cxx_flags ${cxx_flags} -Wno-cast-qual)
  277. set(c_flags ${c_flags} -Wno-cast-qual)
  278. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} ${MPI_C_LIBRARIES})
  279. set(LLAMA_EXTRA_INCLUDES ${LLAMA_EXTRA_INCLUDES} ${MPI_C_INCLUDE_DIRS})
  280. # Even if you're only using the C header, C++ programs may bring in MPI
  281. # C++ functions, so more linkage is needed
  282. if (MPI_CXX_FOUND)
  283. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} ${MPI_CXX_LIBRARIES})
  284. endif()
  285. else()
  286. message(WARNING "MPI not found")
  287. endif()
  288. endif()
  289. if (LLAMA_CLBLAST)
  290. find_package(CLBlast)
  291. if (CLBlast_FOUND)
  292. message(STATUS "CLBlast found")
  293. set(GGML_SOURCES_OPENCL ggml-opencl.cpp ggml-opencl.h)
  294. add_compile_definitions(GGML_USE_CLBLAST)
  295. set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} clblast)
  296. else()
  297. message(WARNING "CLBlast not found")
  298. endif()
  299. endif()
  300. if (LLAMA_ALL_WARNINGS)
  301. if (NOT MSVC)
  302. set(c_flags
  303. -Wall
  304. -Wextra
  305. -Wpedantic
  306. -Wcast-qual
  307. -Wdouble-promotion
  308. -Wshadow
  309. -Wstrict-prototypes
  310. -Wpointer-arith
  311. -Wmissing-prototypes
  312. )
  313. set(cxx_flags
  314. -Wall
  315. -Wextra
  316. -Wpedantic
  317. -Wcast-qual
  318. -Wno-unused-function
  319. -Wno-multichar
  320. )
  321. else()
  322. # todo : msvc
  323. endif()
  324. add_compile_options(
  325. "$<$<COMPILE_LANGUAGE:C>:${c_flags}>"
  326. "$<$<COMPILE_LANGUAGE:CXX>:${cxx_flags}>"
  327. )
  328. endif()
  329. if (MSVC)
  330. add_compile_definitions(_CRT_SECURE_NO_WARNINGS)
  331. if (BUILD_SHARED_LIBS)
  332. set(CMAKE_WINDOWS_EXPORT_ALL_SYMBOLS ON)
  333. endif()
  334. endif()
  335. if (LLAMA_LTO)
  336. include(CheckIPOSupported)
  337. check_ipo_supported(RESULT result OUTPUT output)
  338. if (result)
  339. set(CMAKE_INTERPROCEDURAL_OPTIMIZATION TRUE)
  340. else()
  341. message(WARNING "IPO is not supported: ${output}")
  342. endif()
  343. endif()
  344. # Architecture specific
  345. # TODO: probably these flags need to be tweaked on some architectures
  346. # feel free to update the Makefile for your architecture and send a pull request or issue
  347. message(STATUS "CMAKE_SYSTEM_PROCESSOR: ${CMAKE_SYSTEM_PROCESSOR}")
  348. if (NOT MSVC)
  349. if (LLAMA_STATIC)
  350. add_link_options(-static)
  351. if (MINGW)
  352. add_link_options(-static-libgcc -static-libstdc++)
  353. endif()
  354. endif()
  355. if (LLAMA_GPROF)
  356. add_compile_options(-pg)
  357. endif()
  358. if (LLAMA_NATIVE)
  359. add_compile_options(-march=native)
  360. endif()
  361. endif()
  362. if (${CMAKE_SYSTEM_PROCESSOR} MATCHES "arm" OR ${CMAKE_SYSTEM_PROCESSOR} MATCHES "aarch64")
  363. message(STATUS "ARM detected")
  364. if (MSVC)
  365. # TODO: arm msvc?
  366. else()
  367. if (${CMAKE_SYSTEM_PROCESSOR} MATCHES "armv6")
  368. # Raspberry Pi 1, Zero
  369. add_compile_options(-mfpu=neon-fp-armv8 -mfp16-format=ieee -mno-unaligned-access)
  370. endif()
  371. if (${CMAKE_SYSTEM_PROCESSOR} MATCHES "armv7")
  372. # Raspberry Pi 2
  373. add_compile_options(-mfpu=neon-fp-armv8 -mfp16-format=ieee -mno-unaligned-access -funsafe-math-optimizations)
  374. endif()
  375. if (${CMAKE_SYSTEM_PROCESSOR} MATCHES "armv8")
  376. # Raspberry Pi 3, 4, Zero 2 (32-bit)
  377. add_compile_options(-mfp16-format=ieee -mno-unaligned-access)
  378. endif()
  379. endif()
  380. elseif (${CMAKE_SYSTEM_PROCESSOR} MATCHES "^(x86_64|i686|AMD64)$")
  381. message(STATUS "x86 detected")
  382. if (MSVC)
  383. if (LLAMA_AVX512)
  384. add_compile_options($<$<COMPILE_LANGUAGE:C>:/arch:AVX512>)
  385. add_compile_options($<$<COMPILE_LANGUAGE:CXX>:/arch:AVX512>)
  386. # MSVC has no compile-time flags enabling specific
  387. # AVX512 extensions, neither it defines the
  388. # macros corresponding to the extensions.
  389. # Do it manually.
  390. if (LLAMA_AVX512_VBMI)
  391. add_compile_definitions($<$<COMPILE_LANGUAGE:C>:__AVX512VBMI__>)
  392. add_compile_definitions($<$<COMPILE_LANGUAGE:CXX>:__AVX512VBMI__>)
  393. endif()
  394. if (LLAMA_AVX512_VNNI)
  395. add_compile_definitions($<$<COMPILE_LANGUAGE:C>:__AVX512VNNI__>)
  396. add_compile_definitions($<$<COMPILE_LANGUAGE:CXX>:__AVX512VNNI__>)
  397. endif()
  398. elseif (LLAMA_AVX2)
  399. add_compile_options($<$<COMPILE_LANGUAGE:C>:/arch:AVX2>)
  400. add_compile_options($<$<COMPILE_LANGUAGE:CXX>:/arch:AVX2>)
  401. elseif (LLAMA_AVX)
  402. add_compile_options($<$<COMPILE_LANGUAGE:C>:/arch:AVX>)
  403. add_compile_options($<$<COMPILE_LANGUAGE:CXX>:/arch:AVX>)
  404. endif()
  405. else()
  406. if (LLAMA_F16C)
  407. add_compile_options(-mf16c)
  408. endif()
  409. if (LLAMA_FMA)
  410. add_compile_options(-mfma)
  411. endif()
  412. if (LLAMA_AVX)
  413. add_compile_options(-mavx)
  414. endif()
  415. if (LLAMA_AVX2)
  416. add_compile_options(-mavx2)
  417. endif()
  418. if (LLAMA_AVX512)
  419. add_compile_options(-mavx512f)
  420. add_compile_options(-mavx512bw)
  421. endif()
  422. if (LLAMA_AVX512_VBMI)
  423. add_compile_options(-mavx512vbmi)
  424. endif()
  425. if (LLAMA_AVX512_VNNI)
  426. add_compile_options(-mavx512vnni)
  427. endif()
  428. endif()
  429. elseif (${CMAKE_SYSTEM_PROCESSOR} MATCHES "ppc64")
  430. message(STATUS "PowerPC detected")
  431. add_compile_options(-mcpu=native -mtune=native)
  432. #TODO: Add targets for Power8/Power9 (Altivec/VSX) and Power10(MMA) and query for big endian systems (ppc64/le/be)
  433. else()
  434. message(STATUS "Unknown architecture")
  435. endif()
  436. #
  437. # libraries
  438. #
  439. # ggml
  440. add_library(ggml OBJECT
  441. ggml.c
  442. ggml.h
  443. ggml-alloc.c
  444. ggml-alloc.h
  445. ${GGML_SOURCES_CUDA}
  446. ${GGML_SOURCES_OPENCL}
  447. ${GGML_SOURCES_METAL}
  448. ${GGML_SOURCES_MPI}
  449. ${GGML_SOURCES_EXTRA}
  450. )
  451. target_include_directories(ggml PUBLIC . ${LLAMA_EXTRA_INCLUDES})
  452. target_compile_features(ggml PUBLIC c_std_11) # don't bump
  453. target_link_libraries(ggml PUBLIC Threads::Threads ${LLAMA_EXTRA_LIBS})
  454. add_library(ggml_static STATIC $<TARGET_OBJECTS:ggml>)
  455. if (BUILD_SHARED_LIBS)
  456. set_target_properties(ggml PROPERTIES POSITION_INDEPENDENT_CODE ON)
  457. add_library(ggml_shared SHARED $<TARGET_OBJECTS:ggml>)
  458. target_link_libraries(ggml_shared PUBLIC Threads::Threads ${LLAMA_EXTRA_LIBS})
  459. install(TARGETS ggml_shared LIBRARY)
  460. endif()
  461. # llama
  462. add_library(llama
  463. llama.cpp
  464. llama.h
  465. )
  466. target_include_directories(llama PUBLIC .)
  467. target_compile_features(llama PUBLIC cxx_std_11) # don't bump
  468. target_link_libraries(llama PRIVATE
  469. ggml
  470. ${LLAMA_EXTRA_LIBS}
  471. )
  472. if (BUILD_SHARED_LIBS)
  473. set_target_properties(llama PROPERTIES POSITION_INDEPENDENT_CODE ON)
  474. target_compile_definitions(llama PRIVATE LLAMA_SHARED LLAMA_BUILD)
  475. if (LLAMA_METAL)
  476. set_target_properties(llama PROPERTIES RESOURCE "${CMAKE_CURRENT_SOURCE_DIR}/ggml-metal.metal")
  477. endif()
  478. install(TARGETS llama LIBRARY)
  479. endif()
  480. #
  481. # install
  482. #
  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. if (LLAMA_METAL)
  507. install(
  508. FILES ggml-metal.metal
  509. PERMISSIONS
  510. OWNER_READ
  511. OWNER_WRITE
  512. GROUP_READ
  513. WORLD_READ
  514. DESTINATION ${CMAKE_INSTALL_BINDIR})
  515. endif()
  516. #
  517. # programs, examples and tests
  518. #
  519. add_subdirectory(common)
  520. if (LLAMA_BUILD_TESTS AND NOT CMAKE_JS_VERSION)
  521. include(CTest)
  522. add_subdirectory(tests)
  523. endif ()
  524. if (LLAMA_BUILD_EXAMPLES)
  525. add_subdirectory(examples)
  526. add_subdirectory(pocs)
  527. endif()