blob: 7fbccf7160e17808c84f5089e64cc8ca4d01243b (
plain)
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
|
#include "llvm/CodeGen/GlobalISel/InstructionSelect.h"
#include "GISelMITest.h"
#include "llvm/CodeGen/GlobalISel/InstructionSelector.h"
#include "llvm/CodeGen/TargetPassConfig.h"
#include "llvm/IR/LegacyPassManager.h"
namespace {
class EraseMockInstructionSelector : public InstructionSelector {
public:
int NumSelected = 0;
SmallVector<MachineInstr *> MIs;
bool select(MachineInstr &MI) override {
++NumSelected;
switch (NumSelected) {
case 1:
EXPECT_EQ(&MI, MIs[8]);
// Erase previous instructions
MIs[7]->eraseFromParent();
MIs[6]->eraseFromParent();
// Don't erase this MI before step 3 to prevent DCE
return true;
case 2:
EXPECT_EQ(&MI, MIs[5]);
// Erase previous instructions reversed
MIs[3]->eraseFromParent();
MIs[4]->eraseFromParent();
MI.eraseFromParent();
return true;
case 3:
EXPECT_EQ(&MI, MIs[2]);
MIs[8]->eraseFromParent();
// Erase first instructions
MIs[0]->eraseFromParent();
MIs[1]->eraseFromParent();
MI.eraseFromParent();
return true;
default:
ADD_FAILURE();
return false;
}
}
void setupGeneratedPerFunctionState(MachineFunction &MF) override {}
};
TEST_F(AArch64GISelMITest, TestInstructionSelectErase) {
StringRef MIRString = R"(
$x0 = COPY %2(s64)
$x0 = COPY %2(s64)
$x0 = COPY %2(s64)
$x0 = COPY %2(s64)
$x0 = COPY %2(s64)
$x0 = COPY %2(s64)
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
legacy::PassManager PM;
std::unique_ptr<TargetPassConfig> TPC(TM->createPassConfig(PM));
EraseMockInstructionSelector ISel;
ISel.TPC = TPC.get();
for (auto &MI : *EntryMBB) {
ISel.MIs.push_back(&MI);
}
InstructionSelect ISelPass;
ISelPass.setInstructionSelector(&ISel);
ISelPass.selectMachineFunction(*MF);
EXPECT_EQ(ISel.NumSelected, 3);
}
} // namespace
|
