ggml-alloc : fix list of allocated tensors with GGML_ALLOCATOR_DEBUG (#9573)

This commit is contained in:
slaren 2024-09-21 14:24:23 +02:00 committed by GitHub
parent 41f477879f
commit d09770cae7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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) {