blob: 6cb5d29714f041c77f62ba5719d448abf37bca6b (
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
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
|
#include "slang-ir-peephole.h"
#include "slang-ir-inst-pass-base.h"
namespace Slang
{
struct PeepholeContext : InstPassBase
{
PeepholeContext(IRModule* inModule)
: InstPassBase(inModule)
{}
bool changed = false;
void processInst(IRInst* inst)
{
switch (inst->getOp())
{
case kIROp_GetResultError:
if (inst->getOperand(0)->getOp() == kIROp_MakeResultError)
{
inst->replaceUsesWith(inst->getOperand(0)->getOperand(0));
changed = true;
}
break;
case kIROp_GetResultValue:
if (inst->getOperand(0)->getOp() == kIROp_MakeResultValue)
{
inst->replaceUsesWith(inst->getOperand(0)->getOperand(0));
inst->removeAndDeallocate();
changed = true;
}
break;
case kIROp_IsResultError:
if (inst->getOperand(0)->getOp() == kIROp_MakeResultError)
{
IRBuilder builder(&sharedBuilderStorage);
inst->replaceUsesWith(builder.getBoolValue(true));
inst->removeAndDeallocate();
changed = true;
}
else if (inst->getOperand(0)->getOp() == kIROp_MakeResultValue)
{
IRBuilder builder(&sharedBuilderStorage);
inst->replaceUsesWith(builder.getBoolValue(false));
inst->removeAndDeallocate();
changed = true;
}
break;
case kIROp_GetTupleElement:
if (inst->getOperand(0)->getOp() == kIROp_MakeTuple)
{
auto element = inst->getOperand(1);
if (auto intLit = as<IRIntLit>(element))
{
inst->replaceUsesWith(inst->getOperand(0)->getOperand((UInt)intLit->value.intVal));
inst->removeAndDeallocate();
changed = true;
}
}
break;
case kIROp_FieldExtract:
if (inst->getOperand(0)->getOp() == kIROp_makeStruct)
{
auto field = as<IRFieldExtract>(inst)->field.get();
Index fieldIndex = -1;
auto structType = as<IRStructType>(inst->getOperand(0)->getDataType());
if (structType)
{
Index i = 0;
for (auto sfield : structType->getFields())
{
if (sfield == field)
{
fieldIndex = i;
break;
}
i++;
}
if (fieldIndex != -1 && fieldIndex < (Index)inst->getOperand(0)->getOperandCount())
{
inst->replaceUsesWith(inst->getOperand(0)->getOperand((UInt)fieldIndex));
inst->removeAndDeallocate();
changed = true;
}
}
}
break;
default:
break;
}
}
bool processModule()
{
SharedIRBuilder* sharedBuilder = &sharedBuilderStorage;
sharedBuilder->init(module);
changed = false;
processAllInsts([this](IRInst* inst) { processInst(inst); });
return changed;
}
};
bool peepholeOptimize(IRModule* module)
{
PeepholeContext context = PeepholeContext(module);
return context.processModule();
}
} // namespace Slang
|
