From 2729087efe1f3e75383eceee09273168da1b8809 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Tue, 14 Feb 2023 09:20:12 +0000 Subject: [PATCH] Fix a ref count bug in swapping --- cacheflow/master/block_manager.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/cacheflow/master/block_manager.py b/cacheflow/master/block_manager.py index 17d299c866ad4..415fe09aadffa 100644 --- a/cacheflow/master/block_manager.py +++ b/cacheflow/master/block_manager.py @@ -166,6 +166,7 @@ class BlockSpaceManager: for cpu_block in block_table: if cpu_block in mapping: gpu_block = mapping[cpu_block] + gpu_block.ref_count += 1 else: gpu_block = self.gpu_allocator.allocate() mapping[cpu_block] = gpu_block @@ -196,6 +197,7 @@ class BlockSpaceManager: for gpu_block in block_table: if gpu_block in mapping: cpu_block = mapping[gpu_block] + cpu_block.ref_count += 1 else: cpu_block = self.cpu_allocator.allocate() mapping[gpu_block] = cpu_block