Bench: add wasm support

Author: hhugo

benchmarks/common.ml

@@ -149,6 +149,8 @@ module Spec : sig
 
   val js_of_ocaml_js_string : t
 
+  val wasm_of_ocaml : t
+
   val byte_unsafe : t
 
   val opt_unsafe : t
@@ -214,6 +216,8 @@ end = struct
 
   let js_of_ocaml_js_string = create "jsstring" ".js"
 
+  let wasm_of_ocaml = create "wasm_of_ocaml" ".js"
+
   let byte_unsafe = create "unsafe/byte" ""
 
   let opt_unsafe = create "unsafe/opt" ""

benchmarks/run.ml

@@ -253,9 +253,19 @@ let _ =
          | `Cps -> "--effects=cps"
          | `Double_translation -> "--effects=double-translation"))
   in
+  let compile_wasmoo ?(effects = `None) opts =
+    compile
+      (Format.sprintf
+         "wasm_of_ocaml -q %s %s"
+         opts
+         (match effects with
+         | `None -> ""
+         | `Cps -> "--effects=cps"))
+  in
   Format.eprintf "Compile@.";
   compile "ocamlc" src Spec.ml code Spec.byte;
   compile "ocamlopt" src Spec.ml code Spec.opt;
+  compile_wasmoo "" code Spec.byte code Spec.wasm_of_ocaml;
   compile_jsoo "" code Spec.byte code Spec.js_of_ocaml;
   compile_jsoo "--opt=3" code Spec.byte code Spec.js_of_ocaml_o3;
   compile_jsoo "--enable=use-js-string" code Spec.byte code Spec.js_of_ocaml_js_string;
@@ -278,6 +288,7 @@ let _ =
   ml_size param src Spec.ml sizes Spec.ml;
   file_size param code Spec.byte sizes Spec.byte;
   file_size param code Spec.js_of_ocaml sizes (Spec.sub_spec Spec.js_of_ocaml "full");
+  file_size param code Spec.wasm_of_ocaml sizes (Spec.sub_spec Spec.wasm_of_ocaml "full");
   compr_file_size
     param
     code
@@ -296,6 +307,12 @@ let _ =
     Spec.js_of_ocaml_effects_double_translation
     sizes
     (Spec.sub_spec Spec.js_of_ocaml_effects_double_translation "gzipped");
+  compr_file_size
+    param
+    code
+    Spec.wasm_of_ocaml
+    sizes
+    (Spec.sub_spec Spec.wasm_of_ocaml "gzipped");
   bzip2_file_size
     param
     code
@@ -314,6 +331,12 @@ let _ =
     Spec.js_of_ocaml
     sizes
     (Spec.sub_spec Spec.js_of_ocaml "bzip2");
+  bzip2_file_size
+    param
+    code
+    Spec.wasm_of_ocaml
+    sizes
+    (Spec.sub_spec Spec.wasm_of_ocaml "bzip2");
   runtime_size
     param
     code
@@ -354,6 +377,7 @@ let _ =
         ; Some Spec.js_of_ocaml_call
         ; Some Spec.js_of_ocaml_effects_cps
         ; Some Spec.js_of_ocaml_effects_double_translation
+        ; Some Spec.wasm_of_ocaml
         ] )
     else if effects
     then
@@ -368,7 +392,7 @@ let _ =
       ( (match interpreters with
         | i :: _ -> [ i ]
         | [] -> [])
-      , [ Some Spec.js_of_ocaml ] )
+      , [ Some Spec.js_of_ocaml; Some Spec.wasm_of_ocaml ] )
   in
   List.iter compilers ~f:(fun (comp, dir) ->
       measure param src (Filename.concat times dir) Spec.js comp;

compiler/bin-wasm_of_ocaml/compile.ml

@@ -463,7 +463,9 @@ let run
          in
          if times () then Format.eprintf "  parsing: %a@." Timer.print t1;
          Fs.with_intermediate_file
-           (Filename.temp_file (Filename.chop_extension output_file) ".wat")
+           (Filename.temp_file
+              (Filename.basename (Filename.chop_extension output_file))
+              ".wat")
          @@ fun wat_file ->
          let dir = Filename.chop_extension output_file ^ ".assets" in
          Link.gen_dir dir