test/Bindings/Ocaml/vectorize_opts.ml

   1 (* RUN: rm -rf %t.builddir
   2  * RUN: mkdir -p %t.builddir
   3  * RUN: cp %s %t.builddir
   4  * RUN: %ocamlopt -warn-error A llvm.cmxa llvm_vectorize.cmxa llvm_target.cmxa %t.builddir/vectorize_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_vectorize
  15 open Llvm_target
  16
  17 let context = global_context ()
  18 let void_type = Llvm.void_type context
  19
  20 (* Tiny unit test framework - really just to help find which line is busted *)
  21 let print_checkpoints = false
  22
  23 let suite name f =
  24   if print_checkpoints then
  25     prerr_endline (name ^ ":");
  26   f ()
  27
  28
  29 (*===-- Fixture -----------------------------------------------------------===*)
  30
  31 let filename = Sys.argv.(1)
  32 let m = create_module context filename
  33
  34
  35 (*===-- Transforms --------------------------------------------------------===*)
  36
  37 let test_transforms () =
  38   let (++) x f = ignore (f x); x in
  39
  40   let fty = function_type void_type [| |] in
  41   let fn = define_function "fn" fty m in
  42   ignore (build_ret_void (builder_at_end context (entry_block fn)));
  43
  44   let td = DataLayout.create (target_triple m) in
  45
  46   ignore (PassManager.create ()
  47            ++ DataLayout.add td
  48            ++ add_bb_vectorize
  49            ++ add_loop_vectorize
  50            ++ add_slp_vectorize
  51            ++ PassManager.run_module m
  52            ++ PassManager.dispose);
  53
  54   DataLayout.dispose td
  55
  56
  57 (*===-- Driver ------------------------------------------------------------===*)
  58
  59 let _ =
  60   suite "transforms" test_transforms;
  61   dispose_module m