From 33e5a6612e0cb0c6ee441ef5033f25a20e11c87b Mon Sep 17 00:00:00 2001
From: slaren <slarengh@gmail.com>
Date: Sat, 21 Sep 2024 14:24:23 +0200
Subject: [PATCH] ggml-alloc : fix list of allocated tensors with
 GGML_ALLOCATOR_DEBUG (llama/9573)

---
 ggml/src/ggml-alloc.c | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/ggml/src/ggml-alloc.c b/ggml/src/ggml-alloc.c
index e485326a..70187b9b 100644
--- a/ggml/src/ggml-alloc.c
+++ b/ggml/src/ggml-alloc.c
@@ -294,6 +294,12 @@ static void ggml_dyn_tallocr_reset(struct ggml_dyn_tallocr * alloc) {
     alloc->free_blocks[0].offset = 0;
     alloc->free_blocks[0].size = SIZE_MAX/2; // restrict maximum size of a measure allocator to half size_t max to avoid overflows
     alloc->max_size = 0;
+
+#ifdef GGML_ALLOCATOR_DEBUG
+    for (int i = 0; i < 1024; i++) {
+        alloc->allocated_tensors[i].tensor = NULL;
+    }
+#endif
 }
 
 static struct ggml_dyn_tallocr * ggml_dyn_tallocr_new(size_t alignment) {