From f0b3102e9a31f003934883f8a41641f674f041a1 Mon Sep 17 00:00:00 2001
From: Julian P Samaroo <jpsamaroo@jpsamaroo.me>
Date: Thu, 16 Nov 2023 14:24:16 -0700
Subject: [PATCH] SimpleRecencyAllocator: Decr ctrs in delete_from_device

---
 src/storage.jl | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/storage.jl b/src/storage.jl
index 9dfe6ee..8531389 100644
--- a/src/storage.jl
+++ b/src/storage.jl
@@ -1013,12 +1013,14 @@ function delete_from_device!(sra::SimpleRecencyAllocator, state::RefState, id::I
         if (idx = findfirst(x->x==id, sra.mem_refs)) !== nothing
             delete_from_device!(CPURAMDevice(), state, id)
             deleteat!(sra.mem_refs, idx)
+            sra.mem_size[] -= state.size
         end
         if (idx = findfirst(x->x==id, sra.device_refs)) !== nothing
             if !sra.retain[]
                 delete_from_device!(sra.device, state, id)
             end
             deleteat!(sra.device_refs, idx)
+            sra.device_size[] -= state.size
         end
         delete!(sra.ref_cache, id)
     end