-
Notifications
You must be signed in to change notification settings - Fork 392
/
Pow2.cpp
125 lines (101 loc) · 3.25 KB
/
Pow2.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
/*******************************************************************************
* Copyright (c) 2016, 2021 IBM Corp. and others
*
* This program and the accompanying materials are made available under
* the terms of the Eclipse Public License 2.0 which accompanies this
* distribution and is available at https://www.eclipse.org/legal/epl-2.0/
* or the Apache License, Version 2.0 which accompanies this distribution and
* is available at https://www.apache.org/licenses/LICENSE-2.0.
*
* This Source Code may also be made available under the following
* Secondary Licenses when the conditions for such availability set
* forth in the Eclipse Public License, v. 2.0 are satisfied: GNU
* General Public License, version 2 with the GNU Classpath
* Exception [1] and GNU General Public License, version 2 with the
* OpenJDK Assembly Exception [2].
*
* [1] https://www.gnu.org/software/classpath/license.html
* [2] http://openjdk.java.net/legal/assembly-exception.html
*
* SPDX-License-Identifier: EPL-2.0 OR Apache-2.0 OR GPL-2.0 WITH Classpath-exception-2.0 OR LicenseRef-GPL-2.0 WITH Assembly-exception
*******************************************************************************/
#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <errno.h>
#include "Pow2.hpp"
#include "omrformatconsts.h"
Pow2Method::Pow2Method(OMR::JitBuilder::TypeDictionary *types)
: OMR::JitBuilder::MethodBuilder(types)
{
DefineLine(LINETOSTR(__LINE__));
DefineFile(__FILE__);
DefineName("pow2");
DefineParameter("n", Int64);
DefineReturnType(Int64);
}
bool
Pow2Method::buildIL()
{
Store("a",
ConstInt64(1));
Store("b",
ConstInt64(1));
Store("i",
Load("n"));
Store("keepIterating",
GreaterThan(
Load("i"),
ConstInt64(-1)));
OMR::JitBuilder::IlBuilder *loopBody = NULL;
WhileDoLoop("keepIterating", &loopBody);
loopBody->Store("a",
loopBody-> Load("b"));
loopBody->Store("b",
loopBody-> Add(
loopBody-> Load("a"),
loopBody-> Load("b")));
loopBody->Store("i",
loopBody-> Sub(
loopBody-> Load("i"),
loopBody-> ConstInt64(1)));
loopBody->Store("keepIterating",
loopBody-> GreaterThan(
loopBody-> Load("i"),
loopBody-> ConstInt64(-1)));
Return(
Load("a"));
return true;
}
int
main(int argc, char *argv[])
{
printf("Step 1: initialize JIT\n");
bool initialized = initializeJit();
if (!initialized)
{
fprintf(stderr, "FAIL: could not initialize JIT\n");
exit(-1);
}
printf("Step 2: define relevant types\n");
OMR::JitBuilder::TypeDictionary types;
printf("Step 3: compile method builder\n");
Pow2Method Pow2Method(&types);
void *entry=0;
int32_t rc = compileMethodBuilder(&Pow2Method, &entry);
if (rc != 0)
{
fprintf(stderr,"FAIL: compilation error %d\n", rc);
exit(-2);
}
printf("Step 4: invoke compiled code\n");
Pow2FunctionType *pow2 = (Pow2FunctionType *)entry;
int32_t n = (argc > 1) ? atoi(argv[1]) : 6000000;
int64_t r;
for (int32_t i=0;i < n;i++)
r = pow2((int64_t) 45);
printf("pow2(45) is %" OMR_PRId64 "\n", r);
printf ("Step 5: shutdown JIT\n");
shutdownJit();
printf("PASS\n");
}