From cdb81d4ea16ff80a03dd3fc51a01b97b584275e3 Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Sun, 7 Jan 2024 12:20:04 +0200 Subject: [PATCH] ggml : fix null backend dereference --- ggml-cuda.cu | 2 +- ggml-metal.m | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ggml-cuda.cu b/ggml-cuda.cu index 032d7e986..ee82f68d7 100644 --- a/ggml-cuda.cu +++ b/ggml-cuda.cu @@ -10140,7 +10140,7 @@ ggml_backend_t ggml_backend_cuda_init(int device) { } bool ggml_backend_is_cuda(ggml_backend_t backend) { - return backend->iface.get_name == ggml_backend_cuda_name; + return backend && backend->iface.get_name == ggml_backend_cuda_name; } int ggml_backend_cuda_get_device_count() { diff --git a/ggml-metal.m b/ggml-metal.m index e3d76b9d3..00b377a3e 100644 --- a/ggml-metal.m +++ b/ggml-metal.m @@ -2749,7 +2749,7 @@ ggml_backend_t ggml_backend_metal_init(void) { } bool ggml_backend_is_metal(ggml_backend_t backend) { - return backend->iface.get_name == ggml_backend_metal_name; + return backend && backend->iface.get_name == ggml_backend_metal_name; } void ggml_backend_metal_set_n_cb(ggml_backend_t backend, int n_cb) {