Skip to content

Commit d1aeed0

Browse files
cmake: Add GGML_BMI2 build option
1 parent baa79d6 commit d1aeed0

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

ggml/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,7 @@ option(GGML_CPU_KLEIDIAI "ggml: use KleidiAI optimized kernels if applicable
106106
option(GGML_AVX "ggml: enable AVX" ${INS_ENB})
107107
option(GGML_AVX_VNNI "ggml: enable AVX-VNNI" OFF)
108108
option(GGML_AVX2 "ggml: enable AVX2" ${INS_ENB})
109+
option(GGML_BMI2 "ggml: enable BMI2" ${INS_ENB})
109110
option(GGML_AVX512 "ggml: enable AVX512F" OFF)
110111
option(GGML_AVX512_VBMI "ggml: enable AVX512-VBMI" OFF)
111112
option(GGML_AVX512_VNNI "ggml: enable AVX512-VNNI" OFF)

ggml/src/ggml-cpu/CMakeLists.txt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -233,6 +233,10 @@ function(ggml_add_cpu_backend_variant_impl tag_name)
233233
list(APPEND ARCH_FLAGS -mfma)
234234
list(APPEND ARCH_DEFINITIONS GGML_FMA)
235235
endif()
236+
if (GGML_BMI2)
237+
list(APPEND ARCH_FLAGS -mbmi2)
238+
list(APPEND ARCH_DEFINITIONS GGML_BMI2)
239+
endif()
236240
if (GGML_AVX)
237241
list(APPEND ARCH_FLAGS -mavx)
238242
list(APPEND ARCH_DEFINITIONS GGML_AVX)

0 commit comments

Comments
 (0)