WIP: compute return_values

Author: hhugo

compiler/lib/driver.ml

@@ -63,7 +63,7 @@ let inline profile p =
 
 let specialize_1 (p, info) =
   if debug () then Format.eprintf "Specialize...@.";
-  let return_values = Code.Var.Map.empty in
+  let return_values = Code.return_values p in
   Specialize.f
     ~shape:(fun f -> Flow.the_shape_of ~return_values ~pure:Pure_fun.empty info f)
     ~update_def:(fun x expr -> Flow.Info.update_def info x expr)

compiler/tests-full/shapes.cma.expected.js

@@ -62,11 +62,6 @@
             ? f(a0)
             : runtime.caml_call_gen(f, [a0]);
    }
-   function caml_call2(f, a0, a1){
-    return (f.l >= 0 ? f.l : f.l = f.length) === 2
-            ? f(a0, a1)
-            : runtime.caml_call_gen(f, [a0, a1]);
-   }
    var
     global_data = runtime.caml_get_global_data(),
     Shapes_M2 = global_data.Shapes__M2,
@@ -83,7 +78,7 @@
              };
    }
    var
-    x =  /*<<compiler/tests-full/m3.ml:3:8>>*/ caml_call2(f(0), 0, 0),
+    x =  /*<<compiler/tests-full/m3.ml:3:8>>*/ f(0)(0, 0),
     Shapes_M3 =  /*<<compiler/tests-full/m3.ml:3:18>>*/ [0, f, x];
    runtime.caml_register_global(3, Shapes_M3, "Shapes__M3");
    return;