third_party/rust/gpu-alloc/src/heap.rs


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
32

#[derive(Debug)]
pub(crate) struct Heap {
    size: u64,
    used: u64,
    allocated: u128,
    deallocated: u128,
}

impl Heap {
    pub(crate) fn new(size: u64) -> Self {
        Heap {
            size,
            used: 0,
            allocated: 0,
            deallocated: 0,
        }
    }

    pub(crate) fn size(&mut self) -> u64 {
        self.size
    }

    pub(crate) fn alloc(&mut self, size: u64) {
        self.used += size;
        self.allocated += u128::from(size);
    }

    pub(crate) fn dealloc(&mut self, size: u64) {
        self.used -= size;
        self.deallocated += u128::from(size);
    }
}