@@ -291,16 +291,20 @@ if (GGML_CPU_ALL_VARIANTS)
291
291
if (NOT GGML_BACKEND_DL )
292
292
message (FATAL_ERROR "GGML_CPU_ALL_VARIANTS requires GGML_BACKEND_DL" )
293
293
endif ()
294
- ggml_add_cpu_backend_variant (x64 )
295
- ggml_add_cpu_backend_variant (sse42 SSE42 )
296
- ggml_add_cpu_backend_variant (sandybridge SSE42 AVX )
297
- ggml_add_cpu_backend_variant (haswell SSE42 AVX F16C AVX2 BMI2 FMA )
298
- ggml_add_cpu_backend_variant (skylakex SSE42 AVX F16C AVX2 BMI2 FMA AVX512 )
299
- ggml_add_cpu_backend_variant (icelake SSE42 AVX F16C AVX2 BMI2 FMA AVX512 AVX512_VBMI AVX512_VNNI )
300
- ggml_add_cpu_backend_variant (alderlake SSE42 AVX F16C AVX2 BMI2 FMA AVX_VNNI )
301
- if (NOT MSVC )
302
- # MSVC doesn't support AMX
303
- ggml_add_cpu_backend_variant (sapphirerapids SSE42 AVX F16C AVX2 BMI2 FMA AVX512 AVX512_VBMI AVX512_VNNI AVX512_BF16 AMX_TILE AMX_INT8 )
294
+ if (GGML_SYSTEM_ARCH STREQUAL "x86" )
295
+ ggml_add_cpu_backend_variant (x64 )
296
+ ggml_add_cpu_backend_variant (sse42 SSE42 )
297
+ ggml_add_cpu_backend_variant (sandybridge SSE42 AVX )
298
+ ggml_add_cpu_backend_variant (haswell SSE42 AVX F16C AVX2 BMI2 FMA )
299
+ ggml_add_cpu_backend_variant (skylakex SSE42 AVX F16C AVX2 BMI2 FMA AVX512 )
300
+ ggml_add_cpu_backend_variant (icelake SSE42 AVX F16C AVX2 BMI2 FMA AVX512 AVX512_VBMI AVX512_VNNI )
301
+ ggml_add_cpu_backend_variant (alderlake SSE42 AVX F16C AVX2 BMI2 FMA AVX_VNNI )
302
+ if (NOT MSVC )
303
+ # MSVC doesn't support AMX
304
+ ggml_add_cpu_backend_variant (sapphirerapids SSE42 AVX F16C AVX2 BMI2 FMA AVX512 AVX512_VBMI AVX512_VNNI AVX512_BF16 AMX_TILE AMX_INT8 )
305
+ endif ()
306
+ else ()
307
+ message (FATAL_ERROR "GGML_CPU_ALL_VARIANTS not yet supported on ${GGML_SYSTEM_ARCH} " )
304
308
endif ()
305
309
elseif (GGML_CPU )
306
310
ggml_add_cpu_backend_variant_impl ("" )
0 commit comments