-
Notifications
You must be signed in to change notification settings - Fork 393
/
CompilerUnitTest.hpp
138 lines (117 loc) · 4.33 KB
/
CompilerUnitTest.hpp
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
126
127
128
129
130
131
132
133
134
135
136
137
138
/*******************************************************************************
* Copyright (c) 2020, 2020 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 http://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
*******************************************************************************/
#ifndef COMPILER_UNIT_TEST_HPP
#define COMPILER_UNIT_TEST_HPP
#include <gtest/gtest.h>
#include <exception>
#include "Jit.hpp"
#include "codegen/CodeGenerator.hpp"
#include "compile/Compilation.hpp"
#include "env/ConcreteFE.hpp"
#include "env/SystemSegmentProvider.hpp"
#include "ilgen/IlGenRequest.hpp"
#include "ilgen/IlGeneratorMethodDetails.hpp"
#include "ilgen/TypeDictionary.hpp"
#include "codegen/CodeGenerator_inlines.hpp"
#include "compile/Compilation_inlines.hpp"
#include "ilgen/IlGeneratorMethodDetails_inlines.hpp"
#include "il/ResolvedMethodSymbol.hpp"
namespace TRTest {
class NullIlGenRequest : public TR::IlGenRequest {
TR::IlGeneratorMethodDetails _details;
public:
NullIlGenRequest() : TR::IlGenRequest(_details) {}
virtual TR_IlGenerator *getIlGenerator(
TR::ResolvedMethodSymbol *methodSymbol,
TR_FrontEnd *fe,
TR::Compilation *comp,
TR::SymbolReferenceTable *symRefTab
) {
throw std::runtime_error("The mock JIT environment does not support calling TR::IlGenRequest::getIlGenerator");
}
virtual void print(TR_FrontEnd *fe, TR::FILE *file, const char *suffix) {}
};
class JitInitializer {
public:
JitInitializer() {
initializeJit();
}
~JitInitializer() {
shutdownJit();
}
};
/**
* A base class containing necessary mocking objects for testing compiler and optimizers.
*/
class CompilerUnitTest : public ::testing::Test {
public:
CompilerUnitTest() :
_jitInit(),
_rawAllocator(),
_segmentProvider(1 << 16, _rawAllocator),
_dispatchRegion(_segmentProvider, _rawAllocator),
_trMemory(*OMR::FrontEnd::singleton().persistentMemory(), _dispatchRegion),
_types(),
_options(),
_ilGenRequest(),
_method("compunittest", "0", "test", 0, NULL, _types.NoType, NULL, NULL),
_comp(0, NULL, &OMR::FrontEnd::singleton(), &_method, _ilGenRequest, _options, _dispatchRegion, &_trMemory, TR_OptimizationPlan::alloc(warm)) {
_symbol = TR::ResolvedMethodSymbol::create(_comp.trStackMemory(), &_method, &_comp);
TR::CFG* cfg = new (region()) TR::CFG(&_comp, _symbol, region());
_symbol->setFlowGraph(cfg);
_optimizer = new (region()) TR::Optimizer(&_comp, _symbol, false);
_comp.setOptimizer(_optimizer);
}
TR::Region& region() { return _dispatchRegion; }
protected:
JitInitializer _jitInit;
TR::RawAllocator _rawAllocator;
TR::SystemSegmentProvider _segmentProvider;
TR::Region _dispatchRegion;
TR_Memory _trMemory;
TR::TypeDictionary _types;
TR::Options _options;
NullIlGenRequest _ilGenRequest;
TR::ResolvedMethodSymbol* _symbol;
TR::ResolvedMethod _method;
TR::Compilation _comp;
TR::Optimizer* _optimizer;
};
template <typename T>
class MakeVector {
std::vector<T> _vals;
void add_vals() {}
template <typename... Ts>
void add_vals(T next_val, Ts... more_vals) {
_vals.push_back(next_val);
add_vals(more_vals...);
}
public:
template <typename... Ts>
MakeVector(Ts... vals) {
add_vals(vals...);
}
const std::vector<T>& operator*() const {
return _vals;
}
};
}
#endif