fix clearblocks

UQ-PAC · Jan 23, 2024 · 1f78b30 · 1f78b30
1 parent 4158d49
commit 1f78b30
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/main/scala/ir/Program.scala b/src/main/scala/ir/Program.scala
@@ -224,7 +224,7 @@ class Procedure private (
    * @return an iterator to the new block set
    */
   def replaceBlocks(newBlocks: Iterable[Block]): Unit = {
-    removeBlocks(_blocks)
+    clearBlocks()
     addBlocks(newBlocks)
   }
 
@@ -245,7 +245,8 @@ class Procedure private (
   }
 
   def clearBlocks() : Unit = {
-    removeBlocks(blocks)
+    // O(n) because we are careful to unlink the parents etc.
+    removeBlocks(_blocks)
   }