From 70fd6acc590e3b57195e4ee20296faa75e99549c Mon Sep 17 00:00:00 2001
From: KA7E <kthurmer@gmail.com>
Date: Tue, 4 Jun 2024 14:29:57 -0400
Subject: [PATCH] unset atom clb lookup entry if it has been removed from the
 cluster

---
 vpr/src/pack/re_cluster_util.cpp | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/vpr/src/pack/re_cluster_util.cpp b/vpr/src/pack/re_cluster_util.cpp
index a119408a14f..e194cf2fca9 100644
--- a/vpr/src/pack/re_cluster_util.cpp
+++ b/vpr/src/pack/re_cluster_util.cpp
@@ -721,7 +721,11 @@ static void update_cluster_pb_stats(const t_pack_molecule* molecule,
         }
 
         //Update atom netlist mapping
-        atom_ctx.lookup.set_atom_clb(blk_id, clb_index);
+        if (is_added) {
+            atom_ctx.lookup.set_atom_clb(blk_id, clb_index);
+        } else {
+            atom_ctx.lookup.set_atom_clb(blk_id, ClusterBlockId::INVALID());
+        }
 
         const t_pb* atom_pb = atom_ctx.lookup.atom_pb(blk_id);
         VTR_ASSERT(atom_pb);