llama : add missing call to ggml_backend_load_all() (#22752)
Signed-off-by: Adrien Gallouët <angt@huggingface.co>
This commit is contained in:
@@ -71,12 +71,18 @@ bool llama_supports_mlock(void) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
bool llama_supports_gpu_offload(void) {
|
bool llama_supports_gpu_offload(void) {
|
||||||
|
if (!ggml_backend_reg_count()) {
|
||||||
|
ggml_backend_load_all();
|
||||||
|
}
|
||||||
return ggml_backend_dev_by_type(GGML_BACKEND_DEVICE_TYPE_GPU) != nullptr ||
|
return ggml_backend_dev_by_type(GGML_BACKEND_DEVICE_TYPE_GPU) != nullptr ||
|
||||||
ggml_backend_dev_by_type(GGML_BACKEND_DEVICE_TYPE_IGPU) != nullptr ||
|
ggml_backend_dev_by_type(GGML_BACKEND_DEVICE_TYPE_IGPU) != nullptr ||
|
||||||
llama_supports_rpc();
|
llama_supports_rpc();
|
||||||
}
|
}
|
||||||
|
|
||||||
bool llama_supports_rpc(void) {
|
bool llama_supports_rpc(void) {
|
||||||
|
if (!ggml_backend_reg_count()) {
|
||||||
|
ggml_backend_load_all();
|
||||||
|
}
|
||||||
return ggml_backend_reg_by_name("RPC") != nullptr;
|
return ggml_backend_reg_by_name("RPC") != nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user