blob: 5d1689c3fdde1fc629027e632173c710032a9434 [file] [log] [blame]
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001//===- DAGISelEmitter.cpp - Generate an instruction selector --------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file was developed by Chris Lattner and is distributed under
6// the University of Illinois Open Source License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This tablegen backend emits a DAG instruction selector.
11//
12//===----------------------------------------------------------------------===//
13
14#include "DAGISelEmitter.h"
15#include "Record.h"
16#include "llvm/ADT/StringExtras.h"
17#include "llvm/Support/Debug.h"
Jeff Cohena48283b2005-09-25 19:04:43 +000018#include <algorithm>
Chris Lattner54cb8fd2005-09-07 23:44:43 +000019#include <set>
20using namespace llvm;
21
Chris Lattnerca559d02005-09-08 21:03:01 +000022//===----------------------------------------------------------------------===//
Chris Lattner3c7e18d2005-10-14 06:12:03 +000023// Helpers for working with extended types.
24
25/// FilterVTs - Filter a list of VT's according to a predicate.
26///
27template<typename T>
28static std::vector<MVT::ValueType>
29FilterVTs(const std::vector<MVT::ValueType> &InVTs, T Filter) {
30 std::vector<MVT::ValueType> Result;
31 for (unsigned i = 0, e = InVTs.size(); i != e; ++i)
32 if (Filter(InVTs[i]))
33 Result.push_back(InVTs[i]);
34 return Result;
35}
36
Nate Begemanb73628b2005-12-30 00:12:56 +000037template<typename T>
38static std::vector<unsigned char>
39FilterEVTs(const std::vector<unsigned char> &InVTs, T Filter) {
40 std::vector<unsigned char> Result;
41 for (unsigned i = 0, e = InVTs.size(); i != e; ++i)
42 if (Filter((MVT::ValueType)InVTs[i]))
43 Result.push_back(InVTs[i]);
44 return Result;
Chris Lattner3c7e18d2005-10-14 06:12:03 +000045}
46
Nate Begemanb73628b2005-12-30 00:12:56 +000047static std::vector<unsigned char>
48ConvertVTs(const std::vector<MVT::ValueType> &InVTs) {
49 std::vector<unsigned char> Result;
50 for (unsigned i = 0, e = InVTs.size(); i != e; ++i)
51 Result.push_back(InVTs[i]);
52 return Result;
53}
54
55static bool LHSIsSubsetOfRHS(const std::vector<unsigned char> &LHS,
56 const std::vector<unsigned char> &RHS) {
57 if (LHS.size() > RHS.size()) return false;
58 for (unsigned i = 0, e = LHS.size(); i != e; ++i)
Duraid Madinad47ae092005-12-30 16:41:48 +000059 if (std::find(RHS.begin(), RHS.end(), LHS[i]) == RHS.end())
Nate Begemanb73628b2005-12-30 00:12:56 +000060 return false;
61 return true;
62}
63
64/// isExtIntegerVT - Return true if the specified extended value type vector
65/// contains isInt or an integer value type.
Chris Lattner697f8842006-03-20 05:39:48 +000066static bool isExtIntegerInVTs(const std::vector<unsigned char> &EVTs) {
Nate Begemanb73628b2005-12-30 00:12:56 +000067 assert(!EVTs.empty() && "Cannot check for integer in empty ExtVT list!");
68 return EVTs[0] == MVT::isInt || !(FilterEVTs(EVTs, MVT::isInteger).empty());
69}
70
71/// isExtFloatingPointVT - Return true if the specified extended value type
72/// vector contains isFP or a FP value type.
Chris Lattner697f8842006-03-20 05:39:48 +000073static bool isExtFloatingPointInVTs(const std::vector<unsigned char> &EVTs) {
Nate Begemanb73628b2005-12-30 00:12:56 +000074 assert(!EVTs.empty() && "Cannot check for integer in empty ExtVT list!");
Chris Lattner488580c2006-01-28 19:06:51 +000075 return EVTs[0] == MVT::isFP ||
76 !(FilterEVTs(EVTs, MVT::isFloatingPoint).empty());
Chris Lattner3c7e18d2005-10-14 06:12:03 +000077}
78
79//===----------------------------------------------------------------------===//
Chris Lattner33c92e92005-09-08 21:27:15 +000080// SDTypeConstraint implementation
81//
82
83SDTypeConstraint::SDTypeConstraint(Record *R) {
84 OperandNo = R->getValueAsInt("OperandNum");
85
86 if (R->isSubClassOf("SDTCisVT")) {
87 ConstraintType = SDTCisVT;
88 x.SDTCisVT_Info.VT = getValueType(R->getValueAsDef("VT"));
Chris Lattner5b21be72005-12-09 22:57:42 +000089 } else if (R->isSubClassOf("SDTCisPtrTy")) {
90 ConstraintType = SDTCisPtrTy;
Chris Lattner33c92e92005-09-08 21:27:15 +000091 } else if (R->isSubClassOf("SDTCisInt")) {
92 ConstraintType = SDTCisInt;
93 } else if (R->isSubClassOf("SDTCisFP")) {
94 ConstraintType = SDTCisFP;
95 } else if (R->isSubClassOf("SDTCisSameAs")) {
96 ConstraintType = SDTCisSameAs;
97 x.SDTCisSameAs_Info.OtherOperandNum = R->getValueAsInt("OtherOperandNum");
98 } else if (R->isSubClassOf("SDTCisVTSmallerThanOp")) {
99 ConstraintType = SDTCisVTSmallerThanOp;
100 x.SDTCisVTSmallerThanOp_Info.OtherOperandNum =
101 R->getValueAsInt("OtherOperandNum");
Chris Lattner03ebd802005-10-14 04:53:53 +0000102 } else if (R->isSubClassOf("SDTCisOpSmallerThanOp")) {
103 ConstraintType = SDTCisOpSmallerThanOp;
104 x.SDTCisOpSmallerThanOp_Info.BigOperandNum =
105 R->getValueAsInt("BigOperandNum");
Chris Lattner697f8842006-03-20 05:39:48 +0000106 } else if (R->isSubClassOf("SDTCisIntVectorOfSameSize")) {
107 ConstraintType = SDTCisIntVectorOfSameSize;
108 x.SDTCisIntVectorOfSameSize_Info.OtherOperandNum =
109 R->getValueAsInt("OtherOpNum");
Chris Lattner33c92e92005-09-08 21:27:15 +0000110 } else {
111 std::cerr << "Unrecognized SDTypeConstraint '" << R->getName() << "'!\n";
112 exit(1);
113 }
114}
115
Chris Lattner32707602005-09-08 23:22:48 +0000116/// getOperandNum - Return the node corresponding to operand #OpNo in tree
117/// N, which has NumResults results.
118TreePatternNode *SDTypeConstraint::getOperandNum(unsigned OpNo,
119 TreePatternNode *N,
120 unsigned NumResults) const {
Evan Cheng1c3d19e2005-12-04 08:18:16 +0000121 assert(NumResults <= 1 &&
122 "We only work with nodes with zero or one result so far!");
Chris Lattner32707602005-09-08 23:22:48 +0000123
124 if (OpNo < NumResults)
125 return N; // FIXME: need value #
126 else
127 return N->getChild(OpNo-NumResults);
128}
129
130/// ApplyTypeConstraint - Given a node in a pattern, apply this type
131/// constraint to the nodes operands. This returns true if it makes a
132/// change, false otherwise. If a type contradiction is found, throw an
133/// exception.
134bool SDTypeConstraint::ApplyTypeConstraint(TreePatternNode *N,
135 const SDNodeInfo &NodeInfo,
136 TreePattern &TP) const {
137 unsigned NumResults = NodeInfo.getNumResults();
Evan Cheng1c3d19e2005-12-04 08:18:16 +0000138 assert(NumResults <= 1 &&
139 "We only work with nodes with zero or one result so far!");
Chris Lattner32707602005-09-08 23:22:48 +0000140
141 // Check that the number of operands is sane.
142 if (NodeInfo.getNumOperands() >= 0) {
143 if (N->getNumChildren() != (unsigned)NodeInfo.getNumOperands())
144 TP.error(N->getOperator()->getName() + " node requires exactly " +
145 itostr(NodeInfo.getNumOperands()) + " operands!");
146 }
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000147
148 const CodeGenTarget &CGT = TP.getDAGISelEmitter().getTargetInfo();
Chris Lattner32707602005-09-08 23:22:48 +0000149
150 TreePatternNode *NodeToApply = getOperandNum(OperandNo, N, NumResults);
151
152 switch (ConstraintType) {
153 default: assert(0 && "Unknown constraint type!");
154 case SDTCisVT:
155 // Operand must be a particular type.
156 return NodeToApply->UpdateNodeType(x.SDTCisVT_Info.VT, TP);
Chris Lattner5b21be72005-12-09 22:57:42 +0000157 case SDTCisPtrTy: {
158 // Operand must be same as target pointer type.
159 return NodeToApply->UpdateNodeType(CGT.getPointerType(), TP);
160 }
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000161 case SDTCisInt: {
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000162 // If there is only one integer type supported, this must be it.
Chris Lattnere0583b12005-10-14 05:08:37 +0000163 std::vector<MVT::ValueType> IntVTs =
164 FilterVTs(CGT.getLegalValueTypes(), MVT::isInteger);
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000165
166 // If we found exactly one supported integer type, apply it.
Chris Lattnere0583b12005-10-14 05:08:37 +0000167 if (IntVTs.size() == 1)
168 return NodeToApply->UpdateNodeType(IntVTs[0], TP);
Chris Lattner3c7e18d2005-10-14 06:12:03 +0000169 return NodeToApply->UpdateNodeType(MVT::isInt, TP);
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000170 }
171 case SDTCisFP: {
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000172 // If there is only one FP type supported, this must be it.
Chris Lattnere0583b12005-10-14 05:08:37 +0000173 std::vector<MVT::ValueType> FPVTs =
174 FilterVTs(CGT.getLegalValueTypes(), MVT::isFloatingPoint);
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000175
176 // If we found exactly one supported FP type, apply it.
Chris Lattnere0583b12005-10-14 05:08:37 +0000177 if (FPVTs.size() == 1)
178 return NodeToApply->UpdateNodeType(FPVTs[0], TP);
Chris Lattner3c7e18d2005-10-14 06:12:03 +0000179 return NodeToApply->UpdateNodeType(MVT::isFP, TP);
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000180 }
Chris Lattner32707602005-09-08 23:22:48 +0000181 case SDTCisSameAs: {
182 TreePatternNode *OtherNode =
183 getOperandNum(x.SDTCisSameAs_Info.OtherOperandNum, N, NumResults);
Nate Begemanb73628b2005-12-30 00:12:56 +0000184 return NodeToApply->UpdateNodeType(OtherNode->getExtTypes(), TP) |
185 OtherNode->UpdateNodeType(NodeToApply->getExtTypes(), TP);
Chris Lattner32707602005-09-08 23:22:48 +0000186 }
187 case SDTCisVTSmallerThanOp: {
188 // The NodeToApply must be a leaf node that is a VT. OtherOperandNum must
189 // have an integer type that is smaller than the VT.
190 if (!NodeToApply->isLeaf() ||
191 !dynamic_cast<DefInit*>(NodeToApply->getLeafValue()) ||
192 !static_cast<DefInit*>(NodeToApply->getLeafValue())->getDef()
193 ->isSubClassOf("ValueType"))
194 TP.error(N->getOperator()->getName() + " expects a VT operand!");
195 MVT::ValueType VT =
196 getValueType(static_cast<DefInit*>(NodeToApply->getLeafValue())->getDef());
197 if (!MVT::isInteger(VT))
198 TP.error(N->getOperator()->getName() + " VT operand must be integer!");
199
200 TreePatternNode *OtherNode =
201 getOperandNum(x.SDTCisVTSmallerThanOp_Info.OtherOperandNum, N,NumResults);
Chris Lattner3c7e18d2005-10-14 06:12:03 +0000202
203 // It must be integer.
204 bool MadeChange = false;
205 MadeChange |= OtherNode->UpdateNodeType(MVT::isInt, TP);
206
Nate Begemanb73628b2005-12-30 00:12:56 +0000207 // This code only handles nodes that have one type set. Assert here so
208 // that we can change this if we ever need to deal with multiple value
209 // types at this point.
210 assert(OtherNode->getExtTypes().size() == 1 && "Node has too many types!");
211 if (OtherNode->hasTypeSet() && OtherNode->getTypeNum(0) <= VT)
Chris Lattner32707602005-09-08 23:22:48 +0000212 OtherNode->UpdateNodeType(MVT::Other, TP); // Throw an error.
213 return false;
214 }
Chris Lattner03ebd802005-10-14 04:53:53 +0000215 case SDTCisOpSmallerThanOp: {
Chris Lattner603d78c2005-10-14 06:25:00 +0000216 TreePatternNode *BigOperand =
217 getOperandNum(x.SDTCisOpSmallerThanOp_Info.BigOperandNum, N, NumResults);
218
219 // Both operands must be integer or FP, but we don't care which.
220 bool MadeChange = false;
221
Nate Begemanb73628b2005-12-30 00:12:56 +0000222 // This code does not currently handle nodes which have multiple types,
223 // where some types are integer, and some are fp. Assert that this is not
224 // the case.
225 assert(!(isExtIntegerInVTs(NodeToApply->getExtTypes()) &&
226 isExtFloatingPointInVTs(NodeToApply->getExtTypes())) &&
227 !(isExtIntegerInVTs(BigOperand->getExtTypes()) &&
228 isExtFloatingPointInVTs(BigOperand->getExtTypes())) &&
229 "SDTCisOpSmallerThanOp does not handle mixed int/fp types!");
230 if (isExtIntegerInVTs(NodeToApply->getExtTypes()))
Chris Lattner603d78c2005-10-14 06:25:00 +0000231 MadeChange |= BigOperand->UpdateNodeType(MVT::isInt, TP);
Nate Begemanb73628b2005-12-30 00:12:56 +0000232 else if (isExtFloatingPointInVTs(NodeToApply->getExtTypes()))
Chris Lattner603d78c2005-10-14 06:25:00 +0000233 MadeChange |= BigOperand->UpdateNodeType(MVT::isFP, TP);
Nate Begemanb73628b2005-12-30 00:12:56 +0000234 if (isExtIntegerInVTs(BigOperand->getExtTypes()))
Chris Lattner603d78c2005-10-14 06:25:00 +0000235 MadeChange |= NodeToApply->UpdateNodeType(MVT::isInt, TP);
Nate Begemanb73628b2005-12-30 00:12:56 +0000236 else if (isExtFloatingPointInVTs(BigOperand->getExtTypes()))
Chris Lattner603d78c2005-10-14 06:25:00 +0000237 MadeChange |= NodeToApply->UpdateNodeType(MVT::isFP, TP);
238
239 std::vector<MVT::ValueType> VTs = CGT.getLegalValueTypes();
240
Nate Begemanb73628b2005-12-30 00:12:56 +0000241 if (isExtIntegerInVTs(NodeToApply->getExtTypes())) {
Chris Lattner603d78c2005-10-14 06:25:00 +0000242 VTs = FilterVTs(VTs, MVT::isInteger);
Nate Begemanb73628b2005-12-30 00:12:56 +0000243 } else if (isExtFloatingPointInVTs(NodeToApply->getExtTypes())) {
Chris Lattner603d78c2005-10-14 06:25:00 +0000244 VTs = FilterVTs(VTs, MVT::isFloatingPoint);
245 } else {
246 VTs.clear();
247 }
248
249 switch (VTs.size()) {
250 default: // Too many VT's to pick from.
251 case 0: break; // No info yet.
252 case 1:
253 // Only one VT of this flavor. Cannot ever satisify the constraints.
254 return NodeToApply->UpdateNodeType(MVT::Other, TP); // throw
255 case 2:
256 // If we have exactly two possible types, the little operand must be the
257 // small one, the big operand should be the big one. Common with
258 // float/double for example.
259 assert(VTs[0] < VTs[1] && "Should be sorted!");
260 MadeChange |= NodeToApply->UpdateNodeType(VTs[0], TP);
261 MadeChange |= BigOperand->UpdateNodeType(VTs[1], TP);
262 break;
263 }
264 return MadeChange;
Chris Lattner03ebd802005-10-14 04:53:53 +0000265 }
Chris Lattner697f8842006-03-20 05:39:48 +0000266 case SDTCisIntVectorOfSameSize: {
267 TreePatternNode *OtherOperand =
268 getOperandNum(x.SDTCisIntVectorOfSameSize_Info.OtherOperandNum,
269 N, NumResults);
270 if (OtherOperand->hasTypeSet()) {
271 if (!MVT::isVector(OtherOperand->getTypeNum(0)))
272 TP.error(N->getOperator()->getName() + " VT operand must be a vector!");
273 MVT::ValueType IVT = OtherOperand->getTypeNum(0);
274 IVT = MVT::getIntVectorWithNumElements(MVT::getVectorNumElements(IVT));
275 return NodeToApply->UpdateNodeType(IVT, TP);
276 }
277 return false;
278 }
Chris Lattner32707602005-09-08 23:22:48 +0000279 }
280 return false;
281}
282
283
Chris Lattner33c92e92005-09-08 21:27:15 +0000284//===----------------------------------------------------------------------===//
Chris Lattnerca559d02005-09-08 21:03:01 +0000285// SDNodeInfo implementation
286//
287SDNodeInfo::SDNodeInfo(Record *R) : Def(R) {
288 EnumName = R->getValueAsString("Opcode");
289 SDClassName = R->getValueAsString("SDClass");
Chris Lattner33c92e92005-09-08 21:27:15 +0000290 Record *TypeProfile = R->getValueAsDef("TypeProfile");
291 NumResults = TypeProfile->getValueAsInt("NumResults");
292 NumOperands = TypeProfile->getValueAsInt("NumOperands");
293
Chris Lattnera1a68ae2005-09-28 18:28:29 +0000294 // Parse the properties.
295 Properties = 0;
Chris Lattnerb0e103d2005-10-28 22:49:02 +0000296 std::vector<Record*> PropList = R->getValueAsListOfDefs("Properties");
Chris Lattner6bc0d742005-10-28 22:43:25 +0000297 for (unsigned i = 0, e = PropList.size(); i != e; ++i) {
298 if (PropList[i]->getName() == "SDNPCommutative") {
Chris Lattnera1a68ae2005-09-28 18:28:29 +0000299 Properties |= 1 << SDNPCommutative;
Chris Lattner6bc0d742005-10-28 22:43:25 +0000300 } else if (PropList[i]->getName() == "SDNPAssociative") {
Chris Lattner7cf2fe62005-09-28 20:58:06 +0000301 Properties |= 1 << SDNPAssociative;
Evan Cheng1c3d19e2005-12-04 08:18:16 +0000302 } else if (PropList[i]->getName() == "SDNPHasChain") {
303 Properties |= 1 << SDNPHasChain;
Evan Cheng51fecc82006-01-09 18:27:06 +0000304 } else if (PropList[i]->getName() == "SDNPOutFlag") {
305 Properties |= 1 << SDNPOutFlag;
306 } else if (PropList[i]->getName() == "SDNPInFlag") {
307 Properties |= 1 << SDNPInFlag;
308 } else if (PropList[i]->getName() == "SDNPOptInFlag") {
309 Properties |= 1 << SDNPOptInFlag;
Chris Lattnera1a68ae2005-09-28 18:28:29 +0000310 } else {
Chris Lattner6bc0d742005-10-28 22:43:25 +0000311 std::cerr << "Unknown SD Node property '" << PropList[i]->getName()
Chris Lattnera1a68ae2005-09-28 18:28:29 +0000312 << "' on node '" << R->getName() << "'!\n";
313 exit(1);
314 }
315 }
316
317
Chris Lattner33c92e92005-09-08 21:27:15 +0000318 // Parse the type constraints.
Chris Lattnerb0e103d2005-10-28 22:49:02 +0000319 std::vector<Record*> ConstraintList =
320 TypeProfile->getValueAsListOfDefs("Constraints");
321 TypeConstraints.assign(ConstraintList.begin(), ConstraintList.end());
Chris Lattnerca559d02005-09-08 21:03:01 +0000322}
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000323
324//===----------------------------------------------------------------------===//
325// TreePatternNode implementation
326//
327
328TreePatternNode::~TreePatternNode() {
329#if 0 // FIXME: implement refcounted tree nodes!
330 for (unsigned i = 0, e = getNumChildren(); i != e; ++i)
331 delete getChild(i);
332#endif
333}
334
Chris Lattner32707602005-09-08 23:22:48 +0000335/// UpdateNodeType - Set the node type of N to VT if VT contains
336/// information. If N already contains a conflicting type, then throw an
337/// exception. This returns true if any information was updated.
338///
Nate Begemanb73628b2005-12-30 00:12:56 +0000339bool TreePatternNode::UpdateNodeType(const std::vector<unsigned char> &ExtVTs,
340 TreePattern &TP) {
341 assert(!ExtVTs.empty() && "Cannot update node type with empty type vector!");
342
343 if (ExtVTs[0] == MVT::isUnknown || LHSIsSubsetOfRHS(getExtTypes(), ExtVTs))
344 return false;
345 if (isTypeCompletelyUnknown() || LHSIsSubsetOfRHS(ExtVTs, getExtTypes())) {
346 setTypes(ExtVTs);
Chris Lattner32707602005-09-08 23:22:48 +0000347 return true;
348 }
349
Nate Begemanb73628b2005-12-30 00:12:56 +0000350 if (ExtVTs[0] == MVT::isInt && isExtIntegerInVTs(getExtTypes())) {
351 assert(hasTypeSet() && "should be handled above!");
352 std::vector<unsigned char> FVTs = FilterEVTs(getExtTypes(), MVT::isInteger);
353 if (getExtTypes() == FVTs)
354 return false;
355 setTypes(FVTs);
356 return true;
357 }
358 if (ExtVTs[0] == MVT::isFP && isExtFloatingPointInVTs(getExtTypes())) {
359 assert(hasTypeSet() && "should be handled above!");
Chris Lattner488580c2006-01-28 19:06:51 +0000360 std::vector<unsigned char> FVTs =
361 FilterEVTs(getExtTypes(), MVT::isFloatingPoint);
Nate Begemanb73628b2005-12-30 00:12:56 +0000362 if (getExtTypes() == FVTs)
363 return false;
364 setTypes(FVTs);
365 return true;
366 }
Chris Lattner3c7e18d2005-10-14 06:12:03 +0000367
368 // If we know this is an int or fp type, and we are told it is a specific one,
369 // take the advice.
Nate Begemanb73628b2005-12-30 00:12:56 +0000370 //
371 // Similarly, we should probably set the type here to the intersection of
372 // {isInt|isFP} and ExtVTs
373 if ((getExtTypeNum(0) == MVT::isInt && isExtIntegerInVTs(ExtVTs)) ||
374 (getExtTypeNum(0) == MVT::isFP && isExtFloatingPointInVTs(ExtVTs))) {
375 setTypes(ExtVTs);
Chris Lattner3c7e18d2005-10-14 06:12:03 +0000376 return true;
377 }
378
Chris Lattner1531f202005-10-26 16:59:37 +0000379 if (isLeaf()) {
380 dump();
Evan Chengbcecf332005-12-17 01:19:28 +0000381 std::cerr << " ";
Chris Lattner1531f202005-10-26 16:59:37 +0000382 TP.error("Type inference contradiction found in node!");
383 } else {
384 TP.error("Type inference contradiction found in node " +
385 getOperator()->getName() + "!");
386 }
Chris Lattner32707602005-09-08 23:22:48 +0000387 return true; // unreachable
388}
389
390
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000391void TreePatternNode::print(std::ostream &OS) const {
392 if (isLeaf()) {
393 OS << *getLeafValue();
394 } else {
395 OS << "(" << getOperator()->getName();
396 }
397
Nate Begemanb73628b2005-12-30 00:12:56 +0000398 // FIXME: At some point we should handle printing all the value types for
399 // nodes that are multiply typed.
400 switch (getExtTypeNum(0)) {
Chris Lattner3c7e18d2005-10-14 06:12:03 +0000401 case MVT::Other: OS << ":Other"; break;
402 case MVT::isInt: OS << ":isInt"; break;
403 case MVT::isFP : OS << ":isFP"; break;
404 case MVT::isUnknown: ; /*OS << ":?";*/ break;
Nate Begemanb73628b2005-12-30 00:12:56 +0000405 default: OS << ":" << getTypeNum(0); break;
Chris Lattner3c7e18d2005-10-14 06:12:03 +0000406 }
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000407
408 if (!isLeaf()) {
409 if (getNumChildren() != 0) {
410 OS << " ";
411 getChild(0)->print(OS);
412 for (unsigned i = 1, e = getNumChildren(); i != e; ++i) {
413 OS << ", ";
414 getChild(i)->print(OS);
415 }
416 }
417 OS << ")";
418 }
419
420 if (!PredicateFn.empty())
Chris Lattner24eeeb82005-09-13 21:51:00 +0000421 OS << "<<P:" << PredicateFn << ">>";
Chris Lattnerb0276202005-09-14 22:55:26 +0000422 if (TransformFn)
423 OS << "<<X:" << TransformFn->getName() << ">>";
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000424 if (!getName().empty())
425 OS << ":$" << getName();
426
427}
428void TreePatternNode::dump() const {
429 print(std::cerr);
430}
431
Chris Lattnere46e17b2005-09-29 19:28:10 +0000432/// isIsomorphicTo - Return true if this node is recursively isomorphic to
433/// the specified node. For this comparison, all of the state of the node
434/// is considered, except for the assigned name. Nodes with differing names
435/// that are otherwise identical are considered isomorphic.
436bool TreePatternNode::isIsomorphicTo(const TreePatternNode *N) const {
437 if (N == this) return true;
Nate Begemanb73628b2005-12-30 00:12:56 +0000438 if (N->isLeaf() != isLeaf() || getExtTypes() != N->getExtTypes() ||
Chris Lattnere46e17b2005-09-29 19:28:10 +0000439 getPredicateFn() != N->getPredicateFn() ||
440 getTransformFn() != N->getTransformFn())
441 return false;
442
443 if (isLeaf()) {
444 if (DefInit *DI = dynamic_cast<DefInit*>(getLeafValue()))
445 if (DefInit *NDI = dynamic_cast<DefInit*>(N->getLeafValue()))
446 return DI->getDef() == NDI->getDef();
447 return getLeafValue() == N->getLeafValue();
448 }
449
450 if (N->getOperator() != getOperator() ||
451 N->getNumChildren() != getNumChildren()) return false;
452 for (unsigned i = 0, e = getNumChildren(); i != e; ++i)
453 if (!getChild(i)->isIsomorphicTo(N->getChild(i)))
454 return false;
455 return true;
456}
457
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000458/// clone - Make a copy of this tree and all of its children.
459///
460TreePatternNode *TreePatternNode::clone() const {
461 TreePatternNode *New;
462 if (isLeaf()) {
463 New = new TreePatternNode(getLeafValue());
464 } else {
465 std::vector<TreePatternNode*> CChildren;
466 CChildren.reserve(Children.size());
467 for (unsigned i = 0, e = getNumChildren(); i != e; ++i)
468 CChildren.push_back(getChild(i)->clone());
469 New = new TreePatternNode(getOperator(), CChildren);
470 }
471 New->setName(getName());
Nate Begemanb73628b2005-12-30 00:12:56 +0000472 New->setTypes(getExtTypes());
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000473 New->setPredicateFn(getPredicateFn());
Chris Lattner24eeeb82005-09-13 21:51:00 +0000474 New->setTransformFn(getTransformFn());
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000475 return New;
476}
477
Chris Lattner32707602005-09-08 23:22:48 +0000478/// SubstituteFormalArguments - Replace the formal arguments in this tree
479/// with actual values specified by ArgMap.
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000480void TreePatternNode::
481SubstituteFormalArguments(std::map<std::string, TreePatternNode*> &ArgMap) {
482 if (isLeaf()) return;
483
484 for (unsigned i = 0, e = getNumChildren(); i != e; ++i) {
485 TreePatternNode *Child = getChild(i);
486 if (Child->isLeaf()) {
487 Init *Val = Child->getLeafValue();
488 if (dynamic_cast<DefInit*>(Val) &&
489 static_cast<DefInit*>(Val)->getDef()->getName() == "node") {
490 // We found a use of a formal argument, replace it with its value.
491 Child = ArgMap[Child->getName()];
492 assert(Child && "Couldn't find formal argument!");
493 setChild(i, Child);
494 }
495 } else {
496 getChild(i)->SubstituteFormalArguments(ArgMap);
497 }
498 }
499}
500
501
502/// InlinePatternFragments - If this pattern refers to any pattern
503/// fragments, inline them into place, giving us a pattern without any
504/// PatFrag references.
505TreePatternNode *TreePatternNode::InlinePatternFragments(TreePattern &TP) {
506 if (isLeaf()) return this; // nothing to do.
507 Record *Op = getOperator();
508
509 if (!Op->isSubClassOf("PatFrag")) {
510 // Just recursively inline children nodes.
511 for (unsigned i = 0, e = getNumChildren(); i != e; ++i)
512 setChild(i, getChild(i)->InlinePatternFragments(TP));
513 return this;
514 }
515
516 // Otherwise, we found a reference to a fragment. First, look up its
517 // TreePattern record.
518 TreePattern *Frag = TP.getDAGISelEmitter().getPatternFragment(Op);
519
520 // Verify that we are passing the right number of operands.
521 if (Frag->getNumArgs() != Children.size())
522 TP.error("'" + Op->getName() + "' fragment requires " +
523 utostr(Frag->getNumArgs()) + " operands!");
524
Chris Lattner37937092005-09-09 01:15:01 +0000525 TreePatternNode *FragTree = Frag->getOnlyTree()->clone();
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000526
527 // Resolve formal arguments to their actual value.
528 if (Frag->getNumArgs()) {
529 // Compute the map of formal to actual arguments.
530 std::map<std::string, TreePatternNode*> ArgMap;
531 for (unsigned i = 0, e = Frag->getNumArgs(); i != e; ++i)
532 ArgMap[Frag->getArgName(i)] = getChild(i)->InlinePatternFragments(TP);
533
534 FragTree->SubstituteFormalArguments(ArgMap);
535 }
536
Chris Lattnerfbf8e572005-09-08 17:45:12 +0000537 FragTree->setName(getName());
Nate Begemanb73628b2005-12-30 00:12:56 +0000538 FragTree->UpdateNodeType(getExtTypes(), TP);
Chris Lattnerfbf8e572005-09-08 17:45:12 +0000539
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000540 // Get a new copy of this fragment to stitch into here.
541 //delete this; // FIXME: implement refcounting!
542 return FragTree;
543}
544
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000545/// getIntrinsicType - Check to see if the specified record has an intrinsic
546/// type which should be applied to it. This infer the type of register
547/// references from the register file information, for example.
548///
Nate Begemanb73628b2005-12-30 00:12:56 +0000549static std::vector<unsigned char> getIntrinsicType(Record *R, bool NotRegisters,
Chris Lattner3c7e18d2005-10-14 06:12:03 +0000550 TreePattern &TP) {
Nate Begemanb73628b2005-12-30 00:12:56 +0000551 // Some common return values
552 std::vector<unsigned char> Unknown(1, MVT::isUnknown);
553 std::vector<unsigned char> Other(1, MVT::Other);
554
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000555 // Check to see if this is a register or a register class...
556 if (R->isSubClassOf("RegisterClass")) {
Nate Begemanb73628b2005-12-30 00:12:56 +0000557 if (NotRegisters)
558 return Unknown;
Nate Begeman6510b222005-12-01 04:51:06 +0000559 const CodeGenRegisterClass &RC =
560 TP.getDAGISelEmitter().getTargetInfo().getRegisterClass(R);
Nate Begemanb73628b2005-12-30 00:12:56 +0000561 return ConvertVTs(RC.getValueTypes());
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000562 } else if (R->isSubClassOf("PatFrag")) {
563 // Pattern fragment types will be resolved when they are inlined.
Nate Begemanb73628b2005-12-30 00:12:56 +0000564 return Unknown;
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000565 } else if (R->isSubClassOf("Register")) {
Evan Cheng37e90052006-01-15 10:04:45 +0000566 if (NotRegisters)
567 return Unknown;
Chris Lattner22faeab2005-12-05 02:36:37 +0000568 // If the register appears in exactly one regclass, and the regclass has one
569 // value type, use it as the known type.
570 const CodeGenTarget &T = TP.getDAGISelEmitter().getTargetInfo();
571 if (const CodeGenRegisterClass *RC = T.getRegisterClassForRegister(R))
Nate Begemanb73628b2005-12-30 00:12:56 +0000572 return ConvertVTs(RC->getValueTypes());
573 return Unknown;
Chris Lattner1531f202005-10-26 16:59:37 +0000574 } else if (R->isSubClassOf("ValueType") || R->isSubClassOf("CondCode")) {
575 // Using a VTSDNode or CondCodeSDNode.
Nate Begemanb73628b2005-12-30 00:12:56 +0000576 return Other;
Evan Cheng0fc71982005-12-08 02:00:36 +0000577 } else if (R->isSubClassOf("ComplexPattern")) {
Evan Cheng57c517d2006-01-17 07:36:41 +0000578 if (NotRegisters)
579 return Unknown;
Nate Begemanb73628b2005-12-30 00:12:56 +0000580 std::vector<unsigned char>
581 ComplexPat(1, TP.getDAGISelEmitter().getComplexPattern(R).getValueType());
582 return ComplexPat;
Evan Cheng01f318b2005-12-14 02:21:57 +0000583 } else if (R->getName() == "node" || R->getName() == "srcvalue") {
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000584 // Placeholder.
Nate Begemanb73628b2005-12-30 00:12:56 +0000585 return Unknown;
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000586 }
587
588 TP.error("Unknown node flavor used in pattern: " + R->getName());
Nate Begemanb73628b2005-12-30 00:12:56 +0000589 return Other;
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000590}
591
Chris Lattner32707602005-09-08 23:22:48 +0000592/// ApplyTypeConstraints - Apply all of the type constraints relevent to
593/// this node and its children in the tree. This returns true if it makes a
594/// change, false otherwise. If a type contradiction is found, throw an
595/// exception.
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000596bool TreePatternNode::ApplyTypeConstraints(TreePattern &TP, bool NotRegisters) {
Chris Lattner5a1df382006-03-24 23:10:39 +0000597 DAGISelEmitter &ISE = TP.getDAGISelEmitter();
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000598 if (isLeaf()) {
Chris Lattner465c7372005-11-03 05:46:11 +0000599 if (DefInit *DI = dynamic_cast<DefInit*>(getLeafValue())) {
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000600 // If it's a regclass or something else known, include the type.
601 return UpdateNodeType(getIntrinsicType(DI->getDef(), NotRegisters, TP),
602 TP);
Chris Lattner465c7372005-11-03 05:46:11 +0000603 } else if (IntInit *II = dynamic_cast<IntInit*>(getLeafValue())) {
604 // Int inits are always integers. :)
605 bool MadeChange = UpdateNodeType(MVT::isInt, TP);
606
607 if (hasTypeSet()) {
Nate Begemanb73628b2005-12-30 00:12:56 +0000608 // At some point, it may make sense for this tree pattern to have
609 // multiple types. Assert here that it does not, so we revisit this
610 // code when appropriate.
611 assert(getExtTypes().size() == 1 && "TreePattern has too many types!");
612
613 unsigned Size = MVT::getSizeInBits(getTypeNum(0));
Chris Lattner465c7372005-11-03 05:46:11 +0000614 // Make sure that the value is representable for this type.
615 if (Size < 32) {
616 int Val = (II->getValue() << (32-Size)) >> (32-Size);
617 if (Val != II->getValue())
618 TP.error("Sign-extended integer value '" + itostr(II->getValue()) +
619 "' is out of range for type 'MVT::" +
Nate Begemanb73628b2005-12-30 00:12:56 +0000620 getEnumName(getTypeNum(0)) + "'!");
Chris Lattner465c7372005-11-03 05:46:11 +0000621 }
622 }
623
624 return MadeChange;
625 }
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000626 return false;
627 }
Chris Lattner32707602005-09-08 23:22:48 +0000628
629 // special handling for set, which isn't really an SDNode.
630 if (getOperator()->getName() == "set") {
631 assert (getNumChildren() == 2 && "Only handle 2 operand set's for now!");
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000632 bool MadeChange = getChild(0)->ApplyTypeConstraints(TP, NotRegisters);
633 MadeChange |= getChild(1)->ApplyTypeConstraints(TP, NotRegisters);
Chris Lattner32707602005-09-08 23:22:48 +0000634
635 // Types of operands must match.
Nate Begemanb73628b2005-12-30 00:12:56 +0000636 MadeChange |= getChild(0)->UpdateNodeType(getChild(1)->getExtTypes(), TP);
637 MadeChange |= getChild(1)->UpdateNodeType(getChild(0)->getExtTypes(), TP);
Chris Lattner32707602005-09-08 23:22:48 +0000638 MadeChange |= UpdateNodeType(MVT::isVoid, TP);
639 return MadeChange;
Chris Lattner5a1df382006-03-24 23:10:39 +0000640 } else if (getOperator() == ISE.get_intrinsic_void_sdnode() ||
641 getOperator() == ISE.get_intrinsic_w_chain_sdnode() ||
642 getOperator() == ISE.get_intrinsic_wo_chain_sdnode()) {
643 unsigned IID =
644 dynamic_cast<IntInit*>(getChild(0)->getLeafValue())->getValue();
645 const CodeGenIntrinsic &Int = ISE.getIntrinsicInfo(IID);
646 bool MadeChange = false;
647
648 // Apply the result type to the node.
649 MadeChange = UpdateNodeType(Int.ArgVTs[0], TP);
650
651 if (getNumChildren() != Int.ArgVTs.size())
Chris Lattner2c4e65d2006-03-27 22:21:18 +0000652 TP.error("Intrinsic '" + Int.Name + "' expects " +
Chris Lattner5a1df382006-03-24 23:10:39 +0000653 utostr(Int.ArgVTs.size()-1) + " operands, not " +
654 utostr(getNumChildren()-1) + " operands!");
655
656 // Apply type info to the intrinsic ID.
657 MVT::ValueType PtrTy = ISE.getTargetInfo().getPointerType();
658 MadeChange |= getChild(0)->UpdateNodeType(PtrTy, TP);
659
660 for (unsigned i = 1, e = getNumChildren(); i != e; ++i) {
661 MVT::ValueType OpVT = Int.ArgVTs[i];
662 MadeChange |= getChild(i)->UpdateNodeType(OpVT, TP);
663 MadeChange |= getChild(i)->ApplyTypeConstraints(TP, NotRegisters);
664 }
665 return MadeChange;
Chris Lattnerabbb6052005-09-15 21:42:00 +0000666 } else if (getOperator()->isSubClassOf("SDNode")) {
Chris Lattner5a1df382006-03-24 23:10:39 +0000667 const SDNodeInfo &NI = ISE.getSDNodeInfo(getOperator());
Chris Lattnerabbb6052005-09-15 21:42:00 +0000668
669 bool MadeChange = NI.ApplyTypeConstraints(this, TP);
670 for (unsigned i = 0, e = getNumChildren(); i != e; ++i)
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000671 MadeChange |= getChild(i)->ApplyTypeConstraints(TP, NotRegisters);
Evan Cheng1c3d19e2005-12-04 08:18:16 +0000672 // Branch, etc. do not produce results and top-level forms in instr pattern
673 // must have void types.
674 if (NI.getNumResults() == 0)
675 MadeChange |= UpdateNodeType(MVT::isVoid, TP);
Chris Lattnerabbb6052005-09-15 21:42:00 +0000676 return MadeChange;
Chris Lattnera28aec12005-09-15 22:23:50 +0000677 } else if (getOperator()->isSubClassOf("Instruction")) {
Chris Lattner5a1df382006-03-24 23:10:39 +0000678 const DAGInstruction &Inst = ISE.getInstruction(getOperator());
Evan Cheng1c3d19e2005-12-04 08:18:16 +0000679 bool MadeChange = false;
680 unsigned NumResults = Inst.getNumResults();
Chris Lattnerae5b3502005-09-15 21:57:35 +0000681
Evan Cheng1c3d19e2005-12-04 08:18:16 +0000682 assert(NumResults <= 1 &&
683 "Only supports zero or one result instrs!");
Chris Lattnera28aec12005-09-15 22:23:50 +0000684 // Apply the result type to the node
Evan Cheng1c3d19e2005-12-04 08:18:16 +0000685 if (NumResults == 0) {
686 MadeChange = UpdateNodeType(MVT::isVoid, TP);
687 } else {
688 Record *ResultNode = Inst.getResult(0);
689 assert(ResultNode->isSubClassOf("RegisterClass") &&
690 "Operands should be register classes!");
Nate Begemanddb39542005-12-01 00:06:14 +0000691
Evan Cheng1c3d19e2005-12-04 08:18:16 +0000692 const CodeGenRegisterClass &RC =
Chris Lattner5a1df382006-03-24 23:10:39 +0000693 ISE.getTargetInfo().getRegisterClass(ResultNode);
Nate Begemanb73628b2005-12-30 00:12:56 +0000694 MadeChange = UpdateNodeType(ConvertVTs(RC.getValueTypes()), TP);
Evan Cheng1c3d19e2005-12-04 08:18:16 +0000695 }
Chris Lattnera28aec12005-09-15 22:23:50 +0000696
697 if (getNumChildren() != Inst.getNumOperands())
698 TP.error("Instruction '" + getOperator()->getName() + " expects " +
699 utostr(Inst.getNumOperands()) + " operands, not " +
700 utostr(getNumChildren()) + " operands!");
701 for (unsigned i = 0, e = getNumChildren(); i != e; ++i) {
Nate Begemanddb39542005-12-01 00:06:14 +0000702 Record *OperandNode = Inst.getOperand(i);
703 MVT::ValueType VT;
704 if (OperandNode->isSubClassOf("RegisterClass")) {
705 const CodeGenRegisterClass &RC =
Chris Lattner5a1df382006-03-24 23:10:39 +0000706 ISE.getTargetInfo().getRegisterClass(OperandNode);
Nate Begemanb73628b2005-12-30 00:12:56 +0000707 //VT = RC.getValueTypeNum(0);
708 MadeChange |=getChild(i)->UpdateNodeType(ConvertVTs(RC.getValueTypes()),
709 TP);
Nate Begemanddb39542005-12-01 00:06:14 +0000710 } else if (OperandNode->isSubClassOf("Operand")) {
711 VT = getValueType(OperandNode->getValueAsDef("Type"));
Nate Begemanb73628b2005-12-30 00:12:56 +0000712 MadeChange |= getChild(i)->UpdateNodeType(VT, TP);
Nate Begemanddb39542005-12-01 00:06:14 +0000713 } else {
714 assert(0 && "Unknown operand type!");
715 abort();
716 }
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000717 MadeChange |= getChild(i)->ApplyTypeConstraints(TP, NotRegisters);
Chris Lattnera28aec12005-09-15 22:23:50 +0000718 }
719 return MadeChange;
720 } else {
721 assert(getOperator()->isSubClassOf("SDNodeXForm") && "Unknown node type!");
722
Evan Chengf26ba692006-03-20 08:09:17 +0000723 // Node transforms always take one operand.
Chris Lattnera28aec12005-09-15 22:23:50 +0000724 if (getNumChildren() != 1)
725 TP.error("Node transform '" + getOperator()->getName() +
726 "' requires one operand!");
Chris Lattner4e2f54d2006-03-21 06:42:58 +0000727
728 // If either the output or input of the xform does not have exact
729 // type info. We assume they must be the same. Otherwise, it is perfectly
730 // legal to transform from one type to a completely different type.
731 if (!hasTypeSet() || !getChild(0)->hasTypeSet()) {
Evan Chengf26ba692006-03-20 08:09:17 +0000732 bool MadeChange = UpdateNodeType(getChild(0)->getExtTypes(), TP);
733 MadeChange |= getChild(0)->UpdateNodeType(getExtTypes(), TP);
734 return MadeChange;
735 }
736 return false;
Chris Lattner32707602005-09-08 23:22:48 +0000737 }
Chris Lattner32707602005-09-08 23:22:48 +0000738}
739
Chris Lattnere97603f2005-09-28 19:27:25 +0000740/// canPatternMatch - If it is impossible for this pattern to match on this
741/// target, fill in Reason and return false. Otherwise, return true. This is
742/// used as a santity check for .td files (to prevent people from writing stuff
743/// that can never possibly work), and to prevent the pattern permuter from
744/// generating stuff that is useless.
Chris Lattner7cf2fe62005-09-28 20:58:06 +0000745bool TreePatternNode::canPatternMatch(std::string &Reason, DAGISelEmitter &ISE){
Chris Lattnere97603f2005-09-28 19:27:25 +0000746 if (isLeaf()) return true;
747
748 for (unsigned i = 0, e = getNumChildren(); i != e; ++i)
749 if (!getChild(i)->canPatternMatch(Reason, ISE))
750 return false;
Evan Cheng0fc71982005-12-08 02:00:36 +0000751
Chris Lattner550525e2006-03-24 21:48:51 +0000752 // If this is an intrinsic, handle cases that would make it not match. For
753 // example, if an operand is required to be an immediate.
754 if (getOperator()->isSubClassOf("Intrinsic")) {
755 // TODO:
756 return true;
757 }
758
Chris Lattnere97603f2005-09-28 19:27:25 +0000759 // If this node is a commutative operator, check that the LHS isn't an
760 // immediate.
761 const SDNodeInfo &NodeInfo = ISE.getSDNodeInfo(getOperator());
762 if (NodeInfo.hasProperty(SDNodeInfo::SDNPCommutative)) {
763 // Scan all of the operands of the node and make sure that only the last one
764 // is a constant node.
765 for (unsigned i = 0, e = getNumChildren()-1; i != e; ++i)
766 if (!getChild(i)->isLeaf() &&
767 getChild(i)->getOperator()->getName() == "imm") {
768 Reason = "Immediate value must be on the RHS of commutative operators!";
769 return false;
770 }
771 }
772
773 return true;
774}
Chris Lattner32707602005-09-08 23:22:48 +0000775
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000776//===----------------------------------------------------------------------===//
777// TreePattern implementation
778//
779
Chris Lattneredbd8712005-10-21 01:19:59 +0000780TreePattern::TreePattern(Record *TheRec, ListInit *RawPat, bool isInput,
Chris Lattneree9f0c32005-09-13 21:20:49 +0000781 DAGISelEmitter &ise) : TheRecord(TheRec), ISE(ise) {
Chris Lattneredbd8712005-10-21 01:19:59 +0000782 isInputPattern = isInput;
Chris Lattnera28aec12005-09-15 22:23:50 +0000783 for (unsigned i = 0, e = RawPat->getSize(); i != e; ++i)
784 Trees.push_back(ParseTreePattern((DagInit*)RawPat->getElement(i)));
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000785}
786
Chris Lattneredbd8712005-10-21 01:19:59 +0000787TreePattern::TreePattern(Record *TheRec, DagInit *Pat, bool isInput,
Chris Lattnera28aec12005-09-15 22:23:50 +0000788 DAGISelEmitter &ise) : TheRecord(TheRec), ISE(ise) {
Chris Lattneredbd8712005-10-21 01:19:59 +0000789 isInputPattern = isInput;
Chris Lattnera28aec12005-09-15 22:23:50 +0000790 Trees.push_back(ParseTreePattern(Pat));
791}
792
Chris Lattneredbd8712005-10-21 01:19:59 +0000793TreePattern::TreePattern(Record *TheRec, TreePatternNode *Pat, bool isInput,
Chris Lattnera28aec12005-09-15 22:23:50 +0000794 DAGISelEmitter &ise) : TheRecord(TheRec), ISE(ise) {
Chris Lattneredbd8712005-10-21 01:19:59 +0000795 isInputPattern = isInput;
Chris Lattnera28aec12005-09-15 22:23:50 +0000796 Trees.push_back(Pat);
797}
798
799
800
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000801void TreePattern::error(const std::string &Msg) const {
Chris Lattnerd8a3bde2005-09-14 20:53:42 +0000802 dump();
Chris Lattneree9f0c32005-09-13 21:20:49 +0000803 throw "In " + TheRecord->getName() + ": " + Msg;
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000804}
805
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000806TreePatternNode *TreePattern::ParseTreePattern(DagInit *Dag) {
Chris Lattner8c063182006-03-30 22:50:40 +0000807 DefInit *OpDef = dynamic_cast<DefInit*>(Dag->getOperator());
808 if (!OpDef) error("Pattern has unexpected operator type!");
809 Record *Operator = OpDef->getDef();
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000810
811 if (Operator->isSubClassOf("ValueType")) {
812 // If the operator is a ValueType, then this must be "type cast" of a leaf
813 // node.
814 if (Dag->getNumArgs() != 1)
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000815 error("Type cast only takes one operand!");
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000816
817 Init *Arg = Dag->getArg(0);
818 TreePatternNode *New;
819 if (DefInit *DI = dynamic_cast<DefInit*>(Arg)) {
Chris Lattner72fe91c2005-09-24 00:40:24 +0000820 Record *R = DI->getDef();
821 if (R->isSubClassOf("SDNode") || R->isSubClassOf("PatFrag")) {
Chris Lattner8c063182006-03-30 22:50:40 +0000822 Dag->setArg(0, new DagInit(DI,
Chris Lattner72fe91c2005-09-24 00:40:24 +0000823 std::vector<std::pair<Init*, std::string> >()));
Chris Lattner12cf9092005-11-16 23:14:54 +0000824 return ParseTreePattern(Dag);
Evan Cheng1c3d19e2005-12-04 08:18:16 +0000825 }
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000826 New = new TreePatternNode(DI);
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000827 } else if (DagInit *DI = dynamic_cast<DagInit*>(Arg)) {
828 New = ParseTreePattern(DI);
Chris Lattner0614b622005-11-02 06:49:14 +0000829 } else if (IntInit *II = dynamic_cast<IntInit*>(Arg)) {
830 New = new TreePatternNode(II);
831 if (!Dag->getArgName(0).empty())
832 error("Constant int argument should not have a name!");
Chris Lattnerffa4fdc2006-03-31 05:25:56 +0000833 } else if (BitsInit *BI = dynamic_cast<BitsInit*>(Arg)) {
834 // Turn this into an IntInit.
835 Init *II = BI->convertInitializerTo(new IntRecTy());
836 if (II == 0 || !dynamic_cast<IntInit*>(II))
837 error("Bits value must be constants!");
838
839 New = new TreePatternNode(dynamic_cast<IntInit*>(II));
840 if (!Dag->getArgName(0).empty())
841 error("Constant int argument should not have a name!");
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000842 } else {
843 Arg->dump();
844 error("Unknown leaf value for tree pattern!");
845 return 0;
846 }
847
Chris Lattner32707602005-09-08 23:22:48 +0000848 // Apply the type cast.
849 New->UpdateNodeType(getValueType(Operator), *this);
Chris Lattner12cf9092005-11-16 23:14:54 +0000850 New->setName(Dag->getArgName(0));
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000851 return New;
852 }
853
854 // Verify that this is something that makes sense for an operator.
855 if (!Operator->isSubClassOf("PatFrag") && !Operator->isSubClassOf("SDNode") &&
Chris Lattnerabbb6052005-09-15 21:42:00 +0000856 !Operator->isSubClassOf("Instruction") &&
857 !Operator->isSubClassOf("SDNodeXForm") &&
Chris Lattner550525e2006-03-24 21:48:51 +0000858 !Operator->isSubClassOf("Intrinsic") &&
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000859 Operator->getName() != "set")
860 error("Unrecognized node '" + Operator->getName() + "'!");
861
Chris Lattneredbd8712005-10-21 01:19:59 +0000862 // Check to see if this is something that is illegal in an input pattern.
863 if (isInputPattern && (Operator->isSubClassOf("Instruction") ||
Chris Lattner5a1df382006-03-24 23:10:39 +0000864 Operator->isSubClassOf("SDNodeXForm")))
Chris Lattneredbd8712005-10-21 01:19:59 +0000865 error("Cannot use '" + Operator->getName() + "' in an input pattern!");
866
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000867 std::vector<TreePatternNode*> Children;
868
869 for (unsigned i = 0, e = Dag->getNumArgs(); i != e; ++i) {
870 Init *Arg = Dag->getArg(i);
871 if (DagInit *DI = dynamic_cast<DagInit*>(Arg)) {
872 Children.push_back(ParseTreePattern(DI));
Chris Lattner12cf9092005-11-16 23:14:54 +0000873 if (Children.back()->getName().empty())
874 Children.back()->setName(Dag->getArgName(i));
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000875 } else if (DefInit *DefI = dynamic_cast<DefInit*>(Arg)) {
876 Record *R = DefI->getDef();
877 // Direct reference to a leaf DagNode or PatFrag? Turn it into a
878 // TreePatternNode if its own.
879 if (R->isSubClassOf("SDNode") || R->isSubClassOf("PatFrag")) {
Chris Lattner8c063182006-03-30 22:50:40 +0000880 Dag->setArg(i, new DagInit(DefI,
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000881 std::vector<std::pair<Init*, std::string> >()));
882 --i; // Revisit this node...
883 } else {
884 TreePatternNode *Node = new TreePatternNode(DefI);
885 Node->setName(Dag->getArgName(i));
886 Children.push_back(Node);
887
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000888 // Input argument?
889 if (R->getName() == "node") {
890 if (Dag->getArgName(i).empty())
891 error("'node' argument requires a name to match with operand list");
892 Args.push_back(Dag->getArgName(i));
893 }
894 }
Chris Lattner5d5a0562005-10-19 04:30:56 +0000895 } else if (IntInit *II = dynamic_cast<IntInit*>(Arg)) {
896 TreePatternNode *Node = new TreePatternNode(II);
897 if (!Dag->getArgName(i).empty())
898 error("Constant int argument should not have a name!");
899 Children.push_back(Node);
Chris Lattnerffa4fdc2006-03-31 05:25:56 +0000900 } else if (BitsInit *BI = dynamic_cast<BitsInit*>(Arg)) {
901 // Turn this into an IntInit.
902 Init *II = BI->convertInitializerTo(new IntRecTy());
903 if (II == 0 || !dynamic_cast<IntInit*>(II))
904 error("Bits value must be constants!");
905
906 TreePatternNode *Node = new TreePatternNode(dynamic_cast<IntInit*>(II));
907 if (!Dag->getArgName(i).empty())
908 error("Constant int argument should not have a name!");
909 Children.push_back(Node);
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000910 } else {
Chris Lattner5d5a0562005-10-19 04:30:56 +0000911 std::cerr << '"';
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000912 Arg->dump();
Chris Lattner5d5a0562005-10-19 04:30:56 +0000913 std::cerr << "\": ";
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000914 error("Unknown leaf value for tree pattern!");
915 }
916 }
917
Chris Lattner5a1df382006-03-24 23:10:39 +0000918 // If the operator is an intrinsic, then this is just syntactic sugar for for
919 // (intrinsic_* <number>, ..children..). Pick the right intrinsic node, and
920 // convert the intrinsic name to a number.
921 if (Operator->isSubClassOf("Intrinsic")) {
922 const CodeGenIntrinsic &Int = getDAGISelEmitter().getIntrinsic(Operator);
923 unsigned IID = getDAGISelEmitter().getIntrinsicID(Operator)+1;
924
925 // If this intrinsic returns void, it must have side-effects and thus a
926 // chain.
927 if (Int.ArgVTs[0] == MVT::isVoid) {
928 Operator = getDAGISelEmitter().get_intrinsic_void_sdnode();
929 } else if (Int.ModRef != CodeGenIntrinsic::NoMem) {
930 // Has side-effects, requires chain.
931 Operator = getDAGISelEmitter().get_intrinsic_w_chain_sdnode();
932 } else {
933 // Otherwise, no chain.
934 Operator = getDAGISelEmitter().get_intrinsic_wo_chain_sdnode();
935 }
936
937 TreePatternNode *IIDNode = new TreePatternNode(new IntInit(IID));
938 Children.insert(Children.begin(), IIDNode);
939 }
940
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000941 return new TreePatternNode(Operator, Children);
942}
943
Chris Lattner32707602005-09-08 23:22:48 +0000944/// InferAllTypes - Infer/propagate as many types throughout the expression
945/// patterns as possible. Return true if all types are infered, false
946/// otherwise. Throw an exception if a type contradiction is found.
947bool TreePattern::InferAllTypes() {
948 bool MadeChange = true;
949 while (MadeChange) {
950 MadeChange = false;
951 for (unsigned i = 0, e = Trees.size(); i != e; ++i)
Chris Lattner0ee7cff2005-10-14 04:11:13 +0000952 MadeChange |= Trees[i]->ApplyTypeConstraints(*this, false);
Chris Lattner32707602005-09-08 23:22:48 +0000953 }
954
955 bool HasUnresolvedTypes = false;
956 for (unsigned i = 0, e = Trees.size(); i != e; ++i)
957 HasUnresolvedTypes |= Trees[i]->ContainsUnresolvedType();
958 return !HasUnresolvedTypes;
959}
960
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000961void TreePattern::print(std::ostream &OS) const {
Chris Lattner54cb8fd2005-09-07 23:44:43 +0000962 OS << getRecord()->getName();
963 if (!Args.empty()) {
964 OS << "(" << Args[0];
965 for (unsigned i = 1, e = Args.size(); i != e; ++i)
966 OS << ", " << Args[i];
967 OS << ")";
968 }
969 OS << ": ";
970
971 if (Trees.size() > 1)
972 OS << "[\n";
973 for (unsigned i = 0, e = Trees.size(); i != e; ++i) {
974 OS << "\t";
975 Trees[i]->print(OS);
976 OS << "\n";
977 }
978
979 if (Trees.size() > 1)
980 OS << "]\n";
981}
982
983void TreePattern::dump() const { print(std::cerr); }
984
985
986
987//===----------------------------------------------------------------------===//
988// DAGISelEmitter implementation
989//
990
Chris Lattnerca559d02005-09-08 21:03:01 +0000991// Parse all of the SDNode definitions for the target, populating SDNodes.
992void DAGISelEmitter::ParseNodeInfo() {
993 std::vector<Record*> Nodes = Records.getAllDerivedDefinitions("SDNode");
994 while (!Nodes.empty()) {
995 SDNodes.insert(std::make_pair(Nodes.back(), Nodes.back()));
996 Nodes.pop_back();
997 }
Chris Lattner5a1df382006-03-24 23:10:39 +0000998
999 // Get the buildin intrinsic nodes.
1000 intrinsic_void_sdnode = getSDNodeNamed("intrinsic_void");
1001 intrinsic_w_chain_sdnode = getSDNodeNamed("intrinsic_w_chain");
1002 intrinsic_wo_chain_sdnode = getSDNodeNamed("intrinsic_wo_chain");
Chris Lattnerca559d02005-09-08 21:03:01 +00001003}
1004
Chris Lattner24eeeb82005-09-13 21:51:00 +00001005/// ParseNodeTransforms - Parse all SDNodeXForm instances into the SDNodeXForms
1006/// map, and emit them to the file as functions.
1007void DAGISelEmitter::ParseNodeTransforms(std::ostream &OS) {
1008 OS << "\n// Node transformations.\n";
1009 std::vector<Record*> Xforms = Records.getAllDerivedDefinitions("SDNodeXForm");
1010 while (!Xforms.empty()) {
1011 Record *XFormNode = Xforms.back();
1012 Record *SDNode = XFormNode->getValueAsDef("Opcode");
1013 std::string Code = XFormNode->getValueAsCode("XFormFunction");
1014 SDNodeXForms.insert(std::make_pair(XFormNode,
1015 std::make_pair(SDNode, Code)));
1016
Chris Lattner1048b7a2005-09-13 22:03:37 +00001017 if (!Code.empty()) {
Chris Lattner24eeeb82005-09-13 21:51:00 +00001018 std::string ClassName = getSDNodeInfo(SDNode).getSDClassName();
1019 const char *C2 = ClassName == "SDNode" ? "N" : "inN";
1020
Chris Lattner1048b7a2005-09-13 22:03:37 +00001021 OS << "inline SDOperand Transform_" << XFormNode->getName()
Chris Lattner24eeeb82005-09-13 21:51:00 +00001022 << "(SDNode *" << C2 << ") {\n";
1023 if (ClassName != "SDNode")
1024 OS << " " << ClassName << " *N = cast<" << ClassName << ">(inN);\n";
1025 OS << Code << "\n}\n";
1026 }
1027
1028 Xforms.pop_back();
1029 }
1030}
1031
Evan Cheng0fc71982005-12-08 02:00:36 +00001032void DAGISelEmitter::ParseComplexPatterns() {
1033 std::vector<Record*> AMs = Records.getAllDerivedDefinitions("ComplexPattern");
1034 while (!AMs.empty()) {
1035 ComplexPatterns.insert(std::make_pair(AMs.back(), AMs.back()));
1036 AMs.pop_back();
1037 }
1038}
Chris Lattner24eeeb82005-09-13 21:51:00 +00001039
1040
Chris Lattnerb39e4be2005-09-15 02:38:02 +00001041/// ParsePatternFragments - Parse all of the PatFrag definitions in the .td
1042/// file, building up the PatternFragments map. After we've collected them all,
1043/// inline fragments together as necessary, so that there are no references left
1044/// inside a pattern fragment to a pattern fragment.
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001045///
1046/// This also emits all of the predicate functions to the output file.
1047///
Chris Lattnerb39e4be2005-09-15 02:38:02 +00001048void DAGISelEmitter::ParsePatternFragments(std::ostream &OS) {
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001049 std::vector<Record*> Fragments = Records.getAllDerivedDefinitions("PatFrag");
1050
1051 // First step, parse all of the fragments and emit predicate functions.
1052 OS << "\n// Predicate functions.\n";
1053 for (unsigned i = 0, e = Fragments.size(); i != e; ++i) {
Chris Lattnera28aec12005-09-15 22:23:50 +00001054 DagInit *Tree = Fragments[i]->getValueAsDag("Fragment");
Chris Lattneredbd8712005-10-21 01:19:59 +00001055 TreePattern *P = new TreePattern(Fragments[i], Tree, true, *this);
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001056 PatternFragments[Fragments[i]] = P;
Chris Lattneree9f0c32005-09-13 21:20:49 +00001057
1058 // Validate the argument list, converting it to map, to discard duplicates.
1059 std::vector<std::string> &Args = P->getArgList();
1060 std::set<std::string> OperandsMap(Args.begin(), Args.end());
1061
1062 if (OperandsMap.count(""))
1063 P->error("Cannot have unnamed 'node' values in pattern fragment!");
1064
1065 // Parse the operands list.
1066 DagInit *OpsList = Fragments[i]->getValueAsDag("Operands");
Chris Lattner8c063182006-03-30 22:50:40 +00001067 DefInit *OpsOp = dynamic_cast<DefInit*>(OpsList->getOperator());
1068 if (!OpsOp || OpsOp->getDef()->getName() != "ops")
Chris Lattneree9f0c32005-09-13 21:20:49 +00001069 P->error("Operands list should start with '(ops ... '!");
1070
1071 // Copy over the arguments.
1072 Args.clear();
1073 for (unsigned j = 0, e = OpsList->getNumArgs(); j != e; ++j) {
1074 if (!dynamic_cast<DefInit*>(OpsList->getArg(j)) ||
1075 static_cast<DefInit*>(OpsList->getArg(j))->
1076 getDef()->getName() != "node")
1077 P->error("Operands list should all be 'node' values.");
1078 if (OpsList->getArgName(j).empty())
1079 P->error("Operands list should have names for each operand!");
1080 if (!OperandsMap.count(OpsList->getArgName(j)))
1081 P->error("'" + OpsList->getArgName(j) +
1082 "' does not occur in pattern or was multiply specified!");
1083 OperandsMap.erase(OpsList->getArgName(j));
1084 Args.push_back(OpsList->getArgName(j));
1085 }
1086
1087 if (!OperandsMap.empty())
1088 P->error("Operands list does not contain an entry for operand '" +
1089 *OperandsMap.begin() + "'!");
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001090
1091 // If there is a code init for this fragment, emit the predicate code and
1092 // keep track of the fact that this fragment uses it.
Chris Lattner24eeeb82005-09-13 21:51:00 +00001093 std::string Code = Fragments[i]->getValueAsCode("Predicate");
1094 if (!Code.empty()) {
Chris Lattner37937092005-09-09 01:15:01 +00001095 assert(!P->getOnlyTree()->isLeaf() && "Can't be a leaf!");
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001096 std::string ClassName =
Chris Lattner37937092005-09-09 01:15:01 +00001097 getSDNodeInfo(P->getOnlyTree()->getOperator()).getSDClassName();
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001098 const char *C2 = ClassName == "SDNode" ? "N" : "inN";
1099
Chris Lattner1048b7a2005-09-13 22:03:37 +00001100 OS << "inline bool Predicate_" << Fragments[i]->getName()
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001101 << "(SDNode *" << C2 << ") {\n";
1102 if (ClassName != "SDNode")
1103 OS << " " << ClassName << " *N = cast<" << ClassName << ">(inN);\n";
Chris Lattner24eeeb82005-09-13 21:51:00 +00001104 OS << Code << "\n}\n";
Chris Lattner37937092005-09-09 01:15:01 +00001105 P->getOnlyTree()->setPredicateFn("Predicate_"+Fragments[i]->getName());
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001106 }
Chris Lattner6de8b532005-09-13 21:59:15 +00001107
1108 // If there is a node transformation corresponding to this, keep track of
1109 // it.
1110 Record *Transform = Fragments[i]->getValueAsDef("OperandTransform");
1111 if (!getSDNodeTransform(Transform).second.empty()) // not noop xform?
Chris Lattnerb0276202005-09-14 22:55:26 +00001112 P->getOnlyTree()->setTransformFn(Transform);
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001113 }
1114
1115 OS << "\n\n";
1116
1117 // Now that we've parsed all of the tree fragments, do a closure on them so
1118 // that there are not references to PatFrags left inside of them.
1119 for (std::map<Record*, TreePattern*>::iterator I = PatternFragments.begin(),
1120 E = PatternFragments.end(); I != E; ++I) {
Chris Lattner32707602005-09-08 23:22:48 +00001121 TreePattern *ThePat = I->second;
1122 ThePat->InlinePatternFragments();
Chris Lattneree9f0c32005-09-13 21:20:49 +00001123
Chris Lattner32707602005-09-08 23:22:48 +00001124 // Infer as many types as possible. Don't worry about it if we don't infer
1125 // all of them, some may depend on the inputs of the pattern.
1126 try {
1127 ThePat->InferAllTypes();
1128 } catch (...) {
1129 // If this pattern fragment is not supported by this target (no types can
1130 // satisfy its constraints), just ignore it. If the bogus pattern is
1131 // actually used by instructions, the type consistency error will be
1132 // reported there.
1133 }
1134
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001135 // If debugging, print out the pattern fragment result.
Chris Lattner32707602005-09-08 23:22:48 +00001136 DEBUG(ThePat->dump());
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001137 }
1138}
1139
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001140/// HandleUse - Given "Pat" a leaf in the pattern, check to see if it is an
Chris Lattnerf1311842005-09-14 23:05:13 +00001141/// instruction input. Return true if this is a real use.
1142static bool HandleUse(TreePattern *I, TreePatternNode *Pat,
Evan Cheng7b05bd52005-12-23 22:11:47 +00001143 std::map<std::string, TreePatternNode*> &InstInputs,
1144 std::vector<Record*> &InstImpInputs) {
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001145 // No name -> not interesting.
Chris Lattner7da852f2005-09-14 22:06:36 +00001146 if (Pat->getName().empty()) {
1147 if (Pat->isLeaf()) {
1148 DefInit *DI = dynamic_cast<DefInit*>(Pat->getLeafValue());
1149 if (DI && DI->getDef()->isSubClassOf("RegisterClass"))
1150 I->error("Input " + DI->getDef()->getName() + " must be named!");
Evan Cheng7b05bd52005-12-23 22:11:47 +00001151 else if (DI && DI->getDef()->isSubClassOf("Register"))
1152 InstImpInputs.push_back(DI->getDef());
Chris Lattner7da852f2005-09-14 22:06:36 +00001153 }
Chris Lattnerf1311842005-09-14 23:05:13 +00001154 return false;
Chris Lattner7da852f2005-09-14 22:06:36 +00001155 }
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001156
1157 Record *Rec;
1158 if (Pat->isLeaf()) {
1159 DefInit *DI = dynamic_cast<DefInit*>(Pat->getLeafValue());
1160 if (!DI) I->error("Input $" + Pat->getName() + " must be an identifier!");
1161 Rec = DI->getDef();
1162 } else {
1163 assert(Pat->getNumChildren() == 0 && "can't be a use with children!");
1164 Rec = Pat->getOperator();
1165 }
1166
Evan Cheng01f318b2005-12-14 02:21:57 +00001167 // SRCVALUE nodes are ignored.
1168 if (Rec->getName() == "srcvalue")
1169 return false;
1170
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001171 TreePatternNode *&Slot = InstInputs[Pat->getName()];
1172 if (!Slot) {
1173 Slot = Pat;
1174 } else {
1175 Record *SlotRec;
1176 if (Slot->isLeaf()) {
Chris Lattnerb9f01eb2005-09-16 00:29:46 +00001177 SlotRec = dynamic_cast<DefInit*>(Slot->getLeafValue())->getDef();
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001178 } else {
1179 assert(Slot->getNumChildren() == 0 && "can't be a use with children!");
1180 SlotRec = Slot->getOperator();
1181 }
1182
1183 // Ensure that the inputs agree if we've already seen this input.
1184 if (Rec != SlotRec)
1185 I->error("All $" + Pat->getName() + " inputs must agree with each other");
Nate Begemanb73628b2005-12-30 00:12:56 +00001186 if (Slot->getExtTypes() != Pat->getExtTypes())
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001187 I->error("All $" + Pat->getName() + " inputs must agree with each other");
1188 }
Chris Lattnerf1311842005-09-14 23:05:13 +00001189 return true;
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001190}
1191
1192/// FindPatternInputsAndOutputs - Scan the specified TreePatternNode (which is
1193/// part of "I", the instruction), computing the set of inputs and outputs of
1194/// the pattern. Report errors if we see anything naughty.
1195void DAGISelEmitter::
1196FindPatternInputsAndOutputs(TreePattern *I, TreePatternNode *Pat,
1197 std::map<std::string, TreePatternNode*> &InstInputs,
Chris Lattner947604b2006-03-24 21:52:20 +00001198 std::map<std::string, TreePatternNode*>&InstResults,
Evan Cheng7b05bd52005-12-23 22:11:47 +00001199 std::vector<Record*> &InstImpInputs,
Evan Chengbcecf332005-12-17 01:19:28 +00001200 std::vector<Record*> &InstImpResults) {
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001201 if (Pat->isLeaf()) {
Evan Cheng7b05bd52005-12-23 22:11:47 +00001202 bool isUse = HandleUse(I, Pat, InstInputs, InstImpInputs);
Chris Lattnerf1311842005-09-14 23:05:13 +00001203 if (!isUse && Pat->getTransformFn())
1204 I->error("Cannot specify a transform function for a non-input value!");
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001205 return;
1206 } else if (Pat->getOperator()->getName() != "set") {
1207 // If this is not a set, verify that the children nodes are not void typed,
1208 // and recurse.
1209 for (unsigned i = 0, e = Pat->getNumChildren(); i != e; ++i) {
Nate Begemanb73628b2005-12-30 00:12:56 +00001210 if (Pat->getChild(i)->getExtTypeNum(0) == MVT::isVoid)
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001211 I->error("Cannot have void nodes inside of patterns!");
Evan Chengbcecf332005-12-17 01:19:28 +00001212 FindPatternInputsAndOutputs(I, Pat->getChild(i), InstInputs, InstResults,
Evan Cheng7b05bd52005-12-23 22:11:47 +00001213 InstImpInputs, InstImpResults);
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001214 }
1215
1216 // If this is a non-leaf node with no children, treat it basically as if
1217 // it were a leaf. This handles nodes like (imm).
Chris Lattnerf1311842005-09-14 23:05:13 +00001218 bool isUse = false;
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001219 if (Pat->getNumChildren() == 0)
Evan Cheng7b05bd52005-12-23 22:11:47 +00001220 isUse = HandleUse(I, Pat, InstInputs, InstImpInputs);
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001221
Chris Lattnerf1311842005-09-14 23:05:13 +00001222 if (!isUse && Pat->getTransformFn())
1223 I->error("Cannot specify a transform function for a non-input value!");
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001224 return;
1225 }
1226
1227 // Otherwise, this is a set, validate and collect instruction results.
1228 if (Pat->getNumChildren() == 0)
1229 I->error("set requires operands!");
1230 else if (Pat->getNumChildren() & 1)
1231 I->error("set requires an even number of operands");
1232
Chris Lattnerf1311842005-09-14 23:05:13 +00001233 if (Pat->getTransformFn())
1234 I->error("Cannot specify a transform function on a set node!");
1235
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001236 // Check the set destinations.
1237 unsigned NumValues = Pat->getNumChildren()/2;
1238 for (unsigned i = 0; i != NumValues; ++i) {
1239 TreePatternNode *Dest = Pat->getChild(i);
1240 if (!Dest->isLeaf())
Evan Cheng86217892005-12-12 19:37:43 +00001241 I->error("set destination should be a register!");
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001242
1243 DefInit *Val = dynamic_cast<DefInit*>(Dest->getLeafValue());
1244 if (!Val)
Evan Cheng86217892005-12-12 19:37:43 +00001245 I->error("set destination should be a register!");
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001246
Evan Chengbcecf332005-12-17 01:19:28 +00001247 if (Val->getDef()->isSubClassOf("RegisterClass")) {
1248 if (Dest->getName().empty())
1249 I->error("set destination must have a name!");
1250 if (InstResults.count(Dest->getName()))
1251 I->error("cannot set '" + Dest->getName() +"' multiple times");
Evan Cheng420132e2006-03-20 06:04:09 +00001252 InstResults[Dest->getName()] = Dest;
Evan Cheng7b05bd52005-12-23 22:11:47 +00001253 } else if (Val->getDef()->isSubClassOf("Register")) {
Evan Chengbcecf332005-12-17 01:19:28 +00001254 InstImpResults.push_back(Val->getDef());
1255 } else {
1256 I->error("set destination should be a register!");
1257 }
1258
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001259 // Verify and collect info from the computation.
1260 FindPatternInputsAndOutputs(I, Pat->getChild(i+NumValues),
Evan Cheng7b05bd52005-12-23 22:11:47 +00001261 InstInputs, InstResults,
1262 InstImpInputs, InstImpResults);
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001263 }
1264}
1265
Chris Lattnerb39e4be2005-09-15 02:38:02 +00001266/// ParseInstructions - Parse all of the instructions, inlining and resolving
1267/// any fragments involved. This populates the Instructions list with fully
1268/// resolved instructions.
1269void DAGISelEmitter::ParseInstructions() {
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001270 std::vector<Record*> Instrs = Records.getAllDerivedDefinitions("Instruction");
1271
1272 for (unsigned i = 0, e = Instrs.size(); i != e; ++i) {
Chris Lattner0c0cfa72005-10-19 01:27:22 +00001273 ListInit *LI = 0;
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001274
Chris Lattner0c0cfa72005-10-19 01:27:22 +00001275 if (dynamic_cast<ListInit*>(Instrs[i]->getValueInit("Pattern")))
1276 LI = Instrs[i]->getValueAsListInit("Pattern");
1277
1278 // If there is no pattern, only collect minimal information about the
1279 // instruction for its operand list. We have to assume that there is one
1280 // result, as we have no detailed info.
1281 if (!LI || LI->getSize() == 0) {
Nate Begemanddb39542005-12-01 00:06:14 +00001282 std::vector<Record*> Results;
1283 std::vector<Record*> Operands;
Chris Lattner0c0cfa72005-10-19 01:27:22 +00001284
1285 CodeGenInstruction &InstInfo =Target.getInstruction(Instrs[i]->getName());
Evan Cheng1c3d19e2005-12-04 08:18:16 +00001286
Evan Cheng3a217f32005-12-22 02:35:21 +00001287 if (InstInfo.OperandList.size() != 0) {
Evan Cheng3a217f32005-12-22 02:35:21 +00001288 // FIXME: temporary hack...
Evan Cheng2b4ea792005-12-26 09:11:45 +00001289 if (InstInfo.noResults) {
Evan Cheng3a217f32005-12-22 02:35:21 +00001290 // These produce no results
1291 for (unsigned j = 0, e = InstInfo.OperandList.size(); j < e; ++j)
1292 Operands.push_back(InstInfo.OperandList[j].Rec);
1293 } else {
1294 // Assume the first operand is the result.
1295 Results.push_back(InstInfo.OperandList[0].Rec);
Chris Lattner0c0cfa72005-10-19 01:27:22 +00001296
Evan Cheng3a217f32005-12-22 02:35:21 +00001297 // The rest are inputs.
1298 for (unsigned j = 1, e = InstInfo.OperandList.size(); j < e; ++j)
1299 Operands.push_back(InstInfo.OperandList[j].Rec);
1300 }
Evan Cheng1c3d19e2005-12-04 08:18:16 +00001301 }
Chris Lattner0c0cfa72005-10-19 01:27:22 +00001302
1303 // Create and insert the instruction.
Evan Chengbcecf332005-12-17 01:19:28 +00001304 std::vector<Record*> ImpResults;
1305 std::vector<Record*> ImpOperands;
Chris Lattner0c0cfa72005-10-19 01:27:22 +00001306 Instructions.insert(std::make_pair(Instrs[i],
Evan Cheng7b05bd52005-12-23 22:11:47 +00001307 DAGInstruction(0, Results, Operands, ImpResults,
1308 ImpOperands)));
Chris Lattner0c0cfa72005-10-19 01:27:22 +00001309 continue; // no pattern.
1310 }
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001311
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001312 // Parse the instruction.
Chris Lattneredbd8712005-10-21 01:19:59 +00001313 TreePattern *I = new TreePattern(Instrs[i], LI, true, *this);
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001314 // Inline pattern fragments into it.
Chris Lattner32707602005-09-08 23:22:48 +00001315 I->InlinePatternFragments();
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001316
Chris Lattner95f6b762005-09-08 23:26:30 +00001317 // Infer as many types as possible. If we cannot infer all of them, we can
1318 // never do anything with this instruction pattern: report it to the user.
Chris Lattnerabbb6052005-09-15 21:42:00 +00001319 if (!I->InferAllTypes())
Chris Lattner32707602005-09-08 23:22:48 +00001320 I->error("Could not infer all types in pattern!");
Chris Lattnerf2a17a72005-09-09 01:11:44 +00001321
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001322 // InstInputs - Keep track of all of the inputs of the instruction, along
1323 // with the record they are declared as.
1324 std::map<std::string, TreePatternNode*> InstInputs;
1325
1326 // InstResults - Keep track of all the virtual registers that are 'set'
Chris Lattner5f8cb2a2005-09-14 02:11:12 +00001327 // in the instruction, including what reg class they are.
Evan Cheng420132e2006-03-20 06:04:09 +00001328 std::map<std::string, TreePatternNode*> InstResults;
Evan Cheng7b05bd52005-12-23 22:11:47 +00001329
1330 std::vector<Record*> InstImpInputs;
Evan Chengbcecf332005-12-17 01:19:28 +00001331 std::vector<Record*> InstImpResults;
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001332
Chris Lattner1f39e292005-09-14 00:09:24 +00001333 // Verify that the top-level forms in the instruction are of void type, and
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001334 // fill in the InstResults map.
Chris Lattner1f39e292005-09-14 00:09:24 +00001335 for (unsigned j = 0, e = I->getNumTrees(); j != e; ++j) {
1336 TreePatternNode *Pat = I->getTree(j);
Nate Begemanb73628b2005-12-30 00:12:56 +00001337 if (Pat->getExtTypeNum(0) != MVT::isVoid)
Chris Lattnerf2a17a72005-09-09 01:11:44 +00001338 I->error("Top-level forms in instruction pattern should have"
1339 " void types");
Chris Lattner5f8cb2a2005-09-14 02:11:12 +00001340
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001341 // Find inputs and outputs, and verify the structure of the uses/defs.
Evan Chengbcecf332005-12-17 01:19:28 +00001342 FindPatternInputsAndOutputs(I, Pat, InstInputs, InstResults,
Evan Cheng7b05bd52005-12-23 22:11:47 +00001343 InstImpInputs, InstImpResults);
Chris Lattner1f39e292005-09-14 00:09:24 +00001344 }
Chris Lattner5f8cb2a2005-09-14 02:11:12 +00001345
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001346 // Now that we have inputs and outputs of the pattern, inspect the operands
1347 // list for the instruction. This determines the order that operands are
1348 // added to the machine instruction the node corresponds to.
1349 unsigned NumResults = InstResults.size();
Chris Lattner39e8af92005-09-14 18:19:25 +00001350
1351 // Parse the operands list from the (ops) list, validating it.
1352 std::vector<std::string> &Args = I->getArgList();
1353 assert(Args.empty() && "Args list should still be empty here!");
1354 CodeGenInstruction &CGI = Target.getInstruction(Instrs[i]->getName());
1355
1356 // Check that all of the results occur first in the list.
Nate Begemanddb39542005-12-01 00:06:14 +00001357 std::vector<Record*> Results;
Evan Cheng420132e2006-03-20 06:04:09 +00001358 TreePatternNode *Res0Node = NULL;
Chris Lattner39e8af92005-09-14 18:19:25 +00001359 for (unsigned i = 0; i != NumResults; ++i) {
Chris Lattner3a7319d2005-09-14 21:04:12 +00001360 if (i == CGI.OperandList.size())
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001361 I->error("'" + InstResults.begin()->first +
1362 "' set but does not appear in operand list!");
Chris Lattner39e8af92005-09-14 18:19:25 +00001363 const std::string &OpName = CGI.OperandList[i].Name;
Chris Lattner39e8af92005-09-14 18:19:25 +00001364
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001365 // Check that it exists in InstResults.
Evan Cheng420132e2006-03-20 06:04:09 +00001366 TreePatternNode *RNode = InstResults[OpName];
Chris Lattner5c4c7742006-03-25 22:12:44 +00001367 if (RNode == 0)
1368 I->error("Operand $" + OpName + " does not exist in operand list!");
1369
Evan Cheng420132e2006-03-20 06:04:09 +00001370 if (i == 0)
1371 Res0Node = RNode;
1372 Record *R = dynamic_cast<DefInit*>(RNode->getLeafValue())->getDef();
Chris Lattner39e8af92005-09-14 18:19:25 +00001373 if (R == 0)
1374 I->error("Operand $" + OpName + " should be a set destination: all "
1375 "outputs must occur before inputs in operand list!");
1376
1377 if (CGI.OperandList[i].Rec != R)
1378 I->error("Operand $" + OpName + " class mismatch!");
1379
Chris Lattnerae6d8282005-09-15 21:51:12 +00001380 // Remember the return type.
Nate Begemanddb39542005-12-01 00:06:14 +00001381 Results.push_back(CGI.OperandList[i].Rec);
Chris Lattnerae6d8282005-09-15 21:51:12 +00001382
Chris Lattner39e8af92005-09-14 18:19:25 +00001383 // Okay, this one checks out.
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001384 InstResults.erase(OpName);
1385 }
1386
Chris Lattner0b592252005-09-14 21:59:34 +00001387 // Loop over the inputs next. Make a copy of InstInputs so we can destroy
1388 // the copy while we're checking the inputs.
1389 std::map<std::string, TreePatternNode*> InstInputsCheck(InstInputs);
Chris Lattnerb0276202005-09-14 22:55:26 +00001390
1391 std::vector<TreePatternNode*> ResultNodeOperands;
Nate Begemanddb39542005-12-01 00:06:14 +00001392 std::vector<Record*> Operands;
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001393 for (unsigned i = NumResults, e = CGI.OperandList.size(); i != e; ++i) {
1394 const std::string &OpName = CGI.OperandList[i].Name;
1395 if (OpName.empty())
1396 I->error("Operand #" + utostr(i) + " in operands list has no name!");
1397
Chris Lattner0b592252005-09-14 21:59:34 +00001398 if (!InstInputsCheck.count(OpName))
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001399 I->error("Operand $" + OpName +
1400 " does not appear in the instruction pattern");
Chris Lattner0b592252005-09-14 21:59:34 +00001401 TreePatternNode *InVal = InstInputsCheck[OpName];
Chris Lattnerb0276202005-09-14 22:55:26 +00001402 InstInputsCheck.erase(OpName); // It occurred, remove from map.
Nate Begemanddb39542005-12-01 00:06:14 +00001403
1404 if (InVal->isLeaf() &&
1405 dynamic_cast<DefInit*>(InVal->getLeafValue())) {
1406 Record *InRec = static_cast<DefInit*>(InVal->getLeafValue())->getDef();
Evan Cheng0fc71982005-12-08 02:00:36 +00001407 if (CGI.OperandList[i].Rec != InRec &&
1408 !InRec->isSubClassOf("ComplexPattern"))
Chris Lattner488580c2006-01-28 19:06:51 +00001409 I->error("Operand $" + OpName + "'s register class disagrees"
1410 " between the operand and pattern");
Nate Begemanddb39542005-12-01 00:06:14 +00001411 }
1412 Operands.push_back(CGI.OperandList[i].Rec);
Chris Lattnerb0276202005-09-14 22:55:26 +00001413
Chris Lattner2175c182005-09-14 23:01:59 +00001414 // Construct the result for the dest-pattern operand list.
1415 TreePatternNode *OpNode = InVal->clone();
1416
1417 // No predicate is useful on the result.
1418 OpNode->setPredicateFn("");
1419
1420 // Promote the xform function to be an explicit node if set.
1421 if (Record *Xform = OpNode->getTransformFn()) {
1422 OpNode->setTransformFn(0);
1423 std::vector<TreePatternNode*> Children;
1424 Children.push_back(OpNode);
1425 OpNode = new TreePatternNode(Xform, Children);
1426 }
1427
1428 ResultNodeOperands.push_back(OpNode);
Chris Lattner39e8af92005-09-14 18:19:25 +00001429 }
1430
Chris Lattner0b592252005-09-14 21:59:34 +00001431 if (!InstInputsCheck.empty())
1432 I->error("Input operand $" + InstInputsCheck.begin()->first +
1433 " occurs in pattern but not in operands list!");
Chris Lattnerb0276202005-09-14 22:55:26 +00001434
1435 TreePatternNode *ResultPattern =
1436 new TreePatternNode(I->getRecord(), ResultNodeOperands);
Evan Cheng420132e2006-03-20 06:04:09 +00001437 // Copy fully inferred output node type to instruction result pattern.
1438 if (NumResults > 0)
1439 ResultPattern->setTypes(Res0Node->getExtTypes());
Chris Lattnera28aec12005-09-15 22:23:50 +00001440
1441 // Create and insert the instruction.
Evan Cheng7b05bd52005-12-23 22:11:47 +00001442 DAGInstruction TheInst(I, Results, Operands, InstImpResults, InstImpInputs);
Chris Lattnera28aec12005-09-15 22:23:50 +00001443 Instructions.insert(std::make_pair(I->getRecord(), TheInst));
1444
1445 // Use a temporary tree pattern to infer all types and make sure that the
1446 // constructed result is correct. This depends on the instruction already
1447 // being inserted into the Instructions map.
Chris Lattneredbd8712005-10-21 01:19:59 +00001448 TreePattern Temp(I->getRecord(), ResultPattern, false, *this);
Chris Lattnera28aec12005-09-15 22:23:50 +00001449 Temp.InferAllTypes();
1450
1451 DAGInstruction &TheInsertedInst = Instructions.find(I->getRecord())->second;
1452 TheInsertedInst.setResultPattern(Temp.getOnlyTree());
Chris Lattnerb0276202005-09-14 22:55:26 +00001453
Chris Lattner32707602005-09-08 23:22:48 +00001454 DEBUG(I->dump());
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001455 }
Chris Lattner1f39e292005-09-14 00:09:24 +00001456
Chris Lattnerd8a3bde2005-09-14 20:53:42 +00001457 // If we can, convert the instructions to be patterns that are matched!
Chris Lattnerae5b3502005-09-15 21:57:35 +00001458 for (std::map<Record*, DAGInstruction>::iterator II = Instructions.begin(),
1459 E = Instructions.end(); II != E; ++II) {
Evan Cheng1c3d19e2005-12-04 08:18:16 +00001460 DAGInstruction &TheInst = II->second;
1461 TreePattern *I = TheInst.getPattern();
Chris Lattner0c0cfa72005-10-19 01:27:22 +00001462 if (I == 0) continue; // No pattern.
Evan Chengdd304dd2005-12-05 23:08:55 +00001463
Chris Lattner1f39e292005-09-14 00:09:24 +00001464 if (I->getNumTrees() != 1) {
1465 std::cerr << "CANNOT HANDLE: " << I->getRecord()->getName() << " yet!";
1466 continue;
1467 }
1468 TreePatternNode *Pattern = I->getTree(0);
Evan Cheng1c3d19e2005-12-04 08:18:16 +00001469 TreePatternNode *SrcPattern;
Evan Chengbcecf332005-12-17 01:19:28 +00001470 if (Pattern->getOperator()->getName() == "set") {
Evan Cheng1c3d19e2005-12-04 08:18:16 +00001471 if (Pattern->getNumChildren() != 2)
1472 continue; // Not a set of a single value (not handled so far)
1473
1474 SrcPattern = Pattern->getChild(1)->clone();
Evan Chengbcecf332005-12-17 01:19:28 +00001475 } else{
1476 // Not a set (store or something?)
1477 SrcPattern = Pattern;
Evan Cheng1c3d19e2005-12-04 08:18:16 +00001478 }
Chris Lattnere97603f2005-09-28 19:27:25 +00001479
1480 std::string Reason;
1481 if (!SrcPattern->canPatternMatch(Reason, *this))
1482 I->error("Instruction can never match: " + Reason);
1483
Evan Cheng58e84a62005-12-14 22:02:59 +00001484 Record *Instr = II->first;
Evan Cheng1c3d19e2005-12-04 08:18:16 +00001485 TreePatternNode *DstPattern = TheInst.getResultPattern();
Evan Cheng58e84a62005-12-14 22:02:59 +00001486 PatternsToMatch.
1487 push_back(PatternToMatch(Instr->getValueAsListInit("Predicates"),
1488 SrcPattern, DstPattern));
Chris Lattner1f39e292005-09-14 00:09:24 +00001489 }
Chris Lattner54cb8fd2005-09-07 23:44:43 +00001490}
1491
Chris Lattnerb39e4be2005-09-15 02:38:02 +00001492void DAGISelEmitter::ParsePatterns() {
Chris Lattnerabbb6052005-09-15 21:42:00 +00001493 std::vector<Record*> Patterns = Records.getAllDerivedDefinitions("Pattern");
Chris Lattnerb39e4be2005-09-15 02:38:02 +00001494
Chris Lattnerabbb6052005-09-15 21:42:00 +00001495 for (unsigned i = 0, e = Patterns.size(); i != e; ++i) {
Chris Lattnera28aec12005-09-15 22:23:50 +00001496 DagInit *Tree = Patterns[i]->getValueAsDag("PatternToMatch");
Chris Lattneredbd8712005-10-21 01:19:59 +00001497 TreePattern *Pattern = new TreePattern(Patterns[i], Tree, true, *this);
Chris Lattnerb39e4be2005-09-15 02:38:02 +00001498
Chris Lattnerabbb6052005-09-15 21:42:00 +00001499 // Inline pattern fragments into it.
1500 Pattern->InlinePatternFragments();
1501
1502 // Infer as many types as possible. If we cannot infer all of them, we can
1503 // never do anything with this pattern: report it to the user.
1504 if (!Pattern->InferAllTypes())
1505 Pattern->error("Could not infer all types in pattern!");
Chris Lattner09c03392005-11-17 17:43:52 +00001506
1507 // Validate that the input pattern is correct.
1508 {
1509 std::map<std::string, TreePatternNode*> InstInputs;
Evan Cheng420132e2006-03-20 06:04:09 +00001510 std::map<std::string, TreePatternNode*> InstResults;
Evan Cheng7b05bd52005-12-23 22:11:47 +00001511 std::vector<Record*> InstImpInputs;
Evan Chengbcecf332005-12-17 01:19:28 +00001512 std::vector<Record*> InstImpResults;
Chris Lattner09c03392005-11-17 17:43:52 +00001513 FindPatternInputsAndOutputs(Pattern, Pattern->getOnlyTree(),
Evan Chengbcecf332005-12-17 01:19:28 +00001514 InstInputs, InstResults,
Evan Cheng7b05bd52005-12-23 22:11:47 +00001515 InstImpInputs, InstImpResults);
Chris Lattner09c03392005-11-17 17:43:52 +00001516 }
Chris Lattnerabbb6052005-09-15 21:42:00 +00001517
1518 ListInit *LI = Patterns[i]->getValueAsListInit("ResultInstrs");
1519 if (LI->getSize() == 0) continue; // no pattern.
Chris Lattnerabbb6052005-09-15 21:42:00 +00001520
1521 // Parse the instruction.
Chris Lattneredbd8712005-10-21 01:19:59 +00001522 TreePattern *Result = new TreePattern(Patterns[i], LI, false, *this);
Chris Lattnerabbb6052005-09-15 21:42:00 +00001523
1524 // Inline pattern fragments into it.
1525 Result->InlinePatternFragments();
1526
1527 // Infer as many types as possible. If we cannot infer all of them, we can
1528 // never do anything with this pattern: report it to the user.
Chris Lattnerabbb6052005-09-15 21:42:00 +00001529 if (!Result->InferAllTypes())
Chris Lattnerae5b3502005-09-15 21:57:35 +00001530 Result->error("Could not infer all types in pattern result!");
Chris Lattnerabbb6052005-09-15 21:42:00 +00001531
1532 if (Result->getNumTrees() != 1)
1533 Result->error("Cannot handle instructions producing instructions "
1534 "with temporaries yet!");
Chris Lattnere97603f2005-09-28 19:27:25 +00001535
Evan Cheng3a7a14b2006-03-21 20:44:17 +00001536 // Promote the xform function to be an explicit node if set.
1537 std::vector<TreePatternNode*> ResultNodeOperands;
1538 TreePatternNode *DstPattern = Result->getOnlyTree();
1539 for (unsigned ii = 0, ee = DstPattern->getNumChildren(); ii != ee; ++ii) {
1540 TreePatternNode *OpNode = DstPattern->getChild(ii);
1541 if (Record *Xform = OpNode->getTransformFn()) {
1542 OpNode->setTransformFn(0);
1543 std::vector<TreePatternNode*> Children;
1544 Children.push_back(OpNode);
1545 OpNode = new TreePatternNode(Xform, Children);
1546 }
1547 ResultNodeOperands.push_back(OpNode);
1548 }
Evan Cheng83e1a6a2006-03-23 02:35:32 +00001549 DstPattern = Result->getOnlyTree();
1550 if (!DstPattern->isLeaf())
1551 DstPattern = new TreePatternNode(DstPattern->getOperator(),
1552 ResultNodeOperands);
Evan Cheng3a7a14b2006-03-21 20:44:17 +00001553 DstPattern->setTypes(Result->getOnlyTree()->getExtTypes());
1554 TreePattern Temp(Result->getRecord(), DstPattern, false, *this);
1555 Temp.InferAllTypes();
1556
Chris Lattnere97603f2005-09-28 19:27:25 +00001557 std::string Reason;
1558 if (!Pattern->getOnlyTree()->canPatternMatch(Reason, *this))
1559 Pattern->error("Pattern can never match: " + Reason);
1560
Evan Cheng58e84a62005-12-14 22:02:59 +00001561 PatternsToMatch.
1562 push_back(PatternToMatch(Patterns[i]->getValueAsListInit("Predicates"),
1563 Pattern->getOnlyTree(),
Evan Cheng3a7a14b2006-03-21 20:44:17 +00001564 Temp.getOnlyTree()));
Chris Lattnerabbb6052005-09-15 21:42:00 +00001565 }
Chris Lattnerb39e4be2005-09-15 02:38:02 +00001566}
1567
Chris Lattnere46e17b2005-09-29 19:28:10 +00001568/// CombineChildVariants - Given a bunch of permutations of each child of the
1569/// 'operator' node, put them together in all possible ways.
1570static void CombineChildVariants(TreePatternNode *Orig,
Chris Lattneraf302912005-09-29 22:36:54 +00001571 const std::vector<std::vector<TreePatternNode*> > &ChildVariants,
Chris Lattnere46e17b2005-09-29 19:28:10 +00001572 std::vector<TreePatternNode*> &OutVariants,
1573 DAGISelEmitter &ISE) {
Chris Lattneraf302912005-09-29 22:36:54 +00001574 // Make sure that each operand has at least one variant to choose from.
1575 for (unsigned i = 0, e = ChildVariants.size(); i != e; ++i)
1576 if (ChildVariants[i].empty())
1577 return;
1578
Chris Lattnere46e17b2005-09-29 19:28:10 +00001579 // The end result is an all-pairs construction of the resultant pattern.
1580 std::vector<unsigned> Idxs;
1581 Idxs.resize(ChildVariants.size());
1582 bool NotDone = true;
1583 while (NotDone) {
1584 // Create the variant and add it to the output list.
1585 std::vector<TreePatternNode*> NewChildren;
1586 for (unsigned i = 0, e = ChildVariants.size(); i != e; ++i)
1587 NewChildren.push_back(ChildVariants[i][Idxs[i]]);
1588 TreePatternNode *R = new TreePatternNode(Orig->getOperator(), NewChildren);
1589
1590 // Copy over properties.
1591 R->setName(Orig->getName());
1592 R->setPredicateFn(Orig->getPredicateFn());
1593 R->setTransformFn(Orig->getTransformFn());
Nate Begemanb73628b2005-12-30 00:12:56 +00001594 R->setTypes(Orig->getExtTypes());
Chris Lattnere46e17b2005-09-29 19:28:10 +00001595
1596 // If this pattern cannot every match, do not include it as a variant.
1597 std::string ErrString;
1598 if (!R->canPatternMatch(ErrString, ISE)) {
1599 delete R;
1600 } else {
1601 bool AlreadyExists = false;
1602
1603 // Scan to see if this pattern has already been emitted. We can get
1604 // duplication due to things like commuting:
1605 // (and GPRC:$a, GPRC:$b) -> (and GPRC:$b, GPRC:$a)
1606 // which are the same pattern. Ignore the dups.
1607 for (unsigned i = 0, e = OutVariants.size(); i != e; ++i)
1608 if (R->isIsomorphicTo(OutVariants[i])) {
1609 AlreadyExists = true;
1610 break;
1611 }
1612
1613 if (AlreadyExists)
1614 delete R;
1615 else
1616 OutVariants.push_back(R);
1617 }
1618
1619 // Increment indices to the next permutation.
1620 NotDone = false;
1621 // Look for something we can increment without causing a wrap-around.
1622 for (unsigned IdxsIdx = 0; IdxsIdx != Idxs.size(); ++IdxsIdx) {
1623 if (++Idxs[IdxsIdx] < ChildVariants[IdxsIdx].size()) {
1624 NotDone = true; // Found something to increment.
1625 break;
1626 }
1627 Idxs[IdxsIdx] = 0;
1628 }
1629 }
1630}
1631
Chris Lattneraf302912005-09-29 22:36:54 +00001632/// CombineChildVariants - A helper function for binary operators.
1633///
1634static void CombineChildVariants(TreePatternNode *Orig,
1635 const std::vector<TreePatternNode*> &LHS,
1636 const std::vector<TreePatternNode*> &RHS,
1637 std::vector<TreePatternNode*> &OutVariants,
1638 DAGISelEmitter &ISE) {
1639 std::vector<std::vector<TreePatternNode*> > ChildVariants;
1640 ChildVariants.push_back(LHS);
1641 ChildVariants.push_back(RHS);
1642 CombineChildVariants(Orig, ChildVariants, OutVariants, ISE);
1643}
1644
1645
1646static void GatherChildrenOfAssociativeOpcode(TreePatternNode *N,
1647 std::vector<TreePatternNode *> &Children) {
1648 assert(N->getNumChildren()==2 &&"Associative but doesn't have 2 children!");
1649 Record *Operator = N->getOperator();
1650
1651 // Only permit raw nodes.
1652 if (!N->getName().empty() || !N->getPredicateFn().empty() ||
1653 N->getTransformFn()) {
1654 Children.push_back(N);
1655 return;
1656 }
1657
1658 if (N->getChild(0)->isLeaf() || N->getChild(0)->getOperator() != Operator)
1659 Children.push_back(N->getChild(0));
1660 else
1661 GatherChildrenOfAssociativeOpcode(N->getChild(0), Children);
1662
1663 if (N->getChild(1)->isLeaf() || N->getChild(1)->getOperator() != Operator)
1664 Children.push_back(N->getChild(1));
1665 else
1666 GatherChildrenOfAssociativeOpcode(N->getChild(1), Children);
1667}
1668
Chris Lattnere46e17b2005-09-29 19:28:10 +00001669/// GenerateVariantsOf - Given a pattern N, generate all permutations we can of
1670/// the (potentially recursive) pattern by using algebraic laws.
1671///
1672static void GenerateVariantsOf(TreePatternNode *N,
1673 std::vector<TreePatternNode*> &OutVariants,
1674 DAGISelEmitter &ISE) {
1675 // We cannot permute leaves.
1676 if (N->isLeaf()) {
1677 OutVariants.push_back(N);
1678 return;
1679 }
1680
1681 // Look up interesting info about the node.
Chris Lattner5a1df382006-03-24 23:10:39 +00001682 const SDNodeInfo &NodeInfo = ISE.getSDNodeInfo(N->getOperator());
Chris Lattnere46e17b2005-09-29 19:28:10 +00001683
1684 // If this node is associative, reassociate.
Chris Lattner5a1df382006-03-24 23:10:39 +00001685 if (NodeInfo.hasProperty(SDNodeInfo::SDNPAssociative)) {
Chris Lattneraf302912005-09-29 22:36:54 +00001686 // Reassociate by pulling together all of the linked operators
1687 std::vector<TreePatternNode*> MaximalChildren;
1688 GatherChildrenOfAssociativeOpcode(N, MaximalChildren);
1689
1690 // Only handle child sizes of 3. Otherwise we'll end up trying too many
1691 // permutations.
1692 if (MaximalChildren.size() == 3) {
1693 // Find the variants of all of our maximal children.
1694 std::vector<TreePatternNode*> AVariants, BVariants, CVariants;
1695 GenerateVariantsOf(MaximalChildren[0], AVariants, ISE);
1696 GenerateVariantsOf(MaximalChildren[1], BVariants, ISE);
1697 GenerateVariantsOf(MaximalChildren[2], CVariants, ISE);
1698
1699 // There are only two ways we can permute the tree:
1700 // (A op B) op C and A op (B op C)
1701 // Within these forms, we can also permute A/B/C.
1702
1703 // Generate legal pair permutations of A/B/C.
1704 std::vector<TreePatternNode*> ABVariants;
1705 std::vector<TreePatternNode*> BAVariants;
1706 std::vector<TreePatternNode*> ACVariants;
1707 std::vector<TreePatternNode*> CAVariants;
1708 std::vector<TreePatternNode*> BCVariants;
1709 std::vector<TreePatternNode*> CBVariants;
1710 CombineChildVariants(N, AVariants, BVariants, ABVariants, ISE);
1711 CombineChildVariants(N, BVariants, AVariants, BAVariants, ISE);
1712 CombineChildVariants(N, AVariants, CVariants, ACVariants, ISE);
1713 CombineChildVariants(N, CVariants, AVariants, CAVariants, ISE);
1714 CombineChildVariants(N, BVariants, CVariants, BCVariants, ISE);
1715 CombineChildVariants(N, CVariants, BVariants, CBVariants, ISE);
1716
1717 // Combine those into the result: (x op x) op x
1718 CombineChildVariants(N, ABVariants, CVariants, OutVariants, ISE);
1719 CombineChildVariants(N, BAVariants, CVariants, OutVariants, ISE);
1720 CombineChildVariants(N, ACVariants, BVariants, OutVariants, ISE);
1721 CombineChildVariants(N, CAVariants, BVariants, OutVariants, ISE);
1722 CombineChildVariants(N, BCVariants, AVariants, OutVariants, ISE);
1723 CombineChildVariants(N, CBVariants, AVariants, OutVariants, ISE);
1724
1725 // Combine those into the result: x op (x op x)
1726 CombineChildVariants(N, CVariants, ABVariants, OutVariants, ISE);
1727 CombineChildVariants(N, CVariants, BAVariants, OutVariants, ISE);
1728 CombineChildVariants(N, BVariants, ACVariants, OutVariants, ISE);
1729 CombineChildVariants(N, BVariants, CAVariants, OutVariants, ISE);
1730 CombineChildVariants(N, AVariants, BCVariants, OutVariants, ISE);
1731 CombineChildVariants(N, AVariants, CBVariants, OutVariants, ISE);
1732 return;
1733 }
1734 }
Chris Lattnere46e17b2005-09-29 19:28:10 +00001735
1736 // Compute permutations of all children.
1737 std::vector<std::vector<TreePatternNode*> > ChildVariants;
1738 ChildVariants.resize(N->getNumChildren());
1739 for (unsigned i = 0, e = N->getNumChildren(); i != e; ++i)
1740 GenerateVariantsOf(N->getChild(i), ChildVariants[i], ISE);
1741
1742 // Build all permutations based on how the children were formed.
1743 CombineChildVariants(N, ChildVariants, OutVariants, ISE);
1744
1745 // If this node is commutative, consider the commuted order.
Chris Lattner5a1df382006-03-24 23:10:39 +00001746 if (NodeInfo.hasProperty(SDNodeInfo::SDNPCommutative)) {
Chris Lattnere46e17b2005-09-29 19:28:10 +00001747 assert(N->getNumChildren()==2 &&"Commutative but doesn't have 2 children!");
Chris Lattneraf302912005-09-29 22:36:54 +00001748 // Consider the commuted order.
1749 CombineChildVariants(N, ChildVariants[1], ChildVariants[0],
1750 OutVariants, ISE);
Chris Lattnere46e17b2005-09-29 19:28:10 +00001751 }
1752}
1753
1754
Chris Lattnere97603f2005-09-28 19:27:25 +00001755// GenerateVariants - Generate variants. For example, commutative patterns can
1756// match multiple ways. Add them to PatternsToMatch as well.
1757void DAGISelEmitter::GenerateVariants() {
Chris Lattnere46e17b2005-09-29 19:28:10 +00001758
1759 DEBUG(std::cerr << "Generating instruction variants.\n");
1760
1761 // Loop over all of the patterns we've collected, checking to see if we can
1762 // generate variants of the instruction, through the exploitation of
1763 // identities. This permits the target to provide agressive matching without
1764 // the .td file having to contain tons of variants of instructions.
1765 //
1766 // Note that this loop adds new patterns to the PatternsToMatch list, but we
1767 // intentionally do not reconsider these. Any variants of added patterns have
1768 // already been added.
1769 //
1770 for (unsigned i = 0, e = PatternsToMatch.size(); i != e; ++i) {
1771 std::vector<TreePatternNode*> Variants;
Evan Cheng58e84a62005-12-14 22:02:59 +00001772 GenerateVariantsOf(PatternsToMatch[i].getSrcPattern(), Variants, *this);
Chris Lattnere46e17b2005-09-29 19:28:10 +00001773
1774 assert(!Variants.empty() && "Must create at least original variant!");
Chris Lattnere46e17b2005-09-29 19:28:10 +00001775 Variants.erase(Variants.begin()); // Remove the original pattern.
1776
1777 if (Variants.empty()) // No variants for this pattern.
1778 continue;
1779
1780 DEBUG(std::cerr << "FOUND VARIANTS OF: ";
Evan Cheng58e84a62005-12-14 22:02:59 +00001781 PatternsToMatch[i].getSrcPattern()->dump();
Chris Lattnere46e17b2005-09-29 19:28:10 +00001782 std::cerr << "\n");
1783
1784 for (unsigned v = 0, e = Variants.size(); v != e; ++v) {
1785 TreePatternNode *Variant = Variants[v];
1786
1787 DEBUG(std::cerr << " VAR#" << v << ": ";
1788 Variant->dump();
1789 std::cerr << "\n");
1790
1791 // Scan to see if an instruction or explicit pattern already matches this.
1792 bool AlreadyExists = false;
1793 for (unsigned p = 0, e = PatternsToMatch.size(); p != e; ++p) {
1794 // Check to see if this variant already exists.
Evan Cheng58e84a62005-12-14 22:02:59 +00001795 if (Variant->isIsomorphicTo(PatternsToMatch[p].getSrcPattern())) {
Chris Lattnere46e17b2005-09-29 19:28:10 +00001796 DEBUG(std::cerr << " *** ALREADY EXISTS, ignoring variant.\n");
1797 AlreadyExists = true;
1798 break;
1799 }
1800 }
1801 // If we already have it, ignore the variant.
1802 if (AlreadyExists) continue;
1803
1804 // Otherwise, add it to the list of patterns we have.
Evan Cheng58e84a62005-12-14 22:02:59 +00001805 PatternsToMatch.
1806 push_back(PatternToMatch(PatternsToMatch[i].getPredicates(),
1807 Variant, PatternsToMatch[i].getDstPattern()));
Chris Lattnere46e17b2005-09-29 19:28:10 +00001808 }
1809
1810 DEBUG(std::cerr << "\n");
1811 }
Chris Lattnere97603f2005-09-28 19:27:25 +00001812}
1813
Chris Lattner7cf2fe62005-09-28 20:58:06 +00001814
Evan Cheng0fc71982005-12-08 02:00:36 +00001815// NodeIsComplexPattern - return true if N is a leaf node and a subclass of
1816// ComplexPattern.
1817static bool NodeIsComplexPattern(TreePatternNode *N)
1818{
1819 return (N->isLeaf() &&
1820 dynamic_cast<DefInit*>(N->getLeafValue()) &&
1821 static_cast<DefInit*>(N->getLeafValue())->getDef()->
1822 isSubClassOf("ComplexPattern"));
1823}
1824
1825// NodeGetComplexPattern - return the pointer to the ComplexPattern if N
1826// is a leaf node and a subclass of ComplexPattern, else it returns NULL.
1827static const ComplexPattern *NodeGetComplexPattern(TreePatternNode *N,
1828 DAGISelEmitter &ISE)
1829{
1830 if (N->isLeaf() &&
1831 dynamic_cast<DefInit*>(N->getLeafValue()) &&
1832 static_cast<DefInit*>(N->getLeafValue())->getDef()->
1833 isSubClassOf("ComplexPattern")) {
1834 return &ISE.getComplexPattern(static_cast<DefInit*>(N->getLeafValue())
1835 ->getDef());
1836 }
1837 return NULL;
1838}
1839
Chris Lattner05814af2005-09-28 17:57:56 +00001840/// getPatternSize - Return the 'size' of this pattern. We want to match large
1841/// patterns before small ones. This is used to determine the size of a
1842/// pattern.
Evan Cheng0fc71982005-12-08 02:00:36 +00001843static unsigned getPatternSize(TreePatternNode *P, DAGISelEmitter &ISE) {
Evan Cheng4a7c2842006-01-06 22:19:44 +00001844 assert(isExtIntegerInVTs(P->getExtTypes()) ||
1845 isExtFloatingPointInVTs(P->getExtTypes()) ||
1846 P->getExtTypeNum(0) == MVT::isVoid ||
1847 P->getExtTypeNum(0) == MVT::Flag &&
1848 "Not a valid pattern node to size!");
Evan Chenge1050d62006-01-06 02:30:23 +00001849 unsigned Size = 2; // The node itself.
Evan Cheng657416c2006-02-01 06:06:31 +00001850 // If the root node is a ConstantSDNode, increases its size.
1851 // e.g. (set R32:$dst, 0).
1852 if (P->isLeaf() && dynamic_cast<IntInit*>(P->getLeafValue()))
1853 Size++;
Evan Cheng0fc71982005-12-08 02:00:36 +00001854
1855 // FIXME: This is a hack to statically increase the priority of patterns
1856 // which maps a sub-dag to a complex pattern. e.g. favors LEA over ADD.
1857 // Later we can allow complexity / cost for each pattern to be (optionally)
1858 // specified. To get best possible pattern match we'll need to dynamically
1859 // calculate the complexity of all patterns a dag can potentially map to.
1860 const ComplexPattern *AM = NodeGetComplexPattern(P, ISE);
1861 if (AM)
Evan Cheng4a7c2842006-01-06 22:19:44 +00001862 Size += AM->getNumOperands() * 2;
Chris Lattner3e179802006-02-03 18:06:02 +00001863
1864 // If this node has some predicate function that must match, it adds to the
1865 // complexity of this node.
1866 if (!P->getPredicateFn().empty())
1867 ++Size;
1868
Chris Lattner05814af2005-09-28 17:57:56 +00001869 // Count children in the count if they are also nodes.
1870 for (unsigned i = 0, e = P->getNumChildren(); i != e; ++i) {
1871 TreePatternNode *Child = P->getChild(i);
Nate Begemanb73628b2005-12-30 00:12:56 +00001872 if (!Child->isLeaf() && Child->getExtTypeNum(0) != MVT::Other)
Evan Cheng0fc71982005-12-08 02:00:36 +00001873 Size += getPatternSize(Child, ISE);
1874 else if (Child->isLeaf()) {
1875 if (dynamic_cast<IntInit*>(Child->getLeafValue()))
Chris Lattner3e179802006-02-03 18:06:02 +00001876 Size += 3; // Matches a ConstantSDNode (+2) and a specific value (+1).
Evan Cheng4a7c2842006-01-06 22:19:44 +00001877 else if (NodeIsComplexPattern(Child))
1878 Size += getPatternSize(Child, ISE);
Chris Lattner3e179802006-02-03 18:06:02 +00001879 else if (!Child->getPredicateFn().empty())
1880 ++Size;
Chris Lattner2f041d42005-10-19 04:41:05 +00001881 }
Chris Lattner05814af2005-09-28 17:57:56 +00001882 }
1883
1884 return Size;
1885}
1886
1887/// getResultPatternCost - Compute the number of instructions for this pattern.
1888/// This is a temporary hack. We should really include the instruction
1889/// latencies in this calculation.
Evan Chengfbad7082006-02-18 02:33:09 +00001890static unsigned getResultPatternCost(TreePatternNode *P, DAGISelEmitter &ISE) {
Chris Lattner05814af2005-09-28 17:57:56 +00001891 if (P->isLeaf()) return 0;
1892
Evan Chengfbad7082006-02-18 02:33:09 +00001893 unsigned Cost = 0;
1894 Record *Op = P->getOperator();
1895 if (Op->isSubClassOf("Instruction")) {
1896 Cost++;
1897 CodeGenInstruction &II = ISE.getTargetInfo().getInstruction(Op->getName());
1898 if (II.usesCustomDAGSchedInserter)
1899 Cost += 10;
1900 }
Chris Lattner05814af2005-09-28 17:57:56 +00001901 for (unsigned i = 0, e = P->getNumChildren(); i != e; ++i)
Evan Chengfbad7082006-02-18 02:33:09 +00001902 Cost += getResultPatternCost(P->getChild(i), ISE);
Chris Lattner05814af2005-09-28 17:57:56 +00001903 return Cost;
1904}
1905
1906// PatternSortingPredicate - return true if we prefer to match LHS before RHS.
1907// In particular, we want to match maximal patterns first and lowest cost within
1908// a particular complexity first.
1909struct PatternSortingPredicate {
Evan Cheng0fc71982005-12-08 02:00:36 +00001910 PatternSortingPredicate(DAGISelEmitter &ise) : ISE(ise) {};
1911 DAGISelEmitter &ISE;
1912
Evan Cheng58e84a62005-12-14 22:02:59 +00001913 bool operator()(PatternToMatch *LHS,
1914 PatternToMatch *RHS) {
1915 unsigned LHSSize = getPatternSize(LHS->getSrcPattern(), ISE);
1916 unsigned RHSSize = getPatternSize(RHS->getSrcPattern(), ISE);
Chris Lattner05814af2005-09-28 17:57:56 +00001917 if (LHSSize > RHSSize) return true; // LHS -> bigger -> less cost
1918 if (LHSSize < RHSSize) return false;
1919
1920 // If the patterns have equal complexity, compare generated instruction cost
Evan Chengfbad7082006-02-18 02:33:09 +00001921 return getResultPatternCost(LHS->getDstPattern(), ISE) <
1922 getResultPatternCost(RHS->getDstPattern(), ISE);
Chris Lattner05814af2005-09-28 17:57:56 +00001923 }
1924};
1925
Nate Begeman6510b222005-12-01 04:51:06 +00001926/// getRegisterValueType - Look up and return the first ValueType of specified
1927/// RegisterClass record
Evan Cheng66a48bb2005-12-01 00:18:45 +00001928static MVT::ValueType getRegisterValueType(Record *R, const CodeGenTarget &T) {
Chris Lattner22faeab2005-12-05 02:36:37 +00001929 if (const CodeGenRegisterClass *RC = T.getRegisterClassForRegister(R))
1930 return RC->getValueTypeNum(0);
Evan Cheng66a48bb2005-12-01 00:18:45 +00001931 return MVT::Other;
1932}
1933
Chris Lattner72fe91c2005-09-24 00:40:24 +00001934
Chris Lattner0ee7cff2005-10-14 04:11:13 +00001935/// RemoveAllTypes - A quick recursive walk over a pattern which removes all
1936/// type information from it.
1937static void RemoveAllTypes(TreePatternNode *N) {
Nate Begemanb73628b2005-12-30 00:12:56 +00001938 N->removeTypes();
Chris Lattner0ee7cff2005-10-14 04:11:13 +00001939 if (!N->isLeaf())
1940 for (unsigned i = 0, e = N->getNumChildren(); i != e; ++i)
1941 RemoveAllTypes(N->getChild(i));
1942}
Chris Lattner72fe91c2005-09-24 00:40:24 +00001943
Chris Lattner0614b622005-11-02 06:49:14 +00001944Record *DAGISelEmitter::getSDNodeNamed(const std::string &Name) const {
1945 Record *N = Records.getDef(Name);
Chris Lattner5a1df382006-03-24 23:10:39 +00001946 if (!N || !N->isSubClassOf("SDNode")) {
1947 std::cerr << "Error getting SDNode '" << Name << "'!\n";
1948 exit(1);
1949 }
Chris Lattner0614b622005-11-02 06:49:14 +00001950 return N;
1951}
1952
Evan Cheng51fecc82006-01-09 18:27:06 +00001953/// NodeHasProperty - return true if TreePatternNode has the specified
1954/// property.
1955static bool NodeHasProperty(TreePatternNode *N, SDNodeInfo::SDNP Property,
1956 DAGISelEmitter &ISE)
Evan Cheng7b05bd52005-12-23 22:11:47 +00001957{
1958 if (N->isLeaf()) return false;
1959 Record *Operator = N->getOperator();
1960 if (!Operator->isSubClassOf("SDNode")) return false;
1961
1962 const SDNodeInfo &NodeInfo = ISE.getSDNodeInfo(Operator);
Evan Cheng51fecc82006-01-09 18:27:06 +00001963 return NodeInfo.hasProperty(Property);
Evan Cheng7b05bd52005-12-23 22:11:47 +00001964}
1965
Evan Cheng51fecc82006-01-09 18:27:06 +00001966static bool PatternHasProperty(TreePatternNode *N, SDNodeInfo::SDNP Property,
1967 DAGISelEmitter &ISE)
Evan Cheng7b05bd52005-12-23 22:11:47 +00001968{
Evan Cheng51fecc82006-01-09 18:27:06 +00001969 if (NodeHasProperty(N, Property, ISE))
Evan Cheng7b05bd52005-12-23 22:11:47 +00001970 return true;
Evan Cheng51fecc82006-01-09 18:27:06 +00001971
1972 for (unsigned i = 0, e = N->getNumChildren(); i != e; ++i) {
1973 TreePatternNode *Child = N->getChild(i);
1974 if (PatternHasProperty(Child, Property, ISE))
1975 return true;
Evan Cheng7b05bd52005-12-23 22:11:47 +00001976 }
1977
1978 return false;
1979}
1980
Evan Chengb915f312005-12-09 22:45:35 +00001981class PatternCodeEmitter {
1982private:
1983 DAGISelEmitter &ISE;
1984
Evan Cheng58e84a62005-12-14 22:02:59 +00001985 // Predicates.
1986 ListInit *Predicates;
1987 // Instruction selector pattern.
1988 TreePatternNode *Pattern;
1989 // Matched instruction.
1990 TreePatternNode *Instruction;
Chris Lattner8a0604b2006-01-28 20:31:24 +00001991
Evan Chengb915f312005-12-09 22:45:35 +00001992 // Node to name mapping
Evan Chengf805c2e2006-01-12 19:35:54 +00001993 std::map<std::string, std::string> VariableMap;
1994 // Node to operator mapping
1995 std::map<std::string, Record*> OperatorMap;
Evan Chengb915f312005-12-09 22:45:35 +00001996 // Names of all the folded nodes which produce chains.
Evan Cheng1b80f4d2005-12-19 07:18:51 +00001997 std::vector<std::pair<std::string, unsigned> > FoldedChains;
Evan Chengb4ad33c2006-01-19 01:55:45 +00001998 std::set<std::string> Duplicates;
Evan Chengb915f312005-12-09 22:45:35 +00001999
Chris Lattner8a0604b2006-01-28 20:31:24 +00002000 /// GeneratedCode - This is the buffer that we emit code to. The first bool
2001 /// indicates whether this is an exit predicate (something that should be
2002 /// tested, and if true, the match fails) [when true] or normal code to emit
2003 /// [when false].
2004 std::vector<std::pair<bool, std::string> > &GeneratedCode;
Evan Cheng21ad3922006-02-07 00:37:41 +00002005 /// GeneratedDecl - This is the set of all SDOperand declarations needed for
2006 /// the set of patterns for each top-level opcode.
Evan Chengd7805a72006-02-09 07:16:09 +00002007 std::set<std::pair<bool, std::string> > &GeneratedDecl;
Chris Lattner8a0604b2006-01-28 20:31:24 +00002008
Evan Chenge4a8a6e2006-02-03 06:22:41 +00002009 std::string ChainName;
Evan Chenged66e852006-03-09 08:19:11 +00002010 bool NewTF;
Evan Chenge41bf822006-02-05 06:43:12 +00002011 bool DoReplace;
Chris Lattner8a0604b2006-01-28 20:31:24 +00002012 unsigned TmpNo;
2013
2014 void emitCheck(const std::string &S) {
2015 if (!S.empty())
2016 GeneratedCode.push_back(std::make_pair(true, S));
2017 }
2018 void emitCode(const std::string &S) {
2019 if (!S.empty())
2020 GeneratedCode.push_back(std::make_pair(false, S));
2021 }
Evan Chengd7805a72006-02-09 07:16:09 +00002022 void emitDecl(const std::string &S, bool isSDNode=false) {
Evan Cheng21ad3922006-02-07 00:37:41 +00002023 assert(!S.empty() && "Invalid declaration");
Evan Chengd7805a72006-02-09 07:16:09 +00002024 GeneratedDecl.insert(std::make_pair(isSDNode, S));
Evan Cheng21ad3922006-02-07 00:37:41 +00002025 }
Evan Chengb915f312005-12-09 22:45:35 +00002026public:
Evan Cheng58e84a62005-12-14 22:02:59 +00002027 PatternCodeEmitter(DAGISelEmitter &ise, ListInit *preds,
2028 TreePatternNode *pattern, TreePatternNode *instr,
Evan Cheng21ad3922006-02-07 00:37:41 +00002029 std::vector<std::pair<bool, std::string> > &gc,
Evan Chengd7805a72006-02-09 07:16:09 +00002030 std::set<std::pair<bool, std::string> > &gd,
Evan Cheng21ad3922006-02-07 00:37:41 +00002031 bool dorep)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002032 : ISE(ise), Predicates(preds), Pattern(pattern), Instruction(instr),
Evan Chenged66e852006-03-09 08:19:11 +00002033 GeneratedCode(gc), GeneratedDecl(gd),
2034 NewTF(false), DoReplace(dorep), TmpNo(0) {}
Evan Chengb915f312005-12-09 22:45:35 +00002035
2036 /// EmitMatchCode - Emit a matcher for N, going to the label for PatternNo
2037 /// if the match fails. At this point, we already know that the opcode for N
2038 /// matches, and the SDNode for the result has the RootName specified name.
Evan Chenge41bf822006-02-05 06:43:12 +00002039 void EmitMatchCode(TreePatternNode *N, TreePatternNode *P,
2040 const std::string &RootName, const std::string &ParentName,
2041 const std::string &ChainSuffix, bool &FoundChain) {
2042 bool isRoot = (P == NULL);
Evan Cheng58e84a62005-12-14 22:02:59 +00002043 // Emit instruction predicates. Each predicate is just a string for now.
2044 if (isRoot) {
Chris Lattner8a0604b2006-01-28 20:31:24 +00002045 std::string PredicateCheck;
Evan Cheng58e84a62005-12-14 22:02:59 +00002046 for (unsigned i = 0, e = Predicates->getSize(); i != e; ++i) {
2047 if (DefInit *Pred = dynamic_cast<DefInit*>(Predicates->getElement(i))) {
2048 Record *Def = Pred->getDef();
Chris Lattner8a0604b2006-01-28 20:31:24 +00002049 if (!Def->isSubClassOf("Predicate")) {
Evan Cheng58e84a62005-12-14 22:02:59 +00002050 Def->dump();
2051 assert(0 && "Unknown predicate type!");
2052 }
Chris Lattner8a0604b2006-01-28 20:31:24 +00002053 if (!PredicateCheck.empty())
Chris Lattner67a202b2006-01-28 20:43:52 +00002054 PredicateCheck += " || ";
2055 PredicateCheck += "(" + Def->getValueAsString("CondString") + ")";
Evan Cheng58e84a62005-12-14 22:02:59 +00002056 }
2057 }
Chris Lattner8a0604b2006-01-28 20:31:24 +00002058
2059 emitCheck(PredicateCheck);
Evan Cheng58e84a62005-12-14 22:02:59 +00002060 }
2061
Evan Chengb915f312005-12-09 22:45:35 +00002062 if (N->isLeaf()) {
2063 if (IntInit *II = dynamic_cast<IntInit*>(N->getLeafValue())) {
Chris Lattner8a0604b2006-01-28 20:31:24 +00002064 emitCheck("cast<ConstantSDNode>(" + RootName +
Chris Lattner67a202b2006-01-28 20:43:52 +00002065 ")->getSignExtended() == " + itostr(II->getValue()));
Evan Chengb915f312005-12-09 22:45:35 +00002066 return;
2067 } else if (!NodeIsComplexPattern(N)) {
2068 assert(0 && "Cannot match this as a leaf value!");
2069 abort();
2070 }
2071 }
2072
Chris Lattner488580c2006-01-28 19:06:51 +00002073 // If this node has a name associated with it, capture it in VariableMap. If
Evan Chengb915f312005-12-09 22:45:35 +00002074 // we already saw this in the pattern, emit code to verify dagness.
2075 if (!N->getName().empty()) {
2076 std::string &VarMapEntry = VariableMap[N->getName()];
2077 if (VarMapEntry.empty()) {
2078 VarMapEntry = RootName;
2079 } else {
2080 // If we get here, this is a second reference to a specific name. Since
2081 // we already have checked that the first reference is valid, we don't
2082 // have to recursively match it, just check that it's the same as the
2083 // previously named thing.
Chris Lattner67a202b2006-01-28 20:43:52 +00002084 emitCheck(VarMapEntry + " == " + RootName);
Evan Chengb915f312005-12-09 22:45:35 +00002085 return;
2086 }
Evan Chengf805c2e2006-01-12 19:35:54 +00002087
2088 if (!N->isLeaf())
2089 OperatorMap[N->getName()] = N->getOperator();
Evan Chengb915f312005-12-09 22:45:35 +00002090 }
2091
2092
2093 // Emit code to load the child nodes and match their contents recursively.
2094 unsigned OpNo = 0;
Evan Chenge41bf822006-02-05 06:43:12 +00002095 bool NodeHasChain = NodeHasProperty (N, SDNodeInfo::SDNPHasChain, ISE);
2096 bool HasChain = PatternHasProperty(N, SDNodeInfo::SDNPHasChain, ISE);
2097 bool HasOutFlag = PatternHasProperty(N, SDNodeInfo::SDNPOutFlag, ISE);
Evan Cheng1feeeec2006-01-26 19:13:45 +00002098 bool EmittedUseCheck = false;
2099 bool EmittedSlctedCheck = false;
Evan Cheng86217892005-12-12 19:37:43 +00002100 if (HasChain) {
Evan Cheng76356d92006-01-20 01:11:03 +00002101 if (NodeHasChain)
2102 OpNo = 1;
Evan Chengb915f312005-12-09 22:45:35 +00002103 if (!isRoot) {
Evan Cheng1129e872005-12-10 00:09:17 +00002104 const SDNodeInfo &CInfo = ISE.getSDNodeInfo(N->getOperator());
Chris Lattner8a0604b2006-01-28 20:31:24 +00002105 // Multiple uses of actual result?
Chris Lattner67a202b2006-01-28 20:43:52 +00002106 emitCheck(RootName + ".hasOneUse()");
Evan Cheng1feeeec2006-01-26 19:13:45 +00002107 EmittedUseCheck = true;
2108 // hasOneUse() check is not strong enough. If the original node has
2109 // already been selected, it may have been replaced with another.
Chris Lattner8a0604b2006-01-28 20:31:24 +00002110 for (unsigned j = 0; j != CInfo.getNumResults(); j++)
Chris Lattner67a202b2006-01-28 20:43:52 +00002111 emitCheck("!CodeGenMap.count(" + RootName + ".getValue(" + utostr(j) +
Chris Lattner8a0604b2006-01-28 20:31:24 +00002112 "))");
2113
Evan Cheng1feeeec2006-01-26 19:13:45 +00002114 EmittedSlctedCheck = true;
Evan Chenge41bf822006-02-05 06:43:12 +00002115 if (NodeHasChain) {
2116 // FIXME: Don't fold if 1) the parent node writes a flag, 2) the node
2117 // has a chain use.
2118 // This a workaround for this problem:
2119 //
2120 // [ch, r : ld]
2121 // ^ ^
2122 // | |
2123 // [XX]--/ \- [flag : cmp]
2124 // ^ ^
2125 // | |
2126 // \---[br flag]-
2127 //
2128 // cmp + br should be considered as a single node as they are flagged
2129 // together. So, if the ld is folded into the cmp, the XX node in the
2130 // graph is now both an operand and a use of the ld/cmp/br node.
2131 if (NodeHasProperty(P, SDNodeInfo::SDNPOutFlag, ISE))
2132 emitCheck(ParentName + ".Val->isOnlyUse(" + RootName + ".Val)");
2133
2134 // If the immediate use can somehow reach this node through another
2135 // path, then can't fold it either or it will create a cycle.
2136 // e.g. In the following diagram, XX can reach ld through YY. If
2137 // ld is folded into XX, then YY is both a predecessor and a successor
2138 // of XX.
2139 //
2140 // [ld]
2141 // ^ ^
2142 // | |
2143 // / \---
2144 // / [YY]
2145 // | ^
2146 // [XX]-------|
2147 const SDNodeInfo &PInfo = ISE.getSDNodeInfo(P->getOperator());
2148 if (PInfo.getNumOperands() > 1 ||
2149 PInfo.hasProperty(SDNodeInfo::SDNPHasChain) ||
2150 PInfo.hasProperty(SDNodeInfo::SDNPInFlag) ||
2151 PInfo.hasProperty(SDNodeInfo::SDNPOptInFlag))
Evan Cheng6f8aaf22006-03-07 08:31:27 +00002152 if (PInfo.getNumOperands() > 1) {
2153 emitCheck("!isNonImmUse(" + ParentName + ".Val, " + RootName +
2154 ".Val)");
2155 } else {
2156 emitCheck("(" + ParentName + ".getNumOperands() == 1 || !" +
2157 "isNonImmUse(" + ParentName + ".Val, " + RootName +
2158 ".Val))");
2159 }
Evan Chenge41bf822006-02-05 06:43:12 +00002160 }
Evan Chengb915f312005-12-09 22:45:35 +00002161 }
Evan Chenge41bf822006-02-05 06:43:12 +00002162
Evan Chengc15d18c2006-01-27 22:13:45 +00002163 if (NodeHasChain) {
Evan Chenge4a8a6e2006-02-03 06:22:41 +00002164 ChainName = "Chain" + ChainSuffix;
Evan Cheng21ad3922006-02-07 00:37:41 +00002165 emitDecl(ChainName);
Evan Chenged66e852006-03-09 08:19:11 +00002166 if (FoundChain) {
2167 // FIXME: temporary workaround for a common case where chain
2168 // is a TokenFactor and the previous "inner" chain is an operand.
2169 NewTF = true;
2170 emitDecl("OldTF", true);
2171 emitCheck("(" + ChainName + " = UpdateFoldedChain(CurDAG, " +
2172 RootName + ".Val, Chain.Val, OldTF)).Val");
2173 } else {
2174 FoundChain = true;
2175 emitCode(ChainName + " = " + RootName + ".getOperand(0);");
2176 }
Evan Cheng1cf6db22006-01-06 00:41:12 +00002177 }
Evan Chengb915f312005-12-09 22:45:35 +00002178 }
2179
Evan Cheng54597732006-01-26 00:22:25 +00002180 // Don't fold any node which reads or writes a flag and has multiple uses.
Evan Chenge41bf822006-02-05 06:43:12 +00002181 // FIXME: We really need to separate the concepts of flag and "glue". Those
Evan Cheng54597732006-01-26 00:22:25 +00002182 // real flag results, e.g. X86CMP output, can have multiple uses.
Evan Chenge41bf822006-02-05 06:43:12 +00002183 // FIXME: If the optional incoming flag does not exist. Then it is ok to
2184 // fold it.
Evan Cheng1feeeec2006-01-26 19:13:45 +00002185 if (!isRoot &&
Evan Cheng54597732006-01-26 00:22:25 +00002186 (PatternHasProperty(N, SDNodeInfo::SDNPInFlag, ISE) ||
2187 PatternHasProperty(N, SDNodeInfo::SDNPOptInFlag, ISE) ||
2188 PatternHasProperty(N, SDNodeInfo::SDNPOutFlag, ISE))) {
Evan Cheng1feeeec2006-01-26 19:13:45 +00002189 const SDNodeInfo &CInfo = ISE.getSDNodeInfo(N->getOperator());
2190 if (!EmittedUseCheck) {
Chris Lattner8a0604b2006-01-28 20:31:24 +00002191 // Multiple uses of actual result?
Chris Lattner67a202b2006-01-28 20:43:52 +00002192 emitCheck(RootName + ".hasOneUse()");
Evan Cheng54597732006-01-26 00:22:25 +00002193 }
Evan Cheng1feeeec2006-01-26 19:13:45 +00002194 if (!EmittedSlctedCheck)
2195 // hasOneUse() check is not strong enough. If the original node has
2196 // already been selected, it may have been replaced with another.
Chris Lattner8a0604b2006-01-28 20:31:24 +00002197 for (unsigned j = 0; j < CInfo.getNumResults(); j++)
Chris Lattner67a202b2006-01-28 20:43:52 +00002198 emitCheck("!CodeGenMap.count(" + RootName + ".getValue(" + utostr(j) +
Chris Lattner8a0604b2006-01-28 20:31:24 +00002199 "))");
Evan Cheng54597732006-01-26 00:22:25 +00002200 }
2201
Evan Chengb915f312005-12-09 22:45:35 +00002202 for (unsigned i = 0, e = N->getNumChildren(); i != e; ++i, ++OpNo) {
Evan Cheng21ad3922006-02-07 00:37:41 +00002203 emitDecl(RootName + utostr(OpNo));
2204 emitCode(RootName + utostr(OpNo) + " = " +
Chris Lattner8a0604b2006-01-28 20:31:24 +00002205 RootName + ".getOperand(" +utostr(OpNo) + ");");
Evan Chengb915f312005-12-09 22:45:35 +00002206 TreePatternNode *Child = N->getChild(i);
2207
2208 if (!Child->isLeaf()) {
2209 // If it's not a leaf, recursively match.
2210 const SDNodeInfo &CInfo = ISE.getSDNodeInfo(Child->getOperator());
Chris Lattner67a202b2006-01-28 20:43:52 +00002211 emitCheck(RootName + utostr(OpNo) + ".getOpcode() == " +
Chris Lattner8a0604b2006-01-28 20:31:24 +00002212 CInfo.getEnumName());
Evan Chenge41bf822006-02-05 06:43:12 +00002213 EmitMatchCode(Child, N, RootName + utostr(OpNo), RootName,
2214 ChainSuffix + utostr(OpNo), FoundChain);
Chris Lattner8a0604b2006-01-28 20:31:24 +00002215 if (NodeHasProperty(Child, SDNodeInfo::SDNPHasChain, ISE))
Evan Cheng1b80f4d2005-12-19 07:18:51 +00002216 FoldedChains.push_back(std::make_pair(RootName + utostr(OpNo),
2217 CInfo.getNumResults()));
Evan Chengb915f312005-12-09 22:45:35 +00002218 } else {
Chris Lattner488580c2006-01-28 19:06:51 +00002219 // If this child has a name associated with it, capture it in VarMap. If
Evan Chengb915f312005-12-09 22:45:35 +00002220 // we already saw this in the pattern, emit code to verify dagness.
2221 if (!Child->getName().empty()) {
2222 std::string &VarMapEntry = VariableMap[Child->getName()];
2223 if (VarMapEntry.empty()) {
2224 VarMapEntry = RootName + utostr(OpNo);
2225 } else {
Jeff Cohen9b0ffca2006-01-27 22:22:28 +00002226 // If we get here, this is a second reference to a specific name.
2227 // Since we already have checked that the first reference is valid,
2228 // we don't have to recursively match it, just check that it's the
2229 // same as the previously named thing.
Chris Lattner67a202b2006-01-28 20:43:52 +00002230 emitCheck(VarMapEntry + " == " + RootName + utostr(OpNo));
Evan Chengb4ad33c2006-01-19 01:55:45 +00002231 Duplicates.insert(RootName + utostr(OpNo));
Evan Chengb915f312005-12-09 22:45:35 +00002232 continue;
2233 }
2234 }
2235
2236 // Handle leaves of various types.
2237 if (DefInit *DI = dynamic_cast<DefInit*>(Child->getLeafValue())) {
2238 Record *LeafRec = DI->getDef();
2239 if (LeafRec->isSubClassOf("RegisterClass")) {
2240 // Handle register references. Nothing to do here.
2241 } else if (LeafRec->isSubClassOf("Register")) {
Evan Cheng97938882005-12-22 02:24:50 +00002242 // Handle register references.
Evan Chengb915f312005-12-09 22:45:35 +00002243 } else if (LeafRec->isSubClassOf("ComplexPattern")) {
2244 // Handle complex pattern. Nothing to do here.
Evan Cheng01f318b2005-12-14 02:21:57 +00002245 } else if (LeafRec->getName() == "srcvalue") {
2246 // Place holder for SRCVALUE nodes. Nothing to do here.
Evan Chengb915f312005-12-09 22:45:35 +00002247 } else if (LeafRec->isSubClassOf("ValueType")) {
2248 // Make sure this is the specified value type.
Chris Lattner8a0604b2006-01-28 20:31:24 +00002249 emitCheck("cast<VTSDNode>(" + RootName + utostr(OpNo) +
Chris Lattner67a202b2006-01-28 20:43:52 +00002250 ")->getVT() == MVT::" + LeafRec->getName());
Evan Chengb915f312005-12-09 22:45:35 +00002251 } else if (LeafRec->isSubClassOf("CondCode")) {
2252 // Make sure this is the specified cond code.
Chris Lattner8a0604b2006-01-28 20:31:24 +00002253 emitCheck("cast<CondCodeSDNode>(" + RootName + utostr(OpNo) +
Chris Lattner67a202b2006-01-28 20:43:52 +00002254 ")->get() == ISD::" + LeafRec->getName());
Evan Chengb915f312005-12-09 22:45:35 +00002255 } else {
2256 Child->dump();
Evan Cheng97938882005-12-22 02:24:50 +00002257 std::cerr << " ";
Evan Chengb915f312005-12-09 22:45:35 +00002258 assert(0 && "Unknown leaf type!");
2259 }
Chris Lattner488580c2006-01-28 19:06:51 +00002260 } else if (IntInit *II =
2261 dynamic_cast<IntInit*>(Child->getLeafValue())) {
Chris Lattner8bc74722006-01-29 04:25:26 +00002262 emitCheck("isa<ConstantSDNode>(" + RootName + utostr(OpNo) + ")");
2263 unsigned CTmp = TmpNo++;
Andrew Lenharth8e517732006-01-29 05:22:37 +00002264 emitCode("int64_t CN"+utostr(CTmp)+" = cast<ConstantSDNode>("+
Chris Lattner8bc74722006-01-29 04:25:26 +00002265 RootName + utostr(OpNo) + ")->getSignExtended();");
2266
2267 emitCheck("CN" + utostr(CTmp) + " == " +itostr(II->getValue()));
Evan Chengb915f312005-12-09 22:45:35 +00002268 } else {
2269 Child->dump();
2270 assert(0 && "Unknown leaf type!");
2271 }
2272 }
2273 }
2274
2275 // If there is a node predicate for this, emit the call.
Chris Lattner8a0604b2006-01-28 20:31:24 +00002276 if (!N->getPredicateFn().empty())
Chris Lattner67a202b2006-01-28 20:43:52 +00002277 emitCheck(N->getPredicateFn() + "(" + RootName + ".Val)");
Evan Chengb915f312005-12-09 22:45:35 +00002278 }
2279
2280 /// EmitResultCode - Emit the action for a pattern. Now that it has matched
2281 /// we actually have to build a DAG!
2282 std::pair<unsigned, unsigned>
Chris Lattner947604b2006-03-24 21:52:20 +00002283 EmitResultCode(TreePatternNode *N, bool LikeLeaf = false,
2284 bool isRoot = false) {
Evan Chengb915f312005-12-09 22:45:35 +00002285 // This is something selected from the pattern we matched.
2286 if (!N->getName().empty()) {
Evan Chengb915f312005-12-09 22:45:35 +00002287 std::string &Val = VariableMap[N->getName()];
2288 assert(!Val.empty() &&
2289 "Variable referenced but not defined and not caught earlier!");
2290 if (Val[0] == 'T' && Val[1] == 'm' && Val[2] == 'p') {
2291 // Already selected this operand, just return the tmpval.
2292 return std::make_pair(1, atoi(Val.c_str()+3));
2293 }
2294
2295 const ComplexPattern *CP;
2296 unsigned ResNo = TmpNo++;
2297 unsigned NumRes = 1;
2298 if (!N->isLeaf() && N->getOperator()->getName() == "imm") {
Nate Begemanb73628b2005-12-30 00:12:56 +00002299 assert(N->getExtTypes().size() == 1 && "Multiple types not handled!");
Chris Lattner78593132006-01-29 20:01:35 +00002300 std::string CastType;
Nate Begemanb73628b2005-12-30 00:12:56 +00002301 switch (N->getTypeNum(0)) {
Chris Lattner8a0604b2006-01-28 20:31:24 +00002302 default: assert(0 && "Unknown type for constant node!");
Chris Lattner78593132006-01-29 20:01:35 +00002303 case MVT::i1: CastType = "bool"; break;
2304 case MVT::i8: CastType = "unsigned char"; break;
2305 case MVT::i16: CastType = "unsigned short"; break;
2306 case MVT::i32: CastType = "unsigned"; break;
2307 case MVT::i64: CastType = "uint64_t"; break;
Evan Chengb915f312005-12-09 22:45:35 +00002308 }
Chris Lattner78593132006-01-29 20:01:35 +00002309 emitCode(CastType + " Tmp" + utostr(ResNo) + "C = (" + CastType +
Andrew Lenharth2cba57c2006-01-29 05:17:22 +00002310 ")cast<ConstantSDNode>(" + Val + ")->getValue();");
Evan Cheng21ad3922006-02-07 00:37:41 +00002311 emitDecl("Tmp" + utostr(ResNo));
2312 emitCode("Tmp" + utostr(ResNo) +
Chris Lattner8a0604b2006-01-28 20:31:24 +00002313 " = CurDAG->getTargetConstant(Tmp" + utostr(ResNo) +
2314 "C, MVT::" + getEnumName(N->getTypeNum(0)) + ");");
Evan Chengbb48e332006-01-12 07:54:57 +00002315 } else if (!N->isLeaf() && N->getOperator()->getName() == "texternalsym"){
Evan Chengf805c2e2006-01-12 19:35:54 +00002316 Record *Op = OperatorMap[N->getName()];
2317 // Transform ExternalSymbol to TargetExternalSymbol
2318 if (Op && Op->getName() == "externalsym") {
Evan Cheng21ad3922006-02-07 00:37:41 +00002319 emitDecl("Tmp" + utostr(ResNo));
2320 emitCode("Tmp" + utostr(ResNo) + " = CurDAG->getTarget"
Chris Lattner8a0604b2006-01-28 20:31:24 +00002321 "ExternalSymbol(cast<ExternalSymbolSDNode>(" +
2322 Val + ")->getSymbol(), MVT::" +
2323 getEnumName(N->getTypeNum(0)) + ");");
2324 } else {
Evan Cheng21ad3922006-02-07 00:37:41 +00002325 emitDecl("Tmp" + utostr(ResNo));
2326 emitCode("Tmp" + utostr(ResNo) + " = " + Val + ";");
Chris Lattner8a0604b2006-01-28 20:31:24 +00002327 }
Evan Chengb915f312005-12-09 22:45:35 +00002328 } else if (!N->isLeaf() && N->getOperator()->getName() == "tglobaladdr") {
Evan Chengf805c2e2006-01-12 19:35:54 +00002329 Record *Op = OperatorMap[N->getName()];
2330 // Transform GlobalAddress to TargetGlobalAddress
2331 if (Op && Op->getName() == "globaladdr") {
Evan Cheng21ad3922006-02-07 00:37:41 +00002332 emitDecl("Tmp" + utostr(ResNo));
2333 emitCode("Tmp" + utostr(ResNo) + " = CurDAG->getTarget"
Chris Lattner8a0604b2006-01-28 20:31:24 +00002334 "GlobalAddress(cast<GlobalAddressSDNode>(" + Val +
2335 ")->getGlobal(), MVT::" + getEnumName(N->getTypeNum(0)) +
2336 ");");
2337 } else {
Evan Cheng21ad3922006-02-07 00:37:41 +00002338 emitDecl("Tmp" + utostr(ResNo));
2339 emitCode("Tmp" + utostr(ResNo) + " = " + Val + ";");
Chris Lattner8a0604b2006-01-28 20:31:24 +00002340 }
Chris Lattner4e3c8e512006-01-03 22:55:16 +00002341 } else if (!N->isLeaf() && N->getOperator()->getName() == "texternalsym"){
Evan Cheng21ad3922006-02-07 00:37:41 +00002342 emitDecl("Tmp" + utostr(ResNo));
2343 emitCode("Tmp" + utostr(ResNo) + " = " + Val + ";");
Evan Chengbb48e332006-01-12 07:54:57 +00002344 } else if (!N->isLeaf() && N->getOperator()->getName() == "tconstpool") {
Evan Cheng21ad3922006-02-07 00:37:41 +00002345 emitDecl("Tmp" + utostr(ResNo));
2346 emitCode("Tmp" + utostr(ResNo) + " = " + Val + ";");
Evan Chengb915f312005-12-09 22:45:35 +00002347 } else if (N->isLeaf() && (CP = NodeGetComplexPattern(N, ISE))) {
2348 std::string Fn = CP->getSelectFunc();
2349 NumRes = CP->getNumOperands();
Evan Cheng21ad3922006-02-07 00:37:41 +00002350 for (unsigned i = 0; i < NumRes; ++i)
2351 emitDecl("Tmp" + utostr(i+ResNo));
Chris Lattner8a0604b2006-01-28 20:31:24 +00002352
Evan Cheng21ad3922006-02-07 00:37:41 +00002353 std::string Code = Fn + "(" + Val;
Jeff Cohen60e91872006-01-04 03:23:30 +00002354 for (unsigned i = 0; i < NumRes; i++)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002355 Code += ", Tmp" + utostr(i + ResNo);
2356 emitCheck(Code + ")");
Evan Cheng9c4815a2006-02-04 08:50:49 +00002357
Evan Cheng2216d8a2006-02-05 05:22:18 +00002358 for (unsigned i = 0; i < NumRes; ++i)
Evan Cheng34167212006-02-09 00:37:58 +00002359 emitCode("Select(Tmp" + utostr(i+ResNo) + ", Tmp" +
Evan Cheng2216d8a2006-02-05 05:22:18 +00002360 utostr(i+ResNo) + ");");
Evan Cheng9c4815a2006-02-04 08:50:49 +00002361
Evan Chengb915f312005-12-09 22:45:35 +00002362 TmpNo = ResNo + NumRes;
2363 } else {
Evan Cheng21ad3922006-02-07 00:37:41 +00002364 emitDecl("Tmp" + utostr(ResNo));
Evan Cheng863bf5a2006-03-20 22:53:06 +00002365 // This node, probably wrapped in a SDNodeXForms, behaves like a leaf
2366 // node even if it isn't one. Don't select it.
2367 if (LikeLeaf)
2368 emitCode("Tmp" + utostr(ResNo) + " = " + Val + ";");
2369 else
2370 emitCode("Select(Tmp" + utostr(ResNo) + ", " + Val + ");");
Evan Cheng83e1a6a2006-03-23 02:35:32 +00002371
2372 if (isRoot && N->isLeaf()) {
2373 emitCode("Result = Tmp" + utostr(ResNo) + ";");
2374 emitCode("return;");
2375 }
Evan Chengb915f312005-12-09 22:45:35 +00002376 }
2377 // Add Tmp<ResNo> to VariableMap, so that we don't multiply select this
2378 // value if used multiple times by this pattern result.
2379 Val = "Tmp"+utostr(ResNo);
2380 return std::make_pair(NumRes, ResNo);
2381 }
Evan Chengb915f312005-12-09 22:45:35 +00002382 if (N->isLeaf()) {
2383 // If this is an explicit register reference, handle it.
2384 if (DefInit *DI = dynamic_cast<DefInit*>(N->getLeafValue())) {
2385 unsigned ResNo = TmpNo++;
2386 if (DI->getDef()->isSubClassOf("Register")) {
Evan Cheng21ad3922006-02-07 00:37:41 +00002387 emitDecl("Tmp" + utostr(ResNo));
2388 emitCode("Tmp" + utostr(ResNo) + " = CurDAG->getRegister(" +
Chris Lattner8a0604b2006-01-28 20:31:24 +00002389 ISE.getQualifiedName(DI->getDef()) + ", MVT::" +
2390 getEnumName(N->getTypeNum(0)) + ");");
Evan Chengb915f312005-12-09 22:45:35 +00002391 return std::make_pair(1, ResNo);
2392 }
2393 } else if (IntInit *II = dynamic_cast<IntInit*>(N->getLeafValue())) {
2394 unsigned ResNo = TmpNo++;
Nate Begemanb73628b2005-12-30 00:12:56 +00002395 assert(N->getExtTypes().size() == 1 && "Multiple types not handled!");
Evan Cheng21ad3922006-02-07 00:37:41 +00002396 emitDecl("Tmp" + utostr(ResNo));
2397 emitCode("Tmp" + utostr(ResNo) +
Chris Lattner8a0604b2006-01-28 20:31:24 +00002398 " = CurDAG->getTargetConstant(" + itostr(II->getValue()) +
2399 ", MVT::" + getEnumName(N->getTypeNum(0)) + ");");
Evan Chengb915f312005-12-09 22:45:35 +00002400 return std::make_pair(1, ResNo);
2401 }
2402
2403 N->dump();
2404 assert(0 && "Unknown leaf type!");
2405 return std::make_pair(1, ~0U);
2406 }
2407
2408 Record *Op = N->getOperator();
2409 if (Op->isSubClassOf("Instruction")) {
Evan Cheng7b05bd52005-12-23 22:11:47 +00002410 const CodeGenTarget &CGT = ISE.getTargetInfo();
2411 CodeGenInstruction &II = CGT.getInstruction(Op->getName());
Evan Cheng4fba2812005-12-20 07:37:41 +00002412 const DAGInstruction &Inst = ISE.getInstruction(Op);
Evan Cheng7b05bd52005-12-23 22:11:47 +00002413 bool HasImpInputs = Inst.getNumImpOperands() > 0;
2414 bool HasImpResults = Inst.getNumImpResults() > 0;
Evan Cheng54597732006-01-26 00:22:25 +00002415 bool HasOptInFlag = isRoot &&
2416 PatternHasProperty(Pattern, SDNodeInfo::SDNPOptInFlag, ISE);
Evan Cheng51fecc82006-01-09 18:27:06 +00002417 bool HasInFlag = isRoot &&
Evan Cheng54597732006-01-26 00:22:25 +00002418 PatternHasProperty(Pattern, SDNodeInfo::SDNPInFlag, ISE);
2419 bool NodeHasOutFlag = HasImpResults ||
Evan Cheng51fecc82006-01-09 18:27:06 +00002420 (isRoot && PatternHasProperty(Pattern, SDNodeInfo::SDNPOutFlag, ISE));
Evan Cheng823b7522006-01-19 21:57:10 +00002421 bool NodeHasChain =
2422 NodeHasProperty(Pattern, SDNodeInfo::SDNPHasChain, ISE);
Evan Cheng51fecc82006-01-09 18:27:06 +00002423 bool HasChain = II.hasCtrlDep ||
2424 (isRoot && PatternHasProperty(Pattern, SDNodeInfo::SDNPHasChain, ISE));
Evan Cheng4fba2812005-12-20 07:37:41 +00002425
Evan Cheng54597732006-01-26 00:22:25 +00002426 if (HasInFlag || NodeHasOutFlag || HasOptInFlag || HasImpInputs)
Evan Cheng21ad3922006-02-07 00:37:41 +00002427 emitDecl("InFlag");
Evan Cheng54597732006-01-26 00:22:25 +00002428 if (HasOptInFlag)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002429 emitCode("bool HasOptInFlag = false;");
Evan Cheng4fba2812005-12-20 07:37:41 +00002430
Evan Cheng823b7522006-01-19 21:57:10 +00002431 // How many results is this pattern expected to produce?
Evan Chenged66e852006-03-09 08:19:11 +00002432 unsigned PatResults = 0;
Evan Cheng823b7522006-01-19 21:57:10 +00002433 for (unsigned i = 0, e = Pattern->getExtTypes().size(); i != e; i++) {
2434 MVT::ValueType VT = Pattern->getTypeNum(i);
2435 if (VT != MVT::isVoid && VT != MVT::Flag)
Evan Chenged66e852006-03-09 08:19:11 +00002436 PatResults++;
Evan Cheng823b7522006-01-19 21:57:10 +00002437 }
2438
Evan Chengb915f312005-12-09 22:45:35 +00002439 // Determine operand emission order. Complex pattern first.
2440 std::vector<std::pair<unsigned, TreePatternNode*> > EmitOrder;
2441 std::vector<std::pair<unsigned, TreePatternNode*> >::iterator OI;
2442 for (unsigned i = 0, e = N->getNumChildren(); i != e; ++i) {
2443 TreePatternNode *Child = N->getChild(i);
2444 if (i == 0) {
2445 EmitOrder.push_back(std::make_pair(i, Child));
2446 OI = EmitOrder.begin();
2447 } else if (NodeIsComplexPattern(Child)) {
2448 OI = EmitOrder.insert(OI, std::make_pair(i, Child));
2449 } else {
2450 EmitOrder.push_back(std::make_pair(i, Child));
2451 }
2452 }
2453
2454 // Emit all of the operands.
2455 std::vector<std::pair<unsigned, unsigned> > NumTemps(EmitOrder.size());
2456 for (unsigned i = 0, e = EmitOrder.size(); i != e; ++i) {
2457 unsigned OpOrder = EmitOrder[i].first;
2458 TreePatternNode *Child = EmitOrder[i].second;
2459 std::pair<unsigned, unsigned> NumTemp = EmitResultCode(Child);
2460 NumTemps[OpOrder] = NumTemp;
2461 }
2462
2463 // List all the operands in the right order.
2464 std::vector<unsigned> Ops;
2465 for (unsigned i = 0, e = NumTemps.size(); i != e; i++) {
2466 for (unsigned j = 0; j < NumTemps[i].first; j++)
2467 Ops.push_back(NumTemps[i].second + j);
2468 }
2469
Evan Chengb915f312005-12-09 22:45:35 +00002470 // Emit all the chain and CopyToReg stuff.
Evan Chengb2c6d492006-01-11 22:16:13 +00002471 bool ChainEmitted = HasChain;
Evan Cheng7b05bd52005-12-23 22:11:47 +00002472 if (HasChain)
Evan Cheng34167212006-02-09 00:37:58 +00002473 emitCode("Select(" + ChainName + ", " + ChainName + ");");
Evan Cheng54597732006-01-26 00:22:25 +00002474 if (HasInFlag || HasOptInFlag || HasImpInputs)
2475 EmitInFlagSelectCode(Pattern, "N", ChainEmitted, true);
Evan Chengb915f312005-12-09 22:45:35 +00002476
Evan Chengb915f312005-12-09 22:45:35 +00002477 unsigned NumResults = Inst.getNumResults();
2478 unsigned ResNo = TmpNo++;
2479 if (!isRoot) {
Evan Cheng21ad3922006-02-07 00:37:41 +00002480 emitDecl("Tmp" + utostr(ResNo));
Chris Lattner8a0604b2006-01-28 20:31:24 +00002481 std::string Code =
Evan Chengd7805a72006-02-09 07:16:09 +00002482 "Tmp" + utostr(ResNo) + " = SDOperand(CurDAG->getTargetNode(" +
Chris Lattner8a0604b2006-01-28 20:31:24 +00002483 II.Namespace + "::" + II.TheDef->getName();
Nate Begemanb73628b2005-12-30 00:12:56 +00002484 if (N->getTypeNum(0) != MVT::isVoid)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002485 Code += ", MVT::" + getEnumName(N->getTypeNum(0));
Evan Cheng54597732006-01-26 00:22:25 +00002486 if (NodeHasOutFlag)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002487 Code += ", MVT::Flag";
Evan Chengbcecf332005-12-17 01:19:28 +00002488
Evan Chengb915f312005-12-09 22:45:35 +00002489 unsigned LastOp = 0;
2490 for (unsigned i = 0, e = Ops.size(); i != e; ++i) {
2491 LastOp = Ops[i];
Chris Lattner8a0604b2006-01-28 20:31:24 +00002492 Code += ", Tmp" + utostr(LastOp);
Evan Chengb915f312005-12-09 22:45:35 +00002493 }
Evan Chengd7805a72006-02-09 07:16:09 +00002494 emitCode(Code + "), 0);");
Evan Cheng7b05bd52005-12-23 22:11:47 +00002495 if (HasChain) {
Evan Chengb915f312005-12-09 22:45:35 +00002496 // Must have at least one result
Evan Chenge4a8a6e2006-02-03 06:22:41 +00002497 emitCode(ChainName + " = Tmp" + utostr(LastOp) + ".getValue(" +
Chris Lattner8a0604b2006-01-28 20:31:24 +00002498 utostr(NumResults) + ");");
Evan Chengb915f312005-12-09 22:45:35 +00002499 }
Evan Cheng54597732006-01-26 00:22:25 +00002500 } else if (HasChain || NodeHasOutFlag) {
Evan Cheng9789aaa2006-01-24 20:46:50 +00002501 if (HasOptInFlag) {
Evan Cheng9789aaa2006-01-24 20:46:50 +00002502 unsigned FlagNo = (unsigned) NodeHasChain + Pattern->getNumChildren();
Evan Chengd7805a72006-02-09 07:16:09 +00002503 emitDecl("ResNode", true);
Chris Lattner8a0604b2006-01-28 20:31:24 +00002504 emitCode("if (HasOptInFlag)");
Evan Chengd7805a72006-02-09 07:16:09 +00002505 std::string Code = " ResNode = CurDAG->getTargetNode(" +
Chris Lattner8a0604b2006-01-28 20:31:24 +00002506 II.Namespace + "::" + II.TheDef->getName();
Evan Cheng1b80f4d2005-12-19 07:18:51 +00002507
Evan Cheng9789aaa2006-01-24 20:46:50 +00002508 // Output order: results, chain, flags
2509 // Result types.
2510 if (NumResults > 0) {
2511 if (N->getTypeNum(0) != MVT::isVoid)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002512 Code += ", MVT::" + getEnumName(N->getTypeNum(0));
Evan Cheng9789aaa2006-01-24 20:46:50 +00002513 }
2514 if (HasChain)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002515 Code += ", MVT::Other";
Evan Cheng54597732006-01-26 00:22:25 +00002516 if (NodeHasOutFlag)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002517 Code += ", MVT::Flag";
Evan Cheng9789aaa2006-01-24 20:46:50 +00002518
2519 // Inputs.
2520 for (unsigned i = 0, e = Ops.size(); i != e; ++i)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002521 Code += ", Tmp" + utostr(Ops[i]);
Evan Chenge4a8a6e2006-02-03 06:22:41 +00002522 if (HasChain) Code += ", " + ChainName;
Chris Lattner8a0604b2006-01-28 20:31:24 +00002523 emitCode(Code + ", InFlag);");
Evan Cheng9789aaa2006-01-24 20:46:50 +00002524
Chris Lattner8a0604b2006-01-28 20:31:24 +00002525 emitCode("else");
Evan Chengd7805a72006-02-09 07:16:09 +00002526 Code = " ResNode = CurDAG->getTargetNode(" + II.Namespace + "::" +
Chris Lattner8a0604b2006-01-28 20:31:24 +00002527 II.TheDef->getName();
Evan Cheng9789aaa2006-01-24 20:46:50 +00002528
2529 // Output order: results, chain, flags
2530 // Result types.
Chris Lattner8a0604b2006-01-28 20:31:24 +00002531 if (NumResults > 0 && N->getTypeNum(0) != MVT::isVoid)
2532 Code += ", MVT::" + getEnumName(N->getTypeNum(0));
Evan Cheng9789aaa2006-01-24 20:46:50 +00002533 if (HasChain)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002534 Code += ", MVT::Other";
Evan Cheng54597732006-01-26 00:22:25 +00002535 if (NodeHasOutFlag)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002536 Code += ", MVT::Flag";
Evan Cheng9789aaa2006-01-24 20:46:50 +00002537
2538 // Inputs.
2539 for (unsigned i = 0, e = Ops.size(); i != e; ++i)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002540 Code += ", Tmp" + utostr(Ops[i]);
Evan Chenge4a8a6e2006-02-03 06:22:41 +00002541 if (HasChain) Code += ", " + ChainName + ");";
Chris Lattner8a0604b2006-01-28 20:31:24 +00002542 emitCode(Code);
Evan Cheng9789aaa2006-01-24 20:46:50 +00002543 } else {
Evan Chengd7805a72006-02-09 07:16:09 +00002544 emitDecl("ResNode", true);
2545 std::string Code = "ResNode = CurDAG->getTargetNode(" +
Chris Lattner8a0604b2006-01-28 20:31:24 +00002546 II.Namespace + "::" + II.TheDef->getName();
Evan Cheng9789aaa2006-01-24 20:46:50 +00002547
2548 // Output order: results, chain, flags
2549 // Result types.
Chris Lattner8a0604b2006-01-28 20:31:24 +00002550 if (NumResults > 0 && N->getTypeNum(0) != MVT::isVoid)
2551 Code += ", MVT::" + getEnumName(N->getTypeNum(0));
Evan Cheng9789aaa2006-01-24 20:46:50 +00002552 if (HasChain)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002553 Code += ", MVT::Other";
Evan Cheng54597732006-01-26 00:22:25 +00002554 if (NodeHasOutFlag)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002555 Code += ", MVT::Flag";
Evan Cheng9789aaa2006-01-24 20:46:50 +00002556
2557 // Inputs.
2558 for (unsigned i = 0, e = Ops.size(); i != e; ++i)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002559 Code += ", Tmp" + utostr(Ops[i]);
Evan Chenge4a8a6e2006-02-03 06:22:41 +00002560 if (HasChain) Code += ", " + ChainName;
Chris Lattner8a0604b2006-01-28 20:31:24 +00002561 if (HasInFlag || HasImpInputs) Code += ", InFlag";
2562 emitCode(Code + ");");
Evan Chengbcecf332005-12-17 01:19:28 +00002563 }
Evan Cheng1b80f4d2005-12-19 07:18:51 +00002564
Evan Chenged66e852006-03-09 08:19:11 +00002565 if (NewTF)
2566 emitCode("if (OldTF) "
2567 "SelectionDAG::InsertISelMapEntry(CodeGenMap, OldTF, 0, " +
2568 ChainName + ".Val, 0);");
2569
2570 for (unsigned i = 0; i < NumResults; i++)
Evan Cheng67212a02006-02-09 22:12:27 +00002571 emitCode("SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, " +
Evan Chenged66e852006-03-09 08:19:11 +00002572 utostr(i) + ", ResNode, " + utostr(i) + ");");
Evan Chengf9fc25d2005-12-19 22:40:04 +00002573
Evan Cheng54597732006-01-26 00:22:25 +00002574 if (NodeHasOutFlag)
Evan Chengd7805a72006-02-09 07:16:09 +00002575 emitCode("InFlag = SDOperand(ResNode, " +
Evan Chenged66e852006-03-09 08:19:11 +00002576 utostr(NumResults + (unsigned)HasChain) + ");");
Evan Cheng4fba2812005-12-20 07:37:41 +00002577
Chris Lattner8a0604b2006-01-28 20:31:24 +00002578 if (HasImpResults && EmitCopyFromRegs(N, ChainEmitted)) {
Evan Chenged66e852006-03-09 08:19:11 +00002579 emitCode("SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, "
2580 "0, ResNode, 0);");
2581 NumResults = 1;
Evan Cheng97938882005-12-22 02:24:50 +00002582 }
Evan Cheng4fba2812005-12-20 07:37:41 +00002583
Evan Chenge41bf822006-02-05 06:43:12 +00002584 if (NodeHasChain) {
Evan Cheng67212a02006-02-09 22:12:27 +00002585 emitCode("SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, " +
Evan Chenged66e852006-03-09 08:19:11 +00002586 utostr(PatResults) + ", ResNode, " +
2587 utostr(NumResults) + ");");
Evan Chenge41bf822006-02-05 06:43:12 +00002588 if (DoReplace)
Evan Chenged66e852006-03-09 08:19:11 +00002589 emitCode("if (N.ResNo == 0) AddHandleReplacement(N.Val, " +
2590 utostr(PatResults) + ", " + "ResNode, " +
2591 utostr(NumResults) + ");");
Evan Chenge41bf822006-02-05 06:43:12 +00002592 }
2593
Evan Cheng97938882005-12-22 02:24:50 +00002594 if (FoldedChains.size() > 0) {
Chris Lattner8a0604b2006-01-28 20:31:24 +00002595 std::string Code;
Evan Cheng1b80f4d2005-12-19 07:18:51 +00002596 for (unsigned j = 0, e = FoldedChains.size(); j < e; j++)
Evan Cheng67212a02006-02-09 22:12:27 +00002597 emitCode("SelectionDAG::InsertISelMapEntry(CodeGenMap, " +
2598 FoldedChains[j].first + ".Val, " +
2599 utostr(FoldedChains[j].second) + ", ResNode, " +
Evan Chenged66e852006-03-09 08:19:11 +00002600 utostr(NumResults) + ");");
Evan Chenge41bf822006-02-05 06:43:12 +00002601
2602 for (unsigned j = 0, e = FoldedChains.size(); j < e; j++) {
2603 std::string Code =
Evan Cheng67212a02006-02-09 22:12:27 +00002604 FoldedChains[j].first + ".Val, " +
2605 utostr(FoldedChains[j].second) + ", ";
2606 emitCode("AddHandleReplacement(" + Code + "ResNode, " +
Evan Chenged66e852006-03-09 08:19:11 +00002607 utostr(NumResults) + ");");
Evan Chenge41bf822006-02-05 06:43:12 +00002608 }
Evan Chengb915f312005-12-09 22:45:35 +00002609 }
Evan Chengf9fc25d2005-12-19 22:40:04 +00002610
Evan Cheng54597732006-01-26 00:22:25 +00002611 if (NodeHasOutFlag)
Evan Cheng67212a02006-02-09 22:12:27 +00002612 emitCode("SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, " +
Evan Chenged66e852006-03-09 08:19:11 +00002613 utostr(PatResults + (unsigned)NodeHasChain) +
2614 ", InFlag.Val, InFlag.ResNo);");
Evan Cheng97938882005-12-22 02:24:50 +00002615
Evan Chenged66e852006-03-09 08:19:11 +00002616 // User does not expect the instruction would produce a chain!
2617 bool AddedChain = HasChain && !NodeHasChain;
Evan Cheng54597732006-01-26 00:22:25 +00002618 if (AddedChain && NodeHasOutFlag) {
Evan Chenged66e852006-03-09 08:19:11 +00002619 if (PatResults == 0) {
Evan Chengd7805a72006-02-09 07:16:09 +00002620 emitCode("Result = SDOperand(ResNode, N.ResNo+1);");
Evan Cheng97938882005-12-22 02:24:50 +00002621 } else {
Evan Chenged66e852006-03-09 08:19:11 +00002622 emitCode("if (N.ResNo < " + utostr(PatResults) + ")");
Evan Chengd7805a72006-02-09 07:16:09 +00002623 emitCode(" Result = SDOperand(ResNode, N.ResNo);");
Chris Lattner8a0604b2006-01-28 20:31:24 +00002624 emitCode("else");
Evan Chengd7805a72006-02-09 07:16:09 +00002625 emitCode(" Result = SDOperand(ResNode, N.ResNo+1);");
Evan Cheng97938882005-12-22 02:24:50 +00002626 }
Evan Cheng4fba2812005-12-20 07:37:41 +00002627 } else {
Evan Chengd7805a72006-02-09 07:16:09 +00002628 emitCode("Result = SDOperand(ResNode, N.ResNo);");
Evan Cheng4fba2812005-12-20 07:37:41 +00002629 }
Evan Chengb915f312005-12-09 22:45:35 +00002630 } else {
2631 // If this instruction is the root, and if there is only one use of it,
2632 // use SelectNodeTo instead of getTargetNode to avoid an allocation.
Chris Lattner8a0604b2006-01-28 20:31:24 +00002633 emitCode("if (N.Val->hasOneUse()) {");
Evan Cheng34167212006-02-09 00:37:58 +00002634 std::string Code = " Result = CurDAG->SelectNodeTo(N.Val, " +
Chris Lattner8a0604b2006-01-28 20:31:24 +00002635 II.Namespace + "::" + II.TheDef->getName();
Nate Begemanb73628b2005-12-30 00:12:56 +00002636 if (N->getTypeNum(0) != MVT::isVoid)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002637 Code += ", MVT::" + getEnumName(N->getTypeNum(0));
Evan Cheng54597732006-01-26 00:22:25 +00002638 if (NodeHasOutFlag)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002639 Code += ", MVT::Flag";
Evan Chengb915f312005-12-09 22:45:35 +00002640 for (unsigned i = 0, e = Ops.size(); i != e; ++i)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002641 Code += ", Tmp" + utostr(Ops[i]);
Evan Cheng51fecc82006-01-09 18:27:06 +00002642 if (HasInFlag || HasImpInputs)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002643 Code += ", InFlag";
2644 emitCode(Code + ");");
2645 emitCode("} else {");
Evan Chengd7805a72006-02-09 07:16:09 +00002646 emitDecl("ResNode", true);
2647 Code = " ResNode = CurDAG->getTargetNode(" +
Chris Lattner8a0604b2006-01-28 20:31:24 +00002648 II.Namespace + "::" + II.TheDef->getName();
Nate Begemanb73628b2005-12-30 00:12:56 +00002649 if (N->getTypeNum(0) != MVT::isVoid)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002650 Code += ", MVT::" + getEnumName(N->getTypeNum(0));
Evan Cheng54597732006-01-26 00:22:25 +00002651 if (NodeHasOutFlag)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002652 Code += ", MVT::Flag";
Evan Chengb915f312005-12-09 22:45:35 +00002653 for (unsigned i = 0, e = Ops.size(); i != e; ++i)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002654 Code += ", Tmp" + utostr(Ops[i]);
Evan Cheng51fecc82006-01-09 18:27:06 +00002655 if (HasInFlag || HasImpInputs)
Chris Lattner8a0604b2006-01-28 20:31:24 +00002656 Code += ", InFlag";
2657 emitCode(Code + ");");
Evan Cheng67212a02006-02-09 22:12:27 +00002658 emitCode(" SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, N.ResNo, "
2659 "ResNode, 0);");
2660 emitCode(" Result = SDOperand(ResNode, 0);");
Chris Lattner8a0604b2006-01-28 20:31:24 +00002661 emitCode("}");
Evan Chengb915f312005-12-09 22:45:35 +00002662 }
Evan Cheng4fba2812005-12-20 07:37:41 +00002663
Evan Cheng34167212006-02-09 00:37:58 +00002664 if (isRoot)
2665 emitCode("return;");
Evan Chengb915f312005-12-09 22:45:35 +00002666 return std::make_pair(1, ResNo);
2667 } else if (Op->isSubClassOf("SDNodeXForm")) {
2668 assert(N->getNumChildren() == 1 && "node xform should have one child!");
Evan Cheng863bf5a2006-03-20 22:53:06 +00002669 // PatLeaf node - the operand may or may not be a leaf node. But it should
2670 // behave like one.
2671 unsigned OpVal = EmitResultCode(N->getChild(0), true).second;
Evan Chengb915f312005-12-09 22:45:35 +00002672 unsigned ResNo = TmpNo++;
Evan Cheng21ad3922006-02-07 00:37:41 +00002673 emitDecl("Tmp" + utostr(ResNo));
2674 emitCode("Tmp" + utostr(ResNo) + " = Transform_" + Op->getName()
Chris Lattner8a0604b2006-01-28 20:31:24 +00002675 + "(Tmp" + utostr(OpVal) + ".Val);");
Evan Chengb915f312005-12-09 22:45:35 +00002676 if (isRoot) {
Evan Cheng67212a02006-02-09 22:12:27 +00002677 emitCode("SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val,"
2678 "N.ResNo, Tmp" + utostr(ResNo) + ".Val, Tmp" +
2679 utostr(ResNo) + ".ResNo);");
Evan Cheng34167212006-02-09 00:37:58 +00002680 emitCode("Result = Tmp" + utostr(ResNo) + ";");
2681 emitCode("return;");
Evan Chengb915f312005-12-09 22:45:35 +00002682 }
2683 return std::make_pair(1, ResNo);
2684 } else {
2685 N->dump();
Chris Lattner7893f132006-01-11 01:33:49 +00002686 std::cerr << "\n";
2687 throw std::string("Unknown node in result pattern!");
Evan Chengb915f312005-12-09 22:45:35 +00002688 }
2689 }
2690
Chris Lattner488580c2006-01-28 19:06:51 +00002691 /// InsertOneTypeCheck - Insert a type-check for an unresolved type in 'Pat'
2692 /// and add it to the tree. 'Pat' and 'Other' are isomorphic trees except that
Evan Chengb915f312005-12-09 22:45:35 +00002693 /// 'Pat' may be missing types. If we find an unresolved type to add a check
2694 /// for, this returns true otherwise false if Pat has all types.
2695 bool InsertOneTypeCheck(TreePatternNode *Pat, TreePatternNode *Other,
2696 const std::string &Prefix) {
2697 // Did we find one?
2698 if (!Pat->hasTypeSet()) {
2699 // Move a type over from 'other' to 'pat'.
Nate Begemanb73628b2005-12-30 00:12:56 +00002700 Pat->setTypes(Other->getExtTypes());
Chris Lattner67a202b2006-01-28 20:43:52 +00002701 emitCheck(Prefix + ".Val->getValueType(0) == MVT::" +
Chris Lattner8a0604b2006-01-28 20:31:24 +00002702 getName(Pat->getTypeNum(0)));
Evan Chengb915f312005-12-09 22:45:35 +00002703 return true;
Evan Chengb915f312005-12-09 22:45:35 +00002704 }
2705
Evan Cheng51fecc82006-01-09 18:27:06 +00002706 unsigned OpNo =
2707 (unsigned) NodeHasProperty(Pat, SDNodeInfo::SDNPHasChain, ISE);
Evan Chengb915f312005-12-09 22:45:35 +00002708 for (unsigned i = 0, e = Pat->getNumChildren(); i != e; ++i, ++OpNo)
2709 if (InsertOneTypeCheck(Pat->getChild(i), Other->getChild(i),
2710 Prefix + utostr(OpNo)))
2711 return true;
2712 return false;
2713 }
2714
2715private:
Evan Cheng54597732006-01-26 00:22:25 +00002716 /// EmitInFlagSelectCode - Emit the flag operands for the DAG that is
Evan Chengb915f312005-12-09 22:45:35 +00002717 /// being built.
Evan Cheng54597732006-01-26 00:22:25 +00002718 void EmitInFlagSelectCode(TreePatternNode *N, const std::string &RootName,
2719 bool &ChainEmitted, bool isRoot = false) {
Evan Chengb915f312005-12-09 22:45:35 +00002720 const CodeGenTarget &T = ISE.getTargetInfo();
Evan Cheng51fecc82006-01-09 18:27:06 +00002721 unsigned OpNo =
2722 (unsigned) NodeHasProperty(N, SDNodeInfo::SDNPHasChain, ISE);
Evan Cheng54597732006-01-26 00:22:25 +00002723 bool HasInFlag = NodeHasProperty(N, SDNodeInfo::SDNPInFlag, ISE);
2724 bool HasOptInFlag = NodeHasProperty(N, SDNodeInfo::SDNPOptInFlag, ISE);
Evan Chengb915f312005-12-09 22:45:35 +00002725 for (unsigned i = 0, e = N->getNumChildren(); i != e; ++i, ++OpNo) {
2726 TreePatternNode *Child = N->getChild(i);
2727 if (!Child->isLeaf()) {
Evan Cheng54597732006-01-26 00:22:25 +00002728 EmitInFlagSelectCode(Child, RootName + utostr(OpNo), ChainEmitted);
Evan Chengb915f312005-12-09 22:45:35 +00002729 } else {
2730 if (DefInit *DI = dynamic_cast<DefInit*>(Child->getLeafValue())) {
Evan Chengb4ad33c2006-01-19 01:55:45 +00002731 if (!Child->getName().empty()) {
2732 std::string Name = RootName + utostr(OpNo);
2733 if (Duplicates.find(Name) != Duplicates.end())
2734 // A duplicate! Do not emit a copy for this node.
2735 continue;
2736 }
2737
Evan Chengb915f312005-12-09 22:45:35 +00002738 Record *RR = DI->getDef();
2739 if (RR->isSubClassOf("Register")) {
2740 MVT::ValueType RVT = getRegisterValueType(RR, T);
Evan Chengbcecf332005-12-17 01:19:28 +00002741 if (RVT == MVT::Flag) {
Evan Cheng34167212006-02-09 00:37:58 +00002742 emitCode("Select(InFlag, " + RootName + utostr(OpNo) + ");");
Evan Chengb2c6d492006-01-11 22:16:13 +00002743 } else {
2744 if (!ChainEmitted) {
Evan Cheng21ad3922006-02-07 00:37:41 +00002745 emitDecl("Chain");
2746 emitCode("Chain = CurDAG->getEntryNode();");
Evan Chenge4a8a6e2006-02-03 06:22:41 +00002747 ChainName = "Chain";
Evan Chengb2c6d492006-01-11 22:16:13 +00002748 ChainEmitted = true;
2749 }
Evan Cheng34167212006-02-09 00:37:58 +00002750 emitCode("Select(" + RootName + utostr(OpNo) + ", " +
2751 RootName + utostr(OpNo) + ");");
Evan Cheng67212a02006-02-09 22:12:27 +00002752 emitCode("ResNode = CurDAG->getCopyToReg(" + ChainName +
Evan Chenge4a8a6e2006-02-03 06:22:41 +00002753 ", CurDAG->getRegister(" + ISE.getQualifiedName(RR) +
Evan Cheng34167212006-02-09 00:37:58 +00002754 ", MVT::" + getEnumName(RVT) + "), " +
Evan Chengd7805a72006-02-09 07:16:09 +00002755 RootName + utostr(OpNo) + ", InFlag).Val;");
Evan Cheng67212a02006-02-09 22:12:27 +00002756 emitCode(ChainName + " = SDOperand(ResNode, 0);");
2757 emitCode("InFlag = SDOperand(ResNode, 1);");
Evan Chengb915f312005-12-09 22:45:35 +00002758 }
2759 }
2760 }
2761 }
2762 }
Evan Cheng54597732006-01-26 00:22:25 +00002763
2764 if (HasInFlag || HasOptInFlag) {
Chris Lattner8a0604b2006-01-28 20:31:24 +00002765 std::string Code;
Evan Cheng54597732006-01-26 00:22:25 +00002766 if (HasOptInFlag) {
Chris Lattner8a0604b2006-01-28 20:31:24 +00002767 emitCode("if (" + RootName + ".getNumOperands() == " + utostr(OpNo+1) +
2768 ") {");
2769 Code = " ";
Evan Cheng54597732006-01-26 00:22:25 +00002770 }
Evan Cheng34167212006-02-09 00:37:58 +00002771 emitCode(Code + "Select(InFlag, " + RootName +
2772 ".getOperand(" + utostr(OpNo) + "));");
Evan Cheng54597732006-01-26 00:22:25 +00002773 if (HasOptInFlag) {
Chris Lattner8a0604b2006-01-28 20:31:24 +00002774 emitCode(" HasOptInFlag = true;");
2775 emitCode("}");
Evan Cheng54597732006-01-26 00:22:25 +00002776 }
2777 }
Evan Chengb915f312005-12-09 22:45:35 +00002778 }
Evan Cheng4fba2812005-12-20 07:37:41 +00002779
2780 /// EmitCopyFromRegs - Emit code to copy result to physical registers
Evan Cheng7b05bd52005-12-23 22:11:47 +00002781 /// as specified by the instruction. It returns true if any copy is
2782 /// emitted.
Evan Chengb2c6d492006-01-11 22:16:13 +00002783 bool EmitCopyFromRegs(TreePatternNode *N, bool &ChainEmitted) {
Evan Cheng7b05bd52005-12-23 22:11:47 +00002784 bool RetVal = false;
Evan Cheng4fba2812005-12-20 07:37:41 +00002785 Record *Op = N->getOperator();
2786 if (Op->isSubClassOf("Instruction")) {
2787 const DAGInstruction &Inst = ISE.getInstruction(Op);
2788 const CodeGenTarget &CGT = ISE.getTargetInfo();
2789 CodeGenInstruction &II = CGT.getInstruction(Op->getName());
2790 unsigned NumImpResults = Inst.getNumImpResults();
2791 for (unsigned i = 0; i < NumImpResults; i++) {
2792 Record *RR = Inst.getImpResult(i);
2793 if (RR->isSubClassOf("Register")) {
2794 MVT::ValueType RVT = getRegisterValueType(RR, CGT);
2795 if (RVT != MVT::Flag) {
Evan Chengb2c6d492006-01-11 22:16:13 +00002796 if (!ChainEmitted) {
Evan Cheng21ad3922006-02-07 00:37:41 +00002797 emitDecl("Chain");
2798 emitCode("Chain = CurDAG->getEntryNode();");
Evan Chengb2c6d492006-01-11 22:16:13 +00002799 ChainEmitted = true;
Evan Chenge4a8a6e2006-02-03 06:22:41 +00002800 ChainName = "Chain";
Evan Cheng4fba2812005-12-20 07:37:41 +00002801 }
Evan Chengd7805a72006-02-09 07:16:09 +00002802 emitCode("ResNode = CurDAG->getCopyFromReg(" + ChainName + ", " +
Chris Lattner8a0604b2006-01-28 20:31:24 +00002803 ISE.getQualifiedName(RR) + ", MVT::" + getEnumName(RVT) +
Evan Chengd7805a72006-02-09 07:16:09 +00002804 ", InFlag).Val;");
2805 emitCode(ChainName + " = SDOperand(ResNode, 1);");
2806 emitCode("InFlag = SDOperand(ResNode, 2);");
Evan Cheng7b05bd52005-12-23 22:11:47 +00002807 RetVal = true;
Evan Cheng4fba2812005-12-20 07:37:41 +00002808 }
2809 }
2810 }
2811 }
Evan Cheng7b05bd52005-12-23 22:11:47 +00002812 return RetVal;
Evan Cheng4fba2812005-12-20 07:37:41 +00002813 }
Evan Chengb915f312005-12-09 22:45:35 +00002814};
2815
Chris Lattnerd1ff35a2005-09-23 21:33:23 +00002816/// EmitCodeForPattern - Given a pattern to match, emit code to the specified
2817/// stream to match the pattern, and generate the code for the match if it
Chris Lattner355408b2006-01-29 02:43:35 +00002818/// succeeds. Returns true if the pattern is not guaranteed to match.
Chris Lattner8bc74722006-01-29 04:25:26 +00002819void DAGISelEmitter::GenerateCodeForPattern(PatternToMatch &Pattern,
Evan Chenge41bf822006-02-05 06:43:12 +00002820 std::vector<std::pair<bool, std::string> > &GeneratedCode,
Evan Chengd7805a72006-02-09 07:16:09 +00002821 std::set<std::pair<bool, std::string> > &GeneratedDecl,
Evan Chenge41bf822006-02-05 06:43:12 +00002822 bool DoReplace) {
Evan Cheng58e84a62005-12-14 22:02:59 +00002823 PatternCodeEmitter Emitter(*this, Pattern.getPredicates(),
2824 Pattern.getSrcPattern(), Pattern.getDstPattern(),
Evan Cheng21ad3922006-02-07 00:37:41 +00002825 GeneratedCode, GeneratedDecl, DoReplace);
Evan Chengb915f312005-12-09 22:45:35 +00002826
Chris Lattner8fc35682005-09-23 23:16:51 +00002827 // Emit the matcher, capturing named arguments in VariableMap.
Evan Cheng7b05bd52005-12-23 22:11:47 +00002828 bool FoundChain = false;
Evan Chenge41bf822006-02-05 06:43:12 +00002829 Emitter.EmitMatchCode(Pattern.getSrcPattern(), NULL, "N", "", "", FoundChain);
Evan Chengb915f312005-12-09 22:45:35 +00002830
Chris Lattner0ee7cff2005-10-14 04:11:13 +00002831 // TP - Get *SOME* tree pattern, we don't care which.
2832 TreePattern &TP = *PatternFragments.begin()->second;
Chris Lattner296dfe32005-09-24 00:50:51 +00002833
Chris Lattner0ee7cff2005-10-14 04:11:13 +00002834 // At this point, we know that we structurally match the pattern, but the
2835 // types of the nodes may not match. Figure out the fewest number of type
2836 // comparisons we need to emit. For example, if there is only one integer
2837 // type supported by a target, there should be no type comparisons at all for
2838 // integer patterns!
2839 //
2840 // To figure out the fewest number of type checks needed, clone the pattern,
2841 // remove the types, then perform type inference on the pattern as a whole.
2842 // If there are unresolved types, emit an explicit check for those types,
2843 // apply the type to the tree, then rerun type inference. Iterate until all
2844 // types are resolved.
2845 //
Evan Cheng58e84a62005-12-14 22:02:59 +00002846 TreePatternNode *Pat = Pattern.getSrcPattern()->clone();
Chris Lattner0ee7cff2005-10-14 04:11:13 +00002847 RemoveAllTypes(Pat);
Chris Lattner7e82f132005-10-15 21:34:21 +00002848
2849 do {
2850 // Resolve/propagate as many types as possible.
2851 try {
2852 bool MadeChange = true;
2853 while (MadeChange)
Chris Lattner488580c2006-01-28 19:06:51 +00002854 MadeChange = Pat->ApplyTypeConstraints(TP,
2855 true/*Ignore reg constraints*/);
Chris Lattner7e82f132005-10-15 21:34:21 +00002856 } catch (...) {
2857 assert(0 && "Error: could not find consistent types for something we"
2858 " already decided was ok!");
2859 abort();
2860 }
Chris Lattner0ee7cff2005-10-14 04:11:13 +00002861
Chris Lattner7e82f132005-10-15 21:34:21 +00002862 // Insert a check for an unresolved type and add it to the tree. If we find
2863 // an unresolved type to add a check for, this returns true and we iterate,
2864 // otherwise we are done.
Evan Cheng58e84a62005-12-14 22:02:59 +00002865 } while (Emitter.InsertOneTypeCheck(Pat, Pattern.getSrcPattern(), "N"));
Evan Cheng1c3d19e2005-12-04 08:18:16 +00002866
Evan Cheng863bf5a2006-03-20 22:53:06 +00002867 Emitter.EmitResultCode(Pattern.getDstPattern(), false, true /*the root*/);
Chris Lattner0ee7cff2005-10-14 04:11:13 +00002868 delete Pat;
Chris Lattner3f7e9142005-09-23 20:52:47 +00002869}
2870
Chris Lattner24e00a42006-01-29 04:41:05 +00002871/// EraseCodeLine - Erase one code line from all of the patterns. If removing
2872/// a line causes any of them to be empty, remove them and return true when
2873/// done.
2874static bool EraseCodeLine(std::vector<std::pair<PatternToMatch*,
2875 std::vector<std::pair<bool, std::string> > > >
2876 &Patterns) {
2877 bool ErasedPatterns = false;
2878 for (unsigned i = 0, e = Patterns.size(); i != e; ++i) {
2879 Patterns[i].second.pop_back();
2880 if (Patterns[i].second.empty()) {
2881 Patterns.erase(Patterns.begin()+i);
2882 --i; --e;
2883 ErasedPatterns = true;
2884 }
2885 }
2886 return ErasedPatterns;
2887}
2888
Chris Lattner8bc74722006-01-29 04:25:26 +00002889/// EmitPatterns - Emit code for at least one pattern, but try to group common
2890/// code together between the patterns.
2891void DAGISelEmitter::EmitPatterns(std::vector<std::pair<PatternToMatch*,
2892 std::vector<std::pair<bool, std::string> > > >
2893 &Patterns, unsigned Indent,
2894 std::ostream &OS) {
2895 typedef std::pair<bool, std::string> CodeLine;
2896 typedef std::vector<CodeLine> CodeList;
2897 typedef std::vector<std::pair<PatternToMatch*, CodeList> > PatternList;
2898
2899 if (Patterns.empty()) return;
2900
Chris Lattner24e00a42006-01-29 04:41:05 +00002901 // Figure out how many patterns share the next code line. Explicitly copy
2902 // FirstCodeLine so that we don't invalidate a reference when changing
2903 // Patterns.
2904 const CodeLine FirstCodeLine = Patterns.back().second.back();
Chris Lattner8bc74722006-01-29 04:25:26 +00002905 unsigned LastMatch = Patterns.size()-1;
2906 while (LastMatch != 0 && Patterns[LastMatch-1].second.back() == FirstCodeLine)
2907 --LastMatch;
2908
2909 // If not all patterns share this line, split the list into two pieces. The
2910 // first chunk will use this line, the second chunk won't.
2911 if (LastMatch != 0) {
2912 PatternList Shared(Patterns.begin()+LastMatch, Patterns.end());
2913 PatternList Other(Patterns.begin(), Patterns.begin()+LastMatch);
2914
2915 // FIXME: Emit braces?
2916 if (Shared.size() == 1) {
2917 PatternToMatch &Pattern = *Shared.back().first;
2918 OS << "\n" << std::string(Indent, ' ') << "// Pattern: ";
2919 Pattern.getSrcPattern()->print(OS);
2920 OS << "\n" << std::string(Indent, ' ') << "// Emits: ";
2921 Pattern.getDstPattern()->print(OS);
2922 OS << "\n";
2923 OS << std::string(Indent, ' ') << "// Pattern complexity = "
2924 << getPatternSize(Pattern.getSrcPattern(), *this) << " cost = "
Evan Chengfbad7082006-02-18 02:33:09 +00002925 << getResultPatternCost(Pattern.getDstPattern(), *this) << "\n";
Chris Lattner8bc74722006-01-29 04:25:26 +00002926 }
2927 if (!FirstCodeLine.first) {
2928 OS << std::string(Indent, ' ') << "{\n";
2929 Indent += 2;
2930 }
2931 EmitPatterns(Shared, Indent, OS);
2932 if (!FirstCodeLine.first) {
2933 Indent -= 2;
2934 OS << std::string(Indent, ' ') << "}\n";
2935 }
2936
2937 if (Other.size() == 1) {
2938 PatternToMatch &Pattern = *Other.back().first;
2939 OS << "\n" << std::string(Indent, ' ') << "// Pattern: ";
2940 Pattern.getSrcPattern()->print(OS);
2941 OS << "\n" << std::string(Indent, ' ') << "// Emits: ";
2942 Pattern.getDstPattern()->print(OS);
2943 OS << "\n";
2944 OS << std::string(Indent, ' ') << "// Pattern complexity = "
2945 << getPatternSize(Pattern.getSrcPattern(), *this) << " cost = "
Evan Chengfbad7082006-02-18 02:33:09 +00002946 << getResultPatternCost(Pattern.getDstPattern(), *this) << "\n";
Chris Lattner8bc74722006-01-29 04:25:26 +00002947 }
2948 EmitPatterns(Other, Indent, OS);
2949 return;
2950 }
2951
Chris Lattner24e00a42006-01-29 04:41:05 +00002952 // Remove this code from all of the patterns that share it.
2953 bool ErasedPatterns = EraseCodeLine(Patterns);
2954
Chris Lattner8bc74722006-01-29 04:25:26 +00002955 bool isPredicate = FirstCodeLine.first;
2956
2957 // Otherwise, every pattern in the list has this line. Emit it.
2958 if (!isPredicate) {
2959 // Normal code.
2960 OS << std::string(Indent, ' ') << FirstCodeLine.second << "\n";
2961 } else {
Chris Lattner24e00a42006-01-29 04:41:05 +00002962 OS << std::string(Indent, ' ') << "if (" << FirstCodeLine.second;
2963
2964 // If the next code line is another predicate, and if all of the pattern
2965 // in this group share the same next line, emit it inline now. Do this
2966 // until we run out of common predicates.
2967 while (!ErasedPatterns && Patterns.back().second.back().first) {
2968 // Check that all of fhe patterns in Patterns end with the same predicate.
2969 bool AllEndWithSamePredicate = true;
2970 for (unsigned i = 0, e = Patterns.size(); i != e; ++i)
2971 if (Patterns[i].second.back() != Patterns.back().second.back()) {
2972 AllEndWithSamePredicate = false;
2973 break;
2974 }
2975 // If all of the predicates aren't the same, we can't share them.
2976 if (!AllEndWithSamePredicate) break;
2977
2978 // Otherwise we can. Emit it shared now.
2979 OS << " &&\n" << std::string(Indent+4, ' ')
2980 << Patterns.back().second.back().second;
2981 ErasedPatterns = EraseCodeLine(Patterns);
Chris Lattner8bc74722006-01-29 04:25:26 +00002982 }
Chris Lattner24e00a42006-01-29 04:41:05 +00002983
2984 OS << ") {\n";
2985 Indent += 2;
Chris Lattner8bc74722006-01-29 04:25:26 +00002986 }
2987
2988 EmitPatterns(Patterns, Indent, OS);
2989
2990 if (isPredicate)
2991 OS << std::string(Indent-2, ' ') << "}\n";
2992}
2993
2994
Chris Lattner37481472005-09-26 21:59:35 +00002995
2996namespace {
2997 /// CompareByRecordName - An ordering predicate that implements less-than by
2998 /// comparing the names records.
2999 struct CompareByRecordName {
3000 bool operator()(const Record *LHS, const Record *RHS) const {
3001 // Sort by name first.
3002 if (LHS->getName() < RHS->getName()) return true;
3003 // If both names are equal, sort by pointer.
3004 return LHS->getName() == RHS->getName() && LHS < RHS;
3005 }
3006 };
3007}
3008
Chris Lattner54cb8fd2005-09-07 23:44:43 +00003009void DAGISelEmitter::EmitInstructionSelector(std::ostream &OS) {
Chris Lattnerb277cbc2005-10-18 04:41:01 +00003010 std::string InstNS = Target.inst_begin()->second.Namespace;
3011 if (!InstNS.empty()) InstNS += "::";
3012
Chris Lattner602f6922006-01-04 00:25:00 +00003013 // Group the patterns by their top-level opcodes.
3014 std::map<Record*, std::vector<PatternToMatch*>,
3015 CompareByRecordName> PatternsByOpcode;
3016 for (unsigned i = 0, e = PatternsToMatch.size(); i != e; ++i) {
3017 TreePatternNode *Node = PatternsToMatch[i].getSrcPattern();
3018 if (!Node->isLeaf()) {
3019 PatternsByOpcode[Node->getOperator()].push_back(&PatternsToMatch[i]);
3020 } else {
3021 const ComplexPattern *CP;
3022 if (IntInit *II =
3023 dynamic_cast<IntInit*>(Node->getLeafValue())) {
3024 PatternsByOpcode[getSDNodeNamed("imm")].push_back(&PatternsToMatch[i]);
3025 } else if ((CP = NodeGetComplexPattern(Node, *this))) {
3026 std::vector<Record*> OpNodes = CP->getRootNodes();
3027 for (unsigned j = 0, e = OpNodes.size(); j != e; j++) {
Chris Lattner488580c2006-01-28 19:06:51 +00003028 PatternsByOpcode[OpNodes[j]]
3029 .insert(PatternsByOpcode[OpNodes[j]].begin(), &PatternsToMatch[i]);
Chris Lattner602f6922006-01-04 00:25:00 +00003030 }
3031 } else {
3032 std::cerr << "Unrecognized opcode '";
3033 Node->dump();
3034 std::cerr << "' on tree pattern '";
Chris Lattner488580c2006-01-28 19:06:51 +00003035 std::cerr <<
3036 PatternsToMatch[i].getDstPattern()->getOperator()->getName();
Chris Lattner602f6922006-01-04 00:25:00 +00003037 std::cerr << "'!\n";
3038 exit(1);
3039 }
3040 }
3041 }
3042
3043 // Emit one Select_* method for each top-level opcode. We do this instead of
3044 // emitting one giant switch statement to support compilers where this will
3045 // result in the recursive functions taking less stack space.
3046 for (std::map<Record*, std::vector<PatternToMatch*>,
3047 CompareByRecordName>::iterator PBOI = PatternsByOpcode.begin(),
3048 E = PatternsByOpcode.end(); PBOI != E; ++PBOI) {
Evan Chenge41bf822006-02-05 06:43:12 +00003049 const std::string &OpName = PBOI->first->getName();
Evan Cheng34167212006-02-09 00:37:58 +00003050 OS << "void Select_" << OpName << "(SDOperand &Result, SDOperand N) {\n";
Chris Lattner602f6922006-01-04 00:25:00 +00003051
3052 const SDNodeInfo &OpcodeInfo = getSDNodeInfo(PBOI->first);
Evan Chenge41bf822006-02-05 06:43:12 +00003053 bool OptSlctOrder =
3054 (OpcodeInfo.hasProperty(SDNodeInfo::SDNPHasChain) &&
3055 OpcodeInfo.getNumResults() > 0);
3056
3057 if (OptSlctOrder) {
Evan Chenge41bf822006-02-05 06:43:12 +00003058 OS << " if (N.ResNo == " << OpcodeInfo.getNumResults()
3059 << " && N.getValue(0).hasOneUse()) {\n"
3060 << " SDOperand Dummy = "
3061 << "CurDAG->getNode(ISD::HANDLENODE, MVT::Other, N);\n"
Evan Cheng67212a02006-02-09 22:12:27 +00003062 << " SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, "
3063 << OpcodeInfo.getNumResults() << ", Dummy.Val, 0);\n"
3064 << " SelectionDAG::InsertISelMapEntry(HandleMap, N.Val, "
3065 << OpcodeInfo.getNumResults() << ", Dummy.Val, 0);\n"
Evan Cheng34167212006-02-09 00:37:58 +00003066 << " Result = Dummy;\n"
3067 << " return;\n"
Evan Chenge41bf822006-02-05 06:43:12 +00003068 << " }\n";
3069 }
3070
Chris Lattner602f6922006-01-04 00:25:00 +00003071 std::vector<PatternToMatch*> &Patterns = PBOI->second;
Chris Lattner355408b2006-01-29 02:43:35 +00003072 assert(!Patterns.empty() && "No patterns but map has entry?");
Chris Lattner602f6922006-01-04 00:25:00 +00003073
3074 // We want to emit all of the matching code now. However, we want to emit
3075 // the matches in order of minimal cost. Sort the patterns so the least
3076 // cost one is at the start.
3077 std::stable_sort(Patterns.begin(), Patterns.end(),
3078 PatternSortingPredicate(*this));
Evan Cheng21ad3922006-02-07 00:37:41 +00003079
Chris Lattner2bd4dd72006-01-29 02:57:39 +00003080 typedef std::vector<std::pair<bool, std::string> > CodeList;
Evan Cheng21ad3922006-02-07 00:37:41 +00003081 typedef std::set<std::string> DeclSet;
Chris Lattner2bd4dd72006-01-29 02:57:39 +00003082
3083 std::vector<std::pair<PatternToMatch*, CodeList> > CodeForPatterns;
Evan Chengd7805a72006-02-09 07:16:09 +00003084 std::set<std::pair<bool, std::string> > GeneratedDecl;
Jeff Cohen9b0ffca2006-01-27 22:22:28 +00003085 for (unsigned i = 0, e = Patterns.size(); i != e; ++i) {
Chris Lattner2bd4dd72006-01-29 02:57:39 +00003086 CodeList GeneratedCode;
Evan Cheng21ad3922006-02-07 00:37:41 +00003087 GenerateCodeForPattern(*Patterns[i], GeneratedCode, GeneratedDecl,
3088 OptSlctOrder);
Chris Lattner2bd4dd72006-01-29 02:57:39 +00003089 CodeForPatterns.push_back(std::make_pair(Patterns[i], GeneratedCode));
3090 }
3091
3092 // Scan the code to see if all of the patterns are reachable and if it is
3093 // possible that the last one might not match.
3094 bool mightNotMatch = true;
3095 for (unsigned i = 0, e = CodeForPatterns.size(); i != e; ++i) {
3096 CodeList &GeneratedCode = CodeForPatterns[i].second;
3097 mightNotMatch = false;
Chris Lattner355408b2006-01-29 02:43:35 +00003098
Chris Lattner2bd4dd72006-01-29 02:57:39 +00003099 for (unsigned j = 0, e = GeneratedCode.size(); j != e; ++j) {
3100 if (GeneratedCode[j].first) { // predicate.
3101 mightNotMatch = true;
3102 break;
3103 }
3104 }
Chris Lattner355408b2006-01-29 02:43:35 +00003105
Chris Lattner2bd4dd72006-01-29 02:57:39 +00003106 // If this pattern definitely matches, and if it isn't the last one, the
3107 // patterns after it CANNOT ever match. Error out.
3108 if (mightNotMatch == false && i != CodeForPatterns.size()-1) {
3109 std::cerr << "Pattern '";
3110 CodeForPatterns[i+1].first->getSrcPattern()->print(OS);
3111 std::cerr << "' is impossible to select!\n";
3112 exit(1);
3113 }
3114 }
Evan Cheng21ad3922006-02-07 00:37:41 +00003115
3116 // Print all declarations.
Chris Lattner947604b2006-03-24 21:52:20 +00003117 for (std::set<std::pair<bool, std::string> >::iterator
3118 I = GeneratedDecl.begin(), E = GeneratedDecl.end(); I != E; ++I)
Evan Chengd7805a72006-02-09 07:16:09 +00003119 if (I->first)
3120 OS << " SDNode *" << I->second << ";\n";
3121 else
3122 OS << " SDOperand " << I->second << "(0, 0);\n";
Evan Cheng21ad3922006-02-07 00:37:41 +00003123
Chris Lattner8bc74722006-01-29 04:25:26 +00003124 // Loop through and reverse all of the CodeList vectors, as we will be
3125 // accessing them from their logical front, but accessing the end of a
3126 // vector is more efficient.
Chris Lattner2bd4dd72006-01-29 02:57:39 +00003127 for (unsigned i = 0, e = CodeForPatterns.size(); i != e; ++i) {
3128 CodeList &GeneratedCode = CodeForPatterns[i].second;
Chris Lattner8bc74722006-01-29 04:25:26 +00003129 std::reverse(GeneratedCode.begin(), GeneratedCode.end());
Jeff Cohen9b0ffca2006-01-27 22:22:28 +00003130 }
Chris Lattner602f6922006-01-04 00:25:00 +00003131
Chris Lattner8bc74722006-01-29 04:25:26 +00003132 // Next, reverse the list of patterns itself for the same reason.
3133 std::reverse(CodeForPatterns.begin(), CodeForPatterns.end());
3134
3135 // Emit all of the patterns now, grouped together to share code.
3136 EmitPatterns(CodeForPatterns, 2, OS);
3137
Chris Lattner2bd4dd72006-01-29 02:57:39 +00003138 // If the last pattern has predicates (which could fail) emit code to catch
3139 // the case where nothing handles a pattern.
Chris Lattnerb026e702006-03-28 00:41:33 +00003140 if (mightNotMatch) {
3141 OS << " std::cerr << \"Cannot yet select: \";\n";
3142 if (OpcodeInfo.getEnumName() != "ISD::INTRINSIC_W_CHAIN" &&
3143 OpcodeInfo.getEnumName() != "ISD::INTRINSIC_WO_CHAIN" &&
3144 OpcodeInfo.getEnumName() != "ISD::INTRINSIC_VOID") {
3145 OS << " N.Val->dump(CurDAG);\n";
3146 } else {
3147 OS << " unsigned iid = cast<ConstantSDNode>(N.getOperand("
3148 "N.getOperand(0).getValueType() == MVT::Other))->getValue();\n"
3149 << " std::cerr << \"intrinsic %\"<< "
3150 "Intrinsic::getName((Intrinsic::ID)iid);\n";
3151 }
3152 OS << " std::cerr << '\\n';\n"
Jeff Cohen9b0ffca2006-01-27 22:22:28 +00003153 << " abort();\n";
Chris Lattnerb026e702006-03-28 00:41:33 +00003154 }
Jeff Cohen9b0ffca2006-01-27 22:22:28 +00003155 OS << "}\n\n";
Chris Lattner602f6922006-01-04 00:25:00 +00003156 }
3157
Chris Lattner54cb8fd2005-09-07 23:44:43 +00003158 // Emit boilerplate.
Evan Cheng34167212006-02-09 00:37:58 +00003159 OS << "void Select_INLINEASM(SDOperand& Result, SDOperand N) {\n"
Chris Lattnerfabcb7a2006-01-26 23:08:55 +00003160 << " std::vector<SDOperand> Ops(N.Val->op_begin(), N.Val->op_end());\n"
Evan Cheng34167212006-02-09 00:37:58 +00003161 << " Select(Ops[0], N.getOperand(0)); // Select the chain.\n\n"
Chris Lattnerfabcb7a2006-01-26 23:08:55 +00003162 << " // Select the flag operand.\n"
3163 << " if (Ops.back().getValueType() == MVT::Flag)\n"
Evan Cheng34167212006-02-09 00:37:58 +00003164 << " Select(Ops.back(), Ops.back());\n"
Chris Lattnerfd105d42006-02-24 02:13:31 +00003165 << " SelectInlineAsmMemoryOperands(Ops, *CurDAG);\n"
Chris Lattnerfabcb7a2006-01-26 23:08:55 +00003166 << " std::vector<MVT::ValueType> VTs;\n"
3167 << " VTs.push_back(MVT::Other);\n"
3168 << " VTs.push_back(MVT::Flag);\n"
3169 << " SDOperand New = CurDAG->getNode(ISD::INLINEASM, VTs, Ops);\n"
Evan Cheng67212a02006-02-09 22:12:27 +00003170 << " SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, 0, New.Val, 0);\n"
3171 << " SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, 1, New.Val, 1);\n"
Evan Cheng34167212006-02-09 00:37:58 +00003172 << " Result = New.getValue(N.ResNo);\n"
3173 << " return;\n"
Chris Lattnerfabcb7a2006-01-26 23:08:55 +00003174 << "}\n\n";
3175
Chris Lattner54cb8fd2005-09-07 23:44:43 +00003176 OS << "// The main instruction selector code.\n"
Evan Cheng34167212006-02-09 00:37:58 +00003177 << "void SelectCode(SDOperand &Result, SDOperand N) {\n"
Chris Lattner547394c2005-09-23 21:53:45 +00003178 << " if (N.getOpcode() >= ISD::BUILTIN_OP_END &&\n"
Chris Lattnerb277cbc2005-10-18 04:41:01 +00003179 << " N.getOpcode() < (ISD::BUILTIN_OP_END+" << InstNS
Evan Cheng34167212006-02-09 00:37:58 +00003180 << "INSTRUCTION_LIST_END)) {\n"
3181 << " Result = N;\n"
3182 << " return; // Already selected.\n"
3183 << " }\n\n"
Chris Lattner755dd092005-12-18 15:28:25 +00003184 << " std::map<SDOperand, SDOperand>::iterator CGMI = CodeGenMap.find(N);\n"
Evan Cheng34167212006-02-09 00:37:58 +00003185 << " if (CGMI != CodeGenMap.end()) {\n"
3186 << " Result = CGMI->second;\n"
3187 << " return;\n"
3188 << " }\n\n"
Chris Lattner547394c2005-09-23 21:53:45 +00003189 << " switch (N.getOpcode()) {\n"
Chris Lattner54cb8fd2005-09-07 23:44:43 +00003190 << " default: break;\n"
3191 << " case ISD::EntryToken: // These leaves remain the same.\n"
Chris Lattner5216c692005-12-18 21:05:44 +00003192 << " case ISD::BasicBlock:\n"
Chris Lattner8020a522006-01-11 19:52:27 +00003193 << " case ISD::Register:\n"
Evan Cheng0a83ed52006-02-05 08:46:14 +00003194 << " case ISD::HANDLENODE:\n"
Evan Cheng2216d8a2006-02-05 05:22:18 +00003195 << " case ISD::TargetConstant:\n"
3196 << " case ISD::TargetConstantPool:\n"
3197 << " case ISD::TargetFrameIndex:\n"
Evan Cheng34167212006-02-09 00:37:58 +00003198 << " case ISD::TargetGlobalAddress: {\n"
3199 << " Result = N;\n"
3200 << " return;\n"
3201 << " }\n"
Chris Lattner54cb8fd2005-09-07 23:44:43 +00003202 << " case ISD::AssertSext:\n"
Chris Lattnerfab37282005-09-26 22:10:24 +00003203 << " case ISD::AssertZext: {\n"
Evan Cheng34167212006-02-09 00:37:58 +00003204 << " SDOperand Tmp0;\n"
3205 << " Select(Tmp0, N.getOperand(0));\n"
Evan Cheng67212a02006-02-09 22:12:27 +00003206 << " if (!N.Val->hasOneUse())\n"
3207 << " SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, N.ResNo, "
3208 << "Tmp0.Val, Tmp0.ResNo);\n"
Evan Cheng34167212006-02-09 00:37:58 +00003209 << " Result = Tmp0;\n"
3210 << " return;\n"
Chris Lattnerf071bb52005-10-25 20:35:14 +00003211 << " }\n"
3212 << " case ISD::TokenFactor:\n"
3213 << " if (N.getNumOperands() == 2) {\n"
Evan Cheng34167212006-02-09 00:37:58 +00003214 << " SDOperand Op0, Op1;\n"
3215 << " Select(Op0, N.getOperand(0));\n"
3216 << " Select(Op1, N.getOperand(1));\n"
Evan Cheng67212a02006-02-09 22:12:27 +00003217 << " Result = \n"
Chris Lattnerf071bb52005-10-25 20:35:14 +00003218 << " CurDAG->getNode(ISD::TokenFactor, MVT::Other, Op0, Op1);\n"
Evan Cheng67212a02006-02-09 22:12:27 +00003219 << " SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, N.ResNo, "
3220 << "Result.Val, Result.ResNo);\n"
Chris Lattnerf071bb52005-10-25 20:35:14 +00003221 << " } else {\n"
3222 << " std::vector<SDOperand> Ops;\n"
Evan Cheng34167212006-02-09 00:37:58 +00003223 << " for (unsigned i = 0, e = N.getNumOperands(); i != e; ++i) {\n"
3224 << " SDOperand Val;\n"
3225 << " Select(Val, N.getOperand(i));\n"
3226 << " Ops.push_back(Val);\n"
3227 << " }\n"
Evan Cheng67212a02006-02-09 22:12:27 +00003228 << " Result = \n"
3229 << " CurDAG->getNode(ISD::TokenFactor, MVT::Other, Ops);\n"
3230 << " SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, N.ResNo, "
3231 << "Result.Val, Result.ResNo);\n"
Chris Lattnerf071bb52005-10-25 20:35:14 +00003232 << " }\n"
Evan Cheng34167212006-02-09 00:37:58 +00003233 << " return;\n"
Chris Lattnerf071bb52005-10-25 20:35:14 +00003234 << " case ISD::CopyFromReg: {\n"
Evan Cheng34167212006-02-09 00:37:58 +00003235 << " SDOperand Chain;\n"
3236 << " Select(Chain, N.getOperand(0));\n"
Chris Lattner755dd092005-12-18 15:28:25 +00003237 << " unsigned Reg = cast<RegisterSDNode>(N.getOperand(1))->getReg();\n"
3238 << " MVT::ValueType VT = N.Val->getValueType(0);\n"
Chris Lattnerdc464de2005-12-18 15:45:51 +00003239 << " if (N.Val->getNumValues() == 2) {\n"
Evan Cheng34167212006-02-09 00:37:58 +00003240 << " if (Chain == N.getOperand(0)) {\n"
3241 << " Result = N; // No change\n"
3242 << " return;\n"
3243 << " }\n"
Chris Lattner755dd092005-12-18 15:28:25 +00003244 << " SDOperand New = CurDAG->getCopyFromReg(Chain, Reg, VT);\n"
Evan Cheng67212a02006-02-09 22:12:27 +00003245 << " SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, 0, "
3246 << "New.Val, 0);\n"
3247 << " SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, 1, "
3248 << "New.Val, 1);\n"
Evan Cheng34167212006-02-09 00:37:58 +00003249 << " Result = New.getValue(N.ResNo);\n"
3250 << " return;\n"
Chris Lattner755dd092005-12-18 15:28:25 +00003251 << " } else {\n"
Evan Cheng34167212006-02-09 00:37:58 +00003252 << " SDOperand Flag;\n"
3253 << " if (N.getNumOperands() == 3) Select(Flag, N.getOperand(2));\n"
Chris Lattnerdc464de2005-12-18 15:45:51 +00003254 << " if (Chain == N.getOperand(0) &&\n"
Evan Cheng34167212006-02-09 00:37:58 +00003255 << " (N.getNumOperands() == 2 || Flag == N.getOperand(2))) {\n"
3256 << " Result = N; // No change\n"
3257 << " return;\n"
3258 << " }\n"
Chris Lattner755dd092005-12-18 15:28:25 +00003259 << " SDOperand New = CurDAG->getCopyFromReg(Chain, Reg, VT, Flag);\n"
Evan Cheng67212a02006-02-09 22:12:27 +00003260 << " SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, 0, "
3261 << "New.Val, 0);\n"
3262 << " SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, 1, "
3263 << "New.Val, 1);\n"
3264 << " SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, 2, "
3265 << "New.Val, 2);\n"
Evan Cheng34167212006-02-09 00:37:58 +00003266 << " Result = New.getValue(N.ResNo);\n"
3267 << " return;\n"
Chris Lattner755dd092005-12-18 15:28:25 +00003268 << " }\n"
Chris Lattnerf071bb52005-10-25 20:35:14 +00003269 << " }\n"
3270 << " case ISD::CopyToReg: {\n"
Evan Cheng34167212006-02-09 00:37:58 +00003271 << " SDOperand Chain;\n"
3272 << " Select(Chain, N.getOperand(0));\n"
Chris Lattnerdc464de2005-12-18 15:45:51 +00003273 << " unsigned Reg = cast<RegisterSDNode>(N.getOperand(1))->getReg();\n"
Evan Cheng34167212006-02-09 00:37:58 +00003274 << " SDOperand Val;\n"
3275 << " Select(Val, N.getOperand(2));\n"
Evan Chengd7805a72006-02-09 07:16:09 +00003276 << " Result = N;\n"
Chris Lattnerdc464de2005-12-18 15:45:51 +00003277 << " if (N.Val->getNumValues() == 1) {\n"
Chris Lattner755dd092005-12-18 15:28:25 +00003278 << " if (Chain != N.getOperand(0) || Val != N.getOperand(2))\n"
Evan Chengd7805a72006-02-09 07:16:09 +00003279 << " Result = CurDAG->getCopyToReg(Chain, Reg, Val);\n"
Evan Cheng67212a02006-02-09 22:12:27 +00003280 << " SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, 0, "
3281 << "Result.Val, 0);\n"
Chris Lattner755dd092005-12-18 15:28:25 +00003282 << " } else {\n"
Chris Lattner7a8054f2005-12-22 20:37:36 +00003283 << " SDOperand Flag(0, 0);\n"
Evan Cheng34167212006-02-09 00:37:58 +00003284 << " if (N.getNumOperands() == 4) Select(Flag, N.getOperand(3));\n"
Chris Lattner755dd092005-12-18 15:28:25 +00003285 << " if (Chain != N.getOperand(0) || Val != N.getOperand(2) ||\n"
Chris Lattnerdc464de2005-12-18 15:45:51 +00003286 << " (N.getNumOperands() == 4 && Flag != N.getOperand(3)))\n"
Evan Chengd7805a72006-02-09 07:16:09 +00003287 << " Result = CurDAG->getCopyToReg(Chain, Reg, Val, Flag);\n"
Evan Cheng67212a02006-02-09 22:12:27 +00003288 << " SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, 0, "
3289 << "Result.Val, 0);\n"
3290 << " SelectionDAG::InsertISelMapEntry(CodeGenMap, N.Val, 1, "
3291 << "Result.Val, 1);\n"
Evan Chengd7805a72006-02-09 07:16:09 +00003292 << " Result = Result.getValue(N.ResNo);\n"
Chris Lattner755dd092005-12-18 15:28:25 +00003293 << " }\n"
Evan Cheng34167212006-02-09 00:37:58 +00003294 << " return;\n"
Chris Lattnerfabcb7a2006-01-26 23:08:55 +00003295 << " }\n"
Chris Lattner947604b2006-03-24 21:52:20 +00003296 << " case ISD::INLINEASM: Select_INLINEASM(Result, N); return;\n";
Chris Lattnerfabcb7a2006-01-26 23:08:55 +00003297
Chris Lattner54cb8fd2005-09-07 23:44:43 +00003298
Chris Lattner602f6922006-01-04 00:25:00 +00003299 // Loop over all of the case statements, emiting a call to each method we
3300 // emitted above.
Chris Lattner37481472005-09-26 21:59:35 +00003301 for (std::map<Record*, std::vector<PatternToMatch*>,
3302 CompareByRecordName>::iterator PBOI = PatternsByOpcode.begin(),
3303 E = PatternsByOpcode.end(); PBOI != E; ++PBOI) {
Chris Lattner81303322005-09-23 19:36:15 +00003304 const SDNodeInfo &OpcodeInfo = getSDNodeInfo(PBOI->first);
Chris Lattner602f6922006-01-04 00:25:00 +00003305 OS << " case " << OpcodeInfo.getEnumName() << ": "
Chris Lattner11966a02006-01-04 00:32:01 +00003306 << std::string(std::max(0, int(24-OpcodeInfo.getEnumName().size())), ' ')
Evan Cheng34167212006-02-09 00:37:58 +00003307 << "Select_" << PBOI->first->getName() << "(Result, N); return;\n";
Chris Lattner81303322005-09-23 19:36:15 +00003308 }
Chris Lattner81303322005-09-23 19:36:15 +00003309
Chris Lattner54cb8fd2005-09-07 23:44:43 +00003310 OS << " } // end of big switch.\n\n"
3311 << " std::cerr << \"Cannot yet select: \";\n"
Chris Lattnerb026e702006-03-28 00:41:33 +00003312 << " if (N.getOpcode() != ISD::INTRINSIC_W_CHAIN &&\n"
3313 << " N.getOpcode() != ISD::INTRINSIC_WO_CHAIN &&\n"
3314 << " N.getOpcode() != ISD::INTRINSIC_VOID) {\n"
Chris Lattner9bf2d3e2006-03-25 06:47:53 +00003315 << " N.Val->dump(CurDAG);\n"
3316 << " } else {\n"
3317 << " unsigned iid = cast<ConstantSDNode>(N.getOperand("
3318 "N.getOperand(0).getValueType() == MVT::Other))->getValue();\n"
3319 << " std::cerr << \"intrinsic %\"<< "
3320 "Intrinsic::getName((Intrinsic::ID)iid);\n"
3321 << " }\n"
Chris Lattner54cb8fd2005-09-07 23:44:43 +00003322 << " std::cerr << '\\n';\n"
3323 << " abort();\n"
3324 << "}\n";
3325}
3326
Chris Lattner54cb8fd2005-09-07 23:44:43 +00003327void DAGISelEmitter::run(std::ostream &OS) {
3328 EmitSourceFileHeader("DAG Instruction Selector for the " + Target.getName() +
3329 " target", OS);
3330
Chris Lattner1f39e292005-09-14 00:09:24 +00003331 OS << "// *** NOTE: This file is #included into the middle of the target\n"
3332 << "// *** instruction selector class. These functions are really "
3333 << "methods.\n\n";
Chris Lattnerb9f01eb2005-09-16 00:29:46 +00003334
Chris Lattner296dfe32005-09-24 00:50:51 +00003335 OS << "// Instance var to keep track of multiply used nodes that have \n"
3336 << "// already been selected.\n"
3337 << "std::map<SDOperand, SDOperand> CodeGenMap;\n";
Evan Chenge41bf822006-02-05 06:43:12 +00003338
3339 OS << "// Instance var to keep track of mapping of chain generating nodes\n"
Evan Cheng7cd19d02006-02-06 08:12:55 +00003340 << "// and their place handle nodes.\n";
3341 OS << "std::map<SDOperand, SDOperand> HandleMap;\n";
3342 OS << "// Instance var to keep track of mapping of place handle nodes\n"
Evan Chenge41bf822006-02-05 06:43:12 +00003343 << "// and their replacement nodes.\n";
3344 OS << "std::map<SDOperand, SDOperand> ReplaceMap;\n";
3345
3346 OS << "\n";
3347 OS << "static void findNonImmUse(SDNode* Use, SDNode* Def, bool &found, "
3348 << "std::set<SDNode *> &Visited) {\n";
3349 OS << " if (found || !Visited.insert(Use).second) return;\n";
3350 OS << " for (unsigned i = 0, e = Use->getNumOperands(); i != e; ++i) {\n";
3351 OS << " SDNode *N = Use->getOperand(i).Val;\n";
3352 OS << " if (N->getNodeDepth() >= Def->getNodeDepth()) {\n";
3353 OS << " if (N != Def) {\n";
3354 OS << " findNonImmUse(N, Def, found, Visited);\n";
3355 OS << " } else {\n";
3356 OS << " found = true;\n";
3357 OS << " break;\n";
3358 OS << " }\n";
3359 OS << " }\n";
3360 OS << " }\n";
3361 OS << "}\n";
3362
3363 OS << "\n";
3364 OS << "static bool isNonImmUse(SDNode* Use, SDNode* Def) {\n";
3365 OS << " std::set<SDNode *> Visited;\n";
3366 OS << " bool found = false;\n";
3367 OS << " for (unsigned i = 0, e = Use->getNumOperands(); i != e; ++i) {\n";
3368 OS << " SDNode *N = Use->getOperand(i).Val;\n";
3369 OS << " if (N != Def) {\n";
3370 OS << " findNonImmUse(N, Def, found, Visited);\n";
3371 OS << " if (found) break;\n";
3372 OS << " }\n";
3373 OS << " }\n";
3374 OS << " return found;\n";
3375 OS << "}\n";
3376
3377 OS << "\n";
Evan Cheng024524f2006-02-06 06:03:35 +00003378 OS << "// AddHandleReplacement - Note the pending replacement node for a\n"
Evan Cheng7cd19d02006-02-06 08:12:55 +00003379 << "// handle node in ReplaceMap.\n";
Evan Cheng67212a02006-02-09 22:12:27 +00003380 OS << "void AddHandleReplacement(SDNode *H, unsigned HNum, SDNode *R, "
3381 << "unsigned RNum) {\n";
3382 OS << " SDOperand N(H, HNum);\n";
Evan Cheng7cd19d02006-02-06 08:12:55 +00003383 OS << " std::map<SDOperand, SDOperand>::iterator HMI = HandleMap.find(N);\n";
3384 OS << " if (HMI != HandleMap.end()) {\n";
Evan Cheng67212a02006-02-09 22:12:27 +00003385 OS << " ReplaceMap[HMI->second] = SDOperand(R, RNum);\n";
Evan Cheng7cd19d02006-02-06 08:12:55 +00003386 OS << " HandleMap.erase(N);\n";
Evan Chenge41bf822006-02-05 06:43:12 +00003387 OS << " }\n";
3388 OS << "}\n";
3389
3390 OS << "\n";
Evan Cheng7cd19d02006-02-06 08:12:55 +00003391 OS << "// SelectDanglingHandles - Select replacements for all `dangling`\n";
3392 OS << "// handles.Some handles do not yet have replacements because the\n";
3393 OS << "// nodes they replacements have only dead readers.\n";
3394 OS << "void SelectDanglingHandles() {\n";
3395 OS << " for (std::map<SDOperand, SDOperand>::iterator I = "
3396 << "HandleMap.begin(),\n"
3397 << " E = HandleMap.end(); I != E; ++I) {\n";
3398 OS << " SDOperand N = I->first;\n";
Evan Cheng34167212006-02-09 00:37:58 +00003399 OS << " SDOperand R;\n";
3400 OS << " Select(R, N.getValue(0));\n";
Evan Cheng67212a02006-02-09 22:12:27 +00003401 OS << " AddHandleReplacement(N.Val, N.ResNo, R.Val, R.ResNo);\n";
Evan Cheng7cd19d02006-02-06 08:12:55 +00003402 OS << " }\n";
3403 OS << "}\n";
3404 OS << "\n";
3405 OS << "// ReplaceHandles - Replace all the handles with the real target\n";
Evan Chenge41bf822006-02-05 06:43:12 +00003406 OS << "// specific nodes.\n";
Evan Cheng7cd19d02006-02-06 08:12:55 +00003407 OS << "void ReplaceHandles() {\n";
Evan Chenge41bf822006-02-05 06:43:12 +00003408 OS << " for (std::map<SDOperand, SDOperand>::iterator I = "
3409 << "ReplaceMap.begin(),\n"
3410 << " E = ReplaceMap.end(); I != E; ++I) {\n";
3411 OS << " SDOperand From = I->first;\n";
3412 OS << " SDOperand To = I->second;\n";
3413 OS << " for (SDNode::use_iterator UI = From.Val->use_begin(), "
3414 << "E = From.Val->use_end(); UI != E; ++UI) {\n";
3415 OS << " SDNode *Use = *UI;\n";
3416 OS << " std::vector<SDOperand> Ops;\n";
3417 OS << " for (unsigned i = 0, e = Use->getNumOperands(); i != e; ++i) {\n";
3418 OS << " SDOperand O = Use->getOperand(i);\n";
3419 OS << " if (O.Val == From.Val)\n";
3420 OS << " Ops.push_back(To);\n";
3421 OS << " else\n";
3422 OS << " Ops.push_back(O);\n";
3423 OS << " }\n";
3424 OS << " SDOperand U = SDOperand(Use, 0);\n";
3425 OS << " CurDAG->UpdateNodeOperands(U, Ops);\n";
3426 OS << " }\n";
3427 OS << " }\n";
3428 OS << "}\n";
3429
3430 OS << "\n";
Evan Chenged66e852006-03-09 08:19:11 +00003431 OS << "// UpdateFoldedChain - return a SDOperand of the new chain created\n";
3432 OS << "// if the folding were to happen. This is called when, for example,\n";
3433 OS << "// a load is folded into a store. If the store's chain is the load,\n";
3434 OS << "// then the resulting node's input chain would be the load's input\n";
3435 OS << "// chain. If the store's chain is a TokenFactor and the load's\n";
3436 OS << "// output chain feeds into in, then the new chain is a TokenFactor\n";
3437 OS << "// with the other operands along with the input chain of the load.\n";
3438 OS << "SDOperand UpdateFoldedChain(SelectionDAG *DAG, SDNode *N, "
3439 << "SDNode *Chain, SDNode* &OldTF) {\n";
3440 OS << " OldTF = NULL;\n";
3441 OS << " if (N == Chain) {\n";
3442 OS << " return N->getOperand(0);\n";
3443 OS << " } else if (Chain->getOpcode() == ISD::TokenFactor &&\n";
3444 OS << " N->isOperand(Chain)) {\n";
3445 OS << " SDOperand Ch = SDOperand(Chain, 0);\n";
3446 OS << " std::map<SDOperand, SDOperand>::iterator CGMI = "
3447 << "CodeGenMap.find(Ch);\n";
3448 OS << " if (CGMI != CodeGenMap.end())\n";
3449 OS << " return SDOperand(0, 0);\n";
3450 OS << " OldTF = Chain;\n";
3451 OS << " std::vector<SDOperand> Ops;\n";
3452 OS << " for (unsigned i = 0; i < Chain->getNumOperands(); ++i) {\n";
3453 OS << " SDOperand Op = Chain->getOperand(i);\n";
3454 OS << " if (Op.Val == N)\n";
3455 OS << " Ops.push_back(N->getOperand(0));\n";
3456 OS << " else\n";
3457 OS << " Ops.push_back(Op);\n";
3458 OS << " }\n";
3459 OS << " return DAG->getNode(ISD::TokenFactor, MVT::Other, Ops);\n";
3460 OS << " }\n";
3461 OS << " return SDOperand(0, 0);\n";
3462 OS << "}\n";
3463
3464 OS << "\n";
Evan Chenge41bf822006-02-05 06:43:12 +00003465 OS << "// SelectRoot - Top level entry to DAG isel.\n";
3466 OS << "SDOperand SelectRoot(SDOperand N) {\n";
Evan Cheng34167212006-02-09 00:37:58 +00003467 OS << " SDOperand ResNode;\n";
3468 OS << " Select(ResNode, N);\n";
Evan Cheng7cd19d02006-02-06 08:12:55 +00003469 OS << " SelectDanglingHandles();\n";
3470 OS << " ReplaceHandles();\n";
Evan Chenge41bf822006-02-05 06:43:12 +00003471 OS << " ReplaceMap.clear();\n";
Evan Cheng34167212006-02-09 00:37:58 +00003472 OS << " return ResNode;\n";
Evan Chenge41bf822006-02-05 06:43:12 +00003473 OS << "}\n";
Chris Lattner296dfe32005-09-24 00:50:51 +00003474
Chris Lattner550525e2006-03-24 21:48:51 +00003475 Intrinsics = LoadIntrinsics(Records);
Chris Lattnerca559d02005-09-08 21:03:01 +00003476 ParseNodeInfo();
Chris Lattner24eeeb82005-09-13 21:51:00 +00003477 ParseNodeTransforms(OS);
Evan Cheng0fc71982005-12-08 02:00:36 +00003478 ParseComplexPatterns();
Chris Lattnerb39e4be2005-09-15 02:38:02 +00003479 ParsePatternFragments(OS);
3480 ParseInstructions();
3481 ParsePatterns();
Chris Lattner3f7e9142005-09-23 20:52:47 +00003482
Chris Lattnere97603f2005-09-28 19:27:25 +00003483 // Generate variants. For example, commutative patterns can match
Chris Lattner3f7e9142005-09-23 20:52:47 +00003484 // multiple ways. Add them to PatternsToMatch as well.
Chris Lattnere97603f2005-09-28 19:27:25 +00003485 GenerateVariants();
Chris Lattnerb39e4be2005-09-15 02:38:02 +00003486
Chris Lattnere46e17b2005-09-29 19:28:10 +00003487
3488 DEBUG(std::cerr << "\n\nALL PATTERNS TO MATCH:\n\n";
3489 for (unsigned i = 0, e = PatternsToMatch.size(); i != e; ++i) {
Evan Cheng58e84a62005-12-14 22:02:59 +00003490 std::cerr << "PATTERN: "; PatternsToMatch[i].getSrcPattern()->dump();
3491 std::cerr << "\nRESULT: ";PatternsToMatch[i].getDstPattern()->dump();
Chris Lattnere46e17b2005-09-29 19:28:10 +00003492 std::cerr << "\n";
3493 });
3494
Chris Lattnerb9f01eb2005-09-16 00:29:46 +00003495 // At this point, we have full information about the 'Patterns' we need to
3496 // parse, both implicitly from instructions as well as from explicit pattern
Chris Lattnere97603f2005-09-28 19:27:25 +00003497 // definitions. Emit the resultant instruction selector.
Chris Lattner54cb8fd2005-09-07 23:44:43 +00003498 EmitInstructionSelector(OS);
3499
3500 for (std::map<Record*, TreePattern*>::iterator I = PatternFragments.begin(),
3501 E = PatternFragments.end(); I != E; ++I)
3502 delete I->second;
3503 PatternFragments.clear();
3504
Chris Lattner54cb8fd2005-09-07 23:44:43 +00003505 Instructions.clear();
3506}