From: Jared Van Bortel Date: Wed, 31 Jan 2024 00:04:37 +0000 (-0500) Subject: kompute : llama-bench support and ggml_cpu_has_kompute() (llama/5226) X-Git-Tag: upstream/0.0.1642~1007 X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=df0c703669617d4fbc323d35702a13ea3fba98dc;p=pkg%2Fggml%2Fsources%2Fggml kompute : llama-bench support and ggml_cpu_has_kompute() (llama/5226) --- diff --git a/include/ggml/ggml.h b/include/ggml/ggml.h index e20b14fa..3b264bfd 100644 --- a/include/ggml/ggml.h +++ b/include/ggml/ggml.h @@ -2271,6 +2271,7 @@ extern "C" { GGML_API int ggml_cpu_has_cublas (void); GGML_API int ggml_cpu_has_clblast (void); GGML_API int ggml_cpu_has_vulkan (void); + GGML_API int ggml_cpu_has_kompute (void); GGML_API int ggml_cpu_has_gpublas (void); GGML_API int ggml_cpu_has_sse3 (void); GGML_API int ggml_cpu_has_ssse3 (void); diff --git a/src/ggml.c b/src/ggml.c index 3499b737..f6e797d7 100644 --- a/src/ggml.c +++ b/src/ggml.c @@ -20473,6 +20473,14 @@ int ggml_cpu_has_vulkan(void) { #endif } +int ggml_cpu_has_kompute(void) { +#if defined(GGML_USE_KOMPUTE) + return 1; +#else + return 0; +#endif +} + int ggml_cpu_has_sycl(void) { #if defined(GGML_USE_SYCL) return 1; @@ -20482,7 +20490,8 @@ int ggml_cpu_has_sycl(void) { } int ggml_cpu_has_gpublas(void) { - return ggml_cpu_has_cublas() || ggml_cpu_has_clblast() || ggml_cpu_has_vulkan() || ggml_cpu_has_sycl(); + return ggml_cpu_has_cublas() || ggml_cpu_has_clblast() || ggml_cpu_has_vulkan() || ggml_cpu_has_kompute() || + ggml_cpu_has_sycl(); } int ggml_cpu_has_sse3(void) {