aboutsummaryrefslogtreecommitdiffstats
path: root/test/Bindings/OCaml/passmgr_builder.ml
blob: 5dd9d4e00e89bb953f04a5520d4aa59215d96357 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
(* RUN: cp %s %T/passmgr_builder.ml
 * RUN: %ocamlc -g -warn-error A -package llvm.passmgr_builder -linkpkg %T/passmgr_builder.ml -o %t
 * RUN: %t %t.bc
 * RUN: %ocamlopt -g -warn-error A -package llvm.passmgr_builder -linkpkg %T/passmgr_builder.ml -o %t
 * RUN: %t %t.bc
 * XFAIL: vg_leak
 *)

(* Note: It takes several seconds for ocamlopt to link an executable with
         libLLVMCore.a, so it's better to write a big test than a bunch of
         little ones. *)

open Llvm
open Llvm_passmgr_builder

let context = global_context ()
let void_type = Llvm.void_type context

(* Tiny unit test framework - really just to help find which line is busted *)
let print_checkpoints = false

let suite name f =
  if print_checkpoints then
    prerr_endline (name ^ ":");
  f ()


(*===-- Fixture -----------------------------------------------------------===*)

let filename = Sys.argv.(1)
let m = create_module context filename


(*===-- Pass Manager Builder ----------------------------------------------===*)

let test_pmbuilder () =
  let (++) x f = ignore (f x); x in

  let module_passmgr = PassManager.create () in
  let func_passmgr   = PassManager.create_function m in
  let lto_passmgr    = PassManager.create () in

  ignore (Llvm_passmgr_builder.create ()
           ++ set_opt_level 3
           ++ set_size_level 1
           ++ set_disable_unit_at_a_time false
           ++ set_disable_unroll_loops false
           ++ use_inliner_with_threshold 10
           ++ populate_function_pass_manager func_passmgr
           ++ populate_module_pass_manager module_passmgr
           ++ populate_lto_pass_manager lto_passmgr
                  ~internalize:false ~run_inliner:false);
  Gc.compact ();

  PassManager.dispose module_passmgr;
  PassManager.dispose func_passmgr;
  PassManager.dispose lto_passmgr


(*===-- Driver ------------------------------------------------------------===*)

let _ =
  suite "pass manager builder" test_pmbuilder;
  dispose_module m