vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added GGML_ALLOCATOR_DEBUG code back

Browse source
This commit is contained in:
l3utterfly 2023-10-07 22:55:16 +08:00
parent 4c3a7cdd46
commit cd0022f70f
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
ggml-alloc.c
View file

@ -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).hash_table, 0, sizeof((*alloc).hash_table));
memset((*alloc).parse_seq, 0, sizeof((*alloc).parse_seq)); 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); ggml_allocr_reset(alloc);
return 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)); memset((*alloc).parse_seq, 0, sizeof((*alloc).parse_seq));
#ifdef GGML_ALLOCATOR_DEBUG #ifdef GGML_ALLOCATOR_DEBUG
(*alloc).allocated_tensors = {0}; memset((*alloc).allocated_tensors, 0, sizeof((*alloc).allocated_tensors));
#endif #endif
ggml_allocr_reset(alloc); ggml_allocr_reset(alloc);