From 590e77394095641e733a7f287f0a9d695700ece5 Mon Sep 17 00:00:00 2001 From: Adam Treat Date: Tue, 13 Feb 2024 14:14:20 -0500 Subject: [PATCH] Add an early return to the get/set tensor when the size is null. Signed-off-by: Adam Treat --- ggml-backend.c | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/ggml-backend.c b/ggml-backend.c index 9ee81b766..cf6f485bf 100644 --- a/ggml-backend.c +++ b/ggml-backend.c @@ -213,6 +213,10 @@ void ggml_backend_tensor_get_async(ggml_backend_t backend, const struct ggml_ten } GGML_CALL void ggml_backend_tensor_set(struct ggml_tensor * tensor, const void * data, size_t offset, size_t size) { + if (!size) { + return; + } + ggml_backend_buffer_t buf = tensor->view_src ? tensor->view_src->buffer : tensor->buffer; GGML_ASSERT(tensor->data != NULL && "tensor not allocated"); @@ -223,6 +227,10 @@ GGML_CALL void ggml_backend_tensor_set(struct ggml_tensor * tensor, const void * } GGML_CALL void ggml_backend_tensor_get(const struct ggml_tensor * tensor, void * data, size_t offset, size_t size) { + if (!size) { + return; + } + ggml_backend_buffer_t buf = tensor->view_src ? tensor->view_src->buffer : tensor->buffer; GGML_ASSERT(tensor->data != NULL && "tensor not allocated");