Home | History | Annotate | Download | only in Ocaml
      1 (* RUN: rm -rf %t.builddir
      2  * RUN: mkdir -p %t.builddir
      3  * RUN: cp %s %t.builddir
      4  * RUN: %ocamlopt -g -warn-error A llvm.cmxa llvm_target.cmxa llvm_executionengine.cmxa %t.builddir/target.ml -o %t
      5  * RUN: %t %t.bc
      6  * REQUIRES: native, object-emission
      7  * XFAIL: vg_leak
      8  *)
      9 
     10 (* Note: It takes several seconds for ocamlopt to link an executable with
     11          libLLVMCore.a, so it's better to write a big test than a bunch of
     12          little ones. *)
     13 
     14 open Llvm
     15 open Llvm_target
     16 
     17 let _ = Llvm_executionengine.initialize_native_target ()
     18 
     19 let context = global_context ()
     20 let i32_type = Llvm.i32_type context
     21 let i64_type = Llvm.i64_type context
     22 
     23 (* Tiny unit test framework - really just to help find which line is busted *)
     24 let print_checkpoints = false
     25 
     26 let _ =
     27   Printexc.record_backtrace true
     28 
     29 let assert_equal a b =
     30   if a <> b then failwith "assert_equal"
     31 
     32 
     33 (*===-- Fixture -----------------------------------------------------------===*)
     34 
     35 let filename = Sys.argv.(1)
     36 let m = create_module context filename
     37 
     38 let target = Target.by_triple (Target.default_triple ())
     39 
     40 let machine = TargetMachine.create (Target.default_triple ()) target
     41 
     42 (*===-- Data Layout -------------------------------------------------------===*)
     43 
     44 let test_target_data () =
     45   let module DL = DataLayout in
     46   let layout = "e-p:32:32-f64:32:64-v64:32:64-v128:32:128-n32-S32" in
     47   let dl     = DL.of_string layout in
     48   let sty    = struct_type context [| i32_type; i64_type |] in
     49 
     50   assert_equal (DL.as_string dl) layout;
     51   assert_equal (DL.byte_order dl) Endian.Little;
     52   assert_equal (DL.pointer_size dl) 4;
     53   assert_equal (DL.intptr_type context dl) i32_type;
     54   assert_equal (DL.qualified_pointer_size 0 dl) 4;
     55   assert_equal (DL.qualified_intptr_type context 0 dl) i32_type;
     56   assert_equal (DL.size_in_bits sty dl) (Int64.of_int 96);
     57   assert_equal (DL.store_size sty dl) (Int64.of_int 12);
     58   assert_equal (DL.abi_size sty dl) (Int64.of_int 12);
     59   assert_equal (DL.stack_align sty dl) 4;
     60   assert_equal (DL.preferred_align sty dl) 8;
     61   assert_equal (DL.preferred_align_of_global (declare_global sty "g" m) dl) 8;
     62   assert_equal (DL.element_at_offset sty (Int64.of_int 1) dl) 0;
     63   assert_equal (DL.offset_of_element sty 1 dl) (Int64.of_int 4);
     64 
     65   let pm = PassManager.create () in
     66   ignore (DL.add_to_pass_manager pm dl)
     67 
     68 
     69 (*===-- Target ------------------------------------------------------------===*)
     70 
     71 let test_target () =
     72   let module T = Target in
     73   ignore (T.succ target);
     74   ignore (T.name target);
     75   ignore (T.description target);
     76   ignore (T.has_jit target);
     77   ignore (T.has_target_machine target);
     78   ignore (T.has_asm_backend target)
     79 
     80 
     81 (*===-- Target Machine ----------------------------------------------------===*)
     82 
     83 let test_target_machine () =
     84   let module TM = TargetMachine in
     85   assert_equal (TM.target machine) target;
     86   assert_equal (TM.triple machine) (Target.default_triple ());
     87   assert_equal (TM.cpu machine) "";
     88   assert_equal (TM.features machine) "";
     89   ignore (TM.data_layout machine);
     90   TM.set_verbose_asm true machine
     91 
     92 
     93 (*===-- Code Emission -----------------------------------------------------===*)
     94 
     95 let test_code_emission () =
     96   TargetMachine.emit_to_file m CodeGenFileType.ObjectFile filename machine;
     97   try
     98     TargetMachine.emit_to_file m CodeGenFileType.ObjectFile
     99                                "/nonexistent/file" machine;
    100     failwith "must raise"
    101   with Llvm_target.Error _ ->
    102     ();
    103 
    104   let buf = TargetMachine.emit_to_memory_buffer m CodeGenFileType.ObjectFile
    105                                                 machine in
    106   Llvm.MemoryBuffer.dispose buf
    107 
    108 
    109 (*===-- Driver ------------------------------------------------------------===*)
    110 
    111 let _ =
    112   test_target_data ();
    113   test_target ();
    114   test_target_machine ();
    115   (* test_code_emission (); *) (* broken without AsmParser support *)
    116   dispose_module m
    117