blob: 945c21fb615c81e3e43d1a83fa5e9d5a3f3c449d [file] [log] [blame]
Nate Begeman4ebd8052005-09-01 23:24:04 +00001//===-- DAGCombiner.cpp - Implement a DAG node combiner -------------------===//
Nate Begeman1d4d4142005-09-01 00:19:25 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file was developed by Nate Begeman and is distributed under the
6// University of Illinois Open Source License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This pass combines dag nodes to form fewer, simpler DAG nodes. It can be run
11// both before and after the DAG is legalized.
12//
13// FIXME: Missing folds
14// sdiv, udiv, srem, urem (X, const) where X is an integer can be expanded into
15// a sequence of multiplies, shifts, and adds. This should be controlled by
16// some kind of hint from the target that int div is expensive.
17// various folds of mulh[s,u] by constants such as -1, powers of 2, etc.
18//
19// FIXME: Should add a corresponding version of fold AND with
20// ZERO_EXTEND/SIGN_EXTEND by converting them to an ANY_EXTEND node which
21// we don't have yet.
22//
23// FIXME: mul (x, const) -> shifts + adds
Nate Begeman1d4d4142005-09-01 00:19:25 +000024// FIXME: undef values
Nate Begeman1d4d4142005-09-01 00:19:25 +000025// FIXME: zero extend when top bits are 0 -> drop it ?
Nate Begeman4ebd8052005-09-01 23:24:04 +000026// FIXME: make truncate see through SIGN_EXTEND and AND
27// FIXME: sext_in_reg(setcc) on targets that return zero or one, and where
28// EVT != MVT::i1 can drop the sext.
Nate Begeman4ebd8052005-09-01 23:24:04 +000029// FIXME: (sra (sra x, c1), c2) -> (sra x, c1+c2)
Nate Begeman646d7e22005-09-02 21:18:40 +000030// FIXME: verify that getNode can't return extends with an operand whose type
31// is >= to that of the extend.
32// FIXME: divide by zero is currently left unfolded. do we want to turn this
33// into an undef?
Nate Begeman1d4d4142005-09-01 00:19:25 +000034//
35//===----------------------------------------------------------------------===//
36
37#define DEBUG_TYPE "dagcombine"
38#include "llvm/ADT/Statistic.h"
39#include "llvm/CodeGen/SelectionDAG.h"
Nate Begeman2300f552005-09-07 00:15:36 +000040#include "llvm/Support/Debug.h"
Nate Begeman1d4d4142005-09-01 00:19:25 +000041#include "llvm/Support/MathExtras.h"
42#include "llvm/Target/TargetLowering.h"
43#include <cmath>
44using namespace llvm;
45
46namespace {
47 Statistic<> NodesCombined ("dagcombiner", "Number of dag nodes combined");
48
49 class DAGCombiner {
50 SelectionDAG &DAG;
51 TargetLowering &TLI;
Nate Begeman4ebd8052005-09-01 23:24:04 +000052 bool AfterLegalize;
Nate Begeman1d4d4142005-09-01 00:19:25 +000053
54 // Worklist of all of the nodes that need to be simplified.
55 std::vector<SDNode*> WorkList;
56
57 /// AddUsersToWorkList - When an instruction is simplified, add all users of
58 /// the instruction to the work lists because they might get more simplified
59 /// now.
60 ///
61 void AddUsersToWorkList(SDNode *N) {
62 for (SDNode::use_iterator UI = N->use_begin(), UE = N->use_end();
Nate Begeman4ebd8052005-09-01 23:24:04 +000063 UI != UE; ++UI)
64 WorkList.push_back(*UI);
Nate Begeman1d4d4142005-09-01 00:19:25 +000065 }
66
67 /// removeFromWorkList - remove all instances of N from the worklist.
68 void removeFromWorkList(SDNode *N) {
69 WorkList.erase(std::remove(WorkList.begin(), WorkList.end(), N),
70 WorkList.end());
71 }
72
73 /// visit - call the node-specific routine that knows how to fold each
74 /// particular type of node.
Nate Begeman83e75ec2005-09-06 04:43:02 +000075 SDOperand visit(SDNode *N);
Nate Begeman1d4d4142005-09-01 00:19:25 +000076
77 // Visitation implementation - Implement dag node combining for different
78 // node types. The semantics are as follows:
79 // Return Value:
Nate Begeman2300f552005-09-07 00:15:36 +000080 // SDOperand.Val == 0 - No change was made
81 // otherwise - N should be replaced by the returned Operand.
Nate Begeman1d4d4142005-09-01 00:19:25 +000082 //
Nate Begeman83e75ec2005-09-06 04:43:02 +000083 SDOperand visitTokenFactor(SDNode *N);
84 SDOperand visitADD(SDNode *N);
85 SDOperand visitSUB(SDNode *N);
86 SDOperand visitMUL(SDNode *N);
87 SDOperand visitSDIV(SDNode *N);
88 SDOperand visitUDIV(SDNode *N);
89 SDOperand visitSREM(SDNode *N);
90 SDOperand visitUREM(SDNode *N);
91 SDOperand visitMULHU(SDNode *N);
92 SDOperand visitMULHS(SDNode *N);
93 SDOperand visitAND(SDNode *N);
94 SDOperand visitOR(SDNode *N);
95 SDOperand visitXOR(SDNode *N);
96 SDOperand visitSHL(SDNode *N);
97 SDOperand visitSRA(SDNode *N);
98 SDOperand visitSRL(SDNode *N);
99 SDOperand visitCTLZ(SDNode *N);
100 SDOperand visitCTTZ(SDNode *N);
101 SDOperand visitCTPOP(SDNode *N);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000102 // select
103 // select_cc
104 // setcc
Nate Begeman83e75ec2005-09-06 04:43:02 +0000105 SDOperand visitSIGN_EXTEND(SDNode *N);
106 SDOperand visitZERO_EXTEND(SDNode *N);
107 SDOperand visitSIGN_EXTEND_INREG(SDNode *N);
108 SDOperand visitTRUNCATE(SDNode *N);
109 SDOperand visitSINT_TO_FP(SDNode *N);
110 SDOperand visitUINT_TO_FP(SDNode *N);
111 SDOperand visitFP_TO_SINT(SDNode *N);
112 SDOperand visitFP_TO_UINT(SDNode *N);
113 SDOperand visitFP_ROUND(SDNode *N);
114 SDOperand visitFP_ROUND_INREG(SDNode *N);
115 SDOperand visitFP_EXTEND(SDNode *N);
116 SDOperand visitFNEG(SDNode *N);
117 SDOperand visitFABS(SDNode *N);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000118 // brcond
119 // brcondtwoway
120 // br_cc
121 // brtwoway_cc
122public:
123 DAGCombiner(SelectionDAG &D)
Nate Begeman646d7e22005-09-02 21:18:40 +0000124 : DAG(D), TLI(D.getTargetLoweringInfo()), AfterLegalize(false) {}
Nate Begeman1d4d4142005-09-01 00:19:25 +0000125
126 /// Run - runs the dag combiner on all nodes in the work list
Nate Begeman4ebd8052005-09-01 23:24:04 +0000127 void Run(bool RunningAfterLegalize);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000128 };
129}
130
131/// MaskedValueIsZero - Return true if 'V & Mask' is known to be zero. We use
132/// this predicate to simplify operations downstream. V and Mask are known to
133/// be the same type.
134static bool MaskedValueIsZero(const SDOperand &Op, uint64_t Mask,
135 const TargetLowering &TLI) {
136 unsigned SrcBits;
137 if (Mask == 0) return true;
138
139 // If we know the result of a setcc has the top bits zero, use this info.
140 switch (Op.getOpcode()) {
Nate Begeman4ebd8052005-09-01 23:24:04 +0000141 case ISD::Constant:
142 return (cast<ConstantSDNode>(Op)->getValue() & Mask) == 0;
143 case ISD::SETCC:
Nate Begeman646d7e22005-09-02 21:18:40 +0000144 // FIXME: teach this about non ZeroOrOne values, such as 0 or -1
Nate Begeman4ebd8052005-09-01 23:24:04 +0000145 return ((Mask & 1) == 0) &&
146 TLI.getSetCCResultContents() == TargetLowering::ZeroOrOneSetCCResult;
147 case ISD::ZEXTLOAD:
148 SrcBits = MVT::getSizeInBits(cast<VTSDNode>(Op.getOperand(3))->getVT());
149 return (Mask & ((1ULL << SrcBits)-1)) == 0; // Returning only the zext bits.
150 case ISD::ZERO_EXTEND:
151 SrcBits = MVT::getSizeInBits(Op.getOperand(0).getValueType());
152 return MaskedValueIsZero(Op.getOperand(0),Mask & ((1ULL << SrcBits)-1),TLI);
153 case ISD::AssertZext:
154 SrcBits = MVT::getSizeInBits(cast<VTSDNode>(Op.getOperand(1))->getVT());
155 return (Mask & ((1ULL << SrcBits)-1)) == 0; // Returning only the zext bits.
156 case ISD::AND:
157 // (X & C1) & C2 == 0 iff C1 & C2 == 0.
158 if (ConstantSDNode *AndRHS = dyn_cast<ConstantSDNode>(Op.getOperand(1)))
159 return MaskedValueIsZero(Op.getOperand(0),AndRHS->getValue() & Mask, TLI);
160 // FALL THROUGH
161 case ISD::OR:
162 case ISD::XOR:
163 return MaskedValueIsZero(Op.getOperand(0), Mask, TLI) &&
164 MaskedValueIsZero(Op.getOperand(1), Mask, TLI);
165 case ISD::SELECT:
166 return MaskedValueIsZero(Op.getOperand(1), Mask, TLI) &&
167 MaskedValueIsZero(Op.getOperand(2), Mask, TLI);
168 case ISD::SELECT_CC:
169 return MaskedValueIsZero(Op.getOperand(2), Mask, TLI) &&
170 MaskedValueIsZero(Op.getOperand(3), Mask, TLI);
171 case ISD::SRL:
172 // (ushr X, C1) & C2 == 0 iff X & (C2 << C1) == 0
173 if (ConstantSDNode *ShAmt = dyn_cast<ConstantSDNode>(Op.getOperand(1))) {
174 uint64_t NewVal = Mask << ShAmt->getValue();
175 SrcBits = MVT::getSizeInBits(Op.getValueType());
176 if (SrcBits != 64) NewVal &= (1ULL << SrcBits)-1;
177 return MaskedValueIsZero(Op.getOperand(0), NewVal, TLI);
178 }
179 return false;
180 case ISD::SHL:
181 // (ushl X, C1) & C2 == 0 iff X & (C2 >> C1) == 0
182 if (ConstantSDNode *ShAmt = dyn_cast<ConstantSDNode>(Op.getOperand(1))) {
183 uint64_t NewVal = Mask >> ShAmt->getValue();
184 return MaskedValueIsZero(Op.getOperand(0), NewVal, TLI);
185 }
186 return false;
187 case ISD::CTTZ:
188 case ISD::CTLZ:
189 case ISD::CTPOP:
190 // Bit counting instructions can not set the high bits of the result
191 // register. The max number of bits sets depends on the input.
192 return (Mask & (MVT::getSizeInBits(Op.getValueType())*2-1)) == 0;
193
194 // TODO we could handle some SRA cases here.
195 default: break;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000196 }
Nate Begeman1d4d4142005-09-01 00:19:25 +0000197 return false;
198}
199
Nate Begeman4ebd8052005-09-01 23:24:04 +0000200// isSetCCEquivalent - Return true if this node is a setcc, or is a select_cc
201// that selects between the values 1 and 0, making it equivalent to a setcc.
Nate Begeman646d7e22005-09-02 21:18:40 +0000202// Also, set the incoming LHS, RHS, and CC references to the appropriate
203// nodes based on the type of node we are checking. This simplifies life a
204// bit for the callers.
205static bool isSetCCEquivalent(SDOperand N, SDOperand &LHS, SDOperand &RHS,
206 SDOperand &CC) {
207 if (N.getOpcode() == ISD::SETCC) {
208 LHS = N.getOperand(0);
209 RHS = N.getOperand(1);
210 CC = N.getOperand(2);
Nate Begeman4ebd8052005-09-01 23:24:04 +0000211 return true;
Nate Begeman646d7e22005-09-02 21:18:40 +0000212 }
Nate Begeman1d4d4142005-09-01 00:19:25 +0000213 if (N.getOpcode() == ISD::SELECT_CC &&
214 N.getOperand(2).getOpcode() == ISD::Constant &&
215 N.getOperand(3).getOpcode() == ISD::Constant &&
216 cast<ConstantSDNode>(N.getOperand(2))->getValue() == 1 &&
Nate Begeman646d7e22005-09-02 21:18:40 +0000217 cast<ConstantSDNode>(N.getOperand(3))->isNullValue()) {
218 LHS = N.getOperand(0);
219 RHS = N.getOperand(1);
220 CC = N.getOperand(4);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000221 return true;
Nate Begeman646d7e22005-09-02 21:18:40 +0000222 }
Nate Begeman1d4d4142005-09-01 00:19:25 +0000223 return false;
224}
225
Nate Begeman99801192005-09-07 23:25:52 +0000226// isOneUseSetCC - Return true if this is a SetCC-equivalent operation with only
227// one use. If this is true, it allows the users to invert the operation for
228// free when it is profitable to do so.
229static bool isOneUseSetCC(SDOperand N) {
Nate Begeman646d7e22005-09-02 21:18:40 +0000230 SDOperand N0, N1, N2;
Nate Begeman646d7e22005-09-02 21:18:40 +0000231 if (isSetCCEquivalent(N, N0, N1, N2) && N.Val->hasOneUse())
Nate Begeman4ebd8052005-09-01 23:24:04 +0000232 return true;
233 return false;
234}
235
236void DAGCombiner::Run(bool RunningAfterLegalize) {
237 // set the instance variable, so that the various visit routines may use it.
238 AfterLegalize = RunningAfterLegalize;
239
Nate Begeman646d7e22005-09-02 21:18:40 +0000240 // Add all the dag nodes to the worklist.
241 WorkList.insert(WorkList.end(), DAG.allnodes_begin(), DAG.allnodes_end());
Nate Begeman83e75ec2005-09-06 04:43:02 +0000242
Nate Begeman1d4d4142005-09-01 00:19:25 +0000243 // while the worklist isn't empty, inspect the node on the end of it and
244 // try and combine it.
245 while (!WorkList.empty()) {
246 SDNode *N = WorkList.back();
247 WorkList.pop_back();
248
249 // If N has no uses, it is dead. Make sure to revisit all N's operands once
250 // N is deleted from the DAG, since they too may now be dead.
Nate Begeman83e75ec2005-09-06 04:43:02 +0000251 // FIXME: is there a better way to keep from deleting the dag root because
252 // we think it has no uses? This works for now...
253 if (N->use_empty() && N != DAG.getRoot().Val) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000254 for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i)
255 WorkList.push_back(N->getOperand(i).Val);
256
257 DAG.DeleteNode(N);
258 removeFromWorkList(N);
259 continue;
260 }
261
Nate Begeman83e75ec2005-09-06 04:43:02 +0000262 SDOperand RV = visit(N);
263 if (RV.Val) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000264 ++NodesCombined;
Nate Begeman646d7e22005-09-02 21:18:40 +0000265 // If we get back the same node we passed in, rather than a new node or
266 // zero, we know that the node must have defined multiple values and
267 // CombineTo was used. Since CombineTo takes care of the worklist
268 // mechanics for us, we have no work to do in this case.
Nate Begeman83e75ec2005-09-06 04:43:02 +0000269 if (RV.Val != N) {
Nate Begeman2300f552005-09-07 00:15:36 +0000270 DEBUG(std::cerr << "\nReplacing "; N->dump();
271 std::cerr << "\nWith: "; RV.Val->dump();
272 std::cerr << '\n');
Nate Begeman99801192005-09-07 23:25:52 +0000273 DAG.ReplaceAllUsesWith(N, std::vector<SDOperand>(1, RV));
Nate Begeman646d7e22005-09-02 21:18:40 +0000274
275 // Push the new node and any users onto the worklist
Nate Begeman83e75ec2005-09-06 04:43:02 +0000276 WorkList.push_back(RV.Val);
277 AddUsersToWorkList(RV.Val);
Nate Begeman646d7e22005-09-02 21:18:40 +0000278
279 // Nodes can end up on the worklist more than once. Make sure we do
280 // not process a node that has been replaced.
281 removeFromWorkList(N);
282 }
Nate Begeman1d4d4142005-09-01 00:19:25 +0000283 }
284 }
285}
286
Nate Begeman83e75ec2005-09-06 04:43:02 +0000287SDOperand DAGCombiner::visit(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000288 switch(N->getOpcode()) {
289 default: break;
Nate Begeman4942a962005-09-01 00:33:32 +0000290 case ISD::TokenFactor: return visitTokenFactor(N);
Nate Begeman646d7e22005-09-02 21:18:40 +0000291 case ISD::ADD: return visitADD(N);
292 case ISD::SUB: return visitSUB(N);
293 case ISD::MUL: return visitMUL(N);
294 case ISD::SDIV: return visitSDIV(N);
295 case ISD::UDIV: return visitUDIV(N);
296 case ISD::SREM: return visitSREM(N);
297 case ISD::UREM: return visitUREM(N);
298 case ISD::MULHU: return visitMULHU(N);
299 case ISD::MULHS: return visitMULHS(N);
300 case ISD::AND: return visitAND(N);
301 case ISD::OR: return visitOR(N);
302 case ISD::XOR: return visitXOR(N);
303 case ISD::SHL: return visitSHL(N);
304 case ISD::SRA: return visitSRA(N);
305 case ISD::SRL: return visitSRL(N);
306 case ISD::CTLZ: return visitCTLZ(N);
307 case ISD::CTTZ: return visitCTTZ(N);
308 case ISD::CTPOP: return visitCTPOP(N);
309 case ISD::SIGN_EXTEND: return visitSIGN_EXTEND(N);
310 case ISD::ZERO_EXTEND: return visitZERO_EXTEND(N);
311 case ISD::SIGN_EXTEND_INREG: return visitSIGN_EXTEND_INREG(N);
312 case ISD::TRUNCATE: return visitTRUNCATE(N);
313 case ISD::SINT_TO_FP: return visitSINT_TO_FP(N);
314 case ISD::UINT_TO_FP: return visitUINT_TO_FP(N);
315 case ISD::FP_TO_SINT: return visitFP_TO_SINT(N);
316 case ISD::FP_TO_UINT: return visitFP_TO_UINT(N);
317 case ISD::FP_ROUND: return visitFP_ROUND(N);
318 case ISD::FP_ROUND_INREG: return visitFP_ROUND_INREG(N);
319 case ISD::FP_EXTEND: return visitFP_EXTEND(N);
320 case ISD::FNEG: return visitFNEG(N);
321 case ISD::FABS: return visitFABS(N);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000322 }
Nate Begeman83e75ec2005-09-06 04:43:02 +0000323 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000324}
325
Nate Begeman83e75ec2005-09-06 04:43:02 +0000326SDOperand DAGCombiner::visitTokenFactor(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000327 // If the token factor has two operands and one is the entry token, replace
328 // the token factor with the other operand.
329 if (N->getNumOperands() == 2) {
330 if (N->getOperand(0).getOpcode() == ISD::EntryToken)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000331 return N->getOperand(1);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000332 if (N->getOperand(1).getOpcode() == ISD::EntryToken)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000333 return N->getOperand(0);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000334 }
Nate Begeman83e75ec2005-09-06 04:43:02 +0000335 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000336}
337
Nate Begeman83e75ec2005-09-06 04:43:02 +0000338SDOperand DAGCombiner::visitADD(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000339 SDOperand N0 = N->getOperand(0);
340 SDOperand N1 = N->getOperand(1);
Nate Begeman646d7e22005-09-02 21:18:40 +0000341 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
342 ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
343 ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N0);
344 ConstantFPSDNode *N1CFP = dyn_cast<ConstantFPSDNode>(N1);
Nate Begemanf89d78d2005-09-07 16:09:19 +0000345 MVT::ValueType VT = N0.getValueType();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000346
347 // fold (add c1, c2) -> c1+c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000348 if (N0C && N1C)
Nate Begemanf89d78d2005-09-07 16:09:19 +0000349 return DAG.getConstant(N0C->getValue() + N1C->getValue(), VT);
Nate Begeman99801192005-09-07 23:25:52 +0000350 // canonicalize constant to RHS
351 if (N0C && !N1C) {
352 std::swap(N0, N1);
353 std::swap(N0C, N1C);
354 }
Nate Begeman1d4d4142005-09-01 00:19:25 +0000355 // fold (add x, 0) -> x
Nate Begeman646d7e22005-09-02 21:18:40 +0000356 if (N1C && N1C->isNullValue())
Nate Begeman83e75ec2005-09-06 04:43:02 +0000357 return N0;
Nate Begemanf89d78d2005-09-07 16:09:19 +0000358 // fold (add (add x, c1), c2) -> (add x, c1+c2)
359 if (N1C && N0.getOpcode() == ISD::ADD &&
360 N0.getOperand(1).getOpcode() == ISD::Constant)
361 return DAG.getNode(ISD::ADD, VT, N0.getOperand(0),
362 DAG.getConstant(N1C->getValue() +
363 cast<ConstantSDNode>(N0.getOperand(1))->getValue(),
364 VT));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000365 // fold floating point (add c1, c2) -> c1+c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000366 if (N0CFP && N1CFP)
Nate Begemanf89d78d2005-09-07 16:09:19 +0000367 return DAG.getConstantFP(N0CFP->getValue() + N1CFP->getValue(), VT);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000368 // fold (A + (-B)) -> A-B
369 if (N1.getOpcode() == ISD::FNEG)
Nate Begemanf89d78d2005-09-07 16:09:19 +0000370 return DAG.getNode(ISD::SUB, VT, N0, N1.getOperand(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000371 // fold ((-A) + B) -> B-A
372 if (N0.getOpcode() == ISD::FNEG)
Nate Begemanf89d78d2005-09-07 16:09:19 +0000373 return DAG.getNode(ISD::SUB, VT, N1, N0.getOperand(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000374 // fold ((0-A) + B) -> B-A
375 if (N0.getOpcode() == ISD::SUB && isa<ConstantSDNode>(N0.getOperand(0)) &&
376 cast<ConstantSDNode>(N0.getOperand(0))->isNullValue())
Nate Begemanf89d78d2005-09-07 16:09:19 +0000377 return DAG.getNode(ISD::SUB, VT, N1, N0.getOperand(1));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000378 // fold (A + (0-B)) -> A-B
379 if (N1.getOpcode() == ISD::SUB && isa<ConstantSDNode>(N1.getOperand(0)) &&
380 cast<ConstantSDNode>(N1.getOperand(0))->isNullValue())
Nate Begemanf89d78d2005-09-07 16:09:19 +0000381 return DAG.getNode(ISD::SUB, VT, N0, N1.getOperand(1));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000382 // fold (A+(B-A)) -> B for non-fp types
383 if (N1.getOpcode() == ISD::SUB && N0 == N1.getOperand(1) &&
384 !MVT::isFloatingPoint(N1.getValueType()))
Nate Begeman83e75ec2005-09-06 04:43:02 +0000385 return N1.getOperand(0);
386 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000387}
388
Nate Begeman83e75ec2005-09-06 04:43:02 +0000389SDOperand DAGCombiner::visitSUB(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000390 SDOperand N0 = N->getOperand(0);
391 SDOperand N1 = N->getOperand(1);
Nate Begeman646d7e22005-09-02 21:18:40 +0000392 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0.Val);
393 ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1.Val);
394 ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N0.Val);
395 ConstantFPSDNode *N1CFP = dyn_cast<ConstantFPSDNode>(N1.Val);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000396
397 // fold (sub c1, c2) -> c1-c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000398 if (N0C && N1C)
399 return DAG.getConstant(N0C->getValue() - N1C->getValue(),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000400 N->getValueType(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000401 // fold (sub x, 0) -> x
Nate Begeman646d7e22005-09-02 21:18:40 +0000402 if (N1C && N1C->isNullValue())
Nate Begeman83e75ec2005-09-06 04:43:02 +0000403 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000404 // fold floating point (sub c1, c2) -> c1-c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000405 if (N0CFP && N1CFP)
406 return DAG.getConstantFP(N0CFP->getValue() - N1CFP->getValue(),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000407 N->getValueType(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000408 // fold (A+B)-A -> B
409 if (N0.getOpcode() == ISD::ADD && N0.getOperand(0) == N1 &&
410 !MVT::isFloatingPoint(N1.getValueType()))
Nate Begeman83e75ec2005-09-06 04:43:02 +0000411 return N0.getOperand(1);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000412 // fold (A+B)-B -> A
413 if (N0.getOpcode() == ISD::ADD && N0.getOperand(1) == N1 &&
414 !MVT::isFloatingPoint(N1.getValueType()))
Nate Begeman83e75ec2005-09-06 04:43:02 +0000415 return N0.getOperand(0);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000416 // fold (A-(-B)) -> A+B
417 if (N1.getOpcode() == ISD::FNEG)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000418 return DAG.getNode(ISD::ADD, N0.getValueType(), N0, N1.getOperand(0));
419 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000420}
421
Nate Begeman83e75ec2005-09-06 04:43:02 +0000422SDOperand DAGCombiner::visitMUL(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000423 SDOperand N0 = N->getOperand(0);
424 SDOperand N1 = N->getOperand(1);
Nate Begeman646d7e22005-09-02 21:18:40 +0000425 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
426 ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
427 ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N0);
428 ConstantFPSDNode *N1CFP = dyn_cast<ConstantFPSDNode>(N1);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000429
430 // fold (mul c1, c2) -> c1*c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000431 if (N0C && N1C)
432 return DAG.getConstant(N0C->getValue() * N1C->getValue(),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000433 N->getValueType(0));
Nate Begeman99801192005-09-07 23:25:52 +0000434 // canonicalize constant to RHS
435 if (N0C && !N1C) {
436 std::swap(N0, N1);
437 std::swap(N0C, N1C);
438 }
Nate Begeman1d4d4142005-09-01 00:19:25 +0000439 // fold (mul x, 0) -> 0
Nate Begeman646d7e22005-09-02 21:18:40 +0000440 if (N1C && N1C->isNullValue())
Nate Begeman83e75ec2005-09-06 04:43:02 +0000441 return N1;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000442 // fold (mul x, -1) -> 0-x
Nate Begeman646d7e22005-09-02 21:18:40 +0000443 if (N1C && N1C->isAllOnesValue())
Nate Begeman1d4d4142005-09-01 00:19:25 +0000444 return DAG.getNode(ISD::SUB, N->getValueType(0),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000445 DAG.getConstant(0, N->getValueType(0)), N0);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000446 // fold (mul x, (1 << c)) -> x << c
Nate Begeman646d7e22005-09-02 21:18:40 +0000447 if (N1C && isPowerOf2_64(N1C->getValue()))
Nate Begeman1d4d4142005-09-01 00:19:25 +0000448 return DAG.getNode(ISD::SHL, N->getValueType(0), N0,
Nate Begeman646d7e22005-09-02 21:18:40 +0000449 DAG.getConstant(Log2_64(N1C->getValue()),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000450 TLI.getShiftAmountTy()));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000451 // fold floating point (mul c1, c2) -> c1*c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000452 if (N0CFP && N1CFP)
453 return DAG.getConstantFP(N0CFP->getValue() * N1CFP->getValue(),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000454 N->getValueType(0));
455 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000456}
457
Nate Begeman83e75ec2005-09-06 04:43:02 +0000458SDOperand DAGCombiner::visitSDIV(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000459 SDOperand N0 = N->getOperand(0);
460 SDOperand N1 = N->getOperand(1);
Nate Begeman646d7e22005-09-02 21:18:40 +0000461 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0.Val);
462 ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1.Val);
463 ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N0.Val);
464 ConstantFPSDNode *N1CFP = dyn_cast<ConstantFPSDNode>(N1.Val);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000465
466 // fold (sdiv c1, c2) -> c1/c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000467 if (N0C && N1C && !N1C->isNullValue())
468 return DAG.getConstant(N0C->getSignExtended() / N1C->getSignExtended(),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000469 N->getValueType(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000470 // fold floating point (sdiv c1, c2) -> c1/c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000471 if (N0CFP && N1CFP)
472 return DAG.getConstantFP(N0CFP->getValue() / N1CFP->getValue(),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000473 N->getValueType(0));
474 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000475}
476
Nate Begeman83e75ec2005-09-06 04:43:02 +0000477SDOperand DAGCombiner::visitUDIV(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000478 SDOperand N0 = N->getOperand(0);
479 SDOperand N1 = N->getOperand(1);
Nate Begeman646d7e22005-09-02 21:18:40 +0000480 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0.Val);
481 ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1.Val);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000482
483 // fold (udiv c1, c2) -> c1/c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000484 if (N0C && N1C && !N1C->isNullValue())
485 return DAG.getConstant(N0C->getValue() / N1C->getValue(),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000486 N->getValueType(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000487 // fold (udiv x, (1 << c)) -> x >>u c
Nate Begeman646d7e22005-09-02 21:18:40 +0000488 if (N1C && isPowerOf2_64(N1C->getValue()))
Nate Begeman1d4d4142005-09-01 00:19:25 +0000489 return DAG.getNode(ISD::SRL, N->getValueType(0), N0,
Nate Begeman646d7e22005-09-02 21:18:40 +0000490 DAG.getConstant(Log2_64(N1C->getValue()),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000491 TLI.getShiftAmountTy()));
492 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000493}
494
Nate Begeman83e75ec2005-09-06 04:43:02 +0000495SDOperand DAGCombiner::visitSREM(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000496 SDOperand N0 = N->getOperand(0);
497 SDOperand N1 = N->getOperand(1);
Nate Begeman646d7e22005-09-02 21:18:40 +0000498 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
499 ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
500 ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N0);
501 ConstantFPSDNode *N1CFP = dyn_cast<ConstantFPSDNode>(N1);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000502
503 // fold (srem c1, c2) -> c1%c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000504 if (N0C && N1C && !N1C->isNullValue())
505 return DAG.getConstant(N0C->getSignExtended() % N1C->getSignExtended(),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000506 N->getValueType(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000507 // fold floating point (srem c1, c2) -> fmod(c1, c2)
Nate Begeman646d7e22005-09-02 21:18:40 +0000508 if (N0CFP && N1CFP)
509 return DAG.getConstantFP(fmod(N0CFP->getValue(),N1CFP->getValue()),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000510 N->getValueType(0));
511 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000512}
513
Nate Begeman83e75ec2005-09-06 04:43:02 +0000514SDOperand DAGCombiner::visitUREM(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000515 SDOperand N0 = N->getOperand(0);
516 SDOperand N1 = N->getOperand(1);
Nate Begeman646d7e22005-09-02 21:18:40 +0000517 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
518 ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000519
520 // fold (urem c1, c2) -> c1%c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000521 if (N0C && N1C && !N1C->isNullValue())
522 return DAG.getConstant(N0C->getValue() % N1C->getValue(),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000523 N->getValueType(0));
Nate Begeman646d7e22005-09-02 21:18:40 +0000524 // FIXME: c2 power of 2 -> mask?
Nate Begeman83e75ec2005-09-06 04:43:02 +0000525 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000526}
527
Nate Begeman83e75ec2005-09-06 04:43:02 +0000528SDOperand DAGCombiner::visitMULHS(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000529 SDOperand N0 = N->getOperand(0);
530 SDOperand N1 = N->getOperand(1);
Nate Begeman646d7e22005-09-02 21:18:40 +0000531 ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000532
533 // fold (mulhs x, 0) -> 0
Nate Begeman646d7e22005-09-02 21:18:40 +0000534 if (N1C && N1C->isNullValue())
Nate Begeman83e75ec2005-09-06 04:43:02 +0000535 return N1;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000536 // fold (mulhs x, 1) -> (sra x, size(x)-1)
Nate Begeman646d7e22005-09-02 21:18:40 +0000537 if (N1C && N1C->getValue() == 1)
Nate Begeman1d4d4142005-09-01 00:19:25 +0000538 return DAG.getNode(ISD::SRA, N0.getValueType(), N0,
539 DAG.getConstant(MVT::getSizeInBits(N0.getValueType())-1,
Nate Begeman83e75ec2005-09-06 04:43:02 +0000540 TLI.getShiftAmountTy()));
541 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000542}
543
Nate Begeman83e75ec2005-09-06 04:43:02 +0000544SDOperand DAGCombiner::visitMULHU(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000545 SDOperand N0 = N->getOperand(0);
546 SDOperand N1 = N->getOperand(1);
Nate Begeman646d7e22005-09-02 21:18:40 +0000547 ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000548
549 // fold (mulhu x, 0) -> 0
Nate Begeman646d7e22005-09-02 21:18:40 +0000550 if (N1C && N1C->isNullValue())
Nate Begeman83e75ec2005-09-06 04:43:02 +0000551 return N1;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000552 // fold (mulhu x, 1) -> 0
Nate Begeman646d7e22005-09-02 21:18:40 +0000553 if (N1C && N1C->getValue() == 1)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000554 return DAG.getConstant(0, N0.getValueType());
555 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000556}
557
Nate Begeman83e75ec2005-09-06 04:43:02 +0000558SDOperand DAGCombiner::visitAND(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000559 SDOperand N0 = N->getOperand(0);
560 SDOperand N1 = N->getOperand(1);
Nate Begeman646d7e22005-09-02 21:18:40 +0000561 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
562 ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000563 MVT::ValueType VT = N1.getValueType();
Nate Begeman83e75ec2005-09-06 04:43:02 +0000564 unsigned OpSizeInBits = MVT::getSizeInBits(VT);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000565
566 // fold (and c1, c2) -> c1&c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000567 if (N0C && N1C)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000568 return DAG.getConstant(N0C->getValue() & N1C->getValue(), VT);
Nate Begeman99801192005-09-07 23:25:52 +0000569 // canonicalize constant to RHS
570 if (N0C && !N1C) {
571 std::swap(N0, N1);
572 std::swap(N0C, N1C);
573 }
Nate Begeman1d4d4142005-09-01 00:19:25 +0000574 // fold (and x, -1) -> x
Nate Begeman646d7e22005-09-02 21:18:40 +0000575 if (N1C && N1C->isAllOnesValue())
Nate Begeman83e75ec2005-09-06 04:43:02 +0000576 return N0;
577 // if (and x, c) is known to be zero, return 0
578 if (N1C && MaskedValueIsZero(SDOperand(N, 0), ~0ULL >> (64-OpSizeInBits),TLI))
579 return DAG.getConstant(0, VT);
580 // fold (and x, c) -> x iff (x & ~c) == 0
581 if (N1C && MaskedValueIsZero(N0,~N1C->getValue() & (~0ULL>>(64-OpSizeInBits)),
582 TLI))
583 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000584 // fold (and (sign_extend_inreg x, i16 to i32), 1) -> (and x, 1)
585 if (N0.getOpcode() == ISD::SIGN_EXTEND_INREG) {
586 unsigned ExtendBits =
587 MVT::getSizeInBits(cast<VTSDNode>(N0.getOperand(1))->getVT());
Nate Begeman646d7e22005-09-02 21:18:40 +0000588 if ((N1C->getValue() & (~0ULL << ExtendBits)) == 0)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000589 return DAG.getNode(ISD::AND, VT, N0.getOperand(0), N1);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000590 }
591 // fold (and (or x, 0xFFFF), 0xFF) -> 0xFF
592 if (N0.getOpcode() == ISD::OR)
593 if (ConstantSDNode *ORI = dyn_cast<ConstantSDNode>(N0.getOperand(1)))
Nate Begeman646d7e22005-09-02 21:18:40 +0000594 if ((ORI->getValue() & N1C->getValue()) == N1C->getValue())
Nate Begeman83e75ec2005-09-06 04:43:02 +0000595 return N1;
596 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000597}
598
Nate Begeman83e75ec2005-09-06 04:43:02 +0000599SDOperand DAGCombiner::visitOR(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000600 SDOperand N0 = N->getOperand(0);
601 SDOperand N1 = N->getOperand(1);
Nate Begeman646d7e22005-09-02 21:18:40 +0000602 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
603 ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
Nate Begeman83e75ec2005-09-06 04:43:02 +0000604 MVT::ValueType VT = N1.getValueType();
605 unsigned OpSizeInBits = MVT::getSizeInBits(VT);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000606
607 // fold (or c1, c2) -> c1|c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000608 if (N0C && N1C)
609 return DAG.getConstant(N0C->getValue() | N1C->getValue(),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000610 N->getValueType(0));
Nate Begeman99801192005-09-07 23:25:52 +0000611 // canonicalize constant to RHS
612 if (N0C && !N1C) {
613 std::swap(N0, N1);
614 std::swap(N0C, N1C);
615 }
Nate Begeman1d4d4142005-09-01 00:19:25 +0000616 // fold (or x, 0) -> x
Nate Begeman646d7e22005-09-02 21:18:40 +0000617 if (N1C && N1C->isNullValue())
Nate Begeman83e75ec2005-09-06 04:43:02 +0000618 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000619 // fold (or x, -1) -> -1
Nate Begeman646d7e22005-09-02 21:18:40 +0000620 if (N1C && N1C->isAllOnesValue())
Nate Begeman83e75ec2005-09-06 04:43:02 +0000621 return N1;
622 // fold (or x, c) -> c iff (x & ~c) == 0
623 if (N1C && MaskedValueIsZero(N0,~N1C->getValue() & (~0ULL>>(64-OpSizeInBits)),
624 TLI))
625 return N1;
626 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000627}
628
Nate Begeman83e75ec2005-09-06 04:43:02 +0000629SDOperand DAGCombiner::visitXOR(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000630 SDOperand N0 = N->getOperand(0);
631 SDOperand N1 = N->getOperand(1);
Nate Begeman646d7e22005-09-02 21:18:40 +0000632 SDOperand LHS, RHS, CC;
633 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
634 ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000635 MVT::ValueType VT = N0.getValueType();
636
637 // fold (xor c1, c2) -> c1^c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000638 if (N0C && N1C)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000639 return DAG.getConstant(N0C->getValue() ^ N1C->getValue(), VT);
Nate Begeman99801192005-09-07 23:25:52 +0000640 // canonicalize constant to RHS
641 if (N0C && !N1C) {
642 std::swap(N0, N1);
643 std::swap(N0C, N1C);
644 }
Nate Begeman1d4d4142005-09-01 00:19:25 +0000645 // fold (xor x, 0) -> x
Nate Begeman646d7e22005-09-02 21:18:40 +0000646 if (N1C && N1C->isNullValue())
Nate Begeman83e75ec2005-09-06 04:43:02 +0000647 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000648 // fold !(x cc y) -> (x !cc y)
Nate Begeman646d7e22005-09-02 21:18:40 +0000649 if (N1C && N1C->getValue() == 1 && isSetCCEquivalent(N0, LHS, RHS, CC)) {
650 bool isInt = MVT::isInteger(LHS.getValueType());
651 ISD::CondCode NotCC = ISD::getSetCCInverse(cast<CondCodeSDNode>(CC)->get(),
652 isInt);
653 if (N0.getOpcode() == ISD::SETCC)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000654 return DAG.getSetCC(VT, LHS, RHS, NotCC);
Nate Begeman646d7e22005-09-02 21:18:40 +0000655 if (N0.getOpcode() == ISD::SELECT_CC)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000656 return DAG.getSelectCC(LHS, RHS, N0.getOperand(2),N0.getOperand(3),NotCC);
Nate Begeman646d7e22005-09-02 21:18:40 +0000657 assert(0 && "Unhandled SetCC Equivalent!");
658 abort();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000659 }
Nate Begeman99801192005-09-07 23:25:52 +0000660 // fold !(x or y) -> (!x and !y) iff x or y are setcc
661 if (N1C && N1C->getValue() == 1 &&
662 (N0.getOpcode() == ISD::OR || N0.getOpcode() == ISD::AND)) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000663 SDOperand LHS = N0.getOperand(0), RHS = N0.getOperand(1);
Nate Begeman99801192005-09-07 23:25:52 +0000664 if (isOneUseSetCC(RHS) || isOneUseSetCC(LHS)) {
665 unsigned NewOpcode = N0.getOpcode() == ISD::AND ? ISD::OR : ISD::AND;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000666 LHS = DAG.getNode(ISD::XOR, VT, LHS, N1); // RHS = ~LHS
667 RHS = DAG.getNode(ISD::XOR, VT, RHS, N1); // RHS = ~RHS
Nate Begeman99801192005-09-07 23:25:52 +0000668 WorkList.push_back(LHS.Val); WorkList.push_back(RHS.Val);
669 return DAG.getNode(NewOpcode, VT, LHS, RHS);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000670 }
671 }
Nate Begeman99801192005-09-07 23:25:52 +0000672 // fold !(x or y) -> (!x and !y) iff x or y are constants
673 if (N1C && N1C->isAllOnesValue() &&
674 (N0.getOpcode() == ISD::OR || N0.getOpcode() == ISD::AND)) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000675 SDOperand LHS = N0.getOperand(0), RHS = N0.getOperand(1);
Nate Begeman99801192005-09-07 23:25:52 +0000676 if (isa<ConstantSDNode>(RHS) || isa<ConstantSDNode>(LHS)) {
677 unsigned NewOpcode = N0.getOpcode() == ISD::AND ? ISD::OR : ISD::AND;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000678 LHS = DAG.getNode(ISD::XOR, VT, LHS, N1); // RHS = ~LHS
679 RHS = DAG.getNode(ISD::XOR, VT, RHS, N1); // RHS = ~RHS
Nate Begeman99801192005-09-07 23:25:52 +0000680 WorkList.push_back(LHS.Val); WorkList.push_back(RHS.Val);
681 return DAG.getNode(NewOpcode, VT, LHS, RHS);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000682 }
683 }
Nate Begeman83e75ec2005-09-06 04:43:02 +0000684 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000685}
686
Nate Begeman83e75ec2005-09-06 04:43:02 +0000687SDOperand DAGCombiner::visitSHL(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000688 SDOperand N0 = N->getOperand(0);
689 SDOperand N1 = N->getOperand(1);
Nate Begeman646d7e22005-09-02 21:18:40 +0000690 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
691 ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000692 MVT::ValueType VT = N0.getValueType();
693 unsigned OpSizeInBits = MVT::getSizeInBits(VT);
694
695 // fold (shl c1, c2) -> c1<<c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000696 if (N0C && N1C)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000697 return DAG.getConstant(N0C->getValue() << N1C->getValue(), VT);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000698 // fold (shl 0, x) -> 0
Nate Begeman646d7e22005-09-02 21:18:40 +0000699 if (N0C && N0C->isNullValue())
Nate Begeman83e75ec2005-09-06 04:43:02 +0000700 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000701 // fold (shl x, c >= size(x)) -> undef
Nate Begeman646d7e22005-09-02 21:18:40 +0000702 if (N1C && N1C->getValue() >= OpSizeInBits)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000703 return DAG.getNode(ISD::UNDEF, VT);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000704 // fold (shl x, 0) -> x
Nate Begeman646d7e22005-09-02 21:18:40 +0000705 if (N1C && N1C->isNullValue())
Nate Begeman83e75ec2005-09-06 04:43:02 +0000706 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000707 // if (shl x, c) is known to be zero, return 0
Nate Begeman83e75ec2005-09-06 04:43:02 +0000708 if (N1C && MaskedValueIsZero(SDOperand(N, 0), ~0ULL >> (64-OpSizeInBits),TLI))
709 return DAG.getConstant(0, VT);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000710 // fold (shl (shl x, c1), c2) -> 0 or (shl x, c1+c2)
Nate Begeman646d7e22005-09-02 21:18:40 +0000711 if (N1C && N0.getOpcode() == ISD::SHL &&
Nate Begeman1d4d4142005-09-01 00:19:25 +0000712 N0.getOperand(1).getOpcode() == ISD::Constant) {
713 uint64_t c1 = cast<ConstantSDNode>(N0.getOperand(1))->getValue();
Nate Begeman646d7e22005-09-02 21:18:40 +0000714 uint64_t c2 = N1C->getValue();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000715 if (c1 + c2 > OpSizeInBits)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000716 return DAG.getConstant(0, VT);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000717 return DAG.getNode(ISD::SHL, VT, N0.getOperand(0),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000718 DAG.getConstant(c1 + c2, N1.getValueType()));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000719 }
720 // fold (shl (srl x, c1), c2) -> (shl (and x, -1 << c1), c2-c1) or
721 // (srl (and x, -1 << c1), c1-c2)
Nate Begeman646d7e22005-09-02 21:18:40 +0000722 if (N1C && N0.getOpcode() == ISD::SRL &&
Nate Begeman1d4d4142005-09-01 00:19:25 +0000723 N0.getOperand(1).getOpcode() == ISD::Constant) {
724 uint64_t c1 = cast<ConstantSDNode>(N0.getOperand(1))->getValue();
Nate Begeman646d7e22005-09-02 21:18:40 +0000725 uint64_t c2 = N1C->getValue();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000726 SDOperand Mask = DAG.getNode(ISD::AND, VT, N0.getOperand(0),
727 DAG.getConstant(~0ULL << c1, VT));
728 if (c2 > c1)
729 return DAG.getNode(ISD::SHL, VT, Mask,
Nate Begeman83e75ec2005-09-06 04:43:02 +0000730 DAG.getConstant(c2-c1, N1.getValueType()));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000731 else
Nate Begeman83e75ec2005-09-06 04:43:02 +0000732 return DAG.getNode(ISD::SRL, VT, Mask,
733 DAG.getConstant(c1-c2, N1.getValueType()));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000734 }
735 // fold (shl (sra x, c1), c1) -> (and x, -1 << c1)
Nate Begeman646d7e22005-09-02 21:18:40 +0000736 if (N1C && N0.getOpcode() == ISD::SRA && N1 == N0.getOperand(1))
Nate Begeman4ebd8052005-09-01 23:24:04 +0000737 return DAG.getNode(ISD::AND, VT, N0.getOperand(0),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000738 DAG.getConstant(~0ULL << N1C->getValue(), VT));
739 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000740}
741
Nate Begeman83e75ec2005-09-06 04:43:02 +0000742SDOperand DAGCombiner::visitSRA(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000743 SDOperand N0 = N->getOperand(0);
744 SDOperand N1 = N->getOperand(1);
Nate Begeman646d7e22005-09-02 21:18:40 +0000745 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
746 ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000747 MVT::ValueType VT = N0.getValueType();
748 unsigned OpSizeInBits = MVT::getSizeInBits(VT);
749
750 // fold (sra c1, c2) -> c1>>c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000751 if (N0C && N1C)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000752 return DAG.getConstant(N0C->getSignExtended() >> N1C->getValue(), VT);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000753 // fold (sra 0, x) -> 0
Nate Begeman646d7e22005-09-02 21:18:40 +0000754 if (N0C && N0C->isNullValue())
Nate Begeman83e75ec2005-09-06 04:43:02 +0000755 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000756 // fold (sra -1, x) -> -1
Nate Begeman646d7e22005-09-02 21:18:40 +0000757 if (N0C && N0C->isAllOnesValue())
Nate Begeman83e75ec2005-09-06 04:43:02 +0000758 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000759 // fold (sra x, c >= size(x)) -> undef
Nate Begeman646d7e22005-09-02 21:18:40 +0000760 if (N1C && N1C->getValue() >= OpSizeInBits)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000761 return DAG.getNode(ISD::UNDEF, VT);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000762 // fold (sra x, 0) -> x
Nate Begeman646d7e22005-09-02 21:18:40 +0000763 if (N1C && N1C->isNullValue())
Nate Begeman83e75ec2005-09-06 04:43:02 +0000764 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000765 // If the sign bit is known to be zero, switch this to a SRL.
Nate Begeman646d7e22005-09-02 21:18:40 +0000766 if (N1C && MaskedValueIsZero(N0, (1ULL << (OpSizeInBits-1)), TLI))
Nate Begeman83e75ec2005-09-06 04:43:02 +0000767 return DAG.getNode(ISD::SRL, VT, N0, N1);
768 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000769}
770
Nate Begeman83e75ec2005-09-06 04:43:02 +0000771SDOperand DAGCombiner::visitSRL(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000772 SDOperand N0 = N->getOperand(0);
773 SDOperand N1 = N->getOperand(1);
Nate Begeman646d7e22005-09-02 21:18:40 +0000774 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
775 ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000776 MVT::ValueType VT = N0.getValueType();
777 unsigned OpSizeInBits = MVT::getSizeInBits(VT);
778
779 // fold (srl c1, c2) -> c1 >>u c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000780 if (N0C && N1C)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000781 return DAG.getConstant(N0C->getValue() >> N1C->getValue(), VT);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000782 // fold (srl 0, x) -> 0
Nate Begeman646d7e22005-09-02 21:18:40 +0000783 if (N0C && N0C->isNullValue())
Nate Begeman83e75ec2005-09-06 04:43:02 +0000784 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000785 // fold (srl x, c >= size(x)) -> undef
Nate Begeman646d7e22005-09-02 21:18:40 +0000786 if (N1C && N1C->getValue() >= OpSizeInBits)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000787 return DAG.getNode(ISD::UNDEF, VT);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000788 // fold (srl x, 0) -> x
Nate Begeman646d7e22005-09-02 21:18:40 +0000789 if (N1C && N1C->isNullValue())
Nate Begeman83e75ec2005-09-06 04:43:02 +0000790 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000791 // if (srl x, c) is known to be zero, return 0
Nate Begeman83e75ec2005-09-06 04:43:02 +0000792 if (N1C && MaskedValueIsZero(SDOperand(N, 0), ~0ULL >> (64-OpSizeInBits),TLI))
793 return DAG.getConstant(0, VT);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000794 // fold (srl (srl x, c1), c2) -> 0 or (srl x, c1+c2)
Nate Begeman646d7e22005-09-02 21:18:40 +0000795 if (N1C && N0.getOpcode() == ISD::SRL &&
Nate Begeman1d4d4142005-09-01 00:19:25 +0000796 N0.getOperand(1).getOpcode() == ISD::Constant) {
797 uint64_t c1 = cast<ConstantSDNode>(N0.getOperand(1))->getValue();
Nate Begeman646d7e22005-09-02 21:18:40 +0000798 uint64_t c2 = N1C->getValue();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000799 if (c1 + c2 > OpSizeInBits)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000800 return DAG.getConstant(0, VT);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000801 return DAG.getNode(ISD::SRL, VT, N0.getOperand(0),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000802 DAG.getConstant(c1 + c2, N1.getValueType()));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000803 }
Nate Begeman83e75ec2005-09-06 04:43:02 +0000804 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000805}
806
Nate Begeman83e75ec2005-09-06 04:43:02 +0000807SDOperand DAGCombiner::visitCTLZ(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000808 SDOperand N0 = N->getOperand(0);
Nate Begeman646d7e22005-09-02 21:18:40 +0000809 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000810
811 // fold (ctlz c1) -> c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000812 if (N0C)
813 return DAG.getConstant(CountLeadingZeros_64(N0C->getValue()),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000814 N0.getValueType());
815 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000816}
817
Nate Begeman83e75ec2005-09-06 04:43:02 +0000818SDOperand DAGCombiner::visitCTTZ(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000819 SDOperand N0 = N->getOperand(0);
Nate Begeman646d7e22005-09-02 21:18:40 +0000820 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000821
822 // fold (cttz c1) -> c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000823 if (N0C)
824 return DAG.getConstant(CountTrailingZeros_64(N0C->getValue()),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000825 N0.getValueType());
826 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000827}
828
Nate Begeman83e75ec2005-09-06 04:43:02 +0000829SDOperand DAGCombiner::visitCTPOP(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000830 SDOperand N0 = N->getOperand(0);
Nate Begeman646d7e22005-09-02 21:18:40 +0000831 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000832
833 // fold (ctpop c1) -> c2
Nate Begeman646d7e22005-09-02 21:18:40 +0000834 if (N0C)
835 return DAG.getConstant(CountPopulation_64(N0C->getValue()),
Nate Begeman83e75ec2005-09-06 04:43:02 +0000836 N0.getValueType());
837 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000838}
839
Nate Begeman83e75ec2005-09-06 04:43:02 +0000840SDOperand DAGCombiner::visitSIGN_EXTEND(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000841 SDOperand N0 = N->getOperand(0);
Nate Begeman646d7e22005-09-02 21:18:40 +0000842 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000843 MVT::ValueType VT = N->getValueType(0);
844
Nate Begeman1d4d4142005-09-01 00:19:25 +0000845 // fold (sext c1) -> c1
Nate Begeman646d7e22005-09-02 21:18:40 +0000846 if (N0C)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000847 return DAG.getConstant(N0C->getSignExtended(), VT);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000848 // fold (sext (sext x)) -> (sext x)
849 if (N0.getOpcode() == ISD::SIGN_EXTEND)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000850 return DAG.getNode(ISD::SIGN_EXTEND, VT, N0.getOperand(0));
851 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000852}
853
Nate Begeman83e75ec2005-09-06 04:43:02 +0000854SDOperand DAGCombiner::visitZERO_EXTEND(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000855 SDOperand N0 = N->getOperand(0);
Nate Begeman646d7e22005-09-02 21:18:40 +0000856 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000857 MVT::ValueType VT = N->getValueType(0);
858
Nate Begeman1d4d4142005-09-01 00:19:25 +0000859 // fold (zext c1) -> c1
Nate Begeman646d7e22005-09-02 21:18:40 +0000860 if (N0C)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000861 return DAG.getConstant(N0C->getValue(), VT);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000862 // fold (zext (zext x)) -> (zext x)
863 if (N0.getOpcode() == ISD::ZERO_EXTEND)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000864 return DAG.getNode(ISD::ZERO_EXTEND, VT, N0.getOperand(0));
865 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000866}
867
Nate Begeman83e75ec2005-09-06 04:43:02 +0000868SDOperand DAGCombiner::visitSIGN_EXTEND_INREG(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000869 SDOperand N0 = N->getOperand(0);
Nate Begeman646d7e22005-09-02 21:18:40 +0000870 SDOperand N1 = N->getOperand(1);
871 SDOperand LHS, RHS, CC;
872 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000873 MVT::ValueType VT = N->getValueType(0);
Nate Begeman646d7e22005-09-02 21:18:40 +0000874 MVT::ValueType EVT = cast<VTSDNode>(N1)->getVT();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000875
Nate Begeman1d4d4142005-09-01 00:19:25 +0000876 // fold (sext_in_reg c1) -> c1
Nate Begeman646d7e22005-09-02 21:18:40 +0000877 if (N0C) {
878 SDOperand Truncate = DAG.getConstant(N0C->getValue(), EVT);
Nate Begeman83e75ec2005-09-06 04:43:02 +0000879 return DAG.getNode(ISD::SIGN_EXTEND, VT, Truncate);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000880 }
Nate Begeman646d7e22005-09-02 21:18:40 +0000881 // fold (sext_in_reg (sext_in_reg x, VT2), VT1) -> (sext_in_reg x, minVT) pt1
Nate Begeman1d4d4142005-09-01 00:19:25 +0000882 if (N0.getOpcode() == ISD::SIGN_EXTEND_INREG &&
Nate Begeman646d7e22005-09-02 21:18:40 +0000883 cast<VTSDNode>(N0.getOperand(1))->getVT() < EVT) {
Nate Begeman83e75ec2005-09-06 04:43:02 +0000884 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000885 }
Nate Begeman646d7e22005-09-02 21:18:40 +0000886 // fold (sext_in_reg (sext_in_reg x, VT2), VT1) -> (sext_in_reg x, minVT) pt2
887 if (N0.getOpcode() == ISD::SIGN_EXTEND_INREG &&
888 EVT < cast<VTSDNode>(N0.getOperand(1))->getVT()) {
Nate Begeman83e75ec2005-09-06 04:43:02 +0000889 return DAG.getNode(ISD::SIGN_EXTEND_INREG, VT, N0.getOperand(0), N1);
Nate Begeman646d7e22005-09-02 21:18:40 +0000890 }
Nate Begeman1d4d4142005-09-01 00:19:25 +0000891 // fold (sext_in_reg (assert_sext x)) -> (assert_sext x)
892 if (N0.getOpcode() == ISD::AssertSext &&
893 cast<VTSDNode>(N0.getOperand(1))->getVT() <= EVT) {
Nate Begeman83e75ec2005-09-06 04:43:02 +0000894 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000895 }
896 // fold (sext_in_reg (sextload x)) -> (sextload x)
897 if (N0.getOpcode() == ISD::SEXTLOAD &&
898 cast<VTSDNode>(N0.getOperand(3))->getVT() <= EVT) {
Nate Begeman83e75ec2005-09-06 04:43:02 +0000899 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000900 }
Nate Begeman4ebd8052005-09-01 23:24:04 +0000901 // fold (sext_in_reg (setcc x)) -> setcc x iff (setcc x) == 0 or -1
Nate Begeman646d7e22005-09-02 21:18:40 +0000902 // FIXME: teach isSetCCEquivalent about 0, -1 and then use it here
Nate Begeman1d4d4142005-09-01 00:19:25 +0000903 if (N0.getOpcode() == ISD::SETCC &&
904 TLI.getSetCCResultContents() ==
905 TargetLowering::ZeroOrNegativeOneSetCCResult)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000906 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000907 // FIXME: this code is currently just ported over from SelectionDAG.cpp
908 // we probably actually want to handle this in two pieces. Rather than
909 // checking all the top bits for zero, just check the sign bit here and turn
910 // it into a zero extend inreg (AND with constant).
911 // then, let the code for AND figure out if the mask is superfluous rather
912 // than doing so here.
913 if (N0.getOpcode() == ISD::AND &&
914 N0.getOperand(1).getOpcode() == ISD::Constant) {
915 uint64_t Mask = cast<ConstantSDNode>(N0.getOperand(1))->getValue();
916 unsigned NumBits = MVT::getSizeInBits(EVT);
917 if ((Mask & (~0ULL << (NumBits-1))) == 0)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000918 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000919 }
Nate Begeman83e75ec2005-09-06 04:43:02 +0000920 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000921}
922
Nate Begeman83e75ec2005-09-06 04:43:02 +0000923SDOperand DAGCombiner::visitTRUNCATE(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000924 SDOperand N0 = N->getOperand(0);
Nate Begeman646d7e22005-09-02 21:18:40 +0000925 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000926 MVT::ValueType VT = N->getValueType(0);
927
928 // noop truncate
929 if (N0.getValueType() == N->getValueType(0))
Nate Begeman83e75ec2005-09-06 04:43:02 +0000930 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +0000931 // fold (truncate c1) -> c1
Nate Begeman646d7e22005-09-02 21:18:40 +0000932 if (N0C)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000933 return DAG.getConstant(N0C->getValue(), VT);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000934 // fold (truncate (truncate x)) -> (truncate x)
935 if (N0.getOpcode() == ISD::TRUNCATE)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000936 return DAG.getNode(ISD::TRUNCATE, VT, N0.getOperand(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000937 // fold (truncate (ext x)) -> (ext x) or (truncate x) or x
938 if (N0.getOpcode() == ISD::ZERO_EXTEND || N0.getOpcode() == ISD::SIGN_EXTEND){
939 if (N0.getValueType() < VT)
940 // if the source is smaller than the dest, we still need an extend
Nate Begeman83e75ec2005-09-06 04:43:02 +0000941 return DAG.getNode(N0.getOpcode(), VT, N0.getOperand(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000942 else if (N0.getValueType() > VT)
943 // if the source is larger than the dest, than we just need the truncate
Nate Begeman83e75ec2005-09-06 04:43:02 +0000944 return DAG.getNode(ISD::TRUNCATE, VT, N0.getOperand(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000945 else
946 // if the source and dest are the same type, we can drop both the extend
947 // and the truncate
Nate Begeman83e75ec2005-09-06 04:43:02 +0000948 return N0.getOperand(0);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000949 }
Nate Begeman83e75ec2005-09-06 04:43:02 +0000950 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000951}
952
Nate Begeman83e75ec2005-09-06 04:43:02 +0000953SDOperand DAGCombiner::visitSINT_TO_FP(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000954 SDOperand N0 = N->getOperand(0);
Nate Begeman646d7e22005-09-02 21:18:40 +0000955 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000956
957 // fold (sint_to_fp c1) -> c1fp
Nate Begeman646d7e22005-09-02 21:18:40 +0000958 if (N0C)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000959 return DAG.getConstantFP(N0C->getSignExtended(), N->getValueType(0));
960 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000961}
962
Nate Begeman83e75ec2005-09-06 04:43:02 +0000963SDOperand DAGCombiner::visitUINT_TO_FP(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +0000964 SDOperand N0 = N->getOperand(0);
Nate Begeman646d7e22005-09-02 21:18:40 +0000965 ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
Nate Begeman1d4d4142005-09-01 00:19:25 +0000966
967 // fold (uint_to_fp c1) -> c1fp
Nate Begeman646d7e22005-09-02 21:18:40 +0000968 if (N0C)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000969 return DAG.getConstantFP(N0C->getValue(), N->getValueType(0));
970 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000971}
972
Nate Begeman83e75ec2005-09-06 04:43:02 +0000973SDOperand DAGCombiner::visitFP_TO_SINT(SDNode *N) {
Nate Begeman646d7e22005-09-02 21:18:40 +0000974 ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N->getOperand(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000975
976 // fold (fp_to_sint c1fp) -> c1
Nate Begeman646d7e22005-09-02 21:18:40 +0000977 if (N0CFP)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000978 return DAG.getConstant((int64_t)N0CFP->getValue(), N->getValueType(0));
979 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000980}
981
Nate Begeman83e75ec2005-09-06 04:43:02 +0000982SDOperand DAGCombiner::visitFP_TO_UINT(SDNode *N) {
Nate Begeman646d7e22005-09-02 21:18:40 +0000983 ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N->getOperand(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000984
985 // fold (fp_to_uint c1fp) -> c1
Nate Begeman646d7e22005-09-02 21:18:40 +0000986 if (N0CFP)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000987 return DAG.getConstant((uint64_t)N0CFP->getValue(), N->getValueType(0));
988 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000989}
990
Nate Begeman83e75ec2005-09-06 04:43:02 +0000991SDOperand DAGCombiner::visitFP_ROUND(SDNode *N) {
Nate Begeman646d7e22005-09-02 21:18:40 +0000992 ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N->getOperand(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +0000993
994 // fold (fp_round c1fp) -> c1fp
Nate Begeman646d7e22005-09-02 21:18:40 +0000995 if (N0CFP)
Nate Begeman83e75ec2005-09-06 04:43:02 +0000996 return DAG.getConstantFP(N0CFP->getValue(), N->getValueType(0));
997 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +0000998}
999
Nate Begeman83e75ec2005-09-06 04:43:02 +00001000SDOperand DAGCombiner::visitFP_ROUND_INREG(SDNode *N) {
Nate Begeman1d4d4142005-09-01 00:19:25 +00001001 SDOperand N0 = N->getOperand(0);
1002 MVT::ValueType VT = N->getValueType(0);
1003 MVT::ValueType EVT = cast<VTSDNode>(N->getOperand(1))->getVT();
Nate Begeman646d7e22005-09-02 21:18:40 +00001004 ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N0);
Nate Begeman1d4d4142005-09-01 00:19:25 +00001005
1006 // noop fp_round_inreg
1007 if (EVT == VT)
Nate Begeman83e75ec2005-09-06 04:43:02 +00001008 return N0;
Nate Begeman1d4d4142005-09-01 00:19:25 +00001009 // fold (fp_round_inreg c1fp) -> c1fp
Nate Begeman646d7e22005-09-02 21:18:40 +00001010 if (N0CFP) {
1011 SDOperand Round = DAG.getConstantFP(N0CFP->getValue(), EVT);
Nate Begeman83e75ec2005-09-06 04:43:02 +00001012 return DAG.getNode(ISD::FP_EXTEND, VT, Round);
Nate Begeman1d4d4142005-09-01 00:19:25 +00001013 }
Nate Begeman83e75ec2005-09-06 04:43:02 +00001014 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +00001015}
1016
Nate Begeman83e75ec2005-09-06 04:43:02 +00001017SDOperand DAGCombiner::visitFP_EXTEND(SDNode *N) {
Nate Begeman646d7e22005-09-02 21:18:40 +00001018 ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N->getOperand(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +00001019
1020 // fold (fp_extend c1fp) -> c1fp
Nate Begeman646d7e22005-09-02 21:18:40 +00001021 if (N0CFP)
Nate Begeman83e75ec2005-09-06 04:43:02 +00001022 return DAG.getConstantFP(N0CFP->getValue(), N->getValueType(0));
1023 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +00001024}
1025
Nate Begeman83e75ec2005-09-06 04:43:02 +00001026SDOperand DAGCombiner::visitFNEG(SDNode *N) {
Nate Begeman646d7e22005-09-02 21:18:40 +00001027 ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N->getOperand(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +00001028 // fold (neg c1) -> -c1
Nate Begeman646d7e22005-09-02 21:18:40 +00001029 if (N0CFP)
Nate Begeman83e75ec2005-09-06 04:43:02 +00001030 return DAG.getConstantFP(-N0CFP->getValue(), N->getValueType(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +00001031 // fold (neg (sub x, y)) -> (sub y, x)
1032 if (N->getOperand(0).getOpcode() == ISD::SUB)
1033 return DAG.getNode(ISD::SUB, N->getValueType(0), N->getOperand(1),
Nate Begeman83e75ec2005-09-06 04:43:02 +00001034 N->getOperand(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +00001035 // fold (neg (neg x)) -> x
1036 if (N->getOperand(0).getOpcode() == ISD::FNEG)
Nate Begeman83e75ec2005-09-06 04:43:02 +00001037 return N->getOperand(0).getOperand(0);
1038 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +00001039}
1040
Nate Begeman83e75ec2005-09-06 04:43:02 +00001041SDOperand DAGCombiner::visitFABS(SDNode *N) {
Nate Begeman646d7e22005-09-02 21:18:40 +00001042 ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N->getOperand(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +00001043 // fold (fabs c1) -> fabs(c1)
Nate Begeman646d7e22005-09-02 21:18:40 +00001044 if (N0CFP)
Nate Begeman83e75ec2005-09-06 04:43:02 +00001045 return DAG.getConstantFP(fabs(N0CFP->getValue()), N->getValueType(0));
Nate Begeman1d4d4142005-09-01 00:19:25 +00001046 // fold (fabs (fabs x)) -> (fabs x)
1047 if (N->getOperand(0).getOpcode() == ISD::FABS)
Nate Begeman83e75ec2005-09-06 04:43:02 +00001048 return N->getOperand(0);
Nate Begeman1d4d4142005-09-01 00:19:25 +00001049 // fold (fabs (fneg x)) -> (fabs x)
1050 if (N->getOperand(0).getOpcode() == ISD::FNEG)
1051 return DAG.getNode(ISD::FABS, N->getValueType(0),
Nate Begeman83e75ec2005-09-06 04:43:02 +00001052 N->getOperand(0).getOperand(0));
1053 return SDOperand();
Nate Begeman1d4d4142005-09-01 00:19:25 +00001054}
1055
Nate Begeman1d4d4142005-09-01 00:19:25 +00001056// SelectionDAG::Combine - This is the entry point for the file.
1057//
Nate Begeman4ebd8052005-09-01 23:24:04 +00001058void SelectionDAG::Combine(bool RunningAfterLegalize) {
Nate Begeman1d4d4142005-09-01 00:19:25 +00001059 /// run - This is the main entry point to this class.
1060 ///
Nate Begeman4ebd8052005-09-01 23:24:04 +00001061 DAGCombiner(*this).Run(RunningAfterLegalize);
Nate Begeman1d4d4142005-09-01 00:19:25 +00001062}