Adding upstream version 86.0.1.upstream/86.0.1upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 36 insertions, 0 deletions

diff --git a/third_party/rust/gpu-alloc/src/heap.rs b/third_party/rust/gpu-alloc/src/heap.rs
new file mode 100644
index 0000000000..16e14c8ea2
--- /dev/null
+++ b/third_party/rust/gpu-alloc/src/heap.rs
@@ -0,0 +1,36 @@
+#[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 budget(&mut self) -> u64 {
+        self.size - self.used
+    }
+
+    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);
+    }
+}