From cd0022f70f7f93220db3146cae401d20df9100d4 Mon Sep 17 00:00:00 2001 From: l3utterfly Date: Sat, 7 Oct 2023 22:55:16 +0800 Subject: [PATCH] added GGML_ALLOCATOR_DEBUG code back --- ggml-alloc.c | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/ggml-alloc.c b/ggml-alloc.c index 66e1eee5b..8d1117b20 100644 --- a/ggml-alloc.c +++ b/ggml-alloc.c @@ -303,6 +303,10 @@ struct ggml_allocr * ggml_allocr_new(void * data, size_t size, size_t alignment) memset((*alloc).hash_table, 0, sizeof((*alloc).hash_table)); memset((*alloc).parse_seq, 0, sizeof((*alloc).parse_seq)); +#ifdef GGML_ALLOCATOR_DEBUG + memset((*alloc).allocated_tensors, 0, sizeof((*alloc).allocated_tensors)); +#endif + ggml_allocr_reset(alloc); return alloc; @@ -380,7 +384,7 @@ struct ggml_allocr * ggml_allocr_new_measure(size_t alignment) { memset((*alloc).parse_seq, 0, sizeof((*alloc).parse_seq)); #ifdef GGML_ALLOCATOR_DEBUG - (*alloc).allocated_tensors = {0}; + memset((*alloc).allocated_tensors, 0, sizeof((*alloc).allocated_tensors)); #endif ggml_allocr_reset(alloc);