Skip to content

Commit 071c312

Browse files
ggml: enable BMI2 on relevant CPU variants
1 parent d1aeed0 commit 071c312

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

ggml/src/CMakeLists.txt

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -289,7 +289,7 @@ function(ggml_add_cpu_backend_variant tag_name)
289289
set(GGML_CPU_TAG_NAME ${tag_name})
290290
# other: OPENMP LLAMAFILE CPU_HBM
291291
foreach (feat NATIVE
292-
AVX AVX2 AVX_VNNI FMA F16C
292+
AVX AVX2 BMI2 AVX_VNNI FMA F16C
293293
AVX512 AVX512_VBMI AVX512_VNNI AVX512_BF16
294294
AMX_TILE AMX_INT8 AMX_BF16)
295295
set(GGML_${feat} OFF)
@@ -309,13 +309,13 @@ if (GGML_CPU_ALL_VARIANTS)
309309
message(FATAL_ERROR "GGML_CPU_ALL_VARIANTS requires GGML_BACKEND_DL")
310310
endif()
311311
ggml_add_cpu_backend_variant(sandybridge AVX)
312-
ggml_add_cpu_backend_variant(haswell AVX F16C AVX2 FMA)
313-
ggml_add_cpu_backend_variant(skylakex AVX F16C AVX2 FMA AVX512)
314-
ggml_add_cpu_backend_variant(icelake AVX F16C AVX2 FMA AVX512 AVX512_VBMI AVX512_VNNI)
315-
ggml_add_cpu_backend_variant(alderlake AVX F16C AVX2 FMA AVX_VNNI)
312+
ggml_add_cpu_backend_variant(haswell AVX F16C AVX2 BMI2 FMA)
313+
ggml_add_cpu_backend_variant(skylakex AVX F16C AVX2 BMI2 FMA AVX512)
314+
ggml_add_cpu_backend_variant(icelake AVX F16C AVX2 BMI2 FMA AVX512 AVX512_VBMI AVX512_VNNI)
315+
ggml_add_cpu_backend_variant(alderlake AVX F16C AVX2 BMI2 FMA AVX_VNNI)
316316
if (NOT MSVC)
317317
# MSVC doesn't support AMX
318-
ggml_add_cpu_backend_variant(sapphirerapids AVX F16C AVX2 FMA AVX512 AVX512_VBMI AVX512_VNNI AVX512_BF16 AMX_TILE AMX_INT8)
318+
ggml_add_cpu_backend_variant(sapphirerapids AVX F16C AVX2 BMI2 FMA AVX512 AVX512_VBMI AVX512_VNNI AVX512_BF16 AMX_TILE AMX_INT8)
319319
endif()
320320
elseif (GGML_CPU)
321321
ggml_add_cpu_backend_variant_impl("")

0 commit comments

Comments
 (0)