boots: track free registers in InstTracker

dinfuehr · Aug 30, 2023 · 96396fc · 96396fc
1 parent ace0e66
commit 96396fc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/dora-boots/assembler.dora b/dora-boots/assembler.dora
@@ -290,4 +290,10 @@ impl RegSet {
     pub fn equals(other: RegSet): Bool {
         self.value == other.value
     }
+
+    pub fn clone(): RegSet {
+        RegSet(
+            self.value,
+        )
+    }
 }
diff --git a/dora-boots/regalloc.dora b/dora-boots/regalloc.dora
@@ -52,7 +52,6 @@ impl SimpleRegisterAllocator {
                 // pick a register
                 let register = self.tracker.allocateRegister(inst).getOrPanic();
                 loc.getOutput().setRegister(register);
-
             }
 
             for inst in block.instructionsIterator() {
@@ -132,6 +131,7 @@ fn ssaDestruction(graph: Graph, codegen: CodeGen) {
 
 pub class InstTracker {
     allocatable: RegSet,
+    free: RegSet,
     data: RegMap[Inst],
     locations: HashMap[Inst, Register],
 }
@@ -140,6 +140,7 @@ impl InstTracker {
     static fn new(allocatable: RegSet): InstTracker {
         InstTracker(
             allocatable,
+            allocatable.clone(),
             RegMap[Inst]::new(allocatable),
             HashMap[Inst, Register]::new(),
         )
@@ -186,6 +187,7 @@ impl InstTracker {
     fn clone(): InstTracker {
         InstTracker(
             self.allocatable,
+            self.free.clone(),
             self.data.clone(),
             self.locations.clone(),
         )