fix: fix metal build (#513)
This commit is contained in:
parent
8f4ab9add3
commit
0d9d6659a7
@ -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);
|
||||||
|
@ -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
|
||||||
|
@ -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
|
||||||
|
Loading…
Reference in New Issue
Block a user