Home | History | Annotate | Download | only in OCaml 
      1 (* RUN: cp %s %T/ipo_opts.ml
      2  * RUN: %ocamlc -g -w +A -package llvm.ipo -linkpkg %T/ipo_opts.ml -o %t
      3  * RUN: %t %t.bc
      4  * RUN: %ocamlopt -g -w +A -package llvm.ipo -linkpkg %T/ipo_opts.ml -o %t
      5  * RUN: %t %t.bc
      6  * XFAIL: vg_leak
      7  *)
      8 
      9 (* Note: It takes several seconds for ocamlopt to link an executable with
     10          libLLVMCore.a, so it's better to write a big test than a bunch of
     11          little ones. *)
     12 
     13 open Llvm
     14 open Llvm_ipo
     15 open Llvm_target
     16 
     17 let context = global_context ()
     18 let void_type = Llvm.void_type context
     19 let i8_type = Llvm.i8_type context
     20 
     21 (* Tiny unit test framework - really just to help find which line is busted *)
     22 let print_checkpoints = false
     23 
     24 let suite name f =
     25   if print_checkpoints then
     26     prerr_endline (name ^ ":");
     27   f ()
     28 
     29 
     30 (*===-- Fixture -----------------------------------------------------------===*)
     31 
     32 let filename = Sys.argv.(1)
     33 let m = create_module context filename
     34 
     35 
     36 (*===-- Transforms --------------------------------------------------------===*)
     37 
     38 let test_transforms () =
     39   let (++) x f = f x; x in
     40 
     41   let fty = function_type i8_type [| |] in
     42   let fn = define_function "fn" fty m in
     43   let fn2 = define_function "fn2" fty m in begin
     44       ignore (build_ret (const_int i8_type 4) (builder_at_end context (entry_block fn)));
     45       let b = builder_at_end context  (entry_block fn2) in
     46       ignore (build_ret (build_call fn [| |] "" b) b);
     47   end;
     48 
     49   ignore (PassManager.create ()
     50            ++ add_argument_promotion
     51            ++ add_constant_merge
     52            ++ add_dead_arg_elimination
     53            ++ add_function_attrs
     54            ++ add_function_inlining
     55            ++ add_always_inliner
     56            ++ add_global_dce
     57            ++ add_global_optimizer
     58            ++ add_ipc_propagation
     59            ++ add_prune_eh
     60            ++ add_ipsccp
     61            ++ add_internalize ~all_but_main:true
     62            ++ add_strip_dead_prototypes
     63            ++ add_strip_symbols
     64            ++ PassManager.run_module m
     65            ++ PassManager.dispose)
     66 
     67 
     68 (*===-- Driver ------------------------------------------------------------===*)
     69 
     70 let _ =
     71   suite "transforms" test_transforms;
     72   dispose_module m
     73