Checkpoint

2023-10-14 16:39:16 -07:00
parent 87d027bf8d
commit 71228b9e09
18 changed files with 402 additions and 78 deletions
--- a/src/ir/strings.rs
+++ b/src/ir/strings.rs
@@ -21,7 +21,12 @@ impl Program {
 impl TopLevel {
    fn register_strings(&self, string_set: &mut HashSet<ArcIntern<String>>) {
        match self {
-            TopLevel::Function(_, _, body) => body.register_strings(string_set),
+            TopLevel::Function(_, _, stmts, body) => {
+                for stmt in stmts.iter() {
+                    stmt.register_strings(string_set);
+                }
+                body.register_strings(string_set);
+            }
            TopLevel::Statement(stmt) => stmt.register_strings(string_set),
        }
    }