fix: fix metal build (#513)

This commit is contained in:
stduhpf 2024-12-28 06:06:17 +01:00 committed by GitHub
parent 8f4ab9add3
commit 0d9d6659a7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 2 additions and 7 deletions

View File

@ -1174,11 +1174,6 @@ public:
ggml_backend_cpu_set_n_threads(backend, n_threads); ggml_backend_cpu_set_n_threads(backend, n_threads);
} }
#ifdef SD_USE_METAL
if (ggml_backend_is_metal(backend)) {
ggml_backend_metal_set_n_cb(backend, n_threads);
}
#endif
ggml_backend_graph_compute(backend, gf); ggml_backend_graph_compute(backend, gf);
#ifdef GGML_PERF #ifdef GGML_PERF
ggml_graph_print(gf); ggml_graph_print(gf);

View File

@ -165,7 +165,7 @@ public:
#endif #endif
#ifdef SD_USE_METAL #ifdef SD_USE_METAL
LOG_DEBUG("Using Metal backend"); LOG_DEBUG("Using Metal backend");
ggml_backend_metal_log_set_callback(ggml_log_callback_default, nullptr); ggml_log_set(ggml_log_callback_default, nullptr);
backend = ggml_backend_metal_init(); backend = ggml_backend_metal_init();
#endif #endif
#ifdef SD_USE_VULKAN #ifdef SD_USE_VULKAN

View File

@ -21,7 +21,7 @@ struct UpscalerGGML {
#endif #endif
#ifdef SD_USE_METAL #ifdef SD_USE_METAL
LOG_DEBUG("Using Metal backend"); LOG_DEBUG("Using Metal backend");
ggml_backend_metal_log_set_callback(ggml_log_callback_default, nullptr); ggml_log_set(ggml_log_callback_default, nullptr);
backend = ggml_backend_metal_init(); backend = ggml_backend_metal_init();
#endif #endif
#ifdef SD_USE_VULKAN #ifdef SD_USE_VULKAN