ggml: Add Apple support for GGML_CPU_ALL_VARIANTS (#14258)
This commit is contained in:
parent
413977de32
commit
ef035803eb
2 changed files with 7 additions and 0 deletions
|
@ -330,6 +330,10 @@ if (GGML_CPU_ALL_VARIANTS)
|
||||||
ggml_add_cpu_backend_variant(android_armv8.2_1 DOTPROD)
|
ggml_add_cpu_backend_variant(android_armv8.2_1 DOTPROD)
|
||||||
ggml_add_cpu_backend_variant(android_armv8.2_2 DOTPROD FP16_VECTOR_ARITHMETIC)
|
ggml_add_cpu_backend_variant(android_armv8.2_2 DOTPROD FP16_VECTOR_ARITHMETIC)
|
||||||
ggml_add_cpu_backend_variant(android_armv8.6_1 DOTPROD FP16_VECTOR_ARITHMETIC MATMUL_INT8)
|
ggml_add_cpu_backend_variant(android_armv8.6_1 DOTPROD FP16_VECTOR_ARITHMETIC MATMUL_INT8)
|
||||||
|
elseif (APPLE)
|
||||||
|
ggml_add_cpu_backend_variant(apple_m1 DOTPROD)
|
||||||
|
ggml_add_cpu_backend_variant(apple_m2_m3 DOTPROD MATMUL_INT8)
|
||||||
|
ggml_add_cpu_backend_variant(apple_m4 DOTPROD MATMUL_INT8 NOSVE SME)
|
||||||
else()
|
else()
|
||||||
message(FATAL_ERROR "Unsupported ARM target OS: ${CMAKE_SYSTEM_NAME}")
|
message(FATAL_ERROR "Unsupported ARM target OS: ${CMAKE_SYSTEM_NAME}")
|
||||||
endif()
|
endif()
|
||||||
|
|
|
@ -190,6 +190,9 @@ function(ggml_add_cpu_backend_variant_impl tag_name)
|
||||||
set(ARCH_TAGS "${ARCH_TAGS}+sve2")
|
set(ARCH_TAGS "${ARCH_TAGS}+sve2")
|
||||||
list(APPEND ARCH_DEFINITIONS GGML_USE_SVE2)
|
list(APPEND ARCH_DEFINITIONS GGML_USE_SVE2)
|
||||||
endif()
|
endif()
|
||||||
|
if (GGML_INTERNAL_NOSVE)
|
||||||
|
set(ARCH_TAGS "${ARCH_TAGS}+nosve")
|
||||||
|
endif()
|
||||||
if (GGML_INTERNAL_SME)
|
if (GGML_INTERNAL_SME)
|
||||||
set(ARM_MCPU "armv9.2-a")
|
set(ARM_MCPU "armv9.2-a")
|
||||||
set(ARCH_TAGS "${ARCH_TAGS}+sme")
|
set(ARCH_TAGS "${ARCH_TAGS}+sme")
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue