Skip to content

Commit ef03580

Browse files
authored
ggml: Add Apple support for GGML_CPU_ALL_VARIANTS (#14258)
1 parent 413977d commit ef03580

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

ggml/src/CMakeLists.txt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -330,6 +330,10 @@ if (GGML_CPU_ALL_VARIANTS)
330330
ggml_add_cpu_backend_variant(android_armv8.2_1 DOTPROD)
331331
ggml_add_cpu_backend_variant(android_armv8.2_2 DOTPROD FP16_VECTOR_ARITHMETIC)
332332
ggml_add_cpu_backend_variant(android_armv8.6_1 DOTPROD FP16_VECTOR_ARITHMETIC MATMUL_INT8)
333+
elseif (APPLE)
334+
ggml_add_cpu_backend_variant(apple_m1 DOTPROD)
335+
ggml_add_cpu_backend_variant(apple_m2_m3 DOTPROD MATMUL_INT8)
336+
ggml_add_cpu_backend_variant(apple_m4 DOTPROD MATMUL_INT8 NOSVE SME)
333337
else()
334338
message(FATAL_ERROR "Unsupported ARM target OS: ${CMAKE_SYSTEM_NAME}")
335339
endif()

ggml/src/ggml-cpu/CMakeLists.txt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -190,6 +190,9 @@ function(ggml_add_cpu_backend_variant_impl tag_name)
190190
set(ARCH_TAGS "${ARCH_TAGS}+sve2")
191191
list(APPEND ARCH_DEFINITIONS GGML_USE_SVE2)
192192
endif()
193+
if (GGML_INTERNAL_NOSVE)
194+
set(ARCH_TAGS "${ARCH_TAGS}+nosve")
195+
endif()
193196
if (GGML_INTERNAL_SME)
194197
set(ARM_MCPU "armv9.2-a")
195198
set(ARCH_TAGS "${ARCH_TAGS}+sme")

0 commit comments

Comments
 (0)