kiper220/mimalloc
1
0
Fork 0
mirror of https://github.com/microsoft/mimalloc.git synced 2025-05-08 00:09:31 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Ensure large and medium segments are droped when allocating a new

Browse source 
segment for medium and large objects
This commit is contained in:
Sergiy Kuryata 2024-12-17 13:17:28 -08:00
parent f01f882dd6
commit c8faa8b0fb
1 changed files with 25 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

49
src/heap.c
View file

@ -656,7 +656,7 @@ bool mi_heap_visit_blocks(const mi_heap_t* heap, bool visit_blocks, mi_block_vis
mi_segment_t* mi_segments_get_segment_to_drop_by_slice(mi_segments_tld_t* tld, size_t alloc_block_size); mi_segment_t* mi_segments_get_segment_to_drop_by_slice(mi_segments_tld_t* tld, size_t alloc_block_size);
const mi_slice_t* mi_segment_slices_end(const mi_segment_t* segment); const mi_slice_t* mi_segment_slices_end(const mi_segment_t* segment);
static mi_segment_t* mi_heap_get_segment_to_drop(mi_heap_t* heap, size_t alloc_block_size) { static mi_segment_t* mi_heap_get_segment_to_drop(mi_heap_t* heap, size_t targetSegmentCount, size_t alloc_block_size) {
mi_segment_t* segment = NULL; mi_segment_t* segment = NULL;
mi_page_kind_t page_kind = mi_page_kind_from_size(alloc_block_size); mi_page_kind_t page_kind = mi_page_kind_from_size(alloc_block_size);
@ -667,26 +667,28 @@ static mi_segment_t* mi_heap_get_segment_to_drop(mi_heap_t* heap, size_t alloc_b
return heap->tld->segments.medium_segment; return heap->tld->segments.medium_segment;
} }
int i = 0; if (heap->tld->segments.count >= targetSegmentCount) {
mi_page_queue_t* fullPageQueue = &heap->pages[MI_BIN_FULL]; int i = 0;
for (mi_page_t* page = fullPageQueue->first; page != NULL; page = page->next) { mi_page_queue_t* fullPageQueue = &heap->pages[MI_BIN_FULL];
mi_segment_t* temp_segment = _mi_ptr_segment(page); for (mi_page_t* page = fullPageQueue->first; page != NULL; page = page->next) {
if (temp_segment->page_kind == MI_PAGE_SMALL) { mi_segment_t* temp_segment = _mi_ptr_segment(page);
if (segment == NULL) { if (temp_segment->page_kind == MI_PAGE_SMALL) {
segment = temp_segment; if (segment == NULL) {
segment = temp_segment;
}
else if (temp_segment->used > segment->used) {
segment = temp_segment;
}
if (i > 3) {
break;
}
i++;
} }
else if (temp_segment->used > segment->used) {
segment = temp_segment;
}
if (i > 3) {
break;
}
i++;
} }
}
if (segment == NULL) { if (segment == NULL) {
segment = mi_segments_get_segment_to_drop_by_slice(&heap->tld->segments, alloc_block_size); segment = mi_segments_get_segment_to_drop_by_slice(&heap->tld->segments, alloc_block_size);
}
} }
return segment; return segment;
@ -730,10 +732,9 @@ static mi_decl_noinline void mi_segment_visit_pages(mi_heap_t* heap, mi_segment_
void mi_heap_drop_segment(mi_heap_t* heap, size_t targetSegmentCount, size_t alloc_block_size) { void mi_heap_drop_segment(mi_heap_t* heap, size_t targetSegmentCount, size_t alloc_block_size) {
bool segmentsDropped = false; bool segmentsDropped = false;
while (heap->tld->segments.count >= targetSegmentCount) { do {
// 1. Find a segment to drop (abandon) using the Full Page queue // 1. Find a segment to drop (abandon) using the Full Page queue
mi_segment_t* segmentToAbandon = mi_heap_get_segment_to_drop(heap, alloc_block_size); mi_segment_t* segmentToAbandon = mi_heap_get_segment_to_drop(heap, targetSegmentCount, alloc_block_size);
if (segmentToAbandon == NULL) { if (segmentToAbandon == NULL) {
break; break;
} }
@ -750,7 +751,7 @@ void mi_heap_drop_segment(mi_heap_t* heap, size_t targetSegmentCount, size_t all
// This will effectively abandon the segment. // This will effectively abandon the segment.
mi_collect_t collect = MI_ABANDON; mi_collect_t collect = MI_ABANDON;
mi_segment_visit_pages(heap, segmentToAbandon, &mi_heap_page_collect, &collect); mi_segment_visit_pages(heap, segmentToAbandon, &mi_heap_page_collect, &collect);
} } while (heap->tld->segments.count >= targetSegmentCount);
if (segmentsDropped) { if (segmentsDropped) {
mi_msecs_t now_msec = _mi_clock_now(); mi_msecs_t now_msec = _mi_clock_now();
@ -769,8 +770,8 @@ void mi_heap_drop_segment_if_required(mi_heap_t* heap, size_t alloc_block_size)
{ {
size_t targetSegmentCount = mi_option_get_size(mi_option_max_segments_per_heap); size_t targetSegmentCount = mi_option_get_size(mi_option_max_segments_per_heap);
if ((targetSegmentCount > 0) && if ((targetSegmentCount > 0) &&
(alloc_block_size <= MI_LARGE_OBJ_SIZE_MAX) && ((alloc_block_size > MI_SMALL_OBJ_SIZE_MAX) ||
(heap->tld->segments.count >= targetSegmentCount)) { (heap->tld->segments.count >= targetSegmentCount))) {
mi_heap_drop_segment(heap, targetSegmentCount, alloc_block_size); mi_heap_drop_segment(heap, targetSegmentCount, alloc_block_size);
} }