Chapter 3: Code generation to LLVM IR.

Note: had to replace build_add with build_fadd, build_mul with build_fmul, etc. to get their example to work because the base versions expect ints, but we convert all our inputs to doubles

Author: danluu

_tags

@@ -1 +1,3 @@
+"+llvm-2.8": include
 <{lexer,parser}.ml>: use_camlp4, pp(camlp4of)
+<*.{byte,native}>: g++, use_llvm, use_llvm_analysis

parser.ml

@@ -33,24 +33,24 @@ let rec parse_primary = parser
       | [< >] -> accumulator
     in
     let rec parse_ident id = parser
-      (* Call. *)
+        (* Call. *)
       | [< 'Token.Kwd '(';
-           args=parse_args [];
-           'Token.Kwd ')' ?? "expected ')'">] ->
-        Ast.Call (id, Array.of_list (List.rev args))
+             args=parse_args [];
+             'Token.Kwd ')' ?? "expected ')'">] ->
+            Ast.Call (id, Array.of_list (List.rev args))
 
-      (* Simple variable ref. *)
+        (* Simple variable ref. *)
       | [< >] -> Ast.Variable id
     in
-    parse_ident id stream
+      parse_ident id stream
 
   | [< >] -> raise (Stream.Error "unknown token when expecting an expression.")
 
 (* binoprhs
  *   ::= ('+' primary)* *)
 and parse_bin_rhs expr_prec lhs stream =
   match Stream.peek stream with
-		(* If this is a binop, find its precedence. *)
+  (* If this is a binop, find its precedence. *)
 		| Some (Token.Kwd c) when Hashtbl.mem binop_precedence c ->
       let token_prec = precedence c in
 
@@ -97,9 +97,9 @@ let parse_prototype =
 
   parser
 		| [< 'Token.Ident id;
-				 'Token.Kwd '(' ?? "expected '(' in prototype";
-				 args=parse_args [];
-				 'Token.Kwd ')' ?? "expected ')' in prototype" >] ->
+       'Token.Kwd '(' ?? "expected '(' in prototype";
+       args=parse_args [];
+       'Token.Kwd ')' ?? "expected ')' in prototype" >] ->
       (* success. *)
       Ast.Prototype (id, Array.of_list (List.rev args))
 
@@ -109,13 +109,13 @@ let parse_prototype =
 (* definition ::= 'def' prototype expression *)
 let parse_definition = parser
   | [< 'Token.Def; p=parse_prototype; e=parse_expr >] ->
-    Ast.Function (p, e)
+      Ast.Function (p, e)
 
 (* toplevelexpr ::= expression *)
 let parse_toplevel = parser
   | [< e=parse_expr >] ->
       (* Make an anonymous proto. *)
-    Ast.Function (Ast.Prototype ("", [||]), e)
+      Ast.Function (Ast.Prototype ("", [||]), e)
 
 (*  external ::= 'extern' prototype *)
 let parse_extern = parser

toplevel.ml

@@ -2,6 +2,8 @@
  * Top-Level parsing and JIT Driver
  *===----------------------------------------------------------------------===*)
 
+open Llvm
+
 (* top ::= definition | external | expression | ';' *)
 let rec main_loop stream =
   match Stream.peek stream with
@@ -16,16 +18,19 @@ let rec main_loop stream =
       begin
         try match token with
 					| Token.Def ->
-            ignore(Parser.parse_definition stream);
+            let e = Parser.parse_definition stream in
             print_endline "parsed a function definition.";
+            dump_value (Codegen.codegen_func e);
 					| Token.Extern ->
-            ignore(Parser.parse_extern stream);
+            let e = Parser.parse_extern stream in
             print_endline "parsed an extern.";
+            dump_value (Codegen.codegen_proto e);
 					| _ ->
             (* Evaluate a top-level expression into an anonymous function. *)
-            ignore(Parser.parse_toplevel stream);
+            let e = Parser.parse_toplevel stream in
             print_endline "parsed a top-level expr";
-        with Stream.Error s ->
+            dump_value (Codegen.codegen_func e);
+        with Stream.Error s | Codegen.Error s ->
           (* Skip token for error recovery. *)
           Stream.junk stream;
           print_endline s;

toy.ml

@@ -3,6 +3,8 @@
  * Main driver code.
  *===----------------------------------------------------------------------===*)
 
+open Llvm
+
 let main () =
   (* Install standard binary operators.
    * 1 is the lowest precedence. *)
@@ -17,6 +19,9 @@ let main () =
 
   (* Run the main "interpreter loop" now. *)
   Toplevel.main_loop stream;
+
+  (* Print out all the generated code. *)
+  dump_module Codegen.the_module
 ;;
 
 main ()