cc: fix bug with regbank snap

2021-07-25 23:58:17 -07:00 · 2021-07-25 23:58:17 -07:00 · b01d6a4873
commit b01d6a4873
parent 8dfa5addfd
1 changed files with 3 additions and 1 deletions
--- a/tools/cc.py
+++ b/tools/cc.py
@ -219,7 +219,9 @@ class RegBank:
            self.loaded(var1, reg0)

    def snap(self):
-        return dict(self.vars), dict(self.cleanup)
+        cleanup = collections.defaultdict(list)
+        cleanup.update({r: list(c) for r, c in self.cleanup.items()})
+        return dict(self.vars), cleanup

    def restore(self, snap):
        self.reset()