From 54b75fb9b163de61e639c514ba7b26119421fdef Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jon=C3=A1=C5=A1=20Fiala?= <jonas.fiala@inf.ethz.ch>
Date: Mon, 27 Nov 2023 14:05:53 +0100
Subject: [PATCH] Update checker.rs

---
 mir-state-analysis/src/coupling_graph/check/checker.rs | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/mir-state-analysis/src/coupling_graph/check/checker.rs b/mir-state-analysis/src/coupling_graph/check/checker.rs
index 39b7ea73309..73b66f621dc 100644
--- a/mir-state-analysis/src/coupling_graph/check/checker.rs
+++ b/mir-state-analysis/src/coupling_graph/check/checker.rs
@@ -107,7 +107,7 @@ pub(crate) fn check<'tcx>(
 
         // Repacks
         for op in fpcs_after.repacks_middle {
-            op.update_free(&mut fpcs.summary, false, rp);
+            op.update_free(&mut fpcs.summary, false, true, rp);
         }
         // Couplings bound set
         let bound: Box<dyn Fn(Place<'tcx>) -> CapabilityKind> =
@@ -121,7 +121,7 @@ pub(crate) fn check<'tcx>(
         fpcs.visit_terminator(data.terminator(), loc);
         // Repacks
         for op in fpcs_after.repacks {
-            op.update_free(&mut fpcs.summary, false, rp);
+            op.update_free(&mut fpcs.summary, false, true, rp);
         }
         // Statement post
         assert!(fpcs.repackings.is_empty());
@@ -153,6 +153,7 @@ pub(crate) fn check<'tcx>(
                 op.update_free(
                     &mut fpcs_from.summary,
                     body.basic_blocks[fpcs_succ.location.block].is_cleanup,
+                    false,
                     rp,
                 );
             }
@@ -213,7 +214,7 @@ impl<'a, 'tcx> CouplingState<'a, 'tcx> {
 
         // Repacks
         for op in fpcs_after.repacks_middle {
-            op.update_free(&mut fpcs.summary, false, rp);
+            op.update_free(&mut fpcs.summary, false, true, rp);
         }
         // Couplings bound set
         let bound: Box<dyn Fn(Place<'tcx>) -> CapabilityKind> =
@@ -228,7 +229,7 @@ impl<'a, 'tcx> CouplingState<'a, 'tcx> {
         assert!(fpcs.repackings.is_empty());
         // Repacks
         for op in fpcs_after.repacks {
-            op.update_free(&mut fpcs.summary, false, rp);
+            op.update_free(&mut fpcs.summary, false, true, rp);
         }
         // Statement post
         assert!(fpcs.repackings.is_empty());