Make SetMCJITOptimizationLevel more of a method and pass options
[oota-llvm.git] / bindings / go / llvm / executionengine_test.go
1 //===- executionengine_test.go - Tests for executionengine ----------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file tests bindings for the executionengine component.
11 //
12 //===----------------------------------------------------------------------===//
13
14 package llvm
15
16 import (
17         "testing"
18 )
19
20 func TestFactorial(t *testing.T) {
21         LinkInMCJIT()
22         InitializeNativeTarget()
23         InitializeNativeAsmPrinter()
24
25         mod := NewModule("fac_module")
26
27         fac_args := []Type{Int32Type()}
28         fac_type := FunctionType(Int32Type(), fac_args, false)
29         fac := AddFunction(mod, "fac", fac_type)
30         fac.SetFunctionCallConv(CCallConv)
31         n := fac.Param(0)
32
33         entry := AddBasicBlock(fac, "entry")
34         iftrue := AddBasicBlock(fac, "iftrue")
35         iffalse := AddBasicBlock(fac, "iffalse")
36         end := AddBasicBlock(fac, "end")
37
38         builder := NewBuilder()
39         defer builder.Dispose()
40
41         builder.SetInsertPointAtEnd(entry)
42         If := builder.CreateICmp(IntEQ, n, ConstInt(Int32Type(), 0, false), "cmptmp")
43         builder.CreateCondBr(If, iftrue, iffalse)
44
45         builder.SetInsertPointAtEnd(iftrue)
46         res_iftrue := ConstInt(Int32Type(), 1, false)
47         builder.CreateBr(end)
48
49         builder.SetInsertPointAtEnd(iffalse)
50         n_minus := builder.CreateSub(n, ConstInt(Int32Type(), 1, false), "subtmp")
51         call_fac_args := []Value{n_minus}
52         call_fac := builder.CreateCall(fac, call_fac_args, "calltmp")
53         res_iffalse := builder.CreateMul(n, call_fac, "multmp")
54         builder.CreateBr(end)
55
56         builder.SetInsertPointAtEnd(end)
57         res := builder.CreatePHI(Int32Type(), "result")
58         phi_vals := []Value{res_iftrue, res_iffalse}
59         phi_blocks := []BasicBlock{iftrue, iffalse}
60         res.AddIncoming(phi_vals, phi_blocks)
61         builder.CreateRet(res)
62
63         err := VerifyModule(mod, ReturnStatusAction)
64         if err != nil {
65                 t.Errorf("Error verifying module: %s", err)
66                 return
67         }
68
69         options := NewMCJITCompilerOptions()
70         options.SetMCJITOptimizationLevel(2)
71         engine, err := NewMCJITCompiler(mod, options)
72         if err != nil {
73                 t.Errorf("Error creating JIT: %s", err)
74                 return
75         }
76         defer engine.Dispose()
77
78         pass := NewPassManager()
79         defer pass.Dispose()
80
81         pass.Add(engine.TargetData())
82         pass.AddConstantPropagationPass()
83         pass.AddInstructionCombiningPass()
84         pass.AddPromoteMemoryToRegisterPass()
85         pass.AddGVNPass()
86         pass.AddCFGSimplificationPass()
87         pass.Run(mod)
88
89         exec_args := []GenericValue{NewGenericValueFromInt(Int32Type(), 10, false)}
90         exec_res := engine.RunFunction(fac, exec_args)
91         var fac10 uint64 = 10 * 9 * 8 * 7 * 6 * 5 * 4 * 3 * 2 * 1
92         if exec_res.Int(false) != fac10 {
93                 t.Errorf("Expected %d, got %d", fac10, exec_res.Int(false))
94         }
95 }