56 lines
1.6 KiB
OCaml
56 lines
1.6 KiB
OCaml
|
(* RUN: rm -rf %t && mkdir -p %t && cp %s %t/vectorize_opts.ml
|
||
|
|
* RUN: %ocamlc -g -w +A -package llvm.vectorize -linkpkg %t/vectorize_opts.ml -o %t/executable
|
||
|
|
* RUN: %t/executable %t/bitcode.bc
|
||
|
|
* RUN: %ocamlopt -g -w +A -package llvm.vectorize -linkpkg %t/vectorize_opts.ml -o %t/executable
|
||
|
|
* RUN: %t/executable %t/bitcode.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_vectorize
|
||
|
|
open Llvm_target
|
||
|
|
|
||
|
|
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
|
||
|
|
|
||
|
|
|
||
|
|
(*===-- Transforms --------------------------------------------------------===*)
|
||
|
|
|
||
|
|
let test_transforms () =
|
||
|
|
let (++) x f = f x; x in
|
||
|
|
|
||
|
|
let fty = function_type void_type [| |] in
|
||
|
|
let fn = define_function "fn" fty m in
|
||
|
|
ignore (build_ret_void (builder_at_end context (entry_block fn)));
|
||
|
|
|
||
|
|
ignore (PassManager.create ()
|
||
|
|
++ add_loop_vectorize
|
||
|
|
++ add_slp_vectorize
|
||
|
|
++ PassManager.run_module m
|
||
|
|
++ PassManager.dispose)
|
||
|
|
|
||
|
|
|
||
|
|
(*===-- Driver ------------------------------------------------------------===*)
|
||
|
|
|
||
|
|
let _ =
|
||
|
|
suite "transforms" test_transforms;
|
||
|
|
dispose_module m
|