hpt-allocator 0.1.2

An internal library for memory allocator for hpt
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

use std::collections::{HashMap, HashSet};

use lru::LruCache;

use crate::{
    ptr::SafePtr,
    storage::{CommonStorage, Storage},
};

pub(crate) fn deallocate_helper(
    cache: &mut LruCache<std::alloc::Layout, Vec<SafePtr>>,
    allocated: &mut HashSet<SafePtr>,
    storage: &mut HashMap<usize, CommonStorage>,
    layout: &std::alloc::Layout,
    ptr: *mut u8,
    should_drop: bool,
    device_id: usize,
) {
    if let Some(storage) = storage.get_mut(&device_id) {
        if storage.decrement_ref(SafePtr { ptr }) && should_drop {
            allocated.remove(&SafePtr { ptr });
            if let Some(ptrs) = cache.get_mut(layout) {
                ptrs.push(SafePtr { ptr });
            } else {
                cache.put(layout.clone(), vec![SafePtr { ptr }]);
            }
        }
    } else {
        panic!("device {} not found in storage", device_id);
    }
}