xref: /aosp_15_r20/external/llvm/test/Bindings/OCaml/vectorize.ml (revision 9880d6810fe72a1726cb53787c6711e909410d58)
1*9880d681SAndroid Build Coastguard Worker(* RUN: cp %s %T/vectorize_opts.ml
2*9880d681SAndroid Build Coastguard Worker * RUN: %ocamlc -g -w +A -package llvm.vectorize -linkpkg %T/vectorize_opts.ml -o %t
3*9880d681SAndroid Build Coastguard Worker * RUN: %t %t.bc
4*9880d681SAndroid Build Coastguard Worker * RUN: %ocamlopt -g -w +A -package llvm.vectorize -linkpkg %T/vectorize_opts.ml -o %t
5*9880d681SAndroid Build Coastguard Worker * RUN: %t %t.bc
6*9880d681SAndroid Build Coastguard Worker * XFAIL: vg_leak
7*9880d681SAndroid Build Coastguard Worker *)
8*9880d681SAndroid Build Coastguard Worker
9*9880d681SAndroid Build Coastguard Worker(* Note: It takes several seconds for ocamlopt to link an executable with
10*9880d681SAndroid Build Coastguard Worker         libLLVMCore.a, so it's better to write a big test than a bunch of
11*9880d681SAndroid Build Coastguard Worker         little ones. *)
12*9880d681SAndroid Build Coastguard Worker
13*9880d681SAndroid Build Coastguard Workeropen Llvm
14*9880d681SAndroid Build Coastguard Workeropen Llvm_vectorize
15*9880d681SAndroid Build Coastguard Workeropen Llvm_target
16*9880d681SAndroid Build Coastguard Worker
17*9880d681SAndroid Build Coastguard Workerlet context = global_context ()
18*9880d681SAndroid Build Coastguard Workerlet void_type = Llvm.void_type context
19*9880d681SAndroid Build Coastguard Worker
20*9880d681SAndroid Build Coastguard Worker(* Tiny unit test framework - really just to help find which line is busted *)
21*9880d681SAndroid Build Coastguard Workerlet print_checkpoints = false
22*9880d681SAndroid Build Coastguard Worker
23*9880d681SAndroid Build Coastguard Workerlet suite name f =
24*9880d681SAndroid Build Coastguard Worker  if print_checkpoints then
25*9880d681SAndroid Build Coastguard Worker    prerr_endline (name ^ ":");
26*9880d681SAndroid Build Coastguard Worker  f ()
27*9880d681SAndroid Build Coastguard Worker
28*9880d681SAndroid Build Coastguard Worker
29*9880d681SAndroid Build Coastguard Worker(*===-- Fixture -----------------------------------------------------------===*)
30*9880d681SAndroid Build Coastguard Worker
31*9880d681SAndroid Build Coastguard Workerlet filename = Sys.argv.(1)
32*9880d681SAndroid Build Coastguard Workerlet m = create_module context filename
33*9880d681SAndroid Build Coastguard Worker
34*9880d681SAndroid Build Coastguard Worker
35*9880d681SAndroid Build Coastguard Worker(*===-- Transforms --------------------------------------------------------===*)
36*9880d681SAndroid Build Coastguard Worker
37*9880d681SAndroid Build Coastguard Workerlet test_transforms () =
38*9880d681SAndroid Build Coastguard Worker  let (++) x f = f x; x in
39*9880d681SAndroid Build Coastguard Worker
40*9880d681SAndroid Build Coastguard Worker  let fty = function_type void_type [| |] in
41*9880d681SAndroid Build Coastguard Worker  let fn = define_function "fn" fty m in
42*9880d681SAndroid Build Coastguard Worker  ignore (build_ret_void (builder_at_end context (entry_block fn)));
43*9880d681SAndroid Build Coastguard Worker
44*9880d681SAndroid Build Coastguard Worker  ignore (PassManager.create ()
45*9880d681SAndroid Build Coastguard Worker           ++ add_bb_vectorize
46*9880d681SAndroid Build Coastguard Worker           ++ add_loop_vectorize
47*9880d681SAndroid Build Coastguard Worker           ++ add_slp_vectorize
48*9880d681SAndroid Build Coastguard Worker           ++ PassManager.run_module m
49*9880d681SAndroid Build Coastguard Worker           ++ PassManager.dispose)
50*9880d681SAndroid Build Coastguard Worker
51*9880d681SAndroid Build Coastguard Worker
52*9880d681SAndroid Build Coastguard Worker(*===-- Driver ------------------------------------------------------------===*)
53*9880d681SAndroid Build Coastguard Worker
54*9880d681SAndroid Build Coastguard Workerlet _ =
55*9880d681SAndroid Build Coastguard Worker  suite "transforms" test_transforms;
56*9880d681SAndroid Build Coastguard Worker  dispose_module m
57