Propagate types for function arguments.

tdenniston · Apr 3, 2015 · def7d0f · def7d0f
1 parent fb755a3
commit def7d0f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/TypeChecker.cpp b/src/TypeChecker.cpp
@@ -56,6 +56,8 @@ void TypeChecker::visit(FunctionCall *node) {
         if (node->function->args[i]->type().defined()) {
             bish_assert((*I)->type() == node->function->args[i]->type()) <<
                 "Invalid argument type for function call " << node->debug_info();
+        } else {
+            node->function->args[i]->set_type((*I)->type());
         }
     }
     node->set_type(node->function->type());

diff --git a/tools/TypeAnnotator.cpp b/tools/TypeAnnotator.cpp
@@ -287,7 +287,7 @@ int main(int argc, char **argv) {
     CodeGenerators::CodeGeneratorConstructor cg_constructor =
         CodeGenerators::get("bash");
     assert(cg_constructor);
-    compile_to(m, cg_constructor(s));
+    compile(m, cg_constructor(s));
 
     TypeAnnotator annotate(std::cout);
     m->accept(&annotate);