1*9880d681SAndroid Build Coastguard Worker //===-- DAGCombiner.cpp - Implement a DAG node combiner -------------------===//
2*9880d681SAndroid Build Coastguard Worker //
3*9880d681SAndroid Build Coastguard Worker // The LLVM Compiler Infrastructure
4*9880d681SAndroid Build Coastguard Worker //
5*9880d681SAndroid Build Coastguard Worker // This file is distributed under the University of Illinois Open Source
6*9880d681SAndroid Build Coastguard Worker // License. See LICENSE.TXT for details.
7*9880d681SAndroid Build Coastguard Worker //
8*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
9*9880d681SAndroid Build Coastguard Worker //
10*9880d681SAndroid Build Coastguard Worker // This pass combines dag nodes to form fewer, simpler DAG nodes. It can be run
11*9880d681SAndroid Build Coastguard Worker // both before and after the DAG is legalized.
12*9880d681SAndroid Build Coastguard Worker //
13*9880d681SAndroid Build Coastguard Worker // This pass is not a substitute for the LLVM IR instcombine pass. This pass is
14*9880d681SAndroid Build Coastguard Worker // primarily intended to handle simplification opportunities that are implicit
15*9880d681SAndroid Build Coastguard Worker // in the LLVM IR and exposed by the various codegen lowering phases.
16*9880d681SAndroid Build Coastguard Worker //
17*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
18*9880d681SAndroid Build Coastguard Worker
19*9880d681SAndroid Build Coastguard Worker #include "llvm/CodeGen/SelectionDAG.h"
20*9880d681SAndroid Build Coastguard Worker #include "llvm/ADT/SetVector.h"
21*9880d681SAndroid Build Coastguard Worker #include "llvm/ADT/SmallBitVector.h"
22*9880d681SAndroid Build Coastguard Worker #include "llvm/ADT/SmallPtrSet.h"
23*9880d681SAndroid Build Coastguard Worker #include "llvm/ADT/Statistic.h"
24*9880d681SAndroid Build Coastguard Worker #include "llvm/Analysis/AliasAnalysis.h"
25*9880d681SAndroid Build Coastguard Worker #include "llvm/CodeGen/MachineFrameInfo.h"
26*9880d681SAndroid Build Coastguard Worker #include "llvm/CodeGen/MachineFunction.h"
27*9880d681SAndroid Build Coastguard Worker #include "llvm/CodeGen/SelectionDAGTargetInfo.h"
28*9880d681SAndroid Build Coastguard Worker #include "llvm/IR/DataLayout.h"
29*9880d681SAndroid Build Coastguard Worker #include "llvm/IR/DerivedTypes.h"
30*9880d681SAndroid Build Coastguard Worker #include "llvm/IR/Function.h"
31*9880d681SAndroid Build Coastguard Worker #include "llvm/IR/LLVMContext.h"
32*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/CommandLine.h"
33*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/Debug.h"
34*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/ErrorHandling.h"
35*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/MathExtras.h"
36*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/raw_ostream.h"
37*9880d681SAndroid Build Coastguard Worker #include "llvm/Target/TargetLowering.h"
38*9880d681SAndroid Build Coastguard Worker #include "llvm/Target/TargetOptions.h"
39*9880d681SAndroid Build Coastguard Worker #include "llvm/Target/TargetRegisterInfo.h"
40*9880d681SAndroid Build Coastguard Worker #include "llvm/Target/TargetSubtargetInfo.h"
41*9880d681SAndroid Build Coastguard Worker #include <algorithm>
42*9880d681SAndroid Build Coastguard Worker using namespace llvm;
43*9880d681SAndroid Build Coastguard Worker
44*9880d681SAndroid Build Coastguard Worker #define DEBUG_TYPE "dagcombine"
45*9880d681SAndroid Build Coastguard Worker
46*9880d681SAndroid Build Coastguard Worker STATISTIC(NodesCombined , "Number of dag nodes combined");
47*9880d681SAndroid Build Coastguard Worker STATISTIC(PreIndexedNodes , "Number of pre-indexed nodes created");
48*9880d681SAndroid Build Coastguard Worker STATISTIC(PostIndexedNodes, "Number of post-indexed nodes created");
49*9880d681SAndroid Build Coastguard Worker STATISTIC(OpsNarrowed , "Number of load/op/store narrowed");
50*9880d681SAndroid Build Coastguard Worker STATISTIC(LdStFP2Int , "Number of fp load/store pairs transformed to int");
51*9880d681SAndroid Build Coastguard Worker STATISTIC(SlicedLoads, "Number of load sliced");
52*9880d681SAndroid Build Coastguard Worker
53*9880d681SAndroid Build Coastguard Worker namespace {
54*9880d681SAndroid Build Coastguard Worker static cl::opt<bool>
55*9880d681SAndroid Build Coastguard Worker CombinerAA("combiner-alias-analysis", cl::Hidden,
56*9880d681SAndroid Build Coastguard Worker cl::desc("Enable DAG combiner alias-analysis heuristics"));
57*9880d681SAndroid Build Coastguard Worker
58*9880d681SAndroid Build Coastguard Worker static cl::opt<bool>
59*9880d681SAndroid Build Coastguard Worker CombinerGlobalAA("combiner-global-alias-analysis", cl::Hidden,
60*9880d681SAndroid Build Coastguard Worker cl::desc("Enable DAG combiner's use of IR alias analysis"));
61*9880d681SAndroid Build Coastguard Worker
62*9880d681SAndroid Build Coastguard Worker static cl::opt<bool>
63*9880d681SAndroid Build Coastguard Worker UseTBAA("combiner-use-tbaa", cl::Hidden, cl::init(true),
64*9880d681SAndroid Build Coastguard Worker cl::desc("Enable DAG combiner's use of TBAA"));
65*9880d681SAndroid Build Coastguard Worker
66*9880d681SAndroid Build Coastguard Worker #ifndef NDEBUG
67*9880d681SAndroid Build Coastguard Worker static cl::opt<std::string>
68*9880d681SAndroid Build Coastguard Worker CombinerAAOnlyFunc("combiner-aa-only-func", cl::Hidden,
69*9880d681SAndroid Build Coastguard Worker cl::desc("Only use DAG-combiner alias analysis in this"
70*9880d681SAndroid Build Coastguard Worker " function"));
71*9880d681SAndroid Build Coastguard Worker #endif
72*9880d681SAndroid Build Coastguard Worker
73*9880d681SAndroid Build Coastguard Worker /// Hidden option to stress test load slicing, i.e., when this option
74*9880d681SAndroid Build Coastguard Worker /// is enabled, load slicing bypasses most of its profitability guards.
75*9880d681SAndroid Build Coastguard Worker static cl::opt<bool>
76*9880d681SAndroid Build Coastguard Worker StressLoadSlicing("combiner-stress-load-slicing", cl::Hidden,
77*9880d681SAndroid Build Coastguard Worker cl::desc("Bypass the profitability model of load "
78*9880d681SAndroid Build Coastguard Worker "slicing"),
79*9880d681SAndroid Build Coastguard Worker cl::init(false));
80*9880d681SAndroid Build Coastguard Worker
81*9880d681SAndroid Build Coastguard Worker static cl::opt<bool>
82*9880d681SAndroid Build Coastguard Worker MaySplitLoadIndex("combiner-split-load-index", cl::Hidden, cl::init(true),
83*9880d681SAndroid Build Coastguard Worker cl::desc("DAG combiner may split indexing from loads"));
84*9880d681SAndroid Build Coastguard Worker
85*9880d681SAndroid Build Coastguard Worker //------------------------------ DAGCombiner ---------------------------------//
86*9880d681SAndroid Build Coastguard Worker
87*9880d681SAndroid Build Coastguard Worker class DAGCombiner {
88*9880d681SAndroid Build Coastguard Worker SelectionDAG &DAG;
89*9880d681SAndroid Build Coastguard Worker const TargetLowering &TLI;
90*9880d681SAndroid Build Coastguard Worker CombineLevel Level;
91*9880d681SAndroid Build Coastguard Worker CodeGenOpt::Level OptLevel;
92*9880d681SAndroid Build Coastguard Worker bool LegalOperations;
93*9880d681SAndroid Build Coastguard Worker bool LegalTypes;
94*9880d681SAndroid Build Coastguard Worker bool ForCodeSize;
95*9880d681SAndroid Build Coastguard Worker
96*9880d681SAndroid Build Coastguard Worker /// \brief Worklist of all of the nodes that need to be simplified.
97*9880d681SAndroid Build Coastguard Worker ///
98*9880d681SAndroid Build Coastguard Worker /// This must behave as a stack -- new nodes to process are pushed onto the
99*9880d681SAndroid Build Coastguard Worker /// back and when processing we pop off of the back.
100*9880d681SAndroid Build Coastguard Worker ///
101*9880d681SAndroid Build Coastguard Worker /// The worklist will not contain duplicates but may contain null entries
102*9880d681SAndroid Build Coastguard Worker /// due to nodes being deleted from the underlying DAG.
103*9880d681SAndroid Build Coastguard Worker SmallVector<SDNode *, 64> Worklist;
104*9880d681SAndroid Build Coastguard Worker
105*9880d681SAndroid Build Coastguard Worker /// \brief Mapping from an SDNode to its position on the worklist.
106*9880d681SAndroid Build Coastguard Worker ///
107*9880d681SAndroid Build Coastguard Worker /// This is used to find and remove nodes from the worklist (by nulling
108*9880d681SAndroid Build Coastguard Worker /// them) when they are deleted from the underlying DAG. It relies on
109*9880d681SAndroid Build Coastguard Worker /// stable indices of nodes within the worklist.
110*9880d681SAndroid Build Coastguard Worker DenseMap<SDNode *, unsigned> WorklistMap;
111*9880d681SAndroid Build Coastguard Worker
112*9880d681SAndroid Build Coastguard Worker /// \brief Set of nodes which have been combined (at least once).
113*9880d681SAndroid Build Coastguard Worker ///
114*9880d681SAndroid Build Coastguard Worker /// This is used to allow us to reliably add any operands of a DAG node
115*9880d681SAndroid Build Coastguard Worker /// which have not yet been combined to the worklist.
116*9880d681SAndroid Build Coastguard Worker SmallPtrSet<SDNode *, 32> CombinedNodes;
117*9880d681SAndroid Build Coastguard Worker
118*9880d681SAndroid Build Coastguard Worker // AA - Used for DAG load/store alias analysis.
119*9880d681SAndroid Build Coastguard Worker AliasAnalysis &AA;
120*9880d681SAndroid Build Coastguard Worker
121*9880d681SAndroid Build Coastguard Worker /// When an instruction is simplified, add all users of the instruction to
122*9880d681SAndroid Build Coastguard Worker /// the work lists because they might get more simplified now.
AddUsersToWorklist(SDNode * N)123*9880d681SAndroid Build Coastguard Worker void AddUsersToWorklist(SDNode *N) {
124*9880d681SAndroid Build Coastguard Worker for (SDNode *Node : N->uses())
125*9880d681SAndroid Build Coastguard Worker AddToWorklist(Node);
126*9880d681SAndroid Build Coastguard Worker }
127*9880d681SAndroid Build Coastguard Worker
128*9880d681SAndroid Build Coastguard Worker /// Call the node-specific routine that folds each particular type of node.
129*9880d681SAndroid Build Coastguard Worker SDValue visit(SDNode *N);
130*9880d681SAndroid Build Coastguard Worker
131*9880d681SAndroid Build Coastguard Worker public:
132*9880d681SAndroid Build Coastguard Worker /// Add to the worklist making sure its instance is at the back (next to be
133*9880d681SAndroid Build Coastguard Worker /// processed.)
AddToWorklist(SDNode * N)134*9880d681SAndroid Build Coastguard Worker void AddToWorklist(SDNode *N) {
135*9880d681SAndroid Build Coastguard Worker // Skip handle nodes as they can't usefully be combined and confuse the
136*9880d681SAndroid Build Coastguard Worker // zero-use deletion strategy.
137*9880d681SAndroid Build Coastguard Worker if (N->getOpcode() == ISD::HANDLENODE)
138*9880d681SAndroid Build Coastguard Worker return;
139*9880d681SAndroid Build Coastguard Worker
140*9880d681SAndroid Build Coastguard Worker if (WorklistMap.insert(std::make_pair(N, Worklist.size())).second)
141*9880d681SAndroid Build Coastguard Worker Worklist.push_back(N);
142*9880d681SAndroid Build Coastguard Worker }
143*9880d681SAndroid Build Coastguard Worker
144*9880d681SAndroid Build Coastguard Worker /// Remove all instances of N from the worklist.
removeFromWorklist(SDNode * N)145*9880d681SAndroid Build Coastguard Worker void removeFromWorklist(SDNode *N) {
146*9880d681SAndroid Build Coastguard Worker CombinedNodes.erase(N);
147*9880d681SAndroid Build Coastguard Worker
148*9880d681SAndroid Build Coastguard Worker auto It = WorklistMap.find(N);
149*9880d681SAndroid Build Coastguard Worker if (It == WorklistMap.end())
150*9880d681SAndroid Build Coastguard Worker return; // Not in the worklist.
151*9880d681SAndroid Build Coastguard Worker
152*9880d681SAndroid Build Coastguard Worker // Null out the entry rather than erasing it to avoid a linear operation.
153*9880d681SAndroid Build Coastguard Worker Worklist[It->second] = nullptr;
154*9880d681SAndroid Build Coastguard Worker WorklistMap.erase(It);
155*9880d681SAndroid Build Coastguard Worker }
156*9880d681SAndroid Build Coastguard Worker
157*9880d681SAndroid Build Coastguard Worker void deleteAndRecombine(SDNode *N);
158*9880d681SAndroid Build Coastguard Worker bool recursivelyDeleteUnusedNodes(SDNode *N);
159*9880d681SAndroid Build Coastguard Worker
160*9880d681SAndroid Build Coastguard Worker /// Replaces all uses of the results of one DAG node with new values.
161*9880d681SAndroid Build Coastguard Worker SDValue CombineTo(SDNode *N, const SDValue *To, unsigned NumTo,
162*9880d681SAndroid Build Coastguard Worker bool AddTo = true);
163*9880d681SAndroid Build Coastguard Worker
164*9880d681SAndroid Build Coastguard Worker /// Replaces all uses of the results of one DAG node with new values.
CombineTo(SDNode * N,SDValue Res,bool AddTo=true)165*9880d681SAndroid Build Coastguard Worker SDValue CombineTo(SDNode *N, SDValue Res, bool AddTo = true) {
166*9880d681SAndroid Build Coastguard Worker return CombineTo(N, &Res, 1, AddTo);
167*9880d681SAndroid Build Coastguard Worker }
168*9880d681SAndroid Build Coastguard Worker
169*9880d681SAndroid Build Coastguard Worker /// Replaces all uses of the results of one DAG node with new values.
CombineTo(SDNode * N,SDValue Res0,SDValue Res1,bool AddTo=true)170*9880d681SAndroid Build Coastguard Worker SDValue CombineTo(SDNode *N, SDValue Res0, SDValue Res1,
171*9880d681SAndroid Build Coastguard Worker bool AddTo = true) {
172*9880d681SAndroid Build Coastguard Worker SDValue To[] = { Res0, Res1 };
173*9880d681SAndroid Build Coastguard Worker return CombineTo(N, To, 2, AddTo);
174*9880d681SAndroid Build Coastguard Worker }
175*9880d681SAndroid Build Coastguard Worker
176*9880d681SAndroid Build Coastguard Worker void CommitTargetLoweringOpt(const TargetLowering::TargetLoweringOpt &TLO);
177*9880d681SAndroid Build Coastguard Worker
178*9880d681SAndroid Build Coastguard Worker private:
179*9880d681SAndroid Build Coastguard Worker
180*9880d681SAndroid Build Coastguard Worker /// Check the specified integer node value to see if it can be simplified or
181*9880d681SAndroid Build Coastguard Worker /// if things it uses can be simplified by bit propagation.
182*9880d681SAndroid Build Coastguard Worker /// If so, return true.
SimplifyDemandedBits(SDValue Op)183*9880d681SAndroid Build Coastguard Worker bool SimplifyDemandedBits(SDValue Op) {
184*9880d681SAndroid Build Coastguard Worker unsigned BitWidth = Op.getValueType().getScalarType().getSizeInBits();
185*9880d681SAndroid Build Coastguard Worker APInt Demanded = APInt::getAllOnesValue(BitWidth);
186*9880d681SAndroid Build Coastguard Worker return SimplifyDemandedBits(Op, Demanded);
187*9880d681SAndroid Build Coastguard Worker }
188*9880d681SAndroid Build Coastguard Worker
189*9880d681SAndroid Build Coastguard Worker bool SimplifyDemandedBits(SDValue Op, const APInt &Demanded);
190*9880d681SAndroid Build Coastguard Worker
191*9880d681SAndroid Build Coastguard Worker bool CombineToPreIndexedLoadStore(SDNode *N);
192*9880d681SAndroid Build Coastguard Worker bool CombineToPostIndexedLoadStore(SDNode *N);
193*9880d681SAndroid Build Coastguard Worker SDValue SplitIndexingFromLoad(LoadSDNode *LD);
194*9880d681SAndroid Build Coastguard Worker bool SliceUpLoad(SDNode *N);
195*9880d681SAndroid Build Coastguard Worker
196*9880d681SAndroid Build Coastguard Worker /// \brief Replace an ISD::EXTRACT_VECTOR_ELT of a load with a narrowed
197*9880d681SAndroid Build Coastguard Worker /// load.
198*9880d681SAndroid Build Coastguard Worker ///
199*9880d681SAndroid Build Coastguard Worker /// \param EVE ISD::EXTRACT_VECTOR_ELT to be replaced.
200*9880d681SAndroid Build Coastguard Worker /// \param InVecVT type of the input vector to EVE with bitcasts resolved.
201*9880d681SAndroid Build Coastguard Worker /// \param EltNo index of the vector element to load.
202*9880d681SAndroid Build Coastguard Worker /// \param OriginalLoad load that EVE came from to be replaced.
203*9880d681SAndroid Build Coastguard Worker /// \returns EVE on success SDValue() on failure.
204*9880d681SAndroid Build Coastguard Worker SDValue ReplaceExtractVectorEltOfLoadWithNarrowedLoad(
205*9880d681SAndroid Build Coastguard Worker SDNode *EVE, EVT InVecVT, SDValue EltNo, LoadSDNode *OriginalLoad);
206*9880d681SAndroid Build Coastguard Worker void ReplaceLoadWithPromotedLoad(SDNode *Load, SDNode *ExtLoad);
207*9880d681SAndroid Build Coastguard Worker SDValue PromoteOperand(SDValue Op, EVT PVT, bool &Replace);
208*9880d681SAndroid Build Coastguard Worker SDValue SExtPromoteOperand(SDValue Op, EVT PVT);
209*9880d681SAndroid Build Coastguard Worker SDValue ZExtPromoteOperand(SDValue Op, EVT PVT);
210*9880d681SAndroid Build Coastguard Worker SDValue PromoteIntBinOp(SDValue Op);
211*9880d681SAndroid Build Coastguard Worker SDValue PromoteIntShiftOp(SDValue Op);
212*9880d681SAndroid Build Coastguard Worker SDValue PromoteExtend(SDValue Op);
213*9880d681SAndroid Build Coastguard Worker bool PromoteLoad(SDValue Op);
214*9880d681SAndroid Build Coastguard Worker
215*9880d681SAndroid Build Coastguard Worker void ExtendSetCCUses(const SmallVectorImpl<SDNode *> &SetCCs, SDValue Trunc,
216*9880d681SAndroid Build Coastguard Worker SDValue ExtLoad, const SDLoc &DL,
217*9880d681SAndroid Build Coastguard Worker ISD::NodeType ExtType);
218*9880d681SAndroid Build Coastguard Worker
219*9880d681SAndroid Build Coastguard Worker /// Call the node-specific routine that knows how to fold each
220*9880d681SAndroid Build Coastguard Worker /// particular type of node. If that doesn't do anything, try the
221*9880d681SAndroid Build Coastguard Worker /// target-specific DAG combines.
222*9880d681SAndroid Build Coastguard Worker SDValue combine(SDNode *N);
223*9880d681SAndroid Build Coastguard Worker
224*9880d681SAndroid Build Coastguard Worker // Visitation implementation - Implement dag node combining for different
225*9880d681SAndroid Build Coastguard Worker // node types. The semantics are as follows:
226*9880d681SAndroid Build Coastguard Worker // Return Value:
227*9880d681SAndroid Build Coastguard Worker // SDValue.getNode() == 0 - No change was made
228*9880d681SAndroid Build Coastguard Worker // SDValue.getNode() == N - N was replaced, is dead and has been handled.
229*9880d681SAndroid Build Coastguard Worker // otherwise - N should be replaced by the returned Operand.
230*9880d681SAndroid Build Coastguard Worker //
231*9880d681SAndroid Build Coastguard Worker SDValue visitTokenFactor(SDNode *N);
232*9880d681SAndroid Build Coastguard Worker SDValue visitMERGE_VALUES(SDNode *N);
233*9880d681SAndroid Build Coastguard Worker SDValue visitADD(SDNode *N);
234*9880d681SAndroid Build Coastguard Worker SDValue visitSUB(SDNode *N);
235*9880d681SAndroid Build Coastguard Worker SDValue visitADDC(SDNode *N);
236*9880d681SAndroid Build Coastguard Worker SDValue visitSUBC(SDNode *N);
237*9880d681SAndroid Build Coastguard Worker SDValue visitADDE(SDNode *N);
238*9880d681SAndroid Build Coastguard Worker SDValue visitSUBE(SDNode *N);
239*9880d681SAndroid Build Coastguard Worker SDValue visitMUL(SDNode *N);
240*9880d681SAndroid Build Coastguard Worker SDValue useDivRem(SDNode *N);
241*9880d681SAndroid Build Coastguard Worker SDValue visitSDIV(SDNode *N);
242*9880d681SAndroid Build Coastguard Worker SDValue visitUDIV(SDNode *N);
243*9880d681SAndroid Build Coastguard Worker SDValue visitREM(SDNode *N);
244*9880d681SAndroid Build Coastguard Worker SDValue visitMULHU(SDNode *N);
245*9880d681SAndroid Build Coastguard Worker SDValue visitMULHS(SDNode *N);
246*9880d681SAndroid Build Coastguard Worker SDValue visitSMUL_LOHI(SDNode *N);
247*9880d681SAndroid Build Coastguard Worker SDValue visitUMUL_LOHI(SDNode *N);
248*9880d681SAndroid Build Coastguard Worker SDValue visitSMULO(SDNode *N);
249*9880d681SAndroid Build Coastguard Worker SDValue visitUMULO(SDNode *N);
250*9880d681SAndroid Build Coastguard Worker SDValue visitIMINMAX(SDNode *N);
251*9880d681SAndroid Build Coastguard Worker SDValue visitAND(SDNode *N);
252*9880d681SAndroid Build Coastguard Worker SDValue visitANDLike(SDValue N0, SDValue N1, SDNode *LocReference);
253*9880d681SAndroid Build Coastguard Worker SDValue visitOR(SDNode *N);
254*9880d681SAndroid Build Coastguard Worker SDValue visitORLike(SDValue N0, SDValue N1, SDNode *LocReference);
255*9880d681SAndroid Build Coastguard Worker SDValue visitXOR(SDNode *N);
256*9880d681SAndroid Build Coastguard Worker SDValue SimplifyVBinOp(SDNode *N);
257*9880d681SAndroid Build Coastguard Worker SDValue visitSHL(SDNode *N);
258*9880d681SAndroid Build Coastguard Worker SDValue visitSRA(SDNode *N);
259*9880d681SAndroid Build Coastguard Worker SDValue visitSRL(SDNode *N);
260*9880d681SAndroid Build Coastguard Worker SDValue visitRotate(SDNode *N);
261*9880d681SAndroid Build Coastguard Worker SDValue visitBSWAP(SDNode *N);
262*9880d681SAndroid Build Coastguard Worker SDValue visitBITREVERSE(SDNode *N);
263*9880d681SAndroid Build Coastguard Worker SDValue visitCTLZ(SDNode *N);
264*9880d681SAndroid Build Coastguard Worker SDValue visitCTLZ_ZERO_UNDEF(SDNode *N);
265*9880d681SAndroid Build Coastguard Worker SDValue visitCTTZ(SDNode *N);
266*9880d681SAndroid Build Coastguard Worker SDValue visitCTTZ_ZERO_UNDEF(SDNode *N);
267*9880d681SAndroid Build Coastguard Worker SDValue visitCTPOP(SDNode *N);
268*9880d681SAndroid Build Coastguard Worker SDValue visitSELECT(SDNode *N);
269*9880d681SAndroid Build Coastguard Worker SDValue visitVSELECT(SDNode *N);
270*9880d681SAndroid Build Coastguard Worker SDValue visitSELECT_CC(SDNode *N);
271*9880d681SAndroid Build Coastguard Worker SDValue visitSETCC(SDNode *N);
272*9880d681SAndroid Build Coastguard Worker SDValue visitSETCCE(SDNode *N);
273*9880d681SAndroid Build Coastguard Worker SDValue visitSIGN_EXTEND(SDNode *N);
274*9880d681SAndroid Build Coastguard Worker SDValue visitZERO_EXTEND(SDNode *N);
275*9880d681SAndroid Build Coastguard Worker SDValue visitANY_EXTEND(SDNode *N);
276*9880d681SAndroid Build Coastguard Worker SDValue visitSIGN_EXTEND_INREG(SDNode *N);
277*9880d681SAndroid Build Coastguard Worker SDValue visitSIGN_EXTEND_VECTOR_INREG(SDNode *N);
278*9880d681SAndroid Build Coastguard Worker SDValue visitZERO_EXTEND_VECTOR_INREG(SDNode *N);
279*9880d681SAndroid Build Coastguard Worker SDValue visitTRUNCATE(SDNode *N);
280*9880d681SAndroid Build Coastguard Worker SDValue visitBITCAST(SDNode *N);
281*9880d681SAndroid Build Coastguard Worker SDValue visitBUILD_PAIR(SDNode *N);
282*9880d681SAndroid Build Coastguard Worker SDValue visitFADD(SDNode *N);
283*9880d681SAndroid Build Coastguard Worker SDValue visitFSUB(SDNode *N);
284*9880d681SAndroid Build Coastguard Worker SDValue visitFMUL(SDNode *N);
285*9880d681SAndroid Build Coastguard Worker SDValue visitFMA(SDNode *N);
286*9880d681SAndroid Build Coastguard Worker SDValue visitFDIV(SDNode *N);
287*9880d681SAndroid Build Coastguard Worker SDValue visitFREM(SDNode *N);
288*9880d681SAndroid Build Coastguard Worker SDValue visitFSQRT(SDNode *N);
289*9880d681SAndroid Build Coastguard Worker SDValue visitFCOPYSIGN(SDNode *N);
290*9880d681SAndroid Build Coastguard Worker SDValue visitSINT_TO_FP(SDNode *N);
291*9880d681SAndroid Build Coastguard Worker SDValue visitUINT_TO_FP(SDNode *N);
292*9880d681SAndroid Build Coastguard Worker SDValue visitFP_TO_SINT(SDNode *N);
293*9880d681SAndroid Build Coastguard Worker SDValue visitFP_TO_UINT(SDNode *N);
294*9880d681SAndroid Build Coastguard Worker SDValue visitFP_ROUND(SDNode *N);
295*9880d681SAndroid Build Coastguard Worker SDValue visitFP_ROUND_INREG(SDNode *N);
296*9880d681SAndroid Build Coastguard Worker SDValue visitFP_EXTEND(SDNode *N);
297*9880d681SAndroid Build Coastguard Worker SDValue visitFNEG(SDNode *N);
298*9880d681SAndroid Build Coastguard Worker SDValue visitFABS(SDNode *N);
299*9880d681SAndroid Build Coastguard Worker SDValue visitFCEIL(SDNode *N);
300*9880d681SAndroid Build Coastguard Worker SDValue visitFTRUNC(SDNode *N);
301*9880d681SAndroid Build Coastguard Worker SDValue visitFFLOOR(SDNode *N);
302*9880d681SAndroid Build Coastguard Worker SDValue visitFMINNUM(SDNode *N);
303*9880d681SAndroid Build Coastguard Worker SDValue visitFMAXNUM(SDNode *N);
304*9880d681SAndroid Build Coastguard Worker SDValue visitBRCOND(SDNode *N);
305*9880d681SAndroid Build Coastguard Worker SDValue visitBR_CC(SDNode *N);
306*9880d681SAndroid Build Coastguard Worker SDValue visitLOAD(SDNode *N);
307*9880d681SAndroid Build Coastguard Worker
308*9880d681SAndroid Build Coastguard Worker SDValue replaceStoreChain(StoreSDNode *ST, SDValue BetterChain);
309*9880d681SAndroid Build Coastguard Worker SDValue replaceStoreOfFPConstant(StoreSDNode *ST);
310*9880d681SAndroid Build Coastguard Worker
311*9880d681SAndroid Build Coastguard Worker SDValue visitSTORE(SDNode *N);
312*9880d681SAndroid Build Coastguard Worker SDValue visitINSERT_VECTOR_ELT(SDNode *N);
313*9880d681SAndroid Build Coastguard Worker SDValue visitEXTRACT_VECTOR_ELT(SDNode *N);
314*9880d681SAndroid Build Coastguard Worker SDValue visitBUILD_VECTOR(SDNode *N);
315*9880d681SAndroid Build Coastguard Worker SDValue visitCONCAT_VECTORS(SDNode *N);
316*9880d681SAndroid Build Coastguard Worker SDValue visitEXTRACT_SUBVECTOR(SDNode *N);
317*9880d681SAndroid Build Coastguard Worker SDValue visitVECTOR_SHUFFLE(SDNode *N);
318*9880d681SAndroid Build Coastguard Worker SDValue visitSCALAR_TO_VECTOR(SDNode *N);
319*9880d681SAndroid Build Coastguard Worker SDValue visitINSERT_SUBVECTOR(SDNode *N);
320*9880d681SAndroid Build Coastguard Worker SDValue visitMLOAD(SDNode *N);
321*9880d681SAndroid Build Coastguard Worker SDValue visitMSTORE(SDNode *N);
322*9880d681SAndroid Build Coastguard Worker SDValue visitMGATHER(SDNode *N);
323*9880d681SAndroid Build Coastguard Worker SDValue visitMSCATTER(SDNode *N);
324*9880d681SAndroid Build Coastguard Worker SDValue visitFP_TO_FP16(SDNode *N);
325*9880d681SAndroid Build Coastguard Worker SDValue visitFP16_TO_FP(SDNode *N);
326*9880d681SAndroid Build Coastguard Worker
327*9880d681SAndroid Build Coastguard Worker SDValue visitFADDForFMACombine(SDNode *N);
328*9880d681SAndroid Build Coastguard Worker SDValue visitFSUBForFMACombine(SDNode *N);
329*9880d681SAndroid Build Coastguard Worker SDValue visitFMULForFMACombine(SDNode *N);
330*9880d681SAndroid Build Coastguard Worker
331*9880d681SAndroid Build Coastguard Worker SDValue XformToShuffleWithZero(SDNode *N);
332*9880d681SAndroid Build Coastguard Worker SDValue ReassociateOps(unsigned Opc, const SDLoc &DL, SDValue LHS,
333*9880d681SAndroid Build Coastguard Worker SDValue RHS);
334*9880d681SAndroid Build Coastguard Worker
335*9880d681SAndroid Build Coastguard Worker SDValue visitShiftByConstant(SDNode *N, ConstantSDNode *Amt);
336*9880d681SAndroid Build Coastguard Worker
337*9880d681SAndroid Build Coastguard Worker bool SimplifySelectOps(SDNode *SELECT, SDValue LHS, SDValue RHS);
338*9880d681SAndroid Build Coastguard Worker SDValue SimplifyBinOpWithSameOpcodeHands(SDNode *N);
339*9880d681SAndroid Build Coastguard Worker SDValue SimplifySelect(const SDLoc &DL, SDValue N0, SDValue N1, SDValue N2);
340*9880d681SAndroid Build Coastguard Worker SDValue SimplifySelectCC(const SDLoc &DL, SDValue N0, SDValue N1,
341*9880d681SAndroid Build Coastguard Worker SDValue N2, SDValue N3, ISD::CondCode CC,
342*9880d681SAndroid Build Coastguard Worker bool NotExtCompare = false);
343*9880d681SAndroid Build Coastguard Worker SDValue SimplifySetCC(EVT VT, SDValue N0, SDValue N1, ISD::CondCode Cond,
344*9880d681SAndroid Build Coastguard Worker const SDLoc &DL, bool foldBooleans = true);
345*9880d681SAndroid Build Coastguard Worker
346*9880d681SAndroid Build Coastguard Worker bool isSetCCEquivalent(SDValue N, SDValue &LHS, SDValue &RHS,
347*9880d681SAndroid Build Coastguard Worker SDValue &CC) const;
348*9880d681SAndroid Build Coastguard Worker bool isOneUseSetCC(SDValue N) const;
349*9880d681SAndroid Build Coastguard Worker
350*9880d681SAndroid Build Coastguard Worker SDValue SimplifyNodeWithTwoResults(SDNode *N, unsigned LoOp,
351*9880d681SAndroid Build Coastguard Worker unsigned HiOp);
352*9880d681SAndroid Build Coastguard Worker SDValue CombineConsecutiveLoads(SDNode *N, EVT VT);
353*9880d681SAndroid Build Coastguard Worker SDValue CombineExtLoad(SDNode *N);
354*9880d681SAndroid Build Coastguard Worker SDValue combineRepeatedFPDivisors(SDNode *N);
355*9880d681SAndroid Build Coastguard Worker SDValue ConstantFoldBITCASTofBUILD_VECTOR(SDNode *, EVT);
356*9880d681SAndroid Build Coastguard Worker SDValue BuildSDIV(SDNode *N);
357*9880d681SAndroid Build Coastguard Worker SDValue BuildSDIVPow2(SDNode *N);
358*9880d681SAndroid Build Coastguard Worker SDValue BuildUDIV(SDNode *N);
359*9880d681SAndroid Build Coastguard Worker SDValue BuildReciprocalEstimate(SDValue Op, SDNodeFlags *Flags);
360*9880d681SAndroid Build Coastguard Worker SDValue buildRsqrtEstimate(SDValue Op, SDNodeFlags *Flags);
361*9880d681SAndroid Build Coastguard Worker SDValue buildSqrtEstimate(SDValue Op, SDNodeFlags *Flags);
362*9880d681SAndroid Build Coastguard Worker SDValue buildSqrtEstimateImpl(SDValue Op, SDNodeFlags *Flags, bool Recip);
363*9880d681SAndroid Build Coastguard Worker SDValue buildSqrtNROneConst(SDValue Op, SDValue Est, unsigned Iterations,
364*9880d681SAndroid Build Coastguard Worker SDNodeFlags *Flags, bool Reciprocal);
365*9880d681SAndroid Build Coastguard Worker SDValue buildSqrtNRTwoConst(SDValue Op, SDValue Est, unsigned Iterations,
366*9880d681SAndroid Build Coastguard Worker SDNodeFlags *Flags, bool Reciprocal);
367*9880d681SAndroid Build Coastguard Worker SDValue MatchBSwapHWordLow(SDNode *N, SDValue N0, SDValue N1,
368*9880d681SAndroid Build Coastguard Worker bool DemandHighBits = true);
369*9880d681SAndroid Build Coastguard Worker SDValue MatchBSwapHWord(SDNode *N, SDValue N0, SDValue N1);
370*9880d681SAndroid Build Coastguard Worker SDNode *MatchRotatePosNeg(SDValue Shifted, SDValue Pos, SDValue Neg,
371*9880d681SAndroid Build Coastguard Worker SDValue InnerPos, SDValue InnerNeg,
372*9880d681SAndroid Build Coastguard Worker unsigned PosOpcode, unsigned NegOpcode,
373*9880d681SAndroid Build Coastguard Worker const SDLoc &DL);
374*9880d681SAndroid Build Coastguard Worker SDNode *MatchRotate(SDValue LHS, SDValue RHS, const SDLoc &DL);
375*9880d681SAndroid Build Coastguard Worker SDValue ReduceLoadWidth(SDNode *N);
376*9880d681SAndroid Build Coastguard Worker SDValue ReduceLoadOpStoreWidth(SDNode *N);
377*9880d681SAndroid Build Coastguard Worker SDValue TransformFPLoadStorePair(SDNode *N);
378*9880d681SAndroid Build Coastguard Worker SDValue reduceBuildVecExtToExtBuildVec(SDNode *N);
379*9880d681SAndroid Build Coastguard Worker SDValue reduceBuildVecConvertToConvertBuildVec(SDNode *N);
380*9880d681SAndroid Build Coastguard Worker
381*9880d681SAndroid Build Coastguard Worker SDValue GetDemandedBits(SDValue V, const APInt &Mask);
382*9880d681SAndroid Build Coastguard Worker
383*9880d681SAndroid Build Coastguard Worker /// Walk up chain skipping non-aliasing memory nodes,
384*9880d681SAndroid Build Coastguard Worker /// looking for aliasing nodes and adding them to the Aliases vector.
385*9880d681SAndroid Build Coastguard Worker void GatherAllAliases(SDNode *N, SDValue OriginalChain,
386*9880d681SAndroid Build Coastguard Worker SmallVectorImpl<SDValue> &Aliases);
387*9880d681SAndroid Build Coastguard Worker
388*9880d681SAndroid Build Coastguard Worker /// Return true if there is any possibility that the two addresses overlap.
389*9880d681SAndroid Build Coastguard Worker bool isAlias(LSBaseSDNode *Op0, LSBaseSDNode *Op1) const;
390*9880d681SAndroid Build Coastguard Worker
391*9880d681SAndroid Build Coastguard Worker /// Walk up chain skipping non-aliasing memory nodes, looking for a better
392*9880d681SAndroid Build Coastguard Worker /// chain (aliasing node.)
393*9880d681SAndroid Build Coastguard Worker SDValue FindBetterChain(SDNode *N, SDValue Chain);
394*9880d681SAndroid Build Coastguard Worker
395*9880d681SAndroid Build Coastguard Worker /// Try to replace a store and any possibly adjacent stores on
396*9880d681SAndroid Build Coastguard Worker /// consecutive chains with better chains. Return true only if St is
397*9880d681SAndroid Build Coastguard Worker /// replaced.
398*9880d681SAndroid Build Coastguard Worker ///
399*9880d681SAndroid Build Coastguard Worker /// Notice that other chains may still be replaced even if the function
400*9880d681SAndroid Build Coastguard Worker /// returns false.
401*9880d681SAndroid Build Coastguard Worker bool findBetterNeighborChains(StoreSDNode *St);
402*9880d681SAndroid Build Coastguard Worker
403*9880d681SAndroid Build Coastguard Worker /// Match "(X shl/srl V1) & V2" where V2 may not be present.
404*9880d681SAndroid Build Coastguard Worker bool MatchRotateHalf(SDValue Op, SDValue &Shift, SDValue &Mask);
405*9880d681SAndroid Build Coastguard Worker
406*9880d681SAndroid Build Coastguard Worker /// Holds a pointer to an LSBaseSDNode as well as information on where it
407*9880d681SAndroid Build Coastguard Worker /// is located in a sequence of memory operations connected by a chain.
408*9880d681SAndroid Build Coastguard Worker struct MemOpLink {
MemOpLink__anon093339a80111::DAGCombiner::MemOpLink409*9880d681SAndroid Build Coastguard Worker MemOpLink (LSBaseSDNode *N, int64_t Offset, unsigned Seq):
410*9880d681SAndroid Build Coastguard Worker MemNode(N), OffsetFromBase(Offset), SequenceNum(Seq) { }
411*9880d681SAndroid Build Coastguard Worker // Ptr to the mem node.
412*9880d681SAndroid Build Coastguard Worker LSBaseSDNode *MemNode;
413*9880d681SAndroid Build Coastguard Worker // Offset from the base ptr.
414*9880d681SAndroid Build Coastguard Worker int64_t OffsetFromBase;
415*9880d681SAndroid Build Coastguard Worker // What is the sequence number of this mem node.
416*9880d681SAndroid Build Coastguard Worker // Lowest mem operand in the DAG starts at zero.
417*9880d681SAndroid Build Coastguard Worker unsigned SequenceNum;
418*9880d681SAndroid Build Coastguard Worker };
419*9880d681SAndroid Build Coastguard Worker
420*9880d681SAndroid Build Coastguard Worker /// This is a helper function for visitMUL to check the profitability
421*9880d681SAndroid Build Coastguard Worker /// of folding (mul (add x, c1), c2) -> (add (mul x, c2), c1*c2).
422*9880d681SAndroid Build Coastguard Worker /// MulNode is the original multiply, AddNode is (add x, c1),
423*9880d681SAndroid Build Coastguard Worker /// and ConstNode is c2.
424*9880d681SAndroid Build Coastguard Worker bool isMulAddWithConstProfitable(SDNode *MulNode,
425*9880d681SAndroid Build Coastguard Worker SDValue &AddNode,
426*9880d681SAndroid Build Coastguard Worker SDValue &ConstNode);
427*9880d681SAndroid Build Coastguard Worker
428*9880d681SAndroid Build Coastguard Worker /// This is a helper function for MergeStoresOfConstantsOrVecElts. Returns a
429*9880d681SAndroid Build Coastguard Worker /// constant build_vector of the stored constant values in Stores.
430*9880d681SAndroid Build Coastguard Worker SDValue getMergedConstantVectorStore(SelectionDAG &DAG, const SDLoc &SL,
431*9880d681SAndroid Build Coastguard Worker ArrayRef<MemOpLink> Stores,
432*9880d681SAndroid Build Coastguard Worker SmallVectorImpl<SDValue> &Chains,
433*9880d681SAndroid Build Coastguard Worker EVT Ty) const;
434*9880d681SAndroid Build Coastguard Worker
435*9880d681SAndroid Build Coastguard Worker /// This is a helper function for visitAND and visitZERO_EXTEND. Returns
436*9880d681SAndroid Build Coastguard Worker /// true if the (and (load x) c) pattern matches an extload. ExtVT returns
437*9880d681SAndroid Build Coastguard Worker /// the type of the loaded value to be extended. LoadedVT returns the type
438*9880d681SAndroid Build Coastguard Worker /// of the original loaded value. NarrowLoad returns whether the load would
439*9880d681SAndroid Build Coastguard Worker /// need to be narrowed in order to match.
440*9880d681SAndroid Build Coastguard Worker bool isAndLoadExtLoad(ConstantSDNode *AndC, LoadSDNode *LoadN,
441*9880d681SAndroid Build Coastguard Worker EVT LoadResultTy, EVT &ExtVT, EVT &LoadedVT,
442*9880d681SAndroid Build Coastguard Worker bool &NarrowLoad);
443*9880d681SAndroid Build Coastguard Worker
444*9880d681SAndroid Build Coastguard Worker /// This is a helper function for MergeConsecutiveStores. When the source
445*9880d681SAndroid Build Coastguard Worker /// elements of the consecutive stores are all constants or all extracted
446*9880d681SAndroid Build Coastguard Worker /// vector elements, try to merge them into one larger store.
447*9880d681SAndroid Build Coastguard Worker /// \return True if a merged store was created.
448*9880d681SAndroid Build Coastguard Worker bool MergeStoresOfConstantsOrVecElts(SmallVectorImpl<MemOpLink> &StoreNodes,
449*9880d681SAndroid Build Coastguard Worker EVT MemVT, unsigned NumStores,
450*9880d681SAndroid Build Coastguard Worker bool IsConstantSrc, bool UseVector);
451*9880d681SAndroid Build Coastguard Worker
452*9880d681SAndroid Build Coastguard Worker /// This is a helper function for MergeConsecutiveStores.
453*9880d681SAndroid Build Coastguard Worker /// Stores that may be merged are placed in StoreNodes.
454*9880d681SAndroid Build Coastguard Worker /// Loads that may alias with those stores are placed in AliasLoadNodes.
455*9880d681SAndroid Build Coastguard Worker void getStoreMergeAndAliasCandidates(
456*9880d681SAndroid Build Coastguard Worker StoreSDNode* St, SmallVectorImpl<MemOpLink> &StoreNodes,
457*9880d681SAndroid Build Coastguard Worker SmallVectorImpl<LSBaseSDNode*> &AliasLoadNodes);
458*9880d681SAndroid Build Coastguard Worker
459*9880d681SAndroid Build Coastguard Worker /// Helper function for MergeConsecutiveStores. Checks if
460*9880d681SAndroid Build Coastguard Worker /// Candidate stores have indirect dependency through their
461*9880d681SAndroid Build Coastguard Worker /// operands. \return True if safe to merge
462*9880d681SAndroid Build Coastguard Worker bool checkMergeStoreCandidatesForDependencies(
463*9880d681SAndroid Build Coastguard Worker SmallVectorImpl<MemOpLink> &StoreNodes);
464*9880d681SAndroid Build Coastguard Worker
465*9880d681SAndroid Build Coastguard Worker /// Merge consecutive store operations into a wide store.
466*9880d681SAndroid Build Coastguard Worker /// This optimization uses wide integers or vectors when possible.
467*9880d681SAndroid Build Coastguard Worker /// \return True if some memory operations were changed.
468*9880d681SAndroid Build Coastguard Worker bool MergeConsecutiveStores(StoreSDNode *N);
469*9880d681SAndroid Build Coastguard Worker
470*9880d681SAndroid Build Coastguard Worker /// \brief Try to transform a truncation where C is a constant:
471*9880d681SAndroid Build Coastguard Worker /// (trunc (and X, C)) -> (and (trunc X), (trunc C))
472*9880d681SAndroid Build Coastguard Worker ///
473*9880d681SAndroid Build Coastguard Worker /// \p N needs to be a truncation and its first operand an AND. Other
474*9880d681SAndroid Build Coastguard Worker /// requirements are checked by the function (e.g. that trunc is
475*9880d681SAndroid Build Coastguard Worker /// single-use) and if missed an empty SDValue is returned.
476*9880d681SAndroid Build Coastguard Worker SDValue distributeTruncateThroughAnd(SDNode *N);
477*9880d681SAndroid Build Coastguard Worker
478*9880d681SAndroid Build Coastguard Worker public:
DAGCombiner(SelectionDAG & D,AliasAnalysis & A,CodeGenOpt::Level OL)479*9880d681SAndroid Build Coastguard Worker DAGCombiner(SelectionDAG &D, AliasAnalysis &A, CodeGenOpt::Level OL)
480*9880d681SAndroid Build Coastguard Worker : DAG(D), TLI(D.getTargetLoweringInfo()), Level(BeforeLegalizeTypes),
481*9880d681SAndroid Build Coastguard Worker OptLevel(OL), LegalOperations(false), LegalTypes(false), AA(A) {
482*9880d681SAndroid Build Coastguard Worker ForCodeSize = DAG.getMachineFunction().getFunction()->optForSize();
483*9880d681SAndroid Build Coastguard Worker }
484*9880d681SAndroid Build Coastguard Worker
485*9880d681SAndroid Build Coastguard Worker /// Runs the dag combiner on all nodes in the work list
486*9880d681SAndroid Build Coastguard Worker void Run(CombineLevel AtLevel);
487*9880d681SAndroid Build Coastguard Worker
getDAG() const488*9880d681SAndroid Build Coastguard Worker SelectionDAG &getDAG() const { return DAG; }
489*9880d681SAndroid Build Coastguard Worker
490*9880d681SAndroid Build Coastguard Worker /// Returns a type large enough to hold any valid shift amount - before type
491*9880d681SAndroid Build Coastguard Worker /// legalization these can be huge.
getShiftAmountTy(EVT LHSTy)492*9880d681SAndroid Build Coastguard Worker EVT getShiftAmountTy(EVT LHSTy) {
493*9880d681SAndroid Build Coastguard Worker assert(LHSTy.isInteger() && "Shift amount is not an integer type!");
494*9880d681SAndroid Build Coastguard Worker if (LHSTy.isVector())
495*9880d681SAndroid Build Coastguard Worker return LHSTy;
496*9880d681SAndroid Build Coastguard Worker auto &DL = DAG.getDataLayout();
497*9880d681SAndroid Build Coastguard Worker return LegalTypes ? TLI.getScalarShiftAmountTy(DL, LHSTy)
498*9880d681SAndroid Build Coastguard Worker : TLI.getPointerTy(DL);
499*9880d681SAndroid Build Coastguard Worker }
500*9880d681SAndroid Build Coastguard Worker
501*9880d681SAndroid Build Coastguard Worker /// This method returns true if we are running before type legalization or
502*9880d681SAndroid Build Coastguard Worker /// if the specified VT is legal.
isTypeLegal(const EVT & VT)503*9880d681SAndroid Build Coastguard Worker bool isTypeLegal(const EVT &VT) {
504*9880d681SAndroid Build Coastguard Worker if (!LegalTypes) return true;
505*9880d681SAndroid Build Coastguard Worker return TLI.isTypeLegal(VT);
506*9880d681SAndroid Build Coastguard Worker }
507*9880d681SAndroid Build Coastguard Worker
508*9880d681SAndroid Build Coastguard Worker /// Convenience wrapper around TargetLowering::getSetCCResultType
getSetCCResultType(EVT VT) const509*9880d681SAndroid Build Coastguard Worker EVT getSetCCResultType(EVT VT) const {
510*9880d681SAndroid Build Coastguard Worker return TLI.getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), VT);
511*9880d681SAndroid Build Coastguard Worker }
512*9880d681SAndroid Build Coastguard Worker };
513*9880d681SAndroid Build Coastguard Worker }
514*9880d681SAndroid Build Coastguard Worker
515*9880d681SAndroid Build Coastguard Worker
516*9880d681SAndroid Build Coastguard Worker namespace {
517*9880d681SAndroid Build Coastguard Worker /// This class is a DAGUpdateListener that removes any deleted
518*9880d681SAndroid Build Coastguard Worker /// nodes from the worklist.
519*9880d681SAndroid Build Coastguard Worker class WorklistRemover : public SelectionDAG::DAGUpdateListener {
520*9880d681SAndroid Build Coastguard Worker DAGCombiner &DC;
521*9880d681SAndroid Build Coastguard Worker public:
WorklistRemover(DAGCombiner & dc)522*9880d681SAndroid Build Coastguard Worker explicit WorklistRemover(DAGCombiner &dc)
523*9880d681SAndroid Build Coastguard Worker : SelectionDAG::DAGUpdateListener(dc.getDAG()), DC(dc) {}
524*9880d681SAndroid Build Coastguard Worker
NodeDeleted(SDNode * N,SDNode * E)525*9880d681SAndroid Build Coastguard Worker void NodeDeleted(SDNode *N, SDNode *E) override {
526*9880d681SAndroid Build Coastguard Worker DC.removeFromWorklist(N);
527*9880d681SAndroid Build Coastguard Worker }
528*9880d681SAndroid Build Coastguard Worker };
529*9880d681SAndroid Build Coastguard Worker }
530*9880d681SAndroid Build Coastguard Worker
531*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
532*9880d681SAndroid Build Coastguard Worker // TargetLowering::DAGCombinerInfo implementation
533*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
534*9880d681SAndroid Build Coastguard Worker
AddToWorklist(SDNode * N)535*9880d681SAndroid Build Coastguard Worker void TargetLowering::DAGCombinerInfo::AddToWorklist(SDNode *N) {
536*9880d681SAndroid Build Coastguard Worker ((DAGCombiner*)DC)->AddToWorklist(N);
537*9880d681SAndroid Build Coastguard Worker }
538*9880d681SAndroid Build Coastguard Worker
RemoveFromWorklist(SDNode * N)539*9880d681SAndroid Build Coastguard Worker void TargetLowering::DAGCombinerInfo::RemoveFromWorklist(SDNode *N) {
540*9880d681SAndroid Build Coastguard Worker ((DAGCombiner*)DC)->removeFromWorklist(N);
541*9880d681SAndroid Build Coastguard Worker }
542*9880d681SAndroid Build Coastguard Worker
543*9880d681SAndroid Build Coastguard Worker SDValue TargetLowering::DAGCombinerInfo::
CombineTo(SDNode * N,ArrayRef<SDValue> To,bool AddTo)544*9880d681SAndroid Build Coastguard Worker CombineTo(SDNode *N, ArrayRef<SDValue> To, bool AddTo) {
545*9880d681SAndroid Build Coastguard Worker return ((DAGCombiner*)DC)->CombineTo(N, &To[0], To.size(), AddTo);
546*9880d681SAndroid Build Coastguard Worker }
547*9880d681SAndroid Build Coastguard Worker
548*9880d681SAndroid Build Coastguard Worker SDValue TargetLowering::DAGCombinerInfo::
CombineTo(SDNode * N,SDValue Res,bool AddTo)549*9880d681SAndroid Build Coastguard Worker CombineTo(SDNode *N, SDValue Res, bool AddTo) {
550*9880d681SAndroid Build Coastguard Worker return ((DAGCombiner*)DC)->CombineTo(N, Res, AddTo);
551*9880d681SAndroid Build Coastguard Worker }
552*9880d681SAndroid Build Coastguard Worker
553*9880d681SAndroid Build Coastguard Worker
554*9880d681SAndroid Build Coastguard Worker SDValue TargetLowering::DAGCombinerInfo::
CombineTo(SDNode * N,SDValue Res0,SDValue Res1,bool AddTo)555*9880d681SAndroid Build Coastguard Worker CombineTo(SDNode *N, SDValue Res0, SDValue Res1, bool AddTo) {
556*9880d681SAndroid Build Coastguard Worker return ((DAGCombiner*)DC)->CombineTo(N, Res0, Res1, AddTo);
557*9880d681SAndroid Build Coastguard Worker }
558*9880d681SAndroid Build Coastguard Worker
559*9880d681SAndroid Build Coastguard Worker void TargetLowering::DAGCombinerInfo::
CommitTargetLoweringOpt(const TargetLowering::TargetLoweringOpt & TLO)560*9880d681SAndroid Build Coastguard Worker CommitTargetLoweringOpt(const TargetLowering::TargetLoweringOpt &TLO) {
561*9880d681SAndroid Build Coastguard Worker return ((DAGCombiner*)DC)->CommitTargetLoweringOpt(TLO);
562*9880d681SAndroid Build Coastguard Worker }
563*9880d681SAndroid Build Coastguard Worker
564*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
565*9880d681SAndroid Build Coastguard Worker // Helper Functions
566*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
567*9880d681SAndroid Build Coastguard Worker
deleteAndRecombine(SDNode * N)568*9880d681SAndroid Build Coastguard Worker void DAGCombiner::deleteAndRecombine(SDNode *N) {
569*9880d681SAndroid Build Coastguard Worker removeFromWorklist(N);
570*9880d681SAndroid Build Coastguard Worker
571*9880d681SAndroid Build Coastguard Worker // If the operands of this node are only used by the node, they will now be
572*9880d681SAndroid Build Coastguard Worker // dead. Make sure to re-visit them and recursively delete dead nodes.
573*9880d681SAndroid Build Coastguard Worker for (const SDValue &Op : N->ops())
574*9880d681SAndroid Build Coastguard Worker // For an operand generating multiple values, one of the values may
575*9880d681SAndroid Build Coastguard Worker // become dead allowing further simplification (e.g. split index
576*9880d681SAndroid Build Coastguard Worker // arithmetic from an indexed load).
577*9880d681SAndroid Build Coastguard Worker if (Op->hasOneUse() || Op->getNumValues() > 1)
578*9880d681SAndroid Build Coastguard Worker AddToWorklist(Op.getNode());
579*9880d681SAndroid Build Coastguard Worker
580*9880d681SAndroid Build Coastguard Worker DAG.DeleteNode(N);
581*9880d681SAndroid Build Coastguard Worker }
582*9880d681SAndroid Build Coastguard Worker
583*9880d681SAndroid Build Coastguard Worker /// Return 1 if we can compute the negated form of the specified expression for
584*9880d681SAndroid Build Coastguard Worker /// the same cost as the expression itself, or 2 if we can compute the negated
585*9880d681SAndroid Build Coastguard Worker /// form more cheaply than the expression itself.
isNegatibleForFree(SDValue Op,bool LegalOperations,const TargetLowering & TLI,const TargetOptions * Options,unsigned Depth=0)586*9880d681SAndroid Build Coastguard Worker static char isNegatibleForFree(SDValue Op, bool LegalOperations,
587*9880d681SAndroid Build Coastguard Worker const TargetLowering &TLI,
588*9880d681SAndroid Build Coastguard Worker const TargetOptions *Options,
589*9880d681SAndroid Build Coastguard Worker unsigned Depth = 0) {
590*9880d681SAndroid Build Coastguard Worker // fneg is removable even if it has multiple uses.
591*9880d681SAndroid Build Coastguard Worker if (Op.getOpcode() == ISD::FNEG) return 2;
592*9880d681SAndroid Build Coastguard Worker
593*9880d681SAndroid Build Coastguard Worker // Don't allow anything with multiple uses.
594*9880d681SAndroid Build Coastguard Worker if (!Op.hasOneUse()) return 0;
595*9880d681SAndroid Build Coastguard Worker
596*9880d681SAndroid Build Coastguard Worker // Don't recurse exponentially.
597*9880d681SAndroid Build Coastguard Worker if (Depth > 6) return 0;
598*9880d681SAndroid Build Coastguard Worker
599*9880d681SAndroid Build Coastguard Worker switch (Op.getOpcode()) {
600*9880d681SAndroid Build Coastguard Worker default: return false;
601*9880d681SAndroid Build Coastguard Worker case ISD::ConstantFP:
602*9880d681SAndroid Build Coastguard Worker // Don't invert constant FP values after legalize. The negated constant
603*9880d681SAndroid Build Coastguard Worker // isn't necessarily legal.
604*9880d681SAndroid Build Coastguard Worker return LegalOperations ? 0 : 1;
605*9880d681SAndroid Build Coastguard Worker case ISD::FADD:
606*9880d681SAndroid Build Coastguard Worker // FIXME: determine better conditions for this xform.
607*9880d681SAndroid Build Coastguard Worker if (!Options->UnsafeFPMath) return 0;
608*9880d681SAndroid Build Coastguard Worker
609*9880d681SAndroid Build Coastguard Worker // After operation legalization, it might not be legal to create new FSUBs.
610*9880d681SAndroid Build Coastguard Worker if (LegalOperations &&
611*9880d681SAndroid Build Coastguard Worker !TLI.isOperationLegalOrCustom(ISD::FSUB, Op.getValueType()))
612*9880d681SAndroid Build Coastguard Worker return 0;
613*9880d681SAndroid Build Coastguard Worker
614*9880d681SAndroid Build Coastguard Worker // fold (fneg (fadd A, B)) -> (fsub (fneg A), B)
615*9880d681SAndroid Build Coastguard Worker if (char V = isNegatibleForFree(Op.getOperand(0), LegalOperations, TLI,
616*9880d681SAndroid Build Coastguard Worker Options, Depth + 1))
617*9880d681SAndroid Build Coastguard Worker return V;
618*9880d681SAndroid Build Coastguard Worker // fold (fneg (fadd A, B)) -> (fsub (fneg B), A)
619*9880d681SAndroid Build Coastguard Worker return isNegatibleForFree(Op.getOperand(1), LegalOperations, TLI, Options,
620*9880d681SAndroid Build Coastguard Worker Depth + 1);
621*9880d681SAndroid Build Coastguard Worker case ISD::FSUB:
622*9880d681SAndroid Build Coastguard Worker // We can't turn -(A-B) into B-A when we honor signed zeros.
623*9880d681SAndroid Build Coastguard Worker if (!Options->UnsafeFPMath) return 0;
624*9880d681SAndroid Build Coastguard Worker
625*9880d681SAndroid Build Coastguard Worker // fold (fneg (fsub A, B)) -> (fsub B, A)
626*9880d681SAndroid Build Coastguard Worker return 1;
627*9880d681SAndroid Build Coastguard Worker
628*9880d681SAndroid Build Coastguard Worker case ISD::FMUL:
629*9880d681SAndroid Build Coastguard Worker case ISD::FDIV:
630*9880d681SAndroid Build Coastguard Worker if (Options->HonorSignDependentRoundingFPMath()) return 0;
631*9880d681SAndroid Build Coastguard Worker
632*9880d681SAndroid Build Coastguard Worker // fold (fneg (fmul X, Y)) -> (fmul (fneg X), Y) or (fmul X, (fneg Y))
633*9880d681SAndroid Build Coastguard Worker if (char V = isNegatibleForFree(Op.getOperand(0), LegalOperations, TLI,
634*9880d681SAndroid Build Coastguard Worker Options, Depth + 1))
635*9880d681SAndroid Build Coastguard Worker return V;
636*9880d681SAndroid Build Coastguard Worker
637*9880d681SAndroid Build Coastguard Worker return isNegatibleForFree(Op.getOperand(1), LegalOperations, TLI, Options,
638*9880d681SAndroid Build Coastguard Worker Depth + 1);
639*9880d681SAndroid Build Coastguard Worker
640*9880d681SAndroid Build Coastguard Worker case ISD::FP_EXTEND:
641*9880d681SAndroid Build Coastguard Worker case ISD::FP_ROUND:
642*9880d681SAndroid Build Coastguard Worker case ISD::FSIN:
643*9880d681SAndroid Build Coastguard Worker return isNegatibleForFree(Op.getOperand(0), LegalOperations, TLI, Options,
644*9880d681SAndroid Build Coastguard Worker Depth + 1);
645*9880d681SAndroid Build Coastguard Worker }
646*9880d681SAndroid Build Coastguard Worker }
647*9880d681SAndroid Build Coastguard Worker
648*9880d681SAndroid Build Coastguard Worker /// If isNegatibleForFree returns true, return the newly negated expression.
GetNegatedExpression(SDValue Op,SelectionDAG & DAG,bool LegalOperations,unsigned Depth=0)649*9880d681SAndroid Build Coastguard Worker static SDValue GetNegatedExpression(SDValue Op, SelectionDAG &DAG,
650*9880d681SAndroid Build Coastguard Worker bool LegalOperations, unsigned Depth = 0) {
651*9880d681SAndroid Build Coastguard Worker const TargetOptions &Options = DAG.getTarget().Options;
652*9880d681SAndroid Build Coastguard Worker // fneg is removable even if it has multiple uses.
653*9880d681SAndroid Build Coastguard Worker if (Op.getOpcode() == ISD::FNEG) return Op.getOperand(0);
654*9880d681SAndroid Build Coastguard Worker
655*9880d681SAndroid Build Coastguard Worker // Don't allow anything with multiple uses.
656*9880d681SAndroid Build Coastguard Worker assert(Op.hasOneUse() && "Unknown reuse!");
657*9880d681SAndroid Build Coastguard Worker
658*9880d681SAndroid Build Coastguard Worker assert(Depth <= 6 && "GetNegatedExpression doesn't match isNegatibleForFree");
659*9880d681SAndroid Build Coastguard Worker
660*9880d681SAndroid Build Coastguard Worker const SDNodeFlags *Flags = Op.getNode()->getFlags();
661*9880d681SAndroid Build Coastguard Worker
662*9880d681SAndroid Build Coastguard Worker switch (Op.getOpcode()) {
663*9880d681SAndroid Build Coastguard Worker default: llvm_unreachable("Unknown code");
664*9880d681SAndroid Build Coastguard Worker case ISD::ConstantFP: {
665*9880d681SAndroid Build Coastguard Worker APFloat V = cast<ConstantFPSDNode>(Op)->getValueAPF();
666*9880d681SAndroid Build Coastguard Worker V.changeSign();
667*9880d681SAndroid Build Coastguard Worker return DAG.getConstantFP(V, SDLoc(Op), Op.getValueType());
668*9880d681SAndroid Build Coastguard Worker }
669*9880d681SAndroid Build Coastguard Worker case ISD::FADD:
670*9880d681SAndroid Build Coastguard Worker // FIXME: determine better conditions for this xform.
671*9880d681SAndroid Build Coastguard Worker assert(Options.UnsafeFPMath);
672*9880d681SAndroid Build Coastguard Worker
673*9880d681SAndroid Build Coastguard Worker // fold (fneg (fadd A, B)) -> (fsub (fneg A), B)
674*9880d681SAndroid Build Coastguard Worker if (isNegatibleForFree(Op.getOperand(0), LegalOperations,
675*9880d681SAndroid Build Coastguard Worker DAG.getTargetLoweringInfo(), &Options, Depth+1))
676*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FSUB, SDLoc(Op), Op.getValueType(),
677*9880d681SAndroid Build Coastguard Worker GetNegatedExpression(Op.getOperand(0), DAG,
678*9880d681SAndroid Build Coastguard Worker LegalOperations, Depth+1),
679*9880d681SAndroid Build Coastguard Worker Op.getOperand(1), Flags);
680*9880d681SAndroid Build Coastguard Worker // fold (fneg (fadd A, B)) -> (fsub (fneg B), A)
681*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FSUB, SDLoc(Op), Op.getValueType(),
682*9880d681SAndroid Build Coastguard Worker GetNegatedExpression(Op.getOperand(1), DAG,
683*9880d681SAndroid Build Coastguard Worker LegalOperations, Depth+1),
684*9880d681SAndroid Build Coastguard Worker Op.getOperand(0), Flags);
685*9880d681SAndroid Build Coastguard Worker case ISD::FSUB:
686*9880d681SAndroid Build Coastguard Worker // We can't turn -(A-B) into B-A when we honor signed zeros.
687*9880d681SAndroid Build Coastguard Worker assert(Options.UnsafeFPMath);
688*9880d681SAndroid Build Coastguard Worker
689*9880d681SAndroid Build Coastguard Worker // fold (fneg (fsub 0, B)) -> B
690*9880d681SAndroid Build Coastguard Worker if (ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(Op.getOperand(0)))
691*9880d681SAndroid Build Coastguard Worker if (N0CFP->isZero())
692*9880d681SAndroid Build Coastguard Worker return Op.getOperand(1);
693*9880d681SAndroid Build Coastguard Worker
694*9880d681SAndroid Build Coastguard Worker // fold (fneg (fsub A, B)) -> (fsub B, A)
695*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FSUB, SDLoc(Op), Op.getValueType(),
696*9880d681SAndroid Build Coastguard Worker Op.getOperand(1), Op.getOperand(0), Flags);
697*9880d681SAndroid Build Coastguard Worker
698*9880d681SAndroid Build Coastguard Worker case ISD::FMUL:
699*9880d681SAndroid Build Coastguard Worker case ISD::FDIV:
700*9880d681SAndroid Build Coastguard Worker assert(!Options.HonorSignDependentRoundingFPMath());
701*9880d681SAndroid Build Coastguard Worker
702*9880d681SAndroid Build Coastguard Worker // fold (fneg (fmul X, Y)) -> (fmul (fneg X), Y)
703*9880d681SAndroid Build Coastguard Worker if (isNegatibleForFree(Op.getOperand(0), LegalOperations,
704*9880d681SAndroid Build Coastguard Worker DAG.getTargetLoweringInfo(), &Options, Depth+1))
705*9880d681SAndroid Build Coastguard Worker return DAG.getNode(Op.getOpcode(), SDLoc(Op), Op.getValueType(),
706*9880d681SAndroid Build Coastguard Worker GetNegatedExpression(Op.getOperand(0), DAG,
707*9880d681SAndroid Build Coastguard Worker LegalOperations, Depth+1),
708*9880d681SAndroid Build Coastguard Worker Op.getOperand(1), Flags);
709*9880d681SAndroid Build Coastguard Worker
710*9880d681SAndroid Build Coastguard Worker // fold (fneg (fmul X, Y)) -> (fmul X, (fneg Y))
711*9880d681SAndroid Build Coastguard Worker return DAG.getNode(Op.getOpcode(), SDLoc(Op), Op.getValueType(),
712*9880d681SAndroid Build Coastguard Worker Op.getOperand(0),
713*9880d681SAndroid Build Coastguard Worker GetNegatedExpression(Op.getOperand(1), DAG,
714*9880d681SAndroid Build Coastguard Worker LegalOperations, Depth+1), Flags);
715*9880d681SAndroid Build Coastguard Worker
716*9880d681SAndroid Build Coastguard Worker case ISD::FP_EXTEND:
717*9880d681SAndroid Build Coastguard Worker case ISD::FSIN:
718*9880d681SAndroid Build Coastguard Worker return DAG.getNode(Op.getOpcode(), SDLoc(Op), Op.getValueType(),
719*9880d681SAndroid Build Coastguard Worker GetNegatedExpression(Op.getOperand(0), DAG,
720*9880d681SAndroid Build Coastguard Worker LegalOperations, Depth+1));
721*9880d681SAndroid Build Coastguard Worker case ISD::FP_ROUND:
722*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FP_ROUND, SDLoc(Op), Op.getValueType(),
723*9880d681SAndroid Build Coastguard Worker GetNegatedExpression(Op.getOperand(0), DAG,
724*9880d681SAndroid Build Coastguard Worker LegalOperations, Depth+1),
725*9880d681SAndroid Build Coastguard Worker Op.getOperand(1));
726*9880d681SAndroid Build Coastguard Worker }
727*9880d681SAndroid Build Coastguard Worker }
728*9880d681SAndroid Build Coastguard Worker
729*9880d681SAndroid Build Coastguard Worker // Return true if this node is a setcc, or is a select_cc
730*9880d681SAndroid Build Coastguard Worker // that selects between the target values used for true and false, making it
731*9880d681SAndroid Build Coastguard Worker // equivalent to a setcc. Also, set the incoming LHS, RHS, and CC references to
732*9880d681SAndroid Build Coastguard Worker // the appropriate nodes based on the type of node we are checking. This
733*9880d681SAndroid Build Coastguard Worker // simplifies life a bit for the callers.
isSetCCEquivalent(SDValue N,SDValue & LHS,SDValue & RHS,SDValue & CC) const734*9880d681SAndroid Build Coastguard Worker bool DAGCombiner::isSetCCEquivalent(SDValue N, SDValue &LHS, SDValue &RHS,
735*9880d681SAndroid Build Coastguard Worker SDValue &CC) const {
736*9880d681SAndroid Build Coastguard Worker if (N.getOpcode() == ISD::SETCC) {
737*9880d681SAndroid Build Coastguard Worker LHS = N.getOperand(0);
738*9880d681SAndroid Build Coastguard Worker RHS = N.getOperand(1);
739*9880d681SAndroid Build Coastguard Worker CC = N.getOperand(2);
740*9880d681SAndroid Build Coastguard Worker return true;
741*9880d681SAndroid Build Coastguard Worker }
742*9880d681SAndroid Build Coastguard Worker
743*9880d681SAndroid Build Coastguard Worker if (N.getOpcode() != ISD::SELECT_CC ||
744*9880d681SAndroid Build Coastguard Worker !TLI.isConstTrueVal(N.getOperand(2).getNode()) ||
745*9880d681SAndroid Build Coastguard Worker !TLI.isConstFalseVal(N.getOperand(3).getNode()))
746*9880d681SAndroid Build Coastguard Worker return false;
747*9880d681SAndroid Build Coastguard Worker
748*9880d681SAndroid Build Coastguard Worker if (TLI.getBooleanContents(N.getValueType()) ==
749*9880d681SAndroid Build Coastguard Worker TargetLowering::UndefinedBooleanContent)
750*9880d681SAndroid Build Coastguard Worker return false;
751*9880d681SAndroid Build Coastguard Worker
752*9880d681SAndroid Build Coastguard Worker LHS = N.getOperand(0);
753*9880d681SAndroid Build Coastguard Worker RHS = N.getOperand(1);
754*9880d681SAndroid Build Coastguard Worker CC = N.getOperand(4);
755*9880d681SAndroid Build Coastguard Worker return true;
756*9880d681SAndroid Build Coastguard Worker }
757*9880d681SAndroid Build Coastguard Worker
758*9880d681SAndroid Build Coastguard Worker /// Return true if this is a SetCC-equivalent operation with only one use.
759*9880d681SAndroid Build Coastguard Worker /// If this is true, it allows the users to invert the operation for free when
760*9880d681SAndroid Build Coastguard Worker /// it is profitable to do so.
isOneUseSetCC(SDValue N) const761*9880d681SAndroid Build Coastguard Worker bool DAGCombiner::isOneUseSetCC(SDValue N) const {
762*9880d681SAndroid Build Coastguard Worker SDValue N0, N1, N2;
763*9880d681SAndroid Build Coastguard Worker if (isSetCCEquivalent(N, N0, N1, N2) && N.getNode()->hasOneUse())
764*9880d681SAndroid Build Coastguard Worker return true;
765*9880d681SAndroid Build Coastguard Worker return false;
766*9880d681SAndroid Build Coastguard Worker }
767*9880d681SAndroid Build Coastguard Worker
768*9880d681SAndroid Build Coastguard Worker // \brief Returns the SDNode if it is a constant float BuildVector
769*9880d681SAndroid Build Coastguard Worker // or constant float.
isConstantFPBuildVectorOrConstantFP(SDValue N)770*9880d681SAndroid Build Coastguard Worker static SDNode *isConstantFPBuildVectorOrConstantFP(SDValue N) {
771*9880d681SAndroid Build Coastguard Worker if (isa<ConstantFPSDNode>(N))
772*9880d681SAndroid Build Coastguard Worker return N.getNode();
773*9880d681SAndroid Build Coastguard Worker if (ISD::isBuildVectorOfConstantFPSDNodes(N.getNode()))
774*9880d681SAndroid Build Coastguard Worker return N.getNode();
775*9880d681SAndroid Build Coastguard Worker return nullptr;
776*9880d681SAndroid Build Coastguard Worker }
777*9880d681SAndroid Build Coastguard Worker
778*9880d681SAndroid Build Coastguard Worker // \brief Returns the SDNode if it is a constant splat BuildVector or constant
779*9880d681SAndroid Build Coastguard Worker // int.
isConstOrConstSplat(SDValue N)780*9880d681SAndroid Build Coastguard Worker static ConstantSDNode *isConstOrConstSplat(SDValue N) {
781*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(N))
782*9880d681SAndroid Build Coastguard Worker return CN;
783*9880d681SAndroid Build Coastguard Worker
784*9880d681SAndroid Build Coastguard Worker if (BuildVectorSDNode *BV = dyn_cast<BuildVectorSDNode>(N)) {
785*9880d681SAndroid Build Coastguard Worker BitVector UndefElements;
786*9880d681SAndroid Build Coastguard Worker ConstantSDNode *CN = BV->getConstantSplatNode(&UndefElements);
787*9880d681SAndroid Build Coastguard Worker
788*9880d681SAndroid Build Coastguard Worker // BuildVectors can truncate their operands. Ignore that case here.
789*9880d681SAndroid Build Coastguard Worker // FIXME: We blindly ignore splats which include undef which is overly
790*9880d681SAndroid Build Coastguard Worker // pessimistic.
791*9880d681SAndroid Build Coastguard Worker if (CN && UndefElements.none() &&
792*9880d681SAndroid Build Coastguard Worker CN->getValueType(0) == N.getValueType().getScalarType())
793*9880d681SAndroid Build Coastguard Worker return CN;
794*9880d681SAndroid Build Coastguard Worker }
795*9880d681SAndroid Build Coastguard Worker
796*9880d681SAndroid Build Coastguard Worker return nullptr;
797*9880d681SAndroid Build Coastguard Worker }
798*9880d681SAndroid Build Coastguard Worker
799*9880d681SAndroid Build Coastguard Worker // \brief Returns the SDNode if it is a constant splat BuildVector or constant
800*9880d681SAndroid Build Coastguard Worker // float.
isConstOrConstSplatFP(SDValue N)801*9880d681SAndroid Build Coastguard Worker static ConstantFPSDNode *isConstOrConstSplatFP(SDValue N) {
802*9880d681SAndroid Build Coastguard Worker if (ConstantFPSDNode *CN = dyn_cast<ConstantFPSDNode>(N))
803*9880d681SAndroid Build Coastguard Worker return CN;
804*9880d681SAndroid Build Coastguard Worker
805*9880d681SAndroid Build Coastguard Worker if (BuildVectorSDNode *BV = dyn_cast<BuildVectorSDNode>(N)) {
806*9880d681SAndroid Build Coastguard Worker BitVector UndefElements;
807*9880d681SAndroid Build Coastguard Worker ConstantFPSDNode *CN = BV->getConstantFPSplatNode(&UndefElements);
808*9880d681SAndroid Build Coastguard Worker
809*9880d681SAndroid Build Coastguard Worker if (CN && UndefElements.none())
810*9880d681SAndroid Build Coastguard Worker return CN;
811*9880d681SAndroid Build Coastguard Worker }
812*9880d681SAndroid Build Coastguard Worker
813*9880d681SAndroid Build Coastguard Worker return nullptr;
814*9880d681SAndroid Build Coastguard Worker }
815*9880d681SAndroid Build Coastguard Worker
ReassociateOps(unsigned Opc,const SDLoc & DL,SDValue N0,SDValue N1)816*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::ReassociateOps(unsigned Opc, const SDLoc &DL, SDValue N0,
817*9880d681SAndroid Build Coastguard Worker SDValue N1) {
818*9880d681SAndroid Build Coastguard Worker EVT VT = N0.getValueType();
819*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == Opc) {
820*9880d681SAndroid Build Coastguard Worker if (SDNode *L = DAG.isConstantIntBuildVectorOrConstantInt(N0.getOperand(1))) {
821*9880d681SAndroid Build Coastguard Worker if (SDNode *R = DAG.isConstantIntBuildVectorOrConstantInt(N1)) {
822*9880d681SAndroid Build Coastguard Worker // reassoc. (op (op x, c1), c2) -> (op x, (op c1, c2))
823*9880d681SAndroid Build Coastguard Worker if (SDValue OpNode = DAG.FoldConstantArithmetic(Opc, DL, VT, L, R))
824*9880d681SAndroid Build Coastguard Worker return DAG.getNode(Opc, DL, VT, N0.getOperand(0), OpNode);
825*9880d681SAndroid Build Coastguard Worker return SDValue();
826*9880d681SAndroid Build Coastguard Worker }
827*9880d681SAndroid Build Coastguard Worker if (N0.hasOneUse()) {
828*9880d681SAndroid Build Coastguard Worker // reassoc. (op (op x, c1), y) -> (op (op x, y), c1) iff x+c1 has one
829*9880d681SAndroid Build Coastguard Worker // use
830*9880d681SAndroid Build Coastguard Worker SDValue OpNode = DAG.getNode(Opc, SDLoc(N0), VT, N0.getOperand(0), N1);
831*9880d681SAndroid Build Coastguard Worker if (!OpNode.getNode())
832*9880d681SAndroid Build Coastguard Worker return SDValue();
833*9880d681SAndroid Build Coastguard Worker AddToWorklist(OpNode.getNode());
834*9880d681SAndroid Build Coastguard Worker return DAG.getNode(Opc, DL, VT, OpNode, N0.getOperand(1));
835*9880d681SAndroid Build Coastguard Worker }
836*9880d681SAndroid Build Coastguard Worker }
837*9880d681SAndroid Build Coastguard Worker }
838*9880d681SAndroid Build Coastguard Worker
839*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == Opc) {
840*9880d681SAndroid Build Coastguard Worker if (SDNode *R = DAG.isConstantIntBuildVectorOrConstantInt(N1.getOperand(1))) {
841*9880d681SAndroid Build Coastguard Worker if (SDNode *L = DAG.isConstantIntBuildVectorOrConstantInt(N0)) {
842*9880d681SAndroid Build Coastguard Worker // reassoc. (op c2, (op x, c1)) -> (op x, (op c1, c2))
843*9880d681SAndroid Build Coastguard Worker if (SDValue OpNode = DAG.FoldConstantArithmetic(Opc, DL, VT, R, L))
844*9880d681SAndroid Build Coastguard Worker return DAG.getNode(Opc, DL, VT, N1.getOperand(0), OpNode);
845*9880d681SAndroid Build Coastguard Worker return SDValue();
846*9880d681SAndroid Build Coastguard Worker }
847*9880d681SAndroid Build Coastguard Worker if (N1.hasOneUse()) {
848*9880d681SAndroid Build Coastguard Worker // reassoc. (op x, (op y, c1)) -> (op (op x, y), c1) iff x+c1 has one
849*9880d681SAndroid Build Coastguard Worker // use
850*9880d681SAndroid Build Coastguard Worker SDValue OpNode = DAG.getNode(Opc, SDLoc(N0), VT, N0, N1.getOperand(0));
851*9880d681SAndroid Build Coastguard Worker if (!OpNode.getNode())
852*9880d681SAndroid Build Coastguard Worker return SDValue();
853*9880d681SAndroid Build Coastguard Worker AddToWorklist(OpNode.getNode());
854*9880d681SAndroid Build Coastguard Worker return DAG.getNode(Opc, DL, VT, OpNode, N1.getOperand(1));
855*9880d681SAndroid Build Coastguard Worker }
856*9880d681SAndroid Build Coastguard Worker }
857*9880d681SAndroid Build Coastguard Worker }
858*9880d681SAndroid Build Coastguard Worker
859*9880d681SAndroid Build Coastguard Worker return SDValue();
860*9880d681SAndroid Build Coastguard Worker }
861*9880d681SAndroid Build Coastguard Worker
CombineTo(SDNode * N,const SDValue * To,unsigned NumTo,bool AddTo)862*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::CombineTo(SDNode *N, const SDValue *To, unsigned NumTo,
863*9880d681SAndroid Build Coastguard Worker bool AddTo) {
864*9880d681SAndroid Build Coastguard Worker assert(N->getNumValues() == NumTo && "Broken CombineTo call!");
865*9880d681SAndroid Build Coastguard Worker ++NodesCombined;
866*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "\nReplacing.1 ";
867*9880d681SAndroid Build Coastguard Worker N->dump(&DAG);
868*9880d681SAndroid Build Coastguard Worker dbgs() << "\nWith: ";
869*9880d681SAndroid Build Coastguard Worker To[0].getNode()->dump(&DAG);
870*9880d681SAndroid Build Coastguard Worker dbgs() << " and " << NumTo-1 << " other values\n");
871*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0, e = NumTo; i != e; ++i)
872*9880d681SAndroid Build Coastguard Worker assert((!To[i].getNode() ||
873*9880d681SAndroid Build Coastguard Worker N->getValueType(i) == To[i].getValueType()) &&
874*9880d681SAndroid Build Coastguard Worker "Cannot combine value to value of different type!");
875*9880d681SAndroid Build Coastguard Worker
876*9880d681SAndroid Build Coastguard Worker WorklistRemover DeadNodes(*this);
877*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesWith(N, To);
878*9880d681SAndroid Build Coastguard Worker if (AddTo) {
879*9880d681SAndroid Build Coastguard Worker // Push the new nodes and any users onto the worklist
880*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0, e = NumTo; i != e; ++i) {
881*9880d681SAndroid Build Coastguard Worker if (To[i].getNode()) {
882*9880d681SAndroid Build Coastguard Worker AddToWorklist(To[i].getNode());
883*9880d681SAndroid Build Coastguard Worker AddUsersToWorklist(To[i].getNode());
884*9880d681SAndroid Build Coastguard Worker }
885*9880d681SAndroid Build Coastguard Worker }
886*9880d681SAndroid Build Coastguard Worker }
887*9880d681SAndroid Build Coastguard Worker
888*9880d681SAndroid Build Coastguard Worker // Finally, if the node is now dead, remove it from the graph. The node
889*9880d681SAndroid Build Coastguard Worker // may not be dead if the replacement process recursively simplified to
890*9880d681SAndroid Build Coastguard Worker // something else needing this node.
891*9880d681SAndroid Build Coastguard Worker if (N->use_empty())
892*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(N);
893*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0);
894*9880d681SAndroid Build Coastguard Worker }
895*9880d681SAndroid Build Coastguard Worker
896*9880d681SAndroid Build Coastguard Worker void DAGCombiner::
CommitTargetLoweringOpt(const TargetLowering::TargetLoweringOpt & TLO)897*9880d681SAndroid Build Coastguard Worker CommitTargetLoweringOpt(const TargetLowering::TargetLoweringOpt &TLO) {
898*9880d681SAndroid Build Coastguard Worker // Replace all uses. If any nodes become isomorphic to other nodes and
899*9880d681SAndroid Build Coastguard Worker // are deleted, make sure to remove them from our worklist.
900*9880d681SAndroid Build Coastguard Worker WorklistRemover DeadNodes(*this);
901*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(TLO.Old, TLO.New);
902*9880d681SAndroid Build Coastguard Worker
903*9880d681SAndroid Build Coastguard Worker // Push the new node and any (possibly new) users onto the worklist.
904*9880d681SAndroid Build Coastguard Worker AddToWorklist(TLO.New.getNode());
905*9880d681SAndroid Build Coastguard Worker AddUsersToWorklist(TLO.New.getNode());
906*9880d681SAndroid Build Coastguard Worker
907*9880d681SAndroid Build Coastguard Worker // Finally, if the node is now dead, remove it from the graph. The node
908*9880d681SAndroid Build Coastguard Worker // may not be dead if the replacement process recursively simplified to
909*9880d681SAndroid Build Coastguard Worker // something else needing this node.
910*9880d681SAndroid Build Coastguard Worker if (TLO.Old.getNode()->use_empty())
911*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(TLO.Old.getNode());
912*9880d681SAndroid Build Coastguard Worker }
913*9880d681SAndroid Build Coastguard Worker
914*9880d681SAndroid Build Coastguard Worker /// Check the specified integer node value to see if it can be simplified or if
915*9880d681SAndroid Build Coastguard Worker /// things it uses can be simplified by bit propagation. If so, return true.
SimplifyDemandedBits(SDValue Op,const APInt & Demanded)916*9880d681SAndroid Build Coastguard Worker bool DAGCombiner::SimplifyDemandedBits(SDValue Op, const APInt &Demanded) {
917*9880d681SAndroid Build Coastguard Worker TargetLowering::TargetLoweringOpt TLO(DAG, LegalTypes, LegalOperations);
918*9880d681SAndroid Build Coastguard Worker APInt KnownZero, KnownOne;
919*9880d681SAndroid Build Coastguard Worker if (!TLI.SimplifyDemandedBits(Op, Demanded, KnownZero, KnownOne, TLO))
920*9880d681SAndroid Build Coastguard Worker return false;
921*9880d681SAndroid Build Coastguard Worker
922*9880d681SAndroid Build Coastguard Worker // Revisit the node.
923*9880d681SAndroid Build Coastguard Worker AddToWorklist(Op.getNode());
924*9880d681SAndroid Build Coastguard Worker
925*9880d681SAndroid Build Coastguard Worker // Replace the old value with the new one.
926*9880d681SAndroid Build Coastguard Worker ++NodesCombined;
927*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "\nReplacing.2 ";
928*9880d681SAndroid Build Coastguard Worker TLO.Old.getNode()->dump(&DAG);
929*9880d681SAndroid Build Coastguard Worker dbgs() << "\nWith: ";
930*9880d681SAndroid Build Coastguard Worker TLO.New.getNode()->dump(&DAG);
931*9880d681SAndroid Build Coastguard Worker dbgs() << '\n');
932*9880d681SAndroid Build Coastguard Worker
933*9880d681SAndroid Build Coastguard Worker CommitTargetLoweringOpt(TLO);
934*9880d681SAndroid Build Coastguard Worker return true;
935*9880d681SAndroid Build Coastguard Worker }
936*9880d681SAndroid Build Coastguard Worker
ReplaceLoadWithPromotedLoad(SDNode * Load,SDNode * ExtLoad)937*9880d681SAndroid Build Coastguard Worker void DAGCombiner::ReplaceLoadWithPromotedLoad(SDNode *Load, SDNode *ExtLoad) {
938*9880d681SAndroid Build Coastguard Worker SDLoc dl(Load);
939*9880d681SAndroid Build Coastguard Worker EVT VT = Load->getValueType(0);
940*9880d681SAndroid Build Coastguard Worker SDValue Trunc = DAG.getNode(ISD::TRUNCATE, dl, VT, SDValue(ExtLoad, 0));
941*9880d681SAndroid Build Coastguard Worker
942*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "\nReplacing.9 ";
943*9880d681SAndroid Build Coastguard Worker Load->dump(&DAG);
944*9880d681SAndroid Build Coastguard Worker dbgs() << "\nWith: ";
945*9880d681SAndroid Build Coastguard Worker Trunc.getNode()->dump(&DAG);
946*9880d681SAndroid Build Coastguard Worker dbgs() << '\n');
947*9880d681SAndroid Build Coastguard Worker WorklistRemover DeadNodes(*this);
948*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(Load, 0), Trunc);
949*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(Load, 1), SDValue(ExtLoad, 1));
950*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(Load);
951*9880d681SAndroid Build Coastguard Worker AddToWorklist(Trunc.getNode());
952*9880d681SAndroid Build Coastguard Worker }
953*9880d681SAndroid Build Coastguard Worker
PromoteOperand(SDValue Op,EVT PVT,bool & Replace)954*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::PromoteOperand(SDValue Op, EVT PVT, bool &Replace) {
955*9880d681SAndroid Build Coastguard Worker Replace = false;
956*9880d681SAndroid Build Coastguard Worker SDLoc dl(Op);
957*9880d681SAndroid Build Coastguard Worker if (ISD::isUNINDEXEDLoad(Op.getNode())) {
958*9880d681SAndroid Build Coastguard Worker LoadSDNode *LD = cast<LoadSDNode>(Op);
959*9880d681SAndroid Build Coastguard Worker EVT MemVT = LD->getMemoryVT();
960*9880d681SAndroid Build Coastguard Worker ISD::LoadExtType ExtType = ISD::isNON_EXTLoad(LD)
961*9880d681SAndroid Build Coastguard Worker ? (TLI.isLoadExtLegal(ISD::ZEXTLOAD, PVT, MemVT) ? ISD::ZEXTLOAD
962*9880d681SAndroid Build Coastguard Worker : ISD::EXTLOAD)
963*9880d681SAndroid Build Coastguard Worker : LD->getExtensionType();
964*9880d681SAndroid Build Coastguard Worker Replace = true;
965*9880d681SAndroid Build Coastguard Worker return DAG.getExtLoad(ExtType, dl, PVT,
966*9880d681SAndroid Build Coastguard Worker LD->getChain(), LD->getBasePtr(),
967*9880d681SAndroid Build Coastguard Worker MemVT, LD->getMemOperand());
968*9880d681SAndroid Build Coastguard Worker }
969*9880d681SAndroid Build Coastguard Worker
970*9880d681SAndroid Build Coastguard Worker unsigned Opc = Op.getOpcode();
971*9880d681SAndroid Build Coastguard Worker switch (Opc) {
972*9880d681SAndroid Build Coastguard Worker default: break;
973*9880d681SAndroid Build Coastguard Worker case ISD::AssertSext:
974*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AssertSext, dl, PVT,
975*9880d681SAndroid Build Coastguard Worker SExtPromoteOperand(Op.getOperand(0), PVT),
976*9880d681SAndroid Build Coastguard Worker Op.getOperand(1));
977*9880d681SAndroid Build Coastguard Worker case ISD::AssertZext:
978*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AssertZext, dl, PVT,
979*9880d681SAndroid Build Coastguard Worker ZExtPromoteOperand(Op.getOperand(0), PVT),
980*9880d681SAndroid Build Coastguard Worker Op.getOperand(1));
981*9880d681SAndroid Build Coastguard Worker case ISD::Constant: {
982*9880d681SAndroid Build Coastguard Worker unsigned ExtOpc =
983*9880d681SAndroid Build Coastguard Worker Op.getValueType().isByteSized() ? ISD::SIGN_EXTEND : ISD::ZERO_EXTEND;
984*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ExtOpc, dl, PVT, Op);
985*9880d681SAndroid Build Coastguard Worker }
986*9880d681SAndroid Build Coastguard Worker }
987*9880d681SAndroid Build Coastguard Worker
988*9880d681SAndroid Build Coastguard Worker if (!TLI.isOperationLegal(ISD::ANY_EXTEND, PVT))
989*9880d681SAndroid Build Coastguard Worker return SDValue();
990*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ANY_EXTEND, dl, PVT, Op);
991*9880d681SAndroid Build Coastguard Worker }
992*9880d681SAndroid Build Coastguard Worker
SExtPromoteOperand(SDValue Op,EVT PVT)993*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::SExtPromoteOperand(SDValue Op, EVT PVT) {
994*9880d681SAndroid Build Coastguard Worker if (!TLI.isOperationLegal(ISD::SIGN_EXTEND_INREG, PVT))
995*9880d681SAndroid Build Coastguard Worker return SDValue();
996*9880d681SAndroid Build Coastguard Worker EVT OldVT = Op.getValueType();
997*9880d681SAndroid Build Coastguard Worker SDLoc dl(Op);
998*9880d681SAndroid Build Coastguard Worker bool Replace = false;
999*9880d681SAndroid Build Coastguard Worker SDValue NewOp = PromoteOperand(Op, PVT, Replace);
1000*9880d681SAndroid Build Coastguard Worker if (!NewOp.getNode())
1001*9880d681SAndroid Build Coastguard Worker return SDValue();
1002*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewOp.getNode());
1003*9880d681SAndroid Build Coastguard Worker
1004*9880d681SAndroid Build Coastguard Worker if (Replace)
1005*9880d681SAndroid Build Coastguard Worker ReplaceLoadWithPromotedLoad(Op.getNode(), NewOp.getNode());
1006*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SIGN_EXTEND_INREG, dl, NewOp.getValueType(), NewOp,
1007*9880d681SAndroid Build Coastguard Worker DAG.getValueType(OldVT));
1008*9880d681SAndroid Build Coastguard Worker }
1009*9880d681SAndroid Build Coastguard Worker
ZExtPromoteOperand(SDValue Op,EVT PVT)1010*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::ZExtPromoteOperand(SDValue Op, EVT PVT) {
1011*9880d681SAndroid Build Coastguard Worker EVT OldVT = Op.getValueType();
1012*9880d681SAndroid Build Coastguard Worker SDLoc dl(Op);
1013*9880d681SAndroid Build Coastguard Worker bool Replace = false;
1014*9880d681SAndroid Build Coastguard Worker SDValue NewOp = PromoteOperand(Op, PVT, Replace);
1015*9880d681SAndroid Build Coastguard Worker if (!NewOp.getNode())
1016*9880d681SAndroid Build Coastguard Worker return SDValue();
1017*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewOp.getNode());
1018*9880d681SAndroid Build Coastguard Worker
1019*9880d681SAndroid Build Coastguard Worker if (Replace)
1020*9880d681SAndroid Build Coastguard Worker ReplaceLoadWithPromotedLoad(Op.getNode(), NewOp.getNode());
1021*9880d681SAndroid Build Coastguard Worker return DAG.getZeroExtendInReg(NewOp, dl, OldVT);
1022*9880d681SAndroid Build Coastguard Worker }
1023*9880d681SAndroid Build Coastguard Worker
1024*9880d681SAndroid Build Coastguard Worker /// Promote the specified integer binary operation if the target indicates it is
1025*9880d681SAndroid Build Coastguard Worker /// beneficial. e.g. On x86, it's usually better to promote i16 operations to
1026*9880d681SAndroid Build Coastguard Worker /// i32 since i16 instructions are longer.
PromoteIntBinOp(SDValue Op)1027*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::PromoteIntBinOp(SDValue Op) {
1028*9880d681SAndroid Build Coastguard Worker if (!LegalOperations)
1029*9880d681SAndroid Build Coastguard Worker return SDValue();
1030*9880d681SAndroid Build Coastguard Worker
1031*9880d681SAndroid Build Coastguard Worker EVT VT = Op.getValueType();
1032*9880d681SAndroid Build Coastguard Worker if (VT.isVector() || !VT.isInteger())
1033*9880d681SAndroid Build Coastguard Worker return SDValue();
1034*9880d681SAndroid Build Coastguard Worker
1035*9880d681SAndroid Build Coastguard Worker // If operation type is 'undesirable', e.g. i16 on x86, consider
1036*9880d681SAndroid Build Coastguard Worker // promoting it.
1037*9880d681SAndroid Build Coastguard Worker unsigned Opc = Op.getOpcode();
1038*9880d681SAndroid Build Coastguard Worker if (TLI.isTypeDesirableForOp(Opc, VT))
1039*9880d681SAndroid Build Coastguard Worker return SDValue();
1040*9880d681SAndroid Build Coastguard Worker
1041*9880d681SAndroid Build Coastguard Worker EVT PVT = VT;
1042*9880d681SAndroid Build Coastguard Worker // Consult target whether it is a good idea to promote this operation and
1043*9880d681SAndroid Build Coastguard Worker // what's the right type to promote it to.
1044*9880d681SAndroid Build Coastguard Worker if (TLI.IsDesirableToPromoteOp(Op, PVT)) {
1045*9880d681SAndroid Build Coastguard Worker assert(PVT != VT && "Don't know what type to promote to!");
1046*9880d681SAndroid Build Coastguard Worker
1047*9880d681SAndroid Build Coastguard Worker bool Replace0 = false;
1048*9880d681SAndroid Build Coastguard Worker SDValue N0 = Op.getOperand(0);
1049*9880d681SAndroid Build Coastguard Worker SDValue NN0 = PromoteOperand(N0, PVT, Replace0);
1050*9880d681SAndroid Build Coastguard Worker if (!NN0.getNode())
1051*9880d681SAndroid Build Coastguard Worker return SDValue();
1052*9880d681SAndroid Build Coastguard Worker
1053*9880d681SAndroid Build Coastguard Worker bool Replace1 = false;
1054*9880d681SAndroid Build Coastguard Worker SDValue N1 = Op.getOperand(1);
1055*9880d681SAndroid Build Coastguard Worker SDValue NN1;
1056*9880d681SAndroid Build Coastguard Worker if (N0 == N1)
1057*9880d681SAndroid Build Coastguard Worker NN1 = NN0;
1058*9880d681SAndroid Build Coastguard Worker else {
1059*9880d681SAndroid Build Coastguard Worker NN1 = PromoteOperand(N1, PVT, Replace1);
1060*9880d681SAndroid Build Coastguard Worker if (!NN1.getNode())
1061*9880d681SAndroid Build Coastguard Worker return SDValue();
1062*9880d681SAndroid Build Coastguard Worker }
1063*9880d681SAndroid Build Coastguard Worker
1064*9880d681SAndroid Build Coastguard Worker AddToWorklist(NN0.getNode());
1065*9880d681SAndroid Build Coastguard Worker if (NN1.getNode())
1066*9880d681SAndroid Build Coastguard Worker AddToWorklist(NN1.getNode());
1067*9880d681SAndroid Build Coastguard Worker
1068*9880d681SAndroid Build Coastguard Worker if (Replace0)
1069*9880d681SAndroid Build Coastguard Worker ReplaceLoadWithPromotedLoad(N0.getNode(), NN0.getNode());
1070*9880d681SAndroid Build Coastguard Worker if (Replace1)
1071*9880d681SAndroid Build Coastguard Worker ReplaceLoadWithPromotedLoad(N1.getNode(), NN1.getNode());
1072*9880d681SAndroid Build Coastguard Worker
1073*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "\nPromoting ";
1074*9880d681SAndroid Build Coastguard Worker Op.getNode()->dump(&DAG));
1075*9880d681SAndroid Build Coastguard Worker SDLoc dl(Op);
1076*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TRUNCATE, dl, VT,
1077*9880d681SAndroid Build Coastguard Worker DAG.getNode(Opc, dl, PVT, NN0, NN1));
1078*9880d681SAndroid Build Coastguard Worker }
1079*9880d681SAndroid Build Coastguard Worker return SDValue();
1080*9880d681SAndroid Build Coastguard Worker }
1081*9880d681SAndroid Build Coastguard Worker
1082*9880d681SAndroid Build Coastguard Worker /// Promote the specified integer shift operation if the target indicates it is
1083*9880d681SAndroid Build Coastguard Worker /// beneficial. e.g. On x86, it's usually better to promote i16 operations to
1084*9880d681SAndroid Build Coastguard Worker /// i32 since i16 instructions are longer.
PromoteIntShiftOp(SDValue Op)1085*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::PromoteIntShiftOp(SDValue Op) {
1086*9880d681SAndroid Build Coastguard Worker if (!LegalOperations)
1087*9880d681SAndroid Build Coastguard Worker return SDValue();
1088*9880d681SAndroid Build Coastguard Worker
1089*9880d681SAndroid Build Coastguard Worker EVT VT = Op.getValueType();
1090*9880d681SAndroid Build Coastguard Worker if (VT.isVector() || !VT.isInteger())
1091*9880d681SAndroid Build Coastguard Worker return SDValue();
1092*9880d681SAndroid Build Coastguard Worker
1093*9880d681SAndroid Build Coastguard Worker // If operation type is 'undesirable', e.g. i16 on x86, consider
1094*9880d681SAndroid Build Coastguard Worker // promoting it.
1095*9880d681SAndroid Build Coastguard Worker unsigned Opc = Op.getOpcode();
1096*9880d681SAndroid Build Coastguard Worker if (TLI.isTypeDesirableForOp(Opc, VT))
1097*9880d681SAndroid Build Coastguard Worker return SDValue();
1098*9880d681SAndroid Build Coastguard Worker
1099*9880d681SAndroid Build Coastguard Worker EVT PVT = VT;
1100*9880d681SAndroid Build Coastguard Worker // Consult target whether it is a good idea to promote this operation and
1101*9880d681SAndroid Build Coastguard Worker // what's the right type to promote it to.
1102*9880d681SAndroid Build Coastguard Worker if (TLI.IsDesirableToPromoteOp(Op, PVT)) {
1103*9880d681SAndroid Build Coastguard Worker assert(PVT != VT && "Don't know what type to promote to!");
1104*9880d681SAndroid Build Coastguard Worker
1105*9880d681SAndroid Build Coastguard Worker bool Replace = false;
1106*9880d681SAndroid Build Coastguard Worker SDValue N0 = Op.getOperand(0);
1107*9880d681SAndroid Build Coastguard Worker if (Opc == ISD::SRA)
1108*9880d681SAndroid Build Coastguard Worker N0 = SExtPromoteOperand(Op.getOperand(0), PVT);
1109*9880d681SAndroid Build Coastguard Worker else if (Opc == ISD::SRL)
1110*9880d681SAndroid Build Coastguard Worker N0 = ZExtPromoteOperand(Op.getOperand(0), PVT);
1111*9880d681SAndroid Build Coastguard Worker else
1112*9880d681SAndroid Build Coastguard Worker N0 = PromoteOperand(N0, PVT, Replace);
1113*9880d681SAndroid Build Coastguard Worker if (!N0.getNode())
1114*9880d681SAndroid Build Coastguard Worker return SDValue();
1115*9880d681SAndroid Build Coastguard Worker
1116*9880d681SAndroid Build Coastguard Worker AddToWorklist(N0.getNode());
1117*9880d681SAndroid Build Coastguard Worker if (Replace)
1118*9880d681SAndroid Build Coastguard Worker ReplaceLoadWithPromotedLoad(Op.getOperand(0).getNode(), N0.getNode());
1119*9880d681SAndroid Build Coastguard Worker
1120*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "\nPromoting ";
1121*9880d681SAndroid Build Coastguard Worker Op.getNode()->dump(&DAG));
1122*9880d681SAndroid Build Coastguard Worker SDLoc dl(Op);
1123*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TRUNCATE, dl, VT,
1124*9880d681SAndroid Build Coastguard Worker DAG.getNode(Opc, dl, PVT, N0, Op.getOperand(1)));
1125*9880d681SAndroid Build Coastguard Worker }
1126*9880d681SAndroid Build Coastguard Worker return SDValue();
1127*9880d681SAndroid Build Coastguard Worker }
1128*9880d681SAndroid Build Coastguard Worker
PromoteExtend(SDValue Op)1129*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::PromoteExtend(SDValue Op) {
1130*9880d681SAndroid Build Coastguard Worker if (!LegalOperations)
1131*9880d681SAndroid Build Coastguard Worker return SDValue();
1132*9880d681SAndroid Build Coastguard Worker
1133*9880d681SAndroid Build Coastguard Worker EVT VT = Op.getValueType();
1134*9880d681SAndroid Build Coastguard Worker if (VT.isVector() || !VT.isInteger())
1135*9880d681SAndroid Build Coastguard Worker return SDValue();
1136*9880d681SAndroid Build Coastguard Worker
1137*9880d681SAndroid Build Coastguard Worker // If operation type is 'undesirable', e.g. i16 on x86, consider
1138*9880d681SAndroid Build Coastguard Worker // promoting it.
1139*9880d681SAndroid Build Coastguard Worker unsigned Opc = Op.getOpcode();
1140*9880d681SAndroid Build Coastguard Worker if (TLI.isTypeDesirableForOp(Opc, VT))
1141*9880d681SAndroid Build Coastguard Worker return SDValue();
1142*9880d681SAndroid Build Coastguard Worker
1143*9880d681SAndroid Build Coastguard Worker EVT PVT = VT;
1144*9880d681SAndroid Build Coastguard Worker // Consult target whether it is a good idea to promote this operation and
1145*9880d681SAndroid Build Coastguard Worker // what's the right type to promote it to.
1146*9880d681SAndroid Build Coastguard Worker if (TLI.IsDesirableToPromoteOp(Op, PVT)) {
1147*9880d681SAndroid Build Coastguard Worker assert(PVT != VT && "Don't know what type to promote to!");
1148*9880d681SAndroid Build Coastguard Worker // fold (aext (aext x)) -> (aext x)
1149*9880d681SAndroid Build Coastguard Worker // fold (aext (zext x)) -> (zext x)
1150*9880d681SAndroid Build Coastguard Worker // fold (aext (sext x)) -> (sext x)
1151*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "\nPromoting ";
1152*9880d681SAndroid Build Coastguard Worker Op.getNode()->dump(&DAG));
1153*9880d681SAndroid Build Coastguard Worker return DAG.getNode(Op.getOpcode(), SDLoc(Op), VT, Op.getOperand(0));
1154*9880d681SAndroid Build Coastguard Worker }
1155*9880d681SAndroid Build Coastguard Worker return SDValue();
1156*9880d681SAndroid Build Coastguard Worker }
1157*9880d681SAndroid Build Coastguard Worker
PromoteLoad(SDValue Op)1158*9880d681SAndroid Build Coastguard Worker bool DAGCombiner::PromoteLoad(SDValue Op) {
1159*9880d681SAndroid Build Coastguard Worker if (!LegalOperations)
1160*9880d681SAndroid Build Coastguard Worker return false;
1161*9880d681SAndroid Build Coastguard Worker
1162*9880d681SAndroid Build Coastguard Worker if (!ISD::isUNINDEXEDLoad(Op.getNode()))
1163*9880d681SAndroid Build Coastguard Worker return false;
1164*9880d681SAndroid Build Coastguard Worker
1165*9880d681SAndroid Build Coastguard Worker EVT VT = Op.getValueType();
1166*9880d681SAndroid Build Coastguard Worker if (VT.isVector() || !VT.isInteger())
1167*9880d681SAndroid Build Coastguard Worker return false;
1168*9880d681SAndroid Build Coastguard Worker
1169*9880d681SAndroid Build Coastguard Worker // If operation type is 'undesirable', e.g. i16 on x86, consider
1170*9880d681SAndroid Build Coastguard Worker // promoting it.
1171*9880d681SAndroid Build Coastguard Worker unsigned Opc = Op.getOpcode();
1172*9880d681SAndroid Build Coastguard Worker if (TLI.isTypeDesirableForOp(Opc, VT))
1173*9880d681SAndroid Build Coastguard Worker return false;
1174*9880d681SAndroid Build Coastguard Worker
1175*9880d681SAndroid Build Coastguard Worker EVT PVT = VT;
1176*9880d681SAndroid Build Coastguard Worker // Consult target whether it is a good idea to promote this operation and
1177*9880d681SAndroid Build Coastguard Worker // what's the right type to promote it to.
1178*9880d681SAndroid Build Coastguard Worker if (TLI.IsDesirableToPromoteOp(Op, PVT)) {
1179*9880d681SAndroid Build Coastguard Worker assert(PVT != VT && "Don't know what type to promote to!");
1180*9880d681SAndroid Build Coastguard Worker
1181*9880d681SAndroid Build Coastguard Worker SDLoc dl(Op);
1182*9880d681SAndroid Build Coastguard Worker SDNode *N = Op.getNode();
1183*9880d681SAndroid Build Coastguard Worker LoadSDNode *LD = cast<LoadSDNode>(N);
1184*9880d681SAndroid Build Coastguard Worker EVT MemVT = LD->getMemoryVT();
1185*9880d681SAndroid Build Coastguard Worker ISD::LoadExtType ExtType = ISD::isNON_EXTLoad(LD)
1186*9880d681SAndroid Build Coastguard Worker ? (TLI.isLoadExtLegal(ISD::ZEXTLOAD, PVT, MemVT) ? ISD::ZEXTLOAD
1187*9880d681SAndroid Build Coastguard Worker : ISD::EXTLOAD)
1188*9880d681SAndroid Build Coastguard Worker : LD->getExtensionType();
1189*9880d681SAndroid Build Coastguard Worker SDValue NewLD = DAG.getExtLoad(ExtType, dl, PVT,
1190*9880d681SAndroid Build Coastguard Worker LD->getChain(), LD->getBasePtr(),
1191*9880d681SAndroid Build Coastguard Worker MemVT, LD->getMemOperand());
1192*9880d681SAndroid Build Coastguard Worker SDValue Result = DAG.getNode(ISD::TRUNCATE, dl, VT, NewLD);
1193*9880d681SAndroid Build Coastguard Worker
1194*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "\nPromoting ";
1195*9880d681SAndroid Build Coastguard Worker N->dump(&DAG);
1196*9880d681SAndroid Build Coastguard Worker dbgs() << "\nTo: ";
1197*9880d681SAndroid Build Coastguard Worker Result.getNode()->dump(&DAG);
1198*9880d681SAndroid Build Coastguard Worker dbgs() << '\n');
1199*9880d681SAndroid Build Coastguard Worker WorklistRemover DeadNodes(*this);
1200*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(N, 0), Result);
1201*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(N, 1), NewLD.getValue(1));
1202*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(N);
1203*9880d681SAndroid Build Coastguard Worker AddToWorklist(Result.getNode());
1204*9880d681SAndroid Build Coastguard Worker return true;
1205*9880d681SAndroid Build Coastguard Worker }
1206*9880d681SAndroid Build Coastguard Worker return false;
1207*9880d681SAndroid Build Coastguard Worker }
1208*9880d681SAndroid Build Coastguard Worker
1209*9880d681SAndroid Build Coastguard Worker /// \brief Recursively delete a node which has no uses and any operands for
1210*9880d681SAndroid Build Coastguard Worker /// which it is the only use.
1211*9880d681SAndroid Build Coastguard Worker ///
1212*9880d681SAndroid Build Coastguard Worker /// Note that this both deletes the nodes and removes them from the worklist.
1213*9880d681SAndroid Build Coastguard Worker /// It also adds any nodes who have had a user deleted to the worklist as they
1214*9880d681SAndroid Build Coastguard Worker /// may now have only one use and subject to other combines.
recursivelyDeleteUnusedNodes(SDNode * N)1215*9880d681SAndroid Build Coastguard Worker bool DAGCombiner::recursivelyDeleteUnusedNodes(SDNode *N) {
1216*9880d681SAndroid Build Coastguard Worker if (!N->use_empty())
1217*9880d681SAndroid Build Coastguard Worker return false;
1218*9880d681SAndroid Build Coastguard Worker
1219*9880d681SAndroid Build Coastguard Worker SmallSetVector<SDNode *, 16> Nodes;
1220*9880d681SAndroid Build Coastguard Worker Nodes.insert(N);
1221*9880d681SAndroid Build Coastguard Worker do {
1222*9880d681SAndroid Build Coastguard Worker N = Nodes.pop_back_val();
1223*9880d681SAndroid Build Coastguard Worker if (!N)
1224*9880d681SAndroid Build Coastguard Worker continue;
1225*9880d681SAndroid Build Coastguard Worker
1226*9880d681SAndroid Build Coastguard Worker if (N->use_empty()) {
1227*9880d681SAndroid Build Coastguard Worker for (const SDValue &ChildN : N->op_values())
1228*9880d681SAndroid Build Coastguard Worker Nodes.insert(ChildN.getNode());
1229*9880d681SAndroid Build Coastguard Worker
1230*9880d681SAndroid Build Coastguard Worker removeFromWorklist(N);
1231*9880d681SAndroid Build Coastguard Worker DAG.DeleteNode(N);
1232*9880d681SAndroid Build Coastguard Worker } else {
1233*9880d681SAndroid Build Coastguard Worker AddToWorklist(N);
1234*9880d681SAndroid Build Coastguard Worker }
1235*9880d681SAndroid Build Coastguard Worker } while (!Nodes.empty());
1236*9880d681SAndroid Build Coastguard Worker return true;
1237*9880d681SAndroid Build Coastguard Worker }
1238*9880d681SAndroid Build Coastguard Worker
1239*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
1240*9880d681SAndroid Build Coastguard Worker // Main DAG Combiner implementation
1241*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
1242*9880d681SAndroid Build Coastguard Worker
Run(CombineLevel AtLevel)1243*9880d681SAndroid Build Coastguard Worker void DAGCombiner::Run(CombineLevel AtLevel) {
1244*9880d681SAndroid Build Coastguard Worker // set the instance variables, so that the various visit routines may use it.
1245*9880d681SAndroid Build Coastguard Worker Level = AtLevel;
1246*9880d681SAndroid Build Coastguard Worker LegalOperations = Level >= AfterLegalizeVectorOps;
1247*9880d681SAndroid Build Coastguard Worker LegalTypes = Level >= AfterLegalizeTypes;
1248*9880d681SAndroid Build Coastguard Worker
1249*9880d681SAndroid Build Coastguard Worker // Add all the dag nodes to the worklist.
1250*9880d681SAndroid Build Coastguard Worker for (SDNode &Node : DAG.allnodes())
1251*9880d681SAndroid Build Coastguard Worker AddToWorklist(&Node);
1252*9880d681SAndroid Build Coastguard Worker
1253*9880d681SAndroid Build Coastguard Worker // Create a dummy node (which is not added to allnodes), that adds a reference
1254*9880d681SAndroid Build Coastguard Worker // to the root node, preventing it from being deleted, and tracking any
1255*9880d681SAndroid Build Coastguard Worker // changes of the root.
1256*9880d681SAndroid Build Coastguard Worker HandleSDNode Dummy(DAG.getRoot());
1257*9880d681SAndroid Build Coastguard Worker
1258*9880d681SAndroid Build Coastguard Worker // While the worklist isn't empty, find a node and try to combine it.
1259*9880d681SAndroid Build Coastguard Worker while (!WorklistMap.empty()) {
1260*9880d681SAndroid Build Coastguard Worker SDNode *N;
1261*9880d681SAndroid Build Coastguard Worker // The Worklist holds the SDNodes in order, but it may contain null entries.
1262*9880d681SAndroid Build Coastguard Worker do {
1263*9880d681SAndroid Build Coastguard Worker N = Worklist.pop_back_val();
1264*9880d681SAndroid Build Coastguard Worker } while (!N);
1265*9880d681SAndroid Build Coastguard Worker
1266*9880d681SAndroid Build Coastguard Worker bool GoodWorklistEntry = WorklistMap.erase(N);
1267*9880d681SAndroid Build Coastguard Worker (void)GoodWorklistEntry;
1268*9880d681SAndroid Build Coastguard Worker assert(GoodWorklistEntry &&
1269*9880d681SAndroid Build Coastguard Worker "Found a worklist entry without a corresponding map entry!");
1270*9880d681SAndroid Build Coastguard Worker
1271*9880d681SAndroid Build Coastguard Worker // If N has no uses, it is dead. Make sure to revisit all N's operands once
1272*9880d681SAndroid Build Coastguard Worker // N is deleted from the DAG, since they too may now be dead or may have a
1273*9880d681SAndroid Build Coastguard Worker // reduced number of uses, allowing other xforms.
1274*9880d681SAndroid Build Coastguard Worker if (recursivelyDeleteUnusedNodes(N))
1275*9880d681SAndroid Build Coastguard Worker continue;
1276*9880d681SAndroid Build Coastguard Worker
1277*9880d681SAndroid Build Coastguard Worker WorklistRemover DeadNodes(*this);
1278*9880d681SAndroid Build Coastguard Worker
1279*9880d681SAndroid Build Coastguard Worker // If this combine is running after legalizing the DAG, re-legalize any
1280*9880d681SAndroid Build Coastguard Worker // nodes pulled off the worklist.
1281*9880d681SAndroid Build Coastguard Worker if (Level == AfterLegalizeDAG) {
1282*9880d681SAndroid Build Coastguard Worker SmallSetVector<SDNode *, 16> UpdatedNodes;
1283*9880d681SAndroid Build Coastguard Worker bool NIsValid = DAG.LegalizeOp(N, UpdatedNodes);
1284*9880d681SAndroid Build Coastguard Worker
1285*9880d681SAndroid Build Coastguard Worker for (SDNode *LN : UpdatedNodes) {
1286*9880d681SAndroid Build Coastguard Worker AddToWorklist(LN);
1287*9880d681SAndroid Build Coastguard Worker AddUsersToWorklist(LN);
1288*9880d681SAndroid Build Coastguard Worker }
1289*9880d681SAndroid Build Coastguard Worker if (!NIsValid)
1290*9880d681SAndroid Build Coastguard Worker continue;
1291*9880d681SAndroid Build Coastguard Worker }
1292*9880d681SAndroid Build Coastguard Worker
1293*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "\nCombining: "; N->dump(&DAG));
1294*9880d681SAndroid Build Coastguard Worker
1295*9880d681SAndroid Build Coastguard Worker // Add any operands of the new node which have not yet been combined to the
1296*9880d681SAndroid Build Coastguard Worker // worklist as well. Because the worklist uniques things already, this
1297*9880d681SAndroid Build Coastguard Worker // won't repeatedly process the same operand.
1298*9880d681SAndroid Build Coastguard Worker CombinedNodes.insert(N);
1299*9880d681SAndroid Build Coastguard Worker for (const SDValue &ChildN : N->op_values())
1300*9880d681SAndroid Build Coastguard Worker if (!CombinedNodes.count(ChildN.getNode()))
1301*9880d681SAndroid Build Coastguard Worker AddToWorklist(ChildN.getNode());
1302*9880d681SAndroid Build Coastguard Worker
1303*9880d681SAndroid Build Coastguard Worker SDValue RV = combine(N);
1304*9880d681SAndroid Build Coastguard Worker
1305*9880d681SAndroid Build Coastguard Worker if (!RV.getNode())
1306*9880d681SAndroid Build Coastguard Worker continue;
1307*9880d681SAndroid Build Coastguard Worker
1308*9880d681SAndroid Build Coastguard Worker ++NodesCombined;
1309*9880d681SAndroid Build Coastguard Worker
1310*9880d681SAndroid Build Coastguard Worker // If we get back the same node we passed in, rather than a new node or
1311*9880d681SAndroid Build Coastguard Worker // zero, we know that the node must have defined multiple values and
1312*9880d681SAndroid Build Coastguard Worker // CombineTo was used. Since CombineTo takes care of the worklist
1313*9880d681SAndroid Build Coastguard Worker // mechanics for us, we have no work to do in this case.
1314*9880d681SAndroid Build Coastguard Worker if (RV.getNode() == N)
1315*9880d681SAndroid Build Coastguard Worker continue;
1316*9880d681SAndroid Build Coastguard Worker
1317*9880d681SAndroid Build Coastguard Worker assert(N->getOpcode() != ISD::DELETED_NODE &&
1318*9880d681SAndroid Build Coastguard Worker RV.getNode()->getOpcode() != ISD::DELETED_NODE &&
1319*9880d681SAndroid Build Coastguard Worker "Node was deleted but visit returned new node!");
1320*9880d681SAndroid Build Coastguard Worker
1321*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << " ... into: ";
1322*9880d681SAndroid Build Coastguard Worker RV.getNode()->dump(&DAG));
1323*9880d681SAndroid Build Coastguard Worker
1324*9880d681SAndroid Build Coastguard Worker if (N->getNumValues() == RV.getNode()->getNumValues())
1325*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesWith(N, RV.getNode());
1326*9880d681SAndroid Build Coastguard Worker else {
1327*9880d681SAndroid Build Coastguard Worker assert(N->getValueType(0) == RV.getValueType() &&
1328*9880d681SAndroid Build Coastguard Worker N->getNumValues() == 1 && "Type mismatch");
1329*9880d681SAndroid Build Coastguard Worker SDValue OpV = RV;
1330*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesWith(N, &OpV);
1331*9880d681SAndroid Build Coastguard Worker }
1332*9880d681SAndroid Build Coastguard Worker
1333*9880d681SAndroid Build Coastguard Worker // Push the new node and any users onto the worklist
1334*9880d681SAndroid Build Coastguard Worker AddToWorklist(RV.getNode());
1335*9880d681SAndroid Build Coastguard Worker AddUsersToWorklist(RV.getNode());
1336*9880d681SAndroid Build Coastguard Worker
1337*9880d681SAndroid Build Coastguard Worker // Finally, if the node is now dead, remove it from the graph. The node
1338*9880d681SAndroid Build Coastguard Worker // may not be dead if the replacement process recursively simplified to
1339*9880d681SAndroid Build Coastguard Worker // something else needing this node. This will also take care of adding any
1340*9880d681SAndroid Build Coastguard Worker // operands which have lost a user to the worklist.
1341*9880d681SAndroid Build Coastguard Worker recursivelyDeleteUnusedNodes(N);
1342*9880d681SAndroid Build Coastguard Worker }
1343*9880d681SAndroid Build Coastguard Worker
1344*9880d681SAndroid Build Coastguard Worker // If the root changed (e.g. it was a dead load, update the root).
1345*9880d681SAndroid Build Coastguard Worker DAG.setRoot(Dummy.getValue());
1346*9880d681SAndroid Build Coastguard Worker DAG.RemoveDeadNodes();
1347*9880d681SAndroid Build Coastguard Worker }
1348*9880d681SAndroid Build Coastguard Worker
visit(SDNode * N)1349*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visit(SDNode *N) {
1350*9880d681SAndroid Build Coastguard Worker switch (N->getOpcode()) {
1351*9880d681SAndroid Build Coastguard Worker default: break;
1352*9880d681SAndroid Build Coastguard Worker case ISD::TokenFactor: return visitTokenFactor(N);
1353*9880d681SAndroid Build Coastguard Worker case ISD::MERGE_VALUES: return visitMERGE_VALUES(N);
1354*9880d681SAndroid Build Coastguard Worker case ISD::ADD: return visitADD(N);
1355*9880d681SAndroid Build Coastguard Worker case ISD::SUB: return visitSUB(N);
1356*9880d681SAndroid Build Coastguard Worker case ISD::ADDC: return visitADDC(N);
1357*9880d681SAndroid Build Coastguard Worker case ISD::SUBC: return visitSUBC(N);
1358*9880d681SAndroid Build Coastguard Worker case ISD::ADDE: return visitADDE(N);
1359*9880d681SAndroid Build Coastguard Worker case ISD::SUBE: return visitSUBE(N);
1360*9880d681SAndroid Build Coastguard Worker case ISD::MUL: return visitMUL(N);
1361*9880d681SAndroid Build Coastguard Worker case ISD::SDIV: return visitSDIV(N);
1362*9880d681SAndroid Build Coastguard Worker case ISD::UDIV: return visitUDIV(N);
1363*9880d681SAndroid Build Coastguard Worker case ISD::SREM:
1364*9880d681SAndroid Build Coastguard Worker case ISD::UREM: return visitREM(N);
1365*9880d681SAndroid Build Coastguard Worker case ISD::MULHU: return visitMULHU(N);
1366*9880d681SAndroid Build Coastguard Worker case ISD::MULHS: return visitMULHS(N);
1367*9880d681SAndroid Build Coastguard Worker case ISD::SMUL_LOHI: return visitSMUL_LOHI(N);
1368*9880d681SAndroid Build Coastguard Worker case ISD::UMUL_LOHI: return visitUMUL_LOHI(N);
1369*9880d681SAndroid Build Coastguard Worker case ISD::SMULO: return visitSMULO(N);
1370*9880d681SAndroid Build Coastguard Worker case ISD::UMULO: return visitUMULO(N);
1371*9880d681SAndroid Build Coastguard Worker case ISD::SMIN:
1372*9880d681SAndroid Build Coastguard Worker case ISD::SMAX:
1373*9880d681SAndroid Build Coastguard Worker case ISD::UMIN:
1374*9880d681SAndroid Build Coastguard Worker case ISD::UMAX: return visitIMINMAX(N);
1375*9880d681SAndroid Build Coastguard Worker case ISD::AND: return visitAND(N);
1376*9880d681SAndroid Build Coastguard Worker case ISD::OR: return visitOR(N);
1377*9880d681SAndroid Build Coastguard Worker case ISD::XOR: return visitXOR(N);
1378*9880d681SAndroid Build Coastguard Worker case ISD::SHL: return visitSHL(N);
1379*9880d681SAndroid Build Coastguard Worker case ISD::SRA: return visitSRA(N);
1380*9880d681SAndroid Build Coastguard Worker case ISD::SRL: return visitSRL(N);
1381*9880d681SAndroid Build Coastguard Worker case ISD::ROTR:
1382*9880d681SAndroid Build Coastguard Worker case ISD::ROTL: return visitRotate(N);
1383*9880d681SAndroid Build Coastguard Worker case ISD::BSWAP: return visitBSWAP(N);
1384*9880d681SAndroid Build Coastguard Worker case ISD::BITREVERSE: return visitBITREVERSE(N);
1385*9880d681SAndroid Build Coastguard Worker case ISD::CTLZ: return visitCTLZ(N);
1386*9880d681SAndroid Build Coastguard Worker case ISD::CTLZ_ZERO_UNDEF: return visitCTLZ_ZERO_UNDEF(N);
1387*9880d681SAndroid Build Coastguard Worker case ISD::CTTZ: return visitCTTZ(N);
1388*9880d681SAndroid Build Coastguard Worker case ISD::CTTZ_ZERO_UNDEF: return visitCTTZ_ZERO_UNDEF(N);
1389*9880d681SAndroid Build Coastguard Worker case ISD::CTPOP: return visitCTPOP(N);
1390*9880d681SAndroid Build Coastguard Worker case ISD::SELECT: return visitSELECT(N);
1391*9880d681SAndroid Build Coastguard Worker case ISD::VSELECT: return visitVSELECT(N);
1392*9880d681SAndroid Build Coastguard Worker case ISD::SELECT_CC: return visitSELECT_CC(N);
1393*9880d681SAndroid Build Coastguard Worker case ISD::SETCC: return visitSETCC(N);
1394*9880d681SAndroid Build Coastguard Worker case ISD::SETCCE: return visitSETCCE(N);
1395*9880d681SAndroid Build Coastguard Worker case ISD::SIGN_EXTEND: return visitSIGN_EXTEND(N);
1396*9880d681SAndroid Build Coastguard Worker case ISD::ZERO_EXTEND: return visitZERO_EXTEND(N);
1397*9880d681SAndroid Build Coastguard Worker case ISD::ANY_EXTEND: return visitANY_EXTEND(N);
1398*9880d681SAndroid Build Coastguard Worker case ISD::SIGN_EXTEND_INREG: return visitSIGN_EXTEND_INREG(N);
1399*9880d681SAndroid Build Coastguard Worker case ISD::SIGN_EXTEND_VECTOR_INREG: return visitSIGN_EXTEND_VECTOR_INREG(N);
1400*9880d681SAndroid Build Coastguard Worker case ISD::ZERO_EXTEND_VECTOR_INREG: return visitZERO_EXTEND_VECTOR_INREG(N);
1401*9880d681SAndroid Build Coastguard Worker case ISD::TRUNCATE: return visitTRUNCATE(N);
1402*9880d681SAndroid Build Coastguard Worker case ISD::BITCAST: return visitBITCAST(N);
1403*9880d681SAndroid Build Coastguard Worker case ISD::BUILD_PAIR: return visitBUILD_PAIR(N);
1404*9880d681SAndroid Build Coastguard Worker case ISD::FADD: return visitFADD(N);
1405*9880d681SAndroid Build Coastguard Worker case ISD::FSUB: return visitFSUB(N);
1406*9880d681SAndroid Build Coastguard Worker case ISD::FMUL: return visitFMUL(N);
1407*9880d681SAndroid Build Coastguard Worker case ISD::FMA: return visitFMA(N);
1408*9880d681SAndroid Build Coastguard Worker case ISD::FDIV: return visitFDIV(N);
1409*9880d681SAndroid Build Coastguard Worker case ISD::FREM: return visitFREM(N);
1410*9880d681SAndroid Build Coastguard Worker case ISD::FSQRT: return visitFSQRT(N);
1411*9880d681SAndroid Build Coastguard Worker case ISD::FCOPYSIGN: return visitFCOPYSIGN(N);
1412*9880d681SAndroid Build Coastguard Worker case ISD::SINT_TO_FP: return visitSINT_TO_FP(N);
1413*9880d681SAndroid Build Coastguard Worker case ISD::UINT_TO_FP: return visitUINT_TO_FP(N);
1414*9880d681SAndroid Build Coastguard Worker case ISD::FP_TO_SINT: return visitFP_TO_SINT(N);
1415*9880d681SAndroid Build Coastguard Worker case ISD::FP_TO_UINT: return visitFP_TO_UINT(N);
1416*9880d681SAndroid Build Coastguard Worker case ISD::FP_ROUND: return visitFP_ROUND(N);
1417*9880d681SAndroid Build Coastguard Worker case ISD::FP_ROUND_INREG: return visitFP_ROUND_INREG(N);
1418*9880d681SAndroid Build Coastguard Worker case ISD::FP_EXTEND: return visitFP_EXTEND(N);
1419*9880d681SAndroid Build Coastguard Worker case ISD::FNEG: return visitFNEG(N);
1420*9880d681SAndroid Build Coastguard Worker case ISD::FABS: return visitFABS(N);
1421*9880d681SAndroid Build Coastguard Worker case ISD::FFLOOR: return visitFFLOOR(N);
1422*9880d681SAndroid Build Coastguard Worker case ISD::FMINNUM: return visitFMINNUM(N);
1423*9880d681SAndroid Build Coastguard Worker case ISD::FMAXNUM: return visitFMAXNUM(N);
1424*9880d681SAndroid Build Coastguard Worker case ISD::FCEIL: return visitFCEIL(N);
1425*9880d681SAndroid Build Coastguard Worker case ISD::FTRUNC: return visitFTRUNC(N);
1426*9880d681SAndroid Build Coastguard Worker case ISD::BRCOND: return visitBRCOND(N);
1427*9880d681SAndroid Build Coastguard Worker case ISD::BR_CC: return visitBR_CC(N);
1428*9880d681SAndroid Build Coastguard Worker case ISD::LOAD: return visitLOAD(N);
1429*9880d681SAndroid Build Coastguard Worker case ISD::STORE: return visitSTORE(N);
1430*9880d681SAndroid Build Coastguard Worker case ISD::INSERT_VECTOR_ELT: return visitINSERT_VECTOR_ELT(N);
1431*9880d681SAndroid Build Coastguard Worker case ISD::EXTRACT_VECTOR_ELT: return visitEXTRACT_VECTOR_ELT(N);
1432*9880d681SAndroid Build Coastguard Worker case ISD::BUILD_VECTOR: return visitBUILD_VECTOR(N);
1433*9880d681SAndroid Build Coastguard Worker case ISD::CONCAT_VECTORS: return visitCONCAT_VECTORS(N);
1434*9880d681SAndroid Build Coastguard Worker case ISD::EXTRACT_SUBVECTOR: return visitEXTRACT_SUBVECTOR(N);
1435*9880d681SAndroid Build Coastguard Worker case ISD::VECTOR_SHUFFLE: return visitVECTOR_SHUFFLE(N);
1436*9880d681SAndroid Build Coastguard Worker case ISD::SCALAR_TO_VECTOR: return visitSCALAR_TO_VECTOR(N);
1437*9880d681SAndroid Build Coastguard Worker case ISD::INSERT_SUBVECTOR: return visitINSERT_SUBVECTOR(N);
1438*9880d681SAndroid Build Coastguard Worker case ISD::MGATHER: return visitMGATHER(N);
1439*9880d681SAndroid Build Coastguard Worker case ISD::MLOAD: return visitMLOAD(N);
1440*9880d681SAndroid Build Coastguard Worker case ISD::MSCATTER: return visitMSCATTER(N);
1441*9880d681SAndroid Build Coastguard Worker case ISD::MSTORE: return visitMSTORE(N);
1442*9880d681SAndroid Build Coastguard Worker case ISD::FP_TO_FP16: return visitFP_TO_FP16(N);
1443*9880d681SAndroid Build Coastguard Worker case ISD::FP16_TO_FP: return visitFP16_TO_FP(N);
1444*9880d681SAndroid Build Coastguard Worker }
1445*9880d681SAndroid Build Coastguard Worker return SDValue();
1446*9880d681SAndroid Build Coastguard Worker }
1447*9880d681SAndroid Build Coastguard Worker
combine(SDNode * N)1448*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::combine(SDNode *N) {
1449*9880d681SAndroid Build Coastguard Worker SDValue RV = visit(N);
1450*9880d681SAndroid Build Coastguard Worker
1451*9880d681SAndroid Build Coastguard Worker // If nothing happened, try a target-specific DAG combine.
1452*9880d681SAndroid Build Coastguard Worker if (!RV.getNode()) {
1453*9880d681SAndroid Build Coastguard Worker assert(N->getOpcode() != ISD::DELETED_NODE &&
1454*9880d681SAndroid Build Coastguard Worker "Node was deleted but visit returned NULL!");
1455*9880d681SAndroid Build Coastguard Worker
1456*9880d681SAndroid Build Coastguard Worker if (N->getOpcode() >= ISD::BUILTIN_OP_END ||
1457*9880d681SAndroid Build Coastguard Worker TLI.hasTargetDAGCombine((ISD::NodeType)N->getOpcode())) {
1458*9880d681SAndroid Build Coastguard Worker
1459*9880d681SAndroid Build Coastguard Worker // Expose the DAG combiner to the target combiner impls.
1460*9880d681SAndroid Build Coastguard Worker TargetLowering::DAGCombinerInfo
1461*9880d681SAndroid Build Coastguard Worker DagCombineInfo(DAG, Level, false, this);
1462*9880d681SAndroid Build Coastguard Worker
1463*9880d681SAndroid Build Coastguard Worker RV = TLI.PerformDAGCombine(N, DagCombineInfo);
1464*9880d681SAndroid Build Coastguard Worker }
1465*9880d681SAndroid Build Coastguard Worker }
1466*9880d681SAndroid Build Coastguard Worker
1467*9880d681SAndroid Build Coastguard Worker // If nothing happened still, try promoting the operation.
1468*9880d681SAndroid Build Coastguard Worker if (!RV.getNode()) {
1469*9880d681SAndroid Build Coastguard Worker switch (N->getOpcode()) {
1470*9880d681SAndroid Build Coastguard Worker default: break;
1471*9880d681SAndroid Build Coastguard Worker case ISD::ADD:
1472*9880d681SAndroid Build Coastguard Worker case ISD::SUB:
1473*9880d681SAndroid Build Coastguard Worker case ISD::MUL:
1474*9880d681SAndroid Build Coastguard Worker case ISD::AND:
1475*9880d681SAndroid Build Coastguard Worker case ISD::OR:
1476*9880d681SAndroid Build Coastguard Worker case ISD::XOR:
1477*9880d681SAndroid Build Coastguard Worker RV = PromoteIntBinOp(SDValue(N, 0));
1478*9880d681SAndroid Build Coastguard Worker break;
1479*9880d681SAndroid Build Coastguard Worker case ISD::SHL:
1480*9880d681SAndroid Build Coastguard Worker case ISD::SRA:
1481*9880d681SAndroid Build Coastguard Worker case ISD::SRL:
1482*9880d681SAndroid Build Coastguard Worker RV = PromoteIntShiftOp(SDValue(N, 0));
1483*9880d681SAndroid Build Coastguard Worker break;
1484*9880d681SAndroid Build Coastguard Worker case ISD::SIGN_EXTEND:
1485*9880d681SAndroid Build Coastguard Worker case ISD::ZERO_EXTEND:
1486*9880d681SAndroid Build Coastguard Worker case ISD::ANY_EXTEND:
1487*9880d681SAndroid Build Coastguard Worker RV = PromoteExtend(SDValue(N, 0));
1488*9880d681SAndroid Build Coastguard Worker break;
1489*9880d681SAndroid Build Coastguard Worker case ISD::LOAD:
1490*9880d681SAndroid Build Coastguard Worker if (PromoteLoad(SDValue(N, 0)))
1491*9880d681SAndroid Build Coastguard Worker RV = SDValue(N, 0);
1492*9880d681SAndroid Build Coastguard Worker break;
1493*9880d681SAndroid Build Coastguard Worker }
1494*9880d681SAndroid Build Coastguard Worker }
1495*9880d681SAndroid Build Coastguard Worker
1496*9880d681SAndroid Build Coastguard Worker // If N is a commutative binary node, try commuting it to enable more
1497*9880d681SAndroid Build Coastguard Worker // sdisel CSE.
1498*9880d681SAndroid Build Coastguard Worker if (!RV.getNode() && SelectionDAG::isCommutativeBinOp(N->getOpcode()) &&
1499*9880d681SAndroid Build Coastguard Worker N->getNumValues() == 1) {
1500*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
1501*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
1502*9880d681SAndroid Build Coastguard Worker
1503*9880d681SAndroid Build Coastguard Worker // Constant operands are canonicalized to RHS.
1504*9880d681SAndroid Build Coastguard Worker if (isa<ConstantSDNode>(N0) || !isa<ConstantSDNode>(N1)) {
1505*9880d681SAndroid Build Coastguard Worker SDValue Ops[] = {N1, N0};
1506*9880d681SAndroid Build Coastguard Worker SDNode *CSENode = DAG.getNodeIfExists(N->getOpcode(), N->getVTList(), Ops,
1507*9880d681SAndroid Build Coastguard Worker N->getFlags());
1508*9880d681SAndroid Build Coastguard Worker if (CSENode)
1509*9880d681SAndroid Build Coastguard Worker return SDValue(CSENode, 0);
1510*9880d681SAndroid Build Coastguard Worker }
1511*9880d681SAndroid Build Coastguard Worker }
1512*9880d681SAndroid Build Coastguard Worker
1513*9880d681SAndroid Build Coastguard Worker return RV;
1514*9880d681SAndroid Build Coastguard Worker }
1515*9880d681SAndroid Build Coastguard Worker
1516*9880d681SAndroid Build Coastguard Worker /// Given a node, return its input chain if it has one, otherwise return a null
1517*9880d681SAndroid Build Coastguard Worker /// sd operand.
getInputChainForNode(SDNode * N)1518*9880d681SAndroid Build Coastguard Worker static SDValue getInputChainForNode(SDNode *N) {
1519*9880d681SAndroid Build Coastguard Worker if (unsigned NumOps = N->getNumOperands()) {
1520*9880d681SAndroid Build Coastguard Worker if (N->getOperand(0).getValueType() == MVT::Other)
1521*9880d681SAndroid Build Coastguard Worker return N->getOperand(0);
1522*9880d681SAndroid Build Coastguard Worker if (N->getOperand(NumOps-1).getValueType() == MVT::Other)
1523*9880d681SAndroid Build Coastguard Worker return N->getOperand(NumOps-1);
1524*9880d681SAndroid Build Coastguard Worker for (unsigned i = 1; i < NumOps-1; ++i)
1525*9880d681SAndroid Build Coastguard Worker if (N->getOperand(i).getValueType() == MVT::Other)
1526*9880d681SAndroid Build Coastguard Worker return N->getOperand(i);
1527*9880d681SAndroid Build Coastguard Worker }
1528*9880d681SAndroid Build Coastguard Worker return SDValue();
1529*9880d681SAndroid Build Coastguard Worker }
1530*9880d681SAndroid Build Coastguard Worker
visitTokenFactor(SDNode * N)1531*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitTokenFactor(SDNode *N) {
1532*9880d681SAndroid Build Coastguard Worker // If N has two operands, where one has an input chain equal to the other,
1533*9880d681SAndroid Build Coastguard Worker // the 'other' chain is redundant.
1534*9880d681SAndroid Build Coastguard Worker if (N->getNumOperands() == 2) {
1535*9880d681SAndroid Build Coastguard Worker if (getInputChainForNode(N->getOperand(0).getNode()) == N->getOperand(1))
1536*9880d681SAndroid Build Coastguard Worker return N->getOperand(0);
1537*9880d681SAndroid Build Coastguard Worker if (getInputChainForNode(N->getOperand(1).getNode()) == N->getOperand(0))
1538*9880d681SAndroid Build Coastguard Worker return N->getOperand(1);
1539*9880d681SAndroid Build Coastguard Worker }
1540*9880d681SAndroid Build Coastguard Worker
1541*9880d681SAndroid Build Coastguard Worker SmallVector<SDNode *, 8> TFs; // List of token factors to visit.
1542*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Ops; // Ops for replacing token factor.
1543*9880d681SAndroid Build Coastguard Worker SmallPtrSet<SDNode*, 16> SeenOps;
1544*9880d681SAndroid Build Coastguard Worker bool Changed = false; // If we should replace this token factor.
1545*9880d681SAndroid Build Coastguard Worker
1546*9880d681SAndroid Build Coastguard Worker // Start out with this token factor.
1547*9880d681SAndroid Build Coastguard Worker TFs.push_back(N);
1548*9880d681SAndroid Build Coastguard Worker
1549*9880d681SAndroid Build Coastguard Worker // Iterate through token factors. The TFs grows when new token factors are
1550*9880d681SAndroid Build Coastguard Worker // encountered.
1551*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < TFs.size(); ++i) {
1552*9880d681SAndroid Build Coastguard Worker SDNode *TF = TFs[i];
1553*9880d681SAndroid Build Coastguard Worker
1554*9880d681SAndroid Build Coastguard Worker // Check each of the operands.
1555*9880d681SAndroid Build Coastguard Worker for (const SDValue &Op : TF->op_values()) {
1556*9880d681SAndroid Build Coastguard Worker
1557*9880d681SAndroid Build Coastguard Worker switch (Op.getOpcode()) {
1558*9880d681SAndroid Build Coastguard Worker case ISD::EntryToken:
1559*9880d681SAndroid Build Coastguard Worker // Entry tokens don't need to be added to the list. They are
1560*9880d681SAndroid Build Coastguard Worker // redundant.
1561*9880d681SAndroid Build Coastguard Worker Changed = true;
1562*9880d681SAndroid Build Coastguard Worker break;
1563*9880d681SAndroid Build Coastguard Worker
1564*9880d681SAndroid Build Coastguard Worker case ISD::TokenFactor:
1565*9880d681SAndroid Build Coastguard Worker if (Op.hasOneUse() &&
1566*9880d681SAndroid Build Coastguard Worker std::find(TFs.begin(), TFs.end(), Op.getNode()) == TFs.end()) {
1567*9880d681SAndroid Build Coastguard Worker // Queue up for processing.
1568*9880d681SAndroid Build Coastguard Worker TFs.push_back(Op.getNode());
1569*9880d681SAndroid Build Coastguard Worker // Clean up in case the token factor is removed.
1570*9880d681SAndroid Build Coastguard Worker AddToWorklist(Op.getNode());
1571*9880d681SAndroid Build Coastguard Worker Changed = true;
1572*9880d681SAndroid Build Coastguard Worker break;
1573*9880d681SAndroid Build Coastguard Worker }
1574*9880d681SAndroid Build Coastguard Worker // Fall thru
1575*9880d681SAndroid Build Coastguard Worker
1576*9880d681SAndroid Build Coastguard Worker default:
1577*9880d681SAndroid Build Coastguard Worker // Only add if it isn't already in the list.
1578*9880d681SAndroid Build Coastguard Worker if (SeenOps.insert(Op.getNode()).second)
1579*9880d681SAndroid Build Coastguard Worker Ops.push_back(Op);
1580*9880d681SAndroid Build Coastguard Worker else
1581*9880d681SAndroid Build Coastguard Worker Changed = true;
1582*9880d681SAndroid Build Coastguard Worker break;
1583*9880d681SAndroid Build Coastguard Worker }
1584*9880d681SAndroid Build Coastguard Worker }
1585*9880d681SAndroid Build Coastguard Worker }
1586*9880d681SAndroid Build Coastguard Worker
1587*9880d681SAndroid Build Coastguard Worker SDValue Result;
1588*9880d681SAndroid Build Coastguard Worker
1589*9880d681SAndroid Build Coastguard Worker // If we've changed things around then replace token factor.
1590*9880d681SAndroid Build Coastguard Worker if (Changed) {
1591*9880d681SAndroid Build Coastguard Worker if (Ops.empty()) {
1592*9880d681SAndroid Build Coastguard Worker // The entry token is the only possible outcome.
1593*9880d681SAndroid Build Coastguard Worker Result = DAG.getEntryNode();
1594*9880d681SAndroid Build Coastguard Worker } else {
1595*9880d681SAndroid Build Coastguard Worker // New and improved token factor.
1596*9880d681SAndroid Build Coastguard Worker Result = DAG.getNode(ISD::TokenFactor, SDLoc(N), MVT::Other, Ops);
1597*9880d681SAndroid Build Coastguard Worker }
1598*9880d681SAndroid Build Coastguard Worker
1599*9880d681SAndroid Build Coastguard Worker // Add users to worklist if AA is enabled, since it may introduce
1600*9880d681SAndroid Build Coastguard Worker // a lot of new chained token factors while removing memory deps.
1601*9880d681SAndroid Build Coastguard Worker bool UseAA = CombinerAA.getNumOccurrences() > 0 ? CombinerAA
1602*9880d681SAndroid Build Coastguard Worker : DAG.getSubtarget().useAA();
1603*9880d681SAndroid Build Coastguard Worker return CombineTo(N, Result, UseAA /*add to worklist*/);
1604*9880d681SAndroid Build Coastguard Worker }
1605*9880d681SAndroid Build Coastguard Worker
1606*9880d681SAndroid Build Coastguard Worker return Result;
1607*9880d681SAndroid Build Coastguard Worker }
1608*9880d681SAndroid Build Coastguard Worker
1609*9880d681SAndroid Build Coastguard Worker /// MERGE_VALUES can always be eliminated.
visitMERGE_VALUES(SDNode * N)1610*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitMERGE_VALUES(SDNode *N) {
1611*9880d681SAndroid Build Coastguard Worker WorklistRemover DeadNodes(*this);
1612*9880d681SAndroid Build Coastguard Worker // Replacing results may cause a different MERGE_VALUES to suddenly
1613*9880d681SAndroid Build Coastguard Worker // be CSE'd with N, and carry its uses with it. Iterate until no
1614*9880d681SAndroid Build Coastguard Worker // uses remain, to ensure that the node can be safely deleted.
1615*9880d681SAndroid Build Coastguard Worker // First add the users of this node to the work list so that they
1616*9880d681SAndroid Build Coastguard Worker // can be tried again once they have new operands.
1617*9880d681SAndroid Build Coastguard Worker AddUsersToWorklist(N);
1618*9880d681SAndroid Build Coastguard Worker do {
1619*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i)
1620*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(N, i), N->getOperand(i));
1621*9880d681SAndroid Build Coastguard Worker } while (!N->use_empty());
1622*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(N);
1623*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
1624*9880d681SAndroid Build Coastguard Worker }
1625*9880d681SAndroid Build Coastguard Worker
1626*9880d681SAndroid Build Coastguard Worker /// If \p N is a ConstantSDNode with isOpaque() == false return it casted to a
1627*9880d681SAndroid Build Coastguard Worker /// ConstantSDNode pointer else nullptr.
getAsNonOpaqueConstant(SDValue N)1628*9880d681SAndroid Build Coastguard Worker static ConstantSDNode *getAsNonOpaqueConstant(SDValue N) {
1629*9880d681SAndroid Build Coastguard Worker ConstantSDNode *Const = dyn_cast<ConstantSDNode>(N);
1630*9880d681SAndroid Build Coastguard Worker return Const != nullptr && !Const->isOpaque() ? Const : nullptr;
1631*9880d681SAndroid Build Coastguard Worker }
1632*9880d681SAndroid Build Coastguard Worker
visitADD(SDNode * N)1633*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitADD(SDNode *N) {
1634*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
1635*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
1636*9880d681SAndroid Build Coastguard Worker EVT VT = N0.getValueType();
1637*9880d681SAndroid Build Coastguard Worker
1638*9880d681SAndroid Build Coastguard Worker // fold vector ops
1639*9880d681SAndroid Build Coastguard Worker if (VT.isVector()) {
1640*9880d681SAndroid Build Coastguard Worker if (SDValue FoldedVOp = SimplifyVBinOp(N))
1641*9880d681SAndroid Build Coastguard Worker return FoldedVOp;
1642*9880d681SAndroid Build Coastguard Worker
1643*9880d681SAndroid Build Coastguard Worker // fold (add x, 0) -> x, vector edition
1644*9880d681SAndroid Build Coastguard Worker if (ISD::isBuildVectorAllZeros(N1.getNode()))
1645*9880d681SAndroid Build Coastguard Worker return N0;
1646*9880d681SAndroid Build Coastguard Worker if (ISD::isBuildVectorAllZeros(N0.getNode()))
1647*9880d681SAndroid Build Coastguard Worker return N1;
1648*9880d681SAndroid Build Coastguard Worker }
1649*9880d681SAndroid Build Coastguard Worker
1650*9880d681SAndroid Build Coastguard Worker // fold (add x, undef) -> undef
1651*9880d681SAndroid Build Coastguard Worker if (N0.isUndef())
1652*9880d681SAndroid Build Coastguard Worker return N0;
1653*9880d681SAndroid Build Coastguard Worker if (N1.isUndef())
1654*9880d681SAndroid Build Coastguard Worker return N1;
1655*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N0)) {
1656*9880d681SAndroid Build Coastguard Worker // canonicalize constant to RHS
1657*9880d681SAndroid Build Coastguard Worker if (!DAG.isConstantIntBuildVectorOrConstantInt(N1))
1658*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ADD, SDLoc(N), VT, N1, N0);
1659*9880d681SAndroid Build Coastguard Worker // fold (add c1, c2) -> c1+c2
1660*9880d681SAndroid Build Coastguard Worker return DAG.FoldConstantArithmetic(ISD::ADD, SDLoc(N), VT,
1661*9880d681SAndroid Build Coastguard Worker N0.getNode(), N1.getNode());
1662*9880d681SAndroid Build Coastguard Worker }
1663*9880d681SAndroid Build Coastguard Worker // fold (add x, 0) -> x
1664*9880d681SAndroid Build Coastguard Worker if (isNullConstant(N1))
1665*9880d681SAndroid Build Coastguard Worker return N0;
1666*9880d681SAndroid Build Coastguard Worker // fold ((c1-A)+c2) -> (c1+c2)-A
1667*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *N1C = getAsNonOpaqueConstant(N1)) {
1668*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SUB)
1669*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *N0C = getAsNonOpaqueConstant(N0.getOperand(0))) {
1670*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
1671*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUB, DL, VT,
1672*9880d681SAndroid Build Coastguard Worker DAG.getConstant(N1C->getAPIntValue()+
1673*9880d681SAndroid Build Coastguard Worker N0C->getAPIntValue(), DL, VT),
1674*9880d681SAndroid Build Coastguard Worker N0.getOperand(1));
1675*9880d681SAndroid Build Coastguard Worker }
1676*9880d681SAndroid Build Coastguard Worker }
1677*9880d681SAndroid Build Coastguard Worker // reassociate add
1678*9880d681SAndroid Build Coastguard Worker if (SDValue RADD = ReassociateOps(ISD::ADD, SDLoc(N), N0, N1))
1679*9880d681SAndroid Build Coastguard Worker return RADD;
1680*9880d681SAndroid Build Coastguard Worker // fold ((0-A) + B) -> B-A
1681*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SUB && isNullConstant(N0.getOperand(0)))
1682*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUB, SDLoc(N), VT, N1, N0.getOperand(1));
1683*9880d681SAndroid Build Coastguard Worker // fold (A + (0-B)) -> A-B
1684*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::SUB && isNullConstant(N1.getOperand(0)))
1685*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUB, SDLoc(N), VT, N0, N1.getOperand(1));
1686*9880d681SAndroid Build Coastguard Worker // fold (A+(B-A)) -> B
1687*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::SUB && N0 == N1.getOperand(1))
1688*9880d681SAndroid Build Coastguard Worker return N1.getOperand(0);
1689*9880d681SAndroid Build Coastguard Worker // fold ((B-A)+A) -> B
1690*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SUB && N1 == N0.getOperand(1))
1691*9880d681SAndroid Build Coastguard Worker return N0.getOperand(0);
1692*9880d681SAndroid Build Coastguard Worker // fold (A+(B-(A+C))) to (B-C)
1693*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::SUB && N1.getOperand(1).getOpcode() == ISD::ADD &&
1694*9880d681SAndroid Build Coastguard Worker N0 == N1.getOperand(1).getOperand(0))
1695*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUB, SDLoc(N), VT, N1.getOperand(0),
1696*9880d681SAndroid Build Coastguard Worker N1.getOperand(1).getOperand(1));
1697*9880d681SAndroid Build Coastguard Worker // fold (A+(B-(C+A))) to (B-C)
1698*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::SUB && N1.getOperand(1).getOpcode() == ISD::ADD &&
1699*9880d681SAndroid Build Coastguard Worker N0 == N1.getOperand(1).getOperand(1))
1700*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUB, SDLoc(N), VT, N1.getOperand(0),
1701*9880d681SAndroid Build Coastguard Worker N1.getOperand(1).getOperand(0));
1702*9880d681SAndroid Build Coastguard Worker // fold (A+((B-A)+or-C)) to (B+or-C)
1703*9880d681SAndroid Build Coastguard Worker if ((N1.getOpcode() == ISD::SUB || N1.getOpcode() == ISD::ADD) &&
1704*9880d681SAndroid Build Coastguard Worker N1.getOperand(0).getOpcode() == ISD::SUB &&
1705*9880d681SAndroid Build Coastguard Worker N0 == N1.getOperand(0).getOperand(1))
1706*9880d681SAndroid Build Coastguard Worker return DAG.getNode(N1.getOpcode(), SDLoc(N), VT,
1707*9880d681SAndroid Build Coastguard Worker N1.getOperand(0).getOperand(0), N1.getOperand(1));
1708*9880d681SAndroid Build Coastguard Worker
1709*9880d681SAndroid Build Coastguard Worker // fold (A-B)+(C-D) to (A+C)-(B+D) when A or C is constant
1710*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SUB && N1.getOpcode() == ISD::SUB) {
1711*9880d681SAndroid Build Coastguard Worker SDValue N00 = N0.getOperand(0);
1712*9880d681SAndroid Build Coastguard Worker SDValue N01 = N0.getOperand(1);
1713*9880d681SAndroid Build Coastguard Worker SDValue N10 = N1.getOperand(0);
1714*9880d681SAndroid Build Coastguard Worker SDValue N11 = N1.getOperand(1);
1715*9880d681SAndroid Build Coastguard Worker
1716*9880d681SAndroid Build Coastguard Worker if (isa<ConstantSDNode>(N00) || isa<ConstantSDNode>(N10))
1717*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUB, SDLoc(N), VT,
1718*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::ADD, SDLoc(N0), VT, N00, N10),
1719*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::ADD, SDLoc(N1), VT, N01, N11));
1720*9880d681SAndroid Build Coastguard Worker }
1721*9880d681SAndroid Build Coastguard Worker
1722*9880d681SAndroid Build Coastguard Worker if (!VT.isVector() && SimplifyDemandedBits(SDValue(N, 0)))
1723*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0);
1724*9880d681SAndroid Build Coastguard Worker
1725*9880d681SAndroid Build Coastguard Worker // fold (a+b) -> (a|b) iff a and b share no bits.
1726*9880d681SAndroid Build Coastguard Worker if ((!LegalOperations || TLI.isOperationLegal(ISD::OR, VT)) &&
1727*9880d681SAndroid Build Coastguard Worker VT.isInteger() && !VT.isVector() && DAG.haveNoCommonBitsSet(N0, N1))
1728*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::OR, SDLoc(N), VT, N0, N1);
1729*9880d681SAndroid Build Coastguard Worker
1730*9880d681SAndroid Build Coastguard Worker // fold (add x, shl(0 - y, n)) -> sub(x, shl(y, n))
1731*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::SHL && N1.getOperand(0).getOpcode() == ISD::SUB &&
1732*9880d681SAndroid Build Coastguard Worker isNullConstant(N1.getOperand(0).getOperand(0)))
1733*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUB, SDLoc(N), VT, N0,
1734*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::SHL, SDLoc(N), VT,
1735*9880d681SAndroid Build Coastguard Worker N1.getOperand(0).getOperand(1),
1736*9880d681SAndroid Build Coastguard Worker N1.getOperand(1)));
1737*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SHL && N0.getOperand(0).getOpcode() == ISD::SUB &&
1738*9880d681SAndroid Build Coastguard Worker isNullConstant(N0.getOperand(0).getOperand(0)))
1739*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUB, SDLoc(N), VT, N1,
1740*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::SHL, SDLoc(N), VT,
1741*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getOperand(1),
1742*9880d681SAndroid Build Coastguard Worker N0.getOperand(1)));
1743*9880d681SAndroid Build Coastguard Worker
1744*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::AND) {
1745*9880d681SAndroid Build Coastguard Worker SDValue AndOp0 = N1.getOperand(0);
1746*9880d681SAndroid Build Coastguard Worker unsigned NumSignBits = DAG.ComputeNumSignBits(AndOp0);
1747*9880d681SAndroid Build Coastguard Worker unsigned DestBits = VT.getScalarType().getSizeInBits();
1748*9880d681SAndroid Build Coastguard Worker
1749*9880d681SAndroid Build Coastguard Worker // (add z, (and (sbbl x, x), 1)) -> (sub z, (sbbl x, x))
1750*9880d681SAndroid Build Coastguard Worker // and similar xforms where the inner op is either ~0 or 0.
1751*9880d681SAndroid Build Coastguard Worker if (NumSignBits == DestBits && isOneConstant(N1->getOperand(1))) {
1752*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
1753*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUB, DL, VT, N->getOperand(0), AndOp0);
1754*9880d681SAndroid Build Coastguard Worker }
1755*9880d681SAndroid Build Coastguard Worker }
1756*9880d681SAndroid Build Coastguard Worker
1757*9880d681SAndroid Build Coastguard Worker // add (sext i1), X -> sub X, (zext i1)
1758*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SIGN_EXTEND &&
1759*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getValueType() == MVT::i1 &&
1760*9880d681SAndroid Build Coastguard Worker !TLI.isOperationLegal(ISD::SIGN_EXTEND, MVT::i1)) {
1761*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
1762*9880d681SAndroid Build Coastguard Worker SDValue ZExt = DAG.getNode(ISD::ZERO_EXTEND, DL, VT, N0.getOperand(0));
1763*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUB, DL, VT, N1, ZExt);
1764*9880d681SAndroid Build Coastguard Worker }
1765*9880d681SAndroid Build Coastguard Worker
1766*9880d681SAndroid Build Coastguard Worker // add X, (sextinreg Y i1) -> sub X, (and Y 1)
1767*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::SIGN_EXTEND_INREG) {
1768*9880d681SAndroid Build Coastguard Worker VTSDNode *TN = cast<VTSDNode>(N1.getOperand(1));
1769*9880d681SAndroid Build Coastguard Worker if (TN->getVT() == MVT::i1) {
1770*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
1771*9880d681SAndroid Build Coastguard Worker SDValue ZExt = DAG.getNode(ISD::AND, DL, VT, N1.getOperand(0),
1772*9880d681SAndroid Build Coastguard Worker DAG.getConstant(1, DL, VT));
1773*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUB, DL, VT, N0, ZExt);
1774*9880d681SAndroid Build Coastguard Worker }
1775*9880d681SAndroid Build Coastguard Worker }
1776*9880d681SAndroid Build Coastguard Worker
1777*9880d681SAndroid Build Coastguard Worker return SDValue();
1778*9880d681SAndroid Build Coastguard Worker }
1779*9880d681SAndroid Build Coastguard Worker
visitADDC(SDNode * N)1780*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitADDC(SDNode *N) {
1781*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
1782*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
1783*9880d681SAndroid Build Coastguard Worker EVT VT = N0.getValueType();
1784*9880d681SAndroid Build Coastguard Worker
1785*9880d681SAndroid Build Coastguard Worker // If the flag result is dead, turn this into an ADD.
1786*9880d681SAndroid Build Coastguard Worker if (!N->hasAnyUseOfValue(1))
1787*9880d681SAndroid Build Coastguard Worker return CombineTo(N, DAG.getNode(ISD::ADD, SDLoc(N), VT, N0, N1),
1788*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::CARRY_FALSE,
1789*9880d681SAndroid Build Coastguard Worker SDLoc(N), MVT::Glue));
1790*9880d681SAndroid Build Coastguard Worker
1791*9880d681SAndroid Build Coastguard Worker // canonicalize constant to RHS.
1792*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
1793*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
1794*9880d681SAndroid Build Coastguard Worker if (N0C && !N1C)
1795*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ADDC, SDLoc(N), N->getVTList(), N1, N0);
1796*9880d681SAndroid Build Coastguard Worker
1797*9880d681SAndroid Build Coastguard Worker // fold (addc x, 0) -> x + no carry out
1798*9880d681SAndroid Build Coastguard Worker if (isNullConstant(N1))
1799*9880d681SAndroid Build Coastguard Worker return CombineTo(N, N0, DAG.getNode(ISD::CARRY_FALSE,
1800*9880d681SAndroid Build Coastguard Worker SDLoc(N), MVT::Glue));
1801*9880d681SAndroid Build Coastguard Worker
1802*9880d681SAndroid Build Coastguard Worker // fold (addc a, b) -> (or a, b), CARRY_FALSE iff a and b share no bits.
1803*9880d681SAndroid Build Coastguard Worker APInt LHSZero, LHSOne;
1804*9880d681SAndroid Build Coastguard Worker APInt RHSZero, RHSOne;
1805*9880d681SAndroid Build Coastguard Worker DAG.computeKnownBits(N0, LHSZero, LHSOne);
1806*9880d681SAndroid Build Coastguard Worker
1807*9880d681SAndroid Build Coastguard Worker if (LHSZero.getBoolValue()) {
1808*9880d681SAndroid Build Coastguard Worker DAG.computeKnownBits(N1, RHSZero, RHSOne);
1809*9880d681SAndroid Build Coastguard Worker
1810*9880d681SAndroid Build Coastguard Worker // If all possibly-set bits on the LHS are clear on the RHS, return an OR.
1811*9880d681SAndroid Build Coastguard Worker // If all possibly-set bits on the RHS are clear on the LHS, return an OR.
1812*9880d681SAndroid Build Coastguard Worker if ((RHSZero & ~LHSZero) == ~LHSZero || (LHSZero & ~RHSZero) == ~RHSZero)
1813*9880d681SAndroid Build Coastguard Worker return CombineTo(N, DAG.getNode(ISD::OR, SDLoc(N), VT, N0, N1),
1814*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::CARRY_FALSE,
1815*9880d681SAndroid Build Coastguard Worker SDLoc(N), MVT::Glue));
1816*9880d681SAndroid Build Coastguard Worker }
1817*9880d681SAndroid Build Coastguard Worker
1818*9880d681SAndroid Build Coastguard Worker return SDValue();
1819*9880d681SAndroid Build Coastguard Worker }
1820*9880d681SAndroid Build Coastguard Worker
visitADDE(SDNode * N)1821*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitADDE(SDNode *N) {
1822*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
1823*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
1824*9880d681SAndroid Build Coastguard Worker SDValue CarryIn = N->getOperand(2);
1825*9880d681SAndroid Build Coastguard Worker
1826*9880d681SAndroid Build Coastguard Worker // canonicalize constant to RHS
1827*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N0C = dyn_cast<ConstantSDNode>(N0);
1828*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
1829*9880d681SAndroid Build Coastguard Worker if (N0C && !N1C)
1830*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ADDE, SDLoc(N), N->getVTList(),
1831*9880d681SAndroid Build Coastguard Worker N1, N0, CarryIn);
1832*9880d681SAndroid Build Coastguard Worker
1833*9880d681SAndroid Build Coastguard Worker // fold (adde x, y, false) -> (addc x, y)
1834*9880d681SAndroid Build Coastguard Worker if (CarryIn.getOpcode() == ISD::CARRY_FALSE)
1835*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ADDC, SDLoc(N), N->getVTList(), N0, N1);
1836*9880d681SAndroid Build Coastguard Worker
1837*9880d681SAndroid Build Coastguard Worker return SDValue();
1838*9880d681SAndroid Build Coastguard Worker }
1839*9880d681SAndroid Build Coastguard Worker
1840*9880d681SAndroid Build Coastguard Worker // Since it may not be valid to emit a fold to zero for vector initializers
1841*9880d681SAndroid Build Coastguard Worker // check if we can before folding.
tryFoldToZero(const SDLoc & DL,const TargetLowering & TLI,EVT VT,SelectionDAG & DAG,bool LegalOperations,bool LegalTypes)1842*9880d681SAndroid Build Coastguard Worker static SDValue tryFoldToZero(const SDLoc &DL, const TargetLowering &TLI, EVT VT,
1843*9880d681SAndroid Build Coastguard Worker SelectionDAG &DAG, bool LegalOperations,
1844*9880d681SAndroid Build Coastguard Worker bool LegalTypes) {
1845*9880d681SAndroid Build Coastguard Worker if (!VT.isVector())
1846*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, DL, VT);
1847*9880d681SAndroid Build Coastguard Worker if (!LegalOperations || TLI.isOperationLegal(ISD::BUILD_VECTOR, VT))
1848*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, DL, VT);
1849*9880d681SAndroid Build Coastguard Worker return SDValue();
1850*9880d681SAndroid Build Coastguard Worker }
1851*9880d681SAndroid Build Coastguard Worker
visitSUB(SDNode * N)1852*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSUB(SDNode *N) {
1853*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
1854*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
1855*9880d681SAndroid Build Coastguard Worker EVT VT = N0.getValueType();
1856*9880d681SAndroid Build Coastguard Worker
1857*9880d681SAndroid Build Coastguard Worker // fold vector ops
1858*9880d681SAndroid Build Coastguard Worker if (VT.isVector()) {
1859*9880d681SAndroid Build Coastguard Worker if (SDValue FoldedVOp = SimplifyVBinOp(N))
1860*9880d681SAndroid Build Coastguard Worker return FoldedVOp;
1861*9880d681SAndroid Build Coastguard Worker
1862*9880d681SAndroid Build Coastguard Worker // fold (sub x, 0) -> x, vector edition
1863*9880d681SAndroid Build Coastguard Worker if (ISD::isBuildVectorAllZeros(N1.getNode()))
1864*9880d681SAndroid Build Coastguard Worker return N0;
1865*9880d681SAndroid Build Coastguard Worker }
1866*9880d681SAndroid Build Coastguard Worker
1867*9880d681SAndroid Build Coastguard Worker // fold (sub x, x) -> 0
1868*9880d681SAndroid Build Coastguard Worker // FIXME: Refactor this and xor and other similar operations together.
1869*9880d681SAndroid Build Coastguard Worker if (N0 == N1)
1870*9880d681SAndroid Build Coastguard Worker return tryFoldToZero(SDLoc(N), TLI, VT, DAG, LegalOperations, LegalTypes);
1871*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N0) &&
1872*9880d681SAndroid Build Coastguard Worker DAG.isConstantIntBuildVectorOrConstantInt(N1)) {
1873*9880d681SAndroid Build Coastguard Worker // fold (sub c1, c2) -> c1-c2
1874*9880d681SAndroid Build Coastguard Worker return DAG.FoldConstantArithmetic(ISD::SUB, SDLoc(N), VT,
1875*9880d681SAndroid Build Coastguard Worker N0.getNode(), N1.getNode());
1876*9880d681SAndroid Build Coastguard Worker }
1877*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N0C = getAsNonOpaqueConstant(N0);
1878*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N1C = getAsNonOpaqueConstant(N1);
1879*9880d681SAndroid Build Coastguard Worker // fold (sub x, c) -> (add x, -c)
1880*9880d681SAndroid Build Coastguard Worker if (N1C) {
1881*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
1882*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ADD, DL, VT, N0,
1883*9880d681SAndroid Build Coastguard Worker DAG.getConstant(-N1C->getAPIntValue(), DL, VT));
1884*9880d681SAndroid Build Coastguard Worker }
1885*9880d681SAndroid Build Coastguard Worker // Canonicalize (sub -1, x) -> ~x, i.e. (xor x, -1)
1886*9880d681SAndroid Build Coastguard Worker if (isAllOnesConstant(N0))
1887*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::XOR, SDLoc(N), VT, N1, N0);
1888*9880d681SAndroid Build Coastguard Worker // fold A-(A-B) -> B
1889*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::SUB && N0 == N1.getOperand(0))
1890*9880d681SAndroid Build Coastguard Worker return N1.getOperand(1);
1891*9880d681SAndroid Build Coastguard Worker // fold (A+B)-A -> B
1892*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::ADD && N0.getOperand(0) == N1)
1893*9880d681SAndroid Build Coastguard Worker return N0.getOperand(1);
1894*9880d681SAndroid Build Coastguard Worker // fold (A+B)-B -> A
1895*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::ADD && N0.getOperand(1) == N1)
1896*9880d681SAndroid Build Coastguard Worker return N0.getOperand(0);
1897*9880d681SAndroid Build Coastguard Worker // fold C2-(A+C1) -> (C2-C1)-A
1898*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N1C1 = N1.getOpcode() != ISD::ADD ? nullptr :
1899*9880d681SAndroid Build Coastguard Worker dyn_cast<ConstantSDNode>(N1.getOperand(1).getNode());
1900*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::ADD && N0C && N1C1) {
1901*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
1902*9880d681SAndroid Build Coastguard Worker SDValue NewC = DAG.getConstant(N0C->getAPIntValue() - N1C1->getAPIntValue(),
1903*9880d681SAndroid Build Coastguard Worker DL, VT);
1904*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUB, DL, VT, NewC,
1905*9880d681SAndroid Build Coastguard Worker N1.getOperand(0));
1906*9880d681SAndroid Build Coastguard Worker }
1907*9880d681SAndroid Build Coastguard Worker // fold ((A+(B+or-C))-B) -> A+or-C
1908*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::ADD &&
1909*9880d681SAndroid Build Coastguard Worker (N0.getOperand(1).getOpcode() == ISD::SUB ||
1910*9880d681SAndroid Build Coastguard Worker N0.getOperand(1).getOpcode() == ISD::ADD) &&
1911*9880d681SAndroid Build Coastguard Worker N0.getOperand(1).getOperand(0) == N1)
1912*9880d681SAndroid Build Coastguard Worker return DAG.getNode(N0.getOperand(1).getOpcode(), SDLoc(N), VT,
1913*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N0.getOperand(1).getOperand(1));
1914*9880d681SAndroid Build Coastguard Worker // fold ((A+(C+B))-B) -> A+C
1915*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::ADD &&
1916*9880d681SAndroid Build Coastguard Worker N0.getOperand(1).getOpcode() == ISD::ADD &&
1917*9880d681SAndroid Build Coastguard Worker N0.getOperand(1).getOperand(1) == N1)
1918*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ADD, SDLoc(N), VT,
1919*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N0.getOperand(1).getOperand(0));
1920*9880d681SAndroid Build Coastguard Worker // fold ((A-(B-C))-C) -> A-B
1921*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SUB &&
1922*9880d681SAndroid Build Coastguard Worker N0.getOperand(1).getOpcode() == ISD::SUB &&
1923*9880d681SAndroid Build Coastguard Worker N0.getOperand(1).getOperand(1) == N1)
1924*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUB, SDLoc(N), VT,
1925*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N0.getOperand(1).getOperand(0));
1926*9880d681SAndroid Build Coastguard Worker
1927*9880d681SAndroid Build Coastguard Worker // If either operand of a sub is undef, the result is undef
1928*9880d681SAndroid Build Coastguard Worker if (N0.isUndef())
1929*9880d681SAndroid Build Coastguard Worker return N0;
1930*9880d681SAndroid Build Coastguard Worker if (N1.isUndef())
1931*9880d681SAndroid Build Coastguard Worker return N1;
1932*9880d681SAndroid Build Coastguard Worker
1933*9880d681SAndroid Build Coastguard Worker // If the relocation model supports it, consider symbol offsets.
1934*9880d681SAndroid Build Coastguard Worker if (GlobalAddressSDNode *GA = dyn_cast<GlobalAddressSDNode>(N0))
1935*9880d681SAndroid Build Coastguard Worker if (!LegalOperations && TLI.isOffsetFoldingLegal(GA)) {
1936*9880d681SAndroid Build Coastguard Worker // fold (sub Sym, c) -> Sym-c
1937*9880d681SAndroid Build Coastguard Worker if (N1C && GA->getOpcode() == ISD::GlobalAddress)
1938*9880d681SAndroid Build Coastguard Worker return DAG.getGlobalAddress(GA->getGlobal(), SDLoc(N1C), VT,
1939*9880d681SAndroid Build Coastguard Worker GA->getOffset() -
1940*9880d681SAndroid Build Coastguard Worker (uint64_t)N1C->getSExtValue());
1941*9880d681SAndroid Build Coastguard Worker // fold (sub Sym+c1, Sym+c2) -> c1-c2
1942*9880d681SAndroid Build Coastguard Worker if (GlobalAddressSDNode *GB = dyn_cast<GlobalAddressSDNode>(N1))
1943*9880d681SAndroid Build Coastguard Worker if (GA->getGlobal() == GB->getGlobal())
1944*9880d681SAndroid Build Coastguard Worker return DAG.getConstant((uint64_t)GA->getOffset() - GB->getOffset(),
1945*9880d681SAndroid Build Coastguard Worker SDLoc(N), VT);
1946*9880d681SAndroid Build Coastguard Worker }
1947*9880d681SAndroid Build Coastguard Worker
1948*9880d681SAndroid Build Coastguard Worker // sub X, (sextinreg Y i1) -> add X, (and Y 1)
1949*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::SIGN_EXTEND_INREG) {
1950*9880d681SAndroid Build Coastguard Worker VTSDNode *TN = cast<VTSDNode>(N1.getOperand(1));
1951*9880d681SAndroid Build Coastguard Worker if (TN->getVT() == MVT::i1) {
1952*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
1953*9880d681SAndroid Build Coastguard Worker SDValue ZExt = DAG.getNode(ISD::AND, DL, VT, N1.getOperand(0),
1954*9880d681SAndroid Build Coastguard Worker DAG.getConstant(1, DL, VT));
1955*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ADD, DL, VT, N0, ZExt);
1956*9880d681SAndroid Build Coastguard Worker }
1957*9880d681SAndroid Build Coastguard Worker }
1958*9880d681SAndroid Build Coastguard Worker
1959*9880d681SAndroid Build Coastguard Worker return SDValue();
1960*9880d681SAndroid Build Coastguard Worker }
1961*9880d681SAndroid Build Coastguard Worker
visitSUBC(SDNode * N)1962*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSUBC(SDNode *N) {
1963*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
1964*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
1965*9880d681SAndroid Build Coastguard Worker EVT VT = N0.getValueType();
1966*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
1967*9880d681SAndroid Build Coastguard Worker
1968*9880d681SAndroid Build Coastguard Worker // If the flag result is dead, turn this into an SUB.
1969*9880d681SAndroid Build Coastguard Worker if (!N->hasAnyUseOfValue(1))
1970*9880d681SAndroid Build Coastguard Worker return CombineTo(N, DAG.getNode(ISD::SUB, DL, VT, N0, N1),
1971*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::CARRY_FALSE, DL, MVT::Glue));
1972*9880d681SAndroid Build Coastguard Worker
1973*9880d681SAndroid Build Coastguard Worker // fold (subc x, x) -> 0 + no borrow
1974*9880d681SAndroid Build Coastguard Worker if (N0 == N1)
1975*9880d681SAndroid Build Coastguard Worker return CombineTo(N, DAG.getConstant(0, DL, VT),
1976*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::CARRY_FALSE, DL, MVT::Glue));
1977*9880d681SAndroid Build Coastguard Worker
1978*9880d681SAndroid Build Coastguard Worker // fold (subc x, 0) -> x + no borrow
1979*9880d681SAndroid Build Coastguard Worker if (isNullConstant(N1))
1980*9880d681SAndroid Build Coastguard Worker return CombineTo(N, N0, DAG.getNode(ISD::CARRY_FALSE, DL, MVT::Glue));
1981*9880d681SAndroid Build Coastguard Worker
1982*9880d681SAndroid Build Coastguard Worker // Canonicalize (sub -1, x) -> ~x, i.e. (xor x, -1) + no borrow
1983*9880d681SAndroid Build Coastguard Worker if (isAllOnesConstant(N0))
1984*9880d681SAndroid Build Coastguard Worker return CombineTo(N, DAG.getNode(ISD::XOR, DL, VT, N1, N0),
1985*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::CARRY_FALSE, DL, MVT::Glue));
1986*9880d681SAndroid Build Coastguard Worker
1987*9880d681SAndroid Build Coastguard Worker return SDValue();
1988*9880d681SAndroid Build Coastguard Worker }
1989*9880d681SAndroid Build Coastguard Worker
visitSUBE(SDNode * N)1990*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSUBE(SDNode *N) {
1991*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
1992*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
1993*9880d681SAndroid Build Coastguard Worker SDValue CarryIn = N->getOperand(2);
1994*9880d681SAndroid Build Coastguard Worker
1995*9880d681SAndroid Build Coastguard Worker // fold (sube x, y, false) -> (subc x, y)
1996*9880d681SAndroid Build Coastguard Worker if (CarryIn.getOpcode() == ISD::CARRY_FALSE)
1997*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUBC, SDLoc(N), N->getVTList(), N0, N1);
1998*9880d681SAndroid Build Coastguard Worker
1999*9880d681SAndroid Build Coastguard Worker return SDValue();
2000*9880d681SAndroid Build Coastguard Worker }
2001*9880d681SAndroid Build Coastguard Worker
visitMUL(SDNode * N)2002*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitMUL(SDNode *N) {
2003*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
2004*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
2005*9880d681SAndroid Build Coastguard Worker EVT VT = N0.getValueType();
2006*9880d681SAndroid Build Coastguard Worker
2007*9880d681SAndroid Build Coastguard Worker // fold (mul x, undef) -> 0
2008*9880d681SAndroid Build Coastguard Worker if (N0.isUndef() || N1.isUndef())
2009*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, SDLoc(N), VT);
2010*9880d681SAndroid Build Coastguard Worker
2011*9880d681SAndroid Build Coastguard Worker bool N0IsConst = false;
2012*9880d681SAndroid Build Coastguard Worker bool N1IsConst = false;
2013*9880d681SAndroid Build Coastguard Worker bool N1IsOpaqueConst = false;
2014*9880d681SAndroid Build Coastguard Worker bool N0IsOpaqueConst = false;
2015*9880d681SAndroid Build Coastguard Worker APInt ConstValue0, ConstValue1;
2016*9880d681SAndroid Build Coastguard Worker // fold vector ops
2017*9880d681SAndroid Build Coastguard Worker if (VT.isVector()) {
2018*9880d681SAndroid Build Coastguard Worker if (SDValue FoldedVOp = SimplifyVBinOp(N))
2019*9880d681SAndroid Build Coastguard Worker return FoldedVOp;
2020*9880d681SAndroid Build Coastguard Worker
2021*9880d681SAndroid Build Coastguard Worker N0IsConst = ISD::isConstantSplatVector(N0.getNode(), ConstValue0);
2022*9880d681SAndroid Build Coastguard Worker N1IsConst = ISD::isConstantSplatVector(N1.getNode(), ConstValue1);
2023*9880d681SAndroid Build Coastguard Worker } else {
2024*9880d681SAndroid Build Coastguard Worker N0IsConst = isa<ConstantSDNode>(N0);
2025*9880d681SAndroid Build Coastguard Worker if (N0IsConst) {
2026*9880d681SAndroid Build Coastguard Worker ConstValue0 = cast<ConstantSDNode>(N0)->getAPIntValue();
2027*9880d681SAndroid Build Coastguard Worker N0IsOpaqueConst = cast<ConstantSDNode>(N0)->isOpaque();
2028*9880d681SAndroid Build Coastguard Worker }
2029*9880d681SAndroid Build Coastguard Worker N1IsConst = isa<ConstantSDNode>(N1);
2030*9880d681SAndroid Build Coastguard Worker if (N1IsConst) {
2031*9880d681SAndroid Build Coastguard Worker ConstValue1 = cast<ConstantSDNode>(N1)->getAPIntValue();
2032*9880d681SAndroid Build Coastguard Worker N1IsOpaqueConst = cast<ConstantSDNode>(N1)->isOpaque();
2033*9880d681SAndroid Build Coastguard Worker }
2034*9880d681SAndroid Build Coastguard Worker }
2035*9880d681SAndroid Build Coastguard Worker
2036*9880d681SAndroid Build Coastguard Worker // fold (mul c1, c2) -> c1*c2
2037*9880d681SAndroid Build Coastguard Worker if (N0IsConst && N1IsConst && !N0IsOpaqueConst && !N1IsOpaqueConst)
2038*9880d681SAndroid Build Coastguard Worker return DAG.FoldConstantArithmetic(ISD::MUL, SDLoc(N), VT,
2039*9880d681SAndroid Build Coastguard Worker N0.getNode(), N1.getNode());
2040*9880d681SAndroid Build Coastguard Worker
2041*9880d681SAndroid Build Coastguard Worker // canonicalize constant to RHS (vector doesn't have to splat)
2042*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N0) &&
2043*9880d681SAndroid Build Coastguard Worker !DAG.isConstantIntBuildVectorOrConstantInt(N1))
2044*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::MUL, SDLoc(N), VT, N1, N0);
2045*9880d681SAndroid Build Coastguard Worker // fold (mul x, 0) -> 0
2046*9880d681SAndroid Build Coastguard Worker if (N1IsConst && ConstValue1 == 0)
2047*9880d681SAndroid Build Coastguard Worker return N1;
2048*9880d681SAndroid Build Coastguard Worker // We require a splat of the entire scalar bit width for non-contiguous
2049*9880d681SAndroid Build Coastguard Worker // bit patterns.
2050*9880d681SAndroid Build Coastguard Worker bool IsFullSplat =
2051*9880d681SAndroid Build Coastguard Worker ConstValue1.getBitWidth() == VT.getScalarType().getSizeInBits();
2052*9880d681SAndroid Build Coastguard Worker // fold (mul x, 1) -> x
2053*9880d681SAndroid Build Coastguard Worker if (N1IsConst && ConstValue1 == 1 && IsFullSplat)
2054*9880d681SAndroid Build Coastguard Worker return N0;
2055*9880d681SAndroid Build Coastguard Worker // fold (mul x, -1) -> 0-x
2056*9880d681SAndroid Build Coastguard Worker if (N1IsConst && ConstValue1.isAllOnesValue()) {
2057*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
2058*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUB, DL, VT,
2059*9880d681SAndroid Build Coastguard Worker DAG.getConstant(0, DL, VT), N0);
2060*9880d681SAndroid Build Coastguard Worker }
2061*9880d681SAndroid Build Coastguard Worker // fold (mul x, (1 << c)) -> x << c
2062*9880d681SAndroid Build Coastguard Worker if (N1IsConst && !N1IsOpaqueConst && ConstValue1.isPowerOf2() &&
2063*9880d681SAndroid Build Coastguard Worker IsFullSplat) {
2064*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
2065*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SHL, DL, VT, N0,
2066*9880d681SAndroid Build Coastguard Worker DAG.getConstant(ConstValue1.logBase2(), DL,
2067*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(N0.getValueType())));
2068*9880d681SAndroid Build Coastguard Worker }
2069*9880d681SAndroid Build Coastguard Worker // fold (mul x, -(1 << c)) -> -(x << c) or (-x) << c
2070*9880d681SAndroid Build Coastguard Worker if (N1IsConst && !N1IsOpaqueConst && (-ConstValue1).isPowerOf2() &&
2071*9880d681SAndroid Build Coastguard Worker IsFullSplat) {
2072*9880d681SAndroid Build Coastguard Worker unsigned Log2Val = (-ConstValue1).logBase2();
2073*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
2074*9880d681SAndroid Build Coastguard Worker // FIXME: If the input is something that is easily negated (e.g. a
2075*9880d681SAndroid Build Coastguard Worker // single-use add), we should put the negate there.
2076*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUB, DL, VT,
2077*9880d681SAndroid Build Coastguard Worker DAG.getConstant(0, DL, VT),
2078*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::SHL, DL, VT, N0,
2079*9880d681SAndroid Build Coastguard Worker DAG.getConstant(Log2Val, DL,
2080*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(N0.getValueType()))));
2081*9880d681SAndroid Build Coastguard Worker }
2082*9880d681SAndroid Build Coastguard Worker
2083*9880d681SAndroid Build Coastguard Worker APInt Val;
2084*9880d681SAndroid Build Coastguard Worker // (mul (shl X, c1), c2) -> (mul X, c2 << c1)
2085*9880d681SAndroid Build Coastguard Worker if (N1IsConst && N0.getOpcode() == ISD::SHL &&
2086*9880d681SAndroid Build Coastguard Worker (ISD::isConstantSplatVector(N0.getOperand(1).getNode(), Val) ||
2087*9880d681SAndroid Build Coastguard Worker isa<ConstantSDNode>(N0.getOperand(1)))) {
2088*9880d681SAndroid Build Coastguard Worker SDValue C3 = DAG.getNode(ISD::SHL, SDLoc(N), VT, N1, N0.getOperand(1));
2089*9880d681SAndroid Build Coastguard Worker AddToWorklist(C3.getNode());
2090*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::MUL, SDLoc(N), VT, N0.getOperand(0), C3);
2091*9880d681SAndroid Build Coastguard Worker }
2092*9880d681SAndroid Build Coastguard Worker
2093*9880d681SAndroid Build Coastguard Worker // Change (mul (shl X, C), Y) -> (shl (mul X, Y), C) when the shift has one
2094*9880d681SAndroid Build Coastguard Worker // use.
2095*9880d681SAndroid Build Coastguard Worker {
2096*9880d681SAndroid Build Coastguard Worker SDValue Sh(nullptr, 0), Y(nullptr, 0);
2097*9880d681SAndroid Build Coastguard Worker // Check for both (mul (shl X, C), Y) and (mul Y, (shl X, C)).
2098*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SHL &&
2099*9880d681SAndroid Build Coastguard Worker (ISD::isConstantSplatVector(N0.getOperand(1).getNode(), Val) ||
2100*9880d681SAndroid Build Coastguard Worker isa<ConstantSDNode>(N0.getOperand(1))) &&
2101*9880d681SAndroid Build Coastguard Worker N0.getNode()->hasOneUse()) {
2102*9880d681SAndroid Build Coastguard Worker Sh = N0; Y = N1;
2103*9880d681SAndroid Build Coastguard Worker } else if (N1.getOpcode() == ISD::SHL &&
2104*9880d681SAndroid Build Coastguard Worker isa<ConstantSDNode>(N1.getOperand(1)) &&
2105*9880d681SAndroid Build Coastguard Worker N1.getNode()->hasOneUse()) {
2106*9880d681SAndroid Build Coastguard Worker Sh = N1; Y = N0;
2107*9880d681SAndroid Build Coastguard Worker }
2108*9880d681SAndroid Build Coastguard Worker
2109*9880d681SAndroid Build Coastguard Worker if (Sh.getNode()) {
2110*9880d681SAndroid Build Coastguard Worker SDValue Mul = DAG.getNode(ISD::MUL, SDLoc(N), VT, Sh.getOperand(0), Y);
2111*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SHL, SDLoc(N), VT, Mul, Sh.getOperand(1));
2112*9880d681SAndroid Build Coastguard Worker }
2113*9880d681SAndroid Build Coastguard Worker }
2114*9880d681SAndroid Build Coastguard Worker
2115*9880d681SAndroid Build Coastguard Worker // fold (mul (add x, c1), c2) -> (add (mul x, c2), c1*c2)
2116*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N1) &&
2117*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::ADD &&
2118*9880d681SAndroid Build Coastguard Worker DAG.isConstantIntBuildVectorOrConstantInt(N0.getOperand(1)) &&
2119*9880d681SAndroid Build Coastguard Worker isMulAddWithConstProfitable(N, N0, N1))
2120*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ADD, SDLoc(N), VT,
2121*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::MUL, SDLoc(N0), VT,
2122*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N1),
2123*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::MUL, SDLoc(N1), VT,
2124*9880d681SAndroid Build Coastguard Worker N0.getOperand(1), N1));
2125*9880d681SAndroid Build Coastguard Worker
2126*9880d681SAndroid Build Coastguard Worker // reassociate mul
2127*9880d681SAndroid Build Coastguard Worker if (SDValue RMUL = ReassociateOps(ISD::MUL, SDLoc(N), N0, N1))
2128*9880d681SAndroid Build Coastguard Worker return RMUL;
2129*9880d681SAndroid Build Coastguard Worker
2130*9880d681SAndroid Build Coastguard Worker return SDValue();
2131*9880d681SAndroid Build Coastguard Worker }
2132*9880d681SAndroid Build Coastguard Worker
2133*9880d681SAndroid Build Coastguard Worker /// Return true if divmod libcall is available.
isDivRemLibcallAvailable(SDNode * Node,bool isSigned,const TargetLowering & TLI)2134*9880d681SAndroid Build Coastguard Worker static bool isDivRemLibcallAvailable(SDNode *Node, bool isSigned,
2135*9880d681SAndroid Build Coastguard Worker const TargetLowering &TLI) {
2136*9880d681SAndroid Build Coastguard Worker RTLIB::Libcall LC;
2137*9880d681SAndroid Build Coastguard Worker EVT NodeType = Node->getValueType(0);
2138*9880d681SAndroid Build Coastguard Worker if (!NodeType.isSimple())
2139*9880d681SAndroid Build Coastguard Worker return false;
2140*9880d681SAndroid Build Coastguard Worker switch (NodeType.getSimpleVT().SimpleTy) {
2141*9880d681SAndroid Build Coastguard Worker default: return false; // No libcall for vector types.
2142*9880d681SAndroid Build Coastguard Worker case MVT::i8: LC= isSigned ? RTLIB::SDIVREM_I8 : RTLIB::UDIVREM_I8; break;
2143*9880d681SAndroid Build Coastguard Worker case MVT::i16: LC= isSigned ? RTLIB::SDIVREM_I16 : RTLIB::UDIVREM_I16; break;
2144*9880d681SAndroid Build Coastguard Worker case MVT::i32: LC= isSigned ? RTLIB::SDIVREM_I32 : RTLIB::UDIVREM_I32; break;
2145*9880d681SAndroid Build Coastguard Worker case MVT::i64: LC= isSigned ? RTLIB::SDIVREM_I64 : RTLIB::UDIVREM_I64; break;
2146*9880d681SAndroid Build Coastguard Worker case MVT::i128: LC= isSigned ? RTLIB::SDIVREM_I128:RTLIB::UDIVREM_I128; break;
2147*9880d681SAndroid Build Coastguard Worker }
2148*9880d681SAndroid Build Coastguard Worker
2149*9880d681SAndroid Build Coastguard Worker return TLI.getLibcallName(LC) != nullptr;
2150*9880d681SAndroid Build Coastguard Worker }
2151*9880d681SAndroid Build Coastguard Worker
2152*9880d681SAndroid Build Coastguard Worker /// Issue divrem if both quotient and remainder are needed.
useDivRem(SDNode * Node)2153*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::useDivRem(SDNode *Node) {
2154*9880d681SAndroid Build Coastguard Worker if (Node->use_empty())
2155*9880d681SAndroid Build Coastguard Worker return SDValue(); // This is a dead node, leave it alone.
2156*9880d681SAndroid Build Coastguard Worker
2157*9880d681SAndroid Build Coastguard Worker unsigned Opcode = Node->getOpcode();
2158*9880d681SAndroid Build Coastguard Worker bool isSigned = (Opcode == ISD::SDIV) || (Opcode == ISD::SREM);
2159*9880d681SAndroid Build Coastguard Worker unsigned DivRemOpc = isSigned ? ISD::SDIVREM : ISD::UDIVREM;
2160*9880d681SAndroid Build Coastguard Worker
2161*9880d681SAndroid Build Coastguard Worker // DivMod lib calls can still work on non-legal types if using lib-calls.
2162*9880d681SAndroid Build Coastguard Worker EVT VT = Node->getValueType(0);
2163*9880d681SAndroid Build Coastguard Worker if (VT.isVector() || !VT.isInteger())
2164*9880d681SAndroid Build Coastguard Worker return SDValue();
2165*9880d681SAndroid Build Coastguard Worker
2166*9880d681SAndroid Build Coastguard Worker if (!TLI.isTypeLegal(VT) && !TLI.isOperationCustom(DivRemOpc, VT))
2167*9880d681SAndroid Build Coastguard Worker return SDValue();
2168*9880d681SAndroid Build Coastguard Worker
2169*9880d681SAndroid Build Coastguard Worker // If DIVREM is going to get expanded into a libcall,
2170*9880d681SAndroid Build Coastguard Worker // but there is no libcall available, then don't combine.
2171*9880d681SAndroid Build Coastguard Worker if (!TLI.isOperationLegalOrCustom(DivRemOpc, VT) &&
2172*9880d681SAndroid Build Coastguard Worker !isDivRemLibcallAvailable(Node, isSigned, TLI))
2173*9880d681SAndroid Build Coastguard Worker return SDValue();
2174*9880d681SAndroid Build Coastguard Worker
2175*9880d681SAndroid Build Coastguard Worker // If div is legal, it's better to do the normal expansion
2176*9880d681SAndroid Build Coastguard Worker unsigned OtherOpcode = 0;
2177*9880d681SAndroid Build Coastguard Worker if ((Opcode == ISD::SDIV) || (Opcode == ISD::UDIV)) {
2178*9880d681SAndroid Build Coastguard Worker OtherOpcode = isSigned ? ISD::SREM : ISD::UREM;
2179*9880d681SAndroid Build Coastguard Worker if (TLI.isOperationLegalOrCustom(Opcode, VT))
2180*9880d681SAndroid Build Coastguard Worker return SDValue();
2181*9880d681SAndroid Build Coastguard Worker } else {
2182*9880d681SAndroid Build Coastguard Worker OtherOpcode = isSigned ? ISD::SDIV : ISD::UDIV;
2183*9880d681SAndroid Build Coastguard Worker if (TLI.isOperationLegalOrCustom(OtherOpcode, VT))
2184*9880d681SAndroid Build Coastguard Worker return SDValue();
2185*9880d681SAndroid Build Coastguard Worker }
2186*9880d681SAndroid Build Coastguard Worker
2187*9880d681SAndroid Build Coastguard Worker SDValue Op0 = Node->getOperand(0);
2188*9880d681SAndroid Build Coastguard Worker SDValue Op1 = Node->getOperand(1);
2189*9880d681SAndroid Build Coastguard Worker SDValue combined;
2190*9880d681SAndroid Build Coastguard Worker for (SDNode::use_iterator UI = Op0.getNode()->use_begin(),
2191*9880d681SAndroid Build Coastguard Worker UE = Op0.getNode()->use_end(); UI != UE; ++UI) {
2192*9880d681SAndroid Build Coastguard Worker SDNode *User = *UI;
2193*9880d681SAndroid Build Coastguard Worker if (User == Node || User->use_empty())
2194*9880d681SAndroid Build Coastguard Worker continue;
2195*9880d681SAndroid Build Coastguard Worker // Convert the other matching node(s), too;
2196*9880d681SAndroid Build Coastguard Worker // otherwise, the DIVREM may get target-legalized into something
2197*9880d681SAndroid Build Coastguard Worker // target-specific that we won't be able to recognize.
2198*9880d681SAndroid Build Coastguard Worker unsigned UserOpc = User->getOpcode();
2199*9880d681SAndroid Build Coastguard Worker if ((UserOpc == Opcode || UserOpc == OtherOpcode || UserOpc == DivRemOpc) &&
2200*9880d681SAndroid Build Coastguard Worker User->getOperand(0) == Op0 &&
2201*9880d681SAndroid Build Coastguard Worker User->getOperand(1) == Op1) {
2202*9880d681SAndroid Build Coastguard Worker if (!combined) {
2203*9880d681SAndroid Build Coastguard Worker if (UserOpc == OtherOpcode) {
2204*9880d681SAndroid Build Coastguard Worker SDVTList VTs = DAG.getVTList(VT, VT);
2205*9880d681SAndroid Build Coastguard Worker combined = DAG.getNode(DivRemOpc, SDLoc(Node), VTs, Op0, Op1);
2206*9880d681SAndroid Build Coastguard Worker } else if (UserOpc == DivRemOpc) {
2207*9880d681SAndroid Build Coastguard Worker combined = SDValue(User, 0);
2208*9880d681SAndroid Build Coastguard Worker } else {
2209*9880d681SAndroid Build Coastguard Worker assert(UserOpc == Opcode);
2210*9880d681SAndroid Build Coastguard Worker continue;
2211*9880d681SAndroid Build Coastguard Worker }
2212*9880d681SAndroid Build Coastguard Worker }
2213*9880d681SAndroid Build Coastguard Worker if (UserOpc == ISD::SDIV || UserOpc == ISD::UDIV)
2214*9880d681SAndroid Build Coastguard Worker CombineTo(User, combined);
2215*9880d681SAndroid Build Coastguard Worker else if (UserOpc == ISD::SREM || UserOpc == ISD::UREM)
2216*9880d681SAndroid Build Coastguard Worker CombineTo(User, combined.getValue(1));
2217*9880d681SAndroid Build Coastguard Worker }
2218*9880d681SAndroid Build Coastguard Worker }
2219*9880d681SAndroid Build Coastguard Worker return combined;
2220*9880d681SAndroid Build Coastguard Worker }
2221*9880d681SAndroid Build Coastguard Worker
visitSDIV(SDNode * N)2222*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSDIV(SDNode *N) {
2223*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
2224*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
2225*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
2226*9880d681SAndroid Build Coastguard Worker
2227*9880d681SAndroid Build Coastguard Worker // fold vector ops
2228*9880d681SAndroid Build Coastguard Worker if (VT.isVector())
2229*9880d681SAndroid Build Coastguard Worker if (SDValue FoldedVOp = SimplifyVBinOp(N))
2230*9880d681SAndroid Build Coastguard Worker return FoldedVOp;
2231*9880d681SAndroid Build Coastguard Worker
2232*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
2233*9880d681SAndroid Build Coastguard Worker
2234*9880d681SAndroid Build Coastguard Worker // fold (sdiv c1, c2) -> c1/c2
2235*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N0C = isConstOrConstSplat(N0);
2236*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N1C = isConstOrConstSplat(N1);
2237*9880d681SAndroid Build Coastguard Worker if (N0C && N1C && !N0C->isOpaque() && !N1C->isOpaque())
2238*9880d681SAndroid Build Coastguard Worker return DAG.FoldConstantArithmetic(ISD::SDIV, DL, VT, N0C, N1C);
2239*9880d681SAndroid Build Coastguard Worker // fold (sdiv X, 1) -> X
2240*9880d681SAndroid Build Coastguard Worker if (N1C && N1C->isOne())
2241*9880d681SAndroid Build Coastguard Worker return N0;
2242*9880d681SAndroid Build Coastguard Worker // fold (sdiv X, -1) -> 0-X
2243*9880d681SAndroid Build Coastguard Worker if (N1C && N1C->isAllOnesValue())
2244*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUB, DL, VT,
2245*9880d681SAndroid Build Coastguard Worker DAG.getConstant(0, DL, VT), N0);
2246*9880d681SAndroid Build Coastguard Worker
2247*9880d681SAndroid Build Coastguard Worker // If we know the sign bits of both operands are zero, strength reduce to a
2248*9880d681SAndroid Build Coastguard Worker // udiv instead. Handles (X&15) /s 4 -> X&15 >> 2
2249*9880d681SAndroid Build Coastguard Worker if (!VT.isVector()) {
2250*9880d681SAndroid Build Coastguard Worker if (DAG.SignBitIsZero(N1) && DAG.SignBitIsZero(N0))
2251*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::UDIV, DL, N1.getValueType(), N0, N1);
2252*9880d681SAndroid Build Coastguard Worker }
2253*9880d681SAndroid Build Coastguard Worker
2254*9880d681SAndroid Build Coastguard Worker // fold (sdiv X, pow2) -> simple ops after legalize
2255*9880d681SAndroid Build Coastguard Worker // FIXME: We check for the exact bit here because the generic lowering gives
2256*9880d681SAndroid Build Coastguard Worker // better results in that case. The target-specific lowering should learn how
2257*9880d681SAndroid Build Coastguard Worker // to handle exact sdivs efficiently.
2258*9880d681SAndroid Build Coastguard Worker if (N1C && !N1C->isNullValue() && !N1C->isOpaque() &&
2259*9880d681SAndroid Build Coastguard Worker !cast<BinaryWithFlagsSDNode>(N)->Flags.hasExact() &&
2260*9880d681SAndroid Build Coastguard Worker (N1C->getAPIntValue().isPowerOf2() ||
2261*9880d681SAndroid Build Coastguard Worker (-N1C->getAPIntValue()).isPowerOf2())) {
2262*9880d681SAndroid Build Coastguard Worker // Target-specific implementation of sdiv x, pow2.
2263*9880d681SAndroid Build Coastguard Worker if (SDValue Res = BuildSDIVPow2(N))
2264*9880d681SAndroid Build Coastguard Worker return Res;
2265*9880d681SAndroid Build Coastguard Worker
2266*9880d681SAndroid Build Coastguard Worker unsigned lg2 = N1C->getAPIntValue().countTrailingZeros();
2267*9880d681SAndroid Build Coastguard Worker
2268*9880d681SAndroid Build Coastguard Worker // Splat the sign bit into the register
2269*9880d681SAndroid Build Coastguard Worker SDValue SGN =
2270*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::SRA, DL, VT, N0,
2271*9880d681SAndroid Build Coastguard Worker DAG.getConstant(VT.getScalarSizeInBits() - 1, DL,
2272*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(N0.getValueType())));
2273*9880d681SAndroid Build Coastguard Worker AddToWorklist(SGN.getNode());
2274*9880d681SAndroid Build Coastguard Worker
2275*9880d681SAndroid Build Coastguard Worker // Add (N0 < 0) ? abs2 - 1 : 0;
2276*9880d681SAndroid Build Coastguard Worker SDValue SRL =
2277*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::SRL, DL, VT, SGN,
2278*9880d681SAndroid Build Coastguard Worker DAG.getConstant(VT.getScalarSizeInBits() - lg2, DL,
2279*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(SGN.getValueType())));
2280*9880d681SAndroid Build Coastguard Worker SDValue ADD = DAG.getNode(ISD::ADD, DL, VT, N0, SRL);
2281*9880d681SAndroid Build Coastguard Worker AddToWorklist(SRL.getNode());
2282*9880d681SAndroid Build Coastguard Worker AddToWorklist(ADD.getNode()); // Divide by pow2
2283*9880d681SAndroid Build Coastguard Worker SDValue SRA = DAG.getNode(ISD::SRA, DL, VT, ADD,
2284*9880d681SAndroid Build Coastguard Worker DAG.getConstant(lg2, DL,
2285*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(ADD.getValueType())));
2286*9880d681SAndroid Build Coastguard Worker
2287*9880d681SAndroid Build Coastguard Worker // If we're dividing by a positive value, we're done. Otherwise, we must
2288*9880d681SAndroid Build Coastguard Worker // negate the result.
2289*9880d681SAndroid Build Coastguard Worker if (N1C->getAPIntValue().isNonNegative())
2290*9880d681SAndroid Build Coastguard Worker return SRA;
2291*9880d681SAndroid Build Coastguard Worker
2292*9880d681SAndroid Build Coastguard Worker AddToWorklist(SRA.getNode());
2293*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SUB, DL, VT, DAG.getConstant(0, DL, VT), SRA);
2294*9880d681SAndroid Build Coastguard Worker }
2295*9880d681SAndroid Build Coastguard Worker
2296*9880d681SAndroid Build Coastguard Worker // If integer divide is expensive and we satisfy the requirements, emit an
2297*9880d681SAndroid Build Coastguard Worker // alternate sequence. Targets may check function attributes for size/speed
2298*9880d681SAndroid Build Coastguard Worker // trade-offs.
2299*9880d681SAndroid Build Coastguard Worker AttributeSet Attr = DAG.getMachineFunction().getFunction()->getAttributes();
2300*9880d681SAndroid Build Coastguard Worker if (N1C && !TLI.isIntDivCheap(N->getValueType(0), Attr))
2301*9880d681SAndroid Build Coastguard Worker if (SDValue Op = BuildSDIV(N))
2302*9880d681SAndroid Build Coastguard Worker return Op;
2303*9880d681SAndroid Build Coastguard Worker
2304*9880d681SAndroid Build Coastguard Worker // sdiv, srem -> sdivrem
2305*9880d681SAndroid Build Coastguard Worker // If the divisor is constant, then return DIVREM only if isIntDivCheap() is true.
2306*9880d681SAndroid Build Coastguard Worker // Otherwise, we break the simplification logic in visitREM().
2307*9880d681SAndroid Build Coastguard Worker if (!N1C || TLI.isIntDivCheap(N->getValueType(0), Attr))
2308*9880d681SAndroid Build Coastguard Worker if (SDValue DivRem = useDivRem(N))
2309*9880d681SAndroid Build Coastguard Worker return DivRem;
2310*9880d681SAndroid Build Coastguard Worker
2311*9880d681SAndroid Build Coastguard Worker // undef / X -> 0
2312*9880d681SAndroid Build Coastguard Worker if (N0.isUndef())
2313*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, DL, VT);
2314*9880d681SAndroid Build Coastguard Worker // X / undef -> undef
2315*9880d681SAndroid Build Coastguard Worker if (N1.isUndef())
2316*9880d681SAndroid Build Coastguard Worker return N1;
2317*9880d681SAndroid Build Coastguard Worker
2318*9880d681SAndroid Build Coastguard Worker return SDValue();
2319*9880d681SAndroid Build Coastguard Worker }
2320*9880d681SAndroid Build Coastguard Worker
visitUDIV(SDNode * N)2321*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitUDIV(SDNode *N) {
2322*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
2323*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
2324*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
2325*9880d681SAndroid Build Coastguard Worker
2326*9880d681SAndroid Build Coastguard Worker // fold vector ops
2327*9880d681SAndroid Build Coastguard Worker if (VT.isVector())
2328*9880d681SAndroid Build Coastguard Worker if (SDValue FoldedVOp = SimplifyVBinOp(N))
2329*9880d681SAndroid Build Coastguard Worker return FoldedVOp;
2330*9880d681SAndroid Build Coastguard Worker
2331*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
2332*9880d681SAndroid Build Coastguard Worker
2333*9880d681SAndroid Build Coastguard Worker // fold (udiv c1, c2) -> c1/c2
2334*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N0C = isConstOrConstSplat(N0);
2335*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N1C = isConstOrConstSplat(N1);
2336*9880d681SAndroid Build Coastguard Worker if (N0C && N1C)
2337*9880d681SAndroid Build Coastguard Worker if (SDValue Folded = DAG.FoldConstantArithmetic(ISD::UDIV, DL, VT,
2338*9880d681SAndroid Build Coastguard Worker N0C, N1C))
2339*9880d681SAndroid Build Coastguard Worker return Folded;
2340*9880d681SAndroid Build Coastguard Worker // fold (udiv x, (1 << c)) -> x >>u c
2341*9880d681SAndroid Build Coastguard Worker if (N1C && !N1C->isOpaque() && N1C->getAPIntValue().isPowerOf2())
2342*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SRL, DL, VT, N0,
2343*9880d681SAndroid Build Coastguard Worker DAG.getConstant(N1C->getAPIntValue().logBase2(), DL,
2344*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(N0.getValueType())));
2345*9880d681SAndroid Build Coastguard Worker
2346*9880d681SAndroid Build Coastguard Worker // fold (udiv x, (shl c, y)) -> x >>u (log2(c)+y) iff c is power of 2
2347*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::SHL) {
2348*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *SHC = getAsNonOpaqueConstant(N1.getOperand(0))) {
2349*9880d681SAndroid Build Coastguard Worker if (SHC->getAPIntValue().isPowerOf2()) {
2350*9880d681SAndroid Build Coastguard Worker EVT ADDVT = N1.getOperand(1).getValueType();
2351*9880d681SAndroid Build Coastguard Worker SDValue Add = DAG.getNode(ISD::ADD, DL, ADDVT,
2352*9880d681SAndroid Build Coastguard Worker N1.getOperand(1),
2353*9880d681SAndroid Build Coastguard Worker DAG.getConstant(SHC->getAPIntValue()
2354*9880d681SAndroid Build Coastguard Worker .logBase2(),
2355*9880d681SAndroid Build Coastguard Worker DL, ADDVT));
2356*9880d681SAndroid Build Coastguard Worker AddToWorklist(Add.getNode());
2357*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SRL, DL, VT, N0, Add);
2358*9880d681SAndroid Build Coastguard Worker }
2359*9880d681SAndroid Build Coastguard Worker }
2360*9880d681SAndroid Build Coastguard Worker }
2361*9880d681SAndroid Build Coastguard Worker
2362*9880d681SAndroid Build Coastguard Worker // fold (udiv x, c) -> alternate
2363*9880d681SAndroid Build Coastguard Worker AttributeSet Attr = DAG.getMachineFunction().getFunction()->getAttributes();
2364*9880d681SAndroid Build Coastguard Worker if (N1C && !TLI.isIntDivCheap(N->getValueType(0), Attr))
2365*9880d681SAndroid Build Coastguard Worker if (SDValue Op = BuildUDIV(N))
2366*9880d681SAndroid Build Coastguard Worker return Op;
2367*9880d681SAndroid Build Coastguard Worker
2368*9880d681SAndroid Build Coastguard Worker // sdiv, srem -> sdivrem
2369*9880d681SAndroid Build Coastguard Worker // If the divisor is constant, then return DIVREM only if isIntDivCheap() is true.
2370*9880d681SAndroid Build Coastguard Worker // Otherwise, we break the simplification logic in visitREM().
2371*9880d681SAndroid Build Coastguard Worker if (!N1C || TLI.isIntDivCheap(N->getValueType(0), Attr))
2372*9880d681SAndroid Build Coastguard Worker if (SDValue DivRem = useDivRem(N))
2373*9880d681SAndroid Build Coastguard Worker return DivRem;
2374*9880d681SAndroid Build Coastguard Worker
2375*9880d681SAndroid Build Coastguard Worker // undef / X -> 0
2376*9880d681SAndroid Build Coastguard Worker if (N0.isUndef())
2377*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, DL, VT);
2378*9880d681SAndroid Build Coastguard Worker // X / undef -> undef
2379*9880d681SAndroid Build Coastguard Worker if (N1.isUndef())
2380*9880d681SAndroid Build Coastguard Worker return N1;
2381*9880d681SAndroid Build Coastguard Worker
2382*9880d681SAndroid Build Coastguard Worker return SDValue();
2383*9880d681SAndroid Build Coastguard Worker }
2384*9880d681SAndroid Build Coastguard Worker
2385*9880d681SAndroid Build Coastguard Worker // handles ISD::SREM and ISD::UREM
visitREM(SDNode * N)2386*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitREM(SDNode *N) {
2387*9880d681SAndroid Build Coastguard Worker unsigned Opcode = N->getOpcode();
2388*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
2389*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
2390*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
2391*9880d681SAndroid Build Coastguard Worker bool isSigned = (Opcode == ISD::SREM);
2392*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
2393*9880d681SAndroid Build Coastguard Worker
2394*9880d681SAndroid Build Coastguard Worker // fold (rem c1, c2) -> c1%c2
2395*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N0C = isConstOrConstSplat(N0);
2396*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N1C = isConstOrConstSplat(N1);
2397*9880d681SAndroid Build Coastguard Worker if (N0C && N1C)
2398*9880d681SAndroid Build Coastguard Worker if (SDValue Folded = DAG.FoldConstantArithmetic(Opcode, DL, VT, N0C, N1C))
2399*9880d681SAndroid Build Coastguard Worker return Folded;
2400*9880d681SAndroid Build Coastguard Worker
2401*9880d681SAndroid Build Coastguard Worker if (isSigned) {
2402*9880d681SAndroid Build Coastguard Worker // If we know the sign bits of both operands are zero, strength reduce to a
2403*9880d681SAndroid Build Coastguard Worker // urem instead. Handles (X & 0x0FFFFFFF) %s 16 -> X&15
2404*9880d681SAndroid Build Coastguard Worker if (!VT.isVector()) {
2405*9880d681SAndroid Build Coastguard Worker if (DAG.SignBitIsZero(N1) && DAG.SignBitIsZero(N0))
2406*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::UREM, DL, VT, N0, N1);
2407*9880d681SAndroid Build Coastguard Worker }
2408*9880d681SAndroid Build Coastguard Worker } else {
2409*9880d681SAndroid Build Coastguard Worker // fold (urem x, pow2) -> (and x, pow2-1)
2410*9880d681SAndroid Build Coastguard Worker if (N1C && !N1C->isNullValue() && !N1C->isOpaque() &&
2411*9880d681SAndroid Build Coastguard Worker N1C->getAPIntValue().isPowerOf2()) {
2412*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, DL, VT, N0,
2413*9880d681SAndroid Build Coastguard Worker DAG.getConstant(N1C->getAPIntValue() - 1, DL, VT));
2414*9880d681SAndroid Build Coastguard Worker }
2415*9880d681SAndroid Build Coastguard Worker // fold (urem x, (shl pow2, y)) -> (and x, (add (shl pow2, y), -1))
2416*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::SHL) {
2417*9880d681SAndroid Build Coastguard Worker ConstantSDNode *SHC = getAsNonOpaqueConstant(N1.getOperand(0));
2418*9880d681SAndroid Build Coastguard Worker if (SHC && SHC->getAPIntValue().isPowerOf2()) {
2419*9880d681SAndroid Build Coastguard Worker APInt NegOne = APInt::getAllOnesValue(VT.getSizeInBits());
2420*9880d681SAndroid Build Coastguard Worker SDValue Add =
2421*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::ADD, DL, VT, N1, DAG.getConstant(NegOne, DL, VT));
2422*9880d681SAndroid Build Coastguard Worker AddToWorklist(Add.getNode());
2423*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, DL, VT, N0, Add);
2424*9880d681SAndroid Build Coastguard Worker }
2425*9880d681SAndroid Build Coastguard Worker }
2426*9880d681SAndroid Build Coastguard Worker }
2427*9880d681SAndroid Build Coastguard Worker
2428*9880d681SAndroid Build Coastguard Worker AttributeSet Attr = DAG.getMachineFunction().getFunction()->getAttributes();
2429*9880d681SAndroid Build Coastguard Worker
2430*9880d681SAndroid Build Coastguard Worker // If X/C can be simplified by the division-by-constant logic, lower
2431*9880d681SAndroid Build Coastguard Worker // X%C to the equivalent of X-X/C*C.
2432*9880d681SAndroid Build Coastguard Worker // To avoid mangling nodes, this simplification requires that the combine()
2433*9880d681SAndroid Build Coastguard Worker // call for the speculative DIV must not cause a DIVREM conversion. We guard
2434*9880d681SAndroid Build Coastguard Worker // against this by skipping the simplification if isIntDivCheap(). When
2435*9880d681SAndroid Build Coastguard Worker // div is not cheap, combine will not return a DIVREM. Regardless,
2436*9880d681SAndroid Build Coastguard Worker // checking cheapness here makes sense since the simplification results in
2437*9880d681SAndroid Build Coastguard Worker // fatter code.
2438*9880d681SAndroid Build Coastguard Worker if (N1C && !N1C->isNullValue() && !TLI.isIntDivCheap(VT, Attr)) {
2439*9880d681SAndroid Build Coastguard Worker unsigned DivOpcode = isSigned ? ISD::SDIV : ISD::UDIV;
2440*9880d681SAndroid Build Coastguard Worker SDValue Div = DAG.getNode(DivOpcode, DL, VT, N0, N1);
2441*9880d681SAndroid Build Coastguard Worker AddToWorklist(Div.getNode());
2442*9880d681SAndroid Build Coastguard Worker SDValue OptimizedDiv = combine(Div.getNode());
2443*9880d681SAndroid Build Coastguard Worker if (OptimizedDiv.getNode() && OptimizedDiv.getNode() != Div.getNode()) {
2444*9880d681SAndroid Build Coastguard Worker assert((OptimizedDiv.getOpcode() != ISD::UDIVREM) &&
2445*9880d681SAndroid Build Coastguard Worker (OptimizedDiv.getOpcode() != ISD::SDIVREM));
2446*9880d681SAndroid Build Coastguard Worker SDValue Mul = DAG.getNode(ISD::MUL, DL, VT, OptimizedDiv, N1);
2447*9880d681SAndroid Build Coastguard Worker SDValue Sub = DAG.getNode(ISD::SUB, DL, VT, N0, Mul);
2448*9880d681SAndroid Build Coastguard Worker AddToWorklist(Mul.getNode());
2449*9880d681SAndroid Build Coastguard Worker return Sub;
2450*9880d681SAndroid Build Coastguard Worker }
2451*9880d681SAndroid Build Coastguard Worker }
2452*9880d681SAndroid Build Coastguard Worker
2453*9880d681SAndroid Build Coastguard Worker // sdiv, srem -> sdivrem
2454*9880d681SAndroid Build Coastguard Worker if (SDValue DivRem = useDivRem(N))
2455*9880d681SAndroid Build Coastguard Worker return DivRem.getValue(1);
2456*9880d681SAndroid Build Coastguard Worker
2457*9880d681SAndroid Build Coastguard Worker // undef % X -> 0
2458*9880d681SAndroid Build Coastguard Worker if (N0.isUndef())
2459*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, DL, VT);
2460*9880d681SAndroid Build Coastguard Worker // X % undef -> undef
2461*9880d681SAndroid Build Coastguard Worker if (N1.isUndef())
2462*9880d681SAndroid Build Coastguard Worker return N1;
2463*9880d681SAndroid Build Coastguard Worker
2464*9880d681SAndroid Build Coastguard Worker return SDValue();
2465*9880d681SAndroid Build Coastguard Worker }
2466*9880d681SAndroid Build Coastguard Worker
visitMULHS(SDNode * N)2467*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitMULHS(SDNode *N) {
2468*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
2469*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
2470*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
2471*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
2472*9880d681SAndroid Build Coastguard Worker
2473*9880d681SAndroid Build Coastguard Worker // fold (mulhs x, 0) -> 0
2474*9880d681SAndroid Build Coastguard Worker if (isNullConstant(N1))
2475*9880d681SAndroid Build Coastguard Worker return N1;
2476*9880d681SAndroid Build Coastguard Worker // fold (mulhs x, 1) -> (sra x, size(x)-1)
2477*9880d681SAndroid Build Coastguard Worker if (isOneConstant(N1)) {
2478*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
2479*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SRA, DL, N0.getValueType(), N0,
2480*9880d681SAndroid Build Coastguard Worker DAG.getConstant(N0.getValueType().getSizeInBits() - 1,
2481*9880d681SAndroid Build Coastguard Worker DL,
2482*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(N0.getValueType())));
2483*9880d681SAndroid Build Coastguard Worker }
2484*9880d681SAndroid Build Coastguard Worker // fold (mulhs x, undef) -> 0
2485*9880d681SAndroid Build Coastguard Worker if (N0.isUndef() || N1.isUndef())
2486*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, SDLoc(N), VT);
2487*9880d681SAndroid Build Coastguard Worker
2488*9880d681SAndroid Build Coastguard Worker // If the type twice as wide is legal, transform the mulhs to a wider multiply
2489*9880d681SAndroid Build Coastguard Worker // plus a shift.
2490*9880d681SAndroid Build Coastguard Worker if (VT.isSimple() && !VT.isVector()) {
2491*9880d681SAndroid Build Coastguard Worker MVT Simple = VT.getSimpleVT();
2492*9880d681SAndroid Build Coastguard Worker unsigned SimpleSize = Simple.getSizeInBits();
2493*9880d681SAndroid Build Coastguard Worker EVT NewVT = EVT::getIntegerVT(*DAG.getContext(), SimpleSize*2);
2494*9880d681SAndroid Build Coastguard Worker if (TLI.isOperationLegal(ISD::MUL, NewVT)) {
2495*9880d681SAndroid Build Coastguard Worker N0 = DAG.getNode(ISD::SIGN_EXTEND, DL, NewVT, N0);
2496*9880d681SAndroid Build Coastguard Worker N1 = DAG.getNode(ISD::SIGN_EXTEND, DL, NewVT, N1);
2497*9880d681SAndroid Build Coastguard Worker N1 = DAG.getNode(ISD::MUL, DL, NewVT, N0, N1);
2498*9880d681SAndroid Build Coastguard Worker N1 = DAG.getNode(ISD::SRL, DL, NewVT, N1,
2499*9880d681SAndroid Build Coastguard Worker DAG.getConstant(SimpleSize, DL,
2500*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(N1.getValueType())));
2501*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TRUNCATE, DL, VT, N1);
2502*9880d681SAndroid Build Coastguard Worker }
2503*9880d681SAndroid Build Coastguard Worker }
2504*9880d681SAndroid Build Coastguard Worker
2505*9880d681SAndroid Build Coastguard Worker return SDValue();
2506*9880d681SAndroid Build Coastguard Worker }
2507*9880d681SAndroid Build Coastguard Worker
visitMULHU(SDNode * N)2508*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitMULHU(SDNode *N) {
2509*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
2510*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
2511*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
2512*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
2513*9880d681SAndroid Build Coastguard Worker
2514*9880d681SAndroid Build Coastguard Worker // fold (mulhu x, 0) -> 0
2515*9880d681SAndroid Build Coastguard Worker if (isNullConstant(N1))
2516*9880d681SAndroid Build Coastguard Worker return N1;
2517*9880d681SAndroid Build Coastguard Worker // fold (mulhu x, 1) -> 0
2518*9880d681SAndroid Build Coastguard Worker if (isOneConstant(N1))
2519*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, DL, N0.getValueType());
2520*9880d681SAndroid Build Coastguard Worker // fold (mulhu x, undef) -> 0
2521*9880d681SAndroid Build Coastguard Worker if (N0.isUndef() || N1.isUndef())
2522*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, DL, VT);
2523*9880d681SAndroid Build Coastguard Worker
2524*9880d681SAndroid Build Coastguard Worker // If the type twice as wide is legal, transform the mulhu to a wider multiply
2525*9880d681SAndroid Build Coastguard Worker // plus a shift.
2526*9880d681SAndroid Build Coastguard Worker if (VT.isSimple() && !VT.isVector()) {
2527*9880d681SAndroid Build Coastguard Worker MVT Simple = VT.getSimpleVT();
2528*9880d681SAndroid Build Coastguard Worker unsigned SimpleSize = Simple.getSizeInBits();
2529*9880d681SAndroid Build Coastguard Worker EVT NewVT = EVT::getIntegerVT(*DAG.getContext(), SimpleSize*2);
2530*9880d681SAndroid Build Coastguard Worker if (TLI.isOperationLegal(ISD::MUL, NewVT)) {
2531*9880d681SAndroid Build Coastguard Worker N0 = DAG.getNode(ISD::ZERO_EXTEND, DL, NewVT, N0);
2532*9880d681SAndroid Build Coastguard Worker N1 = DAG.getNode(ISD::ZERO_EXTEND, DL, NewVT, N1);
2533*9880d681SAndroid Build Coastguard Worker N1 = DAG.getNode(ISD::MUL, DL, NewVT, N0, N1);
2534*9880d681SAndroid Build Coastguard Worker N1 = DAG.getNode(ISD::SRL, DL, NewVT, N1,
2535*9880d681SAndroid Build Coastguard Worker DAG.getConstant(SimpleSize, DL,
2536*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(N1.getValueType())));
2537*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TRUNCATE, DL, VT, N1);
2538*9880d681SAndroid Build Coastguard Worker }
2539*9880d681SAndroid Build Coastguard Worker }
2540*9880d681SAndroid Build Coastguard Worker
2541*9880d681SAndroid Build Coastguard Worker return SDValue();
2542*9880d681SAndroid Build Coastguard Worker }
2543*9880d681SAndroid Build Coastguard Worker
2544*9880d681SAndroid Build Coastguard Worker /// Perform optimizations common to nodes that compute two values. LoOp and HiOp
2545*9880d681SAndroid Build Coastguard Worker /// give the opcodes for the two computations that are being performed. Return
2546*9880d681SAndroid Build Coastguard Worker /// true if a simplification was made.
SimplifyNodeWithTwoResults(SDNode * N,unsigned LoOp,unsigned HiOp)2547*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::SimplifyNodeWithTwoResults(SDNode *N, unsigned LoOp,
2548*9880d681SAndroid Build Coastguard Worker unsigned HiOp) {
2549*9880d681SAndroid Build Coastguard Worker // If the high half is not needed, just compute the low half.
2550*9880d681SAndroid Build Coastguard Worker bool HiExists = N->hasAnyUseOfValue(1);
2551*9880d681SAndroid Build Coastguard Worker if (!HiExists &&
2552*9880d681SAndroid Build Coastguard Worker (!LegalOperations ||
2553*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegalOrCustom(LoOp, N->getValueType(0)))) {
2554*9880d681SAndroid Build Coastguard Worker SDValue Res = DAG.getNode(LoOp, SDLoc(N), N->getValueType(0), N->ops());
2555*9880d681SAndroid Build Coastguard Worker return CombineTo(N, Res, Res);
2556*9880d681SAndroid Build Coastguard Worker }
2557*9880d681SAndroid Build Coastguard Worker
2558*9880d681SAndroid Build Coastguard Worker // If the low half is not needed, just compute the high half.
2559*9880d681SAndroid Build Coastguard Worker bool LoExists = N->hasAnyUseOfValue(0);
2560*9880d681SAndroid Build Coastguard Worker if (!LoExists &&
2561*9880d681SAndroid Build Coastguard Worker (!LegalOperations ||
2562*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegal(HiOp, N->getValueType(1)))) {
2563*9880d681SAndroid Build Coastguard Worker SDValue Res = DAG.getNode(HiOp, SDLoc(N), N->getValueType(1), N->ops());
2564*9880d681SAndroid Build Coastguard Worker return CombineTo(N, Res, Res);
2565*9880d681SAndroid Build Coastguard Worker }
2566*9880d681SAndroid Build Coastguard Worker
2567*9880d681SAndroid Build Coastguard Worker // If both halves are used, return as it is.
2568*9880d681SAndroid Build Coastguard Worker if (LoExists && HiExists)
2569*9880d681SAndroid Build Coastguard Worker return SDValue();
2570*9880d681SAndroid Build Coastguard Worker
2571*9880d681SAndroid Build Coastguard Worker // If the two computed results can be simplified separately, separate them.
2572*9880d681SAndroid Build Coastguard Worker if (LoExists) {
2573*9880d681SAndroid Build Coastguard Worker SDValue Lo = DAG.getNode(LoOp, SDLoc(N), N->getValueType(0), N->ops());
2574*9880d681SAndroid Build Coastguard Worker AddToWorklist(Lo.getNode());
2575*9880d681SAndroid Build Coastguard Worker SDValue LoOpt = combine(Lo.getNode());
2576*9880d681SAndroid Build Coastguard Worker if (LoOpt.getNode() && LoOpt.getNode() != Lo.getNode() &&
2577*9880d681SAndroid Build Coastguard Worker (!LegalOperations ||
2578*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegal(LoOpt.getOpcode(), LoOpt.getValueType())))
2579*9880d681SAndroid Build Coastguard Worker return CombineTo(N, LoOpt, LoOpt);
2580*9880d681SAndroid Build Coastguard Worker }
2581*9880d681SAndroid Build Coastguard Worker
2582*9880d681SAndroid Build Coastguard Worker if (HiExists) {
2583*9880d681SAndroid Build Coastguard Worker SDValue Hi = DAG.getNode(HiOp, SDLoc(N), N->getValueType(1), N->ops());
2584*9880d681SAndroid Build Coastguard Worker AddToWorklist(Hi.getNode());
2585*9880d681SAndroid Build Coastguard Worker SDValue HiOpt = combine(Hi.getNode());
2586*9880d681SAndroid Build Coastguard Worker if (HiOpt.getNode() && HiOpt != Hi &&
2587*9880d681SAndroid Build Coastguard Worker (!LegalOperations ||
2588*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegal(HiOpt.getOpcode(), HiOpt.getValueType())))
2589*9880d681SAndroid Build Coastguard Worker return CombineTo(N, HiOpt, HiOpt);
2590*9880d681SAndroid Build Coastguard Worker }
2591*9880d681SAndroid Build Coastguard Worker
2592*9880d681SAndroid Build Coastguard Worker return SDValue();
2593*9880d681SAndroid Build Coastguard Worker }
2594*9880d681SAndroid Build Coastguard Worker
visitSMUL_LOHI(SDNode * N)2595*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSMUL_LOHI(SDNode *N) {
2596*9880d681SAndroid Build Coastguard Worker if (SDValue Res = SimplifyNodeWithTwoResults(N, ISD::MUL, ISD::MULHS))
2597*9880d681SAndroid Build Coastguard Worker return Res;
2598*9880d681SAndroid Build Coastguard Worker
2599*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
2600*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
2601*9880d681SAndroid Build Coastguard Worker
2602*9880d681SAndroid Build Coastguard Worker // If the type is twice as wide is legal, transform the mulhu to a wider
2603*9880d681SAndroid Build Coastguard Worker // multiply plus a shift.
2604*9880d681SAndroid Build Coastguard Worker if (VT.isSimple() && !VT.isVector()) {
2605*9880d681SAndroid Build Coastguard Worker MVT Simple = VT.getSimpleVT();
2606*9880d681SAndroid Build Coastguard Worker unsigned SimpleSize = Simple.getSizeInBits();
2607*9880d681SAndroid Build Coastguard Worker EVT NewVT = EVT::getIntegerVT(*DAG.getContext(), SimpleSize*2);
2608*9880d681SAndroid Build Coastguard Worker if (TLI.isOperationLegal(ISD::MUL, NewVT)) {
2609*9880d681SAndroid Build Coastguard Worker SDValue Lo = DAG.getNode(ISD::SIGN_EXTEND, DL, NewVT, N->getOperand(0));
2610*9880d681SAndroid Build Coastguard Worker SDValue Hi = DAG.getNode(ISD::SIGN_EXTEND, DL, NewVT, N->getOperand(1));
2611*9880d681SAndroid Build Coastguard Worker Lo = DAG.getNode(ISD::MUL, DL, NewVT, Lo, Hi);
2612*9880d681SAndroid Build Coastguard Worker // Compute the high part as N1.
2613*9880d681SAndroid Build Coastguard Worker Hi = DAG.getNode(ISD::SRL, DL, NewVT, Lo,
2614*9880d681SAndroid Build Coastguard Worker DAG.getConstant(SimpleSize, DL,
2615*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(Lo.getValueType())));
2616*9880d681SAndroid Build Coastguard Worker Hi = DAG.getNode(ISD::TRUNCATE, DL, VT, Hi);
2617*9880d681SAndroid Build Coastguard Worker // Compute the low part as N0.
2618*9880d681SAndroid Build Coastguard Worker Lo = DAG.getNode(ISD::TRUNCATE, DL, VT, Lo);
2619*9880d681SAndroid Build Coastguard Worker return CombineTo(N, Lo, Hi);
2620*9880d681SAndroid Build Coastguard Worker }
2621*9880d681SAndroid Build Coastguard Worker }
2622*9880d681SAndroid Build Coastguard Worker
2623*9880d681SAndroid Build Coastguard Worker return SDValue();
2624*9880d681SAndroid Build Coastguard Worker }
2625*9880d681SAndroid Build Coastguard Worker
visitUMUL_LOHI(SDNode * N)2626*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitUMUL_LOHI(SDNode *N) {
2627*9880d681SAndroid Build Coastguard Worker if (SDValue Res = SimplifyNodeWithTwoResults(N, ISD::MUL, ISD::MULHU))
2628*9880d681SAndroid Build Coastguard Worker return Res;
2629*9880d681SAndroid Build Coastguard Worker
2630*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
2631*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
2632*9880d681SAndroid Build Coastguard Worker
2633*9880d681SAndroid Build Coastguard Worker // If the type is twice as wide is legal, transform the mulhu to a wider
2634*9880d681SAndroid Build Coastguard Worker // multiply plus a shift.
2635*9880d681SAndroid Build Coastguard Worker if (VT.isSimple() && !VT.isVector()) {
2636*9880d681SAndroid Build Coastguard Worker MVT Simple = VT.getSimpleVT();
2637*9880d681SAndroid Build Coastguard Worker unsigned SimpleSize = Simple.getSizeInBits();
2638*9880d681SAndroid Build Coastguard Worker EVT NewVT = EVT::getIntegerVT(*DAG.getContext(), SimpleSize*2);
2639*9880d681SAndroid Build Coastguard Worker if (TLI.isOperationLegal(ISD::MUL, NewVT)) {
2640*9880d681SAndroid Build Coastguard Worker SDValue Lo = DAG.getNode(ISD::ZERO_EXTEND, DL, NewVT, N->getOperand(0));
2641*9880d681SAndroid Build Coastguard Worker SDValue Hi = DAG.getNode(ISD::ZERO_EXTEND, DL, NewVT, N->getOperand(1));
2642*9880d681SAndroid Build Coastguard Worker Lo = DAG.getNode(ISD::MUL, DL, NewVT, Lo, Hi);
2643*9880d681SAndroid Build Coastguard Worker // Compute the high part as N1.
2644*9880d681SAndroid Build Coastguard Worker Hi = DAG.getNode(ISD::SRL, DL, NewVT, Lo,
2645*9880d681SAndroid Build Coastguard Worker DAG.getConstant(SimpleSize, DL,
2646*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(Lo.getValueType())));
2647*9880d681SAndroid Build Coastguard Worker Hi = DAG.getNode(ISD::TRUNCATE, DL, VT, Hi);
2648*9880d681SAndroid Build Coastguard Worker // Compute the low part as N0.
2649*9880d681SAndroid Build Coastguard Worker Lo = DAG.getNode(ISD::TRUNCATE, DL, VT, Lo);
2650*9880d681SAndroid Build Coastguard Worker return CombineTo(N, Lo, Hi);
2651*9880d681SAndroid Build Coastguard Worker }
2652*9880d681SAndroid Build Coastguard Worker }
2653*9880d681SAndroid Build Coastguard Worker
2654*9880d681SAndroid Build Coastguard Worker return SDValue();
2655*9880d681SAndroid Build Coastguard Worker }
2656*9880d681SAndroid Build Coastguard Worker
visitSMULO(SDNode * N)2657*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSMULO(SDNode *N) {
2658*9880d681SAndroid Build Coastguard Worker // (smulo x, 2) -> (saddo x, x)
2659*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *C2 = dyn_cast<ConstantSDNode>(N->getOperand(1)))
2660*9880d681SAndroid Build Coastguard Worker if (C2->getAPIntValue() == 2)
2661*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SADDO, SDLoc(N), N->getVTList(),
2662*9880d681SAndroid Build Coastguard Worker N->getOperand(0), N->getOperand(0));
2663*9880d681SAndroid Build Coastguard Worker
2664*9880d681SAndroid Build Coastguard Worker return SDValue();
2665*9880d681SAndroid Build Coastguard Worker }
2666*9880d681SAndroid Build Coastguard Worker
visitUMULO(SDNode * N)2667*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitUMULO(SDNode *N) {
2668*9880d681SAndroid Build Coastguard Worker // (umulo x, 2) -> (uaddo x, x)
2669*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *C2 = dyn_cast<ConstantSDNode>(N->getOperand(1)))
2670*9880d681SAndroid Build Coastguard Worker if (C2->getAPIntValue() == 2)
2671*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::UADDO, SDLoc(N), N->getVTList(),
2672*9880d681SAndroid Build Coastguard Worker N->getOperand(0), N->getOperand(0));
2673*9880d681SAndroid Build Coastguard Worker
2674*9880d681SAndroid Build Coastguard Worker return SDValue();
2675*9880d681SAndroid Build Coastguard Worker }
2676*9880d681SAndroid Build Coastguard Worker
visitIMINMAX(SDNode * N)2677*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitIMINMAX(SDNode *N) {
2678*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
2679*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
2680*9880d681SAndroid Build Coastguard Worker EVT VT = N0.getValueType();
2681*9880d681SAndroid Build Coastguard Worker
2682*9880d681SAndroid Build Coastguard Worker // fold vector ops
2683*9880d681SAndroid Build Coastguard Worker if (VT.isVector())
2684*9880d681SAndroid Build Coastguard Worker if (SDValue FoldedVOp = SimplifyVBinOp(N))
2685*9880d681SAndroid Build Coastguard Worker return FoldedVOp;
2686*9880d681SAndroid Build Coastguard Worker
2687*9880d681SAndroid Build Coastguard Worker // fold (add c1, c2) -> c1+c2
2688*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N0C = getAsNonOpaqueConstant(N0);
2689*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N1C = getAsNonOpaqueConstant(N1);
2690*9880d681SAndroid Build Coastguard Worker if (N0C && N1C)
2691*9880d681SAndroid Build Coastguard Worker return DAG.FoldConstantArithmetic(N->getOpcode(), SDLoc(N), VT, N0C, N1C);
2692*9880d681SAndroid Build Coastguard Worker
2693*9880d681SAndroid Build Coastguard Worker // canonicalize constant to RHS
2694*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N0) &&
2695*9880d681SAndroid Build Coastguard Worker !DAG.isConstantIntBuildVectorOrConstantInt(N1))
2696*9880d681SAndroid Build Coastguard Worker return DAG.getNode(N->getOpcode(), SDLoc(N), VT, N1, N0);
2697*9880d681SAndroid Build Coastguard Worker
2698*9880d681SAndroid Build Coastguard Worker return SDValue();
2699*9880d681SAndroid Build Coastguard Worker }
2700*9880d681SAndroid Build Coastguard Worker
2701*9880d681SAndroid Build Coastguard Worker /// If this is a binary operator with two operands of the same opcode, try to
2702*9880d681SAndroid Build Coastguard Worker /// simplify it.
SimplifyBinOpWithSameOpcodeHands(SDNode * N)2703*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::SimplifyBinOpWithSameOpcodeHands(SDNode *N) {
2704*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0), N1 = N->getOperand(1);
2705*9880d681SAndroid Build Coastguard Worker EVT VT = N0.getValueType();
2706*9880d681SAndroid Build Coastguard Worker assert(N0.getOpcode() == N1.getOpcode() && "Bad input!");
2707*9880d681SAndroid Build Coastguard Worker
2708*9880d681SAndroid Build Coastguard Worker // Bail early if none of these transforms apply.
2709*9880d681SAndroid Build Coastguard Worker if (N0.getNode()->getNumOperands() == 0) return SDValue();
2710*9880d681SAndroid Build Coastguard Worker
2711*9880d681SAndroid Build Coastguard Worker // For each of OP in AND/OR/XOR:
2712*9880d681SAndroid Build Coastguard Worker // fold (OP (zext x), (zext y)) -> (zext (OP x, y))
2713*9880d681SAndroid Build Coastguard Worker // fold (OP (sext x), (sext y)) -> (sext (OP x, y))
2714*9880d681SAndroid Build Coastguard Worker // fold (OP (aext x), (aext y)) -> (aext (OP x, y))
2715*9880d681SAndroid Build Coastguard Worker // fold (OP (bswap x), (bswap y)) -> (bswap (OP x, y))
2716*9880d681SAndroid Build Coastguard Worker // fold (OP (trunc x), (trunc y)) -> (trunc (OP x, y)) (if trunc isn't free)
2717*9880d681SAndroid Build Coastguard Worker //
2718*9880d681SAndroid Build Coastguard Worker // do not sink logical op inside of a vector extend, since it may combine
2719*9880d681SAndroid Build Coastguard Worker // into a vsetcc.
2720*9880d681SAndroid Build Coastguard Worker EVT Op0VT = N0.getOperand(0).getValueType();
2721*9880d681SAndroid Build Coastguard Worker if ((N0.getOpcode() == ISD::ZERO_EXTEND ||
2722*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::SIGN_EXTEND ||
2723*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::BSWAP ||
2724*9880d681SAndroid Build Coastguard Worker // Avoid infinite looping with PromoteIntBinOp.
2725*9880d681SAndroid Build Coastguard Worker (N0.getOpcode() == ISD::ANY_EXTEND &&
2726*9880d681SAndroid Build Coastguard Worker (!LegalTypes || TLI.isTypeDesirableForOp(N->getOpcode(), Op0VT))) ||
2727*9880d681SAndroid Build Coastguard Worker (N0.getOpcode() == ISD::TRUNCATE &&
2728*9880d681SAndroid Build Coastguard Worker (!TLI.isZExtFree(VT, Op0VT) ||
2729*9880d681SAndroid Build Coastguard Worker !TLI.isTruncateFree(Op0VT, VT)) &&
2730*9880d681SAndroid Build Coastguard Worker TLI.isTypeLegal(Op0VT))) &&
2731*9880d681SAndroid Build Coastguard Worker !VT.isVector() &&
2732*9880d681SAndroid Build Coastguard Worker Op0VT == N1.getOperand(0).getValueType() &&
2733*9880d681SAndroid Build Coastguard Worker (!LegalOperations || TLI.isOperationLegal(N->getOpcode(), Op0VT))) {
2734*9880d681SAndroid Build Coastguard Worker SDValue ORNode = DAG.getNode(N->getOpcode(), SDLoc(N0),
2735*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getValueType(),
2736*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N1.getOperand(0));
2737*9880d681SAndroid Build Coastguard Worker AddToWorklist(ORNode.getNode());
2738*9880d681SAndroid Build Coastguard Worker return DAG.getNode(N0.getOpcode(), SDLoc(N), VT, ORNode);
2739*9880d681SAndroid Build Coastguard Worker }
2740*9880d681SAndroid Build Coastguard Worker
2741*9880d681SAndroid Build Coastguard Worker // For each of OP in SHL/SRL/SRA/AND...
2742*9880d681SAndroid Build Coastguard Worker // fold (and (OP x, z), (OP y, z)) -> (OP (and x, y), z)
2743*9880d681SAndroid Build Coastguard Worker // fold (or (OP x, z), (OP y, z)) -> (OP (or x, y), z)
2744*9880d681SAndroid Build Coastguard Worker // fold (xor (OP x, z), (OP y, z)) -> (OP (xor x, y), z)
2745*9880d681SAndroid Build Coastguard Worker if ((N0.getOpcode() == ISD::SHL || N0.getOpcode() == ISD::SRL ||
2746*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::SRA || N0.getOpcode() == ISD::AND) &&
2747*9880d681SAndroid Build Coastguard Worker N0.getOperand(1) == N1.getOperand(1)) {
2748*9880d681SAndroid Build Coastguard Worker SDValue ORNode = DAG.getNode(N->getOpcode(), SDLoc(N0),
2749*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getValueType(),
2750*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N1.getOperand(0));
2751*9880d681SAndroid Build Coastguard Worker AddToWorklist(ORNode.getNode());
2752*9880d681SAndroid Build Coastguard Worker return DAG.getNode(N0.getOpcode(), SDLoc(N), VT,
2753*9880d681SAndroid Build Coastguard Worker ORNode, N0.getOperand(1));
2754*9880d681SAndroid Build Coastguard Worker }
2755*9880d681SAndroid Build Coastguard Worker
2756*9880d681SAndroid Build Coastguard Worker // Simplify xor/and/or (bitcast(A), bitcast(B)) -> bitcast(op (A,B))
2757*9880d681SAndroid Build Coastguard Worker // Only perform this optimization up until type legalization, before
2758*9880d681SAndroid Build Coastguard Worker // LegalizeVectorOprs. LegalizeVectorOprs promotes vector operations by
2759*9880d681SAndroid Build Coastguard Worker // adding bitcasts. For example (xor v4i32) is promoted to (v2i64), and
2760*9880d681SAndroid Build Coastguard Worker // we don't want to undo this promotion.
2761*9880d681SAndroid Build Coastguard Worker // We also handle SCALAR_TO_VECTOR because xor/or/and operations are cheaper
2762*9880d681SAndroid Build Coastguard Worker // on scalars.
2763*9880d681SAndroid Build Coastguard Worker if ((N0.getOpcode() == ISD::BITCAST ||
2764*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::SCALAR_TO_VECTOR) &&
2765*9880d681SAndroid Build Coastguard Worker Level <= AfterLegalizeTypes) {
2766*9880d681SAndroid Build Coastguard Worker SDValue In0 = N0.getOperand(0);
2767*9880d681SAndroid Build Coastguard Worker SDValue In1 = N1.getOperand(0);
2768*9880d681SAndroid Build Coastguard Worker EVT In0Ty = In0.getValueType();
2769*9880d681SAndroid Build Coastguard Worker EVT In1Ty = In1.getValueType();
2770*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
2771*9880d681SAndroid Build Coastguard Worker // If both incoming values are integers, and the original types are the
2772*9880d681SAndroid Build Coastguard Worker // same.
2773*9880d681SAndroid Build Coastguard Worker if (In0Ty.isInteger() && In1Ty.isInteger() && In0Ty == In1Ty) {
2774*9880d681SAndroid Build Coastguard Worker SDValue Op = DAG.getNode(N->getOpcode(), DL, In0Ty, In0, In1);
2775*9880d681SAndroid Build Coastguard Worker SDValue BC = DAG.getNode(N0.getOpcode(), DL, VT, Op);
2776*9880d681SAndroid Build Coastguard Worker AddToWorklist(Op.getNode());
2777*9880d681SAndroid Build Coastguard Worker return BC;
2778*9880d681SAndroid Build Coastguard Worker }
2779*9880d681SAndroid Build Coastguard Worker }
2780*9880d681SAndroid Build Coastguard Worker
2781*9880d681SAndroid Build Coastguard Worker // Xor/and/or are indifferent to the swizzle operation (shuffle of one value).
2782*9880d681SAndroid Build Coastguard Worker // Simplify xor/and/or (shuff(A), shuff(B)) -> shuff(op (A,B))
2783*9880d681SAndroid Build Coastguard Worker // If both shuffles use the same mask, and both shuffle within a single
2784*9880d681SAndroid Build Coastguard Worker // vector, then it is worthwhile to move the swizzle after the operation.
2785*9880d681SAndroid Build Coastguard Worker // The type-legalizer generates this pattern when loading illegal
2786*9880d681SAndroid Build Coastguard Worker // vector types from memory. In many cases this allows additional shuffle
2787*9880d681SAndroid Build Coastguard Worker // optimizations.
2788*9880d681SAndroid Build Coastguard Worker // There are other cases where moving the shuffle after the xor/and/or
2789*9880d681SAndroid Build Coastguard Worker // is profitable even if shuffles don't perform a swizzle.
2790*9880d681SAndroid Build Coastguard Worker // If both shuffles use the same mask, and both shuffles have the same first
2791*9880d681SAndroid Build Coastguard Worker // or second operand, then it might still be profitable to move the shuffle
2792*9880d681SAndroid Build Coastguard Worker // after the xor/and/or operation.
2793*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::VECTOR_SHUFFLE && Level < AfterLegalizeDAG) {
2794*9880d681SAndroid Build Coastguard Worker ShuffleVectorSDNode *SVN0 = cast<ShuffleVectorSDNode>(N0);
2795*9880d681SAndroid Build Coastguard Worker ShuffleVectorSDNode *SVN1 = cast<ShuffleVectorSDNode>(N1);
2796*9880d681SAndroid Build Coastguard Worker
2797*9880d681SAndroid Build Coastguard Worker assert(N0.getOperand(0).getValueType() == N1.getOperand(0).getValueType() &&
2798*9880d681SAndroid Build Coastguard Worker "Inputs to shuffles are not the same type");
2799*9880d681SAndroid Build Coastguard Worker
2800*9880d681SAndroid Build Coastguard Worker // Check that both shuffles use the same mask. The masks are known to be of
2801*9880d681SAndroid Build Coastguard Worker // the same length because the result vector type is the same.
2802*9880d681SAndroid Build Coastguard Worker // Check also that shuffles have only one use to avoid introducing extra
2803*9880d681SAndroid Build Coastguard Worker // instructions.
2804*9880d681SAndroid Build Coastguard Worker if (SVN0->hasOneUse() && SVN1->hasOneUse() &&
2805*9880d681SAndroid Build Coastguard Worker SVN0->getMask().equals(SVN1->getMask())) {
2806*9880d681SAndroid Build Coastguard Worker SDValue ShOp = N0->getOperand(1);
2807*9880d681SAndroid Build Coastguard Worker
2808*9880d681SAndroid Build Coastguard Worker // Don't try to fold this node if it requires introducing a
2809*9880d681SAndroid Build Coastguard Worker // build vector of all zeros that might be illegal at this stage.
2810*9880d681SAndroid Build Coastguard Worker if (N->getOpcode() == ISD::XOR && !ShOp.isUndef()) {
2811*9880d681SAndroid Build Coastguard Worker if (!LegalTypes)
2812*9880d681SAndroid Build Coastguard Worker ShOp = DAG.getConstant(0, SDLoc(N), VT);
2813*9880d681SAndroid Build Coastguard Worker else
2814*9880d681SAndroid Build Coastguard Worker ShOp = SDValue();
2815*9880d681SAndroid Build Coastguard Worker }
2816*9880d681SAndroid Build Coastguard Worker
2817*9880d681SAndroid Build Coastguard Worker // (AND (shuf (A, C), shuf (B, C)) -> shuf (AND (A, B), C)
2818*9880d681SAndroid Build Coastguard Worker // (OR (shuf (A, C), shuf (B, C)) -> shuf (OR (A, B), C)
2819*9880d681SAndroid Build Coastguard Worker // (XOR (shuf (A, C), shuf (B, C)) -> shuf (XOR (A, B), V_0)
2820*9880d681SAndroid Build Coastguard Worker if (N0.getOperand(1) == N1.getOperand(1) && ShOp.getNode()) {
2821*9880d681SAndroid Build Coastguard Worker SDValue NewNode = DAG.getNode(N->getOpcode(), SDLoc(N), VT,
2822*9880d681SAndroid Build Coastguard Worker N0->getOperand(0), N1->getOperand(0));
2823*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewNode.getNode());
2824*9880d681SAndroid Build Coastguard Worker return DAG.getVectorShuffle(VT, SDLoc(N), NewNode, ShOp,
2825*9880d681SAndroid Build Coastguard Worker SVN0->getMask());
2826*9880d681SAndroid Build Coastguard Worker }
2827*9880d681SAndroid Build Coastguard Worker
2828*9880d681SAndroid Build Coastguard Worker // Don't try to fold this node if it requires introducing a
2829*9880d681SAndroid Build Coastguard Worker // build vector of all zeros that might be illegal at this stage.
2830*9880d681SAndroid Build Coastguard Worker ShOp = N0->getOperand(0);
2831*9880d681SAndroid Build Coastguard Worker if (N->getOpcode() == ISD::XOR && !ShOp.isUndef()) {
2832*9880d681SAndroid Build Coastguard Worker if (!LegalTypes)
2833*9880d681SAndroid Build Coastguard Worker ShOp = DAG.getConstant(0, SDLoc(N), VT);
2834*9880d681SAndroid Build Coastguard Worker else
2835*9880d681SAndroid Build Coastguard Worker ShOp = SDValue();
2836*9880d681SAndroid Build Coastguard Worker }
2837*9880d681SAndroid Build Coastguard Worker
2838*9880d681SAndroid Build Coastguard Worker // (AND (shuf (C, A), shuf (C, B)) -> shuf (C, AND (A, B))
2839*9880d681SAndroid Build Coastguard Worker // (OR (shuf (C, A), shuf (C, B)) -> shuf (C, OR (A, B))
2840*9880d681SAndroid Build Coastguard Worker // (XOR (shuf (C, A), shuf (C, B)) -> shuf (V_0, XOR (A, B))
2841*9880d681SAndroid Build Coastguard Worker if (N0->getOperand(0) == N1->getOperand(0) && ShOp.getNode()) {
2842*9880d681SAndroid Build Coastguard Worker SDValue NewNode = DAG.getNode(N->getOpcode(), SDLoc(N), VT,
2843*9880d681SAndroid Build Coastguard Worker N0->getOperand(1), N1->getOperand(1));
2844*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewNode.getNode());
2845*9880d681SAndroid Build Coastguard Worker return DAG.getVectorShuffle(VT, SDLoc(N), ShOp, NewNode,
2846*9880d681SAndroid Build Coastguard Worker SVN0->getMask());
2847*9880d681SAndroid Build Coastguard Worker }
2848*9880d681SAndroid Build Coastguard Worker }
2849*9880d681SAndroid Build Coastguard Worker }
2850*9880d681SAndroid Build Coastguard Worker
2851*9880d681SAndroid Build Coastguard Worker return SDValue();
2852*9880d681SAndroid Build Coastguard Worker }
2853*9880d681SAndroid Build Coastguard Worker
2854*9880d681SAndroid Build Coastguard Worker /// This contains all DAGCombine rules which reduce two values combined by
2855*9880d681SAndroid Build Coastguard Worker /// an And operation to a single value. This makes them reusable in the context
2856*9880d681SAndroid Build Coastguard Worker /// of visitSELECT(). Rules involving constants are not included as
2857*9880d681SAndroid Build Coastguard Worker /// visitSELECT() already handles those cases.
visitANDLike(SDValue N0,SDValue N1,SDNode * LocReference)2858*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitANDLike(SDValue N0, SDValue N1,
2859*9880d681SAndroid Build Coastguard Worker SDNode *LocReference) {
2860*9880d681SAndroid Build Coastguard Worker EVT VT = N1.getValueType();
2861*9880d681SAndroid Build Coastguard Worker
2862*9880d681SAndroid Build Coastguard Worker // fold (and x, undef) -> 0
2863*9880d681SAndroid Build Coastguard Worker if (N0.isUndef() || N1.isUndef())
2864*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, SDLoc(LocReference), VT);
2865*9880d681SAndroid Build Coastguard Worker // fold (and (setcc x), (setcc y)) -> (setcc (and x, y))
2866*9880d681SAndroid Build Coastguard Worker SDValue LL, LR, RL, RR, CC0, CC1;
2867*9880d681SAndroid Build Coastguard Worker if (isSetCCEquivalent(N0, LL, LR, CC0) && isSetCCEquivalent(N1, RL, RR, CC1)){
2868*9880d681SAndroid Build Coastguard Worker ISD::CondCode Op0 = cast<CondCodeSDNode>(CC0)->get();
2869*9880d681SAndroid Build Coastguard Worker ISD::CondCode Op1 = cast<CondCodeSDNode>(CC1)->get();
2870*9880d681SAndroid Build Coastguard Worker
2871*9880d681SAndroid Build Coastguard Worker if (LR == RR && isa<ConstantSDNode>(LR) && Op0 == Op1 &&
2872*9880d681SAndroid Build Coastguard Worker LL.getValueType().isInteger()) {
2873*9880d681SAndroid Build Coastguard Worker // fold (and (seteq X, 0), (seteq Y, 0)) -> (seteq (or X, Y), 0)
2874*9880d681SAndroid Build Coastguard Worker if (isNullConstant(LR) && Op1 == ISD::SETEQ) {
2875*9880d681SAndroid Build Coastguard Worker SDValue ORNode = DAG.getNode(ISD::OR, SDLoc(N0),
2876*9880d681SAndroid Build Coastguard Worker LR.getValueType(), LL, RL);
2877*9880d681SAndroid Build Coastguard Worker AddToWorklist(ORNode.getNode());
2878*9880d681SAndroid Build Coastguard Worker return DAG.getSetCC(SDLoc(LocReference), VT, ORNode, LR, Op1);
2879*9880d681SAndroid Build Coastguard Worker }
2880*9880d681SAndroid Build Coastguard Worker if (isAllOnesConstant(LR)) {
2881*9880d681SAndroid Build Coastguard Worker // fold (and (seteq X, -1), (seteq Y, -1)) -> (seteq (and X, Y), -1)
2882*9880d681SAndroid Build Coastguard Worker if (Op1 == ISD::SETEQ) {
2883*9880d681SAndroid Build Coastguard Worker SDValue ANDNode = DAG.getNode(ISD::AND, SDLoc(N0),
2884*9880d681SAndroid Build Coastguard Worker LR.getValueType(), LL, RL);
2885*9880d681SAndroid Build Coastguard Worker AddToWorklist(ANDNode.getNode());
2886*9880d681SAndroid Build Coastguard Worker return DAG.getSetCC(SDLoc(LocReference), VT, ANDNode, LR, Op1);
2887*9880d681SAndroid Build Coastguard Worker }
2888*9880d681SAndroid Build Coastguard Worker // fold (and (setgt X, -1), (setgt Y, -1)) -> (setgt (or X, Y), -1)
2889*9880d681SAndroid Build Coastguard Worker if (Op1 == ISD::SETGT) {
2890*9880d681SAndroid Build Coastguard Worker SDValue ORNode = DAG.getNode(ISD::OR, SDLoc(N0),
2891*9880d681SAndroid Build Coastguard Worker LR.getValueType(), LL, RL);
2892*9880d681SAndroid Build Coastguard Worker AddToWorklist(ORNode.getNode());
2893*9880d681SAndroid Build Coastguard Worker return DAG.getSetCC(SDLoc(LocReference), VT, ORNode, LR, Op1);
2894*9880d681SAndroid Build Coastguard Worker }
2895*9880d681SAndroid Build Coastguard Worker }
2896*9880d681SAndroid Build Coastguard Worker }
2897*9880d681SAndroid Build Coastguard Worker // Simplify (and (setne X, 0), (setne X, -1)) -> (setuge (add X, 1), 2)
2898*9880d681SAndroid Build Coastguard Worker if (LL == RL && isa<ConstantSDNode>(LR) && isa<ConstantSDNode>(RR) &&
2899*9880d681SAndroid Build Coastguard Worker Op0 == Op1 && LL.getValueType().isInteger() &&
2900*9880d681SAndroid Build Coastguard Worker Op0 == ISD::SETNE && ((isNullConstant(LR) && isAllOnesConstant(RR)) ||
2901*9880d681SAndroid Build Coastguard Worker (isAllOnesConstant(LR) && isNullConstant(RR)))) {
2902*9880d681SAndroid Build Coastguard Worker SDLoc DL(N0);
2903*9880d681SAndroid Build Coastguard Worker SDValue ADDNode = DAG.getNode(ISD::ADD, DL, LL.getValueType(),
2904*9880d681SAndroid Build Coastguard Worker LL, DAG.getConstant(1, DL,
2905*9880d681SAndroid Build Coastguard Worker LL.getValueType()));
2906*9880d681SAndroid Build Coastguard Worker AddToWorklist(ADDNode.getNode());
2907*9880d681SAndroid Build Coastguard Worker return DAG.getSetCC(SDLoc(LocReference), VT, ADDNode,
2908*9880d681SAndroid Build Coastguard Worker DAG.getConstant(2, DL, LL.getValueType()),
2909*9880d681SAndroid Build Coastguard Worker ISD::SETUGE);
2910*9880d681SAndroid Build Coastguard Worker }
2911*9880d681SAndroid Build Coastguard Worker // canonicalize equivalent to ll == rl
2912*9880d681SAndroid Build Coastguard Worker if (LL == RR && LR == RL) {
2913*9880d681SAndroid Build Coastguard Worker Op1 = ISD::getSetCCSwappedOperands(Op1);
2914*9880d681SAndroid Build Coastguard Worker std::swap(RL, RR);
2915*9880d681SAndroid Build Coastguard Worker }
2916*9880d681SAndroid Build Coastguard Worker if (LL == RL && LR == RR) {
2917*9880d681SAndroid Build Coastguard Worker bool isInteger = LL.getValueType().isInteger();
2918*9880d681SAndroid Build Coastguard Worker ISD::CondCode Result = ISD::getSetCCAndOperation(Op0, Op1, isInteger);
2919*9880d681SAndroid Build Coastguard Worker if (Result != ISD::SETCC_INVALID &&
2920*9880d681SAndroid Build Coastguard Worker (!LegalOperations ||
2921*9880d681SAndroid Build Coastguard Worker (TLI.isCondCodeLegal(Result, LL.getSimpleValueType()) &&
2922*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegal(ISD::SETCC, LL.getValueType())))) {
2923*9880d681SAndroid Build Coastguard Worker EVT CCVT = getSetCCResultType(LL.getValueType());
2924*9880d681SAndroid Build Coastguard Worker if (N0.getValueType() == CCVT ||
2925*9880d681SAndroid Build Coastguard Worker (!LegalOperations && N0.getValueType() == MVT::i1))
2926*9880d681SAndroid Build Coastguard Worker return DAG.getSetCC(SDLoc(LocReference), N0.getValueType(),
2927*9880d681SAndroid Build Coastguard Worker LL, LR, Result);
2928*9880d681SAndroid Build Coastguard Worker }
2929*9880d681SAndroid Build Coastguard Worker }
2930*9880d681SAndroid Build Coastguard Worker }
2931*9880d681SAndroid Build Coastguard Worker
2932*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::ADD && N1.getOpcode() == ISD::SRL &&
2933*9880d681SAndroid Build Coastguard Worker VT.getSizeInBits() <= 64) {
2934*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *ADDI = dyn_cast<ConstantSDNode>(N0.getOperand(1))) {
2935*9880d681SAndroid Build Coastguard Worker APInt ADDC = ADDI->getAPIntValue();
2936*9880d681SAndroid Build Coastguard Worker if (!TLI.isLegalAddImmediate(ADDC.getSExtValue())) {
2937*9880d681SAndroid Build Coastguard Worker // Look for (and (add x, c1), (lshr y, c2)). If C1 wasn't a legal
2938*9880d681SAndroid Build Coastguard Worker // immediate for an add, but it is legal if its top c2 bits are set,
2939*9880d681SAndroid Build Coastguard Worker // transform the ADD so the immediate doesn't need to be materialized
2940*9880d681SAndroid Build Coastguard Worker // in a register.
2941*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *SRLI = dyn_cast<ConstantSDNode>(N1.getOperand(1))) {
2942*9880d681SAndroid Build Coastguard Worker APInt Mask = APInt::getHighBitsSet(VT.getSizeInBits(),
2943*9880d681SAndroid Build Coastguard Worker SRLI->getZExtValue());
2944*9880d681SAndroid Build Coastguard Worker if (DAG.MaskedValueIsZero(N0.getOperand(1), Mask)) {
2945*9880d681SAndroid Build Coastguard Worker ADDC |= Mask;
2946*9880d681SAndroid Build Coastguard Worker if (TLI.isLegalAddImmediate(ADDC.getSExtValue())) {
2947*9880d681SAndroid Build Coastguard Worker SDLoc DL(N0);
2948*9880d681SAndroid Build Coastguard Worker SDValue NewAdd =
2949*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::ADD, DL, VT,
2950*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), DAG.getConstant(ADDC, DL, VT));
2951*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(), NewAdd);
2952*9880d681SAndroid Build Coastguard Worker // Return N so it doesn't get rechecked!
2953*9880d681SAndroid Build Coastguard Worker return SDValue(LocReference, 0);
2954*9880d681SAndroid Build Coastguard Worker }
2955*9880d681SAndroid Build Coastguard Worker }
2956*9880d681SAndroid Build Coastguard Worker }
2957*9880d681SAndroid Build Coastguard Worker }
2958*9880d681SAndroid Build Coastguard Worker }
2959*9880d681SAndroid Build Coastguard Worker }
2960*9880d681SAndroid Build Coastguard Worker
2961*9880d681SAndroid Build Coastguard Worker // Reduce bit extract of low half of an integer to the narrower type.
2962*9880d681SAndroid Build Coastguard Worker // (and (srl i64:x, K), KMask) ->
2963*9880d681SAndroid Build Coastguard Worker // (i64 zero_extend (and (srl (i32 (trunc i64:x)), K)), KMask)
2964*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SRL && N0.hasOneUse()) {
2965*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *CAnd = dyn_cast<ConstantSDNode>(N1)) {
2966*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *CShift = dyn_cast<ConstantSDNode>(N0.getOperand(1))) {
2967*9880d681SAndroid Build Coastguard Worker unsigned Size = VT.getSizeInBits();
2968*9880d681SAndroid Build Coastguard Worker const APInt &AndMask = CAnd->getAPIntValue();
2969*9880d681SAndroid Build Coastguard Worker unsigned ShiftBits = CShift->getZExtValue();
2970*9880d681SAndroid Build Coastguard Worker unsigned MaskBits = AndMask.countTrailingOnes();
2971*9880d681SAndroid Build Coastguard Worker EVT HalfVT = EVT::getIntegerVT(*DAG.getContext(), Size / 2);
2972*9880d681SAndroid Build Coastguard Worker
2973*9880d681SAndroid Build Coastguard Worker if (APIntOps::isMask(AndMask) &&
2974*9880d681SAndroid Build Coastguard Worker // Required bits must not span the two halves of the integer and
2975*9880d681SAndroid Build Coastguard Worker // must fit in the half size type.
2976*9880d681SAndroid Build Coastguard Worker (ShiftBits + MaskBits <= Size / 2) &&
2977*9880d681SAndroid Build Coastguard Worker TLI.isNarrowingProfitable(VT, HalfVT) &&
2978*9880d681SAndroid Build Coastguard Worker TLI.isTypeDesirableForOp(ISD::AND, HalfVT) &&
2979*9880d681SAndroid Build Coastguard Worker TLI.isTypeDesirableForOp(ISD::SRL, HalfVT) &&
2980*9880d681SAndroid Build Coastguard Worker TLI.isTruncateFree(VT, HalfVT) &&
2981*9880d681SAndroid Build Coastguard Worker TLI.isZExtFree(HalfVT, VT)) {
2982*9880d681SAndroid Build Coastguard Worker // The isNarrowingProfitable is to avoid regressions on PPC and
2983*9880d681SAndroid Build Coastguard Worker // AArch64 which match a few 64-bit bit insert / bit extract patterns
2984*9880d681SAndroid Build Coastguard Worker // on downstream users of this. Those patterns could probably be
2985*9880d681SAndroid Build Coastguard Worker // extended to handle extensions mixed in.
2986*9880d681SAndroid Build Coastguard Worker
2987*9880d681SAndroid Build Coastguard Worker SDValue SL(N0);
2988*9880d681SAndroid Build Coastguard Worker assert(ShiftBits != 0 && MaskBits <= Size);
2989*9880d681SAndroid Build Coastguard Worker
2990*9880d681SAndroid Build Coastguard Worker // Extracting the highest bit of the low half.
2991*9880d681SAndroid Build Coastguard Worker EVT ShiftVT = TLI.getShiftAmountTy(HalfVT, DAG.getDataLayout());
2992*9880d681SAndroid Build Coastguard Worker SDValue Trunc = DAG.getNode(ISD::TRUNCATE, SL, HalfVT,
2993*9880d681SAndroid Build Coastguard Worker N0.getOperand(0));
2994*9880d681SAndroid Build Coastguard Worker
2995*9880d681SAndroid Build Coastguard Worker SDValue NewMask = DAG.getConstant(AndMask.trunc(Size / 2), SL, HalfVT);
2996*9880d681SAndroid Build Coastguard Worker SDValue ShiftK = DAG.getConstant(ShiftBits, SL, ShiftVT);
2997*9880d681SAndroid Build Coastguard Worker SDValue Shift = DAG.getNode(ISD::SRL, SL, HalfVT, Trunc, ShiftK);
2998*9880d681SAndroid Build Coastguard Worker SDValue And = DAG.getNode(ISD::AND, SL, HalfVT, Shift, NewMask);
2999*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ZERO_EXTEND, SL, VT, And);
3000*9880d681SAndroid Build Coastguard Worker }
3001*9880d681SAndroid Build Coastguard Worker }
3002*9880d681SAndroid Build Coastguard Worker }
3003*9880d681SAndroid Build Coastguard Worker }
3004*9880d681SAndroid Build Coastguard Worker
3005*9880d681SAndroid Build Coastguard Worker return SDValue();
3006*9880d681SAndroid Build Coastguard Worker }
3007*9880d681SAndroid Build Coastguard Worker
isAndLoadExtLoad(ConstantSDNode * AndC,LoadSDNode * LoadN,EVT LoadResultTy,EVT & ExtVT,EVT & LoadedVT,bool & NarrowLoad)3008*9880d681SAndroid Build Coastguard Worker bool DAGCombiner::isAndLoadExtLoad(ConstantSDNode *AndC, LoadSDNode *LoadN,
3009*9880d681SAndroid Build Coastguard Worker EVT LoadResultTy, EVT &ExtVT, EVT &LoadedVT,
3010*9880d681SAndroid Build Coastguard Worker bool &NarrowLoad) {
3011*9880d681SAndroid Build Coastguard Worker uint32_t ActiveBits = AndC->getAPIntValue().getActiveBits();
3012*9880d681SAndroid Build Coastguard Worker
3013*9880d681SAndroid Build Coastguard Worker if (ActiveBits == 0 || !APIntOps::isMask(ActiveBits, AndC->getAPIntValue()))
3014*9880d681SAndroid Build Coastguard Worker return false;
3015*9880d681SAndroid Build Coastguard Worker
3016*9880d681SAndroid Build Coastguard Worker ExtVT = EVT::getIntegerVT(*DAG.getContext(), ActiveBits);
3017*9880d681SAndroid Build Coastguard Worker LoadedVT = LoadN->getMemoryVT();
3018*9880d681SAndroid Build Coastguard Worker
3019*9880d681SAndroid Build Coastguard Worker if (ExtVT == LoadedVT &&
3020*9880d681SAndroid Build Coastguard Worker (!LegalOperations ||
3021*9880d681SAndroid Build Coastguard Worker TLI.isLoadExtLegal(ISD::ZEXTLOAD, LoadResultTy, ExtVT))) {
3022*9880d681SAndroid Build Coastguard Worker // ZEXTLOAD will match without needing to change the size of the value being
3023*9880d681SAndroid Build Coastguard Worker // loaded.
3024*9880d681SAndroid Build Coastguard Worker NarrowLoad = false;
3025*9880d681SAndroid Build Coastguard Worker return true;
3026*9880d681SAndroid Build Coastguard Worker }
3027*9880d681SAndroid Build Coastguard Worker
3028*9880d681SAndroid Build Coastguard Worker // Do not change the width of a volatile load.
3029*9880d681SAndroid Build Coastguard Worker if (LoadN->isVolatile())
3030*9880d681SAndroid Build Coastguard Worker return false;
3031*9880d681SAndroid Build Coastguard Worker
3032*9880d681SAndroid Build Coastguard Worker // Do not generate loads of non-round integer types since these can
3033*9880d681SAndroid Build Coastguard Worker // be expensive (and would be wrong if the type is not byte sized).
3034*9880d681SAndroid Build Coastguard Worker if (!LoadedVT.bitsGT(ExtVT) || !ExtVT.isRound())
3035*9880d681SAndroid Build Coastguard Worker return false;
3036*9880d681SAndroid Build Coastguard Worker
3037*9880d681SAndroid Build Coastguard Worker if (LegalOperations &&
3038*9880d681SAndroid Build Coastguard Worker !TLI.isLoadExtLegal(ISD::ZEXTLOAD, LoadResultTy, ExtVT))
3039*9880d681SAndroid Build Coastguard Worker return false;
3040*9880d681SAndroid Build Coastguard Worker
3041*9880d681SAndroid Build Coastguard Worker if (!TLI.shouldReduceLoadWidth(LoadN, ISD::ZEXTLOAD, ExtVT))
3042*9880d681SAndroid Build Coastguard Worker return false;
3043*9880d681SAndroid Build Coastguard Worker
3044*9880d681SAndroid Build Coastguard Worker NarrowLoad = true;
3045*9880d681SAndroid Build Coastguard Worker return true;
3046*9880d681SAndroid Build Coastguard Worker }
3047*9880d681SAndroid Build Coastguard Worker
visitAND(SDNode * N)3048*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitAND(SDNode *N) {
3049*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
3050*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
3051*9880d681SAndroid Build Coastguard Worker EVT VT = N1.getValueType();
3052*9880d681SAndroid Build Coastguard Worker
3053*9880d681SAndroid Build Coastguard Worker // fold vector ops
3054*9880d681SAndroid Build Coastguard Worker if (VT.isVector()) {
3055*9880d681SAndroid Build Coastguard Worker if (SDValue FoldedVOp = SimplifyVBinOp(N))
3056*9880d681SAndroid Build Coastguard Worker return FoldedVOp;
3057*9880d681SAndroid Build Coastguard Worker
3058*9880d681SAndroid Build Coastguard Worker // fold (and x, 0) -> 0, vector edition
3059*9880d681SAndroid Build Coastguard Worker if (ISD::isBuildVectorAllZeros(N0.getNode()))
3060*9880d681SAndroid Build Coastguard Worker // do not return N0, because undef node may exist in N0
3061*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(
3062*9880d681SAndroid Build Coastguard Worker APInt::getNullValue(
3063*9880d681SAndroid Build Coastguard Worker N0.getValueType().getScalarType().getSizeInBits()),
3064*9880d681SAndroid Build Coastguard Worker SDLoc(N), N0.getValueType());
3065*9880d681SAndroid Build Coastguard Worker if (ISD::isBuildVectorAllZeros(N1.getNode()))
3066*9880d681SAndroid Build Coastguard Worker // do not return N1, because undef node may exist in N1
3067*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(
3068*9880d681SAndroid Build Coastguard Worker APInt::getNullValue(
3069*9880d681SAndroid Build Coastguard Worker N1.getValueType().getScalarType().getSizeInBits()),
3070*9880d681SAndroid Build Coastguard Worker SDLoc(N), N1.getValueType());
3071*9880d681SAndroid Build Coastguard Worker
3072*9880d681SAndroid Build Coastguard Worker // fold (and x, -1) -> x, vector edition
3073*9880d681SAndroid Build Coastguard Worker if (ISD::isBuildVectorAllOnes(N0.getNode()))
3074*9880d681SAndroid Build Coastguard Worker return N1;
3075*9880d681SAndroid Build Coastguard Worker if (ISD::isBuildVectorAllOnes(N1.getNode()))
3076*9880d681SAndroid Build Coastguard Worker return N0;
3077*9880d681SAndroid Build Coastguard Worker }
3078*9880d681SAndroid Build Coastguard Worker
3079*9880d681SAndroid Build Coastguard Worker // fold (and c1, c2) -> c1&c2
3080*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N0C = getAsNonOpaqueConstant(N0);
3081*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
3082*9880d681SAndroid Build Coastguard Worker if (N0C && N1C && !N1C->isOpaque())
3083*9880d681SAndroid Build Coastguard Worker return DAG.FoldConstantArithmetic(ISD::AND, SDLoc(N), VT, N0C, N1C);
3084*9880d681SAndroid Build Coastguard Worker // canonicalize constant to RHS
3085*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N0) &&
3086*9880d681SAndroid Build Coastguard Worker !DAG.isConstantIntBuildVectorOrConstantInt(N1))
3087*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, SDLoc(N), VT, N1, N0);
3088*9880d681SAndroid Build Coastguard Worker // fold (and x, -1) -> x
3089*9880d681SAndroid Build Coastguard Worker if (isAllOnesConstant(N1))
3090*9880d681SAndroid Build Coastguard Worker return N0;
3091*9880d681SAndroid Build Coastguard Worker // if (and x, c) is known to be zero, return 0
3092*9880d681SAndroid Build Coastguard Worker unsigned BitWidth = VT.getScalarType().getSizeInBits();
3093*9880d681SAndroid Build Coastguard Worker if (N1C && DAG.MaskedValueIsZero(SDValue(N, 0),
3094*9880d681SAndroid Build Coastguard Worker APInt::getAllOnesValue(BitWidth)))
3095*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, SDLoc(N), VT);
3096*9880d681SAndroid Build Coastguard Worker // reassociate and
3097*9880d681SAndroid Build Coastguard Worker if (SDValue RAND = ReassociateOps(ISD::AND, SDLoc(N), N0, N1))
3098*9880d681SAndroid Build Coastguard Worker return RAND;
3099*9880d681SAndroid Build Coastguard Worker // fold (and (or x, C), D) -> D if (C & D) == D
3100*9880d681SAndroid Build Coastguard Worker if (N1C && N0.getOpcode() == ISD::OR)
3101*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *ORI = dyn_cast<ConstantSDNode>(N0.getOperand(1)))
3102*9880d681SAndroid Build Coastguard Worker if ((ORI->getAPIntValue() & N1C->getAPIntValue()) == N1C->getAPIntValue())
3103*9880d681SAndroid Build Coastguard Worker return N1;
3104*9880d681SAndroid Build Coastguard Worker // fold (and (any_ext V), c) -> (zero_ext V) if 'and' only clears top bits.
3105*9880d681SAndroid Build Coastguard Worker if (N1C && N0.getOpcode() == ISD::ANY_EXTEND) {
3106*9880d681SAndroid Build Coastguard Worker SDValue N0Op0 = N0.getOperand(0);
3107*9880d681SAndroid Build Coastguard Worker APInt Mask = ~N1C->getAPIntValue();
3108*9880d681SAndroid Build Coastguard Worker Mask = Mask.trunc(N0Op0.getValueSizeInBits());
3109*9880d681SAndroid Build Coastguard Worker if (DAG.MaskedValueIsZero(N0Op0, Mask)) {
3110*9880d681SAndroid Build Coastguard Worker SDValue Zext = DAG.getNode(ISD::ZERO_EXTEND, SDLoc(N),
3111*9880d681SAndroid Build Coastguard Worker N0.getValueType(), N0Op0);
3112*9880d681SAndroid Build Coastguard Worker
3113*9880d681SAndroid Build Coastguard Worker // Replace uses of the AND with uses of the Zero extend node.
3114*9880d681SAndroid Build Coastguard Worker CombineTo(N, Zext);
3115*9880d681SAndroid Build Coastguard Worker
3116*9880d681SAndroid Build Coastguard Worker // We actually want to replace all uses of the any_extend with the
3117*9880d681SAndroid Build Coastguard Worker // zero_extend, to avoid duplicating things. This will later cause this
3118*9880d681SAndroid Build Coastguard Worker // AND to be folded.
3119*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(), Zext);
3120*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
3121*9880d681SAndroid Build Coastguard Worker }
3122*9880d681SAndroid Build Coastguard Worker }
3123*9880d681SAndroid Build Coastguard Worker // similarly fold (and (X (load ([non_ext|any_ext|zero_ext] V))), c) ->
3124*9880d681SAndroid Build Coastguard Worker // (X (load ([non_ext|zero_ext] V))) if 'and' only clears top bits which must
3125*9880d681SAndroid Build Coastguard Worker // already be zero by virtue of the width of the base type of the load.
3126*9880d681SAndroid Build Coastguard Worker //
3127*9880d681SAndroid Build Coastguard Worker // the 'X' node here can either be nothing or an extract_vector_elt to catch
3128*9880d681SAndroid Build Coastguard Worker // more cases.
3129*9880d681SAndroid Build Coastguard Worker if ((N0.getOpcode() == ISD::EXTRACT_VECTOR_ELT &&
3130*9880d681SAndroid Build Coastguard Worker N0.getValueSizeInBits() == N0.getOperand(0).getScalarValueSizeInBits() &&
3131*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getOpcode() == ISD::LOAD &&
3132*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getResNo() == 0) ||
3133*9880d681SAndroid Build Coastguard Worker (N0.getOpcode() == ISD::LOAD && N0.getResNo() == 0)) {
3134*9880d681SAndroid Build Coastguard Worker LoadSDNode *Load = cast<LoadSDNode>( (N0.getOpcode() == ISD::LOAD) ?
3135*9880d681SAndroid Build Coastguard Worker N0 : N0.getOperand(0) );
3136*9880d681SAndroid Build Coastguard Worker
3137*9880d681SAndroid Build Coastguard Worker // Get the constant (if applicable) the zero'th operand is being ANDed with.
3138*9880d681SAndroid Build Coastguard Worker // This can be a pure constant or a vector splat, in which case we treat the
3139*9880d681SAndroid Build Coastguard Worker // vector as a scalar and use the splat value.
3140*9880d681SAndroid Build Coastguard Worker APInt Constant = APInt::getNullValue(1);
3141*9880d681SAndroid Build Coastguard Worker if (const ConstantSDNode *C = dyn_cast<ConstantSDNode>(N1)) {
3142*9880d681SAndroid Build Coastguard Worker Constant = C->getAPIntValue();
3143*9880d681SAndroid Build Coastguard Worker } else if (BuildVectorSDNode *Vector = dyn_cast<BuildVectorSDNode>(N1)) {
3144*9880d681SAndroid Build Coastguard Worker APInt SplatValue, SplatUndef;
3145*9880d681SAndroid Build Coastguard Worker unsigned SplatBitSize;
3146*9880d681SAndroid Build Coastguard Worker bool HasAnyUndefs;
3147*9880d681SAndroid Build Coastguard Worker bool IsSplat = Vector->isConstantSplat(SplatValue, SplatUndef,
3148*9880d681SAndroid Build Coastguard Worker SplatBitSize, HasAnyUndefs);
3149*9880d681SAndroid Build Coastguard Worker if (IsSplat) {
3150*9880d681SAndroid Build Coastguard Worker // Undef bits can contribute to a possible optimisation if set, so
3151*9880d681SAndroid Build Coastguard Worker // set them.
3152*9880d681SAndroid Build Coastguard Worker SplatValue |= SplatUndef;
3153*9880d681SAndroid Build Coastguard Worker
3154*9880d681SAndroid Build Coastguard Worker // The splat value may be something like "0x00FFFFFF", which means 0 for
3155*9880d681SAndroid Build Coastguard Worker // the first vector value and FF for the rest, repeating. We need a mask
3156*9880d681SAndroid Build Coastguard Worker // that will apply equally to all members of the vector, so AND all the
3157*9880d681SAndroid Build Coastguard Worker // lanes of the constant together.
3158*9880d681SAndroid Build Coastguard Worker EVT VT = Vector->getValueType(0);
3159*9880d681SAndroid Build Coastguard Worker unsigned BitWidth = VT.getVectorElementType().getSizeInBits();
3160*9880d681SAndroid Build Coastguard Worker
3161*9880d681SAndroid Build Coastguard Worker // If the splat value has been compressed to a bitlength lower
3162*9880d681SAndroid Build Coastguard Worker // than the size of the vector lane, we need to re-expand it to
3163*9880d681SAndroid Build Coastguard Worker // the lane size.
3164*9880d681SAndroid Build Coastguard Worker if (BitWidth > SplatBitSize)
3165*9880d681SAndroid Build Coastguard Worker for (SplatValue = SplatValue.zextOrTrunc(BitWidth);
3166*9880d681SAndroid Build Coastguard Worker SplatBitSize < BitWidth;
3167*9880d681SAndroid Build Coastguard Worker SplatBitSize = SplatBitSize * 2)
3168*9880d681SAndroid Build Coastguard Worker SplatValue |= SplatValue.shl(SplatBitSize);
3169*9880d681SAndroid Build Coastguard Worker
3170*9880d681SAndroid Build Coastguard Worker // Make sure that variable 'Constant' is only set if 'SplatBitSize' is a
3171*9880d681SAndroid Build Coastguard Worker // multiple of 'BitWidth'. Otherwise, we could propagate a wrong value.
3172*9880d681SAndroid Build Coastguard Worker if (SplatBitSize % BitWidth == 0) {
3173*9880d681SAndroid Build Coastguard Worker Constant = APInt::getAllOnesValue(BitWidth);
3174*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0, n = SplatBitSize/BitWidth; i < n; ++i)
3175*9880d681SAndroid Build Coastguard Worker Constant &= SplatValue.lshr(i*BitWidth).zextOrTrunc(BitWidth);
3176*9880d681SAndroid Build Coastguard Worker }
3177*9880d681SAndroid Build Coastguard Worker }
3178*9880d681SAndroid Build Coastguard Worker }
3179*9880d681SAndroid Build Coastguard Worker
3180*9880d681SAndroid Build Coastguard Worker // If we want to change an EXTLOAD to a ZEXTLOAD, ensure a ZEXTLOAD is
3181*9880d681SAndroid Build Coastguard Worker // actually legal and isn't going to get expanded, else this is a false
3182*9880d681SAndroid Build Coastguard Worker // optimisation.
3183*9880d681SAndroid Build Coastguard Worker bool CanZextLoadProfitably = TLI.isLoadExtLegal(ISD::ZEXTLOAD,
3184*9880d681SAndroid Build Coastguard Worker Load->getValueType(0),
3185*9880d681SAndroid Build Coastguard Worker Load->getMemoryVT());
3186*9880d681SAndroid Build Coastguard Worker
3187*9880d681SAndroid Build Coastguard Worker // Resize the constant to the same size as the original memory access before
3188*9880d681SAndroid Build Coastguard Worker // extension. If it is still the AllOnesValue then this AND is completely
3189*9880d681SAndroid Build Coastguard Worker // unneeded.
3190*9880d681SAndroid Build Coastguard Worker Constant =
3191*9880d681SAndroid Build Coastguard Worker Constant.zextOrTrunc(Load->getMemoryVT().getScalarType().getSizeInBits());
3192*9880d681SAndroid Build Coastguard Worker
3193*9880d681SAndroid Build Coastguard Worker bool B;
3194*9880d681SAndroid Build Coastguard Worker switch (Load->getExtensionType()) {
3195*9880d681SAndroid Build Coastguard Worker default: B = false; break;
3196*9880d681SAndroid Build Coastguard Worker case ISD::EXTLOAD: B = CanZextLoadProfitably; break;
3197*9880d681SAndroid Build Coastguard Worker case ISD::ZEXTLOAD:
3198*9880d681SAndroid Build Coastguard Worker case ISD::NON_EXTLOAD: B = true; break;
3199*9880d681SAndroid Build Coastguard Worker }
3200*9880d681SAndroid Build Coastguard Worker
3201*9880d681SAndroid Build Coastguard Worker if (B && Constant.isAllOnesValue()) {
3202*9880d681SAndroid Build Coastguard Worker // If the load type was an EXTLOAD, convert to ZEXTLOAD in order to
3203*9880d681SAndroid Build Coastguard Worker // preserve semantics once we get rid of the AND.
3204*9880d681SAndroid Build Coastguard Worker SDValue NewLoad(Load, 0);
3205*9880d681SAndroid Build Coastguard Worker if (Load->getExtensionType() == ISD::EXTLOAD) {
3206*9880d681SAndroid Build Coastguard Worker NewLoad = DAG.getLoad(Load->getAddressingMode(), ISD::ZEXTLOAD,
3207*9880d681SAndroid Build Coastguard Worker Load->getValueType(0), SDLoc(Load),
3208*9880d681SAndroid Build Coastguard Worker Load->getChain(), Load->getBasePtr(),
3209*9880d681SAndroid Build Coastguard Worker Load->getOffset(), Load->getMemoryVT(),
3210*9880d681SAndroid Build Coastguard Worker Load->getMemOperand());
3211*9880d681SAndroid Build Coastguard Worker // Replace uses of the EXTLOAD with the new ZEXTLOAD.
3212*9880d681SAndroid Build Coastguard Worker if (Load->getNumValues() == 3) {
3213*9880d681SAndroid Build Coastguard Worker // PRE/POST_INC loads have 3 values.
3214*9880d681SAndroid Build Coastguard Worker SDValue To[] = { NewLoad.getValue(0), NewLoad.getValue(1),
3215*9880d681SAndroid Build Coastguard Worker NewLoad.getValue(2) };
3216*9880d681SAndroid Build Coastguard Worker CombineTo(Load, To, 3, true);
3217*9880d681SAndroid Build Coastguard Worker } else {
3218*9880d681SAndroid Build Coastguard Worker CombineTo(Load, NewLoad.getValue(0), NewLoad.getValue(1));
3219*9880d681SAndroid Build Coastguard Worker }
3220*9880d681SAndroid Build Coastguard Worker }
3221*9880d681SAndroid Build Coastguard Worker
3222*9880d681SAndroid Build Coastguard Worker // Fold the AND away, taking care not to fold to the old load node if we
3223*9880d681SAndroid Build Coastguard Worker // replaced it.
3224*9880d681SAndroid Build Coastguard Worker CombineTo(N, (N0.getNode() == Load) ? NewLoad : N0);
3225*9880d681SAndroid Build Coastguard Worker
3226*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
3227*9880d681SAndroid Build Coastguard Worker }
3228*9880d681SAndroid Build Coastguard Worker }
3229*9880d681SAndroid Build Coastguard Worker
3230*9880d681SAndroid Build Coastguard Worker // fold (and (load x), 255) -> (zextload x, i8)
3231*9880d681SAndroid Build Coastguard Worker // fold (and (extload x, i16), 255) -> (zextload x, i8)
3232*9880d681SAndroid Build Coastguard Worker // fold (and (any_ext (extload x, i16)), 255) -> (zextload x, i8)
3233*9880d681SAndroid Build Coastguard Worker if (N1C && (N0.getOpcode() == ISD::LOAD ||
3234*9880d681SAndroid Build Coastguard Worker (N0.getOpcode() == ISD::ANY_EXTEND &&
3235*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getOpcode() == ISD::LOAD))) {
3236*9880d681SAndroid Build Coastguard Worker bool HasAnyExt = N0.getOpcode() == ISD::ANY_EXTEND;
3237*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = HasAnyExt
3238*9880d681SAndroid Build Coastguard Worker ? cast<LoadSDNode>(N0.getOperand(0))
3239*9880d681SAndroid Build Coastguard Worker : cast<LoadSDNode>(N0);
3240*9880d681SAndroid Build Coastguard Worker if (LN0->getExtensionType() != ISD::SEXTLOAD &&
3241*9880d681SAndroid Build Coastguard Worker LN0->isUnindexed() && N0.hasOneUse() && SDValue(LN0, 0).hasOneUse()) {
3242*9880d681SAndroid Build Coastguard Worker auto NarrowLoad = false;
3243*9880d681SAndroid Build Coastguard Worker EVT LoadResultTy = HasAnyExt ? LN0->getValueType(0) : VT;
3244*9880d681SAndroid Build Coastguard Worker EVT ExtVT, LoadedVT;
3245*9880d681SAndroid Build Coastguard Worker if (isAndLoadExtLoad(N1C, LN0, LoadResultTy, ExtVT, LoadedVT,
3246*9880d681SAndroid Build Coastguard Worker NarrowLoad)) {
3247*9880d681SAndroid Build Coastguard Worker if (!NarrowLoad) {
3248*9880d681SAndroid Build Coastguard Worker SDValue NewLoad =
3249*9880d681SAndroid Build Coastguard Worker DAG.getExtLoad(ISD::ZEXTLOAD, SDLoc(LN0), LoadResultTy,
3250*9880d681SAndroid Build Coastguard Worker LN0->getChain(), LN0->getBasePtr(), ExtVT,
3251*9880d681SAndroid Build Coastguard Worker LN0->getMemOperand());
3252*9880d681SAndroid Build Coastguard Worker AddToWorklist(N);
3253*9880d681SAndroid Build Coastguard Worker CombineTo(LN0, NewLoad, NewLoad.getValue(1));
3254*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
3255*9880d681SAndroid Build Coastguard Worker } else {
3256*9880d681SAndroid Build Coastguard Worker EVT PtrType = LN0->getOperand(1).getValueType();
3257*9880d681SAndroid Build Coastguard Worker
3258*9880d681SAndroid Build Coastguard Worker unsigned Alignment = LN0->getAlignment();
3259*9880d681SAndroid Build Coastguard Worker SDValue NewPtr = LN0->getBasePtr();
3260*9880d681SAndroid Build Coastguard Worker
3261*9880d681SAndroid Build Coastguard Worker // For big endian targets, we need to add an offset to the pointer
3262*9880d681SAndroid Build Coastguard Worker // to load the correct bytes. For little endian systems, we merely
3263*9880d681SAndroid Build Coastguard Worker // need to read fewer bytes from the same pointer.
3264*9880d681SAndroid Build Coastguard Worker if (DAG.getDataLayout().isBigEndian()) {
3265*9880d681SAndroid Build Coastguard Worker unsigned LVTStoreBytes = LoadedVT.getStoreSize();
3266*9880d681SAndroid Build Coastguard Worker unsigned EVTStoreBytes = ExtVT.getStoreSize();
3267*9880d681SAndroid Build Coastguard Worker unsigned PtrOff = LVTStoreBytes - EVTStoreBytes;
3268*9880d681SAndroid Build Coastguard Worker SDLoc DL(LN0);
3269*9880d681SAndroid Build Coastguard Worker NewPtr = DAG.getNode(ISD::ADD, DL, PtrType,
3270*9880d681SAndroid Build Coastguard Worker NewPtr, DAG.getConstant(PtrOff, DL, PtrType));
3271*9880d681SAndroid Build Coastguard Worker Alignment = MinAlign(Alignment, PtrOff);
3272*9880d681SAndroid Build Coastguard Worker }
3273*9880d681SAndroid Build Coastguard Worker
3274*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewPtr.getNode());
3275*9880d681SAndroid Build Coastguard Worker
3276*9880d681SAndroid Build Coastguard Worker SDValue Load =
3277*9880d681SAndroid Build Coastguard Worker DAG.getExtLoad(ISD::ZEXTLOAD, SDLoc(LN0), LoadResultTy,
3278*9880d681SAndroid Build Coastguard Worker LN0->getChain(), NewPtr,
3279*9880d681SAndroid Build Coastguard Worker LN0->getPointerInfo(),
3280*9880d681SAndroid Build Coastguard Worker ExtVT, LN0->isVolatile(), LN0->isNonTemporal(),
3281*9880d681SAndroid Build Coastguard Worker LN0->isInvariant(), Alignment, LN0->getAAInfo());
3282*9880d681SAndroid Build Coastguard Worker AddToWorklist(N);
3283*9880d681SAndroid Build Coastguard Worker CombineTo(LN0, Load, Load.getValue(1));
3284*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
3285*9880d681SAndroid Build Coastguard Worker }
3286*9880d681SAndroid Build Coastguard Worker }
3287*9880d681SAndroid Build Coastguard Worker }
3288*9880d681SAndroid Build Coastguard Worker }
3289*9880d681SAndroid Build Coastguard Worker
3290*9880d681SAndroid Build Coastguard Worker if (SDValue Combined = visitANDLike(N0, N1, N))
3291*9880d681SAndroid Build Coastguard Worker return Combined;
3292*9880d681SAndroid Build Coastguard Worker
3293*9880d681SAndroid Build Coastguard Worker // Simplify: (and (op x...), (op y...)) -> (op (and x, y))
3294*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == N1.getOpcode())
3295*9880d681SAndroid Build Coastguard Worker if (SDValue Tmp = SimplifyBinOpWithSameOpcodeHands(N))
3296*9880d681SAndroid Build Coastguard Worker return Tmp;
3297*9880d681SAndroid Build Coastguard Worker
3298*9880d681SAndroid Build Coastguard Worker // fold (and (sign_extend_inreg x, i16 to i32), 1) -> (and x, 1)
3299*9880d681SAndroid Build Coastguard Worker // fold (and (sra)) -> (and (srl)) when possible.
3300*9880d681SAndroid Build Coastguard Worker if (!VT.isVector() &&
3301*9880d681SAndroid Build Coastguard Worker SimplifyDemandedBits(SDValue(N, 0)))
3302*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0);
3303*9880d681SAndroid Build Coastguard Worker
3304*9880d681SAndroid Build Coastguard Worker // fold (zext_inreg (extload x)) -> (zextload x)
3305*9880d681SAndroid Build Coastguard Worker if (ISD::isEXTLoad(N0.getNode()) && ISD::isUNINDEXEDLoad(N0.getNode())) {
3306*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = cast<LoadSDNode>(N0);
3307*9880d681SAndroid Build Coastguard Worker EVT MemVT = LN0->getMemoryVT();
3308*9880d681SAndroid Build Coastguard Worker // If we zero all the possible extended bits, then we can turn this into
3309*9880d681SAndroid Build Coastguard Worker // a zextload if we are running before legalize or the operation is legal.
3310*9880d681SAndroid Build Coastguard Worker unsigned BitWidth = N1.getValueType().getScalarType().getSizeInBits();
3311*9880d681SAndroid Build Coastguard Worker if (DAG.MaskedValueIsZero(N1, APInt::getHighBitsSet(BitWidth,
3312*9880d681SAndroid Build Coastguard Worker BitWidth - MemVT.getScalarType().getSizeInBits())) &&
3313*9880d681SAndroid Build Coastguard Worker ((!LegalOperations && !LN0->isVolatile()) ||
3314*9880d681SAndroid Build Coastguard Worker TLI.isLoadExtLegal(ISD::ZEXTLOAD, VT, MemVT))) {
3315*9880d681SAndroid Build Coastguard Worker SDValue ExtLoad = DAG.getExtLoad(ISD::ZEXTLOAD, SDLoc(N0), VT,
3316*9880d681SAndroid Build Coastguard Worker LN0->getChain(), LN0->getBasePtr(),
3317*9880d681SAndroid Build Coastguard Worker MemVT, LN0->getMemOperand());
3318*9880d681SAndroid Build Coastguard Worker AddToWorklist(N);
3319*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(), ExtLoad, ExtLoad.getValue(1));
3320*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
3321*9880d681SAndroid Build Coastguard Worker }
3322*9880d681SAndroid Build Coastguard Worker }
3323*9880d681SAndroid Build Coastguard Worker // fold (zext_inreg (sextload x)) -> (zextload x) iff load has one use
3324*9880d681SAndroid Build Coastguard Worker if (ISD::isSEXTLoad(N0.getNode()) && ISD::isUNINDEXEDLoad(N0.getNode()) &&
3325*9880d681SAndroid Build Coastguard Worker N0.hasOneUse()) {
3326*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = cast<LoadSDNode>(N0);
3327*9880d681SAndroid Build Coastguard Worker EVT MemVT = LN0->getMemoryVT();
3328*9880d681SAndroid Build Coastguard Worker // If we zero all the possible extended bits, then we can turn this into
3329*9880d681SAndroid Build Coastguard Worker // a zextload if we are running before legalize or the operation is legal.
3330*9880d681SAndroid Build Coastguard Worker unsigned BitWidth = N1.getValueType().getScalarType().getSizeInBits();
3331*9880d681SAndroid Build Coastguard Worker if (DAG.MaskedValueIsZero(N1, APInt::getHighBitsSet(BitWidth,
3332*9880d681SAndroid Build Coastguard Worker BitWidth - MemVT.getScalarType().getSizeInBits())) &&
3333*9880d681SAndroid Build Coastguard Worker ((!LegalOperations && !LN0->isVolatile()) ||
3334*9880d681SAndroid Build Coastguard Worker TLI.isLoadExtLegal(ISD::ZEXTLOAD, VT, MemVT))) {
3335*9880d681SAndroid Build Coastguard Worker SDValue ExtLoad = DAG.getExtLoad(ISD::ZEXTLOAD, SDLoc(N0), VT,
3336*9880d681SAndroid Build Coastguard Worker LN0->getChain(), LN0->getBasePtr(),
3337*9880d681SAndroid Build Coastguard Worker MemVT, LN0->getMemOperand());
3338*9880d681SAndroid Build Coastguard Worker AddToWorklist(N);
3339*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(), ExtLoad, ExtLoad.getValue(1));
3340*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
3341*9880d681SAndroid Build Coastguard Worker }
3342*9880d681SAndroid Build Coastguard Worker }
3343*9880d681SAndroid Build Coastguard Worker // fold (and (or (srl N, 8), (shl N, 8)), 0xffff) -> (srl (bswap N), const)
3344*9880d681SAndroid Build Coastguard Worker if (N1C && N1C->getAPIntValue() == 0xffff && N0.getOpcode() == ISD::OR) {
3345*9880d681SAndroid Build Coastguard Worker if (SDValue BSwap = MatchBSwapHWordLow(N0.getNode(), N0.getOperand(0),
3346*9880d681SAndroid Build Coastguard Worker N0.getOperand(1), false))
3347*9880d681SAndroid Build Coastguard Worker return BSwap;
3348*9880d681SAndroid Build Coastguard Worker }
3349*9880d681SAndroid Build Coastguard Worker
3350*9880d681SAndroid Build Coastguard Worker return SDValue();
3351*9880d681SAndroid Build Coastguard Worker }
3352*9880d681SAndroid Build Coastguard Worker
3353*9880d681SAndroid Build Coastguard Worker /// Match (a >> 8) | (a << 8) as (bswap a) >> 16.
MatchBSwapHWordLow(SDNode * N,SDValue N0,SDValue N1,bool DemandHighBits)3354*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::MatchBSwapHWordLow(SDNode *N, SDValue N0, SDValue N1,
3355*9880d681SAndroid Build Coastguard Worker bool DemandHighBits) {
3356*9880d681SAndroid Build Coastguard Worker if (!LegalOperations)
3357*9880d681SAndroid Build Coastguard Worker return SDValue();
3358*9880d681SAndroid Build Coastguard Worker
3359*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
3360*9880d681SAndroid Build Coastguard Worker if (VT != MVT::i64 && VT != MVT::i32 && VT != MVT::i16)
3361*9880d681SAndroid Build Coastguard Worker return SDValue();
3362*9880d681SAndroid Build Coastguard Worker if (!TLI.isOperationLegal(ISD::BSWAP, VT))
3363*9880d681SAndroid Build Coastguard Worker return SDValue();
3364*9880d681SAndroid Build Coastguard Worker
3365*9880d681SAndroid Build Coastguard Worker // Recognize (and (shl a, 8), 0xff), (and (srl a, 8), 0xff00)
3366*9880d681SAndroid Build Coastguard Worker bool LookPassAnd0 = false;
3367*9880d681SAndroid Build Coastguard Worker bool LookPassAnd1 = false;
3368*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::AND && N0.getOperand(0).getOpcode() == ISD::SRL)
3369*9880d681SAndroid Build Coastguard Worker std::swap(N0, N1);
3370*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::AND && N1.getOperand(0).getOpcode() == ISD::SHL)
3371*9880d681SAndroid Build Coastguard Worker std::swap(N0, N1);
3372*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::AND) {
3373*9880d681SAndroid Build Coastguard Worker if (!N0.getNode()->hasOneUse())
3374*9880d681SAndroid Build Coastguard Worker return SDValue();
3375*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N01C = dyn_cast<ConstantSDNode>(N0.getOperand(1));
3376*9880d681SAndroid Build Coastguard Worker if (!N01C || N01C->getZExtValue() != 0xFF00)
3377*9880d681SAndroid Build Coastguard Worker return SDValue();
3378*9880d681SAndroid Build Coastguard Worker N0 = N0.getOperand(0);
3379*9880d681SAndroid Build Coastguard Worker LookPassAnd0 = true;
3380*9880d681SAndroid Build Coastguard Worker }
3381*9880d681SAndroid Build Coastguard Worker
3382*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::AND) {
3383*9880d681SAndroid Build Coastguard Worker if (!N1.getNode()->hasOneUse())
3384*9880d681SAndroid Build Coastguard Worker return SDValue();
3385*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N11C = dyn_cast<ConstantSDNode>(N1.getOperand(1));
3386*9880d681SAndroid Build Coastguard Worker if (!N11C || N11C->getZExtValue() != 0xFF)
3387*9880d681SAndroid Build Coastguard Worker return SDValue();
3388*9880d681SAndroid Build Coastguard Worker N1 = N1.getOperand(0);
3389*9880d681SAndroid Build Coastguard Worker LookPassAnd1 = true;
3390*9880d681SAndroid Build Coastguard Worker }
3391*9880d681SAndroid Build Coastguard Worker
3392*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SRL && N1.getOpcode() == ISD::SHL)
3393*9880d681SAndroid Build Coastguard Worker std::swap(N0, N1);
3394*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() != ISD::SHL || N1.getOpcode() != ISD::SRL)
3395*9880d681SAndroid Build Coastguard Worker return SDValue();
3396*9880d681SAndroid Build Coastguard Worker if (!N0.getNode()->hasOneUse() ||
3397*9880d681SAndroid Build Coastguard Worker !N1.getNode()->hasOneUse())
3398*9880d681SAndroid Build Coastguard Worker return SDValue();
3399*9880d681SAndroid Build Coastguard Worker
3400*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N01C = dyn_cast<ConstantSDNode>(N0.getOperand(1));
3401*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N11C = dyn_cast<ConstantSDNode>(N1.getOperand(1));
3402*9880d681SAndroid Build Coastguard Worker if (!N01C || !N11C)
3403*9880d681SAndroid Build Coastguard Worker return SDValue();
3404*9880d681SAndroid Build Coastguard Worker if (N01C->getZExtValue() != 8 || N11C->getZExtValue() != 8)
3405*9880d681SAndroid Build Coastguard Worker return SDValue();
3406*9880d681SAndroid Build Coastguard Worker
3407*9880d681SAndroid Build Coastguard Worker // Look for (shl (and a, 0xff), 8), (srl (and a, 0xff00), 8)
3408*9880d681SAndroid Build Coastguard Worker SDValue N00 = N0->getOperand(0);
3409*9880d681SAndroid Build Coastguard Worker if (!LookPassAnd0 && N00.getOpcode() == ISD::AND) {
3410*9880d681SAndroid Build Coastguard Worker if (!N00.getNode()->hasOneUse())
3411*9880d681SAndroid Build Coastguard Worker return SDValue();
3412*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N001C = dyn_cast<ConstantSDNode>(N00.getOperand(1));
3413*9880d681SAndroid Build Coastguard Worker if (!N001C || N001C->getZExtValue() != 0xFF)
3414*9880d681SAndroid Build Coastguard Worker return SDValue();
3415*9880d681SAndroid Build Coastguard Worker N00 = N00.getOperand(0);
3416*9880d681SAndroid Build Coastguard Worker LookPassAnd0 = true;
3417*9880d681SAndroid Build Coastguard Worker }
3418*9880d681SAndroid Build Coastguard Worker
3419*9880d681SAndroid Build Coastguard Worker SDValue N10 = N1->getOperand(0);
3420*9880d681SAndroid Build Coastguard Worker if (!LookPassAnd1 && N10.getOpcode() == ISD::AND) {
3421*9880d681SAndroid Build Coastguard Worker if (!N10.getNode()->hasOneUse())
3422*9880d681SAndroid Build Coastguard Worker return SDValue();
3423*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N101C = dyn_cast<ConstantSDNode>(N10.getOperand(1));
3424*9880d681SAndroid Build Coastguard Worker if (!N101C || N101C->getZExtValue() != 0xFF00)
3425*9880d681SAndroid Build Coastguard Worker return SDValue();
3426*9880d681SAndroid Build Coastguard Worker N10 = N10.getOperand(0);
3427*9880d681SAndroid Build Coastguard Worker LookPassAnd1 = true;
3428*9880d681SAndroid Build Coastguard Worker }
3429*9880d681SAndroid Build Coastguard Worker
3430*9880d681SAndroid Build Coastguard Worker if (N00 != N10)
3431*9880d681SAndroid Build Coastguard Worker return SDValue();
3432*9880d681SAndroid Build Coastguard Worker
3433*9880d681SAndroid Build Coastguard Worker // Make sure everything beyond the low halfword gets set to zero since the SRL
3434*9880d681SAndroid Build Coastguard Worker // 16 will clear the top bits.
3435*9880d681SAndroid Build Coastguard Worker unsigned OpSizeInBits = VT.getSizeInBits();
3436*9880d681SAndroid Build Coastguard Worker if (DemandHighBits && OpSizeInBits > 16) {
3437*9880d681SAndroid Build Coastguard Worker // If the left-shift isn't masked out then the only way this is a bswap is
3438*9880d681SAndroid Build Coastguard Worker // if all bits beyond the low 8 are 0. In that case the entire pattern
3439*9880d681SAndroid Build Coastguard Worker // reduces to a left shift anyway: leave it for other parts of the combiner.
3440*9880d681SAndroid Build Coastguard Worker if (!LookPassAnd0)
3441*9880d681SAndroid Build Coastguard Worker return SDValue();
3442*9880d681SAndroid Build Coastguard Worker
3443*9880d681SAndroid Build Coastguard Worker // However, if the right shift isn't masked out then it might be because
3444*9880d681SAndroid Build Coastguard Worker // it's not needed. See if we can spot that too.
3445*9880d681SAndroid Build Coastguard Worker if (!LookPassAnd1 &&
3446*9880d681SAndroid Build Coastguard Worker !DAG.MaskedValueIsZero(
3447*9880d681SAndroid Build Coastguard Worker N10, APInt::getHighBitsSet(OpSizeInBits, OpSizeInBits - 16)))
3448*9880d681SAndroid Build Coastguard Worker return SDValue();
3449*9880d681SAndroid Build Coastguard Worker }
3450*9880d681SAndroid Build Coastguard Worker
3451*9880d681SAndroid Build Coastguard Worker SDValue Res = DAG.getNode(ISD::BSWAP, SDLoc(N), VT, N00);
3452*9880d681SAndroid Build Coastguard Worker if (OpSizeInBits > 16) {
3453*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
3454*9880d681SAndroid Build Coastguard Worker Res = DAG.getNode(ISD::SRL, DL, VT, Res,
3455*9880d681SAndroid Build Coastguard Worker DAG.getConstant(OpSizeInBits - 16, DL,
3456*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(VT)));
3457*9880d681SAndroid Build Coastguard Worker }
3458*9880d681SAndroid Build Coastguard Worker return Res;
3459*9880d681SAndroid Build Coastguard Worker }
3460*9880d681SAndroid Build Coastguard Worker
3461*9880d681SAndroid Build Coastguard Worker /// Return true if the specified node is an element that makes up a 32-bit
3462*9880d681SAndroid Build Coastguard Worker /// packed halfword byteswap.
3463*9880d681SAndroid Build Coastguard Worker /// ((x & 0x000000ff) << 8) |
3464*9880d681SAndroid Build Coastguard Worker /// ((x & 0x0000ff00) >> 8) |
3465*9880d681SAndroid Build Coastguard Worker /// ((x & 0x00ff0000) << 8) |
3466*9880d681SAndroid Build Coastguard Worker /// ((x & 0xff000000) >> 8)
isBSwapHWordElement(SDValue N,MutableArrayRef<SDNode * > Parts)3467*9880d681SAndroid Build Coastguard Worker static bool isBSwapHWordElement(SDValue N, MutableArrayRef<SDNode *> Parts) {
3468*9880d681SAndroid Build Coastguard Worker if (!N.getNode()->hasOneUse())
3469*9880d681SAndroid Build Coastguard Worker return false;
3470*9880d681SAndroid Build Coastguard Worker
3471*9880d681SAndroid Build Coastguard Worker unsigned Opc = N.getOpcode();
3472*9880d681SAndroid Build Coastguard Worker if (Opc != ISD::AND && Opc != ISD::SHL && Opc != ISD::SRL)
3473*9880d681SAndroid Build Coastguard Worker return false;
3474*9880d681SAndroid Build Coastguard Worker
3475*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N.getOperand(1));
3476*9880d681SAndroid Build Coastguard Worker if (!N1C)
3477*9880d681SAndroid Build Coastguard Worker return false;
3478*9880d681SAndroid Build Coastguard Worker
3479*9880d681SAndroid Build Coastguard Worker unsigned Num;
3480*9880d681SAndroid Build Coastguard Worker switch (N1C->getZExtValue()) {
3481*9880d681SAndroid Build Coastguard Worker default:
3482*9880d681SAndroid Build Coastguard Worker return false;
3483*9880d681SAndroid Build Coastguard Worker case 0xFF: Num = 0; break;
3484*9880d681SAndroid Build Coastguard Worker case 0xFF00: Num = 1; break;
3485*9880d681SAndroid Build Coastguard Worker case 0xFF0000: Num = 2; break;
3486*9880d681SAndroid Build Coastguard Worker case 0xFF000000: Num = 3; break;
3487*9880d681SAndroid Build Coastguard Worker }
3488*9880d681SAndroid Build Coastguard Worker
3489*9880d681SAndroid Build Coastguard Worker // Look for (x & 0xff) << 8 as well as ((x << 8) & 0xff00).
3490*9880d681SAndroid Build Coastguard Worker SDValue N0 = N.getOperand(0);
3491*9880d681SAndroid Build Coastguard Worker if (Opc == ISD::AND) {
3492*9880d681SAndroid Build Coastguard Worker if (Num == 0 || Num == 2) {
3493*9880d681SAndroid Build Coastguard Worker // (x >> 8) & 0xff
3494*9880d681SAndroid Build Coastguard Worker // (x >> 8) & 0xff0000
3495*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() != ISD::SRL)
3496*9880d681SAndroid Build Coastguard Worker return false;
3497*9880d681SAndroid Build Coastguard Worker ConstantSDNode *C = dyn_cast<ConstantSDNode>(N0.getOperand(1));
3498*9880d681SAndroid Build Coastguard Worker if (!C || C->getZExtValue() != 8)
3499*9880d681SAndroid Build Coastguard Worker return false;
3500*9880d681SAndroid Build Coastguard Worker } else {
3501*9880d681SAndroid Build Coastguard Worker // (x << 8) & 0xff00
3502*9880d681SAndroid Build Coastguard Worker // (x << 8) & 0xff000000
3503*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() != ISD::SHL)
3504*9880d681SAndroid Build Coastguard Worker return false;
3505*9880d681SAndroid Build Coastguard Worker ConstantSDNode *C = dyn_cast<ConstantSDNode>(N0.getOperand(1));
3506*9880d681SAndroid Build Coastguard Worker if (!C || C->getZExtValue() != 8)
3507*9880d681SAndroid Build Coastguard Worker return false;
3508*9880d681SAndroid Build Coastguard Worker }
3509*9880d681SAndroid Build Coastguard Worker } else if (Opc == ISD::SHL) {
3510*9880d681SAndroid Build Coastguard Worker // (x & 0xff) << 8
3511*9880d681SAndroid Build Coastguard Worker // (x & 0xff0000) << 8
3512*9880d681SAndroid Build Coastguard Worker if (Num != 0 && Num != 2)
3513*9880d681SAndroid Build Coastguard Worker return false;
3514*9880d681SAndroid Build Coastguard Worker ConstantSDNode *C = dyn_cast<ConstantSDNode>(N.getOperand(1));
3515*9880d681SAndroid Build Coastguard Worker if (!C || C->getZExtValue() != 8)
3516*9880d681SAndroid Build Coastguard Worker return false;
3517*9880d681SAndroid Build Coastguard Worker } else { // Opc == ISD::SRL
3518*9880d681SAndroid Build Coastguard Worker // (x & 0xff00) >> 8
3519*9880d681SAndroid Build Coastguard Worker // (x & 0xff000000) >> 8
3520*9880d681SAndroid Build Coastguard Worker if (Num != 1 && Num != 3)
3521*9880d681SAndroid Build Coastguard Worker return false;
3522*9880d681SAndroid Build Coastguard Worker ConstantSDNode *C = dyn_cast<ConstantSDNode>(N.getOperand(1));
3523*9880d681SAndroid Build Coastguard Worker if (!C || C->getZExtValue() != 8)
3524*9880d681SAndroid Build Coastguard Worker return false;
3525*9880d681SAndroid Build Coastguard Worker }
3526*9880d681SAndroid Build Coastguard Worker
3527*9880d681SAndroid Build Coastguard Worker if (Parts[Num])
3528*9880d681SAndroid Build Coastguard Worker return false;
3529*9880d681SAndroid Build Coastguard Worker
3530*9880d681SAndroid Build Coastguard Worker Parts[Num] = N0.getOperand(0).getNode();
3531*9880d681SAndroid Build Coastguard Worker return true;
3532*9880d681SAndroid Build Coastguard Worker }
3533*9880d681SAndroid Build Coastguard Worker
3534*9880d681SAndroid Build Coastguard Worker /// Match a 32-bit packed halfword bswap. That is
3535*9880d681SAndroid Build Coastguard Worker /// ((x & 0x000000ff) << 8) |
3536*9880d681SAndroid Build Coastguard Worker /// ((x & 0x0000ff00) >> 8) |
3537*9880d681SAndroid Build Coastguard Worker /// ((x & 0x00ff0000) << 8) |
3538*9880d681SAndroid Build Coastguard Worker /// ((x & 0xff000000) >> 8)
3539*9880d681SAndroid Build Coastguard Worker /// => (rotl (bswap x), 16)
MatchBSwapHWord(SDNode * N,SDValue N0,SDValue N1)3540*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::MatchBSwapHWord(SDNode *N, SDValue N0, SDValue N1) {
3541*9880d681SAndroid Build Coastguard Worker if (!LegalOperations)
3542*9880d681SAndroid Build Coastguard Worker return SDValue();
3543*9880d681SAndroid Build Coastguard Worker
3544*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
3545*9880d681SAndroid Build Coastguard Worker if (VT != MVT::i32)
3546*9880d681SAndroid Build Coastguard Worker return SDValue();
3547*9880d681SAndroid Build Coastguard Worker if (!TLI.isOperationLegal(ISD::BSWAP, VT))
3548*9880d681SAndroid Build Coastguard Worker return SDValue();
3549*9880d681SAndroid Build Coastguard Worker
3550*9880d681SAndroid Build Coastguard Worker // Look for either
3551*9880d681SAndroid Build Coastguard Worker // (or (or (and), (and)), (or (and), (and)))
3552*9880d681SAndroid Build Coastguard Worker // (or (or (or (and), (and)), (and)), (and))
3553*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() != ISD::OR)
3554*9880d681SAndroid Build Coastguard Worker return SDValue();
3555*9880d681SAndroid Build Coastguard Worker SDValue N00 = N0.getOperand(0);
3556*9880d681SAndroid Build Coastguard Worker SDValue N01 = N0.getOperand(1);
3557*9880d681SAndroid Build Coastguard Worker SDNode *Parts[4] = {};
3558*9880d681SAndroid Build Coastguard Worker
3559*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::OR &&
3560*9880d681SAndroid Build Coastguard Worker N00.getNumOperands() == 2 && N01.getNumOperands() == 2) {
3561*9880d681SAndroid Build Coastguard Worker // (or (or (and), (and)), (or (and), (and)))
3562*9880d681SAndroid Build Coastguard Worker SDValue N000 = N00.getOperand(0);
3563*9880d681SAndroid Build Coastguard Worker if (!isBSwapHWordElement(N000, Parts))
3564*9880d681SAndroid Build Coastguard Worker return SDValue();
3565*9880d681SAndroid Build Coastguard Worker
3566*9880d681SAndroid Build Coastguard Worker SDValue N001 = N00.getOperand(1);
3567*9880d681SAndroid Build Coastguard Worker if (!isBSwapHWordElement(N001, Parts))
3568*9880d681SAndroid Build Coastguard Worker return SDValue();
3569*9880d681SAndroid Build Coastguard Worker SDValue N010 = N01.getOperand(0);
3570*9880d681SAndroid Build Coastguard Worker if (!isBSwapHWordElement(N010, Parts))
3571*9880d681SAndroid Build Coastguard Worker return SDValue();
3572*9880d681SAndroid Build Coastguard Worker SDValue N011 = N01.getOperand(1);
3573*9880d681SAndroid Build Coastguard Worker if (!isBSwapHWordElement(N011, Parts))
3574*9880d681SAndroid Build Coastguard Worker return SDValue();
3575*9880d681SAndroid Build Coastguard Worker } else {
3576*9880d681SAndroid Build Coastguard Worker // (or (or (or (and), (and)), (and)), (and))
3577*9880d681SAndroid Build Coastguard Worker if (!isBSwapHWordElement(N1, Parts))
3578*9880d681SAndroid Build Coastguard Worker return SDValue();
3579*9880d681SAndroid Build Coastguard Worker if (!isBSwapHWordElement(N01, Parts))
3580*9880d681SAndroid Build Coastguard Worker return SDValue();
3581*9880d681SAndroid Build Coastguard Worker if (N00.getOpcode() != ISD::OR)
3582*9880d681SAndroid Build Coastguard Worker return SDValue();
3583*9880d681SAndroid Build Coastguard Worker SDValue N000 = N00.getOperand(0);
3584*9880d681SAndroid Build Coastguard Worker if (!isBSwapHWordElement(N000, Parts))
3585*9880d681SAndroid Build Coastguard Worker return SDValue();
3586*9880d681SAndroid Build Coastguard Worker SDValue N001 = N00.getOperand(1);
3587*9880d681SAndroid Build Coastguard Worker if (!isBSwapHWordElement(N001, Parts))
3588*9880d681SAndroid Build Coastguard Worker return SDValue();
3589*9880d681SAndroid Build Coastguard Worker }
3590*9880d681SAndroid Build Coastguard Worker
3591*9880d681SAndroid Build Coastguard Worker // Make sure the parts are all coming from the same node.
3592*9880d681SAndroid Build Coastguard Worker if (Parts[0] != Parts[1] || Parts[0] != Parts[2] || Parts[0] != Parts[3])
3593*9880d681SAndroid Build Coastguard Worker return SDValue();
3594*9880d681SAndroid Build Coastguard Worker
3595*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
3596*9880d681SAndroid Build Coastguard Worker SDValue BSwap = DAG.getNode(ISD::BSWAP, DL, VT,
3597*9880d681SAndroid Build Coastguard Worker SDValue(Parts[0], 0));
3598*9880d681SAndroid Build Coastguard Worker
3599*9880d681SAndroid Build Coastguard Worker // Result of the bswap should be rotated by 16. If it's not legal, then
3600*9880d681SAndroid Build Coastguard Worker // do (x << 16) | (x >> 16).
3601*9880d681SAndroid Build Coastguard Worker SDValue ShAmt = DAG.getConstant(16, DL, getShiftAmountTy(VT));
3602*9880d681SAndroid Build Coastguard Worker if (TLI.isOperationLegalOrCustom(ISD::ROTL, VT))
3603*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ROTL, DL, VT, BSwap, ShAmt);
3604*9880d681SAndroid Build Coastguard Worker if (TLI.isOperationLegalOrCustom(ISD::ROTR, VT))
3605*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ROTR, DL, VT, BSwap, ShAmt);
3606*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::OR, DL, VT,
3607*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::SHL, DL, VT, BSwap, ShAmt),
3608*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::SRL, DL, VT, BSwap, ShAmt));
3609*9880d681SAndroid Build Coastguard Worker }
3610*9880d681SAndroid Build Coastguard Worker
3611*9880d681SAndroid Build Coastguard Worker /// This contains all DAGCombine rules which reduce two values combined by
3612*9880d681SAndroid Build Coastguard Worker /// an Or operation to a single value \see visitANDLike().
visitORLike(SDValue N0,SDValue N1,SDNode * LocReference)3613*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitORLike(SDValue N0, SDValue N1, SDNode *LocReference) {
3614*9880d681SAndroid Build Coastguard Worker EVT VT = N1.getValueType();
3615*9880d681SAndroid Build Coastguard Worker // fold (or x, undef) -> -1
3616*9880d681SAndroid Build Coastguard Worker if (!LegalOperations &&
3617*9880d681SAndroid Build Coastguard Worker (N0.isUndef() || N1.isUndef())) {
3618*9880d681SAndroid Build Coastguard Worker EVT EltVT = VT.isVector() ? VT.getVectorElementType() : VT;
3619*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(APInt::getAllOnesValue(EltVT.getSizeInBits()),
3620*9880d681SAndroid Build Coastguard Worker SDLoc(LocReference), VT);
3621*9880d681SAndroid Build Coastguard Worker }
3622*9880d681SAndroid Build Coastguard Worker // fold (or (setcc x), (setcc y)) -> (setcc (or x, y))
3623*9880d681SAndroid Build Coastguard Worker SDValue LL, LR, RL, RR, CC0, CC1;
3624*9880d681SAndroid Build Coastguard Worker if (isSetCCEquivalent(N0, LL, LR, CC0) && isSetCCEquivalent(N1, RL, RR, CC1)){
3625*9880d681SAndroid Build Coastguard Worker ISD::CondCode Op0 = cast<CondCodeSDNode>(CC0)->get();
3626*9880d681SAndroid Build Coastguard Worker ISD::CondCode Op1 = cast<CondCodeSDNode>(CC1)->get();
3627*9880d681SAndroid Build Coastguard Worker
3628*9880d681SAndroid Build Coastguard Worker if (LR == RR && Op0 == Op1 && LL.getValueType().isInteger()) {
3629*9880d681SAndroid Build Coastguard Worker // fold (or (setne X, 0), (setne Y, 0)) -> (setne (or X, Y), 0)
3630*9880d681SAndroid Build Coastguard Worker // fold (or (setlt X, 0), (setlt Y, 0)) -> (setne (or X, Y), 0)
3631*9880d681SAndroid Build Coastguard Worker if (isNullConstant(LR) && (Op1 == ISD::SETNE || Op1 == ISD::SETLT)) {
3632*9880d681SAndroid Build Coastguard Worker SDValue ORNode = DAG.getNode(ISD::OR, SDLoc(LR),
3633*9880d681SAndroid Build Coastguard Worker LR.getValueType(), LL, RL);
3634*9880d681SAndroid Build Coastguard Worker AddToWorklist(ORNode.getNode());
3635*9880d681SAndroid Build Coastguard Worker return DAG.getSetCC(SDLoc(LocReference), VT, ORNode, LR, Op1);
3636*9880d681SAndroid Build Coastguard Worker }
3637*9880d681SAndroid Build Coastguard Worker // fold (or (setne X, -1), (setne Y, -1)) -> (setne (and X, Y), -1)
3638*9880d681SAndroid Build Coastguard Worker // fold (or (setgt X, -1), (setgt Y -1)) -> (setgt (and X, Y), -1)
3639*9880d681SAndroid Build Coastguard Worker if (isAllOnesConstant(LR) && (Op1 == ISD::SETNE || Op1 == ISD::SETGT)) {
3640*9880d681SAndroid Build Coastguard Worker SDValue ANDNode = DAG.getNode(ISD::AND, SDLoc(LR),
3641*9880d681SAndroid Build Coastguard Worker LR.getValueType(), LL, RL);
3642*9880d681SAndroid Build Coastguard Worker AddToWorklist(ANDNode.getNode());
3643*9880d681SAndroid Build Coastguard Worker return DAG.getSetCC(SDLoc(LocReference), VT, ANDNode, LR, Op1);
3644*9880d681SAndroid Build Coastguard Worker }
3645*9880d681SAndroid Build Coastguard Worker }
3646*9880d681SAndroid Build Coastguard Worker // canonicalize equivalent to ll == rl
3647*9880d681SAndroid Build Coastguard Worker if (LL == RR && LR == RL) {
3648*9880d681SAndroid Build Coastguard Worker Op1 = ISD::getSetCCSwappedOperands(Op1);
3649*9880d681SAndroid Build Coastguard Worker std::swap(RL, RR);
3650*9880d681SAndroid Build Coastguard Worker }
3651*9880d681SAndroid Build Coastguard Worker if (LL == RL && LR == RR) {
3652*9880d681SAndroid Build Coastguard Worker bool isInteger = LL.getValueType().isInteger();
3653*9880d681SAndroid Build Coastguard Worker ISD::CondCode Result = ISD::getSetCCOrOperation(Op0, Op1, isInteger);
3654*9880d681SAndroid Build Coastguard Worker if (Result != ISD::SETCC_INVALID &&
3655*9880d681SAndroid Build Coastguard Worker (!LegalOperations ||
3656*9880d681SAndroid Build Coastguard Worker (TLI.isCondCodeLegal(Result, LL.getSimpleValueType()) &&
3657*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegal(ISD::SETCC, LL.getValueType())))) {
3658*9880d681SAndroid Build Coastguard Worker EVT CCVT = getSetCCResultType(LL.getValueType());
3659*9880d681SAndroid Build Coastguard Worker if (N0.getValueType() == CCVT ||
3660*9880d681SAndroid Build Coastguard Worker (!LegalOperations && N0.getValueType() == MVT::i1))
3661*9880d681SAndroid Build Coastguard Worker return DAG.getSetCC(SDLoc(LocReference), N0.getValueType(),
3662*9880d681SAndroid Build Coastguard Worker LL, LR, Result);
3663*9880d681SAndroid Build Coastguard Worker }
3664*9880d681SAndroid Build Coastguard Worker }
3665*9880d681SAndroid Build Coastguard Worker }
3666*9880d681SAndroid Build Coastguard Worker
3667*9880d681SAndroid Build Coastguard Worker // (or (and X, C1), (and Y, C2)) -> (and (or X, Y), C3) if possible.
3668*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::AND && N1.getOpcode() == ISD::AND &&
3669*9880d681SAndroid Build Coastguard Worker // Don't increase # computations.
3670*9880d681SAndroid Build Coastguard Worker (N0.getNode()->hasOneUse() || N1.getNode()->hasOneUse())) {
3671*9880d681SAndroid Build Coastguard Worker // We can only do this xform if we know that bits from X that are set in C2
3672*9880d681SAndroid Build Coastguard Worker // but not in C1 are already zero. Likewise for Y.
3673*9880d681SAndroid Build Coastguard Worker if (const ConstantSDNode *N0O1C =
3674*9880d681SAndroid Build Coastguard Worker getAsNonOpaqueConstant(N0.getOperand(1))) {
3675*9880d681SAndroid Build Coastguard Worker if (const ConstantSDNode *N1O1C =
3676*9880d681SAndroid Build Coastguard Worker getAsNonOpaqueConstant(N1.getOperand(1))) {
3677*9880d681SAndroid Build Coastguard Worker // We can only do this xform if we know that bits from X that are set in
3678*9880d681SAndroid Build Coastguard Worker // C2 but not in C1 are already zero. Likewise for Y.
3679*9880d681SAndroid Build Coastguard Worker const APInt &LHSMask = N0O1C->getAPIntValue();
3680*9880d681SAndroid Build Coastguard Worker const APInt &RHSMask = N1O1C->getAPIntValue();
3681*9880d681SAndroid Build Coastguard Worker
3682*9880d681SAndroid Build Coastguard Worker if (DAG.MaskedValueIsZero(N0.getOperand(0), RHSMask&~LHSMask) &&
3683*9880d681SAndroid Build Coastguard Worker DAG.MaskedValueIsZero(N1.getOperand(0), LHSMask&~RHSMask)) {
3684*9880d681SAndroid Build Coastguard Worker SDValue X = DAG.getNode(ISD::OR, SDLoc(N0), VT,
3685*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N1.getOperand(0));
3686*9880d681SAndroid Build Coastguard Worker SDLoc DL(LocReference);
3687*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, DL, VT, X,
3688*9880d681SAndroid Build Coastguard Worker DAG.getConstant(LHSMask | RHSMask, DL, VT));
3689*9880d681SAndroid Build Coastguard Worker }
3690*9880d681SAndroid Build Coastguard Worker }
3691*9880d681SAndroid Build Coastguard Worker }
3692*9880d681SAndroid Build Coastguard Worker }
3693*9880d681SAndroid Build Coastguard Worker
3694*9880d681SAndroid Build Coastguard Worker // (or (and X, M), (and X, N)) -> (and X, (or M, N))
3695*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::AND &&
3696*9880d681SAndroid Build Coastguard Worker N1.getOpcode() == ISD::AND &&
3697*9880d681SAndroid Build Coastguard Worker N0.getOperand(0) == N1.getOperand(0) &&
3698*9880d681SAndroid Build Coastguard Worker // Don't increase # computations.
3699*9880d681SAndroid Build Coastguard Worker (N0.getNode()->hasOneUse() || N1.getNode()->hasOneUse())) {
3700*9880d681SAndroid Build Coastguard Worker SDValue X = DAG.getNode(ISD::OR, SDLoc(N0), VT,
3701*9880d681SAndroid Build Coastguard Worker N0.getOperand(1), N1.getOperand(1));
3702*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, SDLoc(LocReference), VT, N0.getOperand(0), X);
3703*9880d681SAndroid Build Coastguard Worker }
3704*9880d681SAndroid Build Coastguard Worker
3705*9880d681SAndroid Build Coastguard Worker return SDValue();
3706*9880d681SAndroid Build Coastguard Worker }
3707*9880d681SAndroid Build Coastguard Worker
visitOR(SDNode * N)3708*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitOR(SDNode *N) {
3709*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
3710*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
3711*9880d681SAndroid Build Coastguard Worker EVT VT = N1.getValueType();
3712*9880d681SAndroid Build Coastguard Worker
3713*9880d681SAndroid Build Coastguard Worker // fold vector ops
3714*9880d681SAndroid Build Coastguard Worker if (VT.isVector()) {
3715*9880d681SAndroid Build Coastguard Worker if (SDValue FoldedVOp = SimplifyVBinOp(N))
3716*9880d681SAndroid Build Coastguard Worker return FoldedVOp;
3717*9880d681SAndroid Build Coastguard Worker
3718*9880d681SAndroid Build Coastguard Worker // fold (or x, 0) -> x, vector edition
3719*9880d681SAndroid Build Coastguard Worker if (ISD::isBuildVectorAllZeros(N0.getNode()))
3720*9880d681SAndroid Build Coastguard Worker return N1;
3721*9880d681SAndroid Build Coastguard Worker if (ISD::isBuildVectorAllZeros(N1.getNode()))
3722*9880d681SAndroid Build Coastguard Worker return N0;
3723*9880d681SAndroid Build Coastguard Worker
3724*9880d681SAndroid Build Coastguard Worker // fold (or x, -1) -> -1, vector edition
3725*9880d681SAndroid Build Coastguard Worker if (ISD::isBuildVectorAllOnes(N0.getNode()))
3726*9880d681SAndroid Build Coastguard Worker // do not return N0, because undef node may exist in N0
3727*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(
3728*9880d681SAndroid Build Coastguard Worker APInt::getAllOnesValue(
3729*9880d681SAndroid Build Coastguard Worker N0.getValueType().getScalarType().getSizeInBits()),
3730*9880d681SAndroid Build Coastguard Worker SDLoc(N), N0.getValueType());
3731*9880d681SAndroid Build Coastguard Worker if (ISD::isBuildVectorAllOnes(N1.getNode()))
3732*9880d681SAndroid Build Coastguard Worker // do not return N1, because undef node may exist in N1
3733*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(
3734*9880d681SAndroid Build Coastguard Worker APInt::getAllOnesValue(
3735*9880d681SAndroid Build Coastguard Worker N1.getValueType().getScalarType().getSizeInBits()),
3736*9880d681SAndroid Build Coastguard Worker SDLoc(N), N1.getValueType());
3737*9880d681SAndroid Build Coastguard Worker
3738*9880d681SAndroid Build Coastguard Worker // fold (or (shuf A, V_0, MA), (shuf B, V_0, MB)) -> (shuf A, B, Mask)
3739*9880d681SAndroid Build Coastguard Worker // Do this only if the resulting shuffle is legal.
3740*9880d681SAndroid Build Coastguard Worker if (isa<ShuffleVectorSDNode>(N0) &&
3741*9880d681SAndroid Build Coastguard Worker isa<ShuffleVectorSDNode>(N1) &&
3742*9880d681SAndroid Build Coastguard Worker // Avoid folding a node with illegal type.
3743*9880d681SAndroid Build Coastguard Worker TLI.isTypeLegal(VT)) {
3744*9880d681SAndroid Build Coastguard Worker bool ZeroN00 = ISD::isBuildVectorAllZeros(N0.getOperand(0).getNode());
3745*9880d681SAndroid Build Coastguard Worker bool ZeroN01 = ISD::isBuildVectorAllZeros(N0.getOperand(1).getNode());
3746*9880d681SAndroid Build Coastguard Worker bool ZeroN10 = ISD::isBuildVectorAllZeros(N1.getOperand(0).getNode());
3747*9880d681SAndroid Build Coastguard Worker bool ZeroN11 = ISD::isBuildVectorAllZeros(N1.getOperand(1).getNode());
3748*9880d681SAndroid Build Coastguard Worker // Ensure both shuffles have a zero input.
3749*9880d681SAndroid Build Coastguard Worker if ((ZeroN00 || ZeroN01) && (ZeroN10 || ZeroN11)) {
3750*9880d681SAndroid Build Coastguard Worker assert((!ZeroN00 || !ZeroN01) && "Both inputs zero!");
3751*9880d681SAndroid Build Coastguard Worker assert((!ZeroN10 || !ZeroN11) && "Both inputs zero!");
3752*9880d681SAndroid Build Coastguard Worker const ShuffleVectorSDNode *SV0 = cast<ShuffleVectorSDNode>(N0);
3753*9880d681SAndroid Build Coastguard Worker const ShuffleVectorSDNode *SV1 = cast<ShuffleVectorSDNode>(N1);
3754*9880d681SAndroid Build Coastguard Worker bool CanFold = true;
3755*9880d681SAndroid Build Coastguard Worker int NumElts = VT.getVectorNumElements();
3756*9880d681SAndroid Build Coastguard Worker SmallVector<int, 4> Mask(NumElts);
3757*9880d681SAndroid Build Coastguard Worker
3758*9880d681SAndroid Build Coastguard Worker for (int i = 0; i != NumElts; ++i) {
3759*9880d681SAndroid Build Coastguard Worker int M0 = SV0->getMaskElt(i);
3760*9880d681SAndroid Build Coastguard Worker int M1 = SV1->getMaskElt(i);
3761*9880d681SAndroid Build Coastguard Worker
3762*9880d681SAndroid Build Coastguard Worker // Determine if either index is pointing to a zero vector.
3763*9880d681SAndroid Build Coastguard Worker bool M0Zero = M0 < 0 || (ZeroN00 == (M0 < NumElts));
3764*9880d681SAndroid Build Coastguard Worker bool M1Zero = M1 < 0 || (ZeroN10 == (M1 < NumElts));
3765*9880d681SAndroid Build Coastguard Worker
3766*9880d681SAndroid Build Coastguard Worker // If one element is zero and the otherside is undef, keep undef.
3767*9880d681SAndroid Build Coastguard Worker // This also handles the case that both are undef.
3768*9880d681SAndroid Build Coastguard Worker if ((M0Zero && M1 < 0) || (M1Zero && M0 < 0)) {
3769*9880d681SAndroid Build Coastguard Worker Mask[i] = -1;
3770*9880d681SAndroid Build Coastguard Worker continue;
3771*9880d681SAndroid Build Coastguard Worker }
3772*9880d681SAndroid Build Coastguard Worker
3773*9880d681SAndroid Build Coastguard Worker // Make sure only one of the elements is zero.
3774*9880d681SAndroid Build Coastguard Worker if (M0Zero == M1Zero) {
3775*9880d681SAndroid Build Coastguard Worker CanFold = false;
3776*9880d681SAndroid Build Coastguard Worker break;
3777*9880d681SAndroid Build Coastguard Worker }
3778*9880d681SAndroid Build Coastguard Worker
3779*9880d681SAndroid Build Coastguard Worker assert((M0 >= 0 || M1 >= 0) && "Undef index!");
3780*9880d681SAndroid Build Coastguard Worker
3781*9880d681SAndroid Build Coastguard Worker // We have a zero and non-zero element. If the non-zero came from
3782*9880d681SAndroid Build Coastguard Worker // SV0 make the index a LHS index. If it came from SV1, make it
3783*9880d681SAndroid Build Coastguard Worker // a RHS index. We need to mod by NumElts because we don't care
3784*9880d681SAndroid Build Coastguard Worker // which operand it came from in the original shuffles.
3785*9880d681SAndroid Build Coastguard Worker Mask[i] = M1Zero ? M0 % NumElts : (M1 % NumElts) + NumElts;
3786*9880d681SAndroid Build Coastguard Worker }
3787*9880d681SAndroid Build Coastguard Worker
3788*9880d681SAndroid Build Coastguard Worker if (CanFold) {
3789*9880d681SAndroid Build Coastguard Worker SDValue NewLHS = ZeroN00 ? N0.getOperand(1) : N0.getOperand(0);
3790*9880d681SAndroid Build Coastguard Worker SDValue NewRHS = ZeroN10 ? N1.getOperand(1) : N1.getOperand(0);
3791*9880d681SAndroid Build Coastguard Worker
3792*9880d681SAndroid Build Coastguard Worker bool LegalMask = TLI.isShuffleMaskLegal(Mask, VT);
3793*9880d681SAndroid Build Coastguard Worker if (!LegalMask) {
3794*9880d681SAndroid Build Coastguard Worker std::swap(NewLHS, NewRHS);
3795*9880d681SAndroid Build Coastguard Worker ShuffleVectorSDNode::commuteMask(Mask);
3796*9880d681SAndroid Build Coastguard Worker LegalMask = TLI.isShuffleMaskLegal(Mask, VT);
3797*9880d681SAndroid Build Coastguard Worker }
3798*9880d681SAndroid Build Coastguard Worker
3799*9880d681SAndroid Build Coastguard Worker if (LegalMask)
3800*9880d681SAndroid Build Coastguard Worker return DAG.getVectorShuffle(VT, SDLoc(N), NewLHS, NewRHS, Mask);
3801*9880d681SAndroid Build Coastguard Worker }
3802*9880d681SAndroid Build Coastguard Worker }
3803*9880d681SAndroid Build Coastguard Worker }
3804*9880d681SAndroid Build Coastguard Worker }
3805*9880d681SAndroid Build Coastguard Worker
3806*9880d681SAndroid Build Coastguard Worker // fold (or c1, c2) -> c1|c2
3807*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N0C = getAsNonOpaqueConstant(N0);
3808*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
3809*9880d681SAndroid Build Coastguard Worker if (N0C && N1C && !N1C->isOpaque())
3810*9880d681SAndroid Build Coastguard Worker return DAG.FoldConstantArithmetic(ISD::OR, SDLoc(N), VT, N0C, N1C);
3811*9880d681SAndroid Build Coastguard Worker // canonicalize constant to RHS
3812*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N0) &&
3813*9880d681SAndroid Build Coastguard Worker !DAG.isConstantIntBuildVectorOrConstantInt(N1))
3814*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::OR, SDLoc(N), VT, N1, N0);
3815*9880d681SAndroid Build Coastguard Worker // fold (or x, 0) -> x
3816*9880d681SAndroid Build Coastguard Worker if (isNullConstant(N1))
3817*9880d681SAndroid Build Coastguard Worker return N0;
3818*9880d681SAndroid Build Coastguard Worker // fold (or x, -1) -> -1
3819*9880d681SAndroid Build Coastguard Worker if (isAllOnesConstant(N1))
3820*9880d681SAndroid Build Coastguard Worker return N1;
3821*9880d681SAndroid Build Coastguard Worker // fold (or x, c) -> c iff (x & ~c) == 0
3822*9880d681SAndroid Build Coastguard Worker if (N1C && DAG.MaskedValueIsZero(N0, ~N1C->getAPIntValue()))
3823*9880d681SAndroid Build Coastguard Worker return N1;
3824*9880d681SAndroid Build Coastguard Worker
3825*9880d681SAndroid Build Coastguard Worker if (SDValue Combined = visitORLike(N0, N1, N))
3826*9880d681SAndroid Build Coastguard Worker return Combined;
3827*9880d681SAndroid Build Coastguard Worker
3828*9880d681SAndroid Build Coastguard Worker // Recognize halfword bswaps as (bswap + rotl 16) or (bswap + shl 16)
3829*9880d681SAndroid Build Coastguard Worker if (SDValue BSwap = MatchBSwapHWord(N, N0, N1))
3830*9880d681SAndroid Build Coastguard Worker return BSwap;
3831*9880d681SAndroid Build Coastguard Worker if (SDValue BSwap = MatchBSwapHWordLow(N, N0, N1))
3832*9880d681SAndroid Build Coastguard Worker return BSwap;
3833*9880d681SAndroid Build Coastguard Worker
3834*9880d681SAndroid Build Coastguard Worker // reassociate or
3835*9880d681SAndroid Build Coastguard Worker if (SDValue ROR = ReassociateOps(ISD::OR, SDLoc(N), N0, N1))
3836*9880d681SAndroid Build Coastguard Worker return ROR;
3837*9880d681SAndroid Build Coastguard Worker // Canonicalize (or (and X, c1), c2) -> (and (or X, c2), c1|c2)
3838*9880d681SAndroid Build Coastguard Worker // iff (c1 & c2) == 0.
3839*9880d681SAndroid Build Coastguard Worker if (N1C && N0.getOpcode() == ISD::AND && N0.getNode()->hasOneUse() &&
3840*9880d681SAndroid Build Coastguard Worker isa<ConstantSDNode>(N0.getOperand(1))) {
3841*9880d681SAndroid Build Coastguard Worker ConstantSDNode *C1 = cast<ConstantSDNode>(N0.getOperand(1));
3842*9880d681SAndroid Build Coastguard Worker if ((C1->getAPIntValue() & N1C->getAPIntValue()) != 0) {
3843*9880d681SAndroid Build Coastguard Worker if (SDValue COR = DAG.FoldConstantArithmetic(ISD::OR, SDLoc(N1), VT,
3844*9880d681SAndroid Build Coastguard Worker N1C, C1))
3845*9880d681SAndroid Build Coastguard Worker return DAG.getNode(
3846*9880d681SAndroid Build Coastguard Worker ISD::AND, SDLoc(N), VT,
3847*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::OR, SDLoc(N0), VT, N0.getOperand(0), N1), COR);
3848*9880d681SAndroid Build Coastguard Worker return SDValue();
3849*9880d681SAndroid Build Coastguard Worker }
3850*9880d681SAndroid Build Coastguard Worker }
3851*9880d681SAndroid Build Coastguard Worker // Simplify: (or (op x...), (op y...)) -> (op (or x, y))
3852*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == N1.getOpcode())
3853*9880d681SAndroid Build Coastguard Worker if (SDValue Tmp = SimplifyBinOpWithSameOpcodeHands(N))
3854*9880d681SAndroid Build Coastguard Worker return Tmp;
3855*9880d681SAndroid Build Coastguard Worker
3856*9880d681SAndroid Build Coastguard Worker // See if this is some rotate idiom.
3857*9880d681SAndroid Build Coastguard Worker if (SDNode *Rot = MatchRotate(N0, N1, SDLoc(N)))
3858*9880d681SAndroid Build Coastguard Worker return SDValue(Rot, 0);
3859*9880d681SAndroid Build Coastguard Worker
3860*9880d681SAndroid Build Coastguard Worker // Simplify the operands using demanded-bits information.
3861*9880d681SAndroid Build Coastguard Worker if (!VT.isVector() &&
3862*9880d681SAndroid Build Coastguard Worker SimplifyDemandedBits(SDValue(N, 0)))
3863*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0);
3864*9880d681SAndroid Build Coastguard Worker
3865*9880d681SAndroid Build Coastguard Worker return SDValue();
3866*9880d681SAndroid Build Coastguard Worker }
3867*9880d681SAndroid Build Coastguard Worker
3868*9880d681SAndroid Build Coastguard Worker /// Match "(X shl/srl V1) & V2" where V2 may not be present.
MatchRotateHalf(SDValue Op,SDValue & Shift,SDValue & Mask)3869*9880d681SAndroid Build Coastguard Worker bool DAGCombiner::MatchRotateHalf(SDValue Op, SDValue &Shift, SDValue &Mask) {
3870*9880d681SAndroid Build Coastguard Worker if (Op.getOpcode() == ISD::AND) {
3871*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(Op.getOperand(1))) {
3872*9880d681SAndroid Build Coastguard Worker Mask = Op.getOperand(1);
3873*9880d681SAndroid Build Coastguard Worker Op = Op.getOperand(0);
3874*9880d681SAndroid Build Coastguard Worker } else {
3875*9880d681SAndroid Build Coastguard Worker return false;
3876*9880d681SAndroid Build Coastguard Worker }
3877*9880d681SAndroid Build Coastguard Worker }
3878*9880d681SAndroid Build Coastguard Worker
3879*9880d681SAndroid Build Coastguard Worker if (Op.getOpcode() == ISD::SRL || Op.getOpcode() == ISD::SHL) {
3880*9880d681SAndroid Build Coastguard Worker Shift = Op;
3881*9880d681SAndroid Build Coastguard Worker return true;
3882*9880d681SAndroid Build Coastguard Worker }
3883*9880d681SAndroid Build Coastguard Worker
3884*9880d681SAndroid Build Coastguard Worker return false;
3885*9880d681SAndroid Build Coastguard Worker }
3886*9880d681SAndroid Build Coastguard Worker
3887*9880d681SAndroid Build Coastguard Worker // Return true if we can prove that, whenever Neg and Pos are both in the
3888*9880d681SAndroid Build Coastguard Worker // range [0, EltSize), Neg == (Pos == 0 ? 0 : EltSize - Pos). This means that
3889*9880d681SAndroid Build Coastguard Worker // for two opposing shifts shift1 and shift2 and a value X with OpBits bits:
3890*9880d681SAndroid Build Coastguard Worker //
3891*9880d681SAndroid Build Coastguard Worker // (or (shift1 X, Neg), (shift2 X, Pos))
3892*9880d681SAndroid Build Coastguard Worker //
3893*9880d681SAndroid Build Coastguard Worker // reduces to a rotate in direction shift2 by Pos or (equivalently) a rotate
3894*9880d681SAndroid Build Coastguard Worker // in direction shift1 by Neg. The range [0, EltSize) means that we only need
3895*9880d681SAndroid Build Coastguard Worker // to consider shift amounts with defined behavior.
matchRotateSub(SDValue Pos,SDValue Neg,unsigned EltSize)3896*9880d681SAndroid Build Coastguard Worker static bool matchRotateSub(SDValue Pos, SDValue Neg, unsigned EltSize) {
3897*9880d681SAndroid Build Coastguard Worker // If EltSize is a power of 2 then:
3898*9880d681SAndroid Build Coastguard Worker //
3899*9880d681SAndroid Build Coastguard Worker // (a) (Pos == 0 ? 0 : EltSize - Pos) == (EltSize - Pos) & (EltSize - 1)
3900*9880d681SAndroid Build Coastguard Worker // (b) Neg == Neg & (EltSize - 1) whenever Neg is in [0, EltSize).
3901*9880d681SAndroid Build Coastguard Worker //
3902*9880d681SAndroid Build Coastguard Worker // So if EltSize is a power of 2 and Neg is (and Neg', EltSize-1), we check
3903*9880d681SAndroid Build Coastguard Worker // for the stronger condition:
3904*9880d681SAndroid Build Coastguard Worker //
3905*9880d681SAndroid Build Coastguard Worker // Neg & (EltSize - 1) == (EltSize - Pos) & (EltSize - 1) [A]
3906*9880d681SAndroid Build Coastguard Worker //
3907*9880d681SAndroid Build Coastguard Worker // for all Neg and Pos. Since Neg & (EltSize - 1) == Neg' & (EltSize - 1)
3908*9880d681SAndroid Build Coastguard Worker // we can just replace Neg with Neg' for the rest of the function.
3909*9880d681SAndroid Build Coastguard Worker //
3910*9880d681SAndroid Build Coastguard Worker // In other cases we check for the even stronger condition:
3911*9880d681SAndroid Build Coastguard Worker //
3912*9880d681SAndroid Build Coastguard Worker // Neg == EltSize - Pos [B]
3913*9880d681SAndroid Build Coastguard Worker //
3914*9880d681SAndroid Build Coastguard Worker // for all Neg and Pos. Note that the (or ...) then invokes undefined
3915*9880d681SAndroid Build Coastguard Worker // behavior if Pos == 0 (and consequently Neg == EltSize).
3916*9880d681SAndroid Build Coastguard Worker //
3917*9880d681SAndroid Build Coastguard Worker // We could actually use [A] whenever EltSize is a power of 2, but the
3918*9880d681SAndroid Build Coastguard Worker // only extra cases that it would match are those uninteresting ones
3919*9880d681SAndroid Build Coastguard Worker // where Neg and Pos are never in range at the same time. E.g. for
3920*9880d681SAndroid Build Coastguard Worker // EltSize == 32, using [A] would allow a Neg of the form (sub 64, Pos)
3921*9880d681SAndroid Build Coastguard Worker // as well as (sub 32, Pos), but:
3922*9880d681SAndroid Build Coastguard Worker //
3923*9880d681SAndroid Build Coastguard Worker // (or (shift1 X, (sub 64, Pos)), (shift2 X, Pos))
3924*9880d681SAndroid Build Coastguard Worker //
3925*9880d681SAndroid Build Coastguard Worker // always invokes undefined behavior for 32-bit X.
3926*9880d681SAndroid Build Coastguard Worker //
3927*9880d681SAndroid Build Coastguard Worker // Below, Mask == EltSize - 1 when using [A] and is all-ones otherwise.
3928*9880d681SAndroid Build Coastguard Worker unsigned MaskLoBits = 0;
3929*9880d681SAndroid Build Coastguard Worker if (Neg.getOpcode() == ISD::AND && isPowerOf2_64(EltSize)) {
3930*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *NegC = isConstOrConstSplat(Neg.getOperand(1))) {
3931*9880d681SAndroid Build Coastguard Worker if (NegC->getAPIntValue() == EltSize - 1) {
3932*9880d681SAndroid Build Coastguard Worker Neg = Neg.getOperand(0);
3933*9880d681SAndroid Build Coastguard Worker MaskLoBits = Log2_64(EltSize);
3934*9880d681SAndroid Build Coastguard Worker }
3935*9880d681SAndroid Build Coastguard Worker }
3936*9880d681SAndroid Build Coastguard Worker }
3937*9880d681SAndroid Build Coastguard Worker
3938*9880d681SAndroid Build Coastguard Worker // Check whether Neg has the form (sub NegC, NegOp1) for some NegC and NegOp1.
3939*9880d681SAndroid Build Coastguard Worker if (Neg.getOpcode() != ISD::SUB)
3940*9880d681SAndroid Build Coastguard Worker return false;
3941*9880d681SAndroid Build Coastguard Worker ConstantSDNode *NegC = isConstOrConstSplat(Neg.getOperand(0));
3942*9880d681SAndroid Build Coastguard Worker if (!NegC)
3943*9880d681SAndroid Build Coastguard Worker return false;
3944*9880d681SAndroid Build Coastguard Worker SDValue NegOp1 = Neg.getOperand(1);
3945*9880d681SAndroid Build Coastguard Worker
3946*9880d681SAndroid Build Coastguard Worker // On the RHS of [A], if Pos is Pos' & (EltSize - 1), just replace Pos with
3947*9880d681SAndroid Build Coastguard Worker // Pos'. The truncation is redundant for the purpose of the equality.
3948*9880d681SAndroid Build Coastguard Worker if (MaskLoBits && Pos.getOpcode() == ISD::AND)
3949*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *PosC = isConstOrConstSplat(Pos.getOperand(1)))
3950*9880d681SAndroid Build Coastguard Worker if (PosC->getAPIntValue() == EltSize - 1)
3951*9880d681SAndroid Build Coastguard Worker Pos = Pos.getOperand(0);
3952*9880d681SAndroid Build Coastguard Worker
3953*9880d681SAndroid Build Coastguard Worker // The condition we need is now:
3954*9880d681SAndroid Build Coastguard Worker //
3955*9880d681SAndroid Build Coastguard Worker // (NegC - NegOp1) & Mask == (EltSize - Pos) & Mask
3956*9880d681SAndroid Build Coastguard Worker //
3957*9880d681SAndroid Build Coastguard Worker // If NegOp1 == Pos then we need:
3958*9880d681SAndroid Build Coastguard Worker //
3959*9880d681SAndroid Build Coastguard Worker // EltSize & Mask == NegC & Mask
3960*9880d681SAndroid Build Coastguard Worker //
3961*9880d681SAndroid Build Coastguard Worker // (because "x & Mask" is a truncation and distributes through subtraction).
3962*9880d681SAndroid Build Coastguard Worker APInt Width;
3963*9880d681SAndroid Build Coastguard Worker if (Pos == NegOp1)
3964*9880d681SAndroid Build Coastguard Worker Width = NegC->getAPIntValue();
3965*9880d681SAndroid Build Coastguard Worker
3966*9880d681SAndroid Build Coastguard Worker // Check for cases where Pos has the form (add NegOp1, PosC) for some PosC.
3967*9880d681SAndroid Build Coastguard Worker // Then the condition we want to prove becomes:
3968*9880d681SAndroid Build Coastguard Worker //
3969*9880d681SAndroid Build Coastguard Worker // (NegC - NegOp1) & Mask == (EltSize - (NegOp1 + PosC)) & Mask
3970*9880d681SAndroid Build Coastguard Worker //
3971*9880d681SAndroid Build Coastguard Worker // which, again because "x & Mask" is a truncation, becomes:
3972*9880d681SAndroid Build Coastguard Worker //
3973*9880d681SAndroid Build Coastguard Worker // NegC & Mask == (EltSize - PosC) & Mask
3974*9880d681SAndroid Build Coastguard Worker // EltSize & Mask == (NegC + PosC) & Mask
3975*9880d681SAndroid Build Coastguard Worker else if (Pos.getOpcode() == ISD::ADD && Pos.getOperand(0) == NegOp1) {
3976*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *PosC = isConstOrConstSplat(Pos.getOperand(1)))
3977*9880d681SAndroid Build Coastguard Worker Width = PosC->getAPIntValue() + NegC->getAPIntValue();
3978*9880d681SAndroid Build Coastguard Worker else
3979*9880d681SAndroid Build Coastguard Worker return false;
3980*9880d681SAndroid Build Coastguard Worker } else
3981*9880d681SAndroid Build Coastguard Worker return false;
3982*9880d681SAndroid Build Coastguard Worker
3983*9880d681SAndroid Build Coastguard Worker // Now we just need to check that EltSize & Mask == Width & Mask.
3984*9880d681SAndroid Build Coastguard Worker if (MaskLoBits)
3985*9880d681SAndroid Build Coastguard Worker // EltSize & Mask is 0 since Mask is EltSize - 1.
3986*9880d681SAndroid Build Coastguard Worker return Width.getLoBits(MaskLoBits) == 0;
3987*9880d681SAndroid Build Coastguard Worker return Width == EltSize;
3988*9880d681SAndroid Build Coastguard Worker }
3989*9880d681SAndroid Build Coastguard Worker
3990*9880d681SAndroid Build Coastguard Worker // A subroutine of MatchRotate used once we have found an OR of two opposite
3991*9880d681SAndroid Build Coastguard Worker // shifts of Shifted. If Neg == <operand size> - Pos then the OR reduces
3992*9880d681SAndroid Build Coastguard Worker // to both (PosOpcode Shifted, Pos) and (NegOpcode Shifted, Neg), with the
3993*9880d681SAndroid Build Coastguard Worker // former being preferred if supported. InnerPos and InnerNeg are Pos and
3994*9880d681SAndroid Build Coastguard Worker // Neg with outer conversions stripped away.
MatchRotatePosNeg(SDValue Shifted,SDValue Pos,SDValue Neg,SDValue InnerPos,SDValue InnerNeg,unsigned PosOpcode,unsigned NegOpcode,const SDLoc & DL)3995*9880d681SAndroid Build Coastguard Worker SDNode *DAGCombiner::MatchRotatePosNeg(SDValue Shifted, SDValue Pos,
3996*9880d681SAndroid Build Coastguard Worker SDValue Neg, SDValue InnerPos,
3997*9880d681SAndroid Build Coastguard Worker SDValue InnerNeg, unsigned PosOpcode,
3998*9880d681SAndroid Build Coastguard Worker unsigned NegOpcode, const SDLoc &DL) {
3999*9880d681SAndroid Build Coastguard Worker // fold (or (shl x, (*ext y)),
4000*9880d681SAndroid Build Coastguard Worker // (srl x, (*ext (sub 32, y)))) ->
4001*9880d681SAndroid Build Coastguard Worker // (rotl x, y) or (rotr x, (sub 32, y))
4002*9880d681SAndroid Build Coastguard Worker //
4003*9880d681SAndroid Build Coastguard Worker // fold (or (shl x, (*ext (sub 32, y))),
4004*9880d681SAndroid Build Coastguard Worker // (srl x, (*ext y))) ->
4005*9880d681SAndroid Build Coastguard Worker // (rotr x, y) or (rotl x, (sub 32, y))
4006*9880d681SAndroid Build Coastguard Worker EVT VT = Shifted.getValueType();
4007*9880d681SAndroid Build Coastguard Worker if (matchRotateSub(InnerPos, InnerNeg, VT.getScalarSizeInBits())) {
4008*9880d681SAndroid Build Coastguard Worker bool HasPos = TLI.isOperationLegalOrCustom(PosOpcode, VT);
4009*9880d681SAndroid Build Coastguard Worker return DAG.getNode(HasPos ? PosOpcode : NegOpcode, DL, VT, Shifted,
4010*9880d681SAndroid Build Coastguard Worker HasPos ? Pos : Neg).getNode();
4011*9880d681SAndroid Build Coastguard Worker }
4012*9880d681SAndroid Build Coastguard Worker
4013*9880d681SAndroid Build Coastguard Worker return nullptr;
4014*9880d681SAndroid Build Coastguard Worker }
4015*9880d681SAndroid Build Coastguard Worker
4016*9880d681SAndroid Build Coastguard Worker // MatchRotate - Handle an 'or' of two operands. If this is one of the many
4017*9880d681SAndroid Build Coastguard Worker // idioms for rotate, and if the target supports rotation instructions, generate
4018*9880d681SAndroid Build Coastguard Worker // a rot[lr].
MatchRotate(SDValue LHS,SDValue RHS,const SDLoc & DL)4019*9880d681SAndroid Build Coastguard Worker SDNode *DAGCombiner::MatchRotate(SDValue LHS, SDValue RHS, const SDLoc &DL) {
4020*9880d681SAndroid Build Coastguard Worker // Must be a legal type. Expanded 'n promoted things won't work with rotates.
4021*9880d681SAndroid Build Coastguard Worker EVT VT = LHS.getValueType();
4022*9880d681SAndroid Build Coastguard Worker if (!TLI.isTypeLegal(VT)) return nullptr;
4023*9880d681SAndroid Build Coastguard Worker
4024*9880d681SAndroid Build Coastguard Worker // The target must have at least one rotate flavor.
4025*9880d681SAndroid Build Coastguard Worker bool HasROTL = TLI.isOperationLegalOrCustom(ISD::ROTL, VT);
4026*9880d681SAndroid Build Coastguard Worker bool HasROTR = TLI.isOperationLegalOrCustom(ISD::ROTR, VT);
4027*9880d681SAndroid Build Coastguard Worker if (!HasROTL && !HasROTR) return nullptr;
4028*9880d681SAndroid Build Coastguard Worker
4029*9880d681SAndroid Build Coastguard Worker // Match "(X shl/srl V1) & V2" where V2 may not be present.
4030*9880d681SAndroid Build Coastguard Worker SDValue LHSShift; // The shift.
4031*9880d681SAndroid Build Coastguard Worker SDValue LHSMask; // AND value if any.
4032*9880d681SAndroid Build Coastguard Worker if (!MatchRotateHalf(LHS, LHSShift, LHSMask))
4033*9880d681SAndroid Build Coastguard Worker return nullptr; // Not part of a rotate.
4034*9880d681SAndroid Build Coastguard Worker
4035*9880d681SAndroid Build Coastguard Worker SDValue RHSShift; // The shift.
4036*9880d681SAndroid Build Coastguard Worker SDValue RHSMask; // AND value if any.
4037*9880d681SAndroid Build Coastguard Worker if (!MatchRotateHalf(RHS, RHSShift, RHSMask))
4038*9880d681SAndroid Build Coastguard Worker return nullptr; // Not part of a rotate.
4039*9880d681SAndroid Build Coastguard Worker
4040*9880d681SAndroid Build Coastguard Worker if (LHSShift.getOperand(0) != RHSShift.getOperand(0))
4041*9880d681SAndroid Build Coastguard Worker return nullptr; // Not shifting the same value.
4042*9880d681SAndroid Build Coastguard Worker
4043*9880d681SAndroid Build Coastguard Worker if (LHSShift.getOpcode() == RHSShift.getOpcode())
4044*9880d681SAndroid Build Coastguard Worker return nullptr; // Shifts must disagree.
4045*9880d681SAndroid Build Coastguard Worker
4046*9880d681SAndroid Build Coastguard Worker // Canonicalize shl to left side in a shl/srl pair.
4047*9880d681SAndroid Build Coastguard Worker if (RHSShift.getOpcode() == ISD::SHL) {
4048*9880d681SAndroid Build Coastguard Worker std::swap(LHS, RHS);
4049*9880d681SAndroid Build Coastguard Worker std::swap(LHSShift, RHSShift);
4050*9880d681SAndroid Build Coastguard Worker std::swap(LHSMask, RHSMask);
4051*9880d681SAndroid Build Coastguard Worker }
4052*9880d681SAndroid Build Coastguard Worker
4053*9880d681SAndroid Build Coastguard Worker unsigned EltSizeInBits = VT.getScalarSizeInBits();
4054*9880d681SAndroid Build Coastguard Worker SDValue LHSShiftArg = LHSShift.getOperand(0);
4055*9880d681SAndroid Build Coastguard Worker SDValue LHSShiftAmt = LHSShift.getOperand(1);
4056*9880d681SAndroid Build Coastguard Worker SDValue RHSShiftArg = RHSShift.getOperand(0);
4057*9880d681SAndroid Build Coastguard Worker SDValue RHSShiftAmt = RHSShift.getOperand(1);
4058*9880d681SAndroid Build Coastguard Worker
4059*9880d681SAndroid Build Coastguard Worker // fold (or (shl x, C1), (srl x, C2)) -> (rotl x, C1)
4060*9880d681SAndroid Build Coastguard Worker // fold (or (shl x, C1), (srl x, C2)) -> (rotr x, C2)
4061*9880d681SAndroid Build Coastguard Worker if (isConstOrConstSplat(LHSShiftAmt) && isConstOrConstSplat(RHSShiftAmt)) {
4062*9880d681SAndroid Build Coastguard Worker uint64_t LShVal = isConstOrConstSplat(LHSShiftAmt)->getZExtValue();
4063*9880d681SAndroid Build Coastguard Worker uint64_t RShVal = isConstOrConstSplat(RHSShiftAmt)->getZExtValue();
4064*9880d681SAndroid Build Coastguard Worker if ((LShVal + RShVal) != EltSizeInBits)
4065*9880d681SAndroid Build Coastguard Worker return nullptr;
4066*9880d681SAndroid Build Coastguard Worker
4067*9880d681SAndroid Build Coastguard Worker SDValue Rot = DAG.getNode(HasROTL ? ISD::ROTL : ISD::ROTR, DL, VT,
4068*9880d681SAndroid Build Coastguard Worker LHSShiftArg, HasROTL ? LHSShiftAmt : RHSShiftAmt);
4069*9880d681SAndroid Build Coastguard Worker
4070*9880d681SAndroid Build Coastguard Worker // If there is an AND of either shifted operand, apply it to the result.
4071*9880d681SAndroid Build Coastguard Worker if (LHSMask.getNode() || RHSMask.getNode()) {
4072*9880d681SAndroid Build Coastguard Worker APInt AllBits = APInt::getAllOnesValue(EltSizeInBits);
4073*9880d681SAndroid Build Coastguard Worker SDValue Mask = DAG.getConstant(AllBits, DL, VT);
4074*9880d681SAndroid Build Coastguard Worker
4075*9880d681SAndroid Build Coastguard Worker if (LHSMask.getNode()) {
4076*9880d681SAndroid Build Coastguard Worker APInt RHSBits = APInt::getLowBitsSet(EltSizeInBits, LShVal);
4077*9880d681SAndroid Build Coastguard Worker Mask = DAG.getNode(ISD::AND, DL, VT, Mask,
4078*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::OR, DL, VT, LHSMask,
4079*9880d681SAndroid Build Coastguard Worker DAG.getConstant(RHSBits, DL, VT)));
4080*9880d681SAndroid Build Coastguard Worker }
4081*9880d681SAndroid Build Coastguard Worker if (RHSMask.getNode()) {
4082*9880d681SAndroid Build Coastguard Worker APInt LHSBits = APInt::getHighBitsSet(EltSizeInBits, RShVal);
4083*9880d681SAndroid Build Coastguard Worker Mask = DAG.getNode(ISD::AND, DL, VT, Mask,
4084*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::OR, DL, VT, RHSMask,
4085*9880d681SAndroid Build Coastguard Worker DAG.getConstant(LHSBits, DL, VT)));
4086*9880d681SAndroid Build Coastguard Worker }
4087*9880d681SAndroid Build Coastguard Worker
4088*9880d681SAndroid Build Coastguard Worker Rot = DAG.getNode(ISD::AND, DL, VT, Rot, Mask);
4089*9880d681SAndroid Build Coastguard Worker }
4090*9880d681SAndroid Build Coastguard Worker
4091*9880d681SAndroid Build Coastguard Worker return Rot.getNode();
4092*9880d681SAndroid Build Coastguard Worker }
4093*9880d681SAndroid Build Coastguard Worker
4094*9880d681SAndroid Build Coastguard Worker // If there is a mask here, and we have a variable shift, we can't be sure
4095*9880d681SAndroid Build Coastguard Worker // that we're masking out the right stuff.
4096*9880d681SAndroid Build Coastguard Worker if (LHSMask.getNode() || RHSMask.getNode())
4097*9880d681SAndroid Build Coastguard Worker return nullptr;
4098*9880d681SAndroid Build Coastguard Worker
4099*9880d681SAndroid Build Coastguard Worker // If the shift amount is sign/zext/any-extended just peel it off.
4100*9880d681SAndroid Build Coastguard Worker SDValue LExtOp0 = LHSShiftAmt;
4101*9880d681SAndroid Build Coastguard Worker SDValue RExtOp0 = RHSShiftAmt;
4102*9880d681SAndroid Build Coastguard Worker if ((LHSShiftAmt.getOpcode() == ISD::SIGN_EXTEND ||
4103*9880d681SAndroid Build Coastguard Worker LHSShiftAmt.getOpcode() == ISD::ZERO_EXTEND ||
4104*9880d681SAndroid Build Coastguard Worker LHSShiftAmt.getOpcode() == ISD::ANY_EXTEND ||
4105*9880d681SAndroid Build Coastguard Worker LHSShiftAmt.getOpcode() == ISD::TRUNCATE) &&
4106*9880d681SAndroid Build Coastguard Worker (RHSShiftAmt.getOpcode() == ISD::SIGN_EXTEND ||
4107*9880d681SAndroid Build Coastguard Worker RHSShiftAmt.getOpcode() == ISD::ZERO_EXTEND ||
4108*9880d681SAndroid Build Coastguard Worker RHSShiftAmt.getOpcode() == ISD::ANY_EXTEND ||
4109*9880d681SAndroid Build Coastguard Worker RHSShiftAmt.getOpcode() == ISD::TRUNCATE)) {
4110*9880d681SAndroid Build Coastguard Worker LExtOp0 = LHSShiftAmt.getOperand(0);
4111*9880d681SAndroid Build Coastguard Worker RExtOp0 = RHSShiftAmt.getOperand(0);
4112*9880d681SAndroid Build Coastguard Worker }
4113*9880d681SAndroid Build Coastguard Worker
4114*9880d681SAndroid Build Coastguard Worker SDNode *TryL = MatchRotatePosNeg(LHSShiftArg, LHSShiftAmt, RHSShiftAmt,
4115*9880d681SAndroid Build Coastguard Worker LExtOp0, RExtOp0, ISD::ROTL, ISD::ROTR, DL);
4116*9880d681SAndroid Build Coastguard Worker if (TryL)
4117*9880d681SAndroid Build Coastguard Worker return TryL;
4118*9880d681SAndroid Build Coastguard Worker
4119*9880d681SAndroid Build Coastguard Worker SDNode *TryR = MatchRotatePosNeg(RHSShiftArg, RHSShiftAmt, LHSShiftAmt,
4120*9880d681SAndroid Build Coastguard Worker RExtOp0, LExtOp0, ISD::ROTR, ISD::ROTL, DL);
4121*9880d681SAndroid Build Coastguard Worker if (TryR)
4122*9880d681SAndroid Build Coastguard Worker return TryR;
4123*9880d681SAndroid Build Coastguard Worker
4124*9880d681SAndroid Build Coastguard Worker return nullptr;
4125*9880d681SAndroid Build Coastguard Worker }
4126*9880d681SAndroid Build Coastguard Worker
visitXOR(SDNode * N)4127*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitXOR(SDNode *N) {
4128*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
4129*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
4130*9880d681SAndroid Build Coastguard Worker EVT VT = N0.getValueType();
4131*9880d681SAndroid Build Coastguard Worker
4132*9880d681SAndroid Build Coastguard Worker // fold vector ops
4133*9880d681SAndroid Build Coastguard Worker if (VT.isVector()) {
4134*9880d681SAndroid Build Coastguard Worker if (SDValue FoldedVOp = SimplifyVBinOp(N))
4135*9880d681SAndroid Build Coastguard Worker return FoldedVOp;
4136*9880d681SAndroid Build Coastguard Worker
4137*9880d681SAndroid Build Coastguard Worker // fold (xor x, 0) -> x, vector edition
4138*9880d681SAndroid Build Coastguard Worker if (ISD::isBuildVectorAllZeros(N0.getNode()))
4139*9880d681SAndroid Build Coastguard Worker return N1;
4140*9880d681SAndroid Build Coastguard Worker if (ISD::isBuildVectorAllZeros(N1.getNode()))
4141*9880d681SAndroid Build Coastguard Worker return N0;
4142*9880d681SAndroid Build Coastguard Worker }
4143*9880d681SAndroid Build Coastguard Worker
4144*9880d681SAndroid Build Coastguard Worker // fold (xor undef, undef) -> 0. This is a common idiom (misuse).
4145*9880d681SAndroid Build Coastguard Worker if (N0.isUndef() && N1.isUndef())
4146*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, SDLoc(N), VT);
4147*9880d681SAndroid Build Coastguard Worker // fold (xor x, undef) -> undef
4148*9880d681SAndroid Build Coastguard Worker if (N0.isUndef())
4149*9880d681SAndroid Build Coastguard Worker return N0;
4150*9880d681SAndroid Build Coastguard Worker if (N1.isUndef())
4151*9880d681SAndroid Build Coastguard Worker return N1;
4152*9880d681SAndroid Build Coastguard Worker // fold (xor c1, c2) -> c1^c2
4153*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N0C = getAsNonOpaqueConstant(N0);
4154*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N1C = getAsNonOpaqueConstant(N1);
4155*9880d681SAndroid Build Coastguard Worker if (N0C && N1C)
4156*9880d681SAndroid Build Coastguard Worker return DAG.FoldConstantArithmetic(ISD::XOR, SDLoc(N), VT, N0C, N1C);
4157*9880d681SAndroid Build Coastguard Worker // canonicalize constant to RHS
4158*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N0) &&
4159*9880d681SAndroid Build Coastguard Worker !DAG.isConstantIntBuildVectorOrConstantInt(N1))
4160*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::XOR, SDLoc(N), VT, N1, N0);
4161*9880d681SAndroid Build Coastguard Worker // fold (xor x, 0) -> x
4162*9880d681SAndroid Build Coastguard Worker if (isNullConstant(N1))
4163*9880d681SAndroid Build Coastguard Worker return N0;
4164*9880d681SAndroid Build Coastguard Worker // reassociate xor
4165*9880d681SAndroid Build Coastguard Worker if (SDValue RXOR = ReassociateOps(ISD::XOR, SDLoc(N), N0, N1))
4166*9880d681SAndroid Build Coastguard Worker return RXOR;
4167*9880d681SAndroid Build Coastguard Worker
4168*9880d681SAndroid Build Coastguard Worker // fold !(x cc y) -> (x !cc y)
4169*9880d681SAndroid Build Coastguard Worker SDValue LHS, RHS, CC;
4170*9880d681SAndroid Build Coastguard Worker if (TLI.isConstTrueVal(N1.getNode()) && isSetCCEquivalent(N0, LHS, RHS, CC)) {
4171*9880d681SAndroid Build Coastguard Worker bool isInt = LHS.getValueType().isInteger();
4172*9880d681SAndroid Build Coastguard Worker ISD::CondCode NotCC = ISD::getSetCCInverse(cast<CondCodeSDNode>(CC)->get(),
4173*9880d681SAndroid Build Coastguard Worker isInt);
4174*9880d681SAndroid Build Coastguard Worker
4175*9880d681SAndroid Build Coastguard Worker if (!LegalOperations ||
4176*9880d681SAndroid Build Coastguard Worker TLI.isCondCodeLegal(NotCC, LHS.getSimpleValueType())) {
4177*9880d681SAndroid Build Coastguard Worker switch (N0.getOpcode()) {
4178*9880d681SAndroid Build Coastguard Worker default:
4179*9880d681SAndroid Build Coastguard Worker llvm_unreachable("Unhandled SetCC Equivalent!");
4180*9880d681SAndroid Build Coastguard Worker case ISD::SETCC:
4181*9880d681SAndroid Build Coastguard Worker return DAG.getSetCC(SDLoc(N), VT, LHS, RHS, NotCC);
4182*9880d681SAndroid Build Coastguard Worker case ISD::SELECT_CC:
4183*9880d681SAndroid Build Coastguard Worker return DAG.getSelectCC(SDLoc(N), LHS, RHS, N0.getOperand(2),
4184*9880d681SAndroid Build Coastguard Worker N0.getOperand(3), NotCC);
4185*9880d681SAndroid Build Coastguard Worker }
4186*9880d681SAndroid Build Coastguard Worker }
4187*9880d681SAndroid Build Coastguard Worker }
4188*9880d681SAndroid Build Coastguard Worker
4189*9880d681SAndroid Build Coastguard Worker // fold (not (zext (setcc x, y))) -> (zext (not (setcc x, y)))
4190*9880d681SAndroid Build Coastguard Worker if (isOneConstant(N1) && N0.getOpcode() == ISD::ZERO_EXTEND &&
4191*9880d681SAndroid Build Coastguard Worker N0.getNode()->hasOneUse() &&
4192*9880d681SAndroid Build Coastguard Worker isSetCCEquivalent(N0.getOperand(0), LHS, RHS, CC)){
4193*9880d681SAndroid Build Coastguard Worker SDValue V = N0.getOperand(0);
4194*9880d681SAndroid Build Coastguard Worker SDLoc DL(N0);
4195*9880d681SAndroid Build Coastguard Worker V = DAG.getNode(ISD::XOR, DL, V.getValueType(), V,
4196*9880d681SAndroid Build Coastguard Worker DAG.getConstant(1, DL, V.getValueType()));
4197*9880d681SAndroid Build Coastguard Worker AddToWorklist(V.getNode());
4198*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ZERO_EXTEND, SDLoc(N), VT, V);
4199*9880d681SAndroid Build Coastguard Worker }
4200*9880d681SAndroid Build Coastguard Worker
4201*9880d681SAndroid Build Coastguard Worker // fold (not (or x, y)) -> (and (not x), (not y)) iff x or y are setcc
4202*9880d681SAndroid Build Coastguard Worker if (isOneConstant(N1) && VT == MVT::i1 &&
4203*9880d681SAndroid Build Coastguard Worker (N0.getOpcode() == ISD::OR || N0.getOpcode() == ISD::AND)) {
4204*9880d681SAndroid Build Coastguard Worker SDValue LHS = N0.getOperand(0), RHS = N0.getOperand(1);
4205*9880d681SAndroid Build Coastguard Worker if (isOneUseSetCC(RHS) || isOneUseSetCC(LHS)) {
4206*9880d681SAndroid Build Coastguard Worker unsigned NewOpcode = N0.getOpcode() == ISD::AND ? ISD::OR : ISD::AND;
4207*9880d681SAndroid Build Coastguard Worker LHS = DAG.getNode(ISD::XOR, SDLoc(LHS), VT, LHS, N1); // LHS = ~LHS
4208*9880d681SAndroid Build Coastguard Worker RHS = DAG.getNode(ISD::XOR, SDLoc(RHS), VT, RHS, N1); // RHS = ~RHS
4209*9880d681SAndroid Build Coastguard Worker AddToWorklist(LHS.getNode()); AddToWorklist(RHS.getNode());
4210*9880d681SAndroid Build Coastguard Worker return DAG.getNode(NewOpcode, SDLoc(N), VT, LHS, RHS);
4211*9880d681SAndroid Build Coastguard Worker }
4212*9880d681SAndroid Build Coastguard Worker }
4213*9880d681SAndroid Build Coastguard Worker // fold (not (or x, y)) -> (and (not x), (not y)) iff x or y are constants
4214*9880d681SAndroid Build Coastguard Worker if (isAllOnesConstant(N1) &&
4215*9880d681SAndroid Build Coastguard Worker (N0.getOpcode() == ISD::OR || N0.getOpcode() == ISD::AND)) {
4216*9880d681SAndroid Build Coastguard Worker SDValue LHS = N0.getOperand(0), RHS = N0.getOperand(1);
4217*9880d681SAndroid Build Coastguard Worker if (isa<ConstantSDNode>(RHS) || isa<ConstantSDNode>(LHS)) {
4218*9880d681SAndroid Build Coastguard Worker unsigned NewOpcode = N0.getOpcode() == ISD::AND ? ISD::OR : ISD::AND;
4219*9880d681SAndroid Build Coastguard Worker LHS = DAG.getNode(ISD::XOR, SDLoc(LHS), VT, LHS, N1); // LHS = ~LHS
4220*9880d681SAndroid Build Coastguard Worker RHS = DAG.getNode(ISD::XOR, SDLoc(RHS), VT, RHS, N1); // RHS = ~RHS
4221*9880d681SAndroid Build Coastguard Worker AddToWorklist(LHS.getNode()); AddToWorklist(RHS.getNode());
4222*9880d681SAndroid Build Coastguard Worker return DAG.getNode(NewOpcode, SDLoc(N), VT, LHS, RHS);
4223*9880d681SAndroid Build Coastguard Worker }
4224*9880d681SAndroid Build Coastguard Worker }
4225*9880d681SAndroid Build Coastguard Worker // fold (xor (and x, y), y) -> (and (not x), y)
4226*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::AND && N0.getNode()->hasOneUse() &&
4227*9880d681SAndroid Build Coastguard Worker N0->getOperand(1) == N1) {
4228*9880d681SAndroid Build Coastguard Worker SDValue X = N0->getOperand(0);
4229*9880d681SAndroid Build Coastguard Worker SDValue NotX = DAG.getNOT(SDLoc(X), X, VT);
4230*9880d681SAndroid Build Coastguard Worker AddToWorklist(NotX.getNode());
4231*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, SDLoc(N), VT, NotX, N1);
4232*9880d681SAndroid Build Coastguard Worker }
4233*9880d681SAndroid Build Coastguard Worker // fold (xor (xor x, c1), c2) -> (xor x, (xor c1, c2))
4234*9880d681SAndroid Build Coastguard Worker if (N1C && N0.getOpcode() == ISD::XOR) {
4235*9880d681SAndroid Build Coastguard Worker if (const ConstantSDNode *N00C = getAsNonOpaqueConstant(N0.getOperand(0))) {
4236*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
4237*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::XOR, DL, VT, N0.getOperand(1),
4238*9880d681SAndroid Build Coastguard Worker DAG.getConstant(N1C->getAPIntValue() ^
4239*9880d681SAndroid Build Coastguard Worker N00C->getAPIntValue(), DL, VT));
4240*9880d681SAndroid Build Coastguard Worker }
4241*9880d681SAndroid Build Coastguard Worker if (const ConstantSDNode *N01C = getAsNonOpaqueConstant(N0.getOperand(1))) {
4242*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
4243*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::XOR, DL, VT, N0.getOperand(0),
4244*9880d681SAndroid Build Coastguard Worker DAG.getConstant(N1C->getAPIntValue() ^
4245*9880d681SAndroid Build Coastguard Worker N01C->getAPIntValue(), DL, VT));
4246*9880d681SAndroid Build Coastguard Worker }
4247*9880d681SAndroid Build Coastguard Worker }
4248*9880d681SAndroid Build Coastguard Worker // fold (xor x, x) -> 0
4249*9880d681SAndroid Build Coastguard Worker if (N0 == N1)
4250*9880d681SAndroid Build Coastguard Worker return tryFoldToZero(SDLoc(N), TLI, VT, DAG, LegalOperations, LegalTypes);
4251*9880d681SAndroid Build Coastguard Worker
4252*9880d681SAndroid Build Coastguard Worker // fold (xor (shl 1, x), -1) -> (rotl ~1, x)
4253*9880d681SAndroid Build Coastguard Worker // Here is a concrete example of this equivalence:
4254*9880d681SAndroid Build Coastguard Worker // i16 x == 14
4255*9880d681SAndroid Build Coastguard Worker // i16 shl == 1 << 14 == 16384 == 0b0100000000000000
4256*9880d681SAndroid Build Coastguard Worker // i16 xor == ~(1 << 14) == 49151 == 0b1011111111111111
4257*9880d681SAndroid Build Coastguard Worker //
4258*9880d681SAndroid Build Coastguard Worker // =>
4259*9880d681SAndroid Build Coastguard Worker //
4260*9880d681SAndroid Build Coastguard Worker // i16 ~1 == 0b1111111111111110
4261*9880d681SAndroid Build Coastguard Worker // i16 rol(~1, 14) == 0b1011111111111111
4262*9880d681SAndroid Build Coastguard Worker //
4263*9880d681SAndroid Build Coastguard Worker // Some additional tips to help conceptualize this transform:
4264*9880d681SAndroid Build Coastguard Worker // - Try to see the operation as placing a single zero in a value of all ones.
4265*9880d681SAndroid Build Coastguard Worker // - There exists no value for x which would allow the result to contain zero.
4266*9880d681SAndroid Build Coastguard Worker // - Values of x larger than the bitwidth are undefined and do not require a
4267*9880d681SAndroid Build Coastguard Worker // consistent result.
4268*9880d681SAndroid Build Coastguard Worker // - Pushing the zero left requires shifting one bits in from the right.
4269*9880d681SAndroid Build Coastguard Worker // A rotate left of ~1 is a nice way of achieving the desired result.
4270*9880d681SAndroid Build Coastguard Worker if (TLI.isOperationLegalOrCustom(ISD::ROTL, VT) && N0.getOpcode() == ISD::SHL
4271*9880d681SAndroid Build Coastguard Worker && isAllOnesConstant(N1) && isOneConstant(N0.getOperand(0))) {
4272*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
4273*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ROTL, DL, VT, DAG.getConstant(~1, DL, VT),
4274*9880d681SAndroid Build Coastguard Worker N0.getOperand(1));
4275*9880d681SAndroid Build Coastguard Worker }
4276*9880d681SAndroid Build Coastguard Worker
4277*9880d681SAndroid Build Coastguard Worker // Simplify: xor (op x...), (op y...) -> (op (xor x, y))
4278*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == N1.getOpcode())
4279*9880d681SAndroid Build Coastguard Worker if (SDValue Tmp = SimplifyBinOpWithSameOpcodeHands(N))
4280*9880d681SAndroid Build Coastguard Worker return Tmp;
4281*9880d681SAndroid Build Coastguard Worker
4282*9880d681SAndroid Build Coastguard Worker // Simplify the expression using non-local knowledge.
4283*9880d681SAndroid Build Coastguard Worker if (!VT.isVector() &&
4284*9880d681SAndroid Build Coastguard Worker SimplifyDemandedBits(SDValue(N, 0)))
4285*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0);
4286*9880d681SAndroid Build Coastguard Worker
4287*9880d681SAndroid Build Coastguard Worker return SDValue();
4288*9880d681SAndroid Build Coastguard Worker }
4289*9880d681SAndroid Build Coastguard Worker
4290*9880d681SAndroid Build Coastguard Worker /// Handle transforms common to the three shifts, when the shift amount is a
4291*9880d681SAndroid Build Coastguard Worker /// constant.
visitShiftByConstant(SDNode * N,ConstantSDNode * Amt)4292*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitShiftByConstant(SDNode *N, ConstantSDNode *Amt) {
4293*9880d681SAndroid Build Coastguard Worker SDNode *LHS = N->getOperand(0).getNode();
4294*9880d681SAndroid Build Coastguard Worker if (!LHS->hasOneUse()) return SDValue();
4295*9880d681SAndroid Build Coastguard Worker
4296*9880d681SAndroid Build Coastguard Worker // We want to pull some binops through shifts, so that we have (and (shift))
4297*9880d681SAndroid Build Coastguard Worker // instead of (shift (and)), likewise for add, or, xor, etc. This sort of
4298*9880d681SAndroid Build Coastguard Worker // thing happens with address calculations, so it's important to canonicalize
4299*9880d681SAndroid Build Coastguard Worker // it.
4300*9880d681SAndroid Build Coastguard Worker bool HighBitSet = false; // Can we transform this if the high bit is set?
4301*9880d681SAndroid Build Coastguard Worker
4302*9880d681SAndroid Build Coastguard Worker switch (LHS->getOpcode()) {
4303*9880d681SAndroid Build Coastguard Worker default: return SDValue();
4304*9880d681SAndroid Build Coastguard Worker case ISD::OR:
4305*9880d681SAndroid Build Coastguard Worker case ISD::XOR:
4306*9880d681SAndroid Build Coastguard Worker HighBitSet = false; // We can only transform sra if the high bit is clear.
4307*9880d681SAndroid Build Coastguard Worker break;
4308*9880d681SAndroid Build Coastguard Worker case ISD::AND:
4309*9880d681SAndroid Build Coastguard Worker HighBitSet = true; // We can only transform sra if the high bit is set.
4310*9880d681SAndroid Build Coastguard Worker break;
4311*9880d681SAndroid Build Coastguard Worker case ISD::ADD:
4312*9880d681SAndroid Build Coastguard Worker if (N->getOpcode() != ISD::SHL)
4313*9880d681SAndroid Build Coastguard Worker return SDValue(); // only shl(add) not sr[al](add).
4314*9880d681SAndroid Build Coastguard Worker HighBitSet = false; // We can only transform sra if the high bit is clear.
4315*9880d681SAndroid Build Coastguard Worker break;
4316*9880d681SAndroid Build Coastguard Worker }
4317*9880d681SAndroid Build Coastguard Worker
4318*9880d681SAndroid Build Coastguard Worker // We require the RHS of the binop to be a constant and not opaque as well.
4319*9880d681SAndroid Build Coastguard Worker ConstantSDNode *BinOpCst = getAsNonOpaqueConstant(LHS->getOperand(1));
4320*9880d681SAndroid Build Coastguard Worker if (!BinOpCst) return SDValue();
4321*9880d681SAndroid Build Coastguard Worker
4322*9880d681SAndroid Build Coastguard Worker // FIXME: disable this unless the input to the binop is a shift by a constant.
4323*9880d681SAndroid Build Coastguard Worker // If it is not a shift, it pessimizes some common cases like:
4324*9880d681SAndroid Build Coastguard Worker //
4325*9880d681SAndroid Build Coastguard Worker // void foo(int *X, int i) { X[i & 1235] = 1; }
4326*9880d681SAndroid Build Coastguard Worker // int bar(int *X, int i) { return X[i & 255]; }
4327*9880d681SAndroid Build Coastguard Worker SDNode *BinOpLHSVal = LHS->getOperand(0).getNode();
4328*9880d681SAndroid Build Coastguard Worker if ((BinOpLHSVal->getOpcode() != ISD::SHL &&
4329*9880d681SAndroid Build Coastguard Worker BinOpLHSVal->getOpcode() != ISD::SRA &&
4330*9880d681SAndroid Build Coastguard Worker BinOpLHSVal->getOpcode() != ISD::SRL) ||
4331*9880d681SAndroid Build Coastguard Worker !isa<ConstantSDNode>(BinOpLHSVal->getOperand(1)))
4332*9880d681SAndroid Build Coastguard Worker return SDValue();
4333*9880d681SAndroid Build Coastguard Worker
4334*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
4335*9880d681SAndroid Build Coastguard Worker
4336*9880d681SAndroid Build Coastguard Worker // If this is a signed shift right, and the high bit is modified by the
4337*9880d681SAndroid Build Coastguard Worker // logical operation, do not perform the transformation. The highBitSet
4338*9880d681SAndroid Build Coastguard Worker // boolean indicates the value of the high bit of the constant which would
4339*9880d681SAndroid Build Coastguard Worker // cause it to be modified for this operation.
4340*9880d681SAndroid Build Coastguard Worker if (N->getOpcode() == ISD::SRA) {
4341*9880d681SAndroid Build Coastguard Worker bool BinOpRHSSignSet = BinOpCst->getAPIntValue().isNegative();
4342*9880d681SAndroid Build Coastguard Worker if (BinOpRHSSignSet != HighBitSet)
4343*9880d681SAndroid Build Coastguard Worker return SDValue();
4344*9880d681SAndroid Build Coastguard Worker }
4345*9880d681SAndroid Build Coastguard Worker
4346*9880d681SAndroid Build Coastguard Worker if (!TLI.isDesirableToCommuteWithShift(LHS))
4347*9880d681SAndroid Build Coastguard Worker return SDValue();
4348*9880d681SAndroid Build Coastguard Worker
4349*9880d681SAndroid Build Coastguard Worker // Fold the constants, shifting the binop RHS by the shift amount.
4350*9880d681SAndroid Build Coastguard Worker SDValue NewRHS = DAG.getNode(N->getOpcode(), SDLoc(LHS->getOperand(1)),
4351*9880d681SAndroid Build Coastguard Worker N->getValueType(0),
4352*9880d681SAndroid Build Coastguard Worker LHS->getOperand(1), N->getOperand(1));
4353*9880d681SAndroid Build Coastguard Worker assert(isa<ConstantSDNode>(NewRHS) && "Folding was not successful!");
4354*9880d681SAndroid Build Coastguard Worker
4355*9880d681SAndroid Build Coastguard Worker // Create the new shift.
4356*9880d681SAndroid Build Coastguard Worker SDValue NewShift = DAG.getNode(N->getOpcode(),
4357*9880d681SAndroid Build Coastguard Worker SDLoc(LHS->getOperand(0)),
4358*9880d681SAndroid Build Coastguard Worker VT, LHS->getOperand(0), N->getOperand(1));
4359*9880d681SAndroid Build Coastguard Worker
4360*9880d681SAndroid Build Coastguard Worker // Create the new binop.
4361*9880d681SAndroid Build Coastguard Worker return DAG.getNode(LHS->getOpcode(), SDLoc(N), VT, NewShift, NewRHS);
4362*9880d681SAndroid Build Coastguard Worker }
4363*9880d681SAndroid Build Coastguard Worker
distributeTruncateThroughAnd(SDNode * N)4364*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::distributeTruncateThroughAnd(SDNode *N) {
4365*9880d681SAndroid Build Coastguard Worker assert(N->getOpcode() == ISD::TRUNCATE);
4366*9880d681SAndroid Build Coastguard Worker assert(N->getOperand(0).getOpcode() == ISD::AND);
4367*9880d681SAndroid Build Coastguard Worker
4368*9880d681SAndroid Build Coastguard Worker // (truncate:TruncVT (and N00, N01C)) -> (and (truncate:TruncVT N00), TruncC)
4369*9880d681SAndroid Build Coastguard Worker if (N->hasOneUse() && N->getOperand(0).hasOneUse()) {
4370*9880d681SAndroid Build Coastguard Worker SDValue N01 = N->getOperand(0).getOperand(1);
4371*9880d681SAndroid Build Coastguard Worker
4372*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *N01C = isConstOrConstSplat(N01)) {
4373*9880d681SAndroid Build Coastguard Worker if (!N01C->isOpaque()) {
4374*9880d681SAndroid Build Coastguard Worker EVT TruncVT = N->getValueType(0);
4375*9880d681SAndroid Build Coastguard Worker SDValue N00 = N->getOperand(0).getOperand(0);
4376*9880d681SAndroid Build Coastguard Worker APInt TruncC = N01C->getAPIntValue();
4377*9880d681SAndroid Build Coastguard Worker TruncC = TruncC.trunc(TruncVT.getScalarSizeInBits());
4378*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
4379*9880d681SAndroid Build Coastguard Worker
4380*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, DL, TruncVT,
4381*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::TRUNCATE, DL, TruncVT, N00),
4382*9880d681SAndroid Build Coastguard Worker DAG.getConstant(TruncC, DL, TruncVT));
4383*9880d681SAndroid Build Coastguard Worker }
4384*9880d681SAndroid Build Coastguard Worker }
4385*9880d681SAndroid Build Coastguard Worker }
4386*9880d681SAndroid Build Coastguard Worker
4387*9880d681SAndroid Build Coastguard Worker return SDValue();
4388*9880d681SAndroid Build Coastguard Worker }
4389*9880d681SAndroid Build Coastguard Worker
visitRotate(SDNode * N)4390*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitRotate(SDNode *N) {
4391*9880d681SAndroid Build Coastguard Worker // fold (rot* x, (trunc (and y, c))) -> (rot* x, (and (trunc y), (trunc c))).
4392*9880d681SAndroid Build Coastguard Worker if (N->getOperand(1).getOpcode() == ISD::TRUNCATE &&
4393*9880d681SAndroid Build Coastguard Worker N->getOperand(1).getOperand(0).getOpcode() == ISD::AND) {
4394*9880d681SAndroid Build Coastguard Worker if (SDValue NewOp1 =
4395*9880d681SAndroid Build Coastguard Worker distributeTruncateThroughAnd(N->getOperand(1).getNode()))
4396*9880d681SAndroid Build Coastguard Worker return DAG.getNode(N->getOpcode(), SDLoc(N), N->getValueType(0),
4397*9880d681SAndroid Build Coastguard Worker N->getOperand(0), NewOp1);
4398*9880d681SAndroid Build Coastguard Worker }
4399*9880d681SAndroid Build Coastguard Worker return SDValue();
4400*9880d681SAndroid Build Coastguard Worker }
4401*9880d681SAndroid Build Coastguard Worker
visitSHL(SDNode * N)4402*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSHL(SDNode *N) {
4403*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
4404*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
4405*9880d681SAndroid Build Coastguard Worker EVT VT = N0.getValueType();
4406*9880d681SAndroid Build Coastguard Worker unsigned OpSizeInBits = VT.getScalarSizeInBits();
4407*9880d681SAndroid Build Coastguard Worker
4408*9880d681SAndroid Build Coastguard Worker // fold vector ops
4409*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
4410*9880d681SAndroid Build Coastguard Worker if (VT.isVector()) {
4411*9880d681SAndroid Build Coastguard Worker if (SDValue FoldedVOp = SimplifyVBinOp(N))
4412*9880d681SAndroid Build Coastguard Worker return FoldedVOp;
4413*9880d681SAndroid Build Coastguard Worker
4414*9880d681SAndroid Build Coastguard Worker BuildVectorSDNode *N1CV = dyn_cast<BuildVectorSDNode>(N1);
4415*9880d681SAndroid Build Coastguard Worker // If setcc produces all-one true value then:
4416*9880d681SAndroid Build Coastguard Worker // (shl (and (setcc) N01CV) N1CV) -> (and (setcc) N01CV<<N1CV)
4417*9880d681SAndroid Build Coastguard Worker if (N1CV && N1CV->isConstant()) {
4418*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::AND) {
4419*9880d681SAndroid Build Coastguard Worker SDValue N00 = N0->getOperand(0);
4420*9880d681SAndroid Build Coastguard Worker SDValue N01 = N0->getOperand(1);
4421*9880d681SAndroid Build Coastguard Worker BuildVectorSDNode *N01CV = dyn_cast<BuildVectorSDNode>(N01);
4422*9880d681SAndroid Build Coastguard Worker
4423*9880d681SAndroid Build Coastguard Worker if (N01CV && N01CV->isConstant() && N00.getOpcode() == ISD::SETCC &&
4424*9880d681SAndroid Build Coastguard Worker TLI.getBooleanContents(N00.getOperand(0).getValueType()) ==
4425*9880d681SAndroid Build Coastguard Worker TargetLowering::ZeroOrNegativeOneBooleanContent) {
4426*9880d681SAndroid Build Coastguard Worker if (SDValue C = DAG.FoldConstantArithmetic(ISD::SHL, SDLoc(N), VT,
4427*9880d681SAndroid Build Coastguard Worker N01CV, N1CV))
4428*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, SDLoc(N), VT, N00, C);
4429*9880d681SAndroid Build Coastguard Worker }
4430*9880d681SAndroid Build Coastguard Worker } else {
4431*9880d681SAndroid Build Coastguard Worker N1C = isConstOrConstSplat(N1);
4432*9880d681SAndroid Build Coastguard Worker }
4433*9880d681SAndroid Build Coastguard Worker }
4434*9880d681SAndroid Build Coastguard Worker }
4435*9880d681SAndroid Build Coastguard Worker
4436*9880d681SAndroid Build Coastguard Worker // fold (shl c1, c2) -> c1<<c2
4437*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N0C = getAsNonOpaqueConstant(N0);
4438*9880d681SAndroid Build Coastguard Worker if (N0C && N1C && !N1C->isOpaque())
4439*9880d681SAndroid Build Coastguard Worker return DAG.FoldConstantArithmetic(ISD::SHL, SDLoc(N), VT, N0C, N1C);
4440*9880d681SAndroid Build Coastguard Worker // fold (shl 0, x) -> 0
4441*9880d681SAndroid Build Coastguard Worker if (isNullConstant(N0))
4442*9880d681SAndroid Build Coastguard Worker return N0;
4443*9880d681SAndroid Build Coastguard Worker // fold (shl x, c >= size(x)) -> undef
4444*9880d681SAndroid Build Coastguard Worker if (N1C && N1C->getAPIntValue().uge(OpSizeInBits))
4445*9880d681SAndroid Build Coastguard Worker return DAG.getUNDEF(VT);
4446*9880d681SAndroid Build Coastguard Worker // fold (shl x, 0) -> x
4447*9880d681SAndroid Build Coastguard Worker if (N1C && N1C->isNullValue())
4448*9880d681SAndroid Build Coastguard Worker return N0;
4449*9880d681SAndroid Build Coastguard Worker // fold (shl undef, x) -> 0
4450*9880d681SAndroid Build Coastguard Worker if (N0.isUndef())
4451*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, SDLoc(N), VT);
4452*9880d681SAndroid Build Coastguard Worker // if (shl x, c) is known to be zero, return 0
4453*9880d681SAndroid Build Coastguard Worker if (DAG.MaskedValueIsZero(SDValue(N, 0),
4454*9880d681SAndroid Build Coastguard Worker APInt::getAllOnesValue(OpSizeInBits)))
4455*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, SDLoc(N), VT);
4456*9880d681SAndroid Build Coastguard Worker // fold (shl x, (trunc (and y, c))) -> (shl x, (and (trunc y), (trunc c))).
4457*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::TRUNCATE &&
4458*9880d681SAndroid Build Coastguard Worker N1.getOperand(0).getOpcode() == ISD::AND) {
4459*9880d681SAndroid Build Coastguard Worker if (SDValue NewOp1 = distributeTruncateThroughAnd(N1.getNode()))
4460*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SHL, SDLoc(N), VT, N0, NewOp1);
4461*9880d681SAndroid Build Coastguard Worker }
4462*9880d681SAndroid Build Coastguard Worker
4463*9880d681SAndroid Build Coastguard Worker if (N1C && SimplifyDemandedBits(SDValue(N, 0)))
4464*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0);
4465*9880d681SAndroid Build Coastguard Worker
4466*9880d681SAndroid Build Coastguard Worker // fold (shl (shl x, c1), c2) -> 0 or (shl x, (add c1, c2))
4467*9880d681SAndroid Build Coastguard Worker if (N1C && N0.getOpcode() == ISD::SHL) {
4468*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *N0C1 = isConstOrConstSplat(N0.getOperand(1))) {
4469*9880d681SAndroid Build Coastguard Worker uint64_t c1 = N0C1->getZExtValue();
4470*9880d681SAndroid Build Coastguard Worker uint64_t c2 = N1C->getZExtValue();
4471*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
4472*9880d681SAndroid Build Coastguard Worker if (c1 + c2 >= OpSizeInBits)
4473*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, DL, VT);
4474*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SHL, DL, VT, N0.getOperand(0),
4475*9880d681SAndroid Build Coastguard Worker DAG.getConstant(c1 + c2, DL, N1.getValueType()));
4476*9880d681SAndroid Build Coastguard Worker }
4477*9880d681SAndroid Build Coastguard Worker }
4478*9880d681SAndroid Build Coastguard Worker
4479*9880d681SAndroid Build Coastguard Worker // fold (shl (ext (shl x, c1)), c2) -> (ext (shl x, (add c1, c2)))
4480*9880d681SAndroid Build Coastguard Worker // For this to be valid, the second form must not preserve any of the bits
4481*9880d681SAndroid Build Coastguard Worker // that are shifted out by the inner shift in the first form. This means
4482*9880d681SAndroid Build Coastguard Worker // the outer shift size must be >= the number of bits added by the ext.
4483*9880d681SAndroid Build Coastguard Worker // As a corollary, we don't care what kind of ext it is.
4484*9880d681SAndroid Build Coastguard Worker if (N1C && (N0.getOpcode() == ISD::ZERO_EXTEND ||
4485*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::ANY_EXTEND ||
4486*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::SIGN_EXTEND) &&
4487*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getOpcode() == ISD::SHL) {
4488*9880d681SAndroid Build Coastguard Worker SDValue N0Op0 = N0.getOperand(0);
4489*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *N0Op0C1 = isConstOrConstSplat(N0Op0.getOperand(1))) {
4490*9880d681SAndroid Build Coastguard Worker uint64_t c1 = N0Op0C1->getZExtValue();
4491*9880d681SAndroid Build Coastguard Worker uint64_t c2 = N1C->getZExtValue();
4492*9880d681SAndroid Build Coastguard Worker EVT InnerShiftVT = N0Op0.getValueType();
4493*9880d681SAndroid Build Coastguard Worker uint64_t InnerShiftSize = InnerShiftVT.getScalarSizeInBits();
4494*9880d681SAndroid Build Coastguard Worker if (c2 >= OpSizeInBits - InnerShiftSize) {
4495*9880d681SAndroid Build Coastguard Worker SDLoc DL(N0);
4496*9880d681SAndroid Build Coastguard Worker if (c1 + c2 >= OpSizeInBits)
4497*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, DL, VT);
4498*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SHL, DL, VT,
4499*9880d681SAndroid Build Coastguard Worker DAG.getNode(N0.getOpcode(), DL, VT,
4500*9880d681SAndroid Build Coastguard Worker N0Op0->getOperand(0)),
4501*9880d681SAndroid Build Coastguard Worker DAG.getConstant(c1 + c2, DL, N1.getValueType()));
4502*9880d681SAndroid Build Coastguard Worker }
4503*9880d681SAndroid Build Coastguard Worker }
4504*9880d681SAndroid Build Coastguard Worker }
4505*9880d681SAndroid Build Coastguard Worker
4506*9880d681SAndroid Build Coastguard Worker // fold (shl (zext (srl x, C)), C) -> (zext (shl (srl x, C), C))
4507*9880d681SAndroid Build Coastguard Worker // Only fold this if the inner zext has no other uses to avoid increasing
4508*9880d681SAndroid Build Coastguard Worker // the total number of instructions.
4509*9880d681SAndroid Build Coastguard Worker if (N1C && N0.getOpcode() == ISD::ZERO_EXTEND && N0.hasOneUse() &&
4510*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getOpcode() == ISD::SRL) {
4511*9880d681SAndroid Build Coastguard Worker SDValue N0Op0 = N0.getOperand(0);
4512*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *N0Op0C1 = isConstOrConstSplat(N0Op0.getOperand(1))) {
4513*9880d681SAndroid Build Coastguard Worker uint64_t c1 = N0Op0C1->getZExtValue();
4514*9880d681SAndroid Build Coastguard Worker if (c1 < VT.getScalarSizeInBits()) {
4515*9880d681SAndroid Build Coastguard Worker uint64_t c2 = N1C->getZExtValue();
4516*9880d681SAndroid Build Coastguard Worker if (c1 == c2) {
4517*9880d681SAndroid Build Coastguard Worker SDValue NewOp0 = N0.getOperand(0);
4518*9880d681SAndroid Build Coastguard Worker EVT CountVT = NewOp0.getOperand(1).getValueType();
4519*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
4520*9880d681SAndroid Build Coastguard Worker SDValue NewSHL = DAG.getNode(ISD::SHL, DL, NewOp0.getValueType(),
4521*9880d681SAndroid Build Coastguard Worker NewOp0,
4522*9880d681SAndroid Build Coastguard Worker DAG.getConstant(c2, DL, CountVT));
4523*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewSHL.getNode());
4524*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ZERO_EXTEND, SDLoc(N0), VT, NewSHL);
4525*9880d681SAndroid Build Coastguard Worker }
4526*9880d681SAndroid Build Coastguard Worker }
4527*9880d681SAndroid Build Coastguard Worker }
4528*9880d681SAndroid Build Coastguard Worker }
4529*9880d681SAndroid Build Coastguard Worker
4530*9880d681SAndroid Build Coastguard Worker // fold (shl (sr[la] exact X, C1), C2) -> (shl X, (C2-C1)) if C1 <= C2
4531*9880d681SAndroid Build Coastguard Worker // fold (shl (sr[la] exact X, C1), C2) -> (sr[la] X, (C2-C1)) if C1 > C2
4532*9880d681SAndroid Build Coastguard Worker if (N1C && (N0.getOpcode() == ISD::SRL || N0.getOpcode() == ISD::SRA) &&
4533*9880d681SAndroid Build Coastguard Worker cast<BinaryWithFlagsSDNode>(N0)->Flags.hasExact()) {
4534*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *N0C1 = isConstOrConstSplat(N0.getOperand(1))) {
4535*9880d681SAndroid Build Coastguard Worker uint64_t C1 = N0C1->getZExtValue();
4536*9880d681SAndroid Build Coastguard Worker uint64_t C2 = N1C->getZExtValue();
4537*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
4538*9880d681SAndroid Build Coastguard Worker if (C1 <= C2)
4539*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SHL, DL, VT, N0.getOperand(0),
4540*9880d681SAndroid Build Coastguard Worker DAG.getConstant(C2 - C1, DL, N1.getValueType()));
4541*9880d681SAndroid Build Coastguard Worker return DAG.getNode(N0.getOpcode(), DL, VT, N0.getOperand(0),
4542*9880d681SAndroid Build Coastguard Worker DAG.getConstant(C1 - C2, DL, N1.getValueType()));
4543*9880d681SAndroid Build Coastguard Worker }
4544*9880d681SAndroid Build Coastguard Worker }
4545*9880d681SAndroid Build Coastguard Worker
4546*9880d681SAndroid Build Coastguard Worker // fold (shl (srl x, c1), c2) -> (and (shl x, (sub c2, c1), MASK) or
4547*9880d681SAndroid Build Coastguard Worker // (and (srl x, (sub c1, c2), MASK)
4548*9880d681SAndroid Build Coastguard Worker // Only fold this if the inner shift has no other uses -- if it does, folding
4549*9880d681SAndroid Build Coastguard Worker // this will increase the total number of instructions.
4550*9880d681SAndroid Build Coastguard Worker if (N1C && N0.getOpcode() == ISD::SRL && N0.hasOneUse()) {
4551*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *N0C1 = isConstOrConstSplat(N0.getOperand(1))) {
4552*9880d681SAndroid Build Coastguard Worker uint64_t c1 = N0C1->getZExtValue();
4553*9880d681SAndroid Build Coastguard Worker if (c1 < OpSizeInBits) {
4554*9880d681SAndroid Build Coastguard Worker uint64_t c2 = N1C->getZExtValue();
4555*9880d681SAndroid Build Coastguard Worker APInt Mask = APInt::getHighBitsSet(OpSizeInBits, OpSizeInBits - c1);
4556*9880d681SAndroid Build Coastguard Worker SDValue Shift;
4557*9880d681SAndroid Build Coastguard Worker if (c2 > c1) {
4558*9880d681SAndroid Build Coastguard Worker Mask = Mask.shl(c2 - c1);
4559*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
4560*9880d681SAndroid Build Coastguard Worker Shift = DAG.getNode(ISD::SHL, DL, VT, N0.getOperand(0),
4561*9880d681SAndroid Build Coastguard Worker DAG.getConstant(c2 - c1, DL, N1.getValueType()));
4562*9880d681SAndroid Build Coastguard Worker } else {
4563*9880d681SAndroid Build Coastguard Worker Mask = Mask.lshr(c1 - c2);
4564*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
4565*9880d681SAndroid Build Coastguard Worker Shift = DAG.getNode(ISD::SRL, DL, VT, N0.getOperand(0),
4566*9880d681SAndroid Build Coastguard Worker DAG.getConstant(c1 - c2, DL, N1.getValueType()));
4567*9880d681SAndroid Build Coastguard Worker }
4568*9880d681SAndroid Build Coastguard Worker SDLoc DL(N0);
4569*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, DL, VT, Shift,
4570*9880d681SAndroid Build Coastguard Worker DAG.getConstant(Mask, DL, VT));
4571*9880d681SAndroid Build Coastguard Worker }
4572*9880d681SAndroid Build Coastguard Worker }
4573*9880d681SAndroid Build Coastguard Worker }
4574*9880d681SAndroid Build Coastguard Worker // fold (shl (sra x, c1), c1) -> (and x, (shl -1, c1))
4575*9880d681SAndroid Build Coastguard Worker if (N1C && N0.getOpcode() == ISD::SRA && N1 == N0.getOperand(1)) {
4576*9880d681SAndroid Build Coastguard Worker unsigned BitSize = VT.getScalarSizeInBits();
4577*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
4578*9880d681SAndroid Build Coastguard Worker SDValue HiBitsMask =
4579*9880d681SAndroid Build Coastguard Worker DAG.getConstant(APInt::getHighBitsSet(BitSize,
4580*9880d681SAndroid Build Coastguard Worker BitSize - N1C->getZExtValue()),
4581*9880d681SAndroid Build Coastguard Worker DL, VT);
4582*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, DL, VT, N0.getOperand(0),
4583*9880d681SAndroid Build Coastguard Worker HiBitsMask);
4584*9880d681SAndroid Build Coastguard Worker }
4585*9880d681SAndroid Build Coastguard Worker
4586*9880d681SAndroid Build Coastguard Worker // fold (shl (add x, c1), c2) -> (add (shl x, c2), c1 << c2)
4587*9880d681SAndroid Build Coastguard Worker // Variant of version done on multiply, except mul by a power of 2 is turned
4588*9880d681SAndroid Build Coastguard Worker // into a shift.
4589*9880d681SAndroid Build Coastguard Worker APInt Val;
4590*9880d681SAndroid Build Coastguard Worker if (N1C && N0.getOpcode() == ISD::ADD && N0.getNode()->hasOneUse() &&
4591*9880d681SAndroid Build Coastguard Worker (isa<ConstantSDNode>(N0.getOperand(1)) ||
4592*9880d681SAndroid Build Coastguard Worker ISD::isConstantSplatVector(N0.getOperand(1).getNode(), Val))) {
4593*9880d681SAndroid Build Coastguard Worker SDValue Shl0 = DAG.getNode(ISD::SHL, SDLoc(N0), VT, N0.getOperand(0), N1);
4594*9880d681SAndroid Build Coastguard Worker SDValue Shl1 = DAG.getNode(ISD::SHL, SDLoc(N1), VT, N0.getOperand(1), N1);
4595*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ADD, SDLoc(N), VT, Shl0, Shl1);
4596*9880d681SAndroid Build Coastguard Worker }
4597*9880d681SAndroid Build Coastguard Worker
4598*9880d681SAndroid Build Coastguard Worker // fold (shl (mul x, c1), c2) -> (mul x, c1 << c2)
4599*9880d681SAndroid Build Coastguard Worker if (N1C && N0.getOpcode() == ISD::MUL && N0.getNode()->hasOneUse()) {
4600*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *N0C1 = isConstOrConstSplat(N0.getOperand(1))) {
4601*9880d681SAndroid Build Coastguard Worker if (SDValue Folded =
4602*9880d681SAndroid Build Coastguard Worker DAG.FoldConstantArithmetic(ISD::SHL, SDLoc(N1), VT, N0C1, N1C))
4603*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::MUL, SDLoc(N), VT, N0.getOperand(0), Folded);
4604*9880d681SAndroid Build Coastguard Worker }
4605*9880d681SAndroid Build Coastguard Worker }
4606*9880d681SAndroid Build Coastguard Worker
4607*9880d681SAndroid Build Coastguard Worker if (N1C && !N1C->isOpaque())
4608*9880d681SAndroid Build Coastguard Worker if (SDValue NewSHL = visitShiftByConstant(N, N1C))
4609*9880d681SAndroid Build Coastguard Worker return NewSHL;
4610*9880d681SAndroid Build Coastguard Worker
4611*9880d681SAndroid Build Coastguard Worker return SDValue();
4612*9880d681SAndroid Build Coastguard Worker }
4613*9880d681SAndroid Build Coastguard Worker
visitSRA(SDNode * N)4614*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSRA(SDNode *N) {
4615*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
4616*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
4617*9880d681SAndroid Build Coastguard Worker EVT VT = N0.getValueType();
4618*9880d681SAndroid Build Coastguard Worker unsigned OpSizeInBits = VT.getScalarType().getSizeInBits();
4619*9880d681SAndroid Build Coastguard Worker
4620*9880d681SAndroid Build Coastguard Worker // fold vector ops
4621*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
4622*9880d681SAndroid Build Coastguard Worker if (VT.isVector()) {
4623*9880d681SAndroid Build Coastguard Worker if (SDValue FoldedVOp = SimplifyVBinOp(N))
4624*9880d681SAndroid Build Coastguard Worker return FoldedVOp;
4625*9880d681SAndroid Build Coastguard Worker
4626*9880d681SAndroid Build Coastguard Worker N1C = isConstOrConstSplat(N1);
4627*9880d681SAndroid Build Coastguard Worker }
4628*9880d681SAndroid Build Coastguard Worker
4629*9880d681SAndroid Build Coastguard Worker // fold (sra c1, c2) -> (sra c1, c2)
4630*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N0C = getAsNonOpaqueConstant(N0);
4631*9880d681SAndroid Build Coastguard Worker if (N0C && N1C && !N1C->isOpaque())
4632*9880d681SAndroid Build Coastguard Worker return DAG.FoldConstantArithmetic(ISD::SRA, SDLoc(N), VT, N0C, N1C);
4633*9880d681SAndroid Build Coastguard Worker // fold (sra 0, x) -> 0
4634*9880d681SAndroid Build Coastguard Worker if (isNullConstant(N0))
4635*9880d681SAndroid Build Coastguard Worker return N0;
4636*9880d681SAndroid Build Coastguard Worker // fold (sra -1, x) -> -1
4637*9880d681SAndroid Build Coastguard Worker if (isAllOnesConstant(N0))
4638*9880d681SAndroid Build Coastguard Worker return N0;
4639*9880d681SAndroid Build Coastguard Worker // fold (sra x, (setge c, size(x))) -> undef
4640*9880d681SAndroid Build Coastguard Worker if (N1C && N1C->getZExtValue() >= OpSizeInBits)
4641*9880d681SAndroid Build Coastguard Worker return DAG.getUNDEF(VT);
4642*9880d681SAndroid Build Coastguard Worker // fold (sra x, 0) -> x
4643*9880d681SAndroid Build Coastguard Worker if (N1C && N1C->isNullValue())
4644*9880d681SAndroid Build Coastguard Worker return N0;
4645*9880d681SAndroid Build Coastguard Worker // fold (sra (shl x, c1), c1) -> sext_inreg for some c1 and target supports
4646*9880d681SAndroid Build Coastguard Worker // sext_inreg.
4647*9880d681SAndroid Build Coastguard Worker if (N1C && N0.getOpcode() == ISD::SHL && N1 == N0.getOperand(1)) {
4648*9880d681SAndroid Build Coastguard Worker unsigned LowBits = OpSizeInBits - (unsigned)N1C->getZExtValue();
4649*9880d681SAndroid Build Coastguard Worker EVT ExtVT = EVT::getIntegerVT(*DAG.getContext(), LowBits);
4650*9880d681SAndroid Build Coastguard Worker if (VT.isVector())
4651*9880d681SAndroid Build Coastguard Worker ExtVT = EVT::getVectorVT(*DAG.getContext(),
4652*9880d681SAndroid Build Coastguard Worker ExtVT, VT.getVectorNumElements());
4653*9880d681SAndroid Build Coastguard Worker if ((!LegalOperations ||
4654*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegal(ISD::SIGN_EXTEND_INREG, ExtVT)))
4655*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SIGN_EXTEND_INREG, SDLoc(N), VT,
4656*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), DAG.getValueType(ExtVT));
4657*9880d681SAndroid Build Coastguard Worker }
4658*9880d681SAndroid Build Coastguard Worker
4659*9880d681SAndroid Build Coastguard Worker // fold (sra (sra x, c1), c2) -> (sra x, (add c1, c2))
4660*9880d681SAndroid Build Coastguard Worker if (N1C && N0.getOpcode() == ISD::SRA) {
4661*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *C1 = isConstOrConstSplat(N0.getOperand(1))) {
4662*9880d681SAndroid Build Coastguard Worker unsigned Sum = N1C->getZExtValue() + C1->getZExtValue();
4663*9880d681SAndroid Build Coastguard Worker if (Sum >= OpSizeInBits)
4664*9880d681SAndroid Build Coastguard Worker Sum = OpSizeInBits - 1;
4665*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
4666*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SRA, DL, VT, N0.getOperand(0),
4667*9880d681SAndroid Build Coastguard Worker DAG.getConstant(Sum, DL, N1.getValueType()));
4668*9880d681SAndroid Build Coastguard Worker }
4669*9880d681SAndroid Build Coastguard Worker }
4670*9880d681SAndroid Build Coastguard Worker
4671*9880d681SAndroid Build Coastguard Worker // fold (sra (shl X, m), (sub result_size, n))
4672*9880d681SAndroid Build Coastguard Worker // -> (sign_extend (trunc (shl X, (sub (sub result_size, n), m)))) for
4673*9880d681SAndroid Build Coastguard Worker // result_size - n != m.
4674*9880d681SAndroid Build Coastguard Worker // If truncate is free for the target sext(shl) is likely to result in better
4675*9880d681SAndroid Build Coastguard Worker // code.
4676*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SHL && N1C) {
4677*9880d681SAndroid Build Coastguard Worker // Get the two constanst of the shifts, CN0 = m, CN = n.
4678*9880d681SAndroid Build Coastguard Worker const ConstantSDNode *N01C = isConstOrConstSplat(N0.getOperand(1));
4679*9880d681SAndroid Build Coastguard Worker if (N01C) {
4680*9880d681SAndroid Build Coastguard Worker LLVMContext &Ctx = *DAG.getContext();
4681*9880d681SAndroid Build Coastguard Worker // Determine what the truncate's result bitsize and type would be.
4682*9880d681SAndroid Build Coastguard Worker EVT TruncVT = EVT::getIntegerVT(Ctx, OpSizeInBits - N1C->getZExtValue());
4683*9880d681SAndroid Build Coastguard Worker
4684*9880d681SAndroid Build Coastguard Worker if (VT.isVector())
4685*9880d681SAndroid Build Coastguard Worker TruncVT = EVT::getVectorVT(Ctx, TruncVT, VT.getVectorNumElements());
4686*9880d681SAndroid Build Coastguard Worker
4687*9880d681SAndroid Build Coastguard Worker // Determine the residual right-shift amount.
4688*9880d681SAndroid Build Coastguard Worker int ShiftAmt = N1C->getZExtValue() - N01C->getZExtValue();
4689*9880d681SAndroid Build Coastguard Worker
4690*9880d681SAndroid Build Coastguard Worker // If the shift is not a no-op (in which case this should be just a sign
4691*9880d681SAndroid Build Coastguard Worker // extend already), the truncated to type is legal, sign_extend is legal
4692*9880d681SAndroid Build Coastguard Worker // on that type, and the truncate to that type is both legal and free,
4693*9880d681SAndroid Build Coastguard Worker // perform the transform.
4694*9880d681SAndroid Build Coastguard Worker if ((ShiftAmt > 0) &&
4695*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegalOrCustom(ISD::SIGN_EXTEND, TruncVT) &&
4696*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegalOrCustom(ISD::TRUNCATE, VT) &&
4697*9880d681SAndroid Build Coastguard Worker TLI.isTruncateFree(VT, TruncVT)) {
4698*9880d681SAndroid Build Coastguard Worker
4699*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
4700*9880d681SAndroid Build Coastguard Worker SDValue Amt = DAG.getConstant(ShiftAmt, DL,
4701*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(N0.getOperand(0).getValueType()));
4702*9880d681SAndroid Build Coastguard Worker SDValue Shift = DAG.getNode(ISD::SRL, DL, VT,
4703*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), Amt);
4704*9880d681SAndroid Build Coastguard Worker SDValue Trunc = DAG.getNode(ISD::TRUNCATE, DL, TruncVT,
4705*9880d681SAndroid Build Coastguard Worker Shift);
4706*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SIGN_EXTEND, DL,
4707*9880d681SAndroid Build Coastguard Worker N->getValueType(0), Trunc);
4708*9880d681SAndroid Build Coastguard Worker }
4709*9880d681SAndroid Build Coastguard Worker }
4710*9880d681SAndroid Build Coastguard Worker }
4711*9880d681SAndroid Build Coastguard Worker
4712*9880d681SAndroid Build Coastguard Worker // fold (sra x, (trunc (and y, c))) -> (sra x, (and (trunc y), (trunc c))).
4713*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::TRUNCATE &&
4714*9880d681SAndroid Build Coastguard Worker N1.getOperand(0).getOpcode() == ISD::AND) {
4715*9880d681SAndroid Build Coastguard Worker if (SDValue NewOp1 = distributeTruncateThroughAnd(N1.getNode()))
4716*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SRA, SDLoc(N), VT, N0, NewOp1);
4717*9880d681SAndroid Build Coastguard Worker }
4718*9880d681SAndroid Build Coastguard Worker
4719*9880d681SAndroid Build Coastguard Worker // fold (sra (trunc (srl x, c1)), c2) -> (trunc (sra x, c1 + c2))
4720*9880d681SAndroid Build Coastguard Worker // if c1 is equal to the number of bits the trunc removes
4721*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::TRUNCATE &&
4722*9880d681SAndroid Build Coastguard Worker (N0.getOperand(0).getOpcode() == ISD::SRL ||
4723*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getOpcode() == ISD::SRA) &&
4724*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).hasOneUse() &&
4725*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getOperand(1).hasOneUse() &&
4726*9880d681SAndroid Build Coastguard Worker N1C) {
4727*9880d681SAndroid Build Coastguard Worker SDValue N0Op0 = N0.getOperand(0);
4728*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *LargeShift = isConstOrConstSplat(N0Op0.getOperand(1))) {
4729*9880d681SAndroid Build Coastguard Worker unsigned LargeShiftVal = LargeShift->getZExtValue();
4730*9880d681SAndroid Build Coastguard Worker EVT LargeVT = N0Op0.getValueType();
4731*9880d681SAndroid Build Coastguard Worker
4732*9880d681SAndroid Build Coastguard Worker if (LargeVT.getScalarSizeInBits() - OpSizeInBits == LargeShiftVal) {
4733*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
4734*9880d681SAndroid Build Coastguard Worker SDValue Amt =
4735*9880d681SAndroid Build Coastguard Worker DAG.getConstant(LargeShiftVal + N1C->getZExtValue(), DL,
4736*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(N0Op0.getOperand(0).getValueType()));
4737*9880d681SAndroid Build Coastguard Worker SDValue SRA = DAG.getNode(ISD::SRA, DL, LargeVT,
4738*9880d681SAndroid Build Coastguard Worker N0Op0.getOperand(0), Amt);
4739*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TRUNCATE, DL, VT, SRA);
4740*9880d681SAndroid Build Coastguard Worker }
4741*9880d681SAndroid Build Coastguard Worker }
4742*9880d681SAndroid Build Coastguard Worker }
4743*9880d681SAndroid Build Coastguard Worker
4744*9880d681SAndroid Build Coastguard Worker // Simplify, based on bits shifted out of the LHS.
4745*9880d681SAndroid Build Coastguard Worker if (N1C && SimplifyDemandedBits(SDValue(N, 0)))
4746*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0);
4747*9880d681SAndroid Build Coastguard Worker
4748*9880d681SAndroid Build Coastguard Worker
4749*9880d681SAndroid Build Coastguard Worker // If the sign bit is known to be zero, switch this to a SRL.
4750*9880d681SAndroid Build Coastguard Worker if (DAG.SignBitIsZero(N0))
4751*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SRL, SDLoc(N), VT, N0, N1);
4752*9880d681SAndroid Build Coastguard Worker
4753*9880d681SAndroid Build Coastguard Worker if (N1C && !N1C->isOpaque())
4754*9880d681SAndroid Build Coastguard Worker if (SDValue NewSRA = visitShiftByConstant(N, N1C))
4755*9880d681SAndroid Build Coastguard Worker return NewSRA;
4756*9880d681SAndroid Build Coastguard Worker
4757*9880d681SAndroid Build Coastguard Worker return SDValue();
4758*9880d681SAndroid Build Coastguard Worker }
4759*9880d681SAndroid Build Coastguard Worker
visitSRL(SDNode * N)4760*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSRL(SDNode *N) {
4761*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
4762*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
4763*9880d681SAndroid Build Coastguard Worker EVT VT = N0.getValueType();
4764*9880d681SAndroid Build Coastguard Worker unsigned OpSizeInBits = VT.getScalarType().getSizeInBits();
4765*9880d681SAndroid Build Coastguard Worker
4766*9880d681SAndroid Build Coastguard Worker // fold vector ops
4767*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
4768*9880d681SAndroid Build Coastguard Worker if (VT.isVector()) {
4769*9880d681SAndroid Build Coastguard Worker if (SDValue FoldedVOp = SimplifyVBinOp(N))
4770*9880d681SAndroid Build Coastguard Worker return FoldedVOp;
4771*9880d681SAndroid Build Coastguard Worker
4772*9880d681SAndroid Build Coastguard Worker N1C = isConstOrConstSplat(N1);
4773*9880d681SAndroid Build Coastguard Worker }
4774*9880d681SAndroid Build Coastguard Worker
4775*9880d681SAndroid Build Coastguard Worker // fold (srl c1, c2) -> c1 >>u c2
4776*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N0C = getAsNonOpaqueConstant(N0);
4777*9880d681SAndroid Build Coastguard Worker if (N0C && N1C && !N1C->isOpaque())
4778*9880d681SAndroid Build Coastguard Worker return DAG.FoldConstantArithmetic(ISD::SRL, SDLoc(N), VT, N0C, N1C);
4779*9880d681SAndroid Build Coastguard Worker // fold (srl 0, x) -> 0
4780*9880d681SAndroid Build Coastguard Worker if (isNullConstant(N0))
4781*9880d681SAndroid Build Coastguard Worker return N0;
4782*9880d681SAndroid Build Coastguard Worker // fold (srl x, c >= size(x)) -> undef
4783*9880d681SAndroid Build Coastguard Worker if (N1C && N1C->getZExtValue() >= OpSizeInBits)
4784*9880d681SAndroid Build Coastguard Worker return DAG.getUNDEF(VT);
4785*9880d681SAndroid Build Coastguard Worker // fold (srl x, 0) -> x
4786*9880d681SAndroid Build Coastguard Worker if (N1C && N1C->isNullValue())
4787*9880d681SAndroid Build Coastguard Worker return N0;
4788*9880d681SAndroid Build Coastguard Worker // if (srl x, c) is known to be zero, return 0
4789*9880d681SAndroid Build Coastguard Worker if (N1C && DAG.MaskedValueIsZero(SDValue(N, 0),
4790*9880d681SAndroid Build Coastguard Worker APInt::getAllOnesValue(OpSizeInBits)))
4791*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, SDLoc(N), VT);
4792*9880d681SAndroid Build Coastguard Worker
4793*9880d681SAndroid Build Coastguard Worker // fold (srl (srl x, c1), c2) -> 0 or (srl x, (add c1, c2))
4794*9880d681SAndroid Build Coastguard Worker if (N1C && N0.getOpcode() == ISD::SRL) {
4795*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *N01C = isConstOrConstSplat(N0.getOperand(1))) {
4796*9880d681SAndroid Build Coastguard Worker uint64_t c1 = N01C->getZExtValue();
4797*9880d681SAndroid Build Coastguard Worker uint64_t c2 = N1C->getZExtValue();
4798*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
4799*9880d681SAndroid Build Coastguard Worker if (c1 + c2 >= OpSizeInBits)
4800*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, DL, VT);
4801*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SRL, DL, VT, N0.getOperand(0),
4802*9880d681SAndroid Build Coastguard Worker DAG.getConstant(c1 + c2, DL, N1.getValueType()));
4803*9880d681SAndroid Build Coastguard Worker }
4804*9880d681SAndroid Build Coastguard Worker }
4805*9880d681SAndroid Build Coastguard Worker
4806*9880d681SAndroid Build Coastguard Worker // fold (srl (trunc (srl x, c1)), c2) -> 0 or (trunc (srl x, (add c1, c2)))
4807*9880d681SAndroid Build Coastguard Worker if (N1C && N0.getOpcode() == ISD::TRUNCATE &&
4808*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getOpcode() == ISD::SRL &&
4809*9880d681SAndroid Build Coastguard Worker isa<ConstantSDNode>(N0.getOperand(0)->getOperand(1))) {
4810*9880d681SAndroid Build Coastguard Worker uint64_t c1 =
4811*9880d681SAndroid Build Coastguard Worker cast<ConstantSDNode>(N0.getOperand(0)->getOperand(1))->getZExtValue();
4812*9880d681SAndroid Build Coastguard Worker uint64_t c2 = N1C->getZExtValue();
4813*9880d681SAndroid Build Coastguard Worker EVT InnerShiftVT = N0.getOperand(0).getValueType();
4814*9880d681SAndroid Build Coastguard Worker EVT ShiftCountVT = N0.getOperand(0)->getOperand(1).getValueType();
4815*9880d681SAndroid Build Coastguard Worker uint64_t InnerShiftSize = InnerShiftVT.getScalarType().getSizeInBits();
4816*9880d681SAndroid Build Coastguard Worker // This is only valid if the OpSizeInBits + c1 = size of inner shift.
4817*9880d681SAndroid Build Coastguard Worker if (c1 + OpSizeInBits == InnerShiftSize) {
4818*9880d681SAndroid Build Coastguard Worker SDLoc DL(N0);
4819*9880d681SAndroid Build Coastguard Worker if (c1 + c2 >= InnerShiftSize)
4820*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(0, DL, VT);
4821*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TRUNCATE, DL, VT,
4822*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::SRL, DL, InnerShiftVT,
4823*9880d681SAndroid Build Coastguard Worker N0.getOperand(0)->getOperand(0),
4824*9880d681SAndroid Build Coastguard Worker DAG.getConstant(c1 + c2, DL,
4825*9880d681SAndroid Build Coastguard Worker ShiftCountVT)));
4826*9880d681SAndroid Build Coastguard Worker }
4827*9880d681SAndroid Build Coastguard Worker }
4828*9880d681SAndroid Build Coastguard Worker
4829*9880d681SAndroid Build Coastguard Worker // fold (srl (shl x, c), c) -> (and x, cst2)
4830*9880d681SAndroid Build Coastguard Worker if (N1C && N0.getOpcode() == ISD::SHL && N0.getOperand(1) == N1) {
4831*9880d681SAndroid Build Coastguard Worker unsigned BitSize = N0.getScalarValueSizeInBits();
4832*9880d681SAndroid Build Coastguard Worker if (BitSize <= 64) {
4833*9880d681SAndroid Build Coastguard Worker uint64_t ShAmt = N1C->getZExtValue() + 64 - BitSize;
4834*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
4835*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, DL, VT, N0.getOperand(0),
4836*9880d681SAndroid Build Coastguard Worker DAG.getConstant(~0ULL >> ShAmt, DL, VT));
4837*9880d681SAndroid Build Coastguard Worker }
4838*9880d681SAndroid Build Coastguard Worker }
4839*9880d681SAndroid Build Coastguard Worker
4840*9880d681SAndroid Build Coastguard Worker // fold (srl (anyextend x), c) -> (and (anyextend (srl x, c)), mask)
4841*9880d681SAndroid Build Coastguard Worker if (N1C && N0.getOpcode() == ISD::ANY_EXTEND) {
4842*9880d681SAndroid Build Coastguard Worker // Shifting in all undef bits?
4843*9880d681SAndroid Build Coastguard Worker EVT SmallVT = N0.getOperand(0).getValueType();
4844*9880d681SAndroid Build Coastguard Worker unsigned BitSize = SmallVT.getScalarSizeInBits();
4845*9880d681SAndroid Build Coastguard Worker if (N1C->getZExtValue() >= BitSize)
4846*9880d681SAndroid Build Coastguard Worker return DAG.getUNDEF(VT);
4847*9880d681SAndroid Build Coastguard Worker
4848*9880d681SAndroid Build Coastguard Worker if (!LegalTypes || TLI.isTypeDesirableForOp(ISD::SRL, SmallVT)) {
4849*9880d681SAndroid Build Coastguard Worker uint64_t ShiftAmt = N1C->getZExtValue();
4850*9880d681SAndroid Build Coastguard Worker SDLoc DL0(N0);
4851*9880d681SAndroid Build Coastguard Worker SDValue SmallShift = DAG.getNode(ISD::SRL, DL0, SmallVT,
4852*9880d681SAndroid Build Coastguard Worker N0.getOperand(0),
4853*9880d681SAndroid Build Coastguard Worker DAG.getConstant(ShiftAmt, DL0,
4854*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(SmallVT)));
4855*9880d681SAndroid Build Coastguard Worker AddToWorklist(SmallShift.getNode());
4856*9880d681SAndroid Build Coastguard Worker APInt Mask = APInt::getAllOnesValue(OpSizeInBits).lshr(ShiftAmt);
4857*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
4858*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, DL, VT,
4859*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::ANY_EXTEND, DL, VT, SmallShift),
4860*9880d681SAndroid Build Coastguard Worker DAG.getConstant(Mask, DL, VT));
4861*9880d681SAndroid Build Coastguard Worker }
4862*9880d681SAndroid Build Coastguard Worker }
4863*9880d681SAndroid Build Coastguard Worker
4864*9880d681SAndroid Build Coastguard Worker // fold (srl (sra X, Y), 31) -> (srl X, 31). This srl only looks at the sign
4865*9880d681SAndroid Build Coastguard Worker // bit, which is unmodified by sra.
4866*9880d681SAndroid Build Coastguard Worker if (N1C && N1C->getZExtValue() + 1 == OpSizeInBits) {
4867*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SRA)
4868*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SRL, SDLoc(N), VT, N0.getOperand(0), N1);
4869*9880d681SAndroid Build Coastguard Worker }
4870*9880d681SAndroid Build Coastguard Worker
4871*9880d681SAndroid Build Coastguard Worker // fold (srl (ctlz x), "5") -> x iff x has one bit set (the low bit).
4872*9880d681SAndroid Build Coastguard Worker if (N1C && N0.getOpcode() == ISD::CTLZ &&
4873*9880d681SAndroid Build Coastguard Worker N1C->getAPIntValue() == Log2_32(OpSizeInBits)) {
4874*9880d681SAndroid Build Coastguard Worker APInt KnownZero, KnownOne;
4875*9880d681SAndroid Build Coastguard Worker DAG.computeKnownBits(N0.getOperand(0), KnownZero, KnownOne);
4876*9880d681SAndroid Build Coastguard Worker
4877*9880d681SAndroid Build Coastguard Worker // If any of the input bits are KnownOne, then the input couldn't be all
4878*9880d681SAndroid Build Coastguard Worker // zeros, thus the result of the srl will always be zero.
4879*9880d681SAndroid Build Coastguard Worker if (KnownOne.getBoolValue()) return DAG.getConstant(0, SDLoc(N0), VT);
4880*9880d681SAndroid Build Coastguard Worker
4881*9880d681SAndroid Build Coastguard Worker // If all of the bits input the to ctlz node are known to be zero, then
4882*9880d681SAndroid Build Coastguard Worker // the result of the ctlz is "32" and the result of the shift is one.
4883*9880d681SAndroid Build Coastguard Worker APInt UnknownBits = ~KnownZero;
4884*9880d681SAndroid Build Coastguard Worker if (UnknownBits == 0) return DAG.getConstant(1, SDLoc(N0), VT);
4885*9880d681SAndroid Build Coastguard Worker
4886*9880d681SAndroid Build Coastguard Worker // Otherwise, check to see if there is exactly one bit input to the ctlz.
4887*9880d681SAndroid Build Coastguard Worker if ((UnknownBits & (UnknownBits - 1)) == 0) {
4888*9880d681SAndroid Build Coastguard Worker // Okay, we know that only that the single bit specified by UnknownBits
4889*9880d681SAndroid Build Coastguard Worker // could be set on input to the CTLZ node. If this bit is set, the SRL
4890*9880d681SAndroid Build Coastguard Worker // will return 0, if it is clear, it returns 1. Change the CTLZ/SRL pair
4891*9880d681SAndroid Build Coastguard Worker // to an SRL/XOR pair, which is likely to simplify more.
4892*9880d681SAndroid Build Coastguard Worker unsigned ShAmt = UnknownBits.countTrailingZeros();
4893*9880d681SAndroid Build Coastguard Worker SDValue Op = N0.getOperand(0);
4894*9880d681SAndroid Build Coastguard Worker
4895*9880d681SAndroid Build Coastguard Worker if (ShAmt) {
4896*9880d681SAndroid Build Coastguard Worker SDLoc DL(N0);
4897*9880d681SAndroid Build Coastguard Worker Op = DAG.getNode(ISD::SRL, DL, VT, Op,
4898*9880d681SAndroid Build Coastguard Worker DAG.getConstant(ShAmt, DL,
4899*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(Op.getValueType())));
4900*9880d681SAndroid Build Coastguard Worker AddToWorklist(Op.getNode());
4901*9880d681SAndroid Build Coastguard Worker }
4902*9880d681SAndroid Build Coastguard Worker
4903*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
4904*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::XOR, DL, VT,
4905*9880d681SAndroid Build Coastguard Worker Op, DAG.getConstant(1, DL, VT));
4906*9880d681SAndroid Build Coastguard Worker }
4907*9880d681SAndroid Build Coastguard Worker }
4908*9880d681SAndroid Build Coastguard Worker
4909*9880d681SAndroid Build Coastguard Worker // fold (srl x, (trunc (and y, c))) -> (srl x, (and (trunc y), (trunc c))).
4910*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::TRUNCATE &&
4911*9880d681SAndroid Build Coastguard Worker N1.getOperand(0).getOpcode() == ISD::AND) {
4912*9880d681SAndroid Build Coastguard Worker if (SDValue NewOp1 = distributeTruncateThroughAnd(N1.getNode()))
4913*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SRL, SDLoc(N), VT, N0, NewOp1);
4914*9880d681SAndroid Build Coastguard Worker }
4915*9880d681SAndroid Build Coastguard Worker
4916*9880d681SAndroid Build Coastguard Worker // fold operands of srl based on knowledge that the low bits are not
4917*9880d681SAndroid Build Coastguard Worker // demanded.
4918*9880d681SAndroid Build Coastguard Worker if (N1C && SimplifyDemandedBits(SDValue(N, 0)))
4919*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0);
4920*9880d681SAndroid Build Coastguard Worker
4921*9880d681SAndroid Build Coastguard Worker if (N1C && !N1C->isOpaque())
4922*9880d681SAndroid Build Coastguard Worker if (SDValue NewSRL = visitShiftByConstant(N, N1C))
4923*9880d681SAndroid Build Coastguard Worker return NewSRL;
4924*9880d681SAndroid Build Coastguard Worker
4925*9880d681SAndroid Build Coastguard Worker // Attempt to convert a srl of a load into a narrower zero-extending load.
4926*9880d681SAndroid Build Coastguard Worker if (SDValue NarrowLoad = ReduceLoadWidth(N))
4927*9880d681SAndroid Build Coastguard Worker return NarrowLoad;
4928*9880d681SAndroid Build Coastguard Worker
4929*9880d681SAndroid Build Coastguard Worker // Here is a common situation. We want to optimize:
4930*9880d681SAndroid Build Coastguard Worker //
4931*9880d681SAndroid Build Coastguard Worker // %a = ...
4932*9880d681SAndroid Build Coastguard Worker // %b = and i32 %a, 2
4933*9880d681SAndroid Build Coastguard Worker // %c = srl i32 %b, 1
4934*9880d681SAndroid Build Coastguard Worker // brcond i32 %c ...
4935*9880d681SAndroid Build Coastguard Worker //
4936*9880d681SAndroid Build Coastguard Worker // into
4937*9880d681SAndroid Build Coastguard Worker //
4938*9880d681SAndroid Build Coastguard Worker // %a = ...
4939*9880d681SAndroid Build Coastguard Worker // %b = and %a, 2
4940*9880d681SAndroid Build Coastguard Worker // %c = setcc eq %b, 0
4941*9880d681SAndroid Build Coastguard Worker // brcond %c ...
4942*9880d681SAndroid Build Coastguard Worker //
4943*9880d681SAndroid Build Coastguard Worker // However when after the source operand of SRL is optimized into AND, the SRL
4944*9880d681SAndroid Build Coastguard Worker // itself may not be optimized further. Look for it and add the BRCOND into
4945*9880d681SAndroid Build Coastguard Worker // the worklist.
4946*9880d681SAndroid Build Coastguard Worker if (N->hasOneUse()) {
4947*9880d681SAndroid Build Coastguard Worker SDNode *Use = *N->use_begin();
4948*9880d681SAndroid Build Coastguard Worker if (Use->getOpcode() == ISD::BRCOND)
4949*9880d681SAndroid Build Coastguard Worker AddToWorklist(Use);
4950*9880d681SAndroid Build Coastguard Worker else if (Use->getOpcode() == ISD::TRUNCATE && Use->hasOneUse()) {
4951*9880d681SAndroid Build Coastguard Worker // Also look pass the truncate.
4952*9880d681SAndroid Build Coastguard Worker Use = *Use->use_begin();
4953*9880d681SAndroid Build Coastguard Worker if (Use->getOpcode() == ISD::BRCOND)
4954*9880d681SAndroid Build Coastguard Worker AddToWorklist(Use);
4955*9880d681SAndroid Build Coastguard Worker }
4956*9880d681SAndroid Build Coastguard Worker }
4957*9880d681SAndroid Build Coastguard Worker
4958*9880d681SAndroid Build Coastguard Worker return SDValue();
4959*9880d681SAndroid Build Coastguard Worker }
4960*9880d681SAndroid Build Coastguard Worker
visitBSWAP(SDNode * N)4961*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitBSWAP(SDNode *N) {
4962*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
4963*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
4964*9880d681SAndroid Build Coastguard Worker
4965*9880d681SAndroid Build Coastguard Worker // fold (bswap c1) -> c2
4966*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N0))
4967*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::BSWAP, SDLoc(N), VT, N0);
4968*9880d681SAndroid Build Coastguard Worker // fold (bswap (bswap x)) -> x
4969*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::BSWAP)
4970*9880d681SAndroid Build Coastguard Worker return N0->getOperand(0);
4971*9880d681SAndroid Build Coastguard Worker return SDValue();
4972*9880d681SAndroid Build Coastguard Worker }
4973*9880d681SAndroid Build Coastguard Worker
visitBITREVERSE(SDNode * N)4974*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitBITREVERSE(SDNode *N) {
4975*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
4976*9880d681SAndroid Build Coastguard Worker
4977*9880d681SAndroid Build Coastguard Worker // fold (bitreverse (bitreverse x)) -> x
4978*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::BITREVERSE)
4979*9880d681SAndroid Build Coastguard Worker return N0.getOperand(0);
4980*9880d681SAndroid Build Coastguard Worker return SDValue();
4981*9880d681SAndroid Build Coastguard Worker }
4982*9880d681SAndroid Build Coastguard Worker
visitCTLZ(SDNode * N)4983*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitCTLZ(SDNode *N) {
4984*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
4985*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
4986*9880d681SAndroid Build Coastguard Worker
4987*9880d681SAndroid Build Coastguard Worker // fold (ctlz c1) -> c2
4988*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N0))
4989*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::CTLZ, SDLoc(N), VT, N0);
4990*9880d681SAndroid Build Coastguard Worker return SDValue();
4991*9880d681SAndroid Build Coastguard Worker }
4992*9880d681SAndroid Build Coastguard Worker
visitCTLZ_ZERO_UNDEF(SDNode * N)4993*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitCTLZ_ZERO_UNDEF(SDNode *N) {
4994*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
4995*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
4996*9880d681SAndroid Build Coastguard Worker
4997*9880d681SAndroid Build Coastguard Worker // fold (ctlz_zero_undef c1) -> c2
4998*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N0))
4999*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::CTLZ_ZERO_UNDEF, SDLoc(N), VT, N0);
5000*9880d681SAndroid Build Coastguard Worker return SDValue();
5001*9880d681SAndroid Build Coastguard Worker }
5002*9880d681SAndroid Build Coastguard Worker
visitCTTZ(SDNode * N)5003*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitCTTZ(SDNode *N) {
5004*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
5005*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
5006*9880d681SAndroid Build Coastguard Worker
5007*9880d681SAndroid Build Coastguard Worker // fold (cttz c1) -> c2
5008*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N0))
5009*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::CTTZ, SDLoc(N), VT, N0);
5010*9880d681SAndroid Build Coastguard Worker return SDValue();
5011*9880d681SAndroid Build Coastguard Worker }
5012*9880d681SAndroid Build Coastguard Worker
visitCTTZ_ZERO_UNDEF(SDNode * N)5013*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitCTTZ_ZERO_UNDEF(SDNode *N) {
5014*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
5015*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
5016*9880d681SAndroid Build Coastguard Worker
5017*9880d681SAndroid Build Coastguard Worker // fold (cttz_zero_undef c1) -> c2
5018*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N0))
5019*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::CTTZ_ZERO_UNDEF, SDLoc(N), VT, N0);
5020*9880d681SAndroid Build Coastguard Worker return SDValue();
5021*9880d681SAndroid Build Coastguard Worker }
5022*9880d681SAndroid Build Coastguard Worker
visitCTPOP(SDNode * N)5023*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitCTPOP(SDNode *N) {
5024*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
5025*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
5026*9880d681SAndroid Build Coastguard Worker
5027*9880d681SAndroid Build Coastguard Worker // fold (ctpop c1) -> c2
5028*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N0))
5029*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::CTPOP, SDLoc(N), VT, N0);
5030*9880d681SAndroid Build Coastguard Worker return SDValue();
5031*9880d681SAndroid Build Coastguard Worker }
5032*9880d681SAndroid Build Coastguard Worker
5033*9880d681SAndroid Build Coastguard Worker
5034*9880d681SAndroid Build Coastguard Worker /// \brief Generate Min/Max node
combineMinNumMaxNum(const SDLoc & DL,EVT VT,SDValue LHS,SDValue RHS,SDValue True,SDValue False,ISD::CondCode CC,const TargetLowering & TLI,SelectionDAG & DAG)5035*9880d681SAndroid Build Coastguard Worker static SDValue combineMinNumMaxNum(const SDLoc &DL, EVT VT, SDValue LHS,
5036*9880d681SAndroid Build Coastguard Worker SDValue RHS, SDValue True, SDValue False,
5037*9880d681SAndroid Build Coastguard Worker ISD::CondCode CC, const TargetLowering &TLI,
5038*9880d681SAndroid Build Coastguard Worker SelectionDAG &DAG) {
5039*9880d681SAndroid Build Coastguard Worker if (!(LHS == True && RHS == False) && !(LHS == False && RHS == True))
5040*9880d681SAndroid Build Coastguard Worker return SDValue();
5041*9880d681SAndroid Build Coastguard Worker
5042*9880d681SAndroid Build Coastguard Worker switch (CC) {
5043*9880d681SAndroid Build Coastguard Worker case ISD::SETOLT:
5044*9880d681SAndroid Build Coastguard Worker case ISD::SETOLE:
5045*9880d681SAndroid Build Coastguard Worker case ISD::SETLT:
5046*9880d681SAndroid Build Coastguard Worker case ISD::SETLE:
5047*9880d681SAndroid Build Coastguard Worker case ISD::SETULT:
5048*9880d681SAndroid Build Coastguard Worker case ISD::SETULE: {
5049*9880d681SAndroid Build Coastguard Worker unsigned Opcode = (LHS == True) ? ISD::FMINNUM : ISD::FMAXNUM;
5050*9880d681SAndroid Build Coastguard Worker if (TLI.isOperationLegal(Opcode, VT))
5051*9880d681SAndroid Build Coastguard Worker return DAG.getNode(Opcode, DL, VT, LHS, RHS);
5052*9880d681SAndroid Build Coastguard Worker return SDValue();
5053*9880d681SAndroid Build Coastguard Worker }
5054*9880d681SAndroid Build Coastguard Worker case ISD::SETOGT:
5055*9880d681SAndroid Build Coastguard Worker case ISD::SETOGE:
5056*9880d681SAndroid Build Coastguard Worker case ISD::SETGT:
5057*9880d681SAndroid Build Coastguard Worker case ISD::SETGE:
5058*9880d681SAndroid Build Coastguard Worker case ISD::SETUGT:
5059*9880d681SAndroid Build Coastguard Worker case ISD::SETUGE: {
5060*9880d681SAndroid Build Coastguard Worker unsigned Opcode = (LHS == True) ? ISD::FMAXNUM : ISD::FMINNUM;
5061*9880d681SAndroid Build Coastguard Worker if (TLI.isOperationLegal(Opcode, VT))
5062*9880d681SAndroid Build Coastguard Worker return DAG.getNode(Opcode, DL, VT, LHS, RHS);
5063*9880d681SAndroid Build Coastguard Worker return SDValue();
5064*9880d681SAndroid Build Coastguard Worker }
5065*9880d681SAndroid Build Coastguard Worker default:
5066*9880d681SAndroid Build Coastguard Worker return SDValue();
5067*9880d681SAndroid Build Coastguard Worker }
5068*9880d681SAndroid Build Coastguard Worker }
5069*9880d681SAndroid Build Coastguard Worker
visitSELECT(SDNode * N)5070*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSELECT(SDNode *N) {
5071*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
5072*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
5073*9880d681SAndroid Build Coastguard Worker SDValue N2 = N->getOperand(2);
5074*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
5075*9880d681SAndroid Build Coastguard Worker EVT VT0 = N0.getValueType();
5076*9880d681SAndroid Build Coastguard Worker
5077*9880d681SAndroid Build Coastguard Worker // fold (select C, X, X) -> X
5078*9880d681SAndroid Build Coastguard Worker if (N1 == N2)
5079*9880d681SAndroid Build Coastguard Worker return N1;
5080*9880d681SAndroid Build Coastguard Worker if (const ConstantSDNode *N0C = dyn_cast<const ConstantSDNode>(N0)) {
5081*9880d681SAndroid Build Coastguard Worker // fold (select true, X, Y) -> X
5082*9880d681SAndroid Build Coastguard Worker // fold (select false, X, Y) -> Y
5083*9880d681SAndroid Build Coastguard Worker return !N0C->isNullValue() ? N1 : N2;
5084*9880d681SAndroid Build Coastguard Worker }
5085*9880d681SAndroid Build Coastguard Worker // fold (select C, 1, X) -> (or C, X)
5086*9880d681SAndroid Build Coastguard Worker if (VT == MVT::i1 && isOneConstant(N1))
5087*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::OR, SDLoc(N), VT, N0, N2);
5088*9880d681SAndroid Build Coastguard Worker // fold (select C, 0, 1) -> (xor C, 1)
5089*9880d681SAndroid Build Coastguard Worker // We can't do this reliably if integer based booleans have different contents
5090*9880d681SAndroid Build Coastguard Worker // to floating point based booleans. This is because we can't tell whether we
5091*9880d681SAndroid Build Coastguard Worker // have an integer-based boolean or a floating-point-based boolean unless we
5092*9880d681SAndroid Build Coastguard Worker // can find the SETCC that produced it and inspect its operands. This is
5093*9880d681SAndroid Build Coastguard Worker // fairly easy if C is the SETCC node, but it can potentially be
5094*9880d681SAndroid Build Coastguard Worker // undiscoverable (or not reasonably discoverable). For example, it could be
5095*9880d681SAndroid Build Coastguard Worker // in another basic block or it could require searching a complicated
5096*9880d681SAndroid Build Coastguard Worker // expression.
5097*9880d681SAndroid Build Coastguard Worker if (VT.isInteger() &&
5098*9880d681SAndroid Build Coastguard Worker (VT0 == MVT::i1 || (VT0.isInteger() &&
5099*9880d681SAndroid Build Coastguard Worker TLI.getBooleanContents(false, false) ==
5100*9880d681SAndroid Build Coastguard Worker TLI.getBooleanContents(false, true) &&
5101*9880d681SAndroid Build Coastguard Worker TLI.getBooleanContents(false, false) ==
5102*9880d681SAndroid Build Coastguard Worker TargetLowering::ZeroOrOneBooleanContent)) &&
5103*9880d681SAndroid Build Coastguard Worker isNullConstant(N1) && isOneConstant(N2)) {
5104*9880d681SAndroid Build Coastguard Worker SDValue XORNode;
5105*9880d681SAndroid Build Coastguard Worker if (VT == VT0) {
5106*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
5107*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::XOR, DL, VT0,
5108*9880d681SAndroid Build Coastguard Worker N0, DAG.getConstant(1, DL, VT0));
5109*9880d681SAndroid Build Coastguard Worker }
5110*9880d681SAndroid Build Coastguard Worker SDLoc DL0(N0);
5111*9880d681SAndroid Build Coastguard Worker XORNode = DAG.getNode(ISD::XOR, DL0, VT0,
5112*9880d681SAndroid Build Coastguard Worker N0, DAG.getConstant(1, DL0, VT0));
5113*9880d681SAndroid Build Coastguard Worker AddToWorklist(XORNode.getNode());
5114*9880d681SAndroid Build Coastguard Worker if (VT.bitsGT(VT0))
5115*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ZERO_EXTEND, SDLoc(N), VT, XORNode);
5116*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TRUNCATE, SDLoc(N), VT, XORNode);
5117*9880d681SAndroid Build Coastguard Worker }
5118*9880d681SAndroid Build Coastguard Worker // fold (select C, 0, X) -> (and (not C), X)
5119*9880d681SAndroid Build Coastguard Worker if (VT == VT0 && VT == MVT::i1 && isNullConstant(N1)) {
5120*9880d681SAndroid Build Coastguard Worker SDValue NOTNode = DAG.getNOT(SDLoc(N0), N0, VT);
5121*9880d681SAndroid Build Coastguard Worker AddToWorklist(NOTNode.getNode());
5122*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, SDLoc(N), VT, NOTNode, N2);
5123*9880d681SAndroid Build Coastguard Worker }
5124*9880d681SAndroid Build Coastguard Worker // fold (select C, X, 1) -> (or (not C), X)
5125*9880d681SAndroid Build Coastguard Worker if (VT == VT0 && VT == MVT::i1 && isOneConstant(N2)) {
5126*9880d681SAndroid Build Coastguard Worker SDValue NOTNode = DAG.getNOT(SDLoc(N0), N0, VT);
5127*9880d681SAndroid Build Coastguard Worker AddToWorklist(NOTNode.getNode());
5128*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::OR, SDLoc(N), VT, NOTNode, N1);
5129*9880d681SAndroid Build Coastguard Worker }
5130*9880d681SAndroid Build Coastguard Worker // fold (select C, X, 0) -> (and C, X)
5131*9880d681SAndroid Build Coastguard Worker if (VT == MVT::i1 && isNullConstant(N2))
5132*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, SDLoc(N), VT, N0, N1);
5133*9880d681SAndroid Build Coastguard Worker // fold (select X, X, Y) -> (or X, Y)
5134*9880d681SAndroid Build Coastguard Worker // fold (select X, 1, Y) -> (or X, Y)
5135*9880d681SAndroid Build Coastguard Worker if (VT == MVT::i1 && (N0 == N1 || isOneConstant(N1)))
5136*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::OR, SDLoc(N), VT, N0, N2);
5137*9880d681SAndroid Build Coastguard Worker // fold (select X, Y, X) -> (and X, Y)
5138*9880d681SAndroid Build Coastguard Worker // fold (select X, Y, 0) -> (and X, Y)
5139*9880d681SAndroid Build Coastguard Worker if (VT == MVT::i1 && (N0 == N2 || isNullConstant(N2)))
5140*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, SDLoc(N), VT, N0, N1);
5141*9880d681SAndroid Build Coastguard Worker
5142*9880d681SAndroid Build Coastguard Worker // If we can fold this based on the true/false value, do so.
5143*9880d681SAndroid Build Coastguard Worker if (SimplifySelectOps(N, N1, N2))
5144*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Don't revisit N.
5145*9880d681SAndroid Build Coastguard Worker
5146*9880d681SAndroid Build Coastguard Worker if (VT0 == MVT::i1) {
5147*9880d681SAndroid Build Coastguard Worker // The code in this block deals with the following 2 equivalences:
5148*9880d681SAndroid Build Coastguard Worker // select(C0|C1, x, y) <=> select(C0, x, select(C1, x, y))
5149*9880d681SAndroid Build Coastguard Worker // select(C0&C1, x, y) <=> select(C0, select(C1, x, y), y)
5150*9880d681SAndroid Build Coastguard Worker // The target can specify its prefered form with the
5151*9880d681SAndroid Build Coastguard Worker // shouldNormalizeToSelectSequence() callback. However we always transform
5152*9880d681SAndroid Build Coastguard Worker // to the right anyway if we find the inner select exists in the DAG anyway
5153*9880d681SAndroid Build Coastguard Worker // and we always transform to the left side if we know that we can further
5154*9880d681SAndroid Build Coastguard Worker // optimize the combination of the conditions.
5155*9880d681SAndroid Build Coastguard Worker bool normalizeToSequence
5156*9880d681SAndroid Build Coastguard Worker = TLI.shouldNormalizeToSelectSequence(*DAG.getContext(), VT);
5157*9880d681SAndroid Build Coastguard Worker // select (and Cond0, Cond1), X, Y
5158*9880d681SAndroid Build Coastguard Worker // -> select Cond0, (select Cond1, X, Y), Y
5159*9880d681SAndroid Build Coastguard Worker if (N0->getOpcode() == ISD::AND && N0->hasOneUse()) {
5160*9880d681SAndroid Build Coastguard Worker SDValue Cond0 = N0->getOperand(0);
5161*9880d681SAndroid Build Coastguard Worker SDValue Cond1 = N0->getOperand(1);
5162*9880d681SAndroid Build Coastguard Worker SDValue InnerSelect = DAG.getNode(ISD::SELECT, SDLoc(N),
5163*9880d681SAndroid Build Coastguard Worker N1.getValueType(), Cond1, N1, N2);
5164*9880d681SAndroid Build Coastguard Worker if (normalizeToSequence || !InnerSelect.use_empty())
5165*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SELECT, SDLoc(N), N1.getValueType(), Cond0,
5166*9880d681SAndroid Build Coastguard Worker InnerSelect, N2);
5167*9880d681SAndroid Build Coastguard Worker }
5168*9880d681SAndroid Build Coastguard Worker // select (or Cond0, Cond1), X, Y -> select Cond0, X, (select Cond1, X, Y)
5169*9880d681SAndroid Build Coastguard Worker if (N0->getOpcode() == ISD::OR && N0->hasOneUse()) {
5170*9880d681SAndroid Build Coastguard Worker SDValue Cond0 = N0->getOperand(0);
5171*9880d681SAndroid Build Coastguard Worker SDValue Cond1 = N0->getOperand(1);
5172*9880d681SAndroid Build Coastguard Worker SDValue InnerSelect = DAG.getNode(ISD::SELECT, SDLoc(N),
5173*9880d681SAndroid Build Coastguard Worker N1.getValueType(), Cond1, N1, N2);
5174*9880d681SAndroid Build Coastguard Worker if (normalizeToSequence || !InnerSelect.use_empty())
5175*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SELECT, SDLoc(N), N1.getValueType(), Cond0, N1,
5176*9880d681SAndroid Build Coastguard Worker InnerSelect);
5177*9880d681SAndroid Build Coastguard Worker }
5178*9880d681SAndroid Build Coastguard Worker
5179*9880d681SAndroid Build Coastguard Worker // select Cond0, (select Cond1, X, Y), Y -> select (and Cond0, Cond1), X, Y
5180*9880d681SAndroid Build Coastguard Worker if (N1->getOpcode() == ISD::SELECT && N1->hasOneUse()) {
5181*9880d681SAndroid Build Coastguard Worker SDValue N1_0 = N1->getOperand(0);
5182*9880d681SAndroid Build Coastguard Worker SDValue N1_1 = N1->getOperand(1);
5183*9880d681SAndroid Build Coastguard Worker SDValue N1_2 = N1->getOperand(2);
5184*9880d681SAndroid Build Coastguard Worker if (N1_2 == N2 && N0.getValueType() == N1_0.getValueType()) {
5185*9880d681SAndroid Build Coastguard Worker // Create the actual and node if we can generate good code for it.
5186*9880d681SAndroid Build Coastguard Worker if (!normalizeToSequence) {
5187*9880d681SAndroid Build Coastguard Worker SDValue And = DAG.getNode(ISD::AND, SDLoc(N), N0.getValueType(),
5188*9880d681SAndroid Build Coastguard Worker N0, N1_0);
5189*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SELECT, SDLoc(N), N1.getValueType(), And,
5190*9880d681SAndroid Build Coastguard Worker N1_1, N2);
5191*9880d681SAndroid Build Coastguard Worker }
5192*9880d681SAndroid Build Coastguard Worker // Otherwise see if we can optimize the "and" to a better pattern.
5193*9880d681SAndroid Build Coastguard Worker if (SDValue Combined = visitANDLike(N0, N1_0, N))
5194*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SELECT, SDLoc(N), N1.getValueType(), Combined,
5195*9880d681SAndroid Build Coastguard Worker N1_1, N2);
5196*9880d681SAndroid Build Coastguard Worker }
5197*9880d681SAndroid Build Coastguard Worker }
5198*9880d681SAndroid Build Coastguard Worker // select Cond0, X, (select Cond1, X, Y) -> select (or Cond0, Cond1), X, Y
5199*9880d681SAndroid Build Coastguard Worker if (N2->getOpcode() == ISD::SELECT && N2->hasOneUse()) {
5200*9880d681SAndroid Build Coastguard Worker SDValue N2_0 = N2->getOperand(0);
5201*9880d681SAndroid Build Coastguard Worker SDValue N2_1 = N2->getOperand(1);
5202*9880d681SAndroid Build Coastguard Worker SDValue N2_2 = N2->getOperand(2);
5203*9880d681SAndroid Build Coastguard Worker if (N2_1 == N1 && N0.getValueType() == N2_0.getValueType()) {
5204*9880d681SAndroid Build Coastguard Worker // Create the actual or node if we can generate good code for it.
5205*9880d681SAndroid Build Coastguard Worker if (!normalizeToSequence) {
5206*9880d681SAndroid Build Coastguard Worker SDValue Or = DAG.getNode(ISD::OR, SDLoc(N), N0.getValueType(),
5207*9880d681SAndroid Build Coastguard Worker N0, N2_0);
5208*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SELECT, SDLoc(N), N1.getValueType(), Or,
5209*9880d681SAndroid Build Coastguard Worker N1, N2_2);
5210*9880d681SAndroid Build Coastguard Worker }
5211*9880d681SAndroid Build Coastguard Worker // Otherwise see if we can optimize to a better pattern.
5212*9880d681SAndroid Build Coastguard Worker if (SDValue Combined = visitORLike(N0, N2_0, N))
5213*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SELECT, SDLoc(N), N1.getValueType(), Combined,
5214*9880d681SAndroid Build Coastguard Worker N1, N2_2);
5215*9880d681SAndroid Build Coastguard Worker }
5216*9880d681SAndroid Build Coastguard Worker }
5217*9880d681SAndroid Build Coastguard Worker }
5218*9880d681SAndroid Build Coastguard Worker
5219*9880d681SAndroid Build Coastguard Worker // fold selects based on a setcc into other things, such as min/max/abs
5220*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SETCC) {
5221*9880d681SAndroid Build Coastguard Worker // select x, y (fcmp lt x, y) -> fminnum x, y
5222*9880d681SAndroid Build Coastguard Worker // select x, y (fcmp gt x, y) -> fmaxnum x, y
5223*9880d681SAndroid Build Coastguard Worker //
5224*9880d681SAndroid Build Coastguard Worker // This is OK if we don't care about what happens if either operand is a
5225*9880d681SAndroid Build Coastguard Worker // NaN.
5226*9880d681SAndroid Build Coastguard Worker //
5227*9880d681SAndroid Build Coastguard Worker
5228*9880d681SAndroid Build Coastguard Worker // FIXME: Instead of testing for UnsafeFPMath, this should be checking for
5229*9880d681SAndroid Build Coastguard Worker // no signed zeros as well as no nans.
5230*9880d681SAndroid Build Coastguard Worker const TargetOptions &Options = DAG.getTarget().Options;
5231*9880d681SAndroid Build Coastguard Worker if (Options.UnsafeFPMath &&
5232*9880d681SAndroid Build Coastguard Worker VT.isFloatingPoint() && N0.hasOneUse() &&
5233*9880d681SAndroid Build Coastguard Worker DAG.isKnownNeverNaN(N1) && DAG.isKnownNeverNaN(N2)) {
5234*9880d681SAndroid Build Coastguard Worker ISD::CondCode CC = cast<CondCodeSDNode>(N0.getOperand(2))->get();
5235*9880d681SAndroid Build Coastguard Worker
5236*9880d681SAndroid Build Coastguard Worker if (SDValue FMinMax = combineMinNumMaxNum(SDLoc(N), VT, N0.getOperand(0),
5237*9880d681SAndroid Build Coastguard Worker N0.getOperand(1), N1, N2, CC,
5238*9880d681SAndroid Build Coastguard Worker TLI, DAG))
5239*9880d681SAndroid Build Coastguard Worker return FMinMax;
5240*9880d681SAndroid Build Coastguard Worker }
5241*9880d681SAndroid Build Coastguard Worker
5242*9880d681SAndroid Build Coastguard Worker if ((!LegalOperations &&
5243*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegalOrCustom(ISD::SELECT_CC, VT)) ||
5244*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegal(ISD::SELECT_CC, VT))
5245*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SELECT_CC, SDLoc(N), VT,
5246*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N0.getOperand(1),
5247*9880d681SAndroid Build Coastguard Worker N1, N2, N0.getOperand(2));
5248*9880d681SAndroid Build Coastguard Worker return SimplifySelect(SDLoc(N), N0, N1, N2);
5249*9880d681SAndroid Build Coastguard Worker }
5250*9880d681SAndroid Build Coastguard Worker
5251*9880d681SAndroid Build Coastguard Worker return SDValue();
5252*9880d681SAndroid Build Coastguard Worker }
5253*9880d681SAndroid Build Coastguard Worker
5254*9880d681SAndroid Build Coastguard Worker static
SplitVSETCC(const SDNode * N,SelectionDAG & DAG)5255*9880d681SAndroid Build Coastguard Worker std::pair<SDValue, SDValue> SplitVSETCC(const SDNode *N, SelectionDAG &DAG) {
5256*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
5257*9880d681SAndroid Build Coastguard Worker EVT LoVT, HiVT;
5258*9880d681SAndroid Build Coastguard Worker std::tie(LoVT, HiVT) = DAG.GetSplitDestVTs(N->getValueType(0));
5259*9880d681SAndroid Build Coastguard Worker
5260*9880d681SAndroid Build Coastguard Worker // Split the inputs.
5261*9880d681SAndroid Build Coastguard Worker SDValue Lo, Hi, LL, LH, RL, RH;
5262*9880d681SAndroid Build Coastguard Worker std::tie(LL, LH) = DAG.SplitVectorOperand(N, 0);
5263*9880d681SAndroid Build Coastguard Worker std::tie(RL, RH) = DAG.SplitVectorOperand(N, 1);
5264*9880d681SAndroid Build Coastguard Worker
5265*9880d681SAndroid Build Coastguard Worker Lo = DAG.getNode(N->getOpcode(), DL, LoVT, LL, RL, N->getOperand(2));
5266*9880d681SAndroid Build Coastguard Worker Hi = DAG.getNode(N->getOpcode(), DL, HiVT, LH, RH, N->getOperand(2));
5267*9880d681SAndroid Build Coastguard Worker
5268*9880d681SAndroid Build Coastguard Worker return std::make_pair(Lo, Hi);
5269*9880d681SAndroid Build Coastguard Worker }
5270*9880d681SAndroid Build Coastguard Worker
5271*9880d681SAndroid Build Coastguard Worker // This function assumes all the vselect's arguments are CONCAT_VECTOR
5272*9880d681SAndroid Build Coastguard Worker // nodes and that the condition is a BV of ConstantSDNodes (or undefs).
ConvertSelectToConcatVector(SDNode * N,SelectionDAG & DAG)5273*9880d681SAndroid Build Coastguard Worker static SDValue ConvertSelectToConcatVector(SDNode *N, SelectionDAG &DAG) {
5274*9880d681SAndroid Build Coastguard Worker SDLoc dl(N);
5275*9880d681SAndroid Build Coastguard Worker SDValue Cond = N->getOperand(0);
5276*9880d681SAndroid Build Coastguard Worker SDValue LHS = N->getOperand(1);
5277*9880d681SAndroid Build Coastguard Worker SDValue RHS = N->getOperand(2);
5278*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
5279*9880d681SAndroid Build Coastguard Worker int NumElems = VT.getVectorNumElements();
5280*9880d681SAndroid Build Coastguard Worker assert(LHS.getOpcode() == ISD::CONCAT_VECTORS &&
5281*9880d681SAndroid Build Coastguard Worker RHS.getOpcode() == ISD::CONCAT_VECTORS &&
5282*9880d681SAndroid Build Coastguard Worker Cond.getOpcode() == ISD::BUILD_VECTOR);
5283*9880d681SAndroid Build Coastguard Worker
5284*9880d681SAndroid Build Coastguard Worker // CONCAT_VECTOR can take an arbitrary number of arguments. We only care about
5285*9880d681SAndroid Build Coastguard Worker // binary ones here.
5286*9880d681SAndroid Build Coastguard Worker if (LHS->getNumOperands() != 2 || RHS->getNumOperands() != 2)
5287*9880d681SAndroid Build Coastguard Worker return SDValue();
5288*9880d681SAndroid Build Coastguard Worker
5289*9880d681SAndroid Build Coastguard Worker // We're sure we have an even number of elements due to the
5290*9880d681SAndroid Build Coastguard Worker // concat_vectors we have as arguments to vselect.
5291*9880d681SAndroid Build Coastguard Worker // Skip BV elements until we find one that's not an UNDEF
5292*9880d681SAndroid Build Coastguard Worker // After we find an UNDEF element, keep looping until we get to half the
5293*9880d681SAndroid Build Coastguard Worker // length of the BV and see if all the non-undef nodes are the same.
5294*9880d681SAndroid Build Coastguard Worker ConstantSDNode *BottomHalf = nullptr;
5295*9880d681SAndroid Build Coastguard Worker for (int i = 0; i < NumElems / 2; ++i) {
5296*9880d681SAndroid Build Coastguard Worker if (Cond->getOperand(i)->isUndef())
5297*9880d681SAndroid Build Coastguard Worker continue;
5298*9880d681SAndroid Build Coastguard Worker
5299*9880d681SAndroid Build Coastguard Worker if (BottomHalf == nullptr)
5300*9880d681SAndroid Build Coastguard Worker BottomHalf = cast<ConstantSDNode>(Cond.getOperand(i));
5301*9880d681SAndroid Build Coastguard Worker else if (Cond->getOperand(i).getNode() != BottomHalf)
5302*9880d681SAndroid Build Coastguard Worker return SDValue();
5303*9880d681SAndroid Build Coastguard Worker }
5304*9880d681SAndroid Build Coastguard Worker
5305*9880d681SAndroid Build Coastguard Worker // Do the same for the second half of the BuildVector
5306*9880d681SAndroid Build Coastguard Worker ConstantSDNode *TopHalf = nullptr;
5307*9880d681SAndroid Build Coastguard Worker for (int i = NumElems / 2; i < NumElems; ++i) {
5308*9880d681SAndroid Build Coastguard Worker if (Cond->getOperand(i)->isUndef())
5309*9880d681SAndroid Build Coastguard Worker continue;
5310*9880d681SAndroid Build Coastguard Worker
5311*9880d681SAndroid Build Coastguard Worker if (TopHalf == nullptr)
5312*9880d681SAndroid Build Coastguard Worker TopHalf = cast<ConstantSDNode>(Cond.getOperand(i));
5313*9880d681SAndroid Build Coastguard Worker else if (Cond->getOperand(i).getNode() != TopHalf)
5314*9880d681SAndroid Build Coastguard Worker return SDValue();
5315*9880d681SAndroid Build Coastguard Worker }
5316*9880d681SAndroid Build Coastguard Worker
5317*9880d681SAndroid Build Coastguard Worker assert(TopHalf && BottomHalf &&
5318*9880d681SAndroid Build Coastguard Worker "One half of the selector was all UNDEFs and the other was all the "
5319*9880d681SAndroid Build Coastguard Worker "same value. This should have been addressed before this function.");
5320*9880d681SAndroid Build Coastguard Worker return DAG.getNode(
5321*9880d681SAndroid Build Coastguard Worker ISD::CONCAT_VECTORS, dl, VT,
5322*9880d681SAndroid Build Coastguard Worker BottomHalf->isNullValue() ? RHS->getOperand(0) : LHS->getOperand(0),
5323*9880d681SAndroid Build Coastguard Worker TopHalf->isNullValue() ? RHS->getOperand(1) : LHS->getOperand(1));
5324*9880d681SAndroid Build Coastguard Worker }
5325*9880d681SAndroid Build Coastguard Worker
visitMSCATTER(SDNode * N)5326*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitMSCATTER(SDNode *N) {
5327*9880d681SAndroid Build Coastguard Worker
5328*9880d681SAndroid Build Coastguard Worker if (Level >= AfterLegalizeTypes)
5329*9880d681SAndroid Build Coastguard Worker return SDValue();
5330*9880d681SAndroid Build Coastguard Worker
5331*9880d681SAndroid Build Coastguard Worker MaskedScatterSDNode *MSC = cast<MaskedScatterSDNode>(N);
5332*9880d681SAndroid Build Coastguard Worker SDValue Mask = MSC->getMask();
5333*9880d681SAndroid Build Coastguard Worker SDValue Data = MSC->getValue();
5334*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
5335*9880d681SAndroid Build Coastguard Worker
5336*9880d681SAndroid Build Coastguard Worker // If the MSCATTER data type requires splitting and the mask is provided by a
5337*9880d681SAndroid Build Coastguard Worker // SETCC, then split both nodes and its operands before legalization. This
5338*9880d681SAndroid Build Coastguard Worker // prevents the type legalizer from unrolling SETCC into scalar comparisons
5339*9880d681SAndroid Build Coastguard Worker // and enables future optimizations (e.g. min/max pattern matching on X86).
5340*9880d681SAndroid Build Coastguard Worker if (Mask.getOpcode() != ISD::SETCC)
5341*9880d681SAndroid Build Coastguard Worker return SDValue();
5342*9880d681SAndroid Build Coastguard Worker
5343*9880d681SAndroid Build Coastguard Worker // Check if any splitting is required.
5344*9880d681SAndroid Build Coastguard Worker if (TLI.getTypeAction(*DAG.getContext(), Data.getValueType()) !=
5345*9880d681SAndroid Build Coastguard Worker TargetLowering::TypeSplitVector)
5346*9880d681SAndroid Build Coastguard Worker return SDValue();
5347*9880d681SAndroid Build Coastguard Worker SDValue MaskLo, MaskHi, Lo, Hi;
5348*9880d681SAndroid Build Coastguard Worker std::tie(MaskLo, MaskHi) = SplitVSETCC(Mask.getNode(), DAG);
5349*9880d681SAndroid Build Coastguard Worker
5350*9880d681SAndroid Build Coastguard Worker EVT LoVT, HiVT;
5351*9880d681SAndroid Build Coastguard Worker std::tie(LoVT, HiVT) = DAG.GetSplitDestVTs(MSC->getValueType(0));
5352*9880d681SAndroid Build Coastguard Worker
5353*9880d681SAndroid Build Coastguard Worker SDValue Chain = MSC->getChain();
5354*9880d681SAndroid Build Coastguard Worker
5355*9880d681SAndroid Build Coastguard Worker EVT MemoryVT = MSC->getMemoryVT();
5356*9880d681SAndroid Build Coastguard Worker unsigned Alignment = MSC->getOriginalAlignment();
5357*9880d681SAndroid Build Coastguard Worker
5358*9880d681SAndroid Build Coastguard Worker EVT LoMemVT, HiMemVT;
5359*9880d681SAndroid Build Coastguard Worker std::tie(LoMemVT, HiMemVT) = DAG.GetSplitDestVTs(MemoryVT);
5360*9880d681SAndroid Build Coastguard Worker
5361*9880d681SAndroid Build Coastguard Worker SDValue DataLo, DataHi;
5362*9880d681SAndroid Build Coastguard Worker std::tie(DataLo, DataHi) = DAG.SplitVector(Data, DL);
5363*9880d681SAndroid Build Coastguard Worker
5364*9880d681SAndroid Build Coastguard Worker SDValue BasePtr = MSC->getBasePtr();
5365*9880d681SAndroid Build Coastguard Worker SDValue IndexLo, IndexHi;
5366*9880d681SAndroid Build Coastguard Worker std::tie(IndexLo, IndexHi) = DAG.SplitVector(MSC->getIndex(), DL);
5367*9880d681SAndroid Build Coastguard Worker
5368*9880d681SAndroid Build Coastguard Worker MachineMemOperand *MMO = DAG.getMachineFunction().
5369*9880d681SAndroid Build Coastguard Worker getMachineMemOperand(MSC->getPointerInfo(),
5370*9880d681SAndroid Build Coastguard Worker MachineMemOperand::MOStore, LoMemVT.getStoreSize(),
5371*9880d681SAndroid Build Coastguard Worker Alignment, MSC->getAAInfo(), MSC->getRanges());
5372*9880d681SAndroid Build Coastguard Worker
5373*9880d681SAndroid Build Coastguard Worker SDValue OpsLo[] = { Chain, DataLo, MaskLo, BasePtr, IndexLo };
5374*9880d681SAndroid Build Coastguard Worker Lo = DAG.getMaskedScatter(DAG.getVTList(MVT::Other), DataLo.getValueType(),
5375*9880d681SAndroid Build Coastguard Worker DL, OpsLo, MMO);
5376*9880d681SAndroid Build Coastguard Worker
5377*9880d681SAndroid Build Coastguard Worker SDValue OpsHi[] = {Chain, DataHi, MaskHi, BasePtr, IndexHi};
5378*9880d681SAndroid Build Coastguard Worker Hi = DAG.getMaskedScatter(DAG.getVTList(MVT::Other), DataHi.getValueType(),
5379*9880d681SAndroid Build Coastguard Worker DL, OpsHi, MMO);
5380*9880d681SAndroid Build Coastguard Worker
5381*9880d681SAndroid Build Coastguard Worker AddToWorklist(Lo.getNode());
5382*9880d681SAndroid Build Coastguard Worker AddToWorklist(Hi.getNode());
5383*9880d681SAndroid Build Coastguard Worker
5384*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TokenFactor, DL, MVT::Other, Lo, Hi);
5385*9880d681SAndroid Build Coastguard Worker }
5386*9880d681SAndroid Build Coastguard Worker
visitMSTORE(SDNode * N)5387*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitMSTORE(SDNode *N) {
5388*9880d681SAndroid Build Coastguard Worker
5389*9880d681SAndroid Build Coastguard Worker if (Level >= AfterLegalizeTypes)
5390*9880d681SAndroid Build Coastguard Worker return SDValue();
5391*9880d681SAndroid Build Coastguard Worker
5392*9880d681SAndroid Build Coastguard Worker MaskedStoreSDNode *MST = dyn_cast<MaskedStoreSDNode>(N);
5393*9880d681SAndroid Build Coastguard Worker SDValue Mask = MST->getMask();
5394*9880d681SAndroid Build Coastguard Worker SDValue Data = MST->getValue();
5395*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
5396*9880d681SAndroid Build Coastguard Worker
5397*9880d681SAndroid Build Coastguard Worker // If the MSTORE data type requires splitting and the mask is provided by a
5398*9880d681SAndroid Build Coastguard Worker // SETCC, then split both nodes and its operands before legalization. This
5399*9880d681SAndroid Build Coastguard Worker // prevents the type legalizer from unrolling SETCC into scalar comparisons
5400*9880d681SAndroid Build Coastguard Worker // and enables future optimizations (e.g. min/max pattern matching on X86).
5401*9880d681SAndroid Build Coastguard Worker if (Mask.getOpcode() == ISD::SETCC) {
5402*9880d681SAndroid Build Coastguard Worker
5403*9880d681SAndroid Build Coastguard Worker // Check if any splitting is required.
5404*9880d681SAndroid Build Coastguard Worker if (TLI.getTypeAction(*DAG.getContext(), Data.getValueType()) !=
5405*9880d681SAndroid Build Coastguard Worker TargetLowering::TypeSplitVector)
5406*9880d681SAndroid Build Coastguard Worker return SDValue();
5407*9880d681SAndroid Build Coastguard Worker
5408*9880d681SAndroid Build Coastguard Worker SDValue MaskLo, MaskHi, Lo, Hi;
5409*9880d681SAndroid Build Coastguard Worker std::tie(MaskLo, MaskHi) = SplitVSETCC(Mask.getNode(), DAG);
5410*9880d681SAndroid Build Coastguard Worker
5411*9880d681SAndroid Build Coastguard Worker EVT LoVT, HiVT;
5412*9880d681SAndroid Build Coastguard Worker std::tie(LoVT, HiVT) = DAG.GetSplitDestVTs(MST->getValueType(0));
5413*9880d681SAndroid Build Coastguard Worker
5414*9880d681SAndroid Build Coastguard Worker SDValue Chain = MST->getChain();
5415*9880d681SAndroid Build Coastguard Worker SDValue Ptr = MST->getBasePtr();
5416*9880d681SAndroid Build Coastguard Worker
5417*9880d681SAndroid Build Coastguard Worker EVT MemoryVT = MST->getMemoryVT();
5418*9880d681SAndroid Build Coastguard Worker unsigned Alignment = MST->getOriginalAlignment();
5419*9880d681SAndroid Build Coastguard Worker
5420*9880d681SAndroid Build Coastguard Worker // if Alignment is equal to the vector size,
5421*9880d681SAndroid Build Coastguard Worker // take the half of it for the second part
5422*9880d681SAndroid Build Coastguard Worker unsigned SecondHalfAlignment =
5423*9880d681SAndroid Build Coastguard Worker (Alignment == Data->getValueType(0).getSizeInBits()/8) ?
5424*9880d681SAndroid Build Coastguard Worker Alignment/2 : Alignment;
5425*9880d681SAndroid Build Coastguard Worker
5426*9880d681SAndroid Build Coastguard Worker EVT LoMemVT, HiMemVT;
5427*9880d681SAndroid Build Coastguard Worker std::tie(LoMemVT, HiMemVT) = DAG.GetSplitDestVTs(MemoryVT);
5428*9880d681SAndroid Build Coastguard Worker
5429*9880d681SAndroid Build Coastguard Worker SDValue DataLo, DataHi;
5430*9880d681SAndroid Build Coastguard Worker std::tie(DataLo, DataHi) = DAG.SplitVector(Data, DL);
5431*9880d681SAndroid Build Coastguard Worker
5432*9880d681SAndroid Build Coastguard Worker MachineMemOperand *MMO = DAG.getMachineFunction().
5433*9880d681SAndroid Build Coastguard Worker getMachineMemOperand(MST->getPointerInfo(),
5434*9880d681SAndroid Build Coastguard Worker MachineMemOperand::MOStore, LoMemVT.getStoreSize(),
5435*9880d681SAndroid Build Coastguard Worker Alignment, MST->getAAInfo(), MST->getRanges());
5436*9880d681SAndroid Build Coastguard Worker
5437*9880d681SAndroid Build Coastguard Worker Lo = DAG.getMaskedStore(Chain, DL, DataLo, Ptr, MaskLo, LoMemVT, MMO,
5438*9880d681SAndroid Build Coastguard Worker MST->isTruncatingStore());
5439*9880d681SAndroid Build Coastguard Worker
5440*9880d681SAndroid Build Coastguard Worker unsigned IncrementSize = LoMemVT.getSizeInBits()/8;
5441*9880d681SAndroid Build Coastguard Worker Ptr = DAG.getNode(ISD::ADD, DL, Ptr.getValueType(), Ptr,
5442*9880d681SAndroid Build Coastguard Worker DAG.getConstant(IncrementSize, DL, Ptr.getValueType()));
5443*9880d681SAndroid Build Coastguard Worker
5444*9880d681SAndroid Build Coastguard Worker MMO = DAG.getMachineFunction().
5445*9880d681SAndroid Build Coastguard Worker getMachineMemOperand(MST->getPointerInfo(),
5446*9880d681SAndroid Build Coastguard Worker MachineMemOperand::MOStore, HiMemVT.getStoreSize(),
5447*9880d681SAndroid Build Coastguard Worker SecondHalfAlignment, MST->getAAInfo(),
5448*9880d681SAndroid Build Coastguard Worker MST->getRanges());
5449*9880d681SAndroid Build Coastguard Worker
5450*9880d681SAndroid Build Coastguard Worker Hi = DAG.getMaskedStore(Chain, DL, DataHi, Ptr, MaskHi, HiMemVT, MMO,
5451*9880d681SAndroid Build Coastguard Worker MST->isTruncatingStore());
5452*9880d681SAndroid Build Coastguard Worker
5453*9880d681SAndroid Build Coastguard Worker AddToWorklist(Lo.getNode());
5454*9880d681SAndroid Build Coastguard Worker AddToWorklist(Hi.getNode());
5455*9880d681SAndroid Build Coastguard Worker
5456*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TokenFactor, DL, MVT::Other, Lo, Hi);
5457*9880d681SAndroid Build Coastguard Worker }
5458*9880d681SAndroid Build Coastguard Worker return SDValue();
5459*9880d681SAndroid Build Coastguard Worker }
5460*9880d681SAndroid Build Coastguard Worker
visitMGATHER(SDNode * N)5461*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitMGATHER(SDNode *N) {
5462*9880d681SAndroid Build Coastguard Worker
5463*9880d681SAndroid Build Coastguard Worker if (Level >= AfterLegalizeTypes)
5464*9880d681SAndroid Build Coastguard Worker return SDValue();
5465*9880d681SAndroid Build Coastguard Worker
5466*9880d681SAndroid Build Coastguard Worker MaskedGatherSDNode *MGT = dyn_cast<MaskedGatherSDNode>(N);
5467*9880d681SAndroid Build Coastguard Worker SDValue Mask = MGT->getMask();
5468*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
5469*9880d681SAndroid Build Coastguard Worker
5470*9880d681SAndroid Build Coastguard Worker // If the MGATHER result requires splitting and the mask is provided by a
5471*9880d681SAndroid Build Coastguard Worker // SETCC, then split both nodes and its operands before legalization. This
5472*9880d681SAndroid Build Coastguard Worker // prevents the type legalizer from unrolling SETCC into scalar comparisons
5473*9880d681SAndroid Build Coastguard Worker // and enables future optimizations (e.g. min/max pattern matching on X86).
5474*9880d681SAndroid Build Coastguard Worker
5475*9880d681SAndroid Build Coastguard Worker if (Mask.getOpcode() != ISD::SETCC)
5476*9880d681SAndroid Build Coastguard Worker return SDValue();
5477*9880d681SAndroid Build Coastguard Worker
5478*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
5479*9880d681SAndroid Build Coastguard Worker
5480*9880d681SAndroid Build Coastguard Worker // Check if any splitting is required.
5481*9880d681SAndroid Build Coastguard Worker if (TLI.getTypeAction(*DAG.getContext(), VT) !=
5482*9880d681SAndroid Build Coastguard Worker TargetLowering::TypeSplitVector)
5483*9880d681SAndroid Build Coastguard Worker return SDValue();
5484*9880d681SAndroid Build Coastguard Worker
5485*9880d681SAndroid Build Coastguard Worker SDValue MaskLo, MaskHi, Lo, Hi;
5486*9880d681SAndroid Build Coastguard Worker std::tie(MaskLo, MaskHi) = SplitVSETCC(Mask.getNode(), DAG);
5487*9880d681SAndroid Build Coastguard Worker
5488*9880d681SAndroid Build Coastguard Worker SDValue Src0 = MGT->getValue();
5489*9880d681SAndroid Build Coastguard Worker SDValue Src0Lo, Src0Hi;
5490*9880d681SAndroid Build Coastguard Worker std::tie(Src0Lo, Src0Hi) = DAG.SplitVector(Src0, DL);
5491*9880d681SAndroid Build Coastguard Worker
5492*9880d681SAndroid Build Coastguard Worker EVT LoVT, HiVT;
5493*9880d681SAndroid Build Coastguard Worker std::tie(LoVT, HiVT) = DAG.GetSplitDestVTs(VT);
5494*9880d681SAndroid Build Coastguard Worker
5495*9880d681SAndroid Build Coastguard Worker SDValue Chain = MGT->getChain();
5496*9880d681SAndroid Build Coastguard Worker EVT MemoryVT = MGT->getMemoryVT();
5497*9880d681SAndroid Build Coastguard Worker unsigned Alignment = MGT->getOriginalAlignment();
5498*9880d681SAndroid Build Coastguard Worker
5499*9880d681SAndroid Build Coastguard Worker EVT LoMemVT, HiMemVT;
5500*9880d681SAndroid Build Coastguard Worker std::tie(LoMemVT, HiMemVT) = DAG.GetSplitDestVTs(MemoryVT);
5501*9880d681SAndroid Build Coastguard Worker
5502*9880d681SAndroid Build Coastguard Worker SDValue BasePtr = MGT->getBasePtr();
5503*9880d681SAndroid Build Coastguard Worker SDValue Index = MGT->getIndex();
5504*9880d681SAndroid Build Coastguard Worker SDValue IndexLo, IndexHi;
5505*9880d681SAndroid Build Coastguard Worker std::tie(IndexLo, IndexHi) = DAG.SplitVector(Index, DL);
5506*9880d681SAndroid Build Coastguard Worker
5507*9880d681SAndroid Build Coastguard Worker MachineMemOperand *MMO = DAG.getMachineFunction().
5508*9880d681SAndroid Build Coastguard Worker getMachineMemOperand(MGT->getPointerInfo(),
5509*9880d681SAndroid Build Coastguard Worker MachineMemOperand::MOLoad, LoMemVT.getStoreSize(),
5510*9880d681SAndroid Build Coastguard Worker Alignment, MGT->getAAInfo(), MGT->getRanges());
5511*9880d681SAndroid Build Coastguard Worker
5512*9880d681SAndroid Build Coastguard Worker SDValue OpsLo[] = { Chain, Src0Lo, MaskLo, BasePtr, IndexLo };
5513*9880d681SAndroid Build Coastguard Worker Lo = DAG.getMaskedGather(DAG.getVTList(LoVT, MVT::Other), LoVT, DL, OpsLo,
5514*9880d681SAndroid Build Coastguard Worker MMO);
5515*9880d681SAndroid Build Coastguard Worker
5516*9880d681SAndroid Build Coastguard Worker SDValue OpsHi[] = {Chain, Src0Hi, MaskHi, BasePtr, IndexHi};
5517*9880d681SAndroid Build Coastguard Worker Hi = DAG.getMaskedGather(DAG.getVTList(HiVT, MVT::Other), HiVT, DL, OpsHi,
5518*9880d681SAndroid Build Coastguard Worker MMO);
5519*9880d681SAndroid Build Coastguard Worker
5520*9880d681SAndroid Build Coastguard Worker AddToWorklist(Lo.getNode());
5521*9880d681SAndroid Build Coastguard Worker AddToWorklist(Hi.getNode());
5522*9880d681SAndroid Build Coastguard Worker
5523*9880d681SAndroid Build Coastguard Worker // Build a factor node to remember that this load is independent of the
5524*9880d681SAndroid Build Coastguard Worker // other one.
5525*9880d681SAndroid Build Coastguard Worker Chain = DAG.getNode(ISD::TokenFactor, DL, MVT::Other, Lo.getValue(1),
5526*9880d681SAndroid Build Coastguard Worker Hi.getValue(1));
5527*9880d681SAndroid Build Coastguard Worker
5528*9880d681SAndroid Build Coastguard Worker // Legalized the chain result - switch anything that used the old chain to
5529*9880d681SAndroid Build Coastguard Worker // use the new one.
5530*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(MGT, 1), Chain);
5531*9880d681SAndroid Build Coastguard Worker
5532*9880d681SAndroid Build Coastguard Worker SDValue GatherRes = DAG.getNode(ISD::CONCAT_VECTORS, DL, VT, Lo, Hi);
5533*9880d681SAndroid Build Coastguard Worker
5534*9880d681SAndroid Build Coastguard Worker SDValue RetOps[] = { GatherRes, Chain };
5535*9880d681SAndroid Build Coastguard Worker return DAG.getMergeValues(RetOps, DL);
5536*9880d681SAndroid Build Coastguard Worker }
5537*9880d681SAndroid Build Coastguard Worker
visitMLOAD(SDNode * N)5538*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitMLOAD(SDNode *N) {
5539*9880d681SAndroid Build Coastguard Worker
5540*9880d681SAndroid Build Coastguard Worker if (Level >= AfterLegalizeTypes)
5541*9880d681SAndroid Build Coastguard Worker return SDValue();
5542*9880d681SAndroid Build Coastguard Worker
5543*9880d681SAndroid Build Coastguard Worker MaskedLoadSDNode *MLD = dyn_cast<MaskedLoadSDNode>(N);
5544*9880d681SAndroid Build Coastguard Worker SDValue Mask = MLD->getMask();
5545*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
5546*9880d681SAndroid Build Coastguard Worker
5547*9880d681SAndroid Build Coastguard Worker // If the MLOAD result requires splitting and the mask is provided by a
5548*9880d681SAndroid Build Coastguard Worker // SETCC, then split both nodes and its operands before legalization. This
5549*9880d681SAndroid Build Coastguard Worker // prevents the type legalizer from unrolling SETCC into scalar comparisons
5550*9880d681SAndroid Build Coastguard Worker // and enables future optimizations (e.g. min/max pattern matching on X86).
5551*9880d681SAndroid Build Coastguard Worker
5552*9880d681SAndroid Build Coastguard Worker if (Mask.getOpcode() == ISD::SETCC) {
5553*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
5554*9880d681SAndroid Build Coastguard Worker
5555*9880d681SAndroid Build Coastguard Worker // Check if any splitting is required.
5556*9880d681SAndroid Build Coastguard Worker if (TLI.getTypeAction(*DAG.getContext(), VT) !=
5557*9880d681SAndroid Build Coastguard Worker TargetLowering::TypeSplitVector)
5558*9880d681SAndroid Build Coastguard Worker return SDValue();
5559*9880d681SAndroid Build Coastguard Worker
5560*9880d681SAndroid Build Coastguard Worker SDValue MaskLo, MaskHi, Lo, Hi;
5561*9880d681SAndroid Build Coastguard Worker std::tie(MaskLo, MaskHi) = SplitVSETCC(Mask.getNode(), DAG);
5562*9880d681SAndroid Build Coastguard Worker
5563*9880d681SAndroid Build Coastguard Worker SDValue Src0 = MLD->getSrc0();
5564*9880d681SAndroid Build Coastguard Worker SDValue Src0Lo, Src0Hi;
5565*9880d681SAndroid Build Coastguard Worker std::tie(Src0Lo, Src0Hi) = DAG.SplitVector(Src0, DL);
5566*9880d681SAndroid Build Coastguard Worker
5567*9880d681SAndroid Build Coastguard Worker EVT LoVT, HiVT;
5568*9880d681SAndroid Build Coastguard Worker std::tie(LoVT, HiVT) = DAG.GetSplitDestVTs(MLD->getValueType(0));
5569*9880d681SAndroid Build Coastguard Worker
5570*9880d681SAndroid Build Coastguard Worker SDValue Chain = MLD->getChain();
5571*9880d681SAndroid Build Coastguard Worker SDValue Ptr = MLD->getBasePtr();
5572*9880d681SAndroid Build Coastguard Worker EVT MemoryVT = MLD->getMemoryVT();
5573*9880d681SAndroid Build Coastguard Worker unsigned Alignment = MLD->getOriginalAlignment();
5574*9880d681SAndroid Build Coastguard Worker
5575*9880d681SAndroid Build Coastguard Worker // if Alignment is equal to the vector size,
5576*9880d681SAndroid Build Coastguard Worker // take the half of it for the second part
5577*9880d681SAndroid Build Coastguard Worker unsigned SecondHalfAlignment =
5578*9880d681SAndroid Build Coastguard Worker (Alignment == MLD->getValueType(0).getSizeInBits()/8) ?
5579*9880d681SAndroid Build Coastguard Worker Alignment/2 : Alignment;
5580*9880d681SAndroid Build Coastguard Worker
5581*9880d681SAndroid Build Coastguard Worker EVT LoMemVT, HiMemVT;
5582*9880d681SAndroid Build Coastguard Worker std::tie(LoMemVT, HiMemVT) = DAG.GetSplitDestVTs(MemoryVT);
5583*9880d681SAndroid Build Coastguard Worker
5584*9880d681SAndroid Build Coastguard Worker MachineMemOperand *MMO = DAG.getMachineFunction().
5585*9880d681SAndroid Build Coastguard Worker getMachineMemOperand(MLD->getPointerInfo(),
5586*9880d681SAndroid Build Coastguard Worker MachineMemOperand::MOLoad, LoMemVT.getStoreSize(),
5587*9880d681SAndroid Build Coastguard Worker Alignment, MLD->getAAInfo(), MLD->getRanges());
5588*9880d681SAndroid Build Coastguard Worker
5589*9880d681SAndroid Build Coastguard Worker Lo = DAG.getMaskedLoad(LoVT, DL, Chain, Ptr, MaskLo, Src0Lo, LoMemVT, MMO,
5590*9880d681SAndroid Build Coastguard Worker ISD::NON_EXTLOAD);
5591*9880d681SAndroid Build Coastguard Worker
5592*9880d681SAndroid Build Coastguard Worker unsigned IncrementSize = LoMemVT.getSizeInBits()/8;
5593*9880d681SAndroid Build Coastguard Worker Ptr = DAG.getNode(ISD::ADD, DL, Ptr.getValueType(), Ptr,
5594*9880d681SAndroid Build Coastguard Worker DAG.getConstant(IncrementSize, DL, Ptr.getValueType()));
5595*9880d681SAndroid Build Coastguard Worker
5596*9880d681SAndroid Build Coastguard Worker MMO = DAG.getMachineFunction().
5597*9880d681SAndroid Build Coastguard Worker getMachineMemOperand(MLD->getPointerInfo(),
5598*9880d681SAndroid Build Coastguard Worker MachineMemOperand::MOLoad, HiMemVT.getStoreSize(),
5599*9880d681SAndroid Build Coastguard Worker SecondHalfAlignment, MLD->getAAInfo(), MLD->getRanges());
5600*9880d681SAndroid Build Coastguard Worker
5601*9880d681SAndroid Build Coastguard Worker Hi = DAG.getMaskedLoad(HiVT, DL, Chain, Ptr, MaskHi, Src0Hi, HiMemVT, MMO,
5602*9880d681SAndroid Build Coastguard Worker ISD::NON_EXTLOAD);
5603*9880d681SAndroid Build Coastguard Worker
5604*9880d681SAndroid Build Coastguard Worker AddToWorklist(Lo.getNode());
5605*9880d681SAndroid Build Coastguard Worker AddToWorklist(Hi.getNode());
5606*9880d681SAndroid Build Coastguard Worker
5607*9880d681SAndroid Build Coastguard Worker // Build a factor node to remember that this load is independent of the
5608*9880d681SAndroid Build Coastguard Worker // other one.
5609*9880d681SAndroid Build Coastguard Worker Chain = DAG.getNode(ISD::TokenFactor, DL, MVT::Other, Lo.getValue(1),
5610*9880d681SAndroid Build Coastguard Worker Hi.getValue(1));
5611*9880d681SAndroid Build Coastguard Worker
5612*9880d681SAndroid Build Coastguard Worker // Legalized the chain result - switch anything that used the old chain to
5613*9880d681SAndroid Build Coastguard Worker // use the new one.
5614*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(MLD, 1), Chain);
5615*9880d681SAndroid Build Coastguard Worker
5616*9880d681SAndroid Build Coastguard Worker SDValue LoadRes = DAG.getNode(ISD::CONCAT_VECTORS, DL, VT, Lo, Hi);
5617*9880d681SAndroid Build Coastguard Worker
5618*9880d681SAndroid Build Coastguard Worker SDValue RetOps[] = { LoadRes, Chain };
5619*9880d681SAndroid Build Coastguard Worker return DAG.getMergeValues(RetOps, DL);
5620*9880d681SAndroid Build Coastguard Worker }
5621*9880d681SAndroid Build Coastguard Worker return SDValue();
5622*9880d681SAndroid Build Coastguard Worker }
5623*9880d681SAndroid Build Coastguard Worker
visitVSELECT(SDNode * N)5624*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitVSELECT(SDNode *N) {
5625*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
5626*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
5627*9880d681SAndroid Build Coastguard Worker SDValue N2 = N->getOperand(2);
5628*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
5629*9880d681SAndroid Build Coastguard Worker
5630*9880d681SAndroid Build Coastguard Worker // Canonicalize integer abs.
5631*9880d681SAndroid Build Coastguard Worker // vselect (setg[te] X, 0), X, -X ->
5632*9880d681SAndroid Build Coastguard Worker // vselect (setgt X, -1), X, -X ->
5633*9880d681SAndroid Build Coastguard Worker // vselect (setl[te] X, 0), -X, X ->
5634*9880d681SAndroid Build Coastguard Worker // Y = sra (X, size(X)-1); xor (add (X, Y), Y)
5635*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SETCC) {
5636*9880d681SAndroid Build Coastguard Worker SDValue LHS = N0.getOperand(0), RHS = N0.getOperand(1);
5637*9880d681SAndroid Build Coastguard Worker ISD::CondCode CC = cast<CondCodeSDNode>(N0.getOperand(2))->get();
5638*9880d681SAndroid Build Coastguard Worker bool isAbs = false;
5639*9880d681SAndroid Build Coastguard Worker bool RHSIsAllZeros = ISD::isBuildVectorAllZeros(RHS.getNode());
5640*9880d681SAndroid Build Coastguard Worker
5641*9880d681SAndroid Build Coastguard Worker if (((RHSIsAllZeros && (CC == ISD::SETGT || CC == ISD::SETGE)) ||
5642*9880d681SAndroid Build Coastguard Worker (ISD::isBuildVectorAllOnes(RHS.getNode()) && CC == ISD::SETGT)) &&
5643*9880d681SAndroid Build Coastguard Worker N1 == LHS && N2.getOpcode() == ISD::SUB && N1 == N2.getOperand(1))
5644*9880d681SAndroid Build Coastguard Worker isAbs = ISD::isBuildVectorAllZeros(N2.getOperand(0).getNode());
5645*9880d681SAndroid Build Coastguard Worker else if ((RHSIsAllZeros && (CC == ISD::SETLT || CC == ISD::SETLE)) &&
5646*9880d681SAndroid Build Coastguard Worker N2 == LHS && N1.getOpcode() == ISD::SUB && N2 == N1.getOperand(1))
5647*9880d681SAndroid Build Coastguard Worker isAbs = ISD::isBuildVectorAllZeros(N1.getOperand(0).getNode());
5648*9880d681SAndroid Build Coastguard Worker
5649*9880d681SAndroid Build Coastguard Worker if (isAbs) {
5650*9880d681SAndroid Build Coastguard Worker EVT VT = LHS.getValueType();
5651*9880d681SAndroid Build Coastguard Worker SDValue Shift = DAG.getNode(
5652*9880d681SAndroid Build Coastguard Worker ISD::SRA, DL, VT, LHS,
5653*9880d681SAndroid Build Coastguard Worker DAG.getConstant(VT.getScalarType().getSizeInBits() - 1, DL, VT));
5654*9880d681SAndroid Build Coastguard Worker SDValue Add = DAG.getNode(ISD::ADD, DL, VT, LHS, Shift);
5655*9880d681SAndroid Build Coastguard Worker AddToWorklist(Shift.getNode());
5656*9880d681SAndroid Build Coastguard Worker AddToWorklist(Add.getNode());
5657*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::XOR, DL, VT, Add, Shift);
5658*9880d681SAndroid Build Coastguard Worker }
5659*9880d681SAndroid Build Coastguard Worker }
5660*9880d681SAndroid Build Coastguard Worker
5661*9880d681SAndroid Build Coastguard Worker if (SimplifySelectOps(N, N1, N2))
5662*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Don't revisit N.
5663*9880d681SAndroid Build Coastguard Worker
5664*9880d681SAndroid Build Coastguard Worker // If the VSELECT result requires splitting and the mask is provided by a
5665*9880d681SAndroid Build Coastguard Worker // SETCC, then split both nodes and its operands before legalization. This
5666*9880d681SAndroid Build Coastguard Worker // prevents the type legalizer from unrolling SETCC into scalar comparisons
5667*9880d681SAndroid Build Coastguard Worker // and enables future optimizations (e.g. min/max pattern matching on X86).
5668*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SETCC) {
5669*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
5670*9880d681SAndroid Build Coastguard Worker
5671*9880d681SAndroid Build Coastguard Worker // Check if any splitting is required.
5672*9880d681SAndroid Build Coastguard Worker if (TLI.getTypeAction(*DAG.getContext(), VT) !=
5673*9880d681SAndroid Build Coastguard Worker TargetLowering::TypeSplitVector)
5674*9880d681SAndroid Build Coastguard Worker return SDValue();
5675*9880d681SAndroid Build Coastguard Worker
5676*9880d681SAndroid Build Coastguard Worker SDValue Lo, Hi, CCLo, CCHi, LL, LH, RL, RH;
5677*9880d681SAndroid Build Coastguard Worker std::tie(CCLo, CCHi) = SplitVSETCC(N0.getNode(), DAG);
5678*9880d681SAndroid Build Coastguard Worker std::tie(LL, LH) = DAG.SplitVectorOperand(N, 1);
5679*9880d681SAndroid Build Coastguard Worker std::tie(RL, RH) = DAG.SplitVectorOperand(N, 2);
5680*9880d681SAndroid Build Coastguard Worker
5681*9880d681SAndroid Build Coastguard Worker Lo = DAG.getNode(N->getOpcode(), DL, LL.getValueType(), CCLo, LL, RL);
5682*9880d681SAndroid Build Coastguard Worker Hi = DAG.getNode(N->getOpcode(), DL, LH.getValueType(), CCHi, LH, RH);
5683*9880d681SAndroid Build Coastguard Worker
5684*9880d681SAndroid Build Coastguard Worker // Add the new VSELECT nodes to the work list in case they need to be split
5685*9880d681SAndroid Build Coastguard Worker // again.
5686*9880d681SAndroid Build Coastguard Worker AddToWorklist(Lo.getNode());
5687*9880d681SAndroid Build Coastguard Worker AddToWorklist(Hi.getNode());
5688*9880d681SAndroid Build Coastguard Worker
5689*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::CONCAT_VECTORS, DL, VT, Lo, Hi);
5690*9880d681SAndroid Build Coastguard Worker }
5691*9880d681SAndroid Build Coastguard Worker
5692*9880d681SAndroid Build Coastguard Worker // Fold (vselect (build_vector all_ones), N1, N2) -> N1
5693*9880d681SAndroid Build Coastguard Worker if (ISD::isBuildVectorAllOnes(N0.getNode()))
5694*9880d681SAndroid Build Coastguard Worker return N1;
5695*9880d681SAndroid Build Coastguard Worker // Fold (vselect (build_vector all_zeros), N1, N2) -> N2
5696*9880d681SAndroid Build Coastguard Worker if (ISD::isBuildVectorAllZeros(N0.getNode()))
5697*9880d681SAndroid Build Coastguard Worker return N2;
5698*9880d681SAndroid Build Coastguard Worker
5699*9880d681SAndroid Build Coastguard Worker // The ConvertSelectToConcatVector function is assuming both the above
5700*9880d681SAndroid Build Coastguard Worker // checks for (vselect (build_vector all{ones,zeros) ...) have been made
5701*9880d681SAndroid Build Coastguard Worker // and addressed.
5702*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::CONCAT_VECTORS &&
5703*9880d681SAndroid Build Coastguard Worker N2.getOpcode() == ISD::CONCAT_VECTORS &&
5704*9880d681SAndroid Build Coastguard Worker ISD::isBuildVectorOfConstantSDNodes(N0.getNode())) {
5705*9880d681SAndroid Build Coastguard Worker if (SDValue CV = ConvertSelectToConcatVector(N, DAG))
5706*9880d681SAndroid Build Coastguard Worker return CV;
5707*9880d681SAndroid Build Coastguard Worker }
5708*9880d681SAndroid Build Coastguard Worker
5709*9880d681SAndroid Build Coastguard Worker return SDValue();
5710*9880d681SAndroid Build Coastguard Worker }
5711*9880d681SAndroid Build Coastguard Worker
visitSELECT_CC(SDNode * N)5712*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSELECT_CC(SDNode *N) {
5713*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
5714*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
5715*9880d681SAndroid Build Coastguard Worker SDValue N2 = N->getOperand(2);
5716*9880d681SAndroid Build Coastguard Worker SDValue N3 = N->getOperand(3);
5717*9880d681SAndroid Build Coastguard Worker SDValue N4 = N->getOperand(4);
5718*9880d681SAndroid Build Coastguard Worker ISD::CondCode CC = cast<CondCodeSDNode>(N4)->get();
5719*9880d681SAndroid Build Coastguard Worker
5720*9880d681SAndroid Build Coastguard Worker // fold select_cc lhs, rhs, x, x, cc -> x
5721*9880d681SAndroid Build Coastguard Worker if (N2 == N3)
5722*9880d681SAndroid Build Coastguard Worker return N2;
5723*9880d681SAndroid Build Coastguard Worker
5724*9880d681SAndroid Build Coastguard Worker // Determine if the condition we're dealing with is constant
5725*9880d681SAndroid Build Coastguard Worker if (SDValue SCC = SimplifySetCC(getSetCCResultType(N0.getValueType()), N0, N1,
5726*9880d681SAndroid Build Coastguard Worker CC, SDLoc(N), false)) {
5727*9880d681SAndroid Build Coastguard Worker AddToWorklist(SCC.getNode());
5728*9880d681SAndroid Build Coastguard Worker
5729*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *SCCC = dyn_cast<ConstantSDNode>(SCC.getNode())) {
5730*9880d681SAndroid Build Coastguard Worker if (!SCCC->isNullValue())
5731*9880d681SAndroid Build Coastguard Worker return N2; // cond always true -> true val
5732*9880d681SAndroid Build Coastguard Worker else
5733*9880d681SAndroid Build Coastguard Worker return N3; // cond always false -> false val
5734*9880d681SAndroid Build Coastguard Worker } else if (SCC->isUndef()) {
5735*9880d681SAndroid Build Coastguard Worker // When the condition is UNDEF, just return the first operand. This is
5736*9880d681SAndroid Build Coastguard Worker // coherent the DAG creation, no setcc node is created in this case
5737*9880d681SAndroid Build Coastguard Worker return N2;
5738*9880d681SAndroid Build Coastguard Worker } else if (SCC.getOpcode() == ISD::SETCC) {
5739*9880d681SAndroid Build Coastguard Worker // Fold to a simpler select_cc
5740*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SELECT_CC, SDLoc(N), N2.getValueType(),
5741*9880d681SAndroid Build Coastguard Worker SCC.getOperand(0), SCC.getOperand(1), N2, N3,
5742*9880d681SAndroid Build Coastguard Worker SCC.getOperand(2));
5743*9880d681SAndroid Build Coastguard Worker }
5744*9880d681SAndroid Build Coastguard Worker }
5745*9880d681SAndroid Build Coastguard Worker
5746*9880d681SAndroid Build Coastguard Worker // If we can fold this based on the true/false value, do so.
5747*9880d681SAndroid Build Coastguard Worker if (SimplifySelectOps(N, N2, N3))
5748*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Don't revisit N.
5749*9880d681SAndroid Build Coastguard Worker
5750*9880d681SAndroid Build Coastguard Worker // fold select_cc into other things, such as min/max/abs
5751*9880d681SAndroid Build Coastguard Worker return SimplifySelectCC(SDLoc(N), N0, N1, N2, N3, CC);
5752*9880d681SAndroid Build Coastguard Worker }
5753*9880d681SAndroid Build Coastguard Worker
visitSETCC(SDNode * N)5754*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSETCC(SDNode *N) {
5755*9880d681SAndroid Build Coastguard Worker return SimplifySetCC(N->getValueType(0), N->getOperand(0), N->getOperand(1),
5756*9880d681SAndroid Build Coastguard Worker cast<CondCodeSDNode>(N->getOperand(2))->get(),
5757*9880d681SAndroid Build Coastguard Worker SDLoc(N));
5758*9880d681SAndroid Build Coastguard Worker }
5759*9880d681SAndroid Build Coastguard Worker
visitSETCCE(SDNode * N)5760*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSETCCE(SDNode *N) {
5761*9880d681SAndroid Build Coastguard Worker SDValue LHS = N->getOperand(0);
5762*9880d681SAndroid Build Coastguard Worker SDValue RHS = N->getOperand(1);
5763*9880d681SAndroid Build Coastguard Worker SDValue Carry = N->getOperand(2);
5764*9880d681SAndroid Build Coastguard Worker SDValue Cond = N->getOperand(3);
5765*9880d681SAndroid Build Coastguard Worker
5766*9880d681SAndroid Build Coastguard Worker // If Carry is false, fold to a regular SETCC.
5767*9880d681SAndroid Build Coastguard Worker if (Carry.getOpcode() == ISD::CARRY_FALSE)
5768*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SETCC, SDLoc(N), N->getVTList(), LHS, RHS, Cond);
5769*9880d681SAndroid Build Coastguard Worker
5770*9880d681SAndroid Build Coastguard Worker return SDValue();
5771*9880d681SAndroid Build Coastguard Worker }
5772*9880d681SAndroid Build Coastguard Worker
5773*9880d681SAndroid Build Coastguard Worker /// Try to fold a sext/zext/aext dag node into a ConstantSDNode or
5774*9880d681SAndroid Build Coastguard Worker /// a build_vector of constants.
5775*9880d681SAndroid Build Coastguard Worker /// This function is called by the DAGCombiner when visiting sext/zext/aext
5776*9880d681SAndroid Build Coastguard Worker /// dag nodes (see for example method DAGCombiner::visitSIGN_EXTEND).
5777*9880d681SAndroid Build Coastguard Worker /// Vector extends are not folded if operations are legal; this is to
5778*9880d681SAndroid Build Coastguard Worker /// avoid introducing illegal build_vector dag nodes.
tryToFoldExtendOfConstant(SDNode * N,const TargetLowering & TLI,SelectionDAG & DAG,bool LegalTypes,bool LegalOperations)5779*9880d681SAndroid Build Coastguard Worker static SDNode *tryToFoldExtendOfConstant(SDNode *N, const TargetLowering &TLI,
5780*9880d681SAndroid Build Coastguard Worker SelectionDAG &DAG, bool LegalTypes,
5781*9880d681SAndroid Build Coastguard Worker bool LegalOperations) {
5782*9880d681SAndroid Build Coastguard Worker unsigned Opcode = N->getOpcode();
5783*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
5784*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
5785*9880d681SAndroid Build Coastguard Worker
5786*9880d681SAndroid Build Coastguard Worker assert((Opcode == ISD::SIGN_EXTEND || Opcode == ISD::ZERO_EXTEND ||
5787*9880d681SAndroid Build Coastguard Worker Opcode == ISD::ANY_EXTEND || Opcode == ISD::SIGN_EXTEND_VECTOR_INREG ||
5788*9880d681SAndroid Build Coastguard Worker Opcode == ISD::ZERO_EXTEND_VECTOR_INREG)
5789*9880d681SAndroid Build Coastguard Worker && "Expected EXTEND dag node in input!");
5790*9880d681SAndroid Build Coastguard Worker
5791*9880d681SAndroid Build Coastguard Worker // fold (sext c1) -> c1
5792*9880d681SAndroid Build Coastguard Worker // fold (zext c1) -> c1
5793*9880d681SAndroid Build Coastguard Worker // fold (aext c1) -> c1
5794*9880d681SAndroid Build Coastguard Worker if (isa<ConstantSDNode>(N0))
5795*9880d681SAndroid Build Coastguard Worker return DAG.getNode(Opcode, SDLoc(N), VT, N0).getNode();
5796*9880d681SAndroid Build Coastguard Worker
5797*9880d681SAndroid Build Coastguard Worker // fold (sext (build_vector AllConstants) -> (build_vector AllConstants)
5798*9880d681SAndroid Build Coastguard Worker // fold (zext (build_vector AllConstants) -> (build_vector AllConstants)
5799*9880d681SAndroid Build Coastguard Worker // fold (aext (build_vector AllConstants) -> (build_vector AllConstants)
5800*9880d681SAndroid Build Coastguard Worker EVT SVT = VT.getScalarType();
5801*9880d681SAndroid Build Coastguard Worker if (!(VT.isVector() &&
5802*9880d681SAndroid Build Coastguard Worker (!LegalTypes || (!LegalOperations && TLI.isTypeLegal(SVT))) &&
5803*9880d681SAndroid Build Coastguard Worker ISD::isBuildVectorOfConstantSDNodes(N0.getNode())))
5804*9880d681SAndroid Build Coastguard Worker return nullptr;
5805*9880d681SAndroid Build Coastguard Worker
5806*9880d681SAndroid Build Coastguard Worker // We can fold this node into a build_vector.
5807*9880d681SAndroid Build Coastguard Worker unsigned VTBits = SVT.getSizeInBits();
5808*9880d681SAndroid Build Coastguard Worker unsigned EVTBits = N0->getValueType(0).getScalarType().getSizeInBits();
5809*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Elts;
5810*9880d681SAndroid Build Coastguard Worker unsigned NumElts = VT.getVectorNumElements();
5811*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
5812*9880d681SAndroid Build Coastguard Worker
5813*9880d681SAndroid Build Coastguard Worker for (unsigned i=0; i != NumElts; ++i) {
5814*9880d681SAndroid Build Coastguard Worker SDValue Op = N0->getOperand(i);
5815*9880d681SAndroid Build Coastguard Worker if (Op->isUndef()) {
5816*9880d681SAndroid Build Coastguard Worker Elts.push_back(DAG.getUNDEF(SVT));
5817*9880d681SAndroid Build Coastguard Worker continue;
5818*9880d681SAndroid Build Coastguard Worker }
5819*9880d681SAndroid Build Coastguard Worker
5820*9880d681SAndroid Build Coastguard Worker SDLoc DL(Op);
5821*9880d681SAndroid Build Coastguard Worker // Get the constant value and if needed trunc it to the size of the type.
5822*9880d681SAndroid Build Coastguard Worker // Nodes like build_vector might have constants wider than the scalar type.
5823*9880d681SAndroid Build Coastguard Worker APInt C = cast<ConstantSDNode>(Op)->getAPIntValue().zextOrTrunc(EVTBits);
5824*9880d681SAndroid Build Coastguard Worker if (Opcode == ISD::SIGN_EXTEND || Opcode == ISD::SIGN_EXTEND_VECTOR_INREG)
5825*9880d681SAndroid Build Coastguard Worker Elts.push_back(DAG.getConstant(C.sext(VTBits), DL, SVT));
5826*9880d681SAndroid Build Coastguard Worker else
5827*9880d681SAndroid Build Coastguard Worker Elts.push_back(DAG.getConstant(C.zext(VTBits), DL, SVT));
5828*9880d681SAndroid Build Coastguard Worker }
5829*9880d681SAndroid Build Coastguard Worker
5830*9880d681SAndroid Build Coastguard Worker return DAG.getBuildVector(VT, DL, Elts).getNode();
5831*9880d681SAndroid Build Coastguard Worker }
5832*9880d681SAndroid Build Coastguard Worker
5833*9880d681SAndroid Build Coastguard Worker // ExtendUsesToFormExtLoad - Trying to extend uses of a load to enable this:
5834*9880d681SAndroid Build Coastguard Worker // "fold ({s|z|a}ext (load x)) -> ({s|z|a}ext (truncate ({s|z|a}extload x)))"
5835*9880d681SAndroid Build Coastguard Worker // transformation. Returns true if extension are possible and the above
5836*9880d681SAndroid Build Coastguard Worker // mentioned transformation is profitable.
ExtendUsesToFormExtLoad(SDNode * N,SDValue N0,unsigned ExtOpc,SmallVectorImpl<SDNode * > & ExtendNodes,const TargetLowering & TLI)5837*9880d681SAndroid Build Coastguard Worker static bool ExtendUsesToFormExtLoad(SDNode *N, SDValue N0,
5838*9880d681SAndroid Build Coastguard Worker unsigned ExtOpc,
5839*9880d681SAndroid Build Coastguard Worker SmallVectorImpl<SDNode *> &ExtendNodes,
5840*9880d681SAndroid Build Coastguard Worker const TargetLowering &TLI) {
5841*9880d681SAndroid Build Coastguard Worker bool HasCopyToRegUses = false;
5842*9880d681SAndroid Build Coastguard Worker bool isTruncFree = TLI.isTruncateFree(N->getValueType(0), N0.getValueType());
5843*9880d681SAndroid Build Coastguard Worker for (SDNode::use_iterator UI = N0.getNode()->use_begin(),
5844*9880d681SAndroid Build Coastguard Worker UE = N0.getNode()->use_end();
5845*9880d681SAndroid Build Coastguard Worker UI != UE; ++UI) {
5846*9880d681SAndroid Build Coastguard Worker SDNode *User = *UI;
5847*9880d681SAndroid Build Coastguard Worker if (User == N)
5848*9880d681SAndroid Build Coastguard Worker continue;
5849*9880d681SAndroid Build Coastguard Worker if (UI.getUse().getResNo() != N0.getResNo())
5850*9880d681SAndroid Build Coastguard Worker continue;
5851*9880d681SAndroid Build Coastguard Worker // FIXME: Only extend SETCC N, N and SETCC N, c for now.
5852*9880d681SAndroid Build Coastguard Worker if (ExtOpc != ISD::ANY_EXTEND && User->getOpcode() == ISD::SETCC) {
5853*9880d681SAndroid Build Coastguard Worker ISD::CondCode CC = cast<CondCodeSDNode>(User->getOperand(2))->get();
5854*9880d681SAndroid Build Coastguard Worker if (ExtOpc == ISD::ZERO_EXTEND && ISD::isSignedIntSetCC(CC))
5855*9880d681SAndroid Build Coastguard Worker // Sign bits will be lost after a zext.
5856*9880d681SAndroid Build Coastguard Worker return false;
5857*9880d681SAndroid Build Coastguard Worker bool Add = false;
5858*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i != 2; ++i) {
5859*9880d681SAndroid Build Coastguard Worker SDValue UseOp = User->getOperand(i);
5860*9880d681SAndroid Build Coastguard Worker if (UseOp == N0)
5861*9880d681SAndroid Build Coastguard Worker continue;
5862*9880d681SAndroid Build Coastguard Worker if (!isa<ConstantSDNode>(UseOp))
5863*9880d681SAndroid Build Coastguard Worker return false;
5864*9880d681SAndroid Build Coastguard Worker Add = true;
5865*9880d681SAndroid Build Coastguard Worker }
5866*9880d681SAndroid Build Coastguard Worker if (Add)
5867*9880d681SAndroid Build Coastguard Worker ExtendNodes.push_back(User);
5868*9880d681SAndroid Build Coastguard Worker continue;
5869*9880d681SAndroid Build Coastguard Worker }
5870*9880d681SAndroid Build Coastguard Worker // If truncates aren't free and there are users we can't
5871*9880d681SAndroid Build Coastguard Worker // extend, it isn't worthwhile.
5872*9880d681SAndroid Build Coastguard Worker if (!isTruncFree)
5873*9880d681SAndroid Build Coastguard Worker return false;
5874*9880d681SAndroid Build Coastguard Worker // Remember if this value is live-out.
5875*9880d681SAndroid Build Coastguard Worker if (User->getOpcode() == ISD::CopyToReg)
5876*9880d681SAndroid Build Coastguard Worker HasCopyToRegUses = true;
5877*9880d681SAndroid Build Coastguard Worker }
5878*9880d681SAndroid Build Coastguard Worker
5879*9880d681SAndroid Build Coastguard Worker if (HasCopyToRegUses) {
5880*9880d681SAndroid Build Coastguard Worker bool BothLiveOut = false;
5881*9880d681SAndroid Build Coastguard Worker for (SDNode::use_iterator UI = N->use_begin(), UE = N->use_end();
5882*9880d681SAndroid Build Coastguard Worker UI != UE; ++UI) {
5883*9880d681SAndroid Build Coastguard Worker SDUse &Use = UI.getUse();
5884*9880d681SAndroid Build Coastguard Worker if (Use.getResNo() == 0 && Use.getUser()->getOpcode() == ISD::CopyToReg) {
5885*9880d681SAndroid Build Coastguard Worker BothLiveOut = true;
5886*9880d681SAndroid Build Coastguard Worker break;
5887*9880d681SAndroid Build Coastguard Worker }
5888*9880d681SAndroid Build Coastguard Worker }
5889*9880d681SAndroid Build Coastguard Worker if (BothLiveOut)
5890*9880d681SAndroid Build Coastguard Worker // Both unextended and extended values are live out. There had better be
5891*9880d681SAndroid Build Coastguard Worker // a good reason for the transformation.
5892*9880d681SAndroid Build Coastguard Worker return ExtendNodes.size();
5893*9880d681SAndroid Build Coastguard Worker }
5894*9880d681SAndroid Build Coastguard Worker return true;
5895*9880d681SAndroid Build Coastguard Worker }
5896*9880d681SAndroid Build Coastguard Worker
ExtendSetCCUses(const SmallVectorImpl<SDNode * > & SetCCs,SDValue Trunc,SDValue ExtLoad,const SDLoc & DL,ISD::NodeType ExtType)5897*9880d681SAndroid Build Coastguard Worker void DAGCombiner::ExtendSetCCUses(const SmallVectorImpl<SDNode *> &SetCCs,
5898*9880d681SAndroid Build Coastguard Worker SDValue Trunc, SDValue ExtLoad,
5899*9880d681SAndroid Build Coastguard Worker const SDLoc &DL, ISD::NodeType ExtType) {
5900*9880d681SAndroid Build Coastguard Worker // Extend SetCC uses if necessary.
5901*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0, e = SetCCs.size(); i != e; ++i) {
5902*9880d681SAndroid Build Coastguard Worker SDNode *SetCC = SetCCs[i];
5903*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 4> Ops;
5904*9880d681SAndroid Build Coastguard Worker
5905*9880d681SAndroid Build Coastguard Worker for (unsigned j = 0; j != 2; ++j) {
5906*9880d681SAndroid Build Coastguard Worker SDValue SOp = SetCC->getOperand(j);
5907*9880d681SAndroid Build Coastguard Worker if (SOp == Trunc)
5908*9880d681SAndroid Build Coastguard Worker Ops.push_back(ExtLoad);
5909*9880d681SAndroid Build Coastguard Worker else
5910*9880d681SAndroid Build Coastguard Worker Ops.push_back(DAG.getNode(ExtType, DL, ExtLoad->getValueType(0), SOp));
5911*9880d681SAndroid Build Coastguard Worker }
5912*9880d681SAndroid Build Coastguard Worker
5913*9880d681SAndroid Build Coastguard Worker Ops.push_back(SetCC->getOperand(2));
5914*9880d681SAndroid Build Coastguard Worker CombineTo(SetCC, DAG.getNode(ISD::SETCC, DL, SetCC->getValueType(0), Ops));
5915*9880d681SAndroid Build Coastguard Worker }
5916*9880d681SAndroid Build Coastguard Worker }
5917*9880d681SAndroid Build Coastguard Worker
5918*9880d681SAndroid Build Coastguard Worker // FIXME: Bring more similar combines here, common to sext/zext (maybe aext?).
CombineExtLoad(SDNode * N)5919*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::CombineExtLoad(SDNode *N) {
5920*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
5921*9880d681SAndroid Build Coastguard Worker EVT DstVT = N->getValueType(0);
5922*9880d681SAndroid Build Coastguard Worker EVT SrcVT = N0.getValueType();
5923*9880d681SAndroid Build Coastguard Worker
5924*9880d681SAndroid Build Coastguard Worker assert((N->getOpcode() == ISD::SIGN_EXTEND ||
5925*9880d681SAndroid Build Coastguard Worker N->getOpcode() == ISD::ZERO_EXTEND) &&
5926*9880d681SAndroid Build Coastguard Worker "Unexpected node type (not an extend)!");
5927*9880d681SAndroid Build Coastguard Worker
5928*9880d681SAndroid Build Coastguard Worker // fold (sext (load x)) to multiple smaller sextloads; same for zext.
5929*9880d681SAndroid Build Coastguard Worker // For example, on a target with legal v4i32, but illegal v8i32, turn:
5930*9880d681SAndroid Build Coastguard Worker // (v8i32 (sext (v8i16 (load x))))
5931*9880d681SAndroid Build Coastguard Worker // into:
5932*9880d681SAndroid Build Coastguard Worker // (v8i32 (concat_vectors (v4i32 (sextload x)),
5933*9880d681SAndroid Build Coastguard Worker // (v4i32 (sextload (x + 16)))))
5934*9880d681SAndroid Build Coastguard Worker // Where uses of the original load, i.e.:
5935*9880d681SAndroid Build Coastguard Worker // (v8i16 (load x))
5936*9880d681SAndroid Build Coastguard Worker // are replaced with:
5937*9880d681SAndroid Build Coastguard Worker // (v8i16 (truncate
5938*9880d681SAndroid Build Coastguard Worker // (v8i32 (concat_vectors (v4i32 (sextload x)),
5939*9880d681SAndroid Build Coastguard Worker // (v4i32 (sextload (x + 16)))))))
5940*9880d681SAndroid Build Coastguard Worker //
5941*9880d681SAndroid Build Coastguard Worker // This combine is only applicable to illegal, but splittable, vectors.
5942*9880d681SAndroid Build Coastguard Worker // All legal types, and illegal non-vector types, are handled elsewhere.
5943*9880d681SAndroid Build Coastguard Worker // This combine is controlled by TargetLowering::isVectorLoadExtDesirable.
5944*9880d681SAndroid Build Coastguard Worker //
5945*9880d681SAndroid Build Coastguard Worker if (N0->getOpcode() != ISD::LOAD)
5946*9880d681SAndroid Build Coastguard Worker return SDValue();
5947*9880d681SAndroid Build Coastguard Worker
5948*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = cast<LoadSDNode>(N0);
5949*9880d681SAndroid Build Coastguard Worker
5950*9880d681SAndroid Build Coastguard Worker if (!ISD::isNON_EXTLoad(LN0) || !ISD::isUNINDEXEDLoad(LN0) ||
5951*9880d681SAndroid Build Coastguard Worker !N0.hasOneUse() || LN0->isVolatile() || !DstVT.isVector() ||
5952*9880d681SAndroid Build Coastguard Worker !DstVT.isPow2VectorType() || !TLI.isVectorLoadExtDesirable(SDValue(N, 0)))
5953*9880d681SAndroid Build Coastguard Worker return SDValue();
5954*9880d681SAndroid Build Coastguard Worker
5955*9880d681SAndroid Build Coastguard Worker SmallVector<SDNode *, 4> SetCCs;
5956*9880d681SAndroid Build Coastguard Worker if (!ExtendUsesToFormExtLoad(N, N0, N->getOpcode(), SetCCs, TLI))
5957*9880d681SAndroid Build Coastguard Worker return SDValue();
5958*9880d681SAndroid Build Coastguard Worker
5959*9880d681SAndroid Build Coastguard Worker ISD::LoadExtType ExtType =
5960*9880d681SAndroid Build Coastguard Worker N->getOpcode() == ISD::SIGN_EXTEND ? ISD::SEXTLOAD : ISD::ZEXTLOAD;
5961*9880d681SAndroid Build Coastguard Worker
5962*9880d681SAndroid Build Coastguard Worker // Try to split the vector types to get down to legal types.
5963*9880d681SAndroid Build Coastguard Worker EVT SplitSrcVT = SrcVT;
5964*9880d681SAndroid Build Coastguard Worker EVT SplitDstVT = DstVT;
5965*9880d681SAndroid Build Coastguard Worker while (!TLI.isLoadExtLegalOrCustom(ExtType, SplitDstVT, SplitSrcVT) &&
5966*9880d681SAndroid Build Coastguard Worker SplitSrcVT.getVectorNumElements() > 1) {
5967*9880d681SAndroid Build Coastguard Worker SplitDstVT = DAG.GetSplitDestVTs(SplitDstVT).first;
5968*9880d681SAndroid Build Coastguard Worker SplitSrcVT = DAG.GetSplitDestVTs(SplitSrcVT).first;
5969*9880d681SAndroid Build Coastguard Worker }
5970*9880d681SAndroid Build Coastguard Worker
5971*9880d681SAndroid Build Coastguard Worker if (!TLI.isLoadExtLegalOrCustom(ExtType, SplitDstVT, SplitSrcVT))
5972*9880d681SAndroid Build Coastguard Worker return SDValue();
5973*9880d681SAndroid Build Coastguard Worker
5974*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
5975*9880d681SAndroid Build Coastguard Worker const unsigned NumSplits =
5976*9880d681SAndroid Build Coastguard Worker DstVT.getVectorNumElements() / SplitDstVT.getVectorNumElements();
5977*9880d681SAndroid Build Coastguard Worker const unsigned Stride = SplitSrcVT.getStoreSize();
5978*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 4> Loads;
5979*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 4> Chains;
5980*9880d681SAndroid Build Coastguard Worker
5981*9880d681SAndroid Build Coastguard Worker SDValue BasePtr = LN0->getBasePtr();
5982*9880d681SAndroid Build Coastguard Worker for (unsigned Idx = 0; Idx < NumSplits; Idx++) {
5983*9880d681SAndroid Build Coastguard Worker const unsigned Offset = Idx * Stride;
5984*9880d681SAndroid Build Coastguard Worker const unsigned Align = MinAlign(LN0->getAlignment(), Offset);
5985*9880d681SAndroid Build Coastguard Worker
5986*9880d681SAndroid Build Coastguard Worker SDValue SplitLoad = DAG.getExtLoad(
5987*9880d681SAndroid Build Coastguard Worker ExtType, DL, SplitDstVT, LN0->getChain(), BasePtr,
5988*9880d681SAndroid Build Coastguard Worker LN0->getPointerInfo().getWithOffset(Offset), SplitSrcVT,
5989*9880d681SAndroid Build Coastguard Worker LN0->isVolatile(), LN0->isNonTemporal(), LN0->isInvariant(),
5990*9880d681SAndroid Build Coastguard Worker Align, LN0->getAAInfo());
5991*9880d681SAndroid Build Coastguard Worker
5992*9880d681SAndroid Build Coastguard Worker BasePtr = DAG.getNode(ISD::ADD, DL, BasePtr.getValueType(), BasePtr,
5993*9880d681SAndroid Build Coastguard Worker DAG.getConstant(Stride, DL, BasePtr.getValueType()));
5994*9880d681SAndroid Build Coastguard Worker
5995*9880d681SAndroid Build Coastguard Worker Loads.push_back(SplitLoad.getValue(0));
5996*9880d681SAndroid Build Coastguard Worker Chains.push_back(SplitLoad.getValue(1));
5997*9880d681SAndroid Build Coastguard Worker }
5998*9880d681SAndroid Build Coastguard Worker
5999*9880d681SAndroid Build Coastguard Worker SDValue NewChain = DAG.getNode(ISD::TokenFactor, DL, MVT::Other, Chains);
6000*9880d681SAndroid Build Coastguard Worker SDValue NewValue = DAG.getNode(ISD::CONCAT_VECTORS, DL, DstVT, Loads);
6001*9880d681SAndroid Build Coastguard Worker
6002*9880d681SAndroid Build Coastguard Worker CombineTo(N, NewValue);
6003*9880d681SAndroid Build Coastguard Worker
6004*9880d681SAndroid Build Coastguard Worker // Replace uses of the original load (before extension)
6005*9880d681SAndroid Build Coastguard Worker // with a truncate of the concatenated sextloaded vectors.
6006*9880d681SAndroid Build Coastguard Worker SDValue Trunc =
6007*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::TRUNCATE, SDLoc(N0), N0.getValueType(), NewValue);
6008*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(), Trunc, NewChain);
6009*9880d681SAndroid Build Coastguard Worker ExtendSetCCUses(SetCCs, Trunc, NewValue, DL,
6010*9880d681SAndroid Build Coastguard Worker (ISD::NodeType)N->getOpcode());
6011*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
6012*9880d681SAndroid Build Coastguard Worker }
6013*9880d681SAndroid Build Coastguard Worker
visitSIGN_EXTEND(SDNode * N)6014*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSIGN_EXTEND(SDNode *N) {
6015*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
6016*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
6017*9880d681SAndroid Build Coastguard Worker
6018*9880d681SAndroid Build Coastguard Worker if (SDNode *Res = tryToFoldExtendOfConstant(N, TLI, DAG, LegalTypes,
6019*9880d681SAndroid Build Coastguard Worker LegalOperations))
6020*9880d681SAndroid Build Coastguard Worker return SDValue(Res, 0);
6021*9880d681SAndroid Build Coastguard Worker
6022*9880d681SAndroid Build Coastguard Worker // fold (sext (sext x)) -> (sext x)
6023*9880d681SAndroid Build Coastguard Worker // fold (sext (aext x)) -> (sext x)
6024*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SIGN_EXTEND || N0.getOpcode() == ISD::ANY_EXTEND)
6025*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SIGN_EXTEND, SDLoc(N), VT,
6026*9880d681SAndroid Build Coastguard Worker N0.getOperand(0));
6027*9880d681SAndroid Build Coastguard Worker
6028*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::TRUNCATE) {
6029*9880d681SAndroid Build Coastguard Worker // fold (sext (truncate (load x))) -> (sext (smaller load x))
6030*9880d681SAndroid Build Coastguard Worker // fold (sext (truncate (srl (load x), c))) -> (sext (smaller load (x+c/n)))
6031*9880d681SAndroid Build Coastguard Worker if (SDValue NarrowLoad = ReduceLoadWidth(N0.getNode())) {
6032*9880d681SAndroid Build Coastguard Worker SDNode* oye = N0.getNode()->getOperand(0).getNode();
6033*9880d681SAndroid Build Coastguard Worker if (NarrowLoad.getNode() != N0.getNode()) {
6034*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(), NarrowLoad);
6035*9880d681SAndroid Build Coastguard Worker // CombineTo deleted the truncate, if needed, but not what's under it.
6036*9880d681SAndroid Build Coastguard Worker AddToWorklist(oye);
6037*9880d681SAndroid Build Coastguard Worker }
6038*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
6039*9880d681SAndroid Build Coastguard Worker }
6040*9880d681SAndroid Build Coastguard Worker
6041*9880d681SAndroid Build Coastguard Worker // See if the value being truncated is already sign extended. If so, just
6042*9880d681SAndroid Build Coastguard Worker // eliminate the trunc/sext pair.
6043*9880d681SAndroid Build Coastguard Worker SDValue Op = N0.getOperand(0);
6044*9880d681SAndroid Build Coastguard Worker unsigned OpBits = Op.getValueType().getScalarType().getSizeInBits();
6045*9880d681SAndroid Build Coastguard Worker unsigned MidBits = N0.getValueType().getScalarType().getSizeInBits();
6046*9880d681SAndroid Build Coastguard Worker unsigned DestBits = VT.getScalarType().getSizeInBits();
6047*9880d681SAndroid Build Coastguard Worker unsigned NumSignBits = DAG.ComputeNumSignBits(Op);
6048*9880d681SAndroid Build Coastguard Worker
6049*9880d681SAndroid Build Coastguard Worker if (OpBits == DestBits) {
6050*9880d681SAndroid Build Coastguard Worker // Op is i32, Mid is i8, and Dest is i32. If Op has more than 24 sign
6051*9880d681SAndroid Build Coastguard Worker // bits, it is already ready.
6052*9880d681SAndroid Build Coastguard Worker if (NumSignBits > DestBits-MidBits)
6053*9880d681SAndroid Build Coastguard Worker return Op;
6054*9880d681SAndroid Build Coastguard Worker } else if (OpBits < DestBits) {
6055*9880d681SAndroid Build Coastguard Worker // Op is i32, Mid is i8, and Dest is i64. If Op has more than 24 sign
6056*9880d681SAndroid Build Coastguard Worker // bits, just sext from i32.
6057*9880d681SAndroid Build Coastguard Worker if (NumSignBits > OpBits-MidBits)
6058*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SIGN_EXTEND, SDLoc(N), VT, Op);
6059*9880d681SAndroid Build Coastguard Worker } else {
6060*9880d681SAndroid Build Coastguard Worker // Op is i64, Mid is i8, and Dest is i32. If Op has more than 56 sign
6061*9880d681SAndroid Build Coastguard Worker // bits, just truncate to i32.
6062*9880d681SAndroid Build Coastguard Worker if (NumSignBits > OpBits-MidBits)
6063*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TRUNCATE, SDLoc(N), VT, Op);
6064*9880d681SAndroid Build Coastguard Worker }
6065*9880d681SAndroid Build Coastguard Worker
6066*9880d681SAndroid Build Coastguard Worker // fold (sext (truncate x)) -> (sextinreg x).
6067*9880d681SAndroid Build Coastguard Worker if (!LegalOperations || TLI.isOperationLegal(ISD::SIGN_EXTEND_INREG,
6068*9880d681SAndroid Build Coastguard Worker N0.getValueType())) {
6069*9880d681SAndroid Build Coastguard Worker if (OpBits < DestBits)
6070*9880d681SAndroid Build Coastguard Worker Op = DAG.getNode(ISD::ANY_EXTEND, SDLoc(N0), VT, Op);
6071*9880d681SAndroid Build Coastguard Worker else if (OpBits > DestBits)
6072*9880d681SAndroid Build Coastguard Worker Op = DAG.getNode(ISD::TRUNCATE, SDLoc(N0), VT, Op);
6073*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SIGN_EXTEND_INREG, SDLoc(N), VT, Op,
6074*9880d681SAndroid Build Coastguard Worker DAG.getValueType(N0.getValueType()));
6075*9880d681SAndroid Build Coastguard Worker }
6076*9880d681SAndroid Build Coastguard Worker }
6077*9880d681SAndroid Build Coastguard Worker
6078*9880d681SAndroid Build Coastguard Worker // fold (sext (load x)) -> (sext (truncate (sextload x)))
6079*9880d681SAndroid Build Coastguard Worker // Only generate vector extloads when 1) they're legal, and 2) they are
6080*9880d681SAndroid Build Coastguard Worker // deemed desirable by the target.
6081*9880d681SAndroid Build Coastguard Worker if (ISD::isNON_EXTLoad(N0.getNode()) && ISD::isUNINDEXEDLoad(N0.getNode()) &&
6082*9880d681SAndroid Build Coastguard Worker ((!LegalOperations && !VT.isVector() &&
6083*9880d681SAndroid Build Coastguard Worker !cast<LoadSDNode>(N0)->isVolatile()) ||
6084*9880d681SAndroid Build Coastguard Worker TLI.isLoadExtLegal(ISD::SEXTLOAD, VT, N0.getValueType()))) {
6085*9880d681SAndroid Build Coastguard Worker bool DoXform = true;
6086*9880d681SAndroid Build Coastguard Worker SmallVector<SDNode*, 4> SetCCs;
6087*9880d681SAndroid Build Coastguard Worker if (!N0.hasOneUse())
6088*9880d681SAndroid Build Coastguard Worker DoXform = ExtendUsesToFormExtLoad(N, N0, ISD::SIGN_EXTEND, SetCCs, TLI);
6089*9880d681SAndroid Build Coastguard Worker if (VT.isVector())
6090*9880d681SAndroid Build Coastguard Worker DoXform &= TLI.isVectorLoadExtDesirable(SDValue(N, 0));
6091*9880d681SAndroid Build Coastguard Worker if (DoXform) {
6092*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = cast<LoadSDNode>(N0);
6093*9880d681SAndroid Build Coastguard Worker SDValue ExtLoad = DAG.getExtLoad(ISD::SEXTLOAD, SDLoc(N), VT,
6094*9880d681SAndroid Build Coastguard Worker LN0->getChain(),
6095*9880d681SAndroid Build Coastguard Worker LN0->getBasePtr(), N0.getValueType(),
6096*9880d681SAndroid Build Coastguard Worker LN0->getMemOperand());
6097*9880d681SAndroid Build Coastguard Worker CombineTo(N, ExtLoad);
6098*9880d681SAndroid Build Coastguard Worker SDValue Trunc = DAG.getNode(ISD::TRUNCATE, SDLoc(N0),
6099*9880d681SAndroid Build Coastguard Worker N0.getValueType(), ExtLoad);
6100*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(), Trunc, ExtLoad.getValue(1));
6101*9880d681SAndroid Build Coastguard Worker ExtendSetCCUses(SetCCs, Trunc, ExtLoad, SDLoc(N),
6102*9880d681SAndroid Build Coastguard Worker ISD::SIGN_EXTEND);
6103*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
6104*9880d681SAndroid Build Coastguard Worker }
6105*9880d681SAndroid Build Coastguard Worker }
6106*9880d681SAndroid Build Coastguard Worker
6107*9880d681SAndroid Build Coastguard Worker // fold (sext (load x)) to multiple smaller sextloads.
6108*9880d681SAndroid Build Coastguard Worker // Only on illegal but splittable vectors.
6109*9880d681SAndroid Build Coastguard Worker if (SDValue ExtLoad = CombineExtLoad(N))
6110*9880d681SAndroid Build Coastguard Worker return ExtLoad;
6111*9880d681SAndroid Build Coastguard Worker
6112*9880d681SAndroid Build Coastguard Worker // fold (sext (sextload x)) -> (sext (truncate (sextload x)))
6113*9880d681SAndroid Build Coastguard Worker // fold (sext ( extload x)) -> (sext (truncate (sextload x)))
6114*9880d681SAndroid Build Coastguard Worker if ((ISD::isSEXTLoad(N0.getNode()) || ISD::isEXTLoad(N0.getNode())) &&
6115*9880d681SAndroid Build Coastguard Worker ISD::isUNINDEXEDLoad(N0.getNode()) && N0.hasOneUse()) {
6116*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = cast<LoadSDNode>(N0);
6117*9880d681SAndroid Build Coastguard Worker EVT MemVT = LN0->getMemoryVT();
6118*9880d681SAndroid Build Coastguard Worker if ((!LegalOperations && !LN0->isVolatile()) ||
6119*9880d681SAndroid Build Coastguard Worker TLI.isLoadExtLegal(ISD::SEXTLOAD, VT, MemVT)) {
6120*9880d681SAndroid Build Coastguard Worker SDValue ExtLoad = DAG.getExtLoad(ISD::SEXTLOAD, SDLoc(N), VT,
6121*9880d681SAndroid Build Coastguard Worker LN0->getChain(),
6122*9880d681SAndroid Build Coastguard Worker LN0->getBasePtr(), MemVT,
6123*9880d681SAndroid Build Coastguard Worker LN0->getMemOperand());
6124*9880d681SAndroid Build Coastguard Worker CombineTo(N, ExtLoad);
6125*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(),
6126*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::TRUNCATE, SDLoc(N0),
6127*9880d681SAndroid Build Coastguard Worker N0.getValueType(), ExtLoad),
6128*9880d681SAndroid Build Coastguard Worker ExtLoad.getValue(1));
6129*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
6130*9880d681SAndroid Build Coastguard Worker }
6131*9880d681SAndroid Build Coastguard Worker }
6132*9880d681SAndroid Build Coastguard Worker
6133*9880d681SAndroid Build Coastguard Worker // fold (sext (and/or/xor (load x), cst)) ->
6134*9880d681SAndroid Build Coastguard Worker // (and/or/xor (sextload x), (sext cst))
6135*9880d681SAndroid Build Coastguard Worker if ((N0.getOpcode() == ISD::AND || N0.getOpcode() == ISD::OR ||
6136*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::XOR) &&
6137*9880d681SAndroid Build Coastguard Worker isa<LoadSDNode>(N0.getOperand(0)) &&
6138*9880d681SAndroid Build Coastguard Worker N0.getOperand(1).getOpcode() == ISD::Constant &&
6139*9880d681SAndroid Build Coastguard Worker TLI.isLoadExtLegal(ISD::SEXTLOAD, VT, N0.getValueType()) &&
6140*9880d681SAndroid Build Coastguard Worker (!LegalOperations && TLI.isOperationLegal(N0.getOpcode(), VT))) {
6141*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = cast<LoadSDNode>(N0.getOperand(0));
6142*9880d681SAndroid Build Coastguard Worker if (LN0->getExtensionType() != ISD::ZEXTLOAD && LN0->isUnindexed()) {
6143*9880d681SAndroid Build Coastguard Worker bool DoXform = true;
6144*9880d681SAndroid Build Coastguard Worker SmallVector<SDNode*, 4> SetCCs;
6145*9880d681SAndroid Build Coastguard Worker if (!N0.hasOneUse())
6146*9880d681SAndroid Build Coastguard Worker DoXform = ExtendUsesToFormExtLoad(N, N0.getOperand(0), ISD::SIGN_EXTEND,
6147*9880d681SAndroid Build Coastguard Worker SetCCs, TLI);
6148*9880d681SAndroid Build Coastguard Worker if (DoXform) {
6149*9880d681SAndroid Build Coastguard Worker SDValue ExtLoad = DAG.getExtLoad(ISD::SEXTLOAD, SDLoc(LN0), VT,
6150*9880d681SAndroid Build Coastguard Worker LN0->getChain(), LN0->getBasePtr(),
6151*9880d681SAndroid Build Coastguard Worker LN0->getMemoryVT(),
6152*9880d681SAndroid Build Coastguard Worker LN0->getMemOperand());
6153*9880d681SAndroid Build Coastguard Worker APInt Mask = cast<ConstantSDNode>(N0.getOperand(1))->getAPIntValue();
6154*9880d681SAndroid Build Coastguard Worker Mask = Mask.sext(VT.getSizeInBits());
6155*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
6156*9880d681SAndroid Build Coastguard Worker SDValue And = DAG.getNode(N0.getOpcode(), DL, VT,
6157*9880d681SAndroid Build Coastguard Worker ExtLoad, DAG.getConstant(Mask, DL, VT));
6158*9880d681SAndroid Build Coastguard Worker SDValue Trunc = DAG.getNode(ISD::TRUNCATE,
6159*9880d681SAndroid Build Coastguard Worker SDLoc(N0.getOperand(0)),
6160*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getValueType(), ExtLoad);
6161*9880d681SAndroid Build Coastguard Worker CombineTo(N, And);
6162*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getOperand(0).getNode(), Trunc, ExtLoad.getValue(1));
6163*9880d681SAndroid Build Coastguard Worker ExtendSetCCUses(SetCCs, Trunc, ExtLoad, DL,
6164*9880d681SAndroid Build Coastguard Worker ISD::SIGN_EXTEND);
6165*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
6166*9880d681SAndroid Build Coastguard Worker }
6167*9880d681SAndroid Build Coastguard Worker }
6168*9880d681SAndroid Build Coastguard Worker }
6169*9880d681SAndroid Build Coastguard Worker
6170*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SETCC) {
6171*9880d681SAndroid Build Coastguard Worker EVT N0VT = N0.getOperand(0).getValueType();
6172*9880d681SAndroid Build Coastguard Worker // sext(setcc) -> sext_in_reg(vsetcc) for vectors.
6173*9880d681SAndroid Build Coastguard Worker // Only do this before legalize for now.
6174*9880d681SAndroid Build Coastguard Worker if (VT.isVector() && !LegalOperations &&
6175*9880d681SAndroid Build Coastguard Worker TLI.getBooleanContents(N0VT) ==
6176*9880d681SAndroid Build Coastguard Worker TargetLowering::ZeroOrNegativeOneBooleanContent) {
6177*9880d681SAndroid Build Coastguard Worker // On some architectures (such as SSE/NEON/etc) the SETCC result type is
6178*9880d681SAndroid Build Coastguard Worker // of the same size as the compared operands. Only optimize sext(setcc())
6179*9880d681SAndroid Build Coastguard Worker // if this is the case.
6180*9880d681SAndroid Build Coastguard Worker EVT SVT = getSetCCResultType(N0VT);
6181*9880d681SAndroid Build Coastguard Worker
6182*9880d681SAndroid Build Coastguard Worker // We know that the # elements of the results is the same as the
6183*9880d681SAndroid Build Coastguard Worker // # elements of the compare (and the # elements of the compare result
6184*9880d681SAndroid Build Coastguard Worker // for that matter). Check to see that they are the same size. If so,
6185*9880d681SAndroid Build Coastguard Worker // we know that the element size of the sext'd result matches the
6186*9880d681SAndroid Build Coastguard Worker // element size of the compare operands.
6187*9880d681SAndroid Build Coastguard Worker if (VT.getSizeInBits() == SVT.getSizeInBits())
6188*9880d681SAndroid Build Coastguard Worker return DAG.getSetCC(SDLoc(N), VT, N0.getOperand(0),
6189*9880d681SAndroid Build Coastguard Worker N0.getOperand(1),
6190*9880d681SAndroid Build Coastguard Worker cast<CondCodeSDNode>(N0.getOperand(2))->get());
6191*9880d681SAndroid Build Coastguard Worker
6192*9880d681SAndroid Build Coastguard Worker // If the desired elements are smaller or larger than the source
6193*9880d681SAndroid Build Coastguard Worker // elements we can use a matching integer vector type and then
6194*9880d681SAndroid Build Coastguard Worker // truncate/sign extend
6195*9880d681SAndroid Build Coastguard Worker EVT MatchingVectorType = N0VT.changeVectorElementTypeToInteger();
6196*9880d681SAndroid Build Coastguard Worker if (SVT == MatchingVectorType) {
6197*9880d681SAndroid Build Coastguard Worker SDValue VsetCC = DAG.getSetCC(SDLoc(N), MatchingVectorType,
6198*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N0.getOperand(1),
6199*9880d681SAndroid Build Coastguard Worker cast<CondCodeSDNode>(N0.getOperand(2))->get());
6200*9880d681SAndroid Build Coastguard Worker return DAG.getSExtOrTrunc(VsetCC, SDLoc(N), VT);
6201*9880d681SAndroid Build Coastguard Worker }
6202*9880d681SAndroid Build Coastguard Worker }
6203*9880d681SAndroid Build Coastguard Worker
6204*9880d681SAndroid Build Coastguard Worker // sext(setcc x, y, cc) -> (select (setcc x, y, cc), -1, 0)
6205*9880d681SAndroid Build Coastguard Worker unsigned ElementWidth = VT.getScalarType().getSizeInBits();
6206*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
6207*9880d681SAndroid Build Coastguard Worker SDValue NegOne =
6208*9880d681SAndroid Build Coastguard Worker DAG.getConstant(APInt::getAllOnesValue(ElementWidth), DL, VT);
6209*9880d681SAndroid Build Coastguard Worker if (SDValue SCC = SimplifySelectCC(
6210*9880d681SAndroid Build Coastguard Worker DL, N0.getOperand(0), N0.getOperand(1), NegOne,
6211*9880d681SAndroid Build Coastguard Worker DAG.getConstant(0, DL, VT),
6212*9880d681SAndroid Build Coastguard Worker cast<CondCodeSDNode>(N0.getOperand(2))->get(), true))
6213*9880d681SAndroid Build Coastguard Worker return SCC;
6214*9880d681SAndroid Build Coastguard Worker
6215*9880d681SAndroid Build Coastguard Worker if (!VT.isVector()) {
6216*9880d681SAndroid Build Coastguard Worker EVT SetCCVT = getSetCCResultType(N0.getOperand(0).getValueType());
6217*9880d681SAndroid Build Coastguard Worker if (!LegalOperations ||
6218*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegal(ISD::SETCC, N0.getOperand(0).getValueType())) {
6219*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
6220*9880d681SAndroid Build Coastguard Worker ISD::CondCode CC = cast<CondCodeSDNode>(N0.getOperand(2))->get();
6221*9880d681SAndroid Build Coastguard Worker SDValue SetCC = DAG.getSetCC(DL, SetCCVT,
6222*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N0.getOperand(1), CC);
6223*9880d681SAndroid Build Coastguard Worker return DAG.getSelect(DL, VT, SetCC,
6224*9880d681SAndroid Build Coastguard Worker NegOne, DAG.getConstant(0, DL, VT));
6225*9880d681SAndroid Build Coastguard Worker }
6226*9880d681SAndroid Build Coastguard Worker }
6227*9880d681SAndroid Build Coastguard Worker }
6228*9880d681SAndroid Build Coastguard Worker
6229*9880d681SAndroid Build Coastguard Worker // fold (sext x) -> (zext x) if the sign bit is known zero.
6230*9880d681SAndroid Build Coastguard Worker if ((!LegalOperations || TLI.isOperationLegal(ISD::ZERO_EXTEND, VT)) &&
6231*9880d681SAndroid Build Coastguard Worker DAG.SignBitIsZero(N0))
6232*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ZERO_EXTEND, SDLoc(N), VT, N0);
6233*9880d681SAndroid Build Coastguard Worker
6234*9880d681SAndroid Build Coastguard Worker return SDValue();
6235*9880d681SAndroid Build Coastguard Worker }
6236*9880d681SAndroid Build Coastguard Worker
6237*9880d681SAndroid Build Coastguard Worker // isTruncateOf - If N is a truncate of some other value, return true, record
6238*9880d681SAndroid Build Coastguard Worker // the value being truncated in Op and which of Op's bits are zero in KnownZero.
6239*9880d681SAndroid Build Coastguard Worker // This function computes KnownZero to avoid a duplicated call to
6240*9880d681SAndroid Build Coastguard Worker // computeKnownBits in the caller.
isTruncateOf(SelectionDAG & DAG,SDValue N,SDValue & Op,APInt & KnownZero)6241*9880d681SAndroid Build Coastguard Worker static bool isTruncateOf(SelectionDAG &DAG, SDValue N, SDValue &Op,
6242*9880d681SAndroid Build Coastguard Worker APInt &KnownZero) {
6243*9880d681SAndroid Build Coastguard Worker APInt KnownOne;
6244*9880d681SAndroid Build Coastguard Worker if (N->getOpcode() == ISD::TRUNCATE) {
6245*9880d681SAndroid Build Coastguard Worker Op = N->getOperand(0);
6246*9880d681SAndroid Build Coastguard Worker DAG.computeKnownBits(Op, KnownZero, KnownOne);
6247*9880d681SAndroid Build Coastguard Worker return true;
6248*9880d681SAndroid Build Coastguard Worker }
6249*9880d681SAndroid Build Coastguard Worker
6250*9880d681SAndroid Build Coastguard Worker if (N->getOpcode() != ISD::SETCC || N->getValueType(0) != MVT::i1 ||
6251*9880d681SAndroid Build Coastguard Worker cast<CondCodeSDNode>(N->getOperand(2))->get() != ISD::SETNE)
6252*9880d681SAndroid Build Coastguard Worker return false;
6253*9880d681SAndroid Build Coastguard Worker
6254*9880d681SAndroid Build Coastguard Worker SDValue Op0 = N->getOperand(0);
6255*9880d681SAndroid Build Coastguard Worker SDValue Op1 = N->getOperand(1);
6256*9880d681SAndroid Build Coastguard Worker assert(Op0.getValueType() == Op1.getValueType());
6257*9880d681SAndroid Build Coastguard Worker
6258*9880d681SAndroid Build Coastguard Worker if (isNullConstant(Op0))
6259*9880d681SAndroid Build Coastguard Worker Op = Op1;
6260*9880d681SAndroid Build Coastguard Worker else if (isNullConstant(Op1))
6261*9880d681SAndroid Build Coastguard Worker Op = Op0;
6262*9880d681SAndroid Build Coastguard Worker else
6263*9880d681SAndroid Build Coastguard Worker return false;
6264*9880d681SAndroid Build Coastguard Worker
6265*9880d681SAndroid Build Coastguard Worker DAG.computeKnownBits(Op, KnownZero, KnownOne);
6266*9880d681SAndroid Build Coastguard Worker
6267*9880d681SAndroid Build Coastguard Worker if (!(KnownZero | APInt(Op.getValueSizeInBits(), 1)).isAllOnesValue())
6268*9880d681SAndroid Build Coastguard Worker return false;
6269*9880d681SAndroid Build Coastguard Worker
6270*9880d681SAndroid Build Coastguard Worker return true;
6271*9880d681SAndroid Build Coastguard Worker }
6272*9880d681SAndroid Build Coastguard Worker
visitZERO_EXTEND(SDNode * N)6273*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitZERO_EXTEND(SDNode *N) {
6274*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
6275*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
6276*9880d681SAndroid Build Coastguard Worker
6277*9880d681SAndroid Build Coastguard Worker if (SDNode *Res = tryToFoldExtendOfConstant(N, TLI, DAG, LegalTypes,
6278*9880d681SAndroid Build Coastguard Worker LegalOperations))
6279*9880d681SAndroid Build Coastguard Worker return SDValue(Res, 0);
6280*9880d681SAndroid Build Coastguard Worker
6281*9880d681SAndroid Build Coastguard Worker // fold (zext (zext x)) -> (zext x)
6282*9880d681SAndroid Build Coastguard Worker // fold (zext (aext x)) -> (zext x)
6283*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::ZERO_EXTEND || N0.getOpcode() == ISD::ANY_EXTEND)
6284*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ZERO_EXTEND, SDLoc(N), VT,
6285*9880d681SAndroid Build Coastguard Worker N0.getOperand(0));
6286*9880d681SAndroid Build Coastguard Worker
6287*9880d681SAndroid Build Coastguard Worker // fold (zext (truncate x)) -> (zext x) or
6288*9880d681SAndroid Build Coastguard Worker // (zext (truncate x)) -> (truncate x)
6289*9880d681SAndroid Build Coastguard Worker // This is valid when the truncated bits of x are already zero.
6290*9880d681SAndroid Build Coastguard Worker // FIXME: We should extend this to work for vectors too.
6291*9880d681SAndroid Build Coastguard Worker SDValue Op;
6292*9880d681SAndroid Build Coastguard Worker APInt KnownZero;
6293*9880d681SAndroid Build Coastguard Worker if (!VT.isVector() && isTruncateOf(DAG, N0, Op, KnownZero)) {
6294*9880d681SAndroid Build Coastguard Worker APInt TruncatedBits =
6295*9880d681SAndroid Build Coastguard Worker (Op.getValueSizeInBits() == N0.getValueSizeInBits()) ?
6296*9880d681SAndroid Build Coastguard Worker APInt(Op.getValueSizeInBits(), 0) :
6297*9880d681SAndroid Build Coastguard Worker APInt::getBitsSet(Op.getValueSizeInBits(),
6298*9880d681SAndroid Build Coastguard Worker N0.getValueSizeInBits(),
6299*9880d681SAndroid Build Coastguard Worker std::min(Op.getValueSizeInBits(),
6300*9880d681SAndroid Build Coastguard Worker VT.getSizeInBits()));
6301*9880d681SAndroid Build Coastguard Worker if (TruncatedBits == (KnownZero & TruncatedBits)) {
6302*9880d681SAndroid Build Coastguard Worker if (VT.bitsGT(Op.getValueType()))
6303*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ZERO_EXTEND, SDLoc(N), VT, Op);
6304*9880d681SAndroid Build Coastguard Worker if (VT.bitsLT(Op.getValueType()))
6305*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TRUNCATE, SDLoc(N), VT, Op);
6306*9880d681SAndroid Build Coastguard Worker
6307*9880d681SAndroid Build Coastguard Worker return Op;
6308*9880d681SAndroid Build Coastguard Worker }
6309*9880d681SAndroid Build Coastguard Worker }
6310*9880d681SAndroid Build Coastguard Worker
6311*9880d681SAndroid Build Coastguard Worker // fold (zext (truncate (load x))) -> (zext (smaller load x))
6312*9880d681SAndroid Build Coastguard Worker // fold (zext (truncate (srl (load x), c))) -> (zext (small load (x+c/n)))
6313*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::TRUNCATE) {
6314*9880d681SAndroid Build Coastguard Worker if (SDValue NarrowLoad = ReduceLoadWidth(N0.getNode())) {
6315*9880d681SAndroid Build Coastguard Worker SDNode* oye = N0.getNode()->getOperand(0).getNode();
6316*9880d681SAndroid Build Coastguard Worker if (NarrowLoad.getNode() != N0.getNode()) {
6317*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(), NarrowLoad);
6318*9880d681SAndroid Build Coastguard Worker // CombineTo deleted the truncate, if needed, but not what's under it.
6319*9880d681SAndroid Build Coastguard Worker AddToWorklist(oye);
6320*9880d681SAndroid Build Coastguard Worker }
6321*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
6322*9880d681SAndroid Build Coastguard Worker }
6323*9880d681SAndroid Build Coastguard Worker }
6324*9880d681SAndroid Build Coastguard Worker
6325*9880d681SAndroid Build Coastguard Worker // fold (zext (truncate x)) -> (and x, mask)
6326*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::TRUNCATE) {
6327*9880d681SAndroid Build Coastguard Worker // fold (zext (truncate (load x))) -> (zext (smaller load x))
6328*9880d681SAndroid Build Coastguard Worker // fold (zext (truncate (srl (load x), c))) -> (zext (smaller load (x+c/n)))
6329*9880d681SAndroid Build Coastguard Worker if (SDValue NarrowLoad = ReduceLoadWidth(N0.getNode())) {
6330*9880d681SAndroid Build Coastguard Worker SDNode *oye = N0.getNode()->getOperand(0).getNode();
6331*9880d681SAndroid Build Coastguard Worker if (NarrowLoad.getNode() != N0.getNode()) {
6332*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(), NarrowLoad);
6333*9880d681SAndroid Build Coastguard Worker // CombineTo deleted the truncate, if needed, but not what's under it.
6334*9880d681SAndroid Build Coastguard Worker AddToWorklist(oye);
6335*9880d681SAndroid Build Coastguard Worker }
6336*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
6337*9880d681SAndroid Build Coastguard Worker }
6338*9880d681SAndroid Build Coastguard Worker
6339*9880d681SAndroid Build Coastguard Worker EVT SrcVT = N0.getOperand(0).getValueType();
6340*9880d681SAndroid Build Coastguard Worker EVT MinVT = N0.getValueType();
6341*9880d681SAndroid Build Coastguard Worker
6342*9880d681SAndroid Build Coastguard Worker // Try to mask before the extension to avoid having to generate a larger mask,
6343*9880d681SAndroid Build Coastguard Worker // possibly over several sub-vectors.
6344*9880d681SAndroid Build Coastguard Worker if (SrcVT.bitsLT(VT)) {
6345*9880d681SAndroid Build Coastguard Worker if (!LegalOperations || (TLI.isOperationLegal(ISD::AND, SrcVT) &&
6346*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegal(ISD::ZERO_EXTEND, VT))) {
6347*9880d681SAndroid Build Coastguard Worker SDValue Op = N0.getOperand(0);
6348*9880d681SAndroid Build Coastguard Worker Op = DAG.getZeroExtendInReg(Op, SDLoc(N), MinVT.getScalarType());
6349*9880d681SAndroid Build Coastguard Worker AddToWorklist(Op.getNode());
6350*9880d681SAndroid Build Coastguard Worker return DAG.getZExtOrTrunc(Op, SDLoc(N), VT);
6351*9880d681SAndroid Build Coastguard Worker }
6352*9880d681SAndroid Build Coastguard Worker }
6353*9880d681SAndroid Build Coastguard Worker
6354*9880d681SAndroid Build Coastguard Worker if (!LegalOperations || TLI.isOperationLegal(ISD::AND, VT)) {
6355*9880d681SAndroid Build Coastguard Worker SDValue Op = N0.getOperand(0);
6356*9880d681SAndroid Build Coastguard Worker if (SrcVT.bitsLT(VT)) {
6357*9880d681SAndroid Build Coastguard Worker Op = DAG.getNode(ISD::ANY_EXTEND, SDLoc(N), VT, Op);
6358*9880d681SAndroid Build Coastguard Worker AddToWorklist(Op.getNode());
6359*9880d681SAndroid Build Coastguard Worker } else if (SrcVT.bitsGT(VT)) {
6360*9880d681SAndroid Build Coastguard Worker Op = DAG.getNode(ISD::TRUNCATE, SDLoc(N), VT, Op);
6361*9880d681SAndroid Build Coastguard Worker AddToWorklist(Op.getNode());
6362*9880d681SAndroid Build Coastguard Worker }
6363*9880d681SAndroid Build Coastguard Worker return DAG.getZeroExtendInReg(Op, SDLoc(N), MinVT.getScalarType());
6364*9880d681SAndroid Build Coastguard Worker }
6365*9880d681SAndroid Build Coastguard Worker }
6366*9880d681SAndroid Build Coastguard Worker
6367*9880d681SAndroid Build Coastguard Worker // Fold (zext (and (trunc x), cst)) -> (and x, cst),
6368*9880d681SAndroid Build Coastguard Worker // if either of the casts is not free.
6369*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::AND &&
6370*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getOpcode() == ISD::TRUNCATE &&
6371*9880d681SAndroid Build Coastguard Worker N0.getOperand(1).getOpcode() == ISD::Constant &&
6372*9880d681SAndroid Build Coastguard Worker (!TLI.isTruncateFree(N0.getOperand(0).getOperand(0).getValueType(),
6373*9880d681SAndroid Build Coastguard Worker N0.getValueType()) ||
6374*9880d681SAndroid Build Coastguard Worker !TLI.isZExtFree(N0.getValueType(), VT))) {
6375*9880d681SAndroid Build Coastguard Worker SDValue X = N0.getOperand(0).getOperand(0);
6376*9880d681SAndroid Build Coastguard Worker if (X.getValueType().bitsLT(VT)) {
6377*9880d681SAndroid Build Coastguard Worker X = DAG.getNode(ISD::ANY_EXTEND, SDLoc(X), VT, X);
6378*9880d681SAndroid Build Coastguard Worker } else if (X.getValueType().bitsGT(VT)) {
6379*9880d681SAndroid Build Coastguard Worker X = DAG.getNode(ISD::TRUNCATE, SDLoc(X), VT, X);
6380*9880d681SAndroid Build Coastguard Worker }
6381*9880d681SAndroid Build Coastguard Worker APInt Mask = cast<ConstantSDNode>(N0.getOperand(1))->getAPIntValue();
6382*9880d681SAndroid Build Coastguard Worker Mask = Mask.zext(VT.getSizeInBits());
6383*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
6384*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, DL, VT,
6385*9880d681SAndroid Build Coastguard Worker X, DAG.getConstant(Mask, DL, VT));
6386*9880d681SAndroid Build Coastguard Worker }
6387*9880d681SAndroid Build Coastguard Worker
6388*9880d681SAndroid Build Coastguard Worker // fold (zext (load x)) -> (zext (truncate (zextload x)))
6389*9880d681SAndroid Build Coastguard Worker // Only generate vector extloads when 1) they're legal, and 2) they are
6390*9880d681SAndroid Build Coastguard Worker // deemed desirable by the target.
6391*9880d681SAndroid Build Coastguard Worker if (ISD::isNON_EXTLoad(N0.getNode()) && ISD::isUNINDEXEDLoad(N0.getNode()) &&
6392*9880d681SAndroid Build Coastguard Worker ((!LegalOperations && !VT.isVector() &&
6393*9880d681SAndroid Build Coastguard Worker !cast<LoadSDNode>(N0)->isVolatile()) ||
6394*9880d681SAndroid Build Coastguard Worker TLI.isLoadExtLegal(ISD::ZEXTLOAD, VT, N0.getValueType()))) {
6395*9880d681SAndroid Build Coastguard Worker bool DoXform = true;
6396*9880d681SAndroid Build Coastguard Worker SmallVector<SDNode*, 4> SetCCs;
6397*9880d681SAndroid Build Coastguard Worker if (!N0.hasOneUse())
6398*9880d681SAndroid Build Coastguard Worker DoXform = ExtendUsesToFormExtLoad(N, N0, ISD::ZERO_EXTEND, SetCCs, TLI);
6399*9880d681SAndroid Build Coastguard Worker if (VT.isVector())
6400*9880d681SAndroid Build Coastguard Worker DoXform &= TLI.isVectorLoadExtDesirable(SDValue(N, 0));
6401*9880d681SAndroid Build Coastguard Worker if (DoXform) {
6402*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = cast<LoadSDNode>(N0);
6403*9880d681SAndroid Build Coastguard Worker SDValue ExtLoad = DAG.getExtLoad(ISD::ZEXTLOAD, SDLoc(N), VT,
6404*9880d681SAndroid Build Coastguard Worker LN0->getChain(),
6405*9880d681SAndroid Build Coastguard Worker LN0->getBasePtr(), N0.getValueType(),
6406*9880d681SAndroid Build Coastguard Worker LN0->getMemOperand());
6407*9880d681SAndroid Build Coastguard Worker CombineTo(N, ExtLoad);
6408*9880d681SAndroid Build Coastguard Worker SDValue Trunc = DAG.getNode(ISD::TRUNCATE, SDLoc(N0),
6409*9880d681SAndroid Build Coastguard Worker N0.getValueType(), ExtLoad);
6410*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(), Trunc, ExtLoad.getValue(1));
6411*9880d681SAndroid Build Coastguard Worker
6412*9880d681SAndroid Build Coastguard Worker ExtendSetCCUses(SetCCs, Trunc, ExtLoad, SDLoc(N),
6413*9880d681SAndroid Build Coastguard Worker ISD::ZERO_EXTEND);
6414*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
6415*9880d681SAndroid Build Coastguard Worker }
6416*9880d681SAndroid Build Coastguard Worker }
6417*9880d681SAndroid Build Coastguard Worker
6418*9880d681SAndroid Build Coastguard Worker // fold (zext (load x)) to multiple smaller zextloads.
6419*9880d681SAndroid Build Coastguard Worker // Only on illegal but splittable vectors.
6420*9880d681SAndroid Build Coastguard Worker if (SDValue ExtLoad = CombineExtLoad(N))
6421*9880d681SAndroid Build Coastguard Worker return ExtLoad;
6422*9880d681SAndroid Build Coastguard Worker
6423*9880d681SAndroid Build Coastguard Worker // fold (zext (and/or/xor (load x), cst)) ->
6424*9880d681SAndroid Build Coastguard Worker // (and/or/xor (zextload x), (zext cst))
6425*9880d681SAndroid Build Coastguard Worker // Unless (and (load x) cst) will match as a zextload already and has
6426*9880d681SAndroid Build Coastguard Worker // additional users.
6427*9880d681SAndroid Build Coastguard Worker if ((N0.getOpcode() == ISD::AND || N0.getOpcode() == ISD::OR ||
6428*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::XOR) &&
6429*9880d681SAndroid Build Coastguard Worker isa<LoadSDNode>(N0.getOperand(0)) &&
6430*9880d681SAndroid Build Coastguard Worker N0.getOperand(1).getOpcode() == ISD::Constant &&
6431*9880d681SAndroid Build Coastguard Worker TLI.isLoadExtLegal(ISD::ZEXTLOAD, VT, N0.getValueType()) &&
6432*9880d681SAndroid Build Coastguard Worker (!LegalOperations && TLI.isOperationLegal(N0.getOpcode(), VT))) {
6433*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = cast<LoadSDNode>(N0.getOperand(0));
6434*9880d681SAndroid Build Coastguard Worker if (LN0->getExtensionType() != ISD::SEXTLOAD && LN0->isUnindexed()) {
6435*9880d681SAndroid Build Coastguard Worker bool DoXform = true;
6436*9880d681SAndroid Build Coastguard Worker SmallVector<SDNode*, 4> SetCCs;
6437*9880d681SAndroid Build Coastguard Worker if (!N0.hasOneUse()) {
6438*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::AND) {
6439*9880d681SAndroid Build Coastguard Worker auto *AndC = cast<ConstantSDNode>(N0.getOperand(1));
6440*9880d681SAndroid Build Coastguard Worker auto NarrowLoad = false;
6441*9880d681SAndroid Build Coastguard Worker EVT LoadResultTy = AndC->getValueType(0);
6442*9880d681SAndroid Build Coastguard Worker EVT ExtVT, LoadedVT;
6443*9880d681SAndroid Build Coastguard Worker if (isAndLoadExtLoad(AndC, LN0, LoadResultTy, ExtVT, LoadedVT,
6444*9880d681SAndroid Build Coastguard Worker NarrowLoad))
6445*9880d681SAndroid Build Coastguard Worker DoXform = false;
6446*9880d681SAndroid Build Coastguard Worker }
6447*9880d681SAndroid Build Coastguard Worker if (DoXform)
6448*9880d681SAndroid Build Coastguard Worker DoXform = ExtendUsesToFormExtLoad(N, N0.getOperand(0),
6449*9880d681SAndroid Build Coastguard Worker ISD::ZERO_EXTEND, SetCCs, TLI);
6450*9880d681SAndroid Build Coastguard Worker }
6451*9880d681SAndroid Build Coastguard Worker if (DoXform) {
6452*9880d681SAndroid Build Coastguard Worker SDValue ExtLoad = DAG.getExtLoad(ISD::ZEXTLOAD, SDLoc(LN0), VT,
6453*9880d681SAndroid Build Coastguard Worker LN0->getChain(), LN0->getBasePtr(),
6454*9880d681SAndroid Build Coastguard Worker LN0->getMemoryVT(),
6455*9880d681SAndroid Build Coastguard Worker LN0->getMemOperand());
6456*9880d681SAndroid Build Coastguard Worker APInt Mask = cast<ConstantSDNode>(N0.getOperand(1))->getAPIntValue();
6457*9880d681SAndroid Build Coastguard Worker Mask = Mask.zext(VT.getSizeInBits());
6458*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
6459*9880d681SAndroid Build Coastguard Worker SDValue And = DAG.getNode(N0.getOpcode(), DL, VT,
6460*9880d681SAndroid Build Coastguard Worker ExtLoad, DAG.getConstant(Mask, DL, VT));
6461*9880d681SAndroid Build Coastguard Worker SDValue Trunc = DAG.getNode(ISD::TRUNCATE,
6462*9880d681SAndroid Build Coastguard Worker SDLoc(N0.getOperand(0)),
6463*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getValueType(), ExtLoad);
6464*9880d681SAndroid Build Coastguard Worker CombineTo(N, And);
6465*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getOperand(0).getNode(), Trunc, ExtLoad.getValue(1));
6466*9880d681SAndroid Build Coastguard Worker ExtendSetCCUses(SetCCs, Trunc, ExtLoad, DL,
6467*9880d681SAndroid Build Coastguard Worker ISD::ZERO_EXTEND);
6468*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
6469*9880d681SAndroid Build Coastguard Worker }
6470*9880d681SAndroid Build Coastguard Worker }
6471*9880d681SAndroid Build Coastguard Worker }
6472*9880d681SAndroid Build Coastguard Worker
6473*9880d681SAndroid Build Coastguard Worker // fold (zext (zextload x)) -> (zext (truncate (zextload x)))
6474*9880d681SAndroid Build Coastguard Worker // fold (zext ( extload x)) -> (zext (truncate (zextload x)))
6475*9880d681SAndroid Build Coastguard Worker if ((ISD::isZEXTLoad(N0.getNode()) || ISD::isEXTLoad(N0.getNode())) &&
6476*9880d681SAndroid Build Coastguard Worker ISD::isUNINDEXEDLoad(N0.getNode()) && N0.hasOneUse()) {
6477*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = cast<LoadSDNode>(N0);
6478*9880d681SAndroid Build Coastguard Worker EVT MemVT = LN0->getMemoryVT();
6479*9880d681SAndroid Build Coastguard Worker if ((!LegalOperations && !LN0->isVolatile()) ||
6480*9880d681SAndroid Build Coastguard Worker TLI.isLoadExtLegal(ISD::ZEXTLOAD, VT, MemVT)) {
6481*9880d681SAndroid Build Coastguard Worker SDValue ExtLoad = DAG.getExtLoad(ISD::ZEXTLOAD, SDLoc(N), VT,
6482*9880d681SAndroid Build Coastguard Worker LN0->getChain(),
6483*9880d681SAndroid Build Coastguard Worker LN0->getBasePtr(), MemVT,
6484*9880d681SAndroid Build Coastguard Worker LN0->getMemOperand());
6485*9880d681SAndroid Build Coastguard Worker CombineTo(N, ExtLoad);
6486*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(),
6487*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::TRUNCATE, SDLoc(N0), N0.getValueType(),
6488*9880d681SAndroid Build Coastguard Worker ExtLoad),
6489*9880d681SAndroid Build Coastguard Worker ExtLoad.getValue(1));
6490*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
6491*9880d681SAndroid Build Coastguard Worker }
6492*9880d681SAndroid Build Coastguard Worker }
6493*9880d681SAndroid Build Coastguard Worker
6494*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SETCC) {
6495*9880d681SAndroid Build Coastguard Worker // Only do this before legalize for now.
6496*9880d681SAndroid Build Coastguard Worker if (!LegalOperations && VT.isVector() &&
6497*9880d681SAndroid Build Coastguard Worker N0.getValueType().getVectorElementType() == MVT::i1) {
6498*9880d681SAndroid Build Coastguard Worker EVT N00VT = N0.getOperand(0).getValueType();
6499*9880d681SAndroid Build Coastguard Worker if (getSetCCResultType(N00VT) == N0.getValueType())
6500*9880d681SAndroid Build Coastguard Worker return SDValue();
6501*9880d681SAndroid Build Coastguard Worker
6502*9880d681SAndroid Build Coastguard Worker // We know that the # elements of the results is the same as the #
6503*9880d681SAndroid Build Coastguard Worker // elements of the compare (and the # elements of the compare result for
6504*9880d681SAndroid Build Coastguard Worker // that matter). Check to see that they are the same size. If so, we know
6505*9880d681SAndroid Build Coastguard Worker // that the element size of the sext'd result matches the element size of
6506*9880d681SAndroid Build Coastguard Worker // the compare operands.
6507*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
6508*9880d681SAndroid Build Coastguard Worker SDValue VecOnes = DAG.getConstant(1, DL, VT);
6509*9880d681SAndroid Build Coastguard Worker if (VT.getSizeInBits() == N00VT.getSizeInBits()) {
6510*9880d681SAndroid Build Coastguard Worker // zext(setcc) -> (and (vsetcc), (1, 1, ...) for vectors.
6511*9880d681SAndroid Build Coastguard Worker SDValue VSetCC = DAG.getNode(ISD::SETCC, DL, VT, N0.getOperand(0),
6512*9880d681SAndroid Build Coastguard Worker N0.getOperand(1), N0.getOperand(2));
6513*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, DL, VT, VSetCC, VecOnes);
6514*9880d681SAndroid Build Coastguard Worker }
6515*9880d681SAndroid Build Coastguard Worker
6516*9880d681SAndroid Build Coastguard Worker // If the desired elements are smaller or larger than the source
6517*9880d681SAndroid Build Coastguard Worker // elements we can use a matching integer vector type and then
6518*9880d681SAndroid Build Coastguard Worker // truncate/sign extend.
6519*9880d681SAndroid Build Coastguard Worker EVT MatchingElementType = EVT::getIntegerVT(
6520*9880d681SAndroid Build Coastguard Worker *DAG.getContext(), N00VT.getScalarType().getSizeInBits());
6521*9880d681SAndroid Build Coastguard Worker EVT MatchingVectorType = EVT::getVectorVT(
6522*9880d681SAndroid Build Coastguard Worker *DAG.getContext(), MatchingElementType, N00VT.getVectorNumElements());
6523*9880d681SAndroid Build Coastguard Worker SDValue VsetCC =
6524*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::SETCC, DL, MatchingVectorType, N0.getOperand(0),
6525*9880d681SAndroid Build Coastguard Worker N0.getOperand(1), N0.getOperand(2));
6526*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, DL, VT, DAG.getSExtOrTrunc(VsetCC, DL, VT),
6527*9880d681SAndroid Build Coastguard Worker VecOnes);
6528*9880d681SAndroid Build Coastguard Worker }
6529*9880d681SAndroid Build Coastguard Worker
6530*9880d681SAndroid Build Coastguard Worker // zext(setcc x,y,cc) -> select_cc x, y, 1, 0, cc
6531*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
6532*9880d681SAndroid Build Coastguard Worker if (SDValue SCC = SimplifySelectCC(
6533*9880d681SAndroid Build Coastguard Worker DL, N0.getOperand(0), N0.getOperand(1), DAG.getConstant(1, DL, VT),
6534*9880d681SAndroid Build Coastguard Worker DAG.getConstant(0, DL, VT),
6535*9880d681SAndroid Build Coastguard Worker cast<CondCodeSDNode>(N0.getOperand(2))->get(), true))
6536*9880d681SAndroid Build Coastguard Worker return SCC;
6537*9880d681SAndroid Build Coastguard Worker }
6538*9880d681SAndroid Build Coastguard Worker
6539*9880d681SAndroid Build Coastguard Worker // (zext (shl (zext x), cst)) -> (shl (zext x), cst)
6540*9880d681SAndroid Build Coastguard Worker if ((N0.getOpcode() == ISD::SHL || N0.getOpcode() == ISD::SRL) &&
6541*9880d681SAndroid Build Coastguard Worker isa<ConstantSDNode>(N0.getOperand(1)) &&
6542*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getOpcode() == ISD::ZERO_EXTEND &&
6543*9880d681SAndroid Build Coastguard Worker N0.hasOneUse()) {
6544*9880d681SAndroid Build Coastguard Worker SDValue ShAmt = N0.getOperand(1);
6545*9880d681SAndroid Build Coastguard Worker unsigned ShAmtVal = cast<ConstantSDNode>(ShAmt)->getZExtValue();
6546*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SHL) {
6547*9880d681SAndroid Build Coastguard Worker SDValue InnerZExt = N0.getOperand(0);
6548*9880d681SAndroid Build Coastguard Worker // If the original shl may be shifting out bits, do not perform this
6549*9880d681SAndroid Build Coastguard Worker // transformation.
6550*9880d681SAndroid Build Coastguard Worker unsigned KnownZeroBits = InnerZExt.getValueType().getSizeInBits() -
6551*9880d681SAndroid Build Coastguard Worker InnerZExt.getOperand(0).getValueType().getSizeInBits();
6552*9880d681SAndroid Build Coastguard Worker if (ShAmtVal > KnownZeroBits)
6553*9880d681SAndroid Build Coastguard Worker return SDValue();
6554*9880d681SAndroid Build Coastguard Worker }
6555*9880d681SAndroid Build Coastguard Worker
6556*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
6557*9880d681SAndroid Build Coastguard Worker
6558*9880d681SAndroid Build Coastguard Worker // Ensure that the shift amount is wide enough for the shifted value.
6559*9880d681SAndroid Build Coastguard Worker if (VT.getSizeInBits() >= 256)
6560*9880d681SAndroid Build Coastguard Worker ShAmt = DAG.getNode(ISD::ZERO_EXTEND, DL, MVT::i32, ShAmt);
6561*9880d681SAndroid Build Coastguard Worker
6562*9880d681SAndroid Build Coastguard Worker return DAG.getNode(N0.getOpcode(), DL, VT,
6563*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::ZERO_EXTEND, DL, VT, N0.getOperand(0)),
6564*9880d681SAndroid Build Coastguard Worker ShAmt);
6565*9880d681SAndroid Build Coastguard Worker }
6566*9880d681SAndroid Build Coastguard Worker
6567*9880d681SAndroid Build Coastguard Worker return SDValue();
6568*9880d681SAndroid Build Coastguard Worker }
6569*9880d681SAndroid Build Coastguard Worker
visitANY_EXTEND(SDNode * N)6570*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitANY_EXTEND(SDNode *N) {
6571*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
6572*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
6573*9880d681SAndroid Build Coastguard Worker
6574*9880d681SAndroid Build Coastguard Worker if (SDNode *Res = tryToFoldExtendOfConstant(N, TLI, DAG, LegalTypes,
6575*9880d681SAndroid Build Coastguard Worker LegalOperations))
6576*9880d681SAndroid Build Coastguard Worker return SDValue(Res, 0);
6577*9880d681SAndroid Build Coastguard Worker
6578*9880d681SAndroid Build Coastguard Worker // fold (aext (aext x)) -> (aext x)
6579*9880d681SAndroid Build Coastguard Worker // fold (aext (zext x)) -> (zext x)
6580*9880d681SAndroid Build Coastguard Worker // fold (aext (sext x)) -> (sext x)
6581*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::ANY_EXTEND ||
6582*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::ZERO_EXTEND ||
6583*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::SIGN_EXTEND)
6584*9880d681SAndroid Build Coastguard Worker return DAG.getNode(N0.getOpcode(), SDLoc(N), VT, N0.getOperand(0));
6585*9880d681SAndroid Build Coastguard Worker
6586*9880d681SAndroid Build Coastguard Worker // fold (aext (truncate (load x))) -> (aext (smaller load x))
6587*9880d681SAndroid Build Coastguard Worker // fold (aext (truncate (srl (load x), c))) -> (aext (small load (x+c/n)))
6588*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::TRUNCATE) {
6589*9880d681SAndroid Build Coastguard Worker if (SDValue NarrowLoad = ReduceLoadWidth(N0.getNode())) {
6590*9880d681SAndroid Build Coastguard Worker SDNode* oye = N0.getNode()->getOperand(0).getNode();
6591*9880d681SAndroid Build Coastguard Worker if (NarrowLoad.getNode() != N0.getNode()) {
6592*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(), NarrowLoad);
6593*9880d681SAndroid Build Coastguard Worker // CombineTo deleted the truncate, if needed, but not what's under it.
6594*9880d681SAndroid Build Coastguard Worker AddToWorklist(oye);
6595*9880d681SAndroid Build Coastguard Worker }
6596*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
6597*9880d681SAndroid Build Coastguard Worker }
6598*9880d681SAndroid Build Coastguard Worker }
6599*9880d681SAndroid Build Coastguard Worker
6600*9880d681SAndroid Build Coastguard Worker // fold (aext (truncate x))
6601*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::TRUNCATE) {
6602*9880d681SAndroid Build Coastguard Worker SDValue TruncOp = N0.getOperand(0);
6603*9880d681SAndroid Build Coastguard Worker if (TruncOp.getValueType() == VT)
6604*9880d681SAndroid Build Coastguard Worker return TruncOp; // x iff x size == zext size.
6605*9880d681SAndroid Build Coastguard Worker if (TruncOp.getValueType().bitsGT(VT))
6606*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TRUNCATE, SDLoc(N), VT, TruncOp);
6607*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::ANY_EXTEND, SDLoc(N), VT, TruncOp);
6608*9880d681SAndroid Build Coastguard Worker }
6609*9880d681SAndroid Build Coastguard Worker
6610*9880d681SAndroid Build Coastguard Worker // Fold (aext (and (trunc x), cst)) -> (and x, cst)
6611*9880d681SAndroid Build Coastguard Worker // if the trunc is not free.
6612*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::AND &&
6613*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getOpcode() == ISD::TRUNCATE &&
6614*9880d681SAndroid Build Coastguard Worker N0.getOperand(1).getOpcode() == ISD::Constant &&
6615*9880d681SAndroid Build Coastguard Worker !TLI.isTruncateFree(N0.getOperand(0).getOperand(0).getValueType(),
6616*9880d681SAndroid Build Coastguard Worker N0.getValueType())) {
6617*9880d681SAndroid Build Coastguard Worker SDValue X = N0.getOperand(0).getOperand(0);
6618*9880d681SAndroid Build Coastguard Worker if (X.getValueType().bitsLT(VT)) {
6619*9880d681SAndroid Build Coastguard Worker X = DAG.getNode(ISD::ANY_EXTEND, SDLoc(N), VT, X);
6620*9880d681SAndroid Build Coastguard Worker } else if (X.getValueType().bitsGT(VT)) {
6621*9880d681SAndroid Build Coastguard Worker X = DAG.getNode(ISD::TRUNCATE, SDLoc(N), VT, X);
6622*9880d681SAndroid Build Coastguard Worker }
6623*9880d681SAndroid Build Coastguard Worker APInt Mask = cast<ConstantSDNode>(N0.getOperand(1))->getAPIntValue();
6624*9880d681SAndroid Build Coastguard Worker Mask = Mask.zext(VT.getSizeInBits());
6625*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
6626*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, DL, VT,
6627*9880d681SAndroid Build Coastguard Worker X, DAG.getConstant(Mask, DL, VT));
6628*9880d681SAndroid Build Coastguard Worker }
6629*9880d681SAndroid Build Coastguard Worker
6630*9880d681SAndroid Build Coastguard Worker // fold (aext (load x)) -> (aext (truncate (extload x)))
6631*9880d681SAndroid Build Coastguard Worker // None of the supported targets knows how to perform load and any_ext
6632*9880d681SAndroid Build Coastguard Worker // on vectors in one instruction. We only perform this transformation on
6633*9880d681SAndroid Build Coastguard Worker // scalars.
6634*9880d681SAndroid Build Coastguard Worker if (ISD::isNON_EXTLoad(N0.getNode()) && !VT.isVector() &&
6635*9880d681SAndroid Build Coastguard Worker ISD::isUNINDEXEDLoad(N0.getNode()) &&
6636*9880d681SAndroid Build Coastguard Worker TLI.isLoadExtLegal(ISD::EXTLOAD, VT, N0.getValueType())) {
6637*9880d681SAndroid Build Coastguard Worker bool DoXform = true;
6638*9880d681SAndroid Build Coastguard Worker SmallVector<SDNode*, 4> SetCCs;
6639*9880d681SAndroid Build Coastguard Worker if (!N0.hasOneUse())
6640*9880d681SAndroid Build Coastguard Worker DoXform = ExtendUsesToFormExtLoad(N, N0, ISD::ANY_EXTEND, SetCCs, TLI);
6641*9880d681SAndroid Build Coastguard Worker if (DoXform) {
6642*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = cast<LoadSDNode>(N0);
6643*9880d681SAndroid Build Coastguard Worker SDValue ExtLoad = DAG.getExtLoad(ISD::EXTLOAD, SDLoc(N), VT,
6644*9880d681SAndroid Build Coastguard Worker LN0->getChain(),
6645*9880d681SAndroid Build Coastguard Worker LN0->getBasePtr(), N0.getValueType(),
6646*9880d681SAndroid Build Coastguard Worker LN0->getMemOperand());
6647*9880d681SAndroid Build Coastguard Worker CombineTo(N, ExtLoad);
6648*9880d681SAndroid Build Coastguard Worker SDValue Trunc = DAG.getNode(ISD::TRUNCATE, SDLoc(N0),
6649*9880d681SAndroid Build Coastguard Worker N0.getValueType(), ExtLoad);
6650*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(), Trunc, ExtLoad.getValue(1));
6651*9880d681SAndroid Build Coastguard Worker ExtendSetCCUses(SetCCs, Trunc, ExtLoad, SDLoc(N),
6652*9880d681SAndroid Build Coastguard Worker ISD::ANY_EXTEND);
6653*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
6654*9880d681SAndroid Build Coastguard Worker }
6655*9880d681SAndroid Build Coastguard Worker }
6656*9880d681SAndroid Build Coastguard Worker
6657*9880d681SAndroid Build Coastguard Worker // fold (aext (zextload x)) -> (aext (truncate (zextload x)))
6658*9880d681SAndroid Build Coastguard Worker // fold (aext (sextload x)) -> (aext (truncate (sextload x)))
6659*9880d681SAndroid Build Coastguard Worker // fold (aext ( extload x)) -> (aext (truncate (extload x)))
6660*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::LOAD &&
6661*9880d681SAndroid Build Coastguard Worker !ISD::isNON_EXTLoad(N0.getNode()) && ISD::isUNINDEXEDLoad(N0.getNode()) &&
6662*9880d681SAndroid Build Coastguard Worker N0.hasOneUse()) {
6663*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = cast<LoadSDNode>(N0);
6664*9880d681SAndroid Build Coastguard Worker ISD::LoadExtType ExtType = LN0->getExtensionType();
6665*9880d681SAndroid Build Coastguard Worker EVT MemVT = LN0->getMemoryVT();
6666*9880d681SAndroid Build Coastguard Worker if (!LegalOperations || TLI.isLoadExtLegal(ExtType, VT, MemVT)) {
6667*9880d681SAndroid Build Coastguard Worker SDValue ExtLoad = DAG.getExtLoad(ExtType, SDLoc(N),
6668*9880d681SAndroid Build Coastguard Worker VT, LN0->getChain(), LN0->getBasePtr(),
6669*9880d681SAndroid Build Coastguard Worker MemVT, LN0->getMemOperand());
6670*9880d681SAndroid Build Coastguard Worker CombineTo(N, ExtLoad);
6671*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(),
6672*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::TRUNCATE, SDLoc(N0),
6673*9880d681SAndroid Build Coastguard Worker N0.getValueType(), ExtLoad),
6674*9880d681SAndroid Build Coastguard Worker ExtLoad.getValue(1));
6675*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
6676*9880d681SAndroid Build Coastguard Worker }
6677*9880d681SAndroid Build Coastguard Worker }
6678*9880d681SAndroid Build Coastguard Worker
6679*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SETCC) {
6680*9880d681SAndroid Build Coastguard Worker // For vectors:
6681*9880d681SAndroid Build Coastguard Worker // aext(setcc) -> vsetcc
6682*9880d681SAndroid Build Coastguard Worker // aext(setcc) -> truncate(vsetcc)
6683*9880d681SAndroid Build Coastguard Worker // aext(setcc) -> aext(vsetcc)
6684*9880d681SAndroid Build Coastguard Worker // Only do this before legalize for now.
6685*9880d681SAndroid Build Coastguard Worker if (VT.isVector() && !LegalOperations) {
6686*9880d681SAndroid Build Coastguard Worker EVT N0VT = N0.getOperand(0).getValueType();
6687*9880d681SAndroid Build Coastguard Worker // We know that the # elements of the results is the same as the
6688*9880d681SAndroid Build Coastguard Worker // # elements of the compare (and the # elements of the compare result
6689*9880d681SAndroid Build Coastguard Worker // for that matter). Check to see that they are the same size. If so,
6690*9880d681SAndroid Build Coastguard Worker // we know that the element size of the sext'd result matches the
6691*9880d681SAndroid Build Coastguard Worker // element size of the compare operands.
6692*9880d681SAndroid Build Coastguard Worker if (VT.getSizeInBits() == N0VT.getSizeInBits())
6693*9880d681SAndroid Build Coastguard Worker return DAG.getSetCC(SDLoc(N), VT, N0.getOperand(0),
6694*9880d681SAndroid Build Coastguard Worker N0.getOperand(1),
6695*9880d681SAndroid Build Coastguard Worker cast<CondCodeSDNode>(N0.getOperand(2))->get());
6696*9880d681SAndroid Build Coastguard Worker // If the desired elements are smaller or larger than the source
6697*9880d681SAndroid Build Coastguard Worker // elements we can use a matching integer vector type and then
6698*9880d681SAndroid Build Coastguard Worker // truncate/any extend
6699*9880d681SAndroid Build Coastguard Worker else {
6700*9880d681SAndroid Build Coastguard Worker EVT MatchingVectorType = N0VT.changeVectorElementTypeToInteger();
6701*9880d681SAndroid Build Coastguard Worker SDValue VsetCC =
6702*9880d681SAndroid Build Coastguard Worker DAG.getSetCC(SDLoc(N), MatchingVectorType, N0.getOperand(0),
6703*9880d681SAndroid Build Coastguard Worker N0.getOperand(1),
6704*9880d681SAndroid Build Coastguard Worker cast<CondCodeSDNode>(N0.getOperand(2))->get());
6705*9880d681SAndroid Build Coastguard Worker return DAG.getAnyExtOrTrunc(VsetCC, SDLoc(N), VT);
6706*9880d681SAndroid Build Coastguard Worker }
6707*9880d681SAndroid Build Coastguard Worker }
6708*9880d681SAndroid Build Coastguard Worker
6709*9880d681SAndroid Build Coastguard Worker // aext(setcc x,y,cc) -> select_cc x, y, 1, 0, cc
6710*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
6711*9880d681SAndroid Build Coastguard Worker if (SDValue SCC = SimplifySelectCC(
6712*9880d681SAndroid Build Coastguard Worker DL, N0.getOperand(0), N0.getOperand(1), DAG.getConstant(1, DL, VT),
6713*9880d681SAndroid Build Coastguard Worker DAG.getConstant(0, DL, VT),
6714*9880d681SAndroid Build Coastguard Worker cast<CondCodeSDNode>(N0.getOperand(2))->get(), true))
6715*9880d681SAndroid Build Coastguard Worker return SCC;
6716*9880d681SAndroid Build Coastguard Worker }
6717*9880d681SAndroid Build Coastguard Worker
6718*9880d681SAndroid Build Coastguard Worker return SDValue();
6719*9880d681SAndroid Build Coastguard Worker }
6720*9880d681SAndroid Build Coastguard Worker
6721*9880d681SAndroid Build Coastguard Worker /// See if the specified operand can be simplified with the knowledge that only
6722*9880d681SAndroid Build Coastguard Worker /// the bits specified by Mask are used. If so, return the simpler operand,
6723*9880d681SAndroid Build Coastguard Worker /// otherwise return a null SDValue.
GetDemandedBits(SDValue V,const APInt & Mask)6724*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::GetDemandedBits(SDValue V, const APInt &Mask) {
6725*9880d681SAndroid Build Coastguard Worker switch (V.getOpcode()) {
6726*9880d681SAndroid Build Coastguard Worker default: break;
6727*9880d681SAndroid Build Coastguard Worker case ISD::Constant: {
6728*9880d681SAndroid Build Coastguard Worker const ConstantSDNode *CV = cast<ConstantSDNode>(V.getNode());
6729*9880d681SAndroid Build Coastguard Worker assert(CV && "Const value should be ConstSDNode.");
6730*9880d681SAndroid Build Coastguard Worker const APInt &CVal = CV->getAPIntValue();
6731*9880d681SAndroid Build Coastguard Worker APInt NewVal = CVal & Mask;
6732*9880d681SAndroid Build Coastguard Worker if (NewVal != CVal)
6733*9880d681SAndroid Build Coastguard Worker return DAG.getConstant(NewVal, SDLoc(V), V.getValueType());
6734*9880d681SAndroid Build Coastguard Worker break;
6735*9880d681SAndroid Build Coastguard Worker }
6736*9880d681SAndroid Build Coastguard Worker case ISD::OR:
6737*9880d681SAndroid Build Coastguard Worker case ISD::XOR:
6738*9880d681SAndroid Build Coastguard Worker // If the LHS or RHS don't contribute bits to the or, drop them.
6739*9880d681SAndroid Build Coastguard Worker if (DAG.MaskedValueIsZero(V.getOperand(0), Mask))
6740*9880d681SAndroid Build Coastguard Worker return V.getOperand(1);
6741*9880d681SAndroid Build Coastguard Worker if (DAG.MaskedValueIsZero(V.getOperand(1), Mask))
6742*9880d681SAndroid Build Coastguard Worker return V.getOperand(0);
6743*9880d681SAndroid Build Coastguard Worker break;
6744*9880d681SAndroid Build Coastguard Worker case ISD::SRL:
6745*9880d681SAndroid Build Coastguard Worker // Only look at single-use SRLs.
6746*9880d681SAndroid Build Coastguard Worker if (!V.getNode()->hasOneUse())
6747*9880d681SAndroid Build Coastguard Worker break;
6748*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *RHSC = getAsNonOpaqueConstant(V.getOperand(1))) {
6749*9880d681SAndroid Build Coastguard Worker // See if we can recursively simplify the LHS.
6750*9880d681SAndroid Build Coastguard Worker unsigned Amt = RHSC->getZExtValue();
6751*9880d681SAndroid Build Coastguard Worker
6752*9880d681SAndroid Build Coastguard Worker // Watch out for shift count overflow though.
6753*9880d681SAndroid Build Coastguard Worker if (Amt >= Mask.getBitWidth()) break;
6754*9880d681SAndroid Build Coastguard Worker APInt NewMask = Mask << Amt;
6755*9880d681SAndroid Build Coastguard Worker if (SDValue SimplifyLHS = GetDemandedBits(V.getOperand(0), NewMask))
6756*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SRL, SDLoc(V), V.getValueType(),
6757*9880d681SAndroid Build Coastguard Worker SimplifyLHS, V.getOperand(1));
6758*9880d681SAndroid Build Coastguard Worker }
6759*9880d681SAndroid Build Coastguard Worker }
6760*9880d681SAndroid Build Coastguard Worker return SDValue();
6761*9880d681SAndroid Build Coastguard Worker }
6762*9880d681SAndroid Build Coastguard Worker
6763*9880d681SAndroid Build Coastguard Worker /// If the result of a wider load is shifted to right of N bits and then
6764*9880d681SAndroid Build Coastguard Worker /// truncated to a narrower type and where N is a multiple of number of bits of
6765*9880d681SAndroid Build Coastguard Worker /// the narrower type, transform it to a narrower load from address + N / num of
6766*9880d681SAndroid Build Coastguard Worker /// bits of new type. If the result is to be extended, also fold the extension
6767*9880d681SAndroid Build Coastguard Worker /// to form a extending load.
ReduceLoadWidth(SDNode * N)6768*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::ReduceLoadWidth(SDNode *N) {
6769*9880d681SAndroid Build Coastguard Worker unsigned Opc = N->getOpcode();
6770*9880d681SAndroid Build Coastguard Worker
6771*9880d681SAndroid Build Coastguard Worker ISD::LoadExtType ExtType = ISD::NON_EXTLOAD;
6772*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
6773*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
6774*9880d681SAndroid Build Coastguard Worker EVT ExtVT = VT;
6775*9880d681SAndroid Build Coastguard Worker
6776*9880d681SAndroid Build Coastguard Worker // This transformation isn't valid for vector loads.
6777*9880d681SAndroid Build Coastguard Worker if (VT.isVector())
6778*9880d681SAndroid Build Coastguard Worker return SDValue();
6779*9880d681SAndroid Build Coastguard Worker
6780*9880d681SAndroid Build Coastguard Worker // Special case: SIGN_EXTEND_INREG is basically truncating to ExtVT then
6781*9880d681SAndroid Build Coastguard Worker // extended to VT.
6782*9880d681SAndroid Build Coastguard Worker if (Opc == ISD::SIGN_EXTEND_INREG) {
6783*9880d681SAndroid Build Coastguard Worker ExtType = ISD::SEXTLOAD;
6784*9880d681SAndroid Build Coastguard Worker ExtVT = cast<VTSDNode>(N->getOperand(1))->getVT();
6785*9880d681SAndroid Build Coastguard Worker } else if (Opc == ISD::SRL) {
6786*9880d681SAndroid Build Coastguard Worker // Another special-case: SRL is basically zero-extending a narrower value.
6787*9880d681SAndroid Build Coastguard Worker ExtType = ISD::ZEXTLOAD;
6788*9880d681SAndroid Build Coastguard Worker N0 = SDValue(N, 0);
6789*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N01 = dyn_cast<ConstantSDNode>(N0.getOperand(1));
6790*9880d681SAndroid Build Coastguard Worker if (!N01) return SDValue();
6791*9880d681SAndroid Build Coastguard Worker ExtVT = EVT::getIntegerVT(*DAG.getContext(),
6792*9880d681SAndroid Build Coastguard Worker VT.getSizeInBits() - N01->getZExtValue());
6793*9880d681SAndroid Build Coastguard Worker }
6794*9880d681SAndroid Build Coastguard Worker if (LegalOperations && !TLI.isLoadExtLegal(ExtType, VT, ExtVT))
6795*9880d681SAndroid Build Coastguard Worker return SDValue();
6796*9880d681SAndroid Build Coastguard Worker
6797*9880d681SAndroid Build Coastguard Worker unsigned EVTBits = ExtVT.getSizeInBits();
6798*9880d681SAndroid Build Coastguard Worker
6799*9880d681SAndroid Build Coastguard Worker // Do not generate loads of non-round integer types since these can
6800*9880d681SAndroid Build Coastguard Worker // be expensive (and would be wrong if the type is not byte sized).
6801*9880d681SAndroid Build Coastguard Worker if (!ExtVT.isRound())
6802*9880d681SAndroid Build Coastguard Worker return SDValue();
6803*9880d681SAndroid Build Coastguard Worker
6804*9880d681SAndroid Build Coastguard Worker unsigned ShAmt = 0;
6805*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SRL && N0.hasOneUse()) {
6806*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *N01 = dyn_cast<ConstantSDNode>(N0.getOperand(1))) {
6807*9880d681SAndroid Build Coastguard Worker ShAmt = N01->getZExtValue();
6808*9880d681SAndroid Build Coastguard Worker // Is the shift amount a multiple of size of VT?
6809*9880d681SAndroid Build Coastguard Worker if ((ShAmt & (EVTBits-1)) == 0) {
6810*9880d681SAndroid Build Coastguard Worker N0 = N0.getOperand(0);
6811*9880d681SAndroid Build Coastguard Worker // Is the load width a multiple of size of VT?
6812*9880d681SAndroid Build Coastguard Worker if ((N0.getValueType().getSizeInBits() & (EVTBits-1)) != 0)
6813*9880d681SAndroid Build Coastguard Worker return SDValue();
6814*9880d681SAndroid Build Coastguard Worker }
6815*9880d681SAndroid Build Coastguard Worker
6816*9880d681SAndroid Build Coastguard Worker // At this point, we must have a load or else we can't do the transform.
6817*9880d681SAndroid Build Coastguard Worker if (!isa<LoadSDNode>(N0)) return SDValue();
6818*9880d681SAndroid Build Coastguard Worker
6819*9880d681SAndroid Build Coastguard Worker // Because a SRL must be assumed to *need* to zero-extend the high bits
6820*9880d681SAndroid Build Coastguard Worker // (as opposed to anyext the high bits), we can't combine the zextload
6821*9880d681SAndroid Build Coastguard Worker // lowering of SRL and an sextload.
6822*9880d681SAndroid Build Coastguard Worker if (cast<LoadSDNode>(N0)->getExtensionType() == ISD::SEXTLOAD)
6823*9880d681SAndroid Build Coastguard Worker return SDValue();
6824*9880d681SAndroid Build Coastguard Worker
6825*9880d681SAndroid Build Coastguard Worker // If the shift amount is larger than the input type then we're not
6826*9880d681SAndroid Build Coastguard Worker // accessing any of the loaded bytes. If the load was a zextload/extload
6827*9880d681SAndroid Build Coastguard Worker // then the result of the shift+trunc is zero/undef (handled elsewhere).
6828*9880d681SAndroid Build Coastguard Worker if (ShAmt >= cast<LoadSDNode>(N0)->getMemoryVT().getSizeInBits())
6829*9880d681SAndroid Build Coastguard Worker return SDValue();
6830*9880d681SAndroid Build Coastguard Worker }
6831*9880d681SAndroid Build Coastguard Worker }
6832*9880d681SAndroid Build Coastguard Worker
6833*9880d681SAndroid Build Coastguard Worker // If the load is shifted left (and the result isn't shifted back right),
6834*9880d681SAndroid Build Coastguard Worker // we can fold the truncate through the shift.
6835*9880d681SAndroid Build Coastguard Worker unsigned ShLeftAmt = 0;
6836*9880d681SAndroid Build Coastguard Worker if (ShAmt == 0 && N0.getOpcode() == ISD::SHL && N0.hasOneUse() &&
6837*9880d681SAndroid Build Coastguard Worker ExtVT == VT && TLI.isNarrowingProfitable(N0.getValueType(), VT)) {
6838*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *N01 = dyn_cast<ConstantSDNode>(N0.getOperand(1))) {
6839*9880d681SAndroid Build Coastguard Worker ShLeftAmt = N01->getZExtValue();
6840*9880d681SAndroid Build Coastguard Worker N0 = N0.getOperand(0);
6841*9880d681SAndroid Build Coastguard Worker }
6842*9880d681SAndroid Build Coastguard Worker }
6843*9880d681SAndroid Build Coastguard Worker
6844*9880d681SAndroid Build Coastguard Worker // If we haven't found a load, we can't narrow it. Don't transform one with
6845*9880d681SAndroid Build Coastguard Worker // multiple uses, this would require adding a new load.
6846*9880d681SAndroid Build Coastguard Worker if (!isa<LoadSDNode>(N0) || !N0.hasOneUse())
6847*9880d681SAndroid Build Coastguard Worker return SDValue();
6848*9880d681SAndroid Build Coastguard Worker
6849*9880d681SAndroid Build Coastguard Worker // Don't change the width of a volatile load.
6850*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = cast<LoadSDNode>(N0);
6851*9880d681SAndroid Build Coastguard Worker if (LN0->isVolatile())
6852*9880d681SAndroid Build Coastguard Worker return SDValue();
6853*9880d681SAndroid Build Coastguard Worker
6854*9880d681SAndroid Build Coastguard Worker // Verify that we are actually reducing a load width here.
6855*9880d681SAndroid Build Coastguard Worker if (LN0->getMemoryVT().getSizeInBits() < EVTBits)
6856*9880d681SAndroid Build Coastguard Worker return SDValue();
6857*9880d681SAndroid Build Coastguard Worker
6858*9880d681SAndroid Build Coastguard Worker // For the transform to be legal, the load must produce only two values
6859*9880d681SAndroid Build Coastguard Worker // (the value loaded and the chain). Don't transform a pre-increment
6860*9880d681SAndroid Build Coastguard Worker // load, for example, which produces an extra value. Otherwise the
6861*9880d681SAndroid Build Coastguard Worker // transformation is not equivalent, and the downstream logic to replace
6862*9880d681SAndroid Build Coastguard Worker // uses gets things wrong.
6863*9880d681SAndroid Build Coastguard Worker if (LN0->getNumValues() > 2)
6864*9880d681SAndroid Build Coastguard Worker return SDValue();
6865*9880d681SAndroid Build Coastguard Worker
6866*9880d681SAndroid Build Coastguard Worker // If the load that we're shrinking is an extload and we're not just
6867*9880d681SAndroid Build Coastguard Worker // discarding the extension we can't simply shrink the load. Bail.
6868*9880d681SAndroid Build Coastguard Worker // TODO: It would be possible to merge the extensions in some cases.
6869*9880d681SAndroid Build Coastguard Worker if (LN0->getExtensionType() != ISD::NON_EXTLOAD &&
6870*9880d681SAndroid Build Coastguard Worker LN0->getMemoryVT().getSizeInBits() < ExtVT.getSizeInBits() + ShAmt)
6871*9880d681SAndroid Build Coastguard Worker return SDValue();
6872*9880d681SAndroid Build Coastguard Worker
6873*9880d681SAndroid Build Coastguard Worker if (!TLI.shouldReduceLoadWidth(LN0, ExtType, ExtVT))
6874*9880d681SAndroid Build Coastguard Worker return SDValue();
6875*9880d681SAndroid Build Coastguard Worker
6876*9880d681SAndroid Build Coastguard Worker EVT PtrType = N0.getOperand(1).getValueType();
6877*9880d681SAndroid Build Coastguard Worker
6878*9880d681SAndroid Build Coastguard Worker if (PtrType == MVT::Untyped || PtrType.isExtended())
6879*9880d681SAndroid Build Coastguard Worker // It's not possible to generate a constant of extended or untyped type.
6880*9880d681SAndroid Build Coastguard Worker return SDValue();
6881*9880d681SAndroid Build Coastguard Worker
6882*9880d681SAndroid Build Coastguard Worker // For big endian targets, we need to adjust the offset to the pointer to
6883*9880d681SAndroid Build Coastguard Worker // load the correct bytes.
6884*9880d681SAndroid Build Coastguard Worker if (DAG.getDataLayout().isBigEndian()) {
6885*9880d681SAndroid Build Coastguard Worker unsigned LVTStoreBits = LN0->getMemoryVT().getStoreSizeInBits();
6886*9880d681SAndroid Build Coastguard Worker unsigned EVTStoreBits = ExtVT.getStoreSizeInBits();
6887*9880d681SAndroid Build Coastguard Worker ShAmt = LVTStoreBits - EVTStoreBits - ShAmt;
6888*9880d681SAndroid Build Coastguard Worker }
6889*9880d681SAndroid Build Coastguard Worker
6890*9880d681SAndroid Build Coastguard Worker uint64_t PtrOff = ShAmt / 8;
6891*9880d681SAndroid Build Coastguard Worker unsigned NewAlign = MinAlign(LN0->getAlignment(), PtrOff);
6892*9880d681SAndroid Build Coastguard Worker SDLoc DL(LN0);
6893*9880d681SAndroid Build Coastguard Worker // The original load itself didn't wrap, so an offset within it doesn't.
6894*9880d681SAndroid Build Coastguard Worker SDNodeFlags Flags;
6895*9880d681SAndroid Build Coastguard Worker Flags.setNoUnsignedWrap(true);
6896*9880d681SAndroid Build Coastguard Worker SDValue NewPtr = DAG.getNode(ISD::ADD, DL,
6897*9880d681SAndroid Build Coastguard Worker PtrType, LN0->getBasePtr(),
6898*9880d681SAndroid Build Coastguard Worker DAG.getConstant(PtrOff, DL, PtrType),
6899*9880d681SAndroid Build Coastguard Worker &Flags);
6900*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewPtr.getNode());
6901*9880d681SAndroid Build Coastguard Worker
6902*9880d681SAndroid Build Coastguard Worker SDValue Load;
6903*9880d681SAndroid Build Coastguard Worker if (ExtType == ISD::NON_EXTLOAD)
6904*9880d681SAndroid Build Coastguard Worker Load = DAG.getLoad(VT, SDLoc(N0), LN0->getChain(), NewPtr,
6905*9880d681SAndroid Build Coastguard Worker LN0->getPointerInfo().getWithOffset(PtrOff),
6906*9880d681SAndroid Build Coastguard Worker LN0->isVolatile(), LN0->isNonTemporal(),
6907*9880d681SAndroid Build Coastguard Worker LN0->isInvariant(), NewAlign, LN0->getAAInfo());
6908*9880d681SAndroid Build Coastguard Worker else
6909*9880d681SAndroid Build Coastguard Worker Load = DAG.getExtLoad(ExtType, SDLoc(N0), VT, LN0->getChain(),NewPtr,
6910*9880d681SAndroid Build Coastguard Worker LN0->getPointerInfo().getWithOffset(PtrOff),
6911*9880d681SAndroid Build Coastguard Worker ExtVT, LN0->isVolatile(), LN0->isNonTemporal(),
6912*9880d681SAndroid Build Coastguard Worker LN0->isInvariant(), NewAlign, LN0->getAAInfo());
6913*9880d681SAndroid Build Coastguard Worker
6914*9880d681SAndroid Build Coastguard Worker // Replace the old load's chain with the new load's chain.
6915*9880d681SAndroid Build Coastguard Worker WorklistRemover DeadNodes(*this);
6916*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(N0.getValue(1), Load.getValue(1));
6917*9880d681SAndroid Build Coastguard Worker
6918*9880d681SAndroid Build Coastguard Worker // Shift the result left, if we've swallowed a left shift.
6919*9880d681SAndroid Build Coastguard Worker SDValue Result = Load;
6920*9880d681SAndroid Build Coastguard Worker if (ShLeftAmt != 0) {
6921*9880d681SAndroid Build Coastguard Worker EVT ShImmTy = getShiftAmountTy(Result.getValueType());
6922*9880d681SAndroid Build Coastguard Worker if (!isUIntN(ShImmTy.getSizeInBits(), ShLeftAmt))
6923*9880d681SAndroid Build Coastguard Worker ShImmTy = VT;
6924*9880d681SAndroid Build Coastguard Worker // If the shift amount is as large as the result size (but, presumably,
6925*9880d681SAndroid Build Coastguard Worker // no larger than the source) then the useful bits of the result are
6926*9880d681SAndroid Build Coastguard Worker // zero; we can't simply return the shortened shift, because the result
6927*9880d681SAndroid Build Coastguard Worker // of that operation is undefined.
6928*9880d681SAndroid Build Coastguard Worker SDLoc DL(N0);
6929*9880d681SAndroid Build Coastguard Worker if (ShLeftAmt >= VT.getSizeInBits())
6930*9880d681SAndroid Build Coastguard Worker Result = DAG.getConstant(0, DL, VT);
6931*9880d681SAndroid Build Coastguard Worker else
6932*9880d681SAndroid Build Coastguard Worker Result = DAG.getNode(ISD::SHL, DL, VT,
6933*9880d681SAndroid Build Coastguard Worker Result, DAG.getConstant(ShLeftAmt, DL, ShImmTy));
6934*9880d681SAndroid Build Coastguard Worker }
6935*9880d681SAndroid Build Coastguard Worker
6936*9880d681SAndroid Build Coastguard Worker // Return the new loaded value.
6937*9880d681SAndroid Build Coastguard Worker return Result;
6938*9880d681SAndroid Build Coastguard Worker }
6939*9880d681SAndroid Build Coastguard Worker
visitSIGN_EXTEND_INREG(SDNode * N)6940*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSIGN_EXTEND_INREG(SDNode *N) {
6941*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
6942*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
6943*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
6944*9880d681SAndroid Build Coastguard Worker EVT EVT = cast<VTSDNode>(N1)->getVT();
6945*9880d681SAndroid Build Coastguard Worker unsigned VTBits = VT.getScalarType().getSizeInBits();
6946*9880d681SAndroid Build Coastguard Worker unsigned EVTBits = EVT.getScalarType().getSizeInBits();
6947*9880d681SAndroid Build Coastguard Worker
6948*9880d681SAndroid Build Coastguard Worker if (N0.isUndef())
6949*9880d681SAndroid Build Coastguard Worker return DAG.getUNDEF(VT);
6950*9880d681SAndroid Build Coastguard Worker
6951*9880d681SAndroid Build Coastguard Worker // fold (sext_in_reg c1) -> c1
6952*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N0))
6953*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SIGN_EXTEND_INREG, SDLoc(N), VT, N0, N1);
6954*9880d681SAndroid Build Coastguard Worker
6955*9880d681SAndroid Build Coastguard Worker // If the input is already sign extended, just drop the extension.
6956*9880d681SAndroid Build Coastguard Worker if (DAG.ComputeNumSignBits(N0) >= VTBits-EVTBits+1)
6957*9880d681SAndroid Build Coastguard Worker return N0;
6958*9880d681SAndroid Build Coastguard Worker
6959*9880d681SAndroid Build Coastguard Worker // fold (sext_in_reg (sext_in_reg x, VT2), VT1) -> (sext_in_reg x, minVT) pt2
6960*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SIGN_EXTEND_INREG &&
6961*9880d681SAndroid Build Coastguard Worker EVT.bitsLT(cast<VTSDNode>(N0.getOperand(1))->getVT()))
6962*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SIGN_EXTEND_INREG, SDLoc(N), VT,
6963*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N1);
6964*9880d681SAndroid Build Coastguard Worker
6965*9880d681SAndroid Build Coastguard Worker // fold (sext_in_reg (sext x)) -> (sext x)
6966*9880d681SAndroid Build Coastguard Worker // fold (sext_in_reg (aext x)) -> (sext x)
6967*9880d681SAndroid Build Coastguard Worker // if x is small enough.
6968*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SIGN_EXTEND || N0.getOpcode() == ISD::ANY_EXTEND) {
6969*9880d681SAndroid Build Coastguard Worker SDValue N00 = N0.getOperand(0);
6970*9880d681SAndroid Build Coastguard Worker if (N00.getValueType().getScalarType().getSizeInBits() <= EVTBits &&
6971*9880d681SAndroid Build Coastguard Worker (!LegalOperations || TLI.isOperationLegal(ISD::SIGN_EXTEND, VT)))
6972*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SIGN_EXTEND, SDLoc(N), VT, N00, N1);
6973*9880d681SAndroid Build Coastguard Worker }
6974*9880d681SAndroid Build Coastguard Worker
6975*9880d681SAndroid Build Coastguard Worker // fold (sext_in_reg x) -> (zext_in_reg x) if the sign bit is known zero.
6976*9880d681SAndroid Build Coastguard Worker if (DAG.MaskedValueIsZero(N0, APInt::getBitsSet(VTBits, EVTBits-1, EVTBits)))
6977*9880d681SAndroid Build Coastguard Worker return DAG.getZeroExtendInReg(N0, SDLoc(N), EVT);
6978*9880d681SAndroid Build Coastguard Worker
6979*9880d681SAndroid Build Coastguard Worker // fold operands of sext_in_reg based on knowledge that the top bits are not
6980*9880d681SAndroid Build Coastguard Worker // demanded.
6981*9880d681SAndroid Build Coastguard Worker if (SimplifyDemandedBits(SDValue(N, 0)))
6982*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0);
6983*9880d681SAndroid Build Coastguard Worker
6984*9880d681SAndroid Build Coastguard Worker // fold (sext_in_reg (load x)) -> (smaller sextload x)
6985*9880d681SAndroid Build Coastguard Worker // fold (sext_in_reg (srl (load x), c)) -> (smaller sextload (x+c/evtbits))
6986*9880d681SAndroid Build Coastguard Worker if (SDValue NarrowLoad = ReduceLoadWidth(N))
6987*9880d681SAndroid Build Coastguard Worker return NarrowLoad;
6988*9880d681SAndroid Build Coastguard Worker
6989*9880d681SAndroid Build Coastguard Worker // fold (sext_in_reg (srl X, 24), i8) -> (sra X, 24)
6990*9880d681SAndroid Build Coastguard Worker // fold (sext_in_reg (srl X, 23), i8) -> (sra X, 23) iff possible.
6991*9880d681SAndroid Build Coastguard Worker // We already fold "(sext_in_reg (srl X, 25), i8) -> srl X, 25" above.
6992*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SRL) {
6993*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *ShAmt = dyn_cast<ConstantSDNode>(N0.getOperand(1)))
6994*9880d681SAndroid Build Coastguard Worker if (ShAmt->getZExtValue()+EVTBits <= VTBits) {
6995*9880d681SAndroid Build Coastguard Worker // We can turn this into an SRA iff the input to the SRL is already sign
6996*9880d681SAndroid Build Coastguard Worker // extended enough.
6997*9880d681SAndroid Build Coastguard Worker unsigned InSignBits = DAG.ComputeNumSignBits(N0.getOperand(0));
6998*9880d681SAndroid Build Coastguard Worker if (VTBits-(ShAmt->getZExtValue()+EVTBits) < InSignBits)
6999*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SRA, SDLoc(N), VT,
7000*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N0.getOperand(1));
7001*9880d681SAndroid Build Coastguard Worker }
7002*9880d681SAndroid Build Coastguard Worker }
7003*9880d681SAndroid Build Coastguard Worker
7004*9880d681SAndroid Build Coastguard Worker // fold (sext_inreg (extload x)) -> (sextload x)
7005*9880d681SAndroid Build Coastguard Worker if (ISD::isEXTLoad(N0.getNode()) &&
7006*9880d681SAndroid Build Coastguard Worker ISD::isUNINDEXEDLoad(N0.getNode()) &&
7007*9880d681SAndroid Build Coastguard Worker EVT == cast<LoadSDNode>(N0)->getMemoryVT() &&
7008*9880d681SAndroid Build Coastguard Worker ((!LegalOperations && !cast<LoadSDNode>(N0)->isVolatile()) ||
7009*9880d681SAndroid Build Coastguard Worker TLI.isLoadExtLegal(ISD::SEXTLOAD, VT, EVT))) {
7010*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = cast<LoadSDNode>(N0);
7011*9880d681SAndroid Build Coastguard Worker SDValue ExtLoad = DAG.getExtLoad(ISD::SEXTLOAD, SDLoc(N), VT,
7012*9880d681SAndroid Build Coastguard Worker LN0->getChain(),
7013*9880d681SAndroid Build Coastguard Worker LN0->getBasePtr(), EVT,
7014*9880d681SAndroid Build Coastguard Worker LN0->getMemOperand());
7015*9880d681SAndroid Build Coastguard Worker CombineTo(N, ExtLoad);
7016*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(), ExtLoad, ExtLoad.getValue(1));
7017*9880d681SAndroid Build Coastguard Worker AddToWorklist(ExtLoad.getNode());
7018*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
7019*9880d681SAndroid Build Coastguard Worker }
7020*9880d681SAndroid Build Coastguard Worker // fold (sext_inreg (zextload x)) -> (sextload x) iff load has one use
7021*9880d681SAndroid Build Coastguard Worker if (ISD::isZEXTLoad(N0.getNode()) && ISD::isUNINDEXEDLoad(N0.getNode()) &&
7022*9880d681SAndroid Build Coastguard Worker N0.hasOneUse() &&
7023*9880d681SAndroid Build Coastguard Worker EVT == cast<LoadSDNode>(N0)->getMemoryVT() &&
7024*9880d681SAndroid Build Coastguard Worker ((!LegalOperations && !cast<LoadSDNode>(N0)->isVolatile()) ||
7025*9880d681SAndroid Build Coastguard Worker TLI.isLoadExtLegal(ISD::SEXTLOAD, VT, EVT))) {
7026*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = cast<LoadSDNode>(N0);
7027*9880d681SAndroid Build Coastguard Worker SDValue ExtLoad = DAG.getExtLoad(ISD::SEXTLOAD, SDLoc(N), VT,
7028*9880d681SAndroid Build Coastguard Worker LN0->getChain(),
7029*9880d681SAndroid Build Coastguard Worker LN0->getBasePtr(), EVT,
7030*9880d681SAndroid Build Coastguard Worker LN0->getMemOperand());
7031*9880d681SAndroid Build Coastguard Worker CombineTo(N, ExtLoad);
7032*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(), ExtLoad, ExtLoad.getValue(1));
7033*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
7034*9880d681SAndroid Build Coastguard Worker }
7035*9880d681SAndroid Build Coastguard Worker
7036*9880d681SAndroid Build Coastguard Worker // Form (sext_inreg (bswap >> 16)) or (sext_inreg (rotl (bswap) 16))
7037*9880d681SAndroid Build Coastguard Worker if (EVTBits <= 16 && N0.getOpcode() == ISD::OR) {
7038*9880d681SAndroid Build Coastguard Worker if (SDValue BSwap = MatchBSwapHWordLow(N0.getNode(), N0.getOperand(0),
7039*9880d681SAndroid Build Coastguard Worker N0.getOperand(1), false))
7040*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SIGN_EXTEND_INREG, SDLoc(N), VT,
7041*9880d681SAndroid Build Coastguard Worker BSwap, N1);
7042*9880d681SAndroid Build Coastguard Worker }
7043*9880d681SAndroid Build Coastguard Worker
7044*9880d681SAndroid Build Coastguard Worker return SDValue();
7045*9880d681SAndroid Build Coastguard Worker }
7046*9880d681SAndroid Build Coastguard Worker
visitSIGN_EXTEND_VECTOR_INREG(SDNode * N)7047*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSIGN_EXTEND_VECTOR_INREG(SDNode *N) {
7048*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
7049*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
7050*9880d681SAndroid Build Coastguard Worker
7051*9880d681SAndroid Build Coastguard Worker if (N0.isUndef())
7052*9880d681SAndroid Build Coastguard Worker return DAG.getUNDEF(VT);
7053*9880d681SAndroid Build Coastguard Worker
7054*9880d681SAndroid Build Coastguard Worker if (SDNode *Res = tryToFoldExtendOfConstant(N, TLI, DAG, LegalTypes,
7055*9880d681SAndroid Build Coastguard Worker LegalOperations))
7056*9880d681SAndroid Build Coastguard Worker return SDValue(Res, 0);
7057*9880d681SAndroid Build Coastguard Worker
7058*9880d681SAndroid Build Coastguard Worker return SDValue();
7059*9880d681SAndroid Build Coastguard Worker }
7060*9880d681SAndroid Build Coastguard Worker
visitZERO_EXTEND_VECTOR_INREG(SDNode * N)7061*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitZERO_EXTEND_VECTOR_INREG(SDNode *N) {
7062*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
7063*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
7064*9880d681SAndroid Build Coastguard Worker
7065*9880d681SAndroid Build Coastguard Worker if (N0.isUndef())
7066*9880d681SAndroid Build Coastguard Worker return DAG.getUNDEF(VT);
7067*9880d681SAndroid Build Coastguard Worker
7068*9880d681SAndroid Build Coastguard Worker if (SDNode *Res = tryToFoldExtendOfConstant(N, TLI, DAG, LegalTypes,
7069*9880d681SAndroid Build Coastguard Worker LegalOperations))
7070*9880d681SAndroid Build Coastguard Worker return SDValue(Res, 0);
7071*9880d681SAndroid Build Coastguard Worker
7072*9880d681SAndroid Build Coastguard Worker return SDValue();
7073*9880d681SAndroid Build Coastguard Worker }
7074*9880d681SAndroid Build Coastguard Worker
visitTRUNCATE(SDNode * N)7075*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitTRUNCATE(SDNode *N) {
7076*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
7077*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
7078*9880d681SAndroid Build Coastguard Worker bool isLE = DAG.getDataLayout().isLittleEndian();
7079*9880d681SAndroid Build Coastguard Worker
7080*9880d681SAndroid Build Coastguard Worker // noop truncate
7081*9880d681SAndroid Build Coastguard Worker if (N0.getValueType() == N->getValueType(0))
7082*9880d681SAndroid Build Coastguard Worker return N0;
7083*9880d681SAndroid Build Coastguard Worker // fold (truncate c1) -> c1
7084*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N0))
7085*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TRUNCATE, SDLoc(N), VT, N0);
7086*9880d681SAndroid Build Coastguard Worker // fold (truncate (truncate x)) -> (truncate x)
7087*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::TRUNCATE)
7088*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TRUNCATE, SDLoc(N), VT, N0.getOperand(0));
7089*9880d681SAndroid Build Coastguard Worker // fold (truncate (ext x)) -> (ext x) or (truncate x) or x
7090*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::ZERO_EXTEND ||
7091*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::SIGN_EXTEND ||
7092*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::ANY_EXTEND) {
7093*9880d681SAndroid Build Coastguard Worker // if the source is smaller than the dest, we still need an extend.
7094*9880d681SAndroid Build Coastguard Worker if (N0.getOperand(0).getValueType().bitsLT(VT))
7095*9880d681SAndroid Build Coastguard Worker return DAG.getNode(N0.getOpcode(), SDLoc(N), VT, N0.getOperand(0));
7096*9880d681SAndroid Build Coastguard Worker // if the source is larger than the dest, than we just need the truncate.
7097*9880d681SAndroid Build Coastguard Worker if (N0.getOperand(0).getValueType().bitsGT(VT))
7098*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TRUNCATE, SDLoc(N), VT, N0.getOperand(0));
7099*9880d681SAndroid Build Coastguard Worker // if the source and dest are the same type, we can drop both the extend
7100*9880d681SAndroid Build Coastguard Worker // and the truncate.
7101*9880d681SAndroid Build Coastguard Worker return N0.getOperand(0);
7102*9880d681SAndroid Build Coastguard Worker }
7103*9880d681SAndroid Build Coastguard Worker
7104*9880d681SAndroid Build Coastguard Worker // Fold extract-and-trunc into a narrow extract. For example:
7105*9880d681SAndroid Build Coastguard Worker // i64 x = EXTRACT_VECTOR_ELT(v2i64 val, i32 1)
7106*9880d681SAndroid Build Coastguard Worker // i32 y = TRUNCATE(i64 x)
7107*9880d681SAndroid Build Coastguard Worker // -- becomes --
7108*9880d681SAndroid Build Coastguard Worker // v16i8 b = BITCAST (v2i64 val)
7109*9880d681SAndroid Build Coastguard Worker // i8 x = EXTRACT_VECTOR_ELT(v16i8 b, i32 8)
7110*9880d681SAndroid Build Coastguard Worker //
7111*9880d681SAndroid Build Coastguard Worker // Note: We only run this optimization after type legalization (which often
7112*9880d681SAndroid Build Coastguard Worker // creates this pattern) and before operation legalization after which
7113*9880d681SAndroid Build Coastguard Worker // we need to be more careful about the vector instructions that we generate.
7114*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::EXTRACT_VECTOR_ELT &&
7115*9880d681SAndroid Build Coastguard Worker LegalTypes && !LegalOperations && N0->hasOneUse() && VT != MVT::i1) {
7116*9880d681SAndroid Build Coastguard Worker
7117*9880d681SAndroid Build Coastguard Worker EVT VecTy = N0.getOperand(0).getValueType();
7118*9880d681SAndroid Build Coastguard Worker EVT ExTy = N0.getValueType();
7119*9880d681SAndroid Build Coastguard Worker EVT TrTy = N->getValueType(0);
7120*9880d681SAndroid Build Coastguard Worker
7121*9880d681SAndroid Build Coastguard Worker unsigned NumElem = VecTy.getVectorNumElements();
7122*9880d681SAndroid Build Coastguard Worker unsigned SizeRatio = ExTy.getSizeInBits()/TrTy.getSizeInBits();
7123*9880d681SAndroid Build Coastguard Worker
7124*9880d681SAndroid Build Coastguard Worker EVT NVT = EVT::getVectorVT(*DAG.getContext(), TrTy, SizeRatio * NumElem);
7125*9880d681SAndroid Build Coastguard Worker assert(NVT.getSizeInBits() == VecTy.getSizeInBits() && "Invalid Size");
7126*9880d681SAndroid Build Coastguard Worker
7127*9880d681SAndroid Build Coastguard Worker SDValue EltNo = N0->getOperand(1);
7128*9880d681SAndroid Build Coastguard Worker if (isa<ConstantSDNode>(EltNo) && isTypeLegal(NVT)) {
7129*9880d681SAndroid Build Coastguard Worker int Elt = cast<ConstantSDNode>(EltNo)->getZExtValue();
7130*9880d681SAndroid Build Coastguard Worker EVT IndexTy = TLI.getVectorIdxTy(DAG.getDataLayout());
7131*9880d681SAndroid Build Coastguard Worker int Index = isLE ? (Elt*SizeRatio) : (Elt*SizeRatio + (SizeRatio-1));
7132*9880d681SAndroid Build Coastguard Worker
7133*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
7134*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::EXTRACT_VECTOR_ELT, DL, TrTy,
7135*9880d681SAndroid Build Coastguard Worker DAG.getBitcast(NVT, N0.getOperand(0)),
7136*9880d681SAndroid Build Coastguard Worker DAG.getConstant(Index, DL, IndexTy));
7137*9880d681SAndroid Build Coastguard Worker }
7138*9880d681SAndroid Build Coastguard Worker }
7139*9880d681SAndroid Build Coastguard Worker
7140*9880d681SAndroid Build Coastguard Worker // trunc (select c, a, b) -> select c, (trunc a), (trunc b)
7141*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SELECT) {
7142*9880d681SAndroid Build Coastguard Worker EVT SrcVT = N0.getValueType();
7143*9880d681SAndroid Build Coastguard Worker if ((!LegalOperations || TLI.isOperationLegal(ISD::SELECT, SrcVT)) &&
7144*9880d681SAndroid Build Coastguard Worker TLI.isTruncateFree(SrcVT, VT)) {
7145*9880d681SAndroid Build Coastguard Worker SDLoc SL(N0);
7146*9880d681SAndroid Build Coastguard Worker SDValue Cond = N0.getOperand(0);
7147*9880d681SAndroid Build Coastguard Worker SDValue TruncOp0 = DAG.getNode(ISD::TRUNCATE, SL, VT, N0.getOperand(1));
7148*9880d681SAndroid Build Coastguard Worker SDValue TruncOp1 = DAG.getNode(ISD::TRUNCATE, SL, VT, N0.getOperand(2));
7149*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SELECT, SDLoc(N), VT, Cond, TruncOp0, TruncOp1);
7150*9880d681SAndroid Build Coastguard Worker }
7151*9880d681SAndroid Build Coastguard Worker }
7152*9880d681SAndroid Build Coastguard Worker
7153*9880d681SAndroid Build Coastguard Worker // trunc (shl x, K) -> shl (trunc x), K => K < vt.size / 2
7154*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SHL && N0.hasOneUse() &&
7155*9880d681SAndroid Build Coastguard Worker (!LegalOperations || TLI.isOperationLegalOrCustom(ISD::SHL, VT)) &&
7156*9880d681SAndroid Build Coastguard Worker TLI.isTypeDesirableForOp(ISD::SHL, VT)) {
7157*9880d681SAndroid Build Coastguard Worker if (const ConstantSDNode *CAmt = isConstOrConstSplat(N0.getOperand(1))) {
7158*9880d681SAndroid Build Coastguard Worker uint64_t Amt = CAmt->getZExtValue();
7159*9880d681SAndroid Build Coastguard Worker unsigned Size = VT.getSizeInBits();
7160*9880d681SAndroid Build Coastguard Worker
7161*9880d681SAndroid Build Coastguard Worker if (Amt < Size / 2) {
7162*9880d681SAndroid Build Coastguard Worker SDLoc SL(N);
7163*9880d681SAndroid Build Coastguard Worker EVT AmtVT = TLI.getShiftAmountTy(VT, DAG.getDataLayout());
7164*9880d681SAndroid Build Coastguard Worker
7165*9880d681SAndroid Build Coastguard Worker SDValue Trunc = DAG.getNode(ISD::TRUNCATE, SL, VT, N0.getOperand(0));
7166*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SHL, SL, VT, Trunc,
7167*9880d681SAndroid Build Coastguard Worker DAG.getConstant(Amt, SL, AmtVT));
7168*9880d681SAndroid Build Coastguard Worker }
7169*9880d681SAndroid Build Coastguard Worker }
7170*9880d681SAndroid Build Coastguard Worker }
7171*9880d681SAndroid Build Coastguard Worker
7172*9880d681SAndroid Build Coastguard Worker // Fold a series of buildvector, bitcast, and truncate if possible.
7173*9880d681SAndroid Build Coastguard Worker // For example fold
7174*9880d681SAndroid Build Coastguard Worker // (2xi32 trunc (bitcast ((4xi32)buildvector x, x, y, y) 2xi64)) to
7175*9880d681SAndroid Build Coastguard Worker // (2xi32 (buildvector x, y)).
7176*9880d681SAndroid Build Coastguard Worker if (Level == AfterLegalizeVectorOps && VT.isVector() &&
7177*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::BITCAST && N0.hasOneUse() &&
7178*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getOpcode() == ISD::BUILD_VECTOR &&
7179*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).hasOneUse()) {
7180*9880d681SAndroid Build Coastguard Worker
7181*9880d681SAndroid Build Coastguard Worker SDValue BuildVect = N0.getOperand(0);
7182*9880d681SAndroid Build Coastguard Worker EVT BuildVectEltTy = BuildVect.getValueType().getVectorElementType();
7183*9880d681SAndroid Build Coastguard Worker EVT TruncVecEltTy = VT.getVectorElementType();
7184*9880d681SAndroid Build Coastguard Worker
7185*9880d681SAndroid Build Coastguard Worker // Check that the element types match.
7186*9880d681SAndroid Build Coastguard Worker if (BuildVectEltTy == TruncVecEltTy) {
7187*9880d681SAndroid Build Coastguard Worker // Now we only need to compute the offset of the truncated elements.
7188*9880d681SAndroid Build Coastguard Worker unsigned BuildVecNumElts = BuildVect.getNumOperands();
7189*9880d681SAndroid Build Coastguard Worker unsigned TruncVecNumElts = VT.getVectorNumElements();
7190*9880d681SAndroid Build Coastguard Worker unsigned TruncEltOffset = BuildVecNumElts / TruncVecNumElts;
7191*9880d681SAndroid Build Coastguard Worker
7192*9880d681SAndroid Build Coastguard Worker assert((BuildVecNumElts % TruncVecNumElts) == 0 &&
7193*9880d681SAndroid Build Coastguard Worker "Invalid number of elements");
7194*9880d681SAndroid Build Coastguard Worker
7195*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Opnds;
7196*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0, e = BuildVecNumElts; i != e; i += TruncEltOffset)
7197*9880d681SAndroid Build Coastguard Worker Opnds.push_back(BuildVect.getOperand(i));
7198*9880d681SAndroid Build Coastguard Worker
7199*9880d681SAndroid Build Coastguard Worker return DAG.getBuildVector(VT, SDLoc(N), Opnds);
7200*9880d681SAndroid Build Coastguard Worker }
7201*9880d681SAndroid Build Coastguard Worker }
7202*9880d681SAndroid Build Coastguard Worker
7203*9880d681SAndroid Build Coastguard Worker // See if we can simplify the input to this truncate through knowledge that
7204*9880d681SAndroid Build Coastguard Worker // only the low bits are being used.
7205*9880d681SAndroid Build Coastguard Worker // For example "trunc (or (shl x, 8), y)" // -> trunc y
7206*9880d681SAndroid Build Coastguard Worker // Currently we only perform this optimization on scalars because vectors
7207*9880d681SAndroid Build Coastguard Worker // may have different active low bits.
7208*9880d681SAndroid Build Coastguard Worker if (!VT.isVector()) {
7209*9880d681SAndroid Build Coastguard Worker if (SDValue Shorter =
7210*9880d681SAndroid Build Coastguard Worker GetDemandedBits(N0, APInt::getLowBitsSet(N0.getValueSizeInBits(),
7211*9880d681SAndroid Build Coastguard Worker VT.getSizeInBits())))
7212*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TRUNCATE, SDLoc(N), VT, Shorter);
7213*9880d681SAndroid Build Coastguard Worker }
7214*9880d681SAndroid Build Coastguard Worker // fold (truncate (load x)) -> (smaller load x)
7215*9880d681SAndroid Build Coastguard Worker // fold (truncate (srl (load x), c)) -> (smaller load (x+c/evtbits))
7216*9880d681SAndroid Build Coastguard Worker if (!LegalTypes || TLI.isTypeDesirableForOp(N0.getOpcode(), VT)) {
7217*9880d681SAndroid Build Coastguard Worker if (SDValue Reduced = ReduceLoadWidth(N))
7218*9880d681SAndroid Build Coastguard Worker return Reduced;
7219*9880d681SAndroid Build Coastguard Worker
7220*9880d681SAndroid Build Coastguard Worker // Handle the case where the load remains an extending load even
7221*9880d681SAndroid Build Coastguard Worker // after truncation.
7222*9880d681SAndroid Build Coastguard Worker if (N0.hasOneUse() && ISD::isUNINDEXEDLoad(N0.getNode())) {
7223*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = cast<LoadSDNode>(N0);
7224*9880d681SAndroid Build Coastguard Worker if (!LN0->isVolatile() &&
7225*9880d681SAndroid Build Coastguard Worker LN0->getMemoryVT().getStoreSizeInBits() < VT.getSizeInBits()) {
7226*9880d681SAndroid Build Coastguard Worker SDValue NewLoad = DAG.getExtLoad(LN0->getExtensionType(), SDLoc(LN0),
7227*9880d681SAndroid Build Coastguard Worker VT, LN0->getChain(), LN0->getBasePtr(),
7228*9880d681SAndroid Build Coastguard Worker LN0->getMemoryVT(),
7229*9880d681SAndroid Build Coastguard Worker LN0->getMemOperand());
7230*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(N0.getValue(1), NewLoad.getValue(1));
7231*9880d681SAndroid Build Coastguard Worker return NewLoad;
7232*9880d681SAndroid Build Coastguard Worker }
7233*9880d681SAndroid Build Coastguard Worker }
7234*9880d681SAndroid Build Coastguard Worker }
7235*9880d681SAndroid Build Coastguard Worker // fold (trunc (concat ... x ...)) -> (concat ..., (trunc x), ...)),
7236*9880d681SAndroid Build Coastguard Worker // where ... are all 'undef'.
7237*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::CONCAT_VECTORS && !LegalTypes) {
7238*9880d681SAndroid Build Coastguard Worker SmallVector<EVT, 8> VTs;
7239*9880d681SAndroid Build Coastguard Worker SDValue V;
7240*9880d681SAndroid Build Coastguard Worker unsigned Idx = 0;
7241*9880d681SAndroid Build Coastguard Worker unsigned NumDefs = 0;
7242*9880d681SAndroid Build Coastguard Worker
7243*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0, e = N0.getNumOperands(); i != e; ++i) {
7244*9880d681SAndroid Build Coastguard Worker SDValue X = N0.getOperand(i);
7245*9880d681SAndroid Build Coastguard Worker if (!X.isUndef()) {
7246*9880d681SAndroid Build Coastguard Worker V = X;
7247*9880d681SAndroid Build Coastguard Worker Idx = i;
7248*9880d681SAndroid Build Coastguard Worker NumDefs++;
7249*9880d681SAndroid Build Coastguard Worker }
7250*9880d681SAndroid Build Coastguard Worker // Stop if more than one members are non-undef.
7251*9880d681SAndroid Build Coastguard Worker if (NumDefs > 1)
7252*9880d681SAndroid Build Coastguard Worker break;
7253*9880d681SAndroid Build Coastguard Worker VTs.push_back(EVT::getVectorVT(*DAG.getContext(),
7254*9880d681SAndroid Build Coastguard Worker VT.getVectorElementType(),
7255*9880d681SAndroid Build Coastguard Worker X.getValueType().getVectorNumElements()));
7256*9880d681SAndroid Build Coastguard Worker }
7257*9880d681SAndroid Build Coastguard Worker
7258*9880d681SAndroid Build Coastguard Worker if (NumDefs == 0)
7259*9880d681SAndroid Build Coastguard Worker return DAG.getUNDEF(VT);
7260*9880d681SAndroid Build Coastguard Worker
7261*9880d681SAndroid Build Coastguard Worker if (NumDefs == 1) {
7262*9880d681SAndroid Build Coastguard Worker assert(V.getNode() && "The single defined operand is empty!");
7263*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Opnds;
7264*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0, e = VTs.size(); i != e; ++i) {
7265*9880d681SAndroid Build Coastguard Worker if (i != Idx) {
7266*9880d681SAndroid Build Coastguard Worker Opnds.push_back(DAG.getUNDEF(VTs[i]));
7267*9880d681SAndroid Build Coastguard Worker continue;
7268*9880d681SAndroid Build Coastguard Worker }
7269*9880d681SAndroid Build Coastguard Worker SDValue NV = DAG.getNode(ISD::TRUNCATE, SDLoc(V), VTs[i], V);
7270*9880d681SAndroid Build Coastguard Worker AddToWorklist(NV.getNode());
7271*9880d681SAndroid Build Coastguard Worker Opnds.push_back(NV);
7272*9880d681SAndroid Build Coastguard Worker }
7273*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::CONCAT_VECTORS, SDLoc(N), VT, Opnds);
7274*9880d681SAndroid Build Coastguard Worker }
7275*9880d681SAndroid Build Coastguard Worker }
7276*9880d681SAndroid Build Coastguard Worker
7277*9880d681SAndroid Build Coastguard Worker // Fold truncate of a bitcast of a vector to an extract of the low vector
7278*9880d681SAndroid Build Coastguard Worker // element.
7279*9880d681SAndroid Build Coastguard Worker //
7280*9880d681SAndroid Build Coastguard Worker // e.g. trunc (i64 (bitcast v2i32:x)) -> extract_vector_elt v2i32:x, 0
7281*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::BITCAST && !VT.isVector()) {
7282*9880d681SAndroid Build Coastguard Worker SDValue VecSrc = N0.getOperand(0);
7283*9880d681SAndroid Build Coastguard Worker EVT SrcVT = VecSrc.getValueType();
7284*9880d681SAndroid Build Coastguard Worker if (SrcVT.isVector() && SrcVT.getScalarType() == VT &&
7285*9880d681SAndroid Build Coastguard Worker (!LegalOperations ||
7286*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegal(ISD::EXTRACT_VECTOR_ELT, SrcVT))) {
7287*9880d681SAndroid Build Coastguard Worker SDLoc SL(N);
7288*9880d681SAndroid Build Coastguard Worker
7289*9880d681SAndroid Build Coastguard Worker EVT IdxVT = TLI.getVectorIdxTy(DAG.getDataLayout());
7290*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, VT,
7291*9880d681SAndroid Build Coastguard Worker VecSrc, DAG.getConstant(0, SL, IdxVT));
7292*9880d681SAndroid Build Coastguard Worker }
7293*9880d681SAndroid Build Coastguard Worker }
7294*9880d681SAndroid Build Coastguard Worker
7295*9880d681SAndroid Build Coastguard Worker // Simplify the operands using demanded-bits information.
7296*9880d681SAndroid Build Coastguard Worker if (!VT.isVector() &&
7297*9880d681SAndroid Build Coastguard Worker SimplifyDemandedBits(SDValue(N, 0)))
7298*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0);
7299*9880d681SAndroid Build Coastguard Worker
7300*9880d681SAndroid Build Coastguard Worker return SDValue();
7301*9880d681SAndroid Build Coastguard Worker }
7302*9880d681SAndroid Build Coastguard Worker
getBuildPairElt(SDNode * N,unsigned i)7303*9880d681SAndroid Build Coastguard Worker static SDNode *getBuildPairElt(SDNode *N, unsigned i) {
7304*9880d681SAndroid Build Coastguard Worker SDValue Elt = N->getOperand(i);
7305*9880d681SAndroid Build Coastguard Worker if (Elt.getOpcode() != ISD::MERGE_VALUES)
7306*9880d681SAndroid Build Coastguard Worker return Elt.getNode();
7307*9880d681SAndroid Build Coastguard Worker return Elt.getOperand(Elt.getResNo()).getNode();
7308*9880d681SAndroid Build Coastguard Worker }
7309*9880d681SAndroid Build Coastguard Worker
7310*9880d681SAndroid Build Coastguard Worker /// build_pair (load, load) -> load
7311*9880d681SAndroid Build Coastguard Worker /// if load locations are consecutive.
CombineConsecutiveLoads(SDNode * N,EVT VT)7312*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::CombineConsecutiveLoads(SDNode *N, EVT VT) {
7313*9880d681SAndroid Build Coastguard Worker assert(N->getOpcode() == ISD::BUILD_PAIR);
7314*9880d681SAndroid Build Coastguard Worker
7315*9880d681SAndroid Build Coastguard Worker LoadSDNode *LD1 = dyn_cast<LoadSDNode>(getBuildPairElt(N, 0));
7316*9880d681SAndroid Build Coastguard Worker LoadSDNode *LD2 = dyn_cast<LoadSDNode>(getBuildPairElt(N, 1));
7317*9880d681SAndroid Build Coastguard Worker if (!LD1 || !LD2 || !ISD::isNON_EXTLoad(LD1) || !LD1->hasOneUse() ||
7318*9880d681SAndroid Build Coastguard Worker LD1->getAddressSpace() != LD2->getAddressSpace())
7319*9880d681SAndroid Build Coastguard Worker return SDValue();
7320*9880d681SAndroid Build Coastguard Worker EVT LD1VT = LD1->getValueType(0);
7321*9880d681SAndroid Build Coastguard Worker unsigned LD1Bytes = LD1VT.getSizeInBits() / 8;
7322*9880d681SAndroid Build Coastguard Worker if (ISD::isNON_EXTLoad(LD2) && LD2->hasOneUse() &&
7323*9880d681SAndroid Build Coastguard Worker DAG.areNonVolatileConsecutiveLoads(LD2, LD1, LD1Bytes, 1)) {
7324*9880d681SAndroid Build Coastguard Worker unsigned Align = LD1->getAlignment();
7325*9880d681SAndroid Build Coastguard Worker unsigned NewAlign = DAG.getDataLayout().getABITypeAlignment(
7326*9880d681SAndroid Build Coastguard Worker VT.getTypeForEVT(*DAG.getContext()));
7327*9880d681SAndroid Build Coastguard Worker
7328*9880d681SAndroid Build Coastguard Worker if (NewAlign <= Align &&
7329*9880d681SAndroid Build Coastguard Worker (!LegalOperations || TLI.isOperationLegal(ISD::LOAD, VT)))
7330*9880d681SAndroid Build Coastguard Worker return DAG.getLoad(VT, SDLoc(N), LD1->getChain(),
7331*9880d681SAndroid Build Coastguard Worker LD1->getBasePtr(), LD1->getPointerInfo(),
7332*9880d681SAndroid Build Coastguard Worker false, false, false, Align);
7333*9880d681SAndroid Build Coastguard Worker }
7334*9880d681SAndroid Build Coastguard Worker
7335*9880d681SAndroid Build Coastguard Worker return SDValue();
7336*9880d681SAndroid Build Coastguard Worker }
7337*9880d681SAndroid Build Coastguard Worker
getPPCf128HiElementSelector(const SelectionDAG & DAG)7338*9880d681SAndroid Build Coastguard Worker static unsigned getPPCf128HiElementSelector(const SelectionDAG &DAG) {
7339*9880d681SAndroid Build Coastguard Worker // On little-endian machines, bitcasting from ppcf128 to i128 does swap the Hi
7340*9880d681SAndroid Build Coastguard Worker // and Lo parts; on big-endian machines it doesn't.
7341*9880d681SAndroid Build Coastguard Worker return DAG.getDataLayout().isBigEndian() ? 1 : 0;
7342*9880d681SAndroid Build Coastguard Worker }
7343*9880d681SAndroid Build Coastguard Worker
foldBitcastedFPLogic(SDNode * N,SelectionDAG & DAG,const TargetLowering & TLI)7344*9880d681SAndroid Build Coastguard Worker static SDValue foldBitcastedFPLogic(SDNode *N, SelectionDAG &DAG,
7345*9880d681SAndroid Build Coastguard Worker const TargetLowering &TLI) {
7346*9880d681SAndroid Build Coastguard Worker // If this is not a bitcast to an FP type or if the target doesn't have
7347*9880d681SAndroid Build Coastguard Worker // IEEE754-compliant FP logic, we're done.
7348*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
7349*9880d681SAndroid Build Coastguard Worker if (!VT.isFloatingPoint() || !TLI.hasBitPreservingFPLogic(VT))
7350*9880d681SAndroid Build Coastguard Worker return SDValue();
7351*9880d681SAndroid Build Coastguard Worker
7352*9880d681SAndroid Build Coastguard Worker // TODO: Use splat values for the constant-checking below and remove this
7353*9880d681SAndroid Build Coastguard Worker // restriction.
7354*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
7355*9880d681SAndroid Build Coastguard Worker EVT SourceVT = N0.getValueType();
7356*9880d681SAndroid Build Coastguard Worker if (SourceVT.isVector())
7357*9880d681SAndroid Build Coastguard Worker return SDValue();
7358*9880d681SAndroid Build Coastguard Worker
7359*9880d681SAndroid Build Coastguard Worker unsigned FPOpcode;
7360*9880d681SAndroid Build Coastguard Worker APInt SignMask;
7361*9880d681SAndroid Build Coastguard Worker switch (N0.getOpcode()) {
7362*9880d681SAndroid Build Coastguard Worker case ISD::AND:
7363*9880d681SAndroid Build Coastguard Worker FPOpcode = ISD::FABS;
7364*9880d681SAndroid Build Coastguard Worker SignMask = ~APInt::getSignBit(SourceVT.getSizeInBits());
7365*9880d681SAndroid Build Coastguard Worker break;
7366*9880d681SAndroid Build Coastguard Worker case ISD::XOR:
7367*9880d681SAndroid Build Coastguard Worker FPOpcode = ISD::FNEG;
7368*9880d681SAndroid Build Coastguard Worker SignMask = APInt::getSignBit(SourceVT.getSizeInBits());
7369*9880d681SAndroid Build Coastguard Worker break;
7370*9880d681SAndroid Build Coastguard Worker // TODO: ISD::OR --> ISD::FNABS?
7371*9880d681SAndroid Build Coastguard Worker default:
7372*9880d681SAndroid Build Coastguard Worker return SDValue();
7373*9880d681SAndroid Build Coastguard Worker }
7374*9880d681SAndroid Build Coastguard Worker
7375*9880d681SAndroid Build Coastguard Worker // Fold (bitcast int (and (bitcast fp X to int), 0x7fff...) to fp) -> fabs X
7376*9880d681SAndroid Build Coastguard Worker // Fold (bitcast int (xor (bitcast fp X to int), 0x8000...) to fp) -> fneg X
7377*9880d681SAndroid Build Coastguard Worker SDValue LogicOp0 = N0.getOperand(0);
7378*9880d681SAndroid Build Coastguard Worker ConstantSDNode *LogicOp1 = dyn_cast<ConstantSDNode>(N0.getOperand(1));
7379*9880d681SAndroid Build Coastguard Worker if (LogicOp1 && LogicOp1->getAPIntValue() == SignMask &&
7380*9880d681SAndroid Build Coastguard Worker LogicOp0.getOpcode() == ISD::BITCAST &&
7381*9880d681SAndroid Build Coastguard Worker LogicOp0->getOperand(0).getValueType() == VT)
7382*9880d681SAndroid Build Coastguard Worker return DAG.getNode(FPOpcode, SDLoc(N), VT, LogicOp0->getOperand(0));
7383*9880d681SAndroid Build Coastguard Worker
7384*9880d681SAndroid Build Coastguard Worker return SDValue();
7385*9880d681SAndroid Build Coastguard Worker }
7386*9880d681SAndroid Build Coastguard Worker
visitBITCAST(SDNode * N)7387*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitBITCAST(SDNode *N) {
7388*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
7389*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
7390*9880d681SAndroid Build Coastguard Worker
7391*9880d681SAndroid Build Coastguard Worker // If the input is a BUILD_VECTOR with all constant elements, fold this now.
7392*9880d681SAndroid Build Coastguard Worker // Only do this before legalize, since afterward the target may be depending
7393*9880d681SAndroid Build Coastguard Worker // on the bitconvert.
7394*9880d681SAndroid Build Coastguard Worker // First check to see if this is all constant.
7395*9880d681SAndroid Build Coastguard Worker if (!LegalTypes &&
7396*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::BUILD_VECTOR && N0.getNode()->hasOneUse() &&
7397*9880d681SAndroid Build Coastguard Worker VT.isVector()) {
7398*9880d681SAndroid Build Coastguard Worker bool isSimple = cast<BuildVectorSDNode>(N0)->isConstant();
7399*9880d681SAndroid Build Coastguard Worker
7400*9880d681SAndroid Build Coastguard Worker EVT DestEltVT = N->getValueType(0).getVectorElementType();
7401*9880d681SAndroid Build Coastguard Worker assert(!DestEltVT.isVector() &&
7402*9880d681SAndroid Build Coastguard Worker "Element type of vector ValueType must not be vector!");
7403*9880d681SAndroid Build Coastguard Worker if (isSimple)
7404*9880d681SAndroid Build Coastguard Worker return ConstantFoldBITCASTofBUILD_VECTOR(N0.getNode(), DestEltVT);
7405*9880d681SAndroid Build Coastguard Worker }
7406*9880d681SAndroid Build Coastguard Worker
7407*9880d681SAndroid Build Coastguard Worker // If the input is a constant, let getNode fold it.
7408*9880d681SAndroid Build Coastguard Worker if (isa<ConstantSDNode>(N0) || isa<ConstantFPSDNode>(N0)) {
7409*9880d681SAndroid Build Coastguard Worker // If we can't allow illegal operations, we need to check that this is just
7410*9880d681SAndroid Build Coastguard Worker // a fp -> int or int -> conversion and that the resulting operation will
7411*9880d681SAndroid Build Coastguard Worker // be legal.
7412*9880d681SAndroid Build Coastguard Worker if (!LegalOperations ||
7413*9880d681SAndroid Build Coastguard Worker (isa<ConstantSDNode>(N0) && VT.isFloatingPoint() && !VT.isVector() &&
7414*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegal(ISD::ConstantFP, VT)) ||
7415*9880d681SAndroid Build Coastguard Worker (isa<ConstantFPSDNode>(N0) && VT.isInteger() && !VT.isVector() &&
7416*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegal(ISD::Constant, VT)))
7417*9880d681SAndroid Build Coastguard Worker return DAG.getBitcast(VT, N0);
7418*9880d681SAndroid Build Coastguard Worker }
7419*9880d681SAndroid Build Coastguard Worker
7420*9880d681SAndroid Build Coastguard Worker // (conv (conv x, t1), t2) -> (conv x, t2)
7421*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::BITCAST)
7422*9880d681SAndroid Build Coastguard Worker return DAG.getBitcast(VT, N0.getOperand(0));
7423*9880d681SAndroid Build Coastguard Worker
7424*9880d681SAndroid Build Coastguard Worker // fold (conv (load x)) -> (load (conv*)x)
7425*9880d681SAndroid Build Coastguard Worker // If the resultant load doesn't need a higher alignment than the original!
7426*9880d681SAndroid Build Coastguard Worker if (ISD::isNormalLoad(N0.getNode()) && N0.hasOneUse() &&
7427*9880d681SAndroid Build Coastguard Worker // Do not change the width of a volatile load.
7428*9880d681SAndroid Build Coastguard Worker !cast<LoadSDNode>(N0)->isVolatile() &&
7429*9880d681SAndroid Build Coastguard Worker // Do not remove the cast if the types differ in endian layout.
7430*9880d681SAndroid Build Coastguard Worker TLI.hasBigEndianPartOrdering(N0.getValueType(), DAG.getDataLayout()) ==
7431*9880d681SAndroid Build Coastguard Worker TLI.hasBigEndianPartOrdering(VT, DAG.getDataLayout()) &&
7432*9880d681SAndroid Build Coastguard Worker (!LegalOperations || TLI.isOperationLegal(ISD::LOAD, VT)) &&
7433*9880d681SAndroid Build Coastguard Worker TLI.isLoadBitCastBeneficial(N0.getValueType(), VT)) {
7434*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = cast<LoadSDNode>(N0);
7435*9880d681SAndroid Build Coastguard Worker unsigned OrigAlign = LN0->getAlignment();
7436*9880d681SAndroid Build Coastguard Worker
7437*9880d681SAndroid Build Coastguard Worker bool Fast = false;
7438*9880d681SAndroid Build Coastguard Worker if (TLI.allowsMemoryAccess(*DAG.getContext(), DAG.getDataLayout(), VT,
7439*9880d681SAndroid Build Coastguard Worker LN0->getAddressSpace(), OrigAlign, &Fast) &&
7440*9880d681SAndroid Build Coastguard Worker Fast) {
7441*9880d681SAndroid Build Coastguard Worker SDValue Load = DAG.getLoad(VT, SDLoc(N), LN0->getChain(),
7442*9880d681SAndroid Build Coastguard Worker LN0->getBasePtr(), LN0->getPointerInfo(),
7443*9880d681SAndroid Build Coastguard Worker LN0->isVolatile(), LN0->isNonTemporal(),
7444*9880d681SAndroid Build Coastguard Worker LN0->isInvariant(), OrigAlign,
7445*9880d681SAndroid Build Coastguard Worker LN0->getAAInfo());
7446*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(N0.getValue(1), Load.getValue(1));
7447*9880d681SAndroid Build Coastguard Worker return Load;
7448*9880d681SAndroid Build Coastguard Worker }
7449*9880d681SAndroid Build Coastguard Worker }
7450*9880d681SAndroid Build Coastguard Worker
7451*9880d681SAndroid Build Coastguard Worker if (SDValue V = foldBitcastedFPLogic(N, DAG, TLI))
7452*9880d681SAndroid Build Coastguard Worker return V;
7453*9880d681SAndroid Build Coastguard Worker
7454*9880d681SAndroid Build Coastguard Worker // fold (bitconvert (fneg x)) -> (xor (bitconvert x), signbit)
7455*9880d681SAndroid Build Coastguard Worker // fold (bitconvert (fabs x)) -> (and (bitconvert x), (not signbit))
7456*9880d681SAndroid Build Coastguard Worker //
7457*9880d681SAndroid Build Coastguard Worker // For ppc_fp128:
7458*9880d681SAndroid Build Coastguard Worker // fold (bitcast (fneg x)) ->
7459*9880d681SAndroid Build Coastguard Worker // flipbit = signbit
7460*9880d681SAndroid Build Coastguard Worker // (xor (bitcast x) (build_pair flipbit, flipbit))
7461*9880d681SAndroid Build Coastguard Worker //
7462*9880d681SAndroid Build Coastguard Worker // fold (bitcast (fabs x)) ->
7463*9880d681SAndroid Build Coastguard Worker // flipbit = (and (extract_element (bitcast x), 0), signbit)
7464*9880d681SAndroid Build Coastguard Worker // (xor (bitcast x) (build_pair flipbit, flipbit))
7465*9880d681SAndroid Build Coastguard Worker // This often reduces constant pool loads.
7466*9880d681SAndroid Build Coastguard Worker if (((N0.getOpcode() == ISD::FNEG && !TLI.isFNegFree(N0.getValueType())) ||
7467*9880d681SAndroid Build Coastguard Worker (N0.getOpcode() == ISD::FABS && !TLI.isFAbsFree(N0.getValueType()))) &&
7468*9880d681SAndroid Build Coastguard Worker N0.getNode()->hasOneUse() && VT.isInteger() &&
7469*9880d681SAndroid Build Coastguard Worker !VT.isVector() && !N0.getValueType().isVector()) {
7470*9880d681SAndroid Build Coastguard Worker SDValue NewConv = DAG.getBitcast(VT, N0.getOperand(0));
7471*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewConv.getNode());
7472*9880d681SAndroid Build Coastguard Worker
7473*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
7474*9880d681SAndroid Build Coastguard Worker if (N0.getValueType() == MVT::ppcf128 && !LegalTypes) {
7475*9880d681SAndroid Build Coastguard Worker assert(VT.getSizeInBits() == 128);
7476*9880d681SAndroid Build Coastguard Worker SDValue SignBit = DAG.getConstant(
7477*9880d681SAndroid Build Coastguard Worker APInt::getSignBit(VT.getSizeInBits() / 2), SDLoc(N0), MVT::i64);
7478*9880d681SAndroid Build Coastguard Worker SDValue FlipBit;
7479*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FNEG) {
7480*9880d681SAndroid Build Coastguard Worker FlipBit = SignBit;
7481*9880d681SAndroid Build Coastguard Worker AddToWorklist(FlipBit.getNode());
7482*9880d681SAndroid Build Coastguard Worker } else {
7483*9880d681SAndroid Build Coastguard Worker assert(N0.getOpcode() == ISD::FABS);
7484*9880d681SAndroid Build Coastguard Worker SDValue Hi =
7485*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::EXTRACT_ELEMENT, SDLoc(NewConv), MVT::i64, NewConv,
7486*9880d681SAndroid Build Coastguard Worker DAG.getIntPtrConstant(getPPCf128HiElementSelector(DAG),
7487*9880d681SAndroid Build Coastguard Worker SDLoc(NewConv)));
7488*9880d681SAndroid Build Coastguard Worker AddToWorklist(Hi.getNode());
7489*9880d681SAndroid Build Coastguard Worker FlipBit = DAG.getNode(ISD::AND, SDLoc(N0), MVT::i64, Hi, SignBit);
7490*9880d681SAndroid Build Coastguard Worker AddToWorklist(FlipBit.getNode());
7491*9880d681SAndroid Build Coastguard Worker }
7492*9880d681SAndroid Build Coastguard Worker SDValue FlipBits =
7493*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::BUILD_PAIR, SDLoc(N0), VT, FlipBit, FlipBit);
7494*9880d681SAndroid Build Coastguard Worker AddToWorklist(FlipBits.getNode());
7495*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::XOR, DL, VT, NewConv, FlipBits);
7496*9880d681SAndroid Build Coastguard Worker }
7497*9880d681SAndroid Build Coastguard Worker APInt SignBit = APInt::getSignBit(VT.getSizeInBits());
7498*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FNEG)
7499*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::XOR, DL, VT,
7500*9880d681SAndroid Build Coastguard Worker NewConv, DAG.getConstant(SignBit, DL, VT));
7501*9880d681SAndroid Build Coastguard Worker assert(N0.getOpcode() == ISD::FABS);
7502*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, DL, VT,
7503*9880d681SAndroid Build Coastguard Worker NewConv, DAG.getConstant(~SignBit, DL, VT));
7504*9880d681SAndroid Build Coastguard Worker }
7505*9880d681SAndroid Build Coastguard Worker
7506*9880d681SAndroid Build Coastguard Worker // fold (bitconvert (fcopysign cst, x)) ->
7507*9880d681SAndroid Build Coastguard Worker // (or (and (bitconvert x), sign), (and cst, (not sign)))
7508*9880d681SAndroid Build Coastguard Worker // Note that we don't handle (copysign x, cst) because this can always be
7509*9880d681SAndroid Build Coastguard Worker // folded to an fneg or fabs.
7510*9880d681SAndroid Build Coastguard Worker //
7511*9880d681SAndroid Build Coastguard Worker // For ppc_fp128:
7512*9880d681SAndroid Build Coastguard Worker // fold (bitcast (fcopysign cst, x)) ->
7513*9880d681SAndroid Build Coastguard Worker // flipbit = (and (extract_element
7514*9880d681SAndroid Build Coastguard Worker // (xor (bitcast cst), (bitcast x)), 0),
7515*9880d681SAndroid Build Coastguard Worker // signbit)
7516*9880d681SAndroid Build Coastguard Worker // (xor (bitcast cst) (build_pair flipbit, flipbit))
7517*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FCOPYSIGN && N0.getNode()->hasOneUse() &&
7518*9880d681SAndroid Build Coastguard Worker isa<ConstantFPSDNode>(N0.getOperand(0)) &&
7519*9880d681SAndroid Build Coastguard Worker VT.isInteger() && !VT.isVector()) {
7520*9880d681SAndroid Build Coastguard Worker unsigned OrigXWidth = N0.getOperand(1).getValueType().getSizeInBits();
7521*9880d681SAndroid Build Coastguard Worker EVT IntXVT = EVT::getIntegerVT(*DAG.getContext(), OrigXWidth);
7522*9880d681SAndroid Build Coastguard Worker if (isTypeLegal(IntXVT)) {
7523*9880d681SAndroid Build Coastguard Worker SDValue X = DAG.getBitcast(IntXVT, N0.getOperand(1));
7524*9880d681SAndroid Build Coastguard Worker AddToWorklist(X.getNode());
7525*9880d681SAndroid Build Coastguard Worker
7526*9880d681SAndroid Build Coastguard Worker // If X has a different width than the result/lhs, sext it or truncate it.
7527*9880d681SAndroid Build Coastguard Worker unsigned VTWidth = VT.getSizeInBits();
7528*9880d681SAndroid Build Coastguard Worker if (OrigXWidth < VTWidth) {
7529*9880d681SAndroid Build Coastguard Worker X = DAG.getNode(ISD::SIGN_EXTEND, SDLoc(N), VT, X);
7530*9880d681SAndroid Build Coastguard Worker AddToWorklist(X.getNode());
7531*9880d681SAndroid Build Coastguard Worker } else if (OrigXWidth > VTWidth) {
7532*9880d681SAndroid Build Coastguard Worker // To get the sign bit in the right place, we have to shift it right
7533*9880d681SAndroid Build Coastguard Worker // before truncating.
7534*9880d681SAndroid Build Coastguard Worker SDLoc DL(X);
7535*9880d681SAndroid Build Coastguard Worker X = DAG.getNode(ISD::SRL, DL,
7536*9880d681SAndroid Build Coastguard Worker X.getValueType(), X,
7537*9880d681SAndroid Build Coastguard Worker DAG.getConstant(OrigXWidth-VTWidth, DL,
7538*9880d681SAndroid Build Coastguard Worker X.getValueType()));
7539*9880d681SAndroid Build Coastguard Worker AddToWorklist(X.getNode());
7540*9880d681SAndroid Build Coastguard Worker X = DAG.getNode(ISD::TRUNCATE, SDLoc(X), VT, X);
7541*9880d681SAndroid Build Coastguard Worker AddToWorklist(X.getNode());
7542*9880d681SAndroid Build Coastguard Worker }
7543*9880d681SAndroid Build Coastguard Worker
7544*9880d681SAndroid Build Coastguard Worker if (N0.getValueType() == MVT::ppcf128 && !LegalTypes) {
7545*9880d681SAndroid Build Coastguard Worker APInt SignBit = APInt::getSignBit(VT.getSizeInBits() / 2);
7546*9880d681SAndroid Build Coastguard Worker SDValue Cst = DAG.getBitcast(VT, N0.getOperand(0));
7547*9880d681SAndroid Build Coastguard Worker AddToWorklist(Cst.getNode());
7548*9880d681SAndroid Build Coastguard Worker SDValue X = DAG.getBitcast(VT, N0.getOperand(1));
7549*9880d681SAndroid Build Coastguard Worker AddToWorklist(X.getNode());
7550*9880d681SAndroid Build Coastguard Worker SDValue XorResult = DAG.getNode(ISD::XOR, SDLoc(N0), VT, Cst, X);
7551*9880d681SAndroid Build Coastguard Worker AddToWorklist(XorResult.getNode());
7552*9880d681SAndroid Build Coastguard Worker SDValue XorResult64 = DAG.getNode(
7553*9880d681SAndroid Build Coastguard Worker ISD::EXTRACT_ELEMENT, SDLoc(XorResult), MVT::i64, XorResult,
7554*9880d681SAndroid Build Coastguard Worker DAG.getIntPtrConstant(getPPCf128HiElementSelector(DAG),
7555*9880d681SAndroid Build Coastguard Worker SDLoc(XorResult)));
7556*9880d681SAndroid Build Coastguard Worker AddToWorklist(XorResult64.getNode());
7557*9880d681SAndroid Build Coastguard Worker SDValue FlipBit =
7558*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::AND, SDLoc(XorResult64), MVT::i64, XorResult64,
7559*9880d681SAndroid Build Coastguard Worker DAG.getConstant(SignBit, SDLoc(XorResult64), MVT::i64));
7560*9880d681SAndroid Build Coastguard Worker AddToWorklist(FlipBit.getNode());
7561*9880d681SAndroid Build Coastguard Worker SDValue FlipBits =
7562*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::BUILD_PAIR, SDLoc(N0), VT, FlipBit, FlipBit);
7563*9880d681SAndroid Build Coastguard Worker AddToWorklist(FlipBits.getNode());
7564*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::XOR, SDLoc(N), VT, Cst, FlipBits);
7565*9880d681SAndroid Build Coastguard Worker }
7566*9880d681SAndroid Build Coastguard Worker APInt SignBit = APInt::getSignBit(VT.getSizeInBits());
7567*9880d681SAndroid Build Coastguard Worker X = DAG.getNode(ISD::AND, SDLoc(X), VT,
7568*9880d681SAndroid Build Coastguard Worker X, DAG.getConstant(SignBit, SDLoc(X), VT));
7569*9880d681SAndroid Build Coastguard Worker AddToWorklist(X.getNode());
7570*9880d681SAndroid Build Coastguard Worker
7571*9880d681SAndroid Build Coastguard Worker SDValue Cst = DAG.getBitcast(VT, N0.getOperand(0));
7572*9880d681SAndroid Build Coastguard Worker Cst = DAG.getNode(ISD::AND, SDLoc(Cst), VT,
7573*9880d681SAndroid Build Coastguard Worker Cst, DAG.getConstant(~SignBit, SDLoc(Cst), VT));
7574*9880d681SAndroid Build Coastguard Worker AddToWorklist(Cst.getNode());
7575*9880d681SAndroid Build Coastguard Worker
7576*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::OR, SDLoc(N), VT, X, Cst);
7577*9880d681SAndroid Build Coastguard Worker }
7578*9880d681SAndroid Build Coastguard Worker }
7579*9880d681SAndroid Build Coastguard Worker
7580*9880d681SAndroid Build Coastguard Worker // bitconvert(build_pair(ld, ld)) -> ld iff load locations are consecutive.
7581*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::BUILD_PAIR)
7582*9880d681SAndroid Build Coastguard Worker if (SDValue CombineLD = CombineConsecutiveLoads(N0.getNode(), VT))
7583*9880d681SAndroid Build Coastguard Worker return CombineLD;
7584*9880d681SAndroid Build Coastguard Worker
7585*9880d681SAndroid Build Coastguard Worker // Remove double bitcasts from shuffles - this is often a legacy of
7586*9880d681SAndroid Build Coastguard Worker // XformToShuffleWithZero being used to combine bitmaskings (of
7587*9880d681SAndroid Build Coastguard Worker // float vectors bitcast to integer vectors) into shuffles.
7588*9880d681SAndroid Build Coastguard Worker // bitcast(shuffle(bitcast(s0),bitcast(s1))) -> shuffle(s0,s1)
7589*9880d681SAndroid Build Coastguard Worker if (Level < AfterLegalizeDAG && TLI.isTypeLegal(VT) && VT.isVector() &&
7590*9880d681SAndroid Build Coastguard Worker N0->getOpcode() == ISD::VECTOR_SHUFFLE &&
7591*9880d681SAndroid Build Coastguard Worker VT.getVectorNumElements() >= N0.getValueType().getVectorNumElements() &&
7592*9880d681SAndroid Build Coastguard Worker !(VT.getVectorNumElements() % N0.getValueType().getVectorNumElements())) {
7593*9880d681SAndroid Build Coastguard Worker ShuffleVectorSDNode *SVN = cast<ShuffleVectorSDNode>(N0);
7594*9880d681SAndroid Build Coastguard Worker
7595*9880d681SAndroid Build Coastguard Worker // If operands are a bitcast, peek through if it casts the original VT.
7596*9880d681SAndroid Build Coastguard Worker // If operands are a constant, just bitcast back to original VT.
7597*9880d681SAndroid Build Coastguard Worker auto PeekThroughBitcast = [&](SDValue Op) {
7598*9880d681SAndroid Build Coastguard Worker if (Op.getOpcode() == ISD::BITCAST &&
7599*9880d681SAndroid Build Coastguard Worker Op.getOperand(0).getValueType() == VT)
7600*9880d681SAndroid Build Coastguard Worker return SDValue(Op.getOperand(0));
7601*9880d681SAndroid Build Coastguard Worker if (ISD::isBuildVectorOfConstantSDNodes(Op.getNode()) ||
7602*9880d681SAndroid Build Coastguard Worker ISD::isBuildVectorOfConstantFPSDNodes(Op.getNode()))
7603*9880d681SAndroid Build Coastguard Worker return DAG.getBitcast(VT, Op);
7604*9880d681SAndroid Build Coastguard Worker return SDValue();
7605*9880d681SAndroid Build Coastguard Worker };
7606*9880d681SAndroid Build Coastguard Worker
7607*9880d681SAndroid Build Coastguard Worker SDValue SV0 = PeekThroughBitcast(N0->getOperand(0));
7608*9880d681SAndroid Build Coastguard Worker SDValue SV1 = PeekThroughBitcast(N0->getOperand(1));
7609*9880d681SAndroid Build Coastguard Worker if (!(SV0 && SV1))
7610*9880d681SAndroid Build Coastguard Worker return SDValue();
7611*9880d681SAndroid Build Coastguard Worker
7612*9880d681SAndroid Build Coastguard Worker int MaskScale =
7613*9880d681SAndroid Build Coastguard Worker VT.getVectorNumElements() / N0.getValueType().getVectorNumElements();
7614*9880d681SAndroid Build Coastguard Worker SmallVector<int, 8> NewMask;
7615*9880d681SAndroid Build Coastguard Worker for (int M : SVN->getMask())
7616*9880d681SAndroid Build Coastguard Worker for (int i = 0; i != MaskScale; ++i)
7617*9880d681SAndroid Build Coastguard Worker NewMask.push_back(M < 0 ? -1 : M * MaskScale + i);
7618*9880d681SAndroid Build Coastguard Worker
7619*9880d681SAndroid Build Coastguard Worker bool LegalMask = TLI.isShuffleMaskLegal(NewMask, VT);
7620*9880d681SAndroid Build Coastguard Worker if (!LegalMask) {
7621*9880d681SAndroid Build Coastguard Worker std::swap(SV0, SV1);
7622*9880d681SAndroid Build Coastguard Worker ShuffleVectorSDNode::commuteMask(NewMask);
7623*9880d681SAndroid Build Coastguard Worker LegalMask = TLI.isShuffleMaskLegal(NewMask, VT);
7624*9880d681SAndroid Build Coastguard Worker }
7625*9880d681SAndroid Build Coastguard Worker
7626*9880d681SAndroid Build Coastguard Worker if (LegalMask)
7627*9880d681SAndroid Build Coastguard Worker return DAG.getVectorShuffle(VT, SDLoc(N), SV0, SV1, NewMask);
7628*9880d681SAndroid Build Coastguard Worker }
7629*9880d681SAndroid Build Coastguard Worker
7630*9880d681SAndroid Build Coastguard Worker return SDValue();
7631*9880d681SAndroid Build Coastguard Worker }
7632*9880d681SAndroid Build Coastguard Worker
visitBUILD_PAIR(SDNode * N)7633*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitBUILD_PAIR(SDNode *N) {
7634*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
7635*9880d681SAndroid Build Coastguard Worker return CombineConsecutiveLoads(N, VT);
7636*9880d681SAndroid Build Coastguard Worker }
7637*9880d681SAndroid Build Coastguard Worker
7638*9880d681SAndroid Build Coastguard Worker /// We know that BV is a build_vector node with Constant, ConstantFP or Undef
7639*9880d681SAndroid Build Coastguard Worker /// operands. DstEltVT indicates the destination element value type.
7640*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::
ConstantFoldBITCASTofBUILD_VECTOR(SDNode * BV,EVT DstEltVT)7641*9880d681SAndroid Build Coastguard Worker ConstantFoldBITCASTofBUILD_VECTOR(SDNode *BV, EVT DstEltVT) {
7642*9880d681SAndroid Build Coastguard Worker EVT SrcEltVT = BV->getValueType(0).getVectorElementType();
7643*9880d681SAndroid Build Coastguard Worker
7644*9880d681SAndroid Build Coastguard Worker // If this is already the right type, we're done.
7645*9880d681SAndroid Build Coastguard Worker if (SrcEltVT == DstEltVT) return SDValue(BV, 0);
7646*9880d681SAndroid Build Coastguard Worker
7647*9880d681SAndroid Build Coastguard Worker unsigned SrcBitSize = SrcEltVT.getSizeInBits();
7648*9880d681SAndroid Build Coastguard Worker unsigned DstBitSize = DstEltVT.getSizeInBits();
7649*9880d681SAndroid Build Coastguard Worker
7650*9880d681SAndroid Build Coastguard Worker // If this is a conversion of N elements of one type to N elements of another
7651*9880d681SAndroid Build Coastguard Worker // type, convert each element. This handles FP<->INT cases.
7652*9880d681SAndroid Build Coastguard Worker if (SrcBitSize == DstBitSize) {
7653*9880d681SAndroid Build Coastguard Worker EVT VT = EVT::getVectorVT(*DAG.getContext(), DstEltVT,
7654*9880d681SAndroid Build Coastguard Worker BV->getValueType(0).getVectorNumElements());
7655*9880d681SAndroid Build Coastguard Worker
7656*9880d681SAndroid Build Coastguard Worker // Due to the FP element handling below calling this routine recursively,
7657*9880d681SAndroid Build Coastguard Worker // we can end up with a scalar-to-vector node here.
7658*9880d681SAndroid Build Coastguard Worker if (BV->getOpcode() == ISD::SCALAR_TO_VECTOR)
7659*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SCALAR_TO_VECTOR, SDLoc(BV), VT,
7660*9880d681SAndroid Build Coastguard Worker DAG.getBitcast(DstEltVT, BV->getOperand(0)));
7661*9880d681SAndroid Build Coastguard Worker
7662*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Ops;
7663*9880d681SAndroid Build Coastguard Worker for (SDValue Op : BV->op_values()) {
7664*9880d681SAndroid Build Coastguard Worker // If the vector element type is not legal, the BUILD_VECTOR operands
7665*9880d681SAndroid Build Coastguard Worker // are promoted and implicitly truncated. Make that explicit here.
7666*9880d681SAndroid Build Coastguard Worker if (Op.getValueType() != SrcEltVT)
7667*9880d681SAndroid Build Coastguard Worker Op = DAG.getNode(ISD::TRUNCATE, SDLoc(BV), SrcEltVT, Op);
7668*9880d681SAndroid Build Coastguard Worker Ops.push_back(DAG.getBitcast(DstEltVT, Op));
7669*9880d681SAndroid Build Coastguard Worker AddToWorklist(Ops.back().getNode());
7670*9880d681SAndroid Build Coastguard Worker }
7671*9880d681SAndroid Build Coastguard Worker return DAG.getBuildVector(VT, SDLoc(BV), Ops);
7672*9880d681SAndroid Build Coastguard Worker }
7673*9880d681SAndroid Build Coastguard Worker
7674*9880d681SAndroid Build Coastguard Worker // Otherwise, we're growing or shrinking the elements. To avoid having to
7675*9880d681SAndroid Build Coastguard Worker // handle annoying details of growing/shrinking FP values, we convert them to
7676*9880d681SAndroid Build Coastguard Worker // int first.
7677*9880d681SAndroid Build Coastguard Worker if (SrcEltVT.isFloatingPoint()) {
7678*9880d681SAndroid Build Coastguard Worker // Convert the input float vector to a int vector where the elements are the
7679*9880d681SAndroid Build Coastguard Worker // same sizes.
7680*9880d681SAndroid Build Coastguard Worker EVT IntVT = EVT::getIntegerVT(*DAG.getContext(), SrcEltVT.getSizeInBits());
7681*9880d681SAndroid Build Coastguard Worker BV = ConstantFoldBITCASTofBUILD_VECTOR(BV, IntVT).getNode();
7682*9880d681SAndroid Build Coastguard Worker SrcEltVT = IntVT;
7683*9880d681SAndroid Build Coastguard Worker }
7684*9880d681SAndroid Build Coastguard Worker
7685*9880d681SAndroid Build Coastguard Worker // Now we know the input is an integer vector. If the output is a FP type,
7686*9880d681SAndroid Build Coastguard Worker // convert to integer first, then to FP of the right size.
7687*9880d681SAndroid Build Coastguard Worker if (DstEltVT.isFloatingPoint()) {
7688*9880d681SAndroid Build Coastguard Worker EVT TmpVT = EVT::getIntegerVT(*DAG.getContext(), DstEltVT.getSizeInBits());
7689*9880d681SAndroid Build Coastguard Worker SDNode *Tmp = ConstantFoldBITCASTofBUILD_VECTOR(BV, TmpVT).getNode();
7690*9880d681SAndroid Build Coastguard Worker
7691*9880d681SAndroid Build Coastguard Worker // Next, convert to FP elements of the same size.
7692*9880d681SAndroid Build Coastguard Worker return ConstantFoldBITCASTofBUILD_VECTOR(Tmp, DstEltVT);
7693*9880d681SAndroid Build Coastguard Worker }
7694*9880d681SAndroid Build Coastguard Worker
7695*9880d681SAndroid Build Coastguard Worker SDLoc DL(BV);
7696*9880d681SAndroid Build Coastguard Worker
7697*9880d681SAndroid Build Coastguard Worker // Okay, we know the src/dst types are both integers of differing types.
7698*9880d681SAndroid Build Coastguard Worker // Handling growing first.
7699*9880d681SAndroid Build Coastguard Worker assert(SrcEltVT.isInteger() && DstEltVT.isInteger());
7700*9880d681SAndroid Build Coastguard Worker if (SrcBitSize < DstBitSize) {
7701*9880d681SAndroid Build Coastguard Worker unsigned NumInputsPerOutput = DstBitSize/SrcBitSize;
7702*9880d681SAndroid Build Coastguard Worker
7703*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Ops;
7704*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0, e = BV->getNumOperands(); i != e;
7705*9880d681SAndroid Build Coastguard Worker i += NumInputsPerOutput) {
7706*9880d681SAndroid Build Coastguard Worker bool isLE = DAG.getDataLayout().isLittleEndian();
7707*9880d681SAndroid Build Coastguard Worker APInt NewBits = APInt(DstBitSize, 0);
7708*9880d681SAndroid Build Coastguard Worker bool EltIsUndef = true;
7709*9880d681SAndroid Build Coastguard Worker for (unsigned j = 0; j != NumInputsPerOutput; ++j) {
7710*9880d681SAndroid Build Coastguard Worker // Shift the previously computed bits over.
7711*9880d681SAndroid Build Coastguard Worker NewBits <<= SrcBitSize;
7712*9880d681SAndroid Build Coastguard Worker SDValue Op = BV->getOperand(i+ (isLE ? (NumInputsPerOutput-j-1) : j));
7713*9880d681SAndroid Build Coastguard Worker if (Op.isUndef()) continue;
7714*9880d681SAndroid Build Coastguard Worker EltIsUndef = false;
7715*9880d681SAndroid Build Coastguard Worker
7716*9880d681SAndroid Build Coastguard Worker NewBits |= cast<ConstantSDNode>(Op)->getAPIntValue().
7717*9880d681SAndroid Build Coastguard Worker zextOrTrunc(SrcBitSize).zext(DstBitSize);
7718*9880d681SAndroid Build Coastguard Worker }
7719*9880d681SAndroid Build Coastguard Worker
7720*9880d681SAndroid Build Coastguard Worker if (EltIsUndef)
7721*9880d681SAndroid Build Coastguard Worker Ops.push_back(DAG.getUNDEF(DstEltVT));
7722*9880d681SAndroid Build Coastguard Worker else
7723*9880d681SAndroid Build Coastguard Worker Ops.push_back(DAG.getConstant(NewBits, DL, DstEltVT));
7724*9880d681SAndroid Build Coastguard Worker }
7725*9880d681SAndroid Build Coastguard Worker
7726*9880d681SAndroid Build Coastguard Worker EVT VT = EVT::getVectorVT(*DAG.getContext(), DstEltVT, Ops.size());
7727*9880d681SAndroid Build Coastguard Worker return DAG.getBuildVector(VT, DL, Ops);
7728*9880d681SAndroid Build Coastguard Worker }
7729*9880d681SAndroid Build Coastguard Worker
7730*9880d681SAndroid Build Coastguard Worker // Finally, this must be the case where we are shrinking elements: each input
7731*9880d681SAndroid Build Coastguard Worker // turns into multiple outputs.
7732*9880d681SAndroid Build Coastguard Worker unsigned NumOutputsPerInput = SrcBitSize/DstBitSize;
7733*9880d681SAndroid Build Coastguard Worker EVT VT = EVT::getVectorVT(*DAG.getContext(), DstEltVT,
7734*9880d681SAndroid Build Coastguard Worker NumOutputsPerInput*BV->getNumOperands());
7735*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Ops;
7736*9880d681SAndroid Build Coastguard Worker
7737*9880d681SAndroid Build Coastguard Worker for (const SDValue &Op : BV->op_values()) {
7738*9880d681SAndroid Build Coastguard Worker if (Op.isUndef()) {
7739*9880d681SAndroid Build Coastguard Worker Ops.append(NumOutputsPerInput, DAG.getUNDEF(DstEltVT));
7740*9880d681SAndroid Build Coastguard Worker continue;
7741*9880d681SAndroid Build Coastguard Worker }
7742*9880d681SAndroid Build Coastguard Worker
7743*9880d681SAndroid Build Coastguard Worker APInt OpVal = cast<ConstantSDNode>(Op)->
7744*9880d681SAndroid Build Coastguard Worker getAPIntValue().zextOrTrunc(SrcBitSize);
7745*9880d681SAndroid Build Coastguard Worker
7746*9880d681SAndroid Build Coastguard Worker for (unsigned j = 0; j != NumOutputsPerInput; ++j) {
7747*9880d681SAndroid Build Coastguard Worker APInt ThisVal = OpVal.trunc(DstBitSize);
7748*9880d681SAndroid Build Coastguard Worker Ops.push_back(DAG.getConstant(ThisVal, DL, DstEltVT));
7749*9880d681SAndroid Build Coastguard Worker OpVal = OpVal.lshr(DstBitSize);
7750*9880d681SAndroid Build Coastguard Worker }
7751*9880d681SAndroid Build Coastguard Worker
7752*9880d681SAndroid Build Coastguard Worker // For big endian targets, swap the order of the pieces of each element.
7753*9880d681SAndroid Build Coastguard Worker if (DAG.getDataLayout().isBigEndian())
7754*9880d681SAndroid Build Coastguard Worker std::reverse(Ops.end()-NumOutputsPerInput, Ops.end());
7755*9880d681SAndroid Build Coastguard Worker }
7756*9880d681SAndroid Build Coastguard Worker
7757*9880d681SAndroid Build Coastguard Worker return DAG.getBuildVector(VT, DL, Ops);
7758*9880d681SAndroid Build Coastguard Worker }
7759*9880d681SAndroid Build Coastguard Worker
7760*9880d681SAndroid Build Coastguard Worker /// Try to perform FMA combining on a given FADD node.
visitFADDForFMACombine(SDNode * N)7761*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFADDForFMACombine(SDNode *N) {
7762*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
7763*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
7764*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
7765*9880d681SAndroid Build Coastguard Worker SDLoc SL(N);
7766*9880d681SAndroid Build Coastguard Worker
7767*9880d681SAndroid Build Coastguard Worker const TargetOptions &Options = DAG.getTarget().Options;
7768*9880d681SAndroid Build Coastguard Worker bool AllowFusion =
7769*9880d681SAndroid Build Coastguard Worker (Options.AllowFPOpFusion == FPOpFusion::Fast || Options.UnsafeFPMath);
7770*9880d681SAndroid Build Coastguard Worker
7771*9880d681SAndroid Build Coastguard Worker // Floating-point multiply-add with intermediate rounding.
7772*9880d681SAndroid Build Coastguard Worker bool HasFMAD = (LegalOperations && TLI.isOperationLegal(ISD::FMAD, VT));
7773*9880d681SAndroid Build Coastguard Worker
7774*9880d681SAndroid Build Coastguard Worker // Floating-point multiply-add without intermediate rounding.
7775*9880d681SAndroid Build Coastguard Worker bool HasFMA =
7776*9880d681SAndroid Build Coastguard Worker AllowFusion && TLI.isFMAFasterThanFMulAndFAdd(VT) &&
7777*9880d681SAndroid Build Coastguard Worker (!LegalOperations || TLI.isOperationLegalOrCustom(ISD::FMA, VT));
7778*9880d681SAndroid Build Coastguard Worker
7779*9880d681SAndroid Build Coastguard Worker // No valid opcode, do not combine.
7780*9880d681SAndroid Build Coastguard Worker if (!HasFMAD && !HasFMA)
7781*9880d681SAndroid Build Coastguard Worker return SDValue();
7782*9880d681SAndroid Build Coastguard Worker
7783*9880d681SAndroid Build Coastguard Worker const SelectionDAGTargetInfo *STI = DAG.getSubtarget().getSelectionDAGInfo();
7784*9880d681SAndroid Build Coastguard Worker ;
7785*9880d681SAndroid Build Coastguard Worker if (AllowFusion && STI && STI->generateFMAsInMachineCombiner(OptLevel))
7786*9880d681SAndroid Build Coastguard Worker return SDValue();
7787*9880d681SAndroid Build Coastguard Worker
7788*9880d681SAndroid Build Coastguard Worker // Always prefer FMAD to FMA for precision.
7789*9880d681SAndroid Build Coastguard Worker unsigned PreferredFusedOpcode = HasFMAD ? ISD::FMAD : ISD::FMA;
7790*9880d681SAndroid Build Coastguard Worker bool Aggressive = TLI.enableAggressiveFMAFusion(VT);
7791*9880d681SAndroid Build Coastguard Worker bool LookThroughFPExt = TLI.isFPExtFree(VT);
7792*9880d681SAndroid Build Coastguard Worker
7793*9880d681SAndroid Build Coastguard Worker // If we have two choices trying to fold (fadd (fmul u, v), (fmul x, y)),
7794*9880d681SAndroid Build Coastguard Worker // prefer to fold the multiply with fewer uses.
7795*9880d681SAndroid Build Coastguard Worker if (Aggressive && N0.getOpcode() == ISD::FMUL &&
7796*9880d681SAndroid Build Coastguard Worker N1.getOpcode() == ISD::FMUL) {
7797*9880d681SAndroid Build Coastguard Worker if (N0.getNode()->use_size() > N1.getNode()->use_size())
7798*9880d681SAndroid Build Coastguard Worker std::swap(N0, N1);
7799*9880d681SAndroid Build Coastguard Worker }
7800*9880d681SAndroid Build Coastguard Worker
7801*9880d681SAndroid Build Coastguard Worker // fold (fadd (fmul x, y), z) -> (fma x, y, z)
7802*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FMUL &&
7803*9880d681SAndroid Build Coastguard Worker (Aggressive || N0->hasOneUse())) {
7804*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
7805*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N0.getOperand(1), N1);
7806*9880d681SAndroid Build Coastguard Worker }
7807*9880d681SAndroid Build Coastguard Worker
7808*9880d681SAndroid Build Coastguard Worker // fold (fadd x, (fmul y, z)) -> (fma y, z, x)
7809*9880d681SAndroid Build Coastguard Worker // Note: Commutes FADD operands.
7810*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::FMUL &&
7811*9880d681SAndroid Build Coastguard Worker (Aggressive || N1->hasOneUse())) {
7812*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
7813*9880d681SAndroid Build Coastguard Worker N1.getOperand(0), N1.getOperand(1), N0);
7814*9880d681SAndroid Build Coastguard Worker }
7815*9880d681SAndroid Build Coastguard Worker
7816*9880d681SAndroid Build Coastguard Worker // Look through FP_EXTEND nodes to do more combining.
7817*9880d681SAndroid Build Coastguard Worker if (AllowFusion && LookThroughFPExt) {
7818*9880d681SAndroid Build Coastguard Worker // fold (fadd (fpext (fmul x, y)), z) -> (fma (fpext x), (fpext y), z)
7819*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FP_EXTEND) {
7820*9880d681SAndroid Build Coastguard Worker SDValue N00 = N0.getOperand(0);
7821*9880d681SAndroid Build Coastguard Worker if (N00.getOpcode() == ISD::FMUL)
7822*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
7823*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
7824*9880d681SAndroid Build Coastguard Worker N00.getOperand(0)),
7825*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
7826*9880d681SAndroid Build Coastguard Worker N00.getOperand(1)), N1);
7827*9880d681SAndroid Build Coastguard Worker }
7828*9880d681SAndroid Build Coastguard Worker
7829*9880d681SAndroid Build Coastguard Worker // fold (fadd x, (fpext (fmul y, z))) -> (fma (fpext y), (fpext z), x)
7830*9880d681SAndroid Build Coastguard Worker // Note: Commutes FADD operands.
7831*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::FP_EXTEND) {
7832*9880d681SAndroid Build Coastguard Worker SDValue N10 = N1.getOperand(0);
7833*9880d681SAndroid Build Coastguard Worker if (N10.getOpcode() == ISD::FMUL)
7834*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
7835*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
7836*9880d681SAndroid Build Coastguard Worker N10.getOperand(0)),
7837*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
7838*9880d681SAndroid Build Coastguard Worker N10.getOperand(1)), N0);
7839*9880d681SAndroid Build Coastguard Worker }
7840*9880d681SAndroid Build Coastguard Worker }
7841*9880d681SAndroid Build Coastguard Worker
7842*9880d681SAndroid Build Coastguard Worker // More folding opportunities when target permits.
7843*9880d681SAndroid Build Coastguard Worker if ((AllowFusion || HasFMAD) && Aggressive) {
7844*9880d681SAndroid Build Coastguard Worker // fold (fadd (fma x, y, (fmul u, v)), z) -> (fma x, y (fma u, v, z))
7845*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == PreferredFusedOpcode &&
7846*9880d681SAndroid Build Coastguard Worker N0.getOperand(2).getOpcode() == ISD::FMUL) {
7847*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
7848*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N0.getOperand(1),
7849*9880d681SAndroid Build Coastguard Worker DAG.getNode(PreferredFusedOpcode, SL, VT,
7850*9880d681SAndroid Build Coastguard Worker N0.getOperand(2).getOperand(0),
7851*9880d681SAndroid Build Coastguard Worker N0.getOperand(2).getOperand(1),
7852*9880d681SAndroid Build Coastguard Worker N1));
7853*9880d681SAndroid Build Coastguard Worker }
7854*9880d681SAndroid Build Coastguard Worker
7855*9880d681SAndroid Build Coastguard Worker // fold (fadd x, (fma y, z, (fmul u, v)) -> (fma y, z (fma u, v, x))
7856*9880d681SAndroid Build Coastguard Worker if (N1->getOpcode() == PreferredFusedOpcode &&
7857*9880d681SAndroid Build Coastguard Worker N1.getOperand(2).getOpcode() == ISD::FMUL) {
7858*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
7859*9880d681SAndroid Build Coastguard Worker N1.getOperand(0), N1.getOperand(1),
7860*9880d681SAndroid Build Coastguard Worker DAG.getNode(PreferredFusedOpcode, SL, VT,
7861*9880d681SAndroid Build Coastguard Worker N1.getOperand(2).getOperand(0),
7862*9880d681SAndroid Build Coastguard Worker N1.getOperand(2).getOperand(1),
7863*9880d681SAndroid Build Coastguard Worker N0));
7864*9880d681SAndroid Build Coastguard Worker }
7865*9880d681SAndroid Build Coastguard Worker
7866*9880d681SAndroid Build Coastguard Worker if (AllowFusion && LookThroughFPExt) {
7867*9880d681SAndroid Build Coastguard Worker // fold (fadd (fma x, y, (fpext (fmul u, v))), z)
7868*9880d681SAndroid Build Coastguard Worker // -> (fma x, y, (fma (fpext u), (fpext v), z))
7869*9880d681SAndroid Build Coastguard Worker auto FoldFAddFMAFPExtFMul = [&] (
7870*9880d681SAndroid Build Coastguard Worker SDValue X, SDValue Y, SDValue U, SDValue V, SDValue Z) {
7871*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT, X, Y,
7872*9880d681SAndroid Build Coastguard Worker DAG.getNode(PreferredFusedOpcode, SL, VT,
7873*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT, U),
7874*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT, V),
7875*9880d681SAndroid Build Coastguard Worker Z));
7876*9880d681SAndroid Build Coastguard Worker };
7877*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == PreferredFusedOpcode) {
7878*9880d681SAndroid Build Coastguard Worker SDValue N02 = N0.getOperand(2);
7879*9880d681SAndroid Build Coastguard Worker if (N02.getOpcode() == ISD::FP_EXTEND) {
7880*9880d681SAndroid Build Coastguard Worker SDValue N020 = N02.getOperand(0);
7881*9880d681SAndroid Build Coastguard Worker if (N020.getOpcode() == ISD::FMUL)
7882*9880d681SAndroid Build Coastguard Worker return FoldFAddFMAFPExtFMul(N0.getOperand(0), N0.getOperand(1),
7883*9880d681SAndroid Build Coastguard Worker N020.getOperand(0), N020.getOperand(1),
7884*9880d681SAndroid Build Coastguard Worker N1);
7885*9880d681SAndroid Build Coastguard Worker }
7886*9880d681SAndroid Build Coastguard Worker }
7887*9880d681SAndroid Build Coastguard Worker
7888*9880d681SAndroid Build Coastguard Worker // fold (fadd (fpext (fma x, y, (fmul u, v))), z)
7889*9880d681SAndroid Build Coastguard Worker // -> (fma (fpext x), (fpext y), (fma (fpext u), (fpext v), z))
7890*9880d681SAndroid Build Coastguard Worker // FIXME: This turns two single-precision and one double-precision
7891*9880d681SAndroid Build Coastguard Worker // operation into two double-precision operations, which might not be
7892*9880d681SAndroid Build Coastguard Worker // interesting for all targets, especially GPUs.
7893*9880d681SAndroid Build Coastguard Worker auto FoldFAddFPExtFMAFMul = [&] (
7894*9880d681SAndroid Build Coastguard Worker SDValue X, SDValue Y, SDValue U, SDValue V, SDValue Z) {
7895*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
7896*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT, X),
7897*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT, Y),
7898*9880d681SAndroid Build Coastguard Worker DAG.getNode(PreferredFusedOpcode, SL, VT,
7899*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT, U),
7900*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT, V),
7901*9880d681SAndroid Build Coastguard Worker Z));
7902*9880d681SAndroid Build Coastguard Worker };
7903*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FP_EXTEND) {
7904*9880d681SAndroid Build Coastguard Worker SDValue N00 = N0.getOperand(0);
7905*9880d681SAndroid Build Coastguard Worker if (N00.getOpcode() == PreferredFusedOpcode) {
7906*9880d681SAndroid Build Coastguard Worker SDValue N002 = N00.getOperand(2);
7907*9880d681SAndroid Build Coastguard Worker if (N002.getOpcode() == ISD::FMUL)
7908*9880d681SAndroid Build Coastguard Worker return FoldFAddFPExtFMAFMul(N00.getOperand(0), N00.getOperand(1),
7909*9880d681SAndroid Build Coastguard Worker N002.getOperand(0), N002.getOperand(1),
7910*9880d681SAndroid Build Coastguard Worker N1);
7911*9880d681SAndroid Build Coastguard Worker }
7912*9880d681SAndroid Build Coastguard Worker }
7913*9880d681SAndroid Build Coastguard Worker
7914*9880d681SAndroid Build Coastguard Worker // fold (fadd x, (fma y, z, (fpext (fmul u, v)))
7915*9880d681SAndroid Build Coastguard Worker // -> (fma y, z, (fma (fpext u), (fpext v), x))
7916*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == PreferredFusedOpcode) {
7917*9880d681SAndroid Build Coastguard Worker SDValue N12 = N1.getOperand(2);
7918*9880d681SAndroid Build Coastguard Worker if (N12.getOpcode() == ISD::FP_EXTEND) {
7919*9880d681SAndroid Build Coastguard Worker SDValue N120 = N12.getOperand(0);
7920*9880d681SAndroid Build Coastguard Worker if (N120.getOpcode() == ISD::FMUL)
7921*9880d681SAndroid Build Coastguard Worker return FoldFAddFMAFPExtFMul(N1.getOperand(0), N1.getOperand(1),
7922*9880d681SAndroid Build Coastguard Worker N120.getOperand(0), N120.getOperand(1),
7923*9880d681SAndroid Build Coastguard Worker N0);
7924*9880d681SAndroid Build Coastguard Worker }
7925*9880d681SAndroid Build Coastguard Worker }
7926*9880d681SAndroid Build Coastguard Worker
7927*9880d681SAndroid Build Coastguard Worker // fold (fadd x, (fpext (fma y, z, (fmul u, v)))
7928*9880d681SAndroid Build Coastguard Worker // -> (fma (fpext y), (fpext z), (fma (fpext u), (fpext v), x))
7929*9880d681SAndroid Build Coastguard Worker // FIXME: This turns two single-precision and one double-precision
7930*9880d681SAndroid Build Coastguard Worker // operation into two double-precision operations, which might not be
7931*9880d681SAndroid Build Coastguard Worker // interesting for all targets, especially GPUs.
7932*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::FP_EXTEND) {
7933*9880d681SAndroid Build Coastguard Worker SDValue N10 = N1.getOperand(0);
7934*9880d681SAndroid Build Coastguard Worker if (N10.getOpcode() == PreferredFusedOpcode) {
7935*9880d681SAndroid Build Coastguard Worker SDValue N102 = N10.getOperand(2);
7936*9880d681SAndroid Build Coastguard Worker if (N102.getOpcode() == ISD::FMUL)
7937*9880d681SAndroid Build Coastguard Worker return FoldFAddFPExtFMAFMul(N10.getOperand(0), N10.getOperand(1),
7938*9880d681SAndroid Build Coastguard Worker N102.getOperand(0), N102.getOperand(1),
7939*9880d681SAndroid Build Coastguard Worker N0);
7940*9880d681SAndroid Build Coastguard Worker }
7941*9880d681SAndroid Build Coastguard Worker }
7942*9880d681SAndroid Build Coastguard Worker }
7943*9880d681SAndroid Build Coastguard Worker }
7944*9880d681SAndroid Build Coastguard Worker
7945*9880d681SAndroid Build Coastguard Worker return SDValue();
7946*9880d681SAndroid Build Coastguard Worker }
7947*9880d681SAndroid Build Coastguard Worker
7948*9880d681SAndroid Build Coastguard Worker /// Try to perform FMA combining on a given FSUB node.
visitFSUBForFMACombine(SDNode * N)7949*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFSUBForFMACombine(SDNode *N) {
7950*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
7951*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
7952*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
7953*9880d681SAndroid Build Coastguard Worker SDLoc SL(N);
7954*9880d681SAndroid Build Coastguard Worker
7955*9880d681SAndroid Build Coastguard Worker const TargetOptions &Options = DAG.getTarget().Options;
7956*9880d681SAndroid Build Coastguard Worker bool AllowFusion =
7957*9880d681SAndroid Build Coastguard Worker (Options.AllowFPOpFusion == FPOpFusion::Fast || Options.UnsafeFPMath);
7958*9880d681SAndroid Build Coastguard Worker
7959*9880d681SAndroid Build Coastguard Worker // Floating-point multiply-add with intermediate rounding.
7960*9880d681SAndroid Build Coastguard Worker bool HasFMAD = (LegalOperations && TLI.isOperationLegal(ISD::FMAD, VT));
7961*9880d681SAndroid Build Coastguard Worker
7962*9880d681SAndroid Build Coastguard Worker // Floating-point multiply-add without intermediate rounding.
7963*9880d681SAndroid Build Coastguard Worker bool HasFMA =
7964*9880d681SAndroid Build Coastguard Worker AllowFusion && TLI.isFMAFasterThanFMulAndFAdd(VT) &&
7965*9880d681SAndroid Build Coastguard Worker (!LegalOperations || TLI.isOperationLegalOrCustom(ISD::FMA, VT));
7966*9880d681SAndroid Build Coastguard Worker
7967*9880d681SAndroid Build Coastguard Worker // No valid opcode, do not combine.
7968*9880d681SAndroid Build Coastguard Worker if (!HasFMAD && !HasFMA)
7969*9880d681SAndroid Build Coastguard Worker return SDValue();
7970*9880d681SAndroid Build Coastguard Worker
7971*9880d681SAndroid Build Coastguard Worker const SelectionDAGTargetInfo *STI = DAG.getSubtarget().getSelectionDAGInfo();
7972*9880d681SAndroid Build Coastguard Worker if (AllowFusion && STI && STI->generateFMAsInMachineCombiner(OptLevel))
7973*9880d681SAndroid Build Coastguard Worker return SDValue();
7974*9880d681SAndroid Build Coastguard Worker
7975*9880d681SAndroid Build Coastguard Worker // Always prefer FMAD to FMA for precision.
7976*9880d681SAndroid Build Coastguard Worker unsigned PreferredFusedOpcode = HasFMAD ? ISD::FMAD : ISD::FMA;
7977*9880d681SAndroid Build Coastguard Worker bool Aggressive = TLI.enableAggressiveFMAFusion(VT);
7978*9880d681SAndroid Build Coastguard Worker bool LookThroughFPExt = TLI.isFPExtFree(VT);
7979*9880d681SAndroid Build Coastguard Worker
7980*9880d681SAndroid Build Coastguard Worker // fold (fsub (fmul x, y), z) -> (fma x, y, (fneg z))
7981*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FMUL &&
7982*9880d681SAndroid Build Coastguard Worker (Aggressive || N0->hasOneUse())) {
7983*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
7984*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N0.getOperand(1),
7985*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT, N1));
7986*9880d681SAndroid Build Coastguard Worker }
7987*9880d681SAndroid Build Coastguard Worker
7988*9880d681SAndroid Build Coastguard Worker // fold (fsub x, (fmul y, z)) -> (fma (fneg y), z, x)
7989*9880d681SAndroid Build Coastguard Worker // Note: Commutes FSUB operands.
7990*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::FMUL &&
7991*9880d681SAndroid Build Coastguard Worker (Aggressive || N1->hasOneUse()))
7992*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
7993*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT,
7994*9880d681SAndroid Build Coastguard Worker N1.getOperand(0)),
7995*9880d681SAndroid Build Coastguard Worker N1.getOperand(1), N0);
7996*9880d681SAndroid Build Coastguard Worker
7997*9880d681SAndroid Build Coastguard Worker // fold (fsub (fneg (fmul, x, y)), z) -> (fma (fneg x), y, (fneg z))
7998*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FNEG &&
7999*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getOpcode() == ISD::FMUL &&
8000*9880d681SAndroid Build Coastguard Worker (Aggressive || (N0->hasOneUse() && N0.getOperand(0).hasOneUse()))) {
8001*9880d681SAndroid Build Coastguard Worker SDValue N00 = N0.getOperand(0).getOperand(0);
8002*9880d681SAndroid Build Coastguard Worker SDValue N01 = N0.getOperand(0).getOperand(1);
8003*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
8004*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT, N00), N01,
8005*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT, N1));
8006*9880d681SAndroid Build Coastguard Worker }
8007*9880d681SAndroid Build Coastguard Worker
8008*9880d681SAndroid Build Coastguard Worker // Look through FP_EXTEND nodes to do more combining.
8009*9880d681SAndroid Build Coastguard Worker if (AllowFusion && LookThroughFPExt) {
8010*9880d681SAndroid Build Coastguard Worker // fold (fsub (fpext (fmul x, y)), z)
8011*9880d681SAndroid Build Coastguard Worker // -> (fma (fpext x), (fpext y), (fneg z))
8012*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FP_EXTEND) {
8013*9880d681SAndroid Build Coastguard Worker SDValue N00 = N0.getOperand(0);
8014*9880d681SAndroid Build Coastguard Worker if (N00.getOpcode() == ISD::FMUL)
8015*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
8016*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
8017*9880d681SAndroid Build Coastguard Worker N00.getOperand(0)),
8018*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
8019*9880d681SAndroid Build Coastguard Worker N00.getOperand(1)),
8020*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT, N1));
8021*9880d681SAndroid Build Coastguard Worker }
8022*9880d681SAndroid Build Coastguard Worker
8023*9880d681SAndroid Build Coastguard Worker // fold (fsub x, (fpext (fmul y, z)))
8024*9880d681SAndroid Build Coastguard Worker // -> (fma (fneg (fpext y)), (fpext z), x)
8025*9880d681SAndroid Build Coastguard Worker // Note: Commutes FSUB operands.
8026*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::FP_EXTEND) {
8027*9880d681SAndroid Build Coastguard Worker SDValue N10 = N1.getOperand(0);
8028*9880d681SAndroid Build Coastguard Worker if (N10.getOpcode() == ISD::FMUL)
8029*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
8030*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT,
8031*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
8032*9880d681SAndroid Build Coastguard Worker N10.getOperand(0))),
8033*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
8034*9880d681SAndroid Build Coastguard Worker N10.getOperand(1)),
8035*9880d681SAndroid Build Coastguard Worker N0);
8036*9880d681SAndroid Build Coastguard Worker }
8037*9880d681SAndroid Build Coastguard Worker
8038*9880d681SAndroid Build Coastguard Worker // fold (fsub (fpext (fneg (fmul, x, y))), z)
8039*9880d681SAndroid Build Coastguard Worker // -> (fneg (fma (fpext x), (fpext y), z))
8040*9880d681SAndroid Build Coastguard Worker // Note: This could be removed with appropriate canonicalization of the
8041*9880d681SAndroid Build Coastguard Worker // input expression into (fneg (fadd (fpext (fmul, x, y)), z). However, the
8042*9880d681SAndroid Build Coastguard Worker // orthogonal flags -fp-contract=fast and -enable-unsafe-fp-math prevent
8043*9880d681SAndroid Build Coastguard Worker // from implementing the canonicalization in visitFSUB.
8044*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FP_EXTEND) {
8045*9880d681SAndroid Build Coastguard Worker SDValue N00 = N0.getOperand(0);
8046*9880d681SAndroid Build Coastguard Worker if (N00.getOpcode() == ISD::FNEG) {
8047*9880d681SAndroid Build Coastguard Worker SDValue N000 = N00.getOperand(0);
8048*9880d681SAndroid Build Coastguard Worker if (N000.getOpcode() == ISD::FMUL) {
8049*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FNEG, SL, VT,
8050*9880d681SAndroid Build Coastguard Worker DAG.getNode(PreferredFusedOpcode, SL, VT,
8051*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
8052*9880d681SAndroid Build Coastguard Worker N000.getOperand(0)),
8053*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
8054*9880d681SAndroid Build Coastguard Worker N000.getOperand(1)),
8055*9880d681SAndroid Build Coastguard Worker N1));
8056*9880d681SAndroid Build Coastguard Worker }
8057*9880d681SAndroid Build Coastguard Worker }
8058*9880d681SAndroid Build Coastguard Worker }
8059*9880d681SAndroid Build Coastguard Worker
8060*9880d681SAndroid Build Coastguard Worker // fold (fsub (fneg (fpext (fmul, x, y))), z)
8061*9880d681SAndroid Build Coastguard Worker // -> (fneg (fma (fpext x)), (fpext y), z)
8062*9880d681SAndroid Build Coastguard Worker // Note: This could be removed with appropriate canonicalization of the
8063*9880d681SAndroid Build Coastguard Worker // input expression into (fneg (fadd (fpext (fmul, x, y)), z). However, the
8064*9880d681SAndroid Build Coastguard Worker // orthogonal flags -fp-contract=fast and -enable-unsafe-fp-math prevent
8065*9880d681SAndroid Build Coastguard Worker // from implementing the canonicalization in visitFSUB.
8066*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FNEG) {
8067*9880d681SAndroid Build Coastguard Worker SDValue N00 = N0.getOperand(0);
8068*9880d681SAndroid Build Coastguard Worker if (N00.getOpcode() == ISD::FP_EXTEND) {
8069*9880d681SAndroid Build Coastguard Worker SDValue N000 = N00.getOperand(0);
8070*9880d681SAndroid Build Coastguard Worker if (N000.getOpcode() == ISD::FMUL) {
8071*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FNEG, SL, VT,
8072*9880d681SAndroid Build Coastguard Worker DAG.getNode(PreferredFusedOpcode, SL, VT,
8073*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
8074*9880d681SAndroid Build Coastguard Worker N000.getOperand(0)),
8075*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
8076*9880d681SAndroid Build Coastguard Worker N000.getOperand(1)),
8077*9880d681SAndroid Build Coastguard Worker N1));
8078*9880d681SAndroid Build Coastguard Worker }
8079*9880d681SAndroid Build Coastguard Worker }
8080*9880d681SAndroid Build Coastguard Worker }
8081*9880d681SAndroid Build Coastguard Worker
8082*9880d681SAndroid Build Coastguard Worker }
8083*9880d681SAndroid Build Coastguard Worker
8084*9880d681SAndroid Build Coastguard Worker // More folding opportunities when target permits.
8085*9880d681SAndroid Build Coastguard Worker if ((AllowFusion || HasFMAD) && Aggressive) {
8086*9880d681SAndroid Build Coastguard Worker // fold (fsub (fma x, y, (fmul u, v)), z)
8087*9880d681SAndroid Build Coastguard Worker // -> (fma x, y (fma u, v, (fneg z)))
8088*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == PreferredFusedOpcode &&
8089*9880d681SAndroid Build Coastguard Worker N0.getOperand(2).getOpcode() == ISD::FMUL) {
8090*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
8091*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N0.getOperand(1),
8092*9880d681SAndroid Build Coastguard Worker DAG.getNode(PreferredFusedOpcode, SL, VT,
8093*9880d681SAndroid Build Coastguard Worker N0.getOperand(2).getOperand(0),
8094*9880d681SAndroid Build Coastguard Worker N0.getOperand(2).getOperand(1),
8095*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT,
8096*9880d681SAndroid Build Coastguard Worker N1)));
8097*9880d681SAndroid Build Coastguard Worker }
8098*9880d681SAndroid Build Coastguard Worker
8099*9880d681SAndroid Build Coastguard Worker // fold (fsub x, (fma y, z, (fmul u, v)))
8100*9880d681SAndroid Build Coastguard Worker // -> (fma (fneg y), z, (fma (fneg u), v, x))
8101*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == PreferredFusedOpcode &&
8102*9880d681SAndroid Build Coastguard Worker N1.getOperand(2).getOpcode() == ISD::FMUL) {
8103*9880d681SAndroid Build Coastguard Worker SDValue N20 = N1.getOperand(2).getOperand(0);
8104*9880d681SAndroid Build Coastguard Worker SDValue N21 = N1.getOperand(2).getOperand(1);
8105*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
8106*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT,
8107*9880d681SAndroid Build Coastguard Worker N1.getOperand(0)),
8108*9880d681SAndroid Build Coastguard Worker N1.getOperand(1),
8109*9880d681SAndroid Build Coastguard Worker DAG.getNode(PreferredFusedOpcode, SL, VT,
8110*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT, N20),
8111*9880d681SAndroid Build Coastguard Worker
8112*9880d681SAndroid Build Coastguard Worker N21, N0));
8113*9880d681SAndroid Build Coastguard Worker }
8114*9880d681SAndroid Build Coastguard Worker
8115*9880d681SAndroid Build Coastguard Worker if (AllowFusion && LookThroughFPExt) {
8116*9880d681SAndroid Build Coastguard Worker // fold (fsub (fma x, y, (fpext (fmul u, v))), z)
8117*9880d681SAndroid Build Coastguard Worker // -> (fma x, y (fma (fpext u), (fpext v), (fneg z)))
8118*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == PreferredFusedOpcode) {
8119*9880d681SAndroid Build Coastguard Worker SDValue N02 = N0.getOperand(2);
8120*9880d681SAndroid Build Coastguard Worker if (N02.getOpcode() == ISD::FP_EXTEND) {
8121*9880d681SAndroid Build Coastguard Worker SDValue N020 = N02.getOperand(0);
8122*9880d681SAndroid Build Coastguard Worker if (N020.getOpcode() == ISD::FMUL)
8123*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
8124*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N0.getOperand(1),
8125*9880d681SAndroid Build Coastguard Worker DAG.getNode(PreferredFusedOpcode, SL, VT,
8126*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
8127*9880d681SAndroid Build Coastguard Worker N020.getOperand(0)),
8128*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
8129*9880d681SAndroid Build Coastguard Worker N020.getOperand(1)),
8130*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT,
8131*9880d681SAndroid Build Coastguard Worker N1)));
8132*9880d681SAndroid Build Coastguard Worker }
8133*9880d681SAndroid Build Coastguard Worker }
8134*9880d681SAndroid Build Coastguard Worker
8135*9880d681SAndroid Build Coastguard Worker // fold (fsub (fpext (fma x, y, (fmul u, v))), z)
8136*9880d681SAndroid Build Coastguard Worker // -> (fma (fpext x), (fpext y),
8137*9880d681SAndroid Build Coastguard Worker // (fma (fpext u), (fpext v), (fneg z)))
8138*9880d681SAndroid Build Coastguard Worker // FIXME: This turns two single-precision and one double-precision
8139*9880d681SAndroid Build Coastguard Worker // operation into two double-precision operations, which might not be
8140*9880d681SAndroid Build Coastguard Worker // interesting for all targets, especially GPUs.
8141*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FP_EXTEND) {
8142*9880d681SAndroid Build Coastguard Worker SDValue N00 = N0.getOperand(0);
8143*9880d681SAndroid Build Coastguard Worker if (N00.getOpcode() == PreferredFusedOpcode) {
8144*9880d681SAndroid Build Coastguard Worker SDValue N002 = N00.getOperand(2);
8145*9880d681SAndroid Build Coastguard Worker if (N002.getOpcode() == ISD::FMUL)
8146*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
8147*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
8148*9880d681SAndroid Build Coastguard Worker N00.getOperand(0)),
8149*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
8150*9880d681SAndroid Build Coastguard Worker N00.getOperand(1)),
8151*9880d681SAndroid Build Coastguard Worker DAG.getNode(PreferredFusedOpcode, SL, VT,
8152*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
8153*9880d681SAndroid Build Coastguard Worker N002.getOperand(0)),
8154*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
8155*9880d681SAndroid Build Coastguard Worker N002.getOperand(1)),
8156*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT,
8157*9880d681SAndroid Build Coastguard Worker N1)));
8158*9880d681SAndroid Build Coastguard Worker }
8159*9880d681SAndroid Build Coastguard Worker }
8160*9880d681SAndroid Build Coastguard Worker
8161*9880d681SAndroid Build Coastguard Worker // fold (fsub x, (fma y, z, (fpext (fmul u, v))))
8162*9880d681SAndroid Build Coastguard Worker // -> (fma (fneg y), z, (fma (fneg (fpext u)), (fpext v), x))
8163*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == PreferredFusedOpcode &&
8164*9880d681SAndroid Build Coastguard Worker N1.getOperand(2).getOpcode() == ISD::FP_EXTEND) {
8165*9880d681SAndroid Build Coastguard Worker SDValue N120 = N1.getOperand(2).getOperand(0);
8166*9880d681SAndroid Build Coastguard Worker if (N120.getOpcode() == ISD::FMUL) {
8167*9880d681SAndroid Build Coastguard Worker SDValue N1200 = N120.getOperand(0);
8168*9880d681SAndroid Build Coastguard Worker SDValue N1201 = N120.getOperand(1);
8169*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
8170*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT, N1.getOperand(0)),
8171*9880d681SAndroid Build Coastguard Worker N1.getOperand(1),
8172*9880d681SAndroid Build Coastguard Worker DAG.getNode(PreferredFusedOpcode, SL, VT,
8173*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT,
8174*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL,
8175*9880d681SAndroid Build Coastguard Worker VT, N1200)),
8176*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
8177*9880d681SAndroid Build Coastguard Worker N1201),
8178*9880d681SAndroid Build Coastguard Worker N0));
8179*9880d681SAndroid Build Coastguard Worker }
8180*9880d681SAndroid Build Coastguard Worker }
8181*9880d681SAndroid Build Coastguard Worker
8182*9880d681SAndroid Build Coastguard Worker // fold (fsub x, (fpext (fma y, z, (fmul u, v))))
8183*9880d681SAndroid Build Coastguard Worker // -> (fma (fneg (fpext y)), (fpext z),
8184*9880d681SAndroid Build Coastguard Worker // (fma (fneg (fpext u)), (fpext v), x))
8185*9880d681SAndroid Build Coastguard Worker // FIXME: This turns two single-precision and one double-precision
8186*9880d681SAndroid Build Coastguard Worker // operation into two double-precision operations, which might not be
8187*9880d681SAndroid Build Coastguard Worker // interesting for all targets, especially GPUs.
8188*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::FP_EXTEND &&
8189*9880d681SAndroid Build Coastguard Worker N1.getOperand(0).getOpcode() == PreferredFusedOpcode) {
8190*9880d681SAndroid Build Coastguard Worker SDValue N100 = N1.getOperand(0).getOperand(0);
8191*9880d681SAndroid Build Coastguard Worker SDValue N101 = N1.getOperand(0).getOperand(1);
8192*9880d681SAndroid Build Coastguard Worker SDValue N102 = N1.getOperand(0).getOperand(2);
8193*9880d681SAndroid Build Coastguard Worker if (N102.getOpcode() == ISD::FMUL) {
8194*9880d681SAndroid Build Coastguard Worker SDValue N1020 = N102.getOperand(0);
8195*9880d681SAndroid Build Coastguard Worker SDValue N1021 = N102.getOperand(1);
8196*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
8197*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT,
8198*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
8199*9880d681SAndroid Build Coastguard Worker N100)),
8200*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT, N101),
8201*9880d681SAndroid Build Coastguard Worker DAG.getNode(PreferredFusedOpcode, SL, VT,
8202*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT,
8203*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL,
8204*9880d681SAndroid Build Coastguard Worker VT, N1020)),
8205*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_EXTEND, SL, VT,
8206*9880d681SAndroid Build Coastguard Worker N1021),
8207*9880d681SAndroid Build Coastguard Worker N0));
8208*9880d681SAndroid Build Coastguard Worker }
8209*9880d681SAndroid Build Coastguard Worker }
8210*9880d681SAndroid Build Coastguard Worker }
8211*9880d681SAndroid Build Coastguard Worker }
8212*9880d681SAndroid Build Coastguard Worker
8213*9880d681SAndroid Build Coastguard Worker return SDValue();
8214*9880d681SAndroid Build Coastguard Worker }
8215*9880d681SAndroid Build Coastguard Worker
8216*9880d681SAndroid Build Coastguard Worker /// Try to perform FMA combining on a given FMUL node.
visitFMULForFMACombine(SDNode * N)8217*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFMULForFMACombine(SDNode *N) {
8218*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
8219*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
8220*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
8221*9880d681SAndroid Build Coastguard Worker SDLoc SL(N);
8222*9880d681SAndroid Build Coastguard Worker
8223*9880d681SAndroid Build Coastguard Worker assert(N->getOpcode() == ISD::FMUL && "Expected FMUL Operation");
8224*9880d681SAndroid Build Coastguard Worker
8225*9880d681SAndroid Build Coastguard Worker const TargetOptions &Options = DAG.getTarget().Options;
8226*9880d681SAndroid Build Coastguard Worker bool AllowFusion =
8227*9880d681SAndroid Build Coastguard Worker (Options.AllowFPOpFusion == FPOpFusion::Fast || Options.UnsafeFPMath);
8228*9880d681SAndroid Build Coastguard Worker
8229*9880d681SAndroid Build Coastguard Worker // Floating-point multiply-add with intermediate rounding.
8230*9880d681SAndroid Build Coastguard Worker bool HasFMAD = (LegalOperations && TLI.isOperationLegal(ISD::FMAD, VT));
8231*9880d681SAndroid Build Coastguard Worker
8232*9880d681SAndroid Build Coastguard Worker // Floating-point multiply-add without intermediate rounding.
8233*9880d681SAndroid Build Coastguard Worker bool HasFMA =
8234*9880d681SAndroid Build Coastguard Worker AllowFusion && TLI.isFMAFasterThanFMulAndFAdd(VT) &&
8235*9880d681SAndroid Build Coastguard Worker (!LegalOperations || TLI.isOperationLegalOrCustom(ISD::FMA, VT));
8236*9880d681SAndroid Build Coastguard Worker
8237*9880d681SAndroid Build Coastguard Worker // No valid opcode, do not combine.
8238*9880d681SAndroid Build Coastguard Worker if (!HasFMAD && !HasFMA)
8239*9880d681SAndroid Build Coastguard Worker return SDValue();
8240*9880d681SAndroid Build Coastguard Worker
8241*9880d681SAndroid Build Coastguard Worker // Always prefer FMAD to FMA for precision.
8242*9880d681SAndroid Build Coastguard Worker unsigned PreferredFusedOpcode = HasFMAD ? ISD::FMAD : ISD::FMA;
8243*9880d681SAndroid Build Coastguard Worker bool Aggressive = TLI.enableAggressiveFMAFusion(VT);
8244*9880d681SAndroid Build Coastguard Worker
8245*9880d681SAndroid Build Coastguard Worker // fold (fmul (fadd x, +1.0), y) -> (fma x, y, y)
8246*9880d681SAndroid Build Coastguard Worker // fold (fmul (fadd x, -1.0), y) -> (fma x, y, (fneg y))
8247*9880d681SAndroid Build Coastguard Worker auto FuseFADD = [&](SDValue X, SDValue Y) {
8248*9880d681SAndroid Build Coastguard Worker if (X.getOpcode() == ISD::FADD && (Aggressive || X->hasOneUse())) {
8249*9880d681SAndroid Build Coastguard Worker auto XC1 = isConstOrConstSplatFP(X.getOperand(1));
8250*9880d681SAndroid Build Coastguard Worker if (XC1 && XC1->isExactlyValue(+1.0))
8251*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT, X.getOperand(0), Y, Y);
8252*9880d681SAndroid Build Coastguard Worker if (XC1 && XC1->isExactlyValue(-1.0))
8253*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT, X.getOperand(0), Y,
8254*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT, Y));
8255*9880d681SAndroid Build Coastguard Worker }
8256*9880d681SAndroid Build Coastguard Worker return SDValue();
8257*9880d681SAndroid Build Coastguard Worker };
8258*9880d681SAndroid Build Coastguard Worker
8259*9880d681SAndroid Build Coastguard Worker if (SDValue FMA = FuseFADD(N0, N1))
8260*9880d681SAndroid Build Coastguard Worker return FMA;
8261*9880d681SAndroid Build Coastguard Worker if (SDValue FMA = FuseFADD(N1, N0))
8262*9880d681SAndroid Build Coastguard Worker return FMA;
8263*9880d681SAndroid Build Coastguard Worker
8264*9880d681SAndroid Build Coastguard Worker // fold (fmul (fsub +1.0, x), y) -> (fma (fneg x), y, y)
8265*9880d681SAndroid Build Coastguard Worker // fold (fmul (fsub -1.0, x), y) -> (fma (fneg x), y, (fneg y))
8266*9880d681SAndroid Build Coastguard Worker // fold (fmul (fsub x, +1.0), y) -> (fma x, y, (fneg y))
8267*9880d681SAndroid Build Coastguard Worker // fold (fmul (fsub x, -1.0), y) -> (fma x, y, y)
8268*9880d681SAndroid Build Coastguard Worker auto FuseFSUB = [&](SDValue X, SDValue Y) {
8269*9880d681SAndroid Build Coastguard Worker if (X.getOpcode() == ISD::FSUB && (Aggressive || X->hasOneUse())) {
8270*9880d681SAndroid Build Coastguard Worker auto XC0 = isConstOrConstSplatFP(X.getOperand(0));
8271*9880d681SAndroid Build Coastguard Worker if (XC0 && XC0->isExactlyValue(+1.0))
8272*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
8273*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT, X.getOperand(1)), Y,
8274*9880d681SAndroid Build Coastguard Worker Y);
8275*9880d681SAndroid Build Coastguard Worker if (XC0 && XC0->isExactlyValue(-1.0))
8276*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT,
8277*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT, X.getOperand(1)), Y,
8278*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT, Y));
8279*9880d681SAndroid Build Coastguard Worker
8280*9880d681SAndroid Build Coastguard Worker auto XC1 = isConstOrConstSplatFP(X.getOperand(1));
8281*9880d681SAndroid Build Coastguard Worker if (XC1 && XC1->isExactlyValue(+1.0))
8282*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT, X.getOperand(0), Y,
8283*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SL, VT, Y));
8284*9880d681SAndroid Build Coastguard Worker if (XC1 && XC1->isExactlyValue(-1.0))
8285*9880d681SAndroid Build Coastguard Worker return DAG.getNode(PreferredFusedOpcode, SL, VT, X.getOperand(0), Y, Y);
8286*9880d681SAndroid Build Coastguard Worker }
8287*9880d681SAndroid Build Coastguard Worker return SDValue();
8288*9880d681SAndroid Build Coastguard Worker };
8289*9880d681SAndroid Build Coastguard Worker
8290*9880d681SAndroid Build Coastguard Worker if (SDValue FMA = FuseFSUB(N0, N1))
8291*9880d681SAndroid Build Coastguard Worker return FMA;
8292*9880d681SAndroid Build Coastguard Worker if (SDValue FMA = FuseFSUB(N1, N0))
8293*9880d681SAndroid Build Coastguard Worker return FMA;
8294*9880d681SAndroid Build Coastguard Worker
8295*9880d681SAndroid Build Coastguard Worker return SDValue();
8296*9880d681SAndroid Build Coastguard Worker }
8297*9880d681SAndroid Build Coastguard Worker
visitFADD(SDNode * N)8298*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFADD(SDNode *N) {
8299*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
8300*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
8301*9880d681SAndroid Build Coastguard Worker bool N0CFP = isConstantFPBuildVectorOrConstantFP(N0);
8302*9880d681SAndroid Build Coastguard Worker bool N1CFP = isConstantFPBuildVectorOrConstantFP(N1);
8303*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
8304*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
8305*9880d681SAndroid Build Coastguard Worker const TargetOptions &Options = DAG.getTarget().Options;
8306*9880d681SAndroid Build Coastguard Worker const SDNodeFlags *Flags = &cast<BinaryWithFlagsSDNode>(N)->Flags;
8307*9880d681SAndroid Build Coastguard Worker
8308*9880d681SAndroid Build Coastguard Worker // fold vector ops
8309*9880d681SAndroid Build Coastguard Worker if (VT.isVector())
8310*9880d681SAndroid Build Coastguard Worker if (SDValue FoldedVOp = SimplifyVBinOp(N))
8311*9880d681SAndroid Build Coastguard Worker return FoldedVOp;
8312*9880d681SAndroid Build Coastguard Worker
8313*9880d681SAndroid Build Coastguard Worker // fold (fadd c1, c2) -> c1 + c2
8314*9880d681SAndroid Build Coastguard Worker if (N0CFP && N1CFP)
8315*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FADD, DL, VT, N0, N1, Flags);
8316*9880d681SAndroid Build Coastguard Worker
8317*9880d681SAndroid Build Coastguard Worker // canonicalize constant to RHS
8318*9880d681SAndroid Build Coastguard Worker if (N0CFP && !N1CFP)
8319*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FADD, DL, VT, N1, N0, Flags);
8320*9880d681SAndroid Build Coastguard Worker
8321*9880d681SAndroid Build Coastguard Worker // fold (fadd A, (fneg B)) -> (fsub A, B)
8322*9880d681SAndroid Build Coastguard Worker if ((!LegalOperations || TLI.isOperationLegalOrCustom(ISD::FSUB, VT)) &&
8323*9880d681SAndroid Build Coastguard Worker isNegatibleForFree(N1, LegalOperations, TLI, &Options) == 2)
8324*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FSUB, DL, VT, N0,
8325*9880d681SAndroid Build Coastguard Worker GetNegatedExpression(N1, DAG, LegalOperations), Flags);
8326*9880d681SAndroid Build Coastguard Worker
8327*9880d681SAndroid Build Coastguard Worker // fold (fadd (fneg A), B) -> (fsub B, A)
8328*9880d681SAndroid Build Coastguard Worker if ((!LegalOperations || TLI.isOperationLegalOrCustom(ISD::FSUB, VT)) &&
8329*9880d681SAndroid Build Coastguard Worker isNegatibleForFree(N0, LegalOperations, TLI, &Options) == 2)
8330*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FSUB, DL, VT, N1,
8331*9880d681SAndroid Build Coastguard Worker GetNegatedExpression(N0, DAG, LegalOperations), Flags);
8332*9880d681SAndroid Build Coastguard Worker
8333*9880d681SAndroid Build Coastguard Worker // If 'unsafe math' is enabled, fold lots of things.
8334*9880d681SAndroid Build Coastguard Worker if (Options.UnsafeFPMath) {
8335*9880d681SAndroid Build Coastguard Worker // No FP constant should be created after legalization as Instruction
8336*9880d681SAndroid Build Coastguard Worker // Selection pass has a hard time dealing with FP constants.
8337*9880d681SAndroid Build Coastguard Worker bool AllowNewConst = (Level < AfterLegalizeDAG);
8338*9880d681SAndroid Build Coastguard Worker
8339*9880d681SAndroid Build Coastguard Worker // fold (fadd A, 0) -> A
8340*9880d681SAndroid Build Coastguard Worker if (ConstantFPSDNode *N1C = isConstOrConstSplatFP(N1))
8341*9880d681SAndroid Build Coastguard Worker if (N1C->isZero())
8342*9880d681SAndroid Build Coastguard Worker return N0;
8343*9880d681SAndroid Build Coastguard Worker
8344*9880d681SAndroid Build Coastguard Worker // fold (fadd (fadd x, c1), c2) -> (fadd x, (fadd c1, c2))
8345*9880d681SAndroid Build Coastguard Worker if (N1CFP && N0.getOpcode() == ISD::FADD && N0.getNode()->hasOneUse() &&
8346*9880d681SAndroid Build Coastguard Worker isConstantFPBuildVectorOrConstantFP(N0.getOperand(1)))
8347*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FADD, DL, VT, N0.getOperand(0),
8348*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FADD, DL, VT, N0.getOperand(1), N1,
8349*9880d681SAndroid Build Coastguard Worker Flags),
8350*9880d681SAndroid Build Coastguard Worker Flags);
8351*9880d681SAndroid Build Coastguard Worker
8352*9880d681SAndroid Build Coastguard Worker // If allowed, fold (fadd (fneg x), x) -> 0.0
8353*9880d681SAndroid Build Coastguard Worker if (AllowNewConst && N0.getOpcode() == ISD::FNEG && N0.getOperand(0) == N1)
8354*9880d681SAndroid Build Coastguard Worker return DAG.getConstantFP(0.0, DL, VT);
8355*9880d681SAndroid Build Coastguard Worker
8356*9880d681SAndroid Build Coastguard Worker // If allowed, fold (fadd x, (fneg x)) -> 0.0
8357*9880d681SAndroid Build Coastguard Worker if (AllowNewConst && N1.getOpcode() == ISD::FNEG && N1.getOperand(0) == N0)
8358*9880d681SAndroid Build Coastguard Worker return DAG.getConstantFP(0.0, DL, VT);
8359*9880d681SAndroid Build Coastguard Worker
8360*9880d681SAndroid Build Coastguard Worker // We can fold chains of FADD's of the same value into multiplications.
8361*9880d681SAndroid Build Coastguard Worker // This transform is not safe in general because we are reducing the number
8362*9880d681SAndroid Build Coastguard Worker // of rounding steps.
8363*9880d681SAndroid Build Coastguard Worker if (TLI.isOperationLegalOrCustom(ISD::FMUL, VT) && !N0CFP && !N1CFP) {
8364*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FMUL) {
8365*9880d681SAndroid Build Coastguard Worker bool CFP00 = isConstantFPBuildVectorOrConstantFP(N0.getOperand(0));
8366*9880d681SAndroid Build Coastguard Worker bool CFP01 = isConstantFPBuildVectorOrConstantFP(N0.getOperand(1));
8367*9880d681SAndroid Build Coastguard Worker
8368*9880d681SAndroid Build Coastguard Worker // (fadd (fmul x, c), x) -> (fmul x, c+1)
8369*9880d681SAndroid Build Coastguard Worker if (CFP01 && !CFP00 && N0.getOperand(0) == N1) {
8370*9880d681SAndroid Build Coastguard Worker SDValue NewCFP = DAG.getNode(ISD::FADD, DL, VT, N0.getOperand(1),
8371*9880d681SAndroid Build Coastguard Worker DAG.getConstantFP(1.0, DL, VT), Flags);
8372*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT, N1, NewCFP, Flags);
8373*9880d681SAndroid Build Coastguard Worker }
8374*9880d681SAndroid Build Coastguard Worker
8375*9880d681SAndroid Build Coastguard Worker // (fadd (fmul x, c), (fadd x, x)) -> (fmul x, c+2)
8376*9880d681SAndroid Build Coastguard Worker if (CFP01 && !CFP00 && N1.getOpcode() == ISD::FADD &&
8377*9880d681SAndroid Build Coastguard Worker N1.getOperand(0) == N1.getOperand(1) &&
8378*9880d681SAndroid Build Coastguard Worker N0.getOperand(0) == N1.getOperand(0)) {
8379*9880d681SAndroid Build Coastguard Worker SDValue NewCFP = DAG.getNode(ISD::FADD, DL, VT, N0.getOperand(1),
8380*9880d681SAndroid Build Coastguard Worker DAG.getConstantFP(2.0, DL, VT), Flags);
8381*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT, N0.getOperand(0), NewCFP, Flags);
8382*9880d681SAndroid Build Coastguard Worker }
8383*9880d681SAndroid Build Coastguard Worker }
8384*9880d681SAndroid Build Coastguard Worker
8385*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::FMUL) {
8386*9880d681SAndroid Build Coastguard Worker bool CFP10 = isConstantFPBuildVectorOrConstantFP(N1.getOperand(0));
8387*9880d681SAndroid Build Coastguard Worker bool CFP11 = isConstantFPBuildVectorOrConstantFP(N1.getOperand(1));
8388*9880d681SAndroid Build Coastguard Worker
8389*9880d681SAndroid Build Coastguard Worker // (fadd x, (fmul x, c)) -> (fmul x, c+1)
8390*9880d681SAndroid Build Coastguard Worker if (CFP11 && !CFP10 && N1.getOperand(0) == N0) {
8391*9880d681SAndroid Build Coastguard Worker SDValue NewCFP = DAG.getNode(ISD::FADD, DL, VT, N1.getOperand(1),
8392*9880d681SAndroid Build Coastguard Worker DAG.getConstantFP(1.0, DL, VT), Flags);
8393*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT, N0, NewCFP, Flags);
8394*9880d681SAndroid Build Coastguard Worker }
8395*9880d681SAndroid Build Coastguard Worker
8396*9880d681SAndroid Build Coastguard Worker // (fadd (fadd x, x), (fmul x, c)) -> (fmul x, c+2)
8397*9880d681SAndroid Build Coastguard Worker if (CFP11 && !CFP10 && N0.getOpcode() == ISD::FADD &&
8398*9880d681SAndroid Build Coastguard Worker N0.getOperand(0) == N0.getOperand(1) &&
8399*9880d681SAndroid Build Coastguard Worker N1.getOperand(0) == N0.getOperand(0)) {
8400*9880d681SAndroid Build Coastguard Worker SDValue NewCFP = DAG.getNode(ISD::FADD, DL, VT, N1.getOperand(1),
8401*9880d681SAndroid Build Coastguard Worker DAG.getConstantFP(2.0, DL, VT), Flags);
8402*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT, N1.getOperand(0), NewCFP, Flags);
8403*9880d681SAndroid Build Coastguard Worker }
8404*9880d681SAndroid Build Coastguard Worker }
8405*9880d681SAndroid Build Coastguard Worker
8406*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FADD && AllowNewConst) {
8407*9880d681SAndroid Build Coastguard Worker bool CFP00 = isConstantFPBuildVectorOrConstantFP(N0.getOperand(0));
8408*9880d681SAndroid Build Coastguard Worker // (fadd (fadd x, x), x) -> (fmul x, 3.0)
8409*9880d681SAndroid Build Coastguard Worker if (!CFP00 && N0.getOperand(0) == N0.getOperand(1) &&
8410*9880d681SAndroid Build Coastguard Worker (N0.getOperand(0) == N1)) {
8411*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT,
8412*9880d681SAndroid Build Coastguard Worker N1, DAG.getConstantFP(3.0, DL, VT), Flags);
8413*9880d681SAndroid Build Coastguard Worker }
8414*9880d681SAndroid Build Coastguard Worker }
8415*9880d681SAndroid Build Coastguard Worker
8416*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::FADD && AllowNewConst) {
8417*9880d681SAndroid Build Coastguard Worker bool CFP10 = isConstantFPBuildVectorOrConstantFP(N1.getOperand(0));
8418*9880d681SAndroid Build Coastguard Worker // (fadd x, (fadd x, x)) -> (fmul x, 3.0)
8419*9880d681SAndroid Build Coastguard Worker if (!CFP10 && N1.getOperand(0) == N1.getOperand(1) &&
8420*9880d681SAndroid Build Coastguard Worker N1.getOperand(0) == N0) {
8421*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT,
8422*9880d681SAndroid Build Coastguard Worker N0, DAG.getConstantFP(3.0, DL, VT), Flags);
8423*9880d681SAndroid Build Coastguard Worker }
8424*9880d681SAndroid Build Coastguard Worker }
8425*9880d681SAndroid Build Coastguard Worker
8426*9880d681SAndroid Build Coastguard Worker // (fadd (fadd x, x), (fadd x, x)) -> (fmul x, 4.0)
8427*9880d681SAndroid Build Coastguard Worker if (AllowNewConst &&
8428*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::FADD && N1.getOpcode() == ISD::FADD &&
8429*9880d681SAndroid Build Coastguard Worker N0.getOperand(0) == N0.getOperand(1) &&
8430*9880d681SAndroid Build Coastguard Worker N1.getOperand(0) == N1.getOperand(1) &&
8431*9880d681SAndroid Build Coastguard Worker N0.getOperand(0) == N1.getOperand(0)) {
8432*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT, N0.getOperand(0),
8433*9880d681SAndroid Build Coastguard Worker DAG.getConstantFP(4.0, DL, VT), Flags);
8434*9880d681SAndroid Build Coastguard Worker }
8435*9880d681SAndroid Build Coastguard Worker }
8436*9880d681SAndroid Build Coastguard Worker } // enable-unsafe-fp-math
8437*9880d681SAndroid Build Coastguard Worker
8438*9880d681SAndroid Build Coastguard Worker // FADD -> FMA combines:
8439*9880d681SAndroid Build Coastguard Worker if (SDValue Fused = visitFADDForFMACombine(N)) {
8440*9880d681SAndroid Build Coastguard Worker AddToWorklist(Fused.getNode());
8441*9880d681SAndroid Build Coastguard Worker return Fused;
8442*9880d681SAndroid Build Coastguard Worker }
8443*9880d681SAndroid Build Coastguard Worker return SDValue();
8444*9880d681SAndroid Build Coastguard Worker }
8445*9880d681SAndroid Build Coastguard Worker
visitFSUB(SDNode * N)8446*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFSUB(SDNode *N) {
8447*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
8448*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
8449*9880d681SAndroid Build Coastguard Worker ConstantFPSDNode *N0CFP = isConstOrConstSplatFP(N0);
8450*9880d681SAndroid Build Coastguard Worker ConstantFPSDNode *N1CFP = isConstOrConstSplatFP(N1);
8451*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
8452*9880d681SAndroid Build Coastguard Worker SDLoc dl(N);
8453*9880d681SAndroid Build Coastguard Worker const TargetOptions &Options = DAG.getTarget().Options;
8454*9880d681SAndroid Build Coastguard Worker const SDNodeFlags *Flags = &cast<BinaryWithFlagsSDNode>(N)->Flags;
8455*9880d681SAndroid Build Coastguard Worker
8456*9880d681SAndroid Build Coastguard Worker // fold vector ops
8457*9880d681SAndroid Build Coastguard Worker if (VT.isVector())
8458*9880d681SAndroid Build Coastguard Worker if (SDValue FoldedVOp = SimplifyVBinOp(N))
8459*9880d681SAndroid Build Coastguard Worker return FoldedVOp;
8460*9880d681SAndroid Build Coastguard Worker
8461*9880d681SAndroid Build Coastguard Worker // fold (fsub c1, c2) -> c1-c2
8462*9880d681SAndroid Build Coastguard Worker if (N0CFP && N1CFP)
8463*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FSUB, dl, VT, N0, N1, Flags);
8464*9880d681SAndroid Build Coastguard Worker
8465*9880d681SAndroid Build Coastguard Worker // fold (fsub A, (fneg B)) -> (fadd A, B)
8466*9880d681SAndroid Build Coastguard Worker if (isNegatibleForFree(N1, LegalOperations, TLI, &Options))
8467*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FADD, dl, VT, N0,
8468*9880d681SAndroid Build Coastguard Worker GetNegatedExpression(N1, DAG, LegalOperations), Flags);
8469*9880d681SAndroid Build Coastguard Worker
8470*9880d681SAndroid Build Coastguard Worker // If 'unsafe math' is enabled, fold lots of things.
8471*9880d681SAndroid Build Coastguard Worker if (Options.UnsafeFPMath) {
8472*9880d681SAndroid Build Coastguard Worker // (fsub A, 0) -> A
8473*9880d681SAndroid Build Coastguard Worker if (N1CFP && N1CFP->isZero())
8474*9880d681SAndroid Build Coastguard Worker return N0;
8475*9880d681SAndroid Build Coastguard Worker
8476*9880d681SAndroid Build Coastguard Worker // (fsub 0, B) -> -B
8477*9880d681SAndroid Build Coastguard Worker if (N0CFP && N0CFP->isZero()) {
8478*9880d681SAndroid Build Coastguard Worker if (isNegatibleForFree(N1, LegalOperations, TLI, &Options))
8479*9880d681SAndroid Build Coastguard Worker return GetNegatedExpression(N1, DAG, LegalOperations);
8480*9880d681SAndroid Build Coastguard Worker if (!LegalOperations || TLI.isOperationLegal(ISD::FNEG, VT))
8481*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FNEG, dl, VT, N1);
8482*9880d681SAndroid Build Coastguard Worker }
8483*9880d681SAndroid Build Coastguard Worker
8484*9880d681SAndroid Build Coastguard Worker // (fsub x, x) -> 0.0
8485*9880d681SAndroid Build Coastguard Worker if (N0 == N1)
8486*9880d681SAndroid Build Coastguard Worker return DAG.getConstantFP(0.0f, dl, VT);
8487*9880d681SAndroid Build Coastguard Worker
8488*9880d681SAndroid Build Coastguard Worker // (fsub x, (fadd x, y)) -> (fneg y)
8489*9880d681SAndroid Build Coastguard Worker // (fsub x, (fadd y, x)) -> (fneg y)
8490*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::FADD) {
8491*9880d681SAndroid Build Coastguard Worker SDValue N10 = N1->getOperand(0);
8492*9880d681SAndroid Build Coastguard Worker SDValue N11 = N1->getOperand(1);
8493*9880d681SAndroid Build Coastguard Worker
8494*9880d681SAndroid Build Coastguard Worker if (N10 == N0 && isNegatibleForFree(N11, LegalOperations, TLI, &Options))
8495*9880d681SAndroid Build Coastguard Worker return GetNegatedExpression(N11, DAG, LegalOperations);
8496*9880d681SAndroid Build Coastguard Worker
8497*9880d681SAndroid Build Coastguard Worker if (N11 == N0 && isNegatibleForFree(N10, LegalOperations, TLI, &Options))
8498*9880d681SAndroid Build Coastguard Worker return GetNegatedExpression(N10, DAG, LegalOperations);
8499*9880d681SAndroid Build Coastguard Worker }
8500*9880d681SAndroid Build Coastguard Worker }
8501*9880d681SAndroid Build Coastguard Worker
8502*9880d681SAndroid Build Coastguard Worker // FSUB -> FMA combines:
8503*9880d681SAndroid Build Coastguard Worker if (SDValue Fused = visitFSUBForFMACombine(N)) {
8504*9880d681SAndroid Build Coastguard Worker AddToWorklist(Fused.getNode());
8505*9880d681SAndroid Build Coastguard Worker return Fused;
8506*9880d681SAndroid Build Coastguard Worker }
8507*9880d681SAndroid Build Coastguard Worker
8508*9880d681SAndroid Build Coastguard Worker return SDValue();
8509*9880d681SAndroid Build Coastguard Worker }
8510*9880d681SAndroid Build Coastguard Worker
visitFMUL(SDNode * N)8511*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFMUL(SDNode *N) {
8512*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
8513*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
8514*9880d681SAndroid Build Coastguard Worker ConstantFPSDNode *N0CFP = isConstOrConstSplatFP(N0);
8515*9880d681SAndroid Build Coastguard Worker ConstantFPSDNode *N1CFP = isConstOrConstSplatFP(N1);
8516*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
8517*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
8518*9880d681SAndroid Build Coastguard Worker const TargetOptions &Options = DAG.getTarget().Options;
8519*9880d681SAndroid Build Coastguard Worker const SDNodeFlags *Flags = &cast<BinaryWithFlagsSDNode>(N)->Flags;
8520*9880d681SAndroid Build Coastguard Worker
8521*9880d681SAndroid Build Coastguard Worker // fold vector ops
8522*9880d681SAndroid Build Coastguard Worker if (VT.isVector()) {
8523*9880d681SAndroid Build Coastguard Worker // This just handles C1 * C2 for vectors. Other vector folds are below.
8524*9880d681SAndroid Build Coastguard Worker if (SDValue FoldedVOp = SimplifyVBinOp(N))
8525*9880d681SAndroid Build Coastguard Worker return FoldedVOp;
8526*9880d681SAndroid Build Coastguard Worker }
8527*9880d681SAndroid Build Coastguard Worker
8528*9880d681SAndroid Build Coastguard Worker // fold (fmul c1, c2) -> c1*c2
8529*9880d681SAndroid Build Coastguard Worker if (N0CFP && N1CFP)
8530*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT, N0, N1, Flags);
8531*9880d681SAndroid Build Coastguard Worker
8532*9880d681SAndroid Build Coastguard Worker // canonicalize constant to RHS
8533*9880d681SAndroid Build Coastguard Worker if (isConstantFPBuildVectorOrConstantFP(N0) &&
8534*9880d681SAndroid Build Coastguard Worker !isConstantFPBuildVectorOrConstantFP(N1))
8535*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT, N1, N0, Flags);
8536*9880d681SAndroid Build Coastguard Worker
8537*9880d681SAndroid Build Coastguard Worker // fold (fmul A, 1.0) -> A
8538*9880d681SAndroid Build Coastguard Worker if (N1CFP && N1CFP->isExactlyValue(1.0))
8539*9880d681SAndroid Build Coastguard Worker return N0;
8540*9880d681SAndroid Build Coastguard Worker
8541*9880d681SAndroid Build Coastguard Worker if (Options.UnsafeFPMath) {
8542*9880d681SAndroid Build Coastguard Worker // fold (fmul A, 0) -> 0
8543*9880d681SAndroid Build Coastguard Worker if (N1CFP && N1CFP->isZero())
8544*9880d681SAndroid Build Coastguard Worker return N1;
8545*9880d681SAndroid Build Coastguard Worker
8546*9880d681SAndroid Build Coastguard Worker // fold (fmul (fmul x, c1), c2) -> (fmul x, (fmul c1, c2))
8547*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FMUL) {
8548*9880d681SAndroid Build Coastguard Worker // Fold scalars or any vector constants (not just splats).
8549*9880d681SAndroid Build Coastguard Worker // This fold is done in general by InstCombine, but extra fmul insts
8550*9880d681SAndroid Build Coastguard Worker // may have been generated during lowering.
8551*9880d681SAndroid Build Coastguard Worker SDValue N00 = N0.getOperand(0);
8552*9880d681SAndroid Build Coastguard Worker SDValue N01 = N0.getOperand(1);
8553*9880d681SAndroid Build Coastguard Worker auto *BV1 = dyn_cast<BuildVectorSDNode>(N1);
8554*9880d681SAndroid Build Coastguard Worker auto *BV00 = dyn_cast<BuildVectorSDNode>(N00);
8555*9880d681SAndroid Build Coastguard Worker auto *BV01 = dyn_cast<BuildVectorSDNode>(N01);
8556*9880d681SAndroid Build Coastguard Worker
8557*9880d681SAndroid Build Coastguard Worker // Check 1: Make sure that the first operand of the inner multiply is NOT
8558*9880d681SAndroid Build Coastguard Worker // a constant. Otherwise, we may induce infinite looping.
8559*9880d681SAndroid Build Coastguard Worker if (!(isConstOrConstSplatFP(N00) || (BV00 && BV00->isConstant()))) {
8560*9880d681SAndroid Build Coastguard Worker // Check 2: Make sure that the second operand of the inner multiply and
8561*9880d681SAndroid Build Coastguard Worker // the second operand of the outer multiply are constants.
8562*9880d681SAndroid Build Coastguard Worker if ((N1CFP && isConstOrConstSplatFP(N01)) ||
8563*9880d681SAndroid Build Coastguard Worker (BV1 && BV01 && BV1->isConstant() && BV01->isConstant())) {
8564*9880d681SAndroid Build Coastguard Worker SDValue MulConsts = DAG.getNode(ISD::FMUL, DL, VT, N01, N1, Flags);
8565*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT, N00, MulConsts, Flags);
8566*9880d681SAndroid Build Coastguard Worker }
8567*9880d681SAndroid Build Coastguard Worker }
8568*9880d681SAndroid Build Coastguard Worker }
8569*9880d681SAndroid Build Coastguard Worker
8570*9880d681SAndroid Build Coastguard Worker // fold (fmul (fadd x, x), c) -> (fmul x, (fmul 2.0, c))
8571*9880d681SAndroid Build Coastguard Worker // Undo the fmul 2.0, x -> fadd x, x transformation, since if it occurs
8572*9880d681SAndroid Build Coastguard Worker // during an early run of DAGCombiner can prevent folding with fmuls
8573*9880d681SAndroid Build Coastguard Worker // inserted during lowering.
8574*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FADD &&
8575*9880d681SAndroid Build Coastguard Worker (N0.getOperand(0) == N0.getOperand(1)) &&
8576*9880d681SAndroid Build Coastguard Worker N0.hasOneUse()) {
8577*9880d681SAndroid Build Coastguard Worker const SDValue Two = DAG.getConstantFP(2.0, DL, VT);
8578*9880d681SAndroid Build Coastguard Worker SDValue MulConsts = DAG.getNode(ISD::FMUL, DL, VT, Two, N1, Flags);
8579*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT, N0.getOperand(0), MulConsts, Flags);
8580*9880d681SAndroid Build Coastguard Worker }
8581*9880d681SAndroid Build Coastguard Worker }
8582*9880d681SAndroid Build Coastguard Worker
8583*9880d681SAndroid Build Coastguard Worker // fold (fmul X, 2.0) -> (fadd X, X)
8584*9880d681SAndroid Build Coastguard Worker if (N1CFP && N1CFP->isExactlyValue(+2.0))
8585*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FADD, DL, VT, N0, N0, Flags);
8586*9880d681SAndroid Build Coastguard Worker
8587*9880d681SAndroid Build Coastguard Worker // fold (fmul X, -1.0) -> (fneg X)
8588*9880d681SAndroid Build Coastguard Worker if (N1CFP && N1CFP->isExactlyValue(-1.0))
8589*9880d681SAndroid Build Coastguard Worker if (!LegalOperations || TLI.isOperationLegal(ISD::FNEG, VT))
8590*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FNEG, DL, VT, N0);
8591*9880d681SAndroid Build Coastguard Worker
8592*9880d681SAndroid Build Coastguard Worker // fold (fmul (fneg X), (fneg Y)) -> (fmul X, Y)
8593*9880d681SAndroid Build Coastguard Worker if (char LHSNeg = isNegatibleForFree(N0, LegalOperations, TLI, &Options)) {
8594*9880d681SAndroid Build Coastguard Worker if (char RHSNeg = isNegatibleForFree(N1, LegalOperations, TLI, &Options)) {
8595*9880d681SAndroid Build Coastguard Worker // Both can be negated for free, check to see if at least one is cheaper
8596*9880d681SAndroid Build Coastguard Worker // negated.
8597*9880d681SAndroid Build Coastguard Worker if (LHSNeg == 2 || RHSNeg == 2)
8598*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT,
8599*9880d681SAndroid Build Coastguard Worker GetNegatedExpression(N0, DAG, LegalOperations),
8600*9880d681SAndroid Build Coastguard Worker GetNegatedExpression(N1, DAG, LegalOperations),
8601*9880d681SAndroid Build Coastguard Worker Flags);
8602*9880d681SAndroid Build Coastguard Worker }
8603*9880d681SAndroid Build Coastguard Worker }
8604*9880d681SAndroid Build Coastguard Worker
8605*9880d681SAndroid Build Coastguard Worker // FMUL -> FMA combines:
8606*9880d681SAndroid Build Coastguard Worker if (SDValue Fused = visitFMULForFMACombine(N)) {
8607*9880d681SAndroid Build Coastguard Worker AddToWorklist(Fused.getNode());
8608*9880d681SAndroid Build Coastguard Worker return Fused;
8609*9880d681SAndroid Build Coastguard Worker }
8610*9880d681SAndroid Build Coastguard Worker
8611*9880d681SAndroid Build Coastguard Worker return SDValue();
8612*9880d681SAndroid Build Coastguard Worker }
8613*9880d681SAndroid Build Coastguard Worker
visitFMA(SDNode * N)8614*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFMA(SDNode *N) {
8615*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
8616*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
8617*9880d681SAndroid Build Coastguard Worker SDValue N2 = N->getOperand(2);
8618*9880d681SAndroid Build Coastguard Worker ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N0);
8619*9880d681SAndroid Build Coastguard Worker ConstantFPSDNode *N1CFP = dyn_cast<ConstantFPSDNode>(N1);
8620*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
8621*9880d681SAndroid Build Coastguard Worker SDLoc dl(N);
8622*9880d681SAndroid Build Coastguard Worker const TargetOptions &Options = DAG.getTarget().Options;
8623*9880d681SAndroid Build Coastguard Worker
8624*9880d681SAndroid Build Coastguard Worker // Constant fold FMA.
8625*9880d681SAndroid Build Coastguard Worker if (isa<ConstantFPSDNode>(N0) &&
8626*9880d681SAndroid Build Coastguard Worker isa<ConstantFPSDNode>(N1) &&
8627*9880d681SAndroid Build Coastguard Worker isa<ConstantFPSDNode>(N2)) {
8628*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMA, dl, VT, N0, N1, N2);
8629*9880d681SAndroid Build Coastguard Worker }
8630*9880d681SAndroid Build Coastguard Worker
8631*9880d681SAndroid Build Coastguard Worker if (Options.UnsafeFPMath) {
8632*9880d681SAndroid Build Coastguard Worker if (N0CFP && N0CFP->isZero())
8633*9880d681SAndroid Build Coastguard Worker return N2;
8634*9880d681SAndroid Build Coastguard Worker if (N1CFP && N1CFP->isZero())
8635*9880d681SAndroid Build Coastguard Worker return N2;
8636*9880d681SAndroid Build Coastguard Worker }
8637*9880d681SAndroid Build Coastguard Worker // TODO: The FMA node should have flags that propagate to these nodes.
8638*9880d681SAndroid Build Coastguard Worker if (N0CFP && N0CFP->isExactlyValue(1.0))
8639*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FADD, SDLoc(N), VT, N1, N2);
8640*9880d681SAndroid Build Coastguard Worker if (N1CFP && N1CFP->isExactlyValue(1.0))
8641*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FADD, SDLoc(N), VT, N0, N2);
8642*9880d681SAndroid Build Coastguard Worker
8643*9880d681SAndroid Build Coastguard Worker // Canonicalize (fma c, x, y) -> (fma x, c, y)
8644*9880d681SAndroid Build Coastguard Worker if (isConstantFPBuildVectorOrConstantFP(N0) &&
8645*9880d681SAndroid Build Coastguard Worker !isConstantFPBuildVectorOrConstantFP(N1))
8646*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMA, SDLoc(N), VT, N1, N0, N2);
8647*9880d681SAndroid Build Coastguard Worker
8648*9880d681SAndroid Build Coastguard Worker // TODO: FMA nodes should have flags that propagate to the created nodes.
8649*9880d681SAndroid Build Coastguard Worker // For now, create a Flags object for use with all unsafe math transforms.
8650*9880d681SAndroid Build Coastguard Worker SDNodeFlags Flags;
8651*9880d681SAndroid Build Coastguard Worker Flags.setUnsafeAlgebra(true);
8652*9880d681SAndroid Build Coastguard Worker
8653*9880d681SAndroid Build Coastguard Worker if (Options.UnsafeFPMath) {
8654*9880d681SAndroid Build Coastguard Worker // (fma x, c1, (fmul x, c2)) -> (fmul x, c1+c2)
8655*9880d681SAndroid Build Coastguard Worker if (N2.getOpcode() == ISD::FMUL && N0 == N2.getOperand(0) &&
8656*9880d681SAndroid Build Coastguard Worker isConstantFPBuildVectorOrConstantFP(N1) &&
8657*9880d681SAndroid Build Coastguard Worker isConstantFPBuildVectorOrConstantFP(N2.getOperand(1))) {
8658*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, dl, VT, N0,
8659*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FADD, dl, VT, N1, N2.getOperand(1),
8660*9880d681SAndroid Build Coastguard Worker &Flags), &Flags);
8661*9880d681SAndroid Build Coastguard Worker }
8662*9880d681SAndroid Build Coastguard Worker
8663*9880d681SAndroid Build Coastguard Worker // (fma (fmul x, c1), c2, y) -> (fma x, c1*c2, y)
8664*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FMUL &&
8665*9880d681SAndroid Build Coastguard Worker isConstantFPBuildVectorOrConstantFP(N1) &&
8666*9880d681SAndroid Build Coastguard Worker isConstantFPBuildVectorOrConstantFP(N0.getOperand(1))) {
8667*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMA, dl, VT,
8668*9880d681SAndroid Build Coastguard Worker N0.getOperand(0),
8669*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FMUL, dl, VT, N1, N0.getOperand(1),
8670*9880d681SAndroid Build Coastguard Worker &Flags),
8671*9880d681SAndroid Build Coastguard Worker N2);
8672*9880d681SAndroid Build Coastguard Worker }
8673*9880d681SAndroid Build Coastguard Worker }
8674*9880d681SAndroid Build Coastguard Worker
8675*9880d681SAndroid Build Coastguard Worker // (fma x, 1, y) -> (fadd x, y)
8676*9880d681SAndroid Build Coastguard Worker // (fma x, -1, y) -> (fadd (fneg x), y)
8677*9880d681SAndroid Build Coastguard Worker if (N1CFP) {
8678*9880d681SAndroid Build Coastguard Worker if (N1CFP->isExactlyValue(1.0))
8679*9880d681SAndroid Build Coastguard Worker // TODO: The FMA node should have flags that propagate to this node.
8680*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FADD, dl, VT, N0, N2);
8681*9880d681SAndroid Build Coastguard Worker
8682*9880d681SAndroid Build Coastguard Worker if (N1CFP->isExactlyValue(-1.0) &&
8683*9880d681SAndroid Build Coastguard Worker (!LegalOperations || TLI.isOperationLegal(ISD::FNEG, VT))) {
8684*9880d681SAndroid Build Coastguard Worker SDValue RHSNeg = DAG.getNode(ISD::FNEG, dl, VT, N0);
8685*9880d681SAndroid Build Coastguard Worker AddToWorklist(RHSNeg.getNode());
8686*9880d681SAndroid Build Coastguard Worker // TODO: The FMA node should have flags that propagate to this node.
8687*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FADD, dl, VT, N2, RHSNeg);
8688*9880d681SAndroid Build Coastguard Worker }
8689*9880d681SAndroid Build Coastguard Worker }
8690*9880d681SAndroid Build Coastguard Worker
8691*9880d681SAndroid Build Coastguard Worker if (Options.UnsafeFPMath) {
8692*9880d681SAndroid Build Coastguard Worker // (fma x, c, x) -> (fmul x, (c+1))
8693*9880d681SAndroid Build Coastguard Worker if (N1CFP && N0 == N2) {
8694*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, dl, VT, N0,
8695*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FADD, dl, VT,
8696*9880d681SAndroid Build Coastguard Worker N1, DAG.getConstantFP(1.0, dl, VT),
8697*9880d681SAndroid Build Coastguard Worker &Flags), &Flags);
8698*9880d681SAndroid Build Coastguard Worker }
8699*9880d681SAndroid Build Coastguard Worker
8700*9880d681SAndroid Build Coastguard Worker // (fma x, c, (fneg x)) -> (fmul x, (c-1))
8701*9880d681SAndroid Build Coastguard Worker if (N1CFP && N2.getOpcode() == ISD::FNEG && N2.getOperand(0) == N0) {
8702*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, dl, VT, N0,
8703*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FADD, dl, VT,
8704*9880d681SAndroid Build Coastguard Worker N1, DAG.getConstantFP(-1.0, dl, VT),
8705*9880d681SAndroid Build Coastguard Worker &Flags), &Flags);
8706*9880d681SAndroid Build Coastguard Worker }
8707*9880d681SAndroid Build Coastguard Worker }
8708*9880d681SAndroid Build Coastguard Worker
8709*9880d681SAndroid Build Coastguard Worker return SDValue();
8710*9880d681SAndroid Build Coastguard Worker }
8711*9880d681SAndroid Build Coastguard Worker
8712*9880d681SAndroid Build Coastguard Worker // Combine multiple FDIVs with the same divisor into multiple FMULs by the
8713*9880d681SAndroid Build Coastguard Worker // reciprocal.
8714*9880d681SAndroid Build Coastguard Worker // E.g., (a / D; b / D;) -> (recip = 1.0 / D; a * recip; b * recip)
8715*9880d681SAndroid Build Coastguard Worker // Notice that this is not always beneficial. One reason is different target
8716*9880d681SAndroid Build Coastguard Worker // may have different costs for FDIV and FMUL, so sometimes the cost of two
8717*9880d681SAndroid Build Coastguard Worker // FDIVs may be lower than the cost of one FDIV and two FMULs. Another reason
8718*9880d681SAndroid Build Coastguard Worker // is the critical path is increased from "one FDIV" to "one FDIV + one FMUL".
combineRepeatedFPDivisors(SDNode * N)8719*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::combineRepeatedFPDivisors(SDNode *N) {
8720*9880d681SAndroid Build Coastguard Worker bool UnsafeMath = DAG.getTarget().Options.UnsafeFPMath;
8721*9880d681SAndroid Build Coastguard Worker const SDNodeFlags *Flags = N->getFlags();
8722*9880d681SAndroid Build Coastguard Worker if (!UnsafeMath && !Flags->hasAllowReciprocal())
8723*9880d681SAndroid Build Coastguard Worker return SDValue();
8724*9880d681SAndroid Build Coastguard Worker
8725*9880d681SAndroid Build Coastguard Worker // Skip if current node is a reciprocal.
8726*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
8727*9880d681SAndroid Build Coastguard Worker ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N0);
8728*9880d681SAndroid Build Coastguard Worker if (N0CFP && N0CFP->isExactlyValue(1.0))
8729*9880d681SAndroid Build Coastguard Worker return SDValue();
8730*9880d681SAndroid Build Coastguard Worker
8731*9880d681SAndroid Build Coastguard Worker // Exit early if the target does not want this transform or if there can't
8732*9880d681SAndroid Build Coastguard Worker // possibly be enough uses of the divisor to make the transform worthwhile.
8733*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
8734*9880d681SAndroid Build Coastguard Worker unsigned MinUses = TLI.combineRepeatedFPDivisors();
8735*9880d681SAndroid Build Coastguard Worker if (!MinUses || N1->use_size() < MinUses)
8736*9880d681SAndroid Build Coastguard Worker return SDValue();
8737*9880d681SAndroid Build Coastguard Worker
8738*9880d681SAndroid Build Coastguard Worker // Find all FDIV users of the same divisor.
8739*9880d681SAndroid Build Coastguard Worker // Use a set because duplicates may be present in the user list.
8740*9880d681SAndroid Build Coastguard Worker SetVector<SDNode *> Users;
8741*9880d681SAndroid Build Coastguard Worker for (auto *U : N1->uses()) {
8742*9880d681SAndroid Build Coastguard Worker if (U->getOpcode() == ISD::FDIV && U->getOperand(1) == N1) {
8743*9880d681SAndroid Build Coastguard Worker // This division is eligible for optimization only if global unsafe math
8744*9880d681SAndroid Build Coastguard Worker // is enabled or if this division allows reciprocal formation.
8745*9880d681SAndroid Build Coastguard Worker if (UnsafeMath || U->getFlags()->hasAllowReciprocal())
8746*9880d681SAndroid Build Coastguard Worker Users.insert(U);
8747*9880d681SAndroid Build Coastguard Worker }
8748*9880d681SAndroid Build Coastguard Worker }
8749*9880d681SAndroid Build Coastguard Worker
8750*9880d681SAndroid Build Coastguard Worker // Now that we have the actual number of divisor uses, make sure it meets
8751*9880d681SAndroid Build Coastguard Worker // the minimum threshold specified by the target.
8752*9880d681SAndroid Build Coastguard Worker if (Users.size() < MinUses)
8753*9880d681SAndroid Build Coastguard Worker return SDValue();
8754*9880d681SAndroid Build Coastguard Worker
8755*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
8756*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
8757*9880d681SAndroid Build Coastguard Worker SDValue FPOne = DAG.getConstantFP(1.0, DL, VT);
8758*9880d681SAndroid Build Coastguard Worker SDValue Reciprocal = DAG.getNode(ISD::FDIV, DL, VT, FPOne, N1, Flags);
8759*9880d681SAndroid Build Coastguard Worker
8760*9880d681SAndroid Build Coastguard Worker // Dividend / Divisor -> Dividend * Reciprocal
8761*9880d681SAndroid Build Coastguard Worker for (auto *U : Users) {
8762*9880d681SAndroid Build Coastguard Worker SDValue Dividend = U->getOperand(0);
8763*9880d681SAndroid Build Coastguard Worker if (Dividend != FPOne) {
8764*9880d681SAndroid Build Coastguard Worker SDValue NewNode = DAG.getNode(ISD::FMUL, SDLoc(U), VT, Dividend,
8765*9880d681SAndroid Build Coastguard Worker Reciprocal, Flags);
8766*9880d681SAndroid Build Coastguard Worker CombineTo(U, NewNode);
8767*9880d681SAndroid Build Coastguard Worker } else if (U != Reciprocal.getNode()) {
8768*9880d681SAndroid Build Coastguard Worker // In the absence of fast-math-flags, this user node is always the
8769*9880d681SAndroid Build Coastguard Worker // same node as Reciprocal, but with FMF they may be different nodes.
8770*9880d681SAndroid Build Coastguard Worker CombineTo(U, Reciprocal);
8771*9880d681SAndroid Build Coastguard Worker }
8772*9880d681SAndroid Build Coastguard Worker }
8773*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // N was replaced.
8774*9880d681SAndroid Build Coastguard Worker }
8775*9880d681SAndroid Build Coastguard Worker
visitFDIV(SDNode * N)8776*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFDIV(SDNode *N) {
8777*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
8778*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
8779*9880d681SAndroid Build Coastguard Worker ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N0);
8780*9880d681SAndroid Build Coastguard Worker ConstantFPSDNode *N1CFP = dyn_cast<ConstantFPSDNode>(N1);
8781*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
8782*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
8783*9880d681SAndroid Build Coastguard Worker const TargetOptions &Options = DAG.getTarget().Options;
8784*9880d681SAndroid Build Coastguard Worker SDNodeFlags *Flags = &cast<BinaryWithFlagsSDNode>(N)->Flags;
8785*9880d681SAndroid Build Coastguard Worker
8786*9880d681SAndroid Build Coastguard Worker // fold vector ops
8787*9880d681SAndroid Build Coastguard Worker if (VT.isVector())
8788*9880d681SAndroid Build Coastguard Worker if (SDValue FoldedVOp = SimplifyVBinOp(N))
8789*9880d681SAndroid Build Coastguard Worker return FoldedVOp;
8790*9880d681SAndroid Build Coastguard Worker
8791*9880d681SAndroid Build Coastguard Worker // fold (fdiv c1, c2) -> c1/c2
8792*9880d681SAndroid Build Coastguard Worker if (N0CFP && N1CFP)
8793*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FDIV, SDLoc(N), VT, N0, N1, Flags);
8794*9880d681SAndroid Build Coastguard Worker
8795*9880d681SAndroid Build Coastguard Worker if (Options.UnsafeFPMath) {
8796*9880d681SAndroid Build Coastguard Worker // fold (fdiv X, c2) -> fmul X, 1/c2 if losing precision is acceptable.
8797*9880d681SAndroid Build Coastguard Worker if (N1CFP) {
8798*9880d681SAndroid Build Coastguard Worker // Compute the reciprocal 1.0 / c2.
8799*9880d681SAndroid Build Coastguard Worker const APFloat &N1APF = N1CFP->getValueAPF();
8800*9880d681SAndroid Build Coastguard Worker APFloat Recip(N1APF.getSemantics(), 1); // 1.0
8801*9880d681SAndroid Build Coastguard Worker APFloat::opStatus st = Recip.divide(N1APF, APFloat::rmNearestTiesToEven);
8802*9880d681SAndroid Build Coastguard Worker // Only do the transform if the reciprocal is a legal fp immediate that
8803*9880d681SAndroid Build Coastguard Worker // isn't too nasty (eg NaN, denormal, ...).
8804*9880d681SAndroid Build Coastguard Worker if ((st == APFloat::opOK || st == APFloat::opInexact) && // Not too nasty
8805*9880d681SAndroid Build Coastguard Worker (!LegalOperations ||
8806*9880d681SAndroid Build Coastguard Worker // FIXME: custom lowering of ConstantFP might fail (see e.g. ARM
8807*9880d681SAndroid Build Coastguard Worker // backend)... we should handle this gracefully after Legalize.
8808*9880d681SAndroid Build Coastguard Worker // TLI.isOperationLegalOrCustom(llvm::ISD::ConstantFP, VT) ||
8809*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegal(llvm::ISD::ConstantFP, VT) ||
8810*9880d681SAndroid Build Coastguard Worker TLI.isFPImmLegal(Recip, VT)))
8811*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT, N0,
8812*9880d681SAndroid Build Coastguard Worker DAG.getConstantFP(Recip, DL, VT), Flags);
8813*9880d681SAndroid Build Coastguard Worker }
8814*9880d681SAndroid Build Coastguard Worker
8815*9880d681SAndroid Build Coastguard Worker // If this FDIV is part of a reciprocal square root, it may be folded
8816*9880d681SAndroid Build Coastguard Worker // into a target-specific square root estimate instruction.
8817*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::FSQRT) {
8818*9880d681SAndroid Build Coastguard Worker if (SDValue RV = buildRsqrtEstimate(N1.getOperand(0), Flags)) {
8819*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT, N0, RV, Flags);
8820*9880d681SAndroid Build Coastguard Worker }
8821*9880d681SAndroid Build Coastguard Worker } else if (N1.getOpcode() == ISD::FP_EXTEND &&
8822*9880d681SAndroid Build Coastguard Worker N1.getOperand(0).getOpcode() == ISD::FSQRT) {
8823*9880d681SAndroid Build Coastguard Worker if (SDValue RV = buildRsqrtEstimate(N1.getOperand(0).getOperand(0),
8824*9880d681SAndroid Build Coastguard Worker Flags)) {
8825*9880d681SAndroid Build Coastguard Worker RV = DAG.getNode(ISD::FP_EXTEND, SDLoc(N1), VT, RV);
8826*9880d681SAndroid Build Coastguard Worker AddToWorklist(RV.getNode());
8827*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT, N0, RV, Flags);
8828*9880d681SAndroid Build Coastguard Worker }
8829*9880d681SAndroid Build Coastguard Worker } else if (N1.getOpcode() == ISD::FP_ROUND &&
8830*9880d681SAndroid Build Coastguard Worker N1.getOperand(0).getOpcode() == ISD::FSQRT) {
8831*9880d681SAndroid Build Coastguard Worker if (SDValue RV = buildRsqrtEstimate(N1.getOperand(0).getOperand(0),
8832*9880d681SAndroid Build Coastguard Worker Flags)) {
8833*9880d681SAndroid Build Coastguard Worker RV = DAG.getNode(ISD::FP_ROUND, SDLoc(N1), VT, RV, N1.getOperand(1));
8834*9880d681SAndroid Build Coastguard Worker AddToWorklist(RV.getNode());
8835*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT, N0, RV, Flags);
8836*9880d681SAndroid Build Coastguard Worker }
8837*9880d681SAndroid Build Coastguard Worker } else if (N1.getOpcode() == ISD::FMUL) {
8838*9880d681SAndroid Build Coastguard Worker // Look through an FMUL. Even though this won't remove the FDIV directly,
8839*9880d681SAndroid Build Coastguard Worker // it's still worthwhile to get rid of the FSQRT if possible.
8840*9880d681SAndroid Build Coastguard Worker SDValue SqrtOp;
8841*9880d681SAndroid Build Coastguard Worker SDValue OtherOp;
8842*9880d681SAndroid Build Coastguard Worker if (N1.getOperand(0).getOpcode() == ISD::FSQRT) {
8843*9880d681SAndroid Build Coastguard Worker SqrtOp = N1.getOperand(0);
8844*9880d681SAndroid Build Coastguard Worker OtherOp = N1.getOperand(1);
8845*9880d681SAndroid Build Coastguard Worker } else if (N1.getOperand(1).getOpcode() == ISD::FSQRT) {
8846*9880d681SAndroid Build Coastguard Worker SqrtOp = N1.getOperand(1);
8847*9880d681SAndroid Build Coastguard Worker OtherOp = N1.getOperand(0);
8848*9880d681SAndroid Build Coastguard Worker }
8849*9880d681SAndroid Build Coastguard Worker if (SqrtOp.getNode()) {
8850*9880d681SAndroid Build Coastguard Worker // We found a FSQRT, so try to make this fold:
8851*9880d681SAndroid Build Coastguard Worker // x / (y * sqrt(z)) -> x * (rsqrt(z) / y)
8852*9880d681SAndroid Build Coastguard Worker if (SDValue RV = buildRsqrtEstimate(SqrtOp.getOperand(0), Flags)) {
8853*9880d681SAndroid Build Coastguard Worker RV = DAG.getNode(ISD::FDIV, SDLoc(N1), VT, RV, OtherOp, Flags);
8854*9880d681SAndroid Build Coastguard Worker AddToWorklist(RV.getNode());
8855*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT, N0, RV, Flags);
8856*9880d681SAndroid Build Coastguard Worker }
8857*9880d681SAndroid Build Coastguard Worker }
8858*9880d681SAndroid Build Coastguard Worker }
8859*9880d681SAndroid Build Coastguard Worker
8860*9880d681SAndroid Build Coastguard Worker // Fold into a reciprocal estimate and multiply instead of a real divide.
8861*9880d681SAndroid Build Coastguard Worker if (SDValue RV = BuildReciprocalEstimate(N1, Flags)) {
8862*9880d681SAndroid Build Coastguard Worker AddToWorklist(RV.getNode());
8863*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, DL, VT, N0, RV, Flags);
8864*9880d681SAndroid Build Coastguard Worker }
8865*9880d681SAndroid Build Coastguard Worker }
8866*9880d681SAndroid Build Coastguard Worker
8867*9880d681SAndroid Build Coastguard Worker // (fdiv (fneg X), (fneg Y)) -> (fdiv X, Y)
8868*9880d681SAndroid Build Coastguard Worker if (char LHSNeg = isNegatibleForFree(N0, LegalOperations, TLI, &Options)) {
8869*9880d681SAndroid Build Coastguard Worker if (char RHSNeg = isNegatibleForFree(N1, LegalOperations, TLI, &Options)) {
8870*9880d681SAndroid Build Coastguard Worker // Both can be negated for free, check to see if at least one is cheaper
8871*9880d681SAndroid Build Coastguard Worker // negated.
8872*9880d681SAndroid Build Coastguard Worker if (LHSNeg == 2 || RHSNeg == 2)
8873*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FDIV, SDLoc(N), VT,
8874*9880d681SAndroid Build Coastguard Worker GetNegatedExpression(N0, DAG, LegalOperations),
8875*9880d681SAndroid Build Coastguard Worker GetNegatedExpression(N1, DAG, LegalOperations),
8876*9880d681SAndroid Build Coastguard Worker Flags);
8877*9880d681SAndroid Build Coastguard Worker }
8878*9880d681SAndroid Build Coastguard Worker }
8879*9880d681SAndroid Build Coastguard Worker
8880*9880d681SAndroid Build Coastguard Worker if (SDValue CombineRepeatedDivisors = combineRepeatedFPDivisors(N))
8881*9880d681SAndroid Build Coastguard Worker return CombineRepeatedDivisors;
8882*9880d681SAndroid Build Coastguard Worker
8883*9880d681SAndroid Build Coastguard Worker return SDValue();
8884*9880d681SAndroid Build Coastguard Worker }
8885*9880d681SAndroid Build Coastguard Worker
visitFREM(SDNode * N)8886*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFREM(SDNode *N) {
8887*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
8888*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
8889*9880d681SAndroid Build Coastguard Worker ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N0);
8890*9880d681SAndroid Build Coastguard Worker ConstantFPSDNode *N1CFP = dyn_cast<ConstantFPSDNode>(N1);
8891*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
8892*9880d681SAndroid Build Coastguard Worker
8893*9880d681SAndroid Build Coastguard Worker // fold (frem c1, c2) -> fmod(c1,c2)
8894*9880d681SAndroid Build Coastguard Worker if (N0CFP && N1CFP)
8895*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FREM, SDLoc(N), VT, N0, N1,
8896*9880d681SAndroid Build Coastguard Worker &cast<BinaryWithFlagsSDNode>(N)->Flags);
8897*9880d681SAndroid Build Coastguard Worker
8898*9880d681SAndroid Build Coastguard Worker return SDValue();
8899*9880d681SAndroid Build Coastguard Worker }
8900*9880d681SAndroid Build Coastguard Worker
visitFSQRT(SDNode * N)8901*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFSQRT(SDNode *N) {
8902*9880d681SAndroid Build Coastguard Worker if (!DAG.getTarget().Options.UnsafeFPMath || TLI.isFsqrtCheap())
8903*9880d681SAndroid Build Coastguard Worker return SDValue();
8904*9880d681SAndroid Build Coastguard Worker
8905*9880d681SAndroid Build Coastguard Worker // TODO: FSQRT nodes should have flags that propagate to the created nodes.
8906*9880d681SAndroid Build Coastguard Worker // For now, create a Flags object for use with all unsafe math transforms.
8907*9880d681SAndroid Build Coastguard Worker SDNodeFlags Flags;
8908*9880d681SAndroid Build Coastguard Worker Flags.setUnsafeAlgebra(true);
8909*9880d681SAndroid Build Coastguard Worker return buildSqrtEstimate(N->getOperand(0), &Flags);
8910*9880d681SAndroid Build Coastguard Worker }
8911*9880d681SAndroid Build Coastguard Worker
8912*9880d681SAndroid Build Coastguard Worker /// copysign(x, fp_extend(y)) -> copysign(x, y)
8913*9880d681SAndroid Build Coastguard Worker /// copysign(x, fp_round(y)) -> copysign(x, y)
CanCombineFCOPYSIGN_EXTEND_ROUND(SDNode * N)8914*9880d681SAndroid Build Coastguard Worker static inline bool CanCombineFCOPYSIGN_EXTEND_ROUND(SDNode *N) {
8915*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
8916*9880d681SAndroid Build Coastguard Worker if ((N1.getOpcode() == ISD::FP_EXTEND ||
8917*9880d681SAndroid Build Coastguard Worker N1.getOpcode() == ISD::FP_ROUND)) {
8918*9880d681SAndroid Build Coastguard Worker // Do not optimize out type conversion of f128 type yet.
8919*9880d681SAndroid Build Coastguard Worker // For some targets like x86_64, configuration is changed to keep one f128
8920*9880d681SAndroid Build Coastguard Worker // value in one SSE register, but instruction selection cannot handle
8921*9880d681SAndroid Build Coastguard Worker // FCOPYSIGN on SSE registers yet.
8922*9880d681SAndroid Build Coastguard Worker EVT N1VT = N1->getValueType(0);
8923*9880d681SAndroid Build Coastguard Worker EVT N1Op0VT = N1->getOperand(0)->getValueType(0);
8924*9880d681SAndroid Build Coastguard Worker return (N1VT == N1Op0VT || N1Op0VT != MVT::f128);
8925*9880d681SAndroid Build Coastguard Worker }
8926*9880d681SAndroid Build Coastguard Worker return false;
8927*9880d681SAndroid Build Coastguard Worker }
8928*9880d681SAndroid Build Coastguard Worker
visitFCOPYSIGN(SDNode * N)8929*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFCOPYSIGN(SDNode *N) {
8930*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
8931*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
8932*9880d681SAndroid Build Coastguard Worker ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N0);
8933*9880d681SAndroid Build Coastguard Worker ConstantFPSDNode *N1CFP = dyn_cast<ConstantFPSDNode>(N1);
8934*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
8935*9880d681SAndroid Build Coastguard Worker
8936*9880d681SAndroid Build Coastguard Worker if (N0CFP && N1CFP) // Constant fold
8937*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FCOPYSIGN, SDLoc(N), VT, N0, N1);
8938*9880d681SAndroid Build Coastguard Worker
8939*9880d681SAndroid Build Coastguard Worker if (N1CFP) {
8940*9880d681SAndroid Build Coastguard Worker const APFloat& V = N1CFP->getValueAPF();
8941*9880d681SAndroid Build Coastguard Worker // copysign(x, c1) -> fabs(x) iff ispos(c1)
8942*9880d681SAndroid Build Coastguard Worker // copysign(x, c1) -> fneg(fabs(x)) iff isneg(c1)
8943*9880d681SAndroid Build Coastguard Worker if (!V.isNegative()) {
8944*9880d681SAndroid Build Coastguard Worker if (!LegalOperations || TLI.isOperationLegal(ISD::FABS, VT))
8945*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FABS, SDLoc(N), VT, N0);
8946*9880d681SAndroid Build Coastguard Worker } else {
8947*9880d681SAndroid Build Coastguard Worker if (!LegalOperations || TLI.isOperationLegal(ISD::FNEG, VT))
8948*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FNEG, SDLoc(N), VT,
8949*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FABS, SDLoc(N0), VT, N0));
8950*9880d681SAndroid Build Coastguard Worker }
8951*9880d681SAndroid Build Coastguard Worker }
8952*9880d681SAndroid Build Coastguard Worker
8953*9880d681SAndroid Build Coastguard Worker // copysign(fabs(x), y) -> copysign(x, y)
8954*9880d681SAndroid Build Coastguard Worker // copysign(fneg(x), y) -> copysign(x, y)
8955*9880d681SAndroid Build Coastguard Worker // copysign(copysign(x,z), y) -> copysign(x, y)
8956*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FABS || N0.getOpcode() == ISD::FNEG ||
8957*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::FCOPYSIGN)
8958*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FCOPYSIGN, SDLoc(N), VT,
8959*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N1);
8960*9880d681SAndroid Build Coastguard Worker
8961*9880d681SAndroid Build Coastguard Worker // copysign(x, abs(y)) -> abs(x)
8962*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::FABS)
8963*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FABS, SDLoc(N), VT, N0);
8964*9880d681SAndroid Build Coastguard Worker
8965*9880d681SAndroid Build Coastguard Worker // copysign(x, copysign(y,z)) -> copysign(x, z)
8966*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::FCOPYSIGN)
8967*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FCOPYSIGN, SDLoc(N), VT,
8968*9880d681SAndroid Build Coastguard Worker N0, N1.getOperand(1));
8969*9880d681SAndroid Build Coastguard Worker
8970*9880d681SAndroid Build Coastguard Worker // copysign(x, fp_extend(y)) -> copysign(x, y)
8971*9880d681SAndroid Build Coastguard Worker // copysign(x, fp_round(y)) -> copysign(x, y)
8972*9880d681SAndroid Build Coastguard Worker if (CanCombineFCOPYSIGN_EXTEND_ROUND(N))
8973*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FCOPYSIGN, SDLoc(N), VT,
8974*9880d681SAndroid Build Coastguard Worker N0, N1.getOperand(0));
8975*9880d681SAndroid Build Coastguard Worker
8976*9880d681SAndroid Build Coastguard Worker return SDValue();
8977*9880d681SAndroid Build Coastguard Worker }
8978*9880d681SAndroid Build Coastguard Worker
visitSINT_TO_FP(SDNode * N)8979*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSINT_TO_FP(SDNode *N) {
8980*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
8981*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
8982*9880d681SAndroid Build Coastguard Worker EVT OpVT = N0.getValueType();
8983*9880d681SAndroid Build Coastguard Worker
8984*9880d681SAndroid Build Coastguard Worker // fold (sint_to_fp c1) -> c1fp
8985*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N0) &&
8986*9880d681SAndroid Build Coastguard Worker // ...but only if the target supports immediate floating-point values
8987*9880d681SAndroid Build Coastguard Worker (!LegalOperations ||
8988*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegalOrCustom(llvm::ISD::ConstantFP, VT)))
8989*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SINT_TO_FP, SDLoc(N), VT, N0);
8990*9880d681SAndroid Build Coastguard Worker
8991*9880d681SAndroid Build Coastguard Worker // If the input is a legal type, and SINT_TO_FP is not legal on this target,
8992*9880d681SAndroid Build Coastguard Worker // but UINT_TO_FP is legal on this target, try to convert.
8993*9880d681SAndroid Build Coastguard Worker if (!TLI.isOperationLegalOrCustom(ISD::SINT_TO_FP, OpVT) &&
8994*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegalOrCustom(ISD::UINT_TO_FP, OpVT)) {
8995*9880d681SAndroid Build Coastguard Worker // If the sign bit is known to be zero, we can change this to UINT_TO_FP.
8996*9880d681SAndroid Build Coastguard Worker if (DAG.SignBitIsZero(N0))
8997*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::UINT_TO_FP, SDLoc(N), VT, N0);
8998*9880d681SAndroid Build Coastguard Worker }
8999*9880d681SAndroid Build Coastguard Worker
9000*9880d681SAndroid Build Coastguard Worker // The next optimizations are desirable only if SELECT_CC can be lowered.
9001*9880d681SAndroid Build Coastguard Worker if (TLI.isOperationLegalOrCustom(ISD::SELECT_CC, VT) || !LegalOperations) {
9002*9880d681SAndroid Build Coastguard Worker // fold (sint_to_fp (setcc x, y, cc)) -> (select_cc x, y, -1.0, 0.0,, cc)
9003*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SETCC && N0.getValueType() == MVT::i1 &&
9004*9880d681SAndroid Build Coastguard Worker !VT.isVector() &&
9005*9880d681SAndroid Build Coastguard Worker (!LegalOperations ||
9006*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegalOrCustom(llvm::ISD::ConstantFP, VT))) {
9007*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
9008*9880d681SAndroid Build Coastguard Worker SDValue Ops[] =
9009*9880d681SAndroid Build Coastguard Worker { N0.getOperand(0), N0.getOperand(1),
9010*9880d681SAndroid Build Coastguard Worker DAG.getConstantFP(-1.0, DL, VT), DAG.getConstantFP(0.0, DL, VT),
9011*9880d681SAndroid Build Coastguard Worker N0.getOperand(2) };
9012*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SELECT_CC, DL, VT, Ops);
9013*9880d681SAndroid Build Coastguard Worker }
9014*9880d681SAndroid Build Coastguard Worker
9015*9880d681SAndroid Build Coastguard Worker // fold (sint_to_fp (zext (setcc x, y, cc))) ->
9016*9880d681SAndroid Build Coastguard Worker // (select_cc x, y, 1.0, 0.0,, cc)
9017*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::ZERO_EXTEND &&
9018*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getOpcode() == ISD::SETCC &&!VT.isVector() &&
9019*9880d681SAndroid Build Coastguard Worker (!LegalOperations ||
9020*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegalOrCustom(llvm::ISD::ConstantFP, VT))) {
9021*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
9022*9880d681SAndroid Build Coastguard Worker SDValue Ops[] =
9023*9880d681SAndroid Build Coastguard Worker { N0.getOperand(0).getOperand(0), N0.getOperand(0).getOperand(1),
9024*9880d681SAndroid Build Coastguard Worker DAG.getConstantFP(1.0, DL, VT), DAG.getConstantFP(0.0, DL, VT),
9025*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getOperand(2) };
9026*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SELECT_CC, DL, VT, Ops);
9027*9880d681SAndroid Build Coastguard Worker }
9028*9880d681SAndroid Build Coastguard Worker }
9029*9880d681SAndroid Build Coastguard Worker
9030*9880d681SAndroid Build Coastguard Worker return SDValue();
9031*9880d681SAndroid Build Coastguard Worker }
9032*9880d681SAndroid Build Coastguard Worker
visitUINT_TO_FP(SDNode * N)9033*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitUINT_TO_FP(SDNode *N) {
9034*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
9035*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
9036*9880d681SAndroid Build Coastguard Worker EVT OpVT = N0.getValueType();
9037*9880d681SAndroid Build Coastguard Worker
9038*9880d681SAndroid Build Coastguard Worker // fold (uint_to_fp c1) -> c1fp
9039*9880d681SAndroid Build Coastguard Worker if (DAG.isConstantIntBuildVectorOrConstantInt(N0) &&
9040*9880d681SAndroid Build Coastguard Worker // ...but only if the target supports immediate floating-point values
9041*9880d681SAndroid Build Coastguard Worker (!LegalOperations ||
9042*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegalOrCustom(llvm::ISD::ConstantFP, VT)))
9043*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::UINT_TO_FP, SDLoc(N), VT, N0);
9044*9880d681SAndroid Build Coastguard Worker
9045*9880d681SAndroid Build Coastguard Worker // If the input is a legal type, and UINT_TO_FP is not legal on this target,
9046*9880d681SAndroid Build Coastguard Worker // but SINT_TO_FP is legal on this target, try to convert.
9047*9880d681SAndroid Build Coastguard Worker if (!TLI.isOperationLegalOrCustom(ISD::UINT_TO_FP, OpVT) &&
9048*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegalOrCustom(ISD::SINT_TO_FP, OpVT)) {
9049*9880d681SAndroid Build Coastguard Worker // If the sign bit is known to be zero, we can change this to SINT_TO_FP.
9050*9880d681SAndroid Build Coastguard Worker if (DAG.SignBitIsZero(N0))
9051*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SINT_TO_FP, SDLoc(N), VT, N0);
9052*9880d681SAndroid Build Coastguard Worker }
9053*9880d681SAndroid Build Coastguard Worker
9054*9880d681SAndroid Build Coastguard Worker // The next optimizations are desirable only if SELECT_CC can be lowered.
9055*9880d681SAndroid Build Coastguard Worker if (TLI.isOperationLegalOrCustom(ISD::SELECT_CC, VT) || !LegalOperations) {
9056*9880d681SAndroid Build Coastguard Worker // fold (uint_to_fp (setcc x, y, cc)) -> (select_cc x, y, -1.0, 0.0,, cc)
9057*9880d681SAndroid Build Coastguard Worker
9058*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::SETCC && !VT.isVector() &&
9059*9880d681SAndroid Build Coastguard Worker (!LegalOperations ||
9060*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegalOrCustom(llvm::ISD::ConstantFP, VT))) {
9061*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
9062*9880d681SAndroid Build Coastguard Worker SDValue Ops[] =
9063*9880d681SAndroid Build Coastguard Worker { N0.getOperand(0), N0.getOperand(1),
9064*9880d681SAndroid Build Coastguard Worker DAG.getConstantFP(1.0, DL, VT), DAG.getConstantFP(0.0, DL, VT),
9065*9880d681SAndroid Build Coastguard Worker N0.getOperand(2) };
9066*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::SELECT_CC, DL, VT, Ops);
9067*9880d681SAndroid Build Coastguard Worker }
9068*9880d681SAndroid Build Coastguard Worker }
9069*9880d681SAndroid Build Coastguard Worker
9070*9880d681SAndroid Build Coastguard Worker return SDValue();
9071*9880d681SAndroid Build Coastguard Worker }
9072*9880d681SAndroid Build Coastguard Worker
9073*9880d681SAndroid Build Coastguard Worker // Fold (fp_to_{s/u}int ({s/u}int_to_fpx)) -> zext x, sext x, trunc x, or x
FoldIntToFPToInt(SDNode * N,SelectionDAG & DAG)9074*9880d681SAndroid Build Coastguard Worker static SDValue FoldIntToFPToInt(SDNode *N, SelectionDAG &DAG) {
9075*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
9076*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
9077*9880d681SAndroid Build Coastguard Worker
9078*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() != ISD::UINT_TO_FP && N0.getOpcode() != ISD::SINT_TO_FP)
9079*9880d681SAndroid Build Coastguard Worker return SDValue();
9080*9880d681SAndroid Build Coastguard Worker
9081*9880d681SAndroid Build Coastguard Worker SDValue Src = N0.getOperand(0);
9082*9880d681SAndroid Build Coastguard Worker EVT SrcVT = Src.getValueType();
9083*9880d681SAndroid Build Coastguard Worker bool IsInputSigned = N0.getOpcode() == ISD::SINT_TO_FP;
9084*9880d681SAndroid Build Coastguard Worker bool IsOutputSigned = N->getOpcode() == ISD::FP_TO_SINT;
9085*9880d681SAndroid Build Coastguard Worker
9086*9880d681SAndroid Build Coastguard Worker // We can safely assume the conversion won't overflow the output range,
9087*9880d681SAndroid Build Coastguard Worker // because (for example) (uint8_t)18293.f is undefined behavior.
9088*9880d681SAndroid Build Coastguard Worker
9089*9880d681SAndroid Build Coastguard Worker // Since we can assume the conversion won't overflow, our decision as to
9090*9880d681SAndroid Build Coastguard Worker // whether the input will fit in the float should depend on the minimum
9091*9880d681SAndroid Build Coastguard Worker // of the input range and output range.
9092*9880d681SAndroid Build Coastguard Worker
9093*9880d681SAndroid Build Coastguard Worker // This means this is also safe for a signed input and unsigned output, since
9094*9880d681SAndroid Build Coastguard Worker // a negative input would lead to undefined behavior.
9095*9880d681SAndroid Build Coastguard Worker unsigned InputSize = (int)SrcVT.getScalarSizeInBits() - IsInputSigned;
9096*9880d681SAndroid Build Coastguard Worker unsigned OutputSize = (int)VT.getScalarSizeInBits() - IsOutputSigned;
9097*9880d681SAndroid Build Coastguard Worker unsigned ActualSize = std::min(InputSize, OutputSize);
9098*9880d681SAndroid Build Coastguard Worker const fltSemantics &sem = DAG.EVTToAPFloatSemantics(N0.getValueType());
9099*9880d681SAndroid Build Coastguard Worker
9100*9880d681SAndroid Build Coastguard Worker // We can only fold away the float conversion if the input range can be
9101*9880d681SAndroid Build Coastguard Worker // represented exactly in the float range.
9102*9880d681SAndroid Build Coastguard Worker if (APFloat::semanticsPrecision(sem) >= ActualSize) {
9103*9880d681SAndroid Build Coastguard Worker if (VT.getScalarSizeInBits() > SrcVT.getScalarSizeInBits()) {
9104*9880d681SAndroid Build Coastguard Worker unsigned ExtOp = IsInputSigned && IsOutputSigned ? ISD::SIGN_EXTEND
9105*9880d681SAndroid Build Coastguard Worker : ISD::ZERO_EXTEND;
9106*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ExtOp, SDLoc(N), VT, Src);
9107*9880d681SAndroid Build Coastguard Worker }
9108*9880d681SAndroid Build Coastguard Worker if (VT.getScalarSizeInBits() < SrcVT.getScalarSizeInBits())
9109*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TRUNCATE, SDLoc(N), VT, Src);
9110*9880d681SAndroid Build Coastguard Worker return DAG.getBitcast(VT, Src);
9111*9880d681SAndroid Build Coastguard Worker }
9112*9880d681SAndroid Build Coastguard Worker return SDValue();
9113*9880d681SAndroid Build Coastguard Worker }
9114*9880d681SAndroid Build Coastguard Worker
visitFP_TO_SINT(SDNode * N)9115*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFP_TO_SINT(SDNode *N) {
9116*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
9117*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
9118*9880d681SAndroid Build Coastguard Worker
9119*9880d681SAndroid Build Coastguard Worker // fold (fp_to_sint c1fp) -> c1
9120*9880d681SAndroid Build Coastguard Worker if (isConstantFPBuildVectorOrConstantFP(N0))
9121*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FP_TO_SINT, SDLoc(N), VT, N0);
9122*9880d681SAndroid Build Coastguard Worker
9123*9880d681SAndroid Build Coastguard Worker return FoldIntToFPToInt(N, DAG);
9124*9880d681SAndroid Build Coastguard Worker }
9125*9880d681SAndroid Build Coastguard Worker
visitFP_TO_UINT(SDNode * N)9126*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFP_TO_UINT(SDNode *N) {
9127*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
9128*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
9129*9880d681SAndroid Build Coastguard Worker
9130*9880d681SAndroid Build Coastguard Worker // fold (fp_to_uint c1fp) -> c1
9131*9880d681SAndroid Build Coastguard Worker if (isConstantFPBuildVectorOrConstantFP(N0))
9132*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FP_TO_UINT, SDLoc(N), VT, N0);
9133*9880d681SAndroid Build Coastguard Worker
9134*9880d681SAndroid Build Coastguard Worker return FoldIntToFPToInt(N, DAG);
9135*9880d681SAndroid Build Coastguard Worker }
9136*9880d681SAndroid Build Coastguard Worker
visitFP_ROUND(SDNode * N)9137*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFP_ROUND(SDNode *N) {
9138*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
9139*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
9140*9880d681SAndroid Build Coastguard Worker ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N0);
9141*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
9142*9880d681SAndroid Build Coastguard Worker
9143*9880d681SAndroid Build Coastguard Worker // fold (fp_round c1fp) -> c1fp
9144*9880d681SAndroid Build Coastguard Worker if (N0CFP)
9145*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FP_ROUND, SDLoc(N), VT, N0, N1);
9146*9880d681SAndroid Build Coastguard Worker
9147*9880d681SAndroid Build Coastguard Worker // fold (fp_round (fp_extend x)) -> x
9148*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FP_EXTEND && VT == N0.getOperand(0).getValueType())
9149*9880d681SAndroid Build Coastguard Worker return N0.getOperand(0);
9150*9880d681SAndroid Build Coastguard Worker
9151*9880d681SAndroid Build Coastguard Worker // fold (fp_round (fp_round x)) -> (fp_round x)
9152*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FP_ROUND) {
9153*9880d681SAndroid Build Coastguard Worker const bool NIsTrunc = N->getConstantOperandVal(1) == 1;
9154*9880d681SAndroid Build Coastguard Worker const bool N0IsTrunc = N0.getNode()->getConstantOperandVal(1) == 1;
9155*9880d681SAndroid Build Coastguard Worker
9156*9880d681SAndroid Build Coastguard Worker // Skip this folding if it results in an fp_round from f80 to f16.
9157*9880d681SAndroid Build Coastguard Worker //
9158*9880d681SAndroid Build Coastguard Worker // f80 to f16 always generates an expensive (and as yet, unimplemented)
9159*9880d681SAndroid Build Coastguard Worker // libcall to __truncxfhf2 instead of selecting native f16 conversion
9160*9880d681SAndroid Build Coastguard Worker // instructions from f32 or f64. Moreover, the first (value-preserving)
9161*9880d681SAndroid Build Coastguard Worker // fp_round from f80 to either f32 or f64 may become a NOP in platforms like
9162*9880d681SAndroid Build Coastguard Worker // x86.
9163*9880d681SAndroid Build Coastguard Worker if (N0.getOperand(0).getValueType() == MVT::f80 && VT == MVT::f16)
9164*9880d681SAndroid Build Coastguard Worker return SDValue();
9165*9880d681SAndroid Build Coastguard Worker
9166*9880d681SAndroid Build Coastguard Worker // If the first fp_round isn't a value preserving truncation, it might
9167*9880d681SAndroid Build Coastguard Worker // introduce a tie in the second fp_round, that wouldn't occur in the
9168*9880d681SAndroid Build Coastguard Worker // single-step fp_round we want to fold to.
9169*9880d681SAndroid Build Coastguard Worker // In other words, double rounding isn't the same as rounding.
9170*9880d681SAndroid Build Coastguard Worker // Also, this is a value preserving truncation iff both fp_round's are.
9171*9880d681SAndroid Build Coastguard Worker if (DAG.getTarget().Options.UnsafeFPMath || N0IsTrunc) {
9172*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
9173*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FP_ROUND, DL, VT, N0.getOperand(0),
9174*9880d681SAndroid Build Coastguard Worker DAG.getIntPtrConstant(NIsTrunc && N0IsTrunc, DL));
9175*9880d681SAndroid Build Coastguard Worker }
9176*9880d681SAndroid Build Coastguard Worker }
9177*9880d681SAndroid Build Coastguard Worker
9178*9880d681SAndroid Build Coastguard Worker // fold (fp_round (copysign X, Y)) -> (copysign (fp_round X), Y)
9179*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FCOPYSIGN && N0.getNode()->hasOneUse()) {
9180*9880d681SAndroid Build Coastguard Worker SDValue Tmp = DAG.getNode(ISD::FP_ROUND, SDLoc(N0), VT,
9181*9880d681SAndroid Build Coastguard Worker N0.getOperand(0), N1);
9182*9880d681SAndroid Build Coastguard Worker AddToWorklist(Tmp.getNode());
9183*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FCOPYSIGN, SDLoc(N), VT,
9184*9880d681SAndroid Build Coastguard Worker Tmp, N0.getOperand(1));
9185*9880d681SAndroid Build Coastguard Worker }
9186*9880d681SAndroid Build Coastguard Worker
9187*9880d681SAndroid Build Coastguard Worker return SDValue();
9188*9880d681SAndroid Build Coastguard Worker }
9189*9880d681SAndroid Build Coastguard Worker
visitFP_ROUND_INREG(SDNode * N)9190*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFP_ROUND_INREG(SDNode *N) {
9191*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
9192*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
9193*9880d681SAndroid Build Coastguard Worker EVT EVT = cast<VTSDNode>(N->getOperand(1))->getVT();
9194*9880d681SAndroid Build Coastguard Worker ConstantFPSDNode *N0CFP = dyn_cast<ConstantFPSDNode>(N0);
9195*9880d681SAndroid Build Coastguard Worker
9196*9880d681SAndroid Build Coastguard Worker // fold (fp_round_inreg c1fp) -> c1fp
9197*9880d681SAndroid Build Coastguard Worker if (N0CFP && isTypeLegal(EVT)) {
9198*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
9199*9880d681SAndroid Build Coastguard Worker SDValue Round = DAG.getConstantFP(*N0CFP->getConstantFPValue(), DL, EVT);
9200*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FP_EXTEND, DL, VT, Round);
9201*9880d681SAndroid Build Coastguard Worker }
9202*9880d681SAndroid Build Coastguard Worker
9203*9880d681SAndroid Build Coastguard Worker return SDValue();
9204*9880d681SAndroid Build Coastguard Worker }
9205*9880d681SAndroid Build Coastguard Worker
visitFP_EXTEND(SDNode * N)9206*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFP_EXTEND(SDNode *N) {
9207*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
9208*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
9209*9880d681SAndroid Build Coastguard Worker
9210*9880d681SAndroid Build Coastguard Worker // If this is fp_round(fpextend), don't fold it, allow ourselves to be folded.
9211*9880d681SAndroid Build Coastguard Worker if (N->hasOneUse() &&
9212*9880d681SAndroid Build Coastguard Worker N->use_begin()->getOpcode() == ISD::FP_ROUND)
9213*9880d681SAndroid Build Coastguard Worker return SDValue();
9214*9880d681SAndroid Build Coastguard Worker
9215*9880d681SAndroid Build Coastguard Worker // fold (fp_extend c1fp) -> c1fp
9216*9880d681SAndroid Build Coastguard Worker if (isConstantFPBuildVectorOrConstantFP(N0))
9217*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FP_EXTEND, SDLoc(N), VT, N0);
9218*9880d681SAndroid Build Coastguard Worker
9219*9880d681SAndroid Build Coastguard Worker // fold (fp_extend (fp16_to_fp op)) -> (fp16_to_fp op)
9220*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FP16_TO_FP &&
9221*9880d681SAndroid Build Coastguard Worker TLI.getOperationAction(ISD::FP16_TO_FP, VT) == TargetLowering::Legal)
9222*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FP16_TO_FP, SDLoc(N), VT, N0.getOperand(0));
9223*9880d681SAndroid Build Coastguard Worker
9224*9880d681SAndroid Build Coastguard Worker // Turn fp_extend(fp_round(X, 1)) -> x since the fp_round doesn't affect the
9225*9880d681SAndroid Build Coastguard Worker // value of X.
9226*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FP_ROUND
9227*9880d681SAndroid Build Coastguard Worker && N0.getNode()->getConstantOperandVal(1) == 1) {
9228*9880d681SAndroid Build Coastguard Worker SDValue In = N0.getOperand(0);
9229*9880d681SAndroid Build Coastguard Worker if (In.getValueType() == VT) return In;
9230*9880d681SAndroid Build Coastguard Worker if (VT.bitsLT(In.getValueType()))
9231*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FP_ROUND, SDLoc(N), VT,
9232*9880d681SAndroid Build Coastguard Worker In, N0.getOperand(1));
9233*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FP_EXTEND, SDLoc(N), VT, In);
9234*9880d681SAndroid Build Coastguard Worker }
9235*9880d681SAndroid Build Coastguard Worker
9236*9880d681SAndroid Build Coastguard Worker // fold (fpext (load x)) -> (fpext (fptrunc (extload x)))
9237*9880d681SAndroid Build Coastguard Worker if (ISD::isNormalLoad(N0.getNode()) && N0.hasOneUse() &&
9238*9880d681SAndroid Build Coastguard Worker TLI.isLoadExtLegal(ISD::EXTLOAD, VT, N0.getValueType())) {
9239*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = cast<LoadSDNode>(N0);
9240*9880d681SAndroid Build Coastguard Worker SDValue ExtLoad = DAG.getExtLoad(ISD::EXTLOAD, SDLoc(N), VT,
9241*9880d681SAndroid Build Coastguard Worker LN0->getChain(),
9242*9880d681SAndroid Build Coastguard Worker LN0->getBasePtr(), N0.getValueType(),
9243*9880d681SAndroid Build Coastguard Worker LN0->getMemOperand());
9244*9880d681SAndroid Build Coastguard Worker CombineTo(N, ExtLoad);
9245*9880d681SAndroid Build Coastguard Worker CombineTo(N0.getNode(),
9246*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FP_ROUND, SDLoc(N0),
9247*9880d681SAndroid Build Coastguard Worker N0.getValueType(), ExtLoad,
9248*9880d681SAndroid Build Coastguard Worker DAG.getIntPtrConstant(1, SDLoc(N0))),
9249*9880d681SAndroid Build Coastguard Worker ExtLoad.getValue(1));
9250*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
9251*9880d681SAndroid Build Coastguard Worker }
9252*9880d681SAndroid Build Coastguard Worker
9253*9880d681SAndroid Build Coastguard Worker return SDValue();
9254*9880d681SAndroid Build Coastguard Worker }
9255*9880d681SAndroid Build Coastguard Worker
visitFCEIL(SDNode * N)9256*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFCEIL(SDNode *N) {
9257*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
9258*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
9259*9880d681SAndroid Build Coastguard Worker
9260*9880d681SAndroid Build Coastguard Worker // fold (fceil c1) -> fceil(c1)
9261*9880d681SAndroid Build Coastguard Worker if (isConstantFPBuildVectorOrConstantFP(N0))
9262*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FCEIL, SDLoc(N), VT, N0);
9263*9880d681SAndroid Build Coastguard Worker
9264*9880d681SAndroid Build Coastguard Worker return SDValue();
9265*9880d681SAndroid Build Coastguard Worker }
9266*9880d681SAndroid Build Coastguard Worker
visitFTRUNC(SDNode * N)9267*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFTRUNC(SDNode *N) {
9268*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
9269*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
9270*9880d681SAndroid Build Coastguard Worker
9271*9880d681SAndroid Build Coastguard Worker // fold (ftrunc c1) -> ftrunc(c1)
9272*9880d681SAndroid Build Coastguard Worker if (isConstantFPBuildVectorOrConstantFP(N0))
9273*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FTRUNC, SDLoc(N), VT, N0);
9274*9880d681SAndroid Build Coastguard Worker
9275*9880d681SAndroid Build Coastguard Worker return SDValue();
9276*9880d681SAndroid Build Coastguard Worker }
9277*9880d681SAndroid Build Coastguard Worker
visitFFLOOR(SDNode * N)9278*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFFLOOR(SDNode *N) {
9279*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
9280*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
9281*9880d681SAndroid Build Coastguard Worker
9282*9880d681SAndroid Build Coastguard Worker // fold (ffloor c1) -> ffloor(c1)
9283*9880d681SAndroid Build Coastguard Worker if (isConstantFPBuildVectorOrConstantFP(N0))
9284*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FFLOOR, SDLoc(N), VT, N0);
9285*9880d681SAndroid Build Coastguard Worker
9286*9880d681SAndroid Build Coastguard Worker return SDValue();
9287*9880d681SAndroid Build Coastguard Worker }
9288*9880d681SAndroid Build Coastguard Worker
9289*9880d681SAndroid Build Coastguard Worker // FIXME: FNEG and FABS have a lot in common; refactor.
visitFNEG(SDNode * N)9290*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFNEG(SDNode *N) {
9291*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
9292*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
9293*9880d681SAndroid Build Coastguard Worker
9294*9880d681SAndroid Build Coastguard Worker // Constant fold FNEG.
9295*9880d681SAndroid Build Coastguard Worker if (isConstantFPBuildVectorOrConstantFP(N0))
9296*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FNEG, SDLoc(N), VT, N0);
9297*9880d681SAndroid Build Coastguard Worker
9298*9880d681SAndroid Build Coastguard Worker if (isNegatibleForFree(N0, LegalOperations, DAG.getTargetLoweringInfo(),
9299*9880d681SAndroid Build Coastguard Worker &DAG.getTarget().Options))
9300*9880d681SAndroid Build Coastguard Worker return GetNegatedExpression(N0, DAG, LegalOperations);
9301*9880d681SAndroid Build Coastguard Worker
9302*9880d681SAndroid Build Coastguard Worker // Transform fneg(bitconvert(x)) -> bitconvert(x ^ sign) to avoid loading
9303*9880d681SAndroid Build Coastguard Worker // constant pool values.
9304*9880d681SAndroid Build Coastguard Worker if (!TLI.isFNegFree(VT) &&
9305*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::BITCAST &&
9306*9880d681SAndroid Build Coastguard Worker N0.getNode()->hasOneUse()) {
9307*9880d681SAndroid Build Coastguard Worker SDValue Int = N0.getOperand(0);
9308*9880d681SAndroid Build Coastguard Worker EVT IntVT = Int.getValueType();
9309*9880d681SAndroid Build Coastguard Worker if (IntVT.isInteger() && !IntVT.isVector()) {
9310*9880d681SAndroid Build Coastguard Worker APInt SignMask;
9311*9880d681SAndroid Build Coastguard Worker if (N0.getValueType().isVector()) {
9312*9880d681SAndroid Build Coastguard Worker // For a vector, get a mask such as 0x80... per scalar element
9313*9880d681SAndroid Build Coastguard Worker // and splat it.
9314*9880d681SAndroid Build Coastguard Worker SignMask = APInt::getSignBit(N0.getValueType().getScalarSizeInBits());
9315*9880d681SAndroid Build Coastguard Worker SignMask = APInt::getSplat(IntVT.getSizeInBits(), SignMask);
9316*9880d681SAndroid Build Coastguard Worker } else {
9317*9880d681SAndroid Build Coastguard Worker // For a scalar, just generate 0x80...
9318*9880d681SAndroid Build Coastguard Worker SignMask = APInt::getSignBit(IntVT.getSizeInBits());
9319*9880d681SAndroid Build Coastguard Worker }
9320*9880d681SAndroid Build Coastguard Worker SDLoc DL0(N0);
9321*9880d681SAndroid Build Coastguard Worker Int = DAG.getNode(ISD::XOR, DL0, IntVT, Int,
9322*9880d681SAndroid Build Coastguard Worker DAG.getConstant(SignMask, DL0, IntVT));
9323*9880d681SAndroid Build Coastguard Worker AddToWorklist(Int.getNode());
9324*9880d681SAndroid Build Coastguard Worker return DAG.getBitcast(VT, Int);
9325*9880d681SAndroid Build Coastguard Worker }
9326*9880d681SAndroid Build Coastguard Worker }
9327*9880d681SAndroid Build Coastguard Worker
9328*9880d681SAndroid Build Coastguard Worker // (fneg (fmul c, x)) -> (fmul -c, x)
9329*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FMUL &&
9330*9880d681SAndroid Build Coastguard Worker (N0.getNode()->hasOneUse() || !TLI.isFNegFree(VT))) {
9331*9880d681SAndroid Build Coastguard Worker ConstantFPSDNode *CFP1 = dyn_cast<ConstantFPSDNode>(N0.getOperand(1));
9332*9880d681SAndroid Build Coastguard Worker if (CFP1) {
9333*9880d681SAndroid Build Coastguard Worker APFloat CVal = CFP1->getValueAPF();
9334*9880d681SAndroid Build Coastguard Worker CVal.changeSign();
9335*9880d681SAndroid Build Coastguard Worker if (Level >= AfterLegalizeDAG &&
9336*9880d681SAndroid Build Coastguard Worker (TLI.isFPImmLegal(CVal, VT) ||
9337*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegal(ISD::ConstantFP, VT)))
9338*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMUL, SDLoc(N), VT, N0.getOperand(0),
9339*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::FNEG, SDLoc(N), VT,
9340*9880d681SAndroid Build Coastguard Worker N0.getOperand(1)),
9341*9880d681SAndroid Build Coastguard Worker &cast<BinaryWithFlagsSDNode>(N0)->Flags);
9342*9880d681SAndroid Build Coastguard Worker }
9343*9880d681SAndroid Build Coastguard Worker }
9344*9880d681SAndroid Build Coastguard Worker
9345*9880d681SAndroid Build Coastguard Worker return SDValue();
9346*9880d681SAndroid Build Coastguard Worker }
9347*9880d681SAndroid Build Coastguard Worker
visitFMINNUM(SDNode * N)9348*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFMINNUM(SDNode *N) {
9349*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
9350*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
9351*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
9352*9880d681SAndroid Build Coastguard Worker const ConstantFPSDNode *N0CFP = isConstOrConstSplatFP(N0);
9353*9880d681SAndroid Build Coastguard Worker const ConstantFPSDNode *N1CFP = isConstOrConstSplatFP(N1);
9354*9880d681SAndroid Build Coastguard Worker
9355*9880d681SAndroid Build Coastguard Worker if (N0CFP && N1CFP) {
9356*9880d681SAndroid Build Coastguard Worker const APFloat &C0 = N0CFP->getValueAPF();
9357*9880d681SAndroid Build Coastguard Worker const APFloat &C1 = N1CFP->getValueAPF();
9358*9880d681SAndroid Build Coastguard Worker return DAG.getConstantFP(minnum(C0, C1), SDLoc(N), VT);
9359*9880d681SAndroid Build Coastguard Worker }
9360*9880d681SAndroid Build Coastguard Worker
9361*9880d681SAndroid Build Coastguard Worker // Canonicalize to constant on RHS.
9362*9880d681SAndroid Build Coastguard Worker if (isConstantFPBuildVectorOrConstantFP(N0) &&
9363*9880d681SAndroid Build Coastguard Worker !isConstantFPBuildVectorOrConstantFP(N1))
9364*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMINNUM, SDLoc(N), VT, N1, N0);
9365*9880d681SAndroid Build Coastguard Worker
9366*9880d681SAndroid Build Coastguard Worker return SDValue();
9367*9880d681SAndroid Build Coastguard Worker }
9368*9880d681SAndroid Build Coastguard Worker
visitFMAXNUM(SDNode * N)9369*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFMAXNUM(SDNode *N) {
9370*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
9371*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
9372*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
9373*9880d681SAndroid Build Coastguard Worker const ConstantFPSDNode *N0CFP = isConstOrConstSplatFP(N0);
9374*9880d681SAndroid Build Coastguard Worker const ConstantFPSDNode *N1CFP = isConstOrConstSplatFP(N1);
9375*9880d681SAndroid Build Coastguard Worker
9376*9880d681SAndroid Build Coastguard Worker if (N0CFP && N1CFP) {
9377*9880d681SAndroid Build Coastguard Worker const APFloat &C0 = N0CFP->getValueAPF();
9378*9880d681SAndroid Build Coastguard Worker const APFloat &C1 = N1CFP->getValueAPF();
9379*9880d681SAndroid Build Coastguard Worker return DAG.getConstantFP(maxnum(C0, C1), SDLoc(N), VT);
9380*9880d681SAndroid Build Coastguard Worker }
9381*9880d681SAndroid Build Coastguard Worker
9382*9880d681SAndroid Build Coastguard Worker // Canonicalize to constant on RHS.
9383*9880d681SAndroid Build Coastguard Worker if (isConstantFPBuildVectorOrConstantFP(N0) &&
9384*9880d681SAndroid Build Coastguard Worker !isConstantFPBuildVectorOrConstantFP(N1))
9385*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FMAXNUM, SDLoc(N), VT, N1, N0);
9386*9880d681SAndroid Build Coastguard Worker
9387*9880d681SAndroid Build Coastguard Worker return SDValue();
9388*9880d681SAndroid Build Coastguard Worker }
9389*9880d681SAndroid Build Coastguard Worker
visitFABS(SDNode * N)9390*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFABS(SDNode *N) {
9391*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
9392*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
9393*9880d681SAndroid Build Coastguard Worker
9394*9880d681SAndroid Build Coastguard Worker // fold (fabs c1) -> fabs(c1)
9395*9880d681SAndroid Build Coastguard Worker if (isConstantFPBuildVectorOrConstantFP(N0))
9396*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FABS, SDLoc(N), VT, N0);
9397*9880d681SAndroid Build Coastguard Worker
9398*9880d681SAndroid Build Coastguard Worker // fold (fabs (fabs x)) -> (fabs x)
9399*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FABS)
9400*9880d681SAndroid Build Coastguard Worker return N->getOperand(0);
9401*9880d681SAndroid Build Coastguard Worker
9402*9880d681SAndroid Build Coastguard Worker // fold (fabs (fneg x)) -> (fabs x)
9403*9880d681SAndroid Build Coastguard Worker // fold (fabs (fcopysign x, y)) -> (fabs x)
9404*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::FNEG || N0.getOpcode() == ISD::FCOPYSIGN)
9405*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FABS, SDLoc(N), VT, N0.getOperand(0));
9406*9880d681SAndroid Build Coastguard Worker
9407*9880d681SAndroid Build Coastguard Worker // Transform fabs(bitconvert(x)) -> bitconvert(x & ~sign) to avoid loading
9408*9880d681SAndroid Build Coastguard Worker // constant pool values.
9409*9880d681SAndroid Build Coastguard Worker if (!TLI.isFAbsFree(VT) &&
9410*9880d681SAndroid Build Coastguard Worker N0.getOpcode() == ISD::BITCAST &&
9411*9880d681SAndroid Build Coastguard Worker N0.getNode()->hasOneUse()) {
9412*9880d681SAndroid Build Coastguard Worker SDValue Int = N0.getOperand(0);
9413*9880d681SAndroid Build Coastguard Worker EVT IntVT = Int.getValueType();
9414*9880d681SAndroid Build Coastguard Worker if (IntVT.isInteger() && !IntVT.isVector()) {
9415*9880d681SAndroid Build Coastguard Worker APInt SignMask;
9416*9880d681SAndroid Build Coastguard Worker if (N0.getValueType().isVector()) {
9417*9880d681SAndroid Build Coastguard Worker // For a vector, get a mask such as 0x7f... per scalar element
9418*9880d681SAndroid Build Coastguard Worker // and splat it.
9419*9880d681SAndroid Build Coastguard Worker SignMask = ~APInt::getSignBit(N0.getValueType().getScalarSizeInBits());
9420*9880d681SAndroid Build Coastguard Worker SignMask = APInt::getSplat(IntVT.getSizeInBits(), SignMask);
9421*9880d681SAndroid Build Coastguard Worker } else {
9422*9880d681SAndroid Build Coastguard Worker // For a scalar, just generate 0x7f...
9423*9880d681SAndroid Build Coastguard Worker SignMask = ~APInt::getSignBit(IntVT.getSizeInBits());
9424*9880d681SAndroid Build Coastguard Worker }
9425*9880d681SAndroid Build Coastguard Worker SDLoc DL(N0);
9426*9880d681SAndroid Build Coastguard Worker Int = DAG.getNode(ISD::AND, DL, IntVT, Int,
9427*9880d681SAndroid Build Coastguard Worker DAG.getConstant(SignMask, DL, IntVT));
9428*9880d681SAndroid Build Coastguard Worker AddToWorklist(Int.getNode());
9429*9880d681SAndroid Build Coastguard Worker return DAG.getBitcast(N->getValueType(0), Int);
9430*9880d681SAndroid Build Coastguard Worker }
9431*9880d681SAndroid Build Coastguard Worker }
9432*9880d681SAndroid Build Coastguard Worker
9433*9880d681SAndroid Build Coastguard Worker return SDValue();
9434*9880d681SAndroid Build Coastguard Worker }
9435*9880d681SAndroid Build Coastguard Worker
visitBRCOND(SDNode * N)9436*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitBRCOND(SDNode *N) {
9437*9880d681SAndroid Build Coastguard Worker SDValue Chain = N->getOperand(0);
9438*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
9439*9880d681SAndroid Build Coastguard Worker SDValue N2 = N->getOperand(2);
9440*9880d681SAndroid Build Coastguard Worker
9441*9880d681SAndroid Build Coastguard Worker // If N is a constant we could fold this into a fallthrough or unconditional
9442*9880d681SAndroid Build Coastguard Worker // branch. However that doesn't happen very often in normal code, because
9443*9880d681SAndroid Build Coastguard Worker // Instcombine/SimplifyCFG should have handled the available opportunities.
9444*9880d681SAndroid Build Coastguard Worker // If we did this folding here, it would be necessary to update the
9445*9880d681SAndroid Build Coastguard Worker // MachineBasicBlock CFG, which is awkward.
9446*9880d681SAndroid Build Coastguard Worker
9447*9880d681SAndroid Build Coastguard Worker // fold a brcond with a setcc condition into a BR_CC node if BR_CC is legal
9448*9880d681SAndroid Build Coastguard Worker // on the target.
9449*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::SETCC &&
9450*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegalOrCustom(ISD::BR_CC,
9451*9880d681SAndroid Build Coastguard Worker N1.getOperand(0).getValueType())) {
9452*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::BR_CC, SDLoc(N), MVT::Other,
9453*9880d681SAndroid Build Coastguard Worker Chain, N1.getOperand(2),
9454*9880d681SAndroid Build Coastguard Worker N1.getOperand(0), N1.getOperand(1), N2);
9455*9880d681SAndroid Build Coastguard Worker }
9456*9880d681SAndroid Build Coastguard Worker
9457*9880d681SAndroid Build Coastguard Worker if ((N1.hasOneUse() && N1.getOpcode() == ISD::SRL) ||
9458*9880d681SAndroid Build Coastguard Worker ((N1.getOpcode() == ISD::TRUNCATE && N1.hasOneUse()) &&
9459*9880d681SAndroid Build Coastguard Worker (N1.getOperand(0).hasOneUse() &&
9460*9880d681SAndroid Build Coastguard Worker N1.getOperand(0).getOpcode() == ISD::SRL))) {
9461*9880d681SAndroid Build Coastguard Worker SDNode *Trunc = nullptr;
9462*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::TRUNCATE) {
9463*9880d681SAndroid Build Coastguard Worker // Look pass the truncate.
9464*9880d681SAndroid Build Coastguard Worker Trunc = N1.getNode();
9465*9880d681SAndroid Build Coastguard Worker N1 = N1.getOperand(0);
9466*9880d681SAndroid Build Coastguard Worker }
9467*9880d681SAndroid Build Coastguard Worker
9468*9880d681SAndroid Build Coastguard Worker // Match this pattern so that we can generate simpler code:
9469*9880d681SAndroid Build Coastguard Worker //
9470*9880d681SAndroid Build Coastguard Worker // %a = ...
9471*9880d681SAndroid Build Coastguard Worker // %b = and i32 %a, 2
9472*9880d681SAndroid Build Coastguard Worker // %c = srl i32 %b, 1
9473*9880d681SAndroid Build Coastguard Worker // brcond i32 %c ...
9474*9880d681SAndroid Build Coastguard Worker //
9475*9880d681SAndroid Build Coastguard Worker // into
9476*9880d681SAndroid Build Coastguard Worker //
9477*9880d681SAndroid Build Coastguard Worker // %a = ...
9478*9880d681SAndroid Build Coastguard Worker // %b = and i32 %a, 2
9479*9880d681SAndroid Build Coastguard Worker // %c = setcc eq %b, 0
9480*9880d681SAndroid Build Coastguard Worker // brcond %c ...
9481*9880d681SAndroid Build Coastguard Worker //
9482*9880d681SAndroid Build Coastguard Worker // This applies only when the AND constant value has one bit set and the
9483*9880d681SAndroid Build Coastguard Worker // SRL constant is equal to the log2 of the AND constant. The back-end is
9484*9880d681SAndroid Build Coastguard Worker // smart enough to convert the result into a TEST/JMP sequence.
9485*9880d681SAndroid Build Coastguard Worker SDValue Op0 = N1.getOperand(0);
9486*9880d681SAndroid Build Coastguard Worker SDValue Op1 = N1.getOperand(1);
9487*9880d681SAndroid Build Coastguard Worker
9488*9880d681SAndroid Build Coastguard Worker if (Op0.getOpcode() == ISD::AND &&
9489*9880d681SAndroid Build Coastguard Worker Op1.getOpcode() == ISD::Constant) {
9490*9880d681SAndroid Build Coastguard Worker SDValue AndOp1 = Op0.getOperand(1);
9491*9880d681SAndroid Build Coastguard Worker
9492*9880d681SAndroid Build Coastguard Worker if (AndOp1.getOpcode() == ISD::Constant) {
9493*9880d681SAndroid Build Coastguard Worker const APInt &AndConst = cast<ConstantSDNode>(AndOp1)->getAPIntValue();
9494*9880d681SAndroid Build Coastguard Worker
9495*9880d681SAndroid Build Coastguard Worker if (AndConst.isPowerOf2() &&
9496*9880d681SAndroid Build Coastguard Worker cast<ConstantSDNode>(Op1)->getAPIntValue()==AndConst.logBase2()) {
9497*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
9498*9880d681SAndroid Build Coastguard Worker SDValue SetCC =
9499*9880d681SAndroid Build Coastguard Worker DAG.getSetCC(DL,
9500*9880d681SAndroid Build Coastguard Worker getSetCCResultType(Op0.getValueType()),
9501*9880d681SAndroid Build Coastguard Worker Op0, DAG.getConstant(0, DL, Op0.getValueType()),
9502*9880d681SAndroid Build Coastguard Worker ISD::SETNE);
9503*9880d681SAndroid Build Coastguard Worker
9504*9880d681SAndroid Build Coastguard Worker SDValue NewBRCond = DAG.getNode(ISD::BRCOND, DL,
9505*9880d681SAndroid Build Coastguard Worker MVT::Other, Chain, SetCC, N2);
9506*9880d681SAndroid Build Coastguard Worker // Don't add the new BRCond into the worklist or else SimplifySelectCC
9507*9880d681SAndroid Build Coastguard Worker // will convert it back to (X & C1) >> C2.
9508*9880d681SAndroid Build Coastguard Worker CombineTo(N, NewBRCond, false);
9509*9880d681SAndroid Build Coastguard Worker // Truncate is dead.
9510*9880d681SAndroid Build Coastguard Worker if (Trunc)
9511*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(Trunc);
9512*9880d681SAndroid Build Coastguard Worker // Replace the uses of SRL with SETCC
9513*9880d681SAndroid Build Coastguard Worker WorklistRemover DeadNodes(*this);
9514*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(N1, SetCC);
9515*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(N1.getNode());
9516*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
9517*9880d681SAndroid Build Coastguard Worker }
9518*9880d681SAndroid Build Coastguard Worker }
9519*9880d681SAndroid Build Coastguard Worker }
9520*9880d681SAndroid Build Coastguard Worker
9521*9880d681SAndroid Build Coastguard Worker if (Trunc)
9522*9880d681SAndroid Build Coastguard Worker // Restore N1 if the above transformation doesn't match.
9523*9880d681SAndroid Build Coastguard Worker N1 = N->getOperand(1);
9524*9880d681SAndroid Build Coastguard Worker }
9525*9880d681SAndroid Build Coastguard Worker
9526*9880d681SAndroid Build Coastguard Worker // Transform br(xor(x, y)) -> br(x != y)
9527*9880d681SAndroid Build Coastguard Worker // Transform br(xor(xor(x,y), 1)) -> br (x == y)
9528*9880d681SAndroid Build Coastguard Worker if (N1.hasOneUse() && N1.getOpcode() == ISD::XOR) {
9529*9880d681SAndroid Build Coastguard Worker SDNode *TheXor = N1.getNode();
9530*9880d681SAndroid Build Coastguard Worker SDValue Op0 = TheXor->getOperand(0);
9531*9880d681SAndroid Build Coastguard Worker SDValue Op1 = TheXor->getOperand(1);
9532*9880d681SAndroid Build Coastguard Worker if (Op0.getOpcode() == Op1.getOpcode()) {
9533*9880d681SAndroid Build Coastguard Worker // Avoid missing important xor optimizations.
9534*9880d681SAndroid Build Coastguard Worker if (SDValue Tmp = visitXOR(TheXor)) {
9535*9880d681SAndroid Build Coastguard Worker if (Tmp.getNode() != TheXor) {
9536*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "\nReplacing.8 ";
9537*9880d681SAndroid Build Coastguard Worker TheXor->dump(&DAG);
9538*9880d681SAndroid Build Coastguard Worker dbgs() << "\nWith: ";
9539*9880d681SAndroid Build Coastguard Worker Tmp.getNode()->dump(&DAG);
9540*9880d681SAndroid Build Coastguard Worker dbgs() << '\n');
9541*9880d681SAndroid Build Coastguard Worker WorklistRemover DeadNodes(*this);
9542*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(N1, Tmp);
9543*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(TheXor);
9544*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::BRCOND, SDLoc(N),
9545*9880d681SAndroid Build Coastguard Worker MVT::Other, Chain, Tmp, N2);
9546*9880d681SAndroid Build Coastguard Worker }
9547*9880d681SAndroid Build Coastguard Worker
9548*9880d681SAndroid Build Coastguard Worker // visitXOR has changed XOR's operands or replaced the XOR completely,
9549*9880d681SAndroid Build Coastguard Worker // bail out.
9550*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0);
9551*9880d681SAndroid Build Coastguard Worker }
9552*9880d681SAndroid Build Coastguard Worker }
9553*9880d681SAndroid Build Coastguard Worker
9554*9880d681SAndroid Build Coastguard Worker if (Op0.getOpcode() != ISD::SETCC && Op1.getOpcode() != ISD::SETCC) {
9555*9880d681SAndroid Build Coastguard Worker bool Equal = false;
9556*9880d681SAndroid Build Coastguard Worker if (isOneConstant(Op0) && Op0.hasOneUse() &&
9557*9880d681SAndroid Build Coastguard Worker Op0.getOpcode() == ISD::XOR) {
9558*9880d681SAndroid Build Coastguard Worker TheXor = Op0.getNode();
9559*9880d681SAndroid Build Coastguard Worker Equal = true;
9560*9880d681SAndroid Build Coastguard Worker }
9561*9880d681SAndroid Build Coastguard Worker
9562*9880d681SAndroid Build Coastguard Worker EVT SetCCVT = N1.getValueType();
9563*9880d681SAndroid Build Coastguard Worker if (LegalTypes)
9564*9880d681SAndroid Build Coastguard Worker SetCCVT = getSetCCResultType(SetCCVT);
9565*9880d681SAndroid Build Coastguard Worker SDValue SetCC = DAG.getSetCC(SDLoc(TheXor),
9566*9880d681SAndroid Build Coastguard Worker SetCCVT,
9567*9880d681SAndroid Build Coastguard Worker Op0, Op1,
9568*9880d681SAndroid Build Coastguard Worker Equal ? ISD::SETEQ : ISD::SETNE);
9569*9880d681SAndroid Build Coastguard Worker // Replace the uses of XOR with SETCC
9570*9880d681SAndroid Build Coastguard Worker WorklistRemover DeadNodes(*this);
9571*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(N1, SetCC);
9572*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(N1.getNode());
9573*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::BRCOND, SDLoc(N),
9574*9880d681SAndroid Build Coastguard Worker MVT::Other, Chain, SetCC, N2);
9575*9880d681SAndroid Build Coastguard Worker }
9576*9880d681SAndroid Build Coastguard Worker }
9577*9880d681SAndroid Build Coastguard Worker
9578*9880d681SAndroid Build Coastguard Worker return SDValue();
9579*9880d681SAndroid Build Coastguard Worker }
9580*9880d681SAndroid Build Coastguard Worker
9581*9880d681SAndroid Build Coastguard Worker // Operand List for BR_CC: Chain, CondCC, CondLHS, CondRHS, DestBB.
9582*9880d681SAndroid Build Coastguard Worker //
visitBR_CC(SDNode * N)9583*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitBR_CC(SDNode *N) {
9584*9880d681SAndroid Build Coastguard Worker CondCodeSDNode *CC = cast<CondCodeSDNode>(N->getOperand(1));
9585*9880d681SAndroid Build Coastguard Worker SDValue CondLHS = N->getOperand(2), CondRHS = N->getOperand(3);
9586*9880d681SAndroid Build Coastguard Worker
9587*9880d681SAndroid Build Coastguard Worker // If N is a constant we could fold this into a fallthrough or unconditional
9588*9880d681SAndroid Build Coastguard Worker // branch. However that doesn't happen very often in normal code, because
9589*9880d681SAndroid Build Coastguard Worker // Instcombine/SimplifyCFG should have handled the available opportunities.
9590*9880d681SAndroid Build Coastguard Worker // If we did this folding here, it would be necessary to update the
9591*9880d681SAndroid Build Coastguard Worker // MachineBasicBlock CFG, which is awkward.
9592*9880d681SAndroid Build Coastguard Worker
9593*9880d681SAndroid Build Coastguard Worker // Use SimplifySetCC to simplify SETCC's.
9594*9880d681SAndroid Build Coastguard Worker SDValue Simp = SimplifySetCC(getSetCCResultType(CondLHS.getValueType()),
9595*9880d681SAndroid Build Coastguard Worker CondLHS, CondRHS, CC->get(), SDLoc(N),
9596*9880d681SAndroid Build Coastguard Worker false);
9597*9880d681SAndroid Build Coastguard Worker if (Simp.getNode()) AddToWorklist(Simp.getNode());
9598*9880d681SAndroid Build Coastguard Worker
9599*9880d681SAndroid Build Coastguard Worker // fold to a simpler setcc
9600*9880d681SAndroid Build Coastguard Worker if (Simp.getNode() && Simp.getOpcode() == ISD::SETCC)
9601*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::BR_CC, SDLoc(N), MVT::Other,
9602*9880d681SAndroid Build Coastguard Worker N->getOperand(0), Simp.getOperand(2),
9603*9880d681SAndroid Build Coastguard Worker Simp.getOperand(0), Simp.getOperand(1),
9604*9880d681SAndroid Build Coastguard Worker N->getOperand(4));
9605*9880d681SAndroid Build Coastguard Worker
9606*9880d681SAndroid Build Coastguard Worker return SDValue();
9607*9880d681SAndroid Build Coastguard Worker }
9608*9880d681SAndroid Build Coastguard Worker
9609*9880d681SAndroid Build Coastguard Worker /// Return true if 'Use' is a load or a store that uses N as its base pointer
9610*9880d681SAndroid Build Coastguard Worker /// and that N may be folded in the load / store addressing mode.
canFoldInAddressingMode(SDNode * N,SDNode * Use,SelectionDAG & DAG,const TargetLowering & TLI)9611*9880d681SAndroid Build Coastguard Worker static bool canFoldInAddressingMode(SDNode *N, SDNode *Use,
9612*9880d681SAndroid Build Coastguard Worker SelectionDAG &DAG,
9613*9880d681SAndroid Build Coastguard Worker const TargetLowering &TLI) {
9614*9880d681SAndroid Build Coastguard Worker EVT VT;
9615*9880d681SAndroid Build Coastguard Worker unsigned AS;
9616*9880d681SAndroid Build Coastguard Worker
9617*9880d681SAndroid Build Coastguard Worker if (LoadSDNode *LD = dyn_cast<LoadSDNode>(Use)) {
9618*9880d681SAndroid Build Coastguard Worker if (LD->isIndexed() || LD->getBasePtr().getNode() != N)
9619*9880d681SAndroid Build Coastguard Worker return false;
9620*9880d681SAndroid Build Coastguard Worker VT = LD->getMemoryVT();
9621*9880d681SAndroid Build Coastguard Worker AS = LD->getAddressSpace();
9622*9880d681SAndroid Build Coastguard Worker } else if (StoreSDNode *ST = dyn_cast<StoreSDNode>(Use)) {
9623*9880d681SAndroid Build Coastguard Worker if (ST->isIndexed() || ST->getBasePtr().getNode() != N)
9624*9880d681SAndroid Build Coastguard Worker return false;
9625*9880d681SAndroid Build Coastguard Worker VT = ST->getMemoryVT();
9626*9880d681SAndroid Build Coastguard Worker AS = ST->getAddressSpace();
9627*9880d681SAndroid Build Coastguard Worker } else
9628*9880d681SAndroid Build Coastguard Worker return false;
9629*9880d681SAndroid Build Coastguard Worker
9630*9880d681SAndroid Build Coastguard Worker TargetLowering::AddrMode AM;
9631*9880d681SAndroid Build Coastguard Worker if (N->getOpcode() == ISD::ADD) {
9632*9880d681SAndroid Build Coastguard Worker ConstantSDNode *Offset = dyn_cast<ConstantSDNode>(N->getOperand(1));
9633*9880d681SAndroid Build Coastguard Worker if (Offset)
9634*9880d681SAndroid Build Coastguard Worker // [reg +/- imm]
9635*9880d681SAndroid Build Coastguard Worker AM.BaseOffs = Offset->getSExtValue();
9636*9880d681SAndroid Build Coastguard Worker else
9637*9880d681SAndroid Build Coastguard Worker // [reg +/- reg]
9638*9880d681SAndroid Build Coastguard Worker AM.Scale = 1;
9639*9880d681SAndroid Build Coastguard Worker } else if (N->getOpcode() == ISD::SUB) {
9640*9880d681SAndroid Build Coastguard Worker ConstantSDNode *Offset = dyn_cast<ConstantSDNode>(N->getOperand(1));
9641*9880d681SAndroid Build Coastguard Worker if (Offset)
9642*9880d681SAndroid Build Coastguard Worker // [reg +/- imm]
9643*9880d681SAndroid Build Coastguard Worker AM.BaseOffs = -Offset->getSExtValue();
9644*9880d681SAndroid Build Coastguard Worker else
9645*9880d681SAndroid Build Coastguard Worker // [reg +/- reg]
9646*9880d681SAndroid Build Coastguard Worker AM.Scale = 1;
9647*9880d681SAndroid Build Coastguard Worker } else
9648*9880d681SAndroid Build Coastguard Worker return false;
9649*9880d681SAndroid Build Coastguard Worker
9650*9880d681SAndroid Build Coastguard Worker return TLI.isLegalAddressingMode(DAG.getDataLayout(), AM,
9651*9880d681SAndroid Build Coastguard Worker VT.getTypeForEVT(*DAG.getContext()), AS);
9652*9880d681SAndroid Build Coastguard Worker }
9653*9880d681SAndroid Build Coastguard Worker
9654*9880d681SAndroid Build Coastguard Worker /// Try turning a load/store into a pre-indexed load/store when the base
9655*9880d681SAndroid Build Coastguard Worker /// pointer is an add or subtract and it has other uses besides the load/store.
9656*9880d681SAndroid Build Coastguard Worker /// After the transformation, the new indexed load/store has effectively folded
9657*9880d681SAndroid Build Coastguard Worker /// the add/subtract in and all of its other uses are redirected to the
9658*9880d681SAndroid Build Coastguard Worker /// new load/store.
CombineToPreIndexedLoadStore(SDNode * N)9659*9880d681SAndroid Build Coastguard Worker bool DAGCombiner::CombineToPreIndexedLoadStore(SDNode *N) {
9660*9880d681SAndroid Build Coastguard Worker if (Level < AfterLegalizeDAG)
9661*9880d681SAndroid Build Coastguard Worker return false;
9662*9880d681SAndroid Build Coastguard Worker
9663*9880d681SAndroid Build Coastguard Worker bool isLoad = true;
9664*9880d681SAndroid Build Coastguard Worker SDValue Ptr;
9665*9880d681SAndroid Build Coastguard Worker EVT VT;
9666*9880d681SAndroid Build Coastguard Worker if (LoadSDNode *LD = dyn_cast<LoadSDNode>(N)) {
9667*9880d681SAndroid Build Coastguard Worker if (LD->isIndexed())
9668*9880d681SAndroid Build Coastguard Worker return false;
9669*9880d681SAndroid Build Coastguard Worker VT = LD->getMemoryVT();
9670*9880d681SAndroid Build Coastguard Worker if (!TLI.isIndexedLoadLegal(ISD::PRE_INC, VT) &&
9671*9880d681SAndroid Build Coastguard Worker !TLI.isIndexedLoadLegal(ISD::PRE_DEC, VT))
9672*9880d681SAndroid Build Coastguard Worker return false;
9673*9880d681SAndroid Build Coastguard Worker Ptr = LD->getBasePtr();
9674*9880d681SAndroid Build Coastguard Worker } else if (StoreSDNode *ST = dyn_cast<StoreSDNode>(N)) {
9675*9880d681SAndroid Build Coastguard Worker if (ST->isIndexed())
9676*9880d681SAndroid Build Coastguard Worker return false;
9677*9880d681SAndroid Build Coastguard Worker VT = ST->getMemoryVT();
9678*9880d681SAndroid Build Coastguard Worker if (!TLI.isIndexedStoreLegal(ISD::PRE_INC, VT) &&
9679*9880d681SAndroid Build Coastguard Worker !TLI.isIndexedStoreLegal(ISD::PRE_DEC, VT))
9680*9880d681SAndroid Build Coastguard Worker return false;
9681*9880d681SAndroid Build Coastguard Worker Ptr = ST->getBasePtr();
9682*9880d681SAndroid Build Coastguard Worker isLoad = false;
9683*9880d681SAndroid Build Coastguard Worker } else {
9684*9880d681SAndroid Build Coastguard Worker return false;
9685*9880d681SAndroid Build Coastguard Worker }
9686*9880d681SAndroid Build Coastguard Worker
9687*9880d681SAndroid Build Coastguard Worker // If the pointer is not an add/sub, or if it doesn't have multiple uses, bail
9688*9880d681SAndroid Build Coastguard Worker // out. There is no reason to make this a preinc/predec.
9689*9880d681SAndroid Build Coastguard Worker if ((Ptr.getOpcode() != ISD::ADD && Ptr.getOpcode() != ISD::SUB) ||
9690*9880d681SAndroid Build Coastguard Worker Ptr.getNode()->hasOneUse())
9691*9880d681SAndroid Build Coastguard Worker return false;
9692*9880d681SAndroid Build Coastguard Worker
9693*9880d681SAndroid Build Coastguard Worker // Ask the target to do addressing mode selection.
9694*9880d681SAndroid Build Coastguard Worker SDValue BasePtr;
9695*9880d681SAndroid Build Coastguard Worker SDValue Offset;
9696*9880d681SAndroid Build Coastguard Worker ISD::MemIndexedMode AM = ISD::UNINDEXED;
9697*9880d681SAndroid Build Coastguard Worker if (!TLI.getPreIndexedAddressParts(N, BasePtr, Offset, AM, DAG))
9698*9880d681SAndroid Build Coastguard Worker return false;
9699*9880d681SAndroid Build Coastguard Worker
9700*9880d681SAndroid Build Coastguard Worker // Backends without true r+i pre-indexed forms may need to pass a
9701*9880d681SAndroid Build Coastguard Worker // constant base with a variable offset so that constant coercion
9702*9880d681SAndroid Build Coastguard Worker // will work with the patterns in canonical form.
9703*9880d681SAndroid Build Coastguard Worker bool Swapped = false;
9704*9880d681SAndroid Build Coastguard Worker if (isa<ConstantSDNode>(BasePtr)) {
9705*9880d681SAndroid Build Coastguard Worker std::swap(BasePtr, Offset);
9706*9880d681SAndroid Build Coastguard Worker Swapped = true;
9707*9880d681SAndroid Build Coastguard Worker }
9708*9880d681SAndroid Build Coastguard Worker
9709*9880d681SAndroid Build Coastguard Worker // Don't create a indexed load / store with zero offset.
9710*9880d681SAndroid Build Coastguard Worker if (isNullConstant(Offset))
9711*9880d681SAndroid Build Coastguard Worker return false;
9712*9880d681SAndroid Build Coastguard Worker
9713*9880d681SAndroid Build Coastguard Worker // Try turning it into a pre-indexed load / store except when:
9714*9880d681SAndroid Build Coastguard Worker // 1) The new base ptr is a frame index.
9715*9880d681SAndroid Build Coastguard Worker // 2) If N is a store and the new base ptr is either the same as or is a
9716*9880d681SAndroid Build Coastguard Worker // predecessor of the value being stored.
9717*9880d681SAndroid Build Coastguard Worker // 3) Another use of old base ptr is a predecessor of N. If ptr is folded
9718*9880d681SAndroid Build Coastguard Worker // that would create a cycle.
9719*9880d681SAndroid Build Coastguard Worker // 4) All uses are load / store ops that use it as old base ptr.
9720*9880d681SAndroid Build Coastguard Worker
9721*9880d681SAndroid Build Coastguard Worker // Check #1. Preinc'ing a frame index would require copying the stack pointer
9722*9880d681SAndroid Build Coastguard Worker // (plus the implicit offset) to a register to preinc anyway.
9723*9880d681SAndroid Build Coastguard Worker if (isa<FrameIndexSDNode>(BasePtr) || isa<RegisterSDNode>(BasePtr))
9724*9880d681SAndroid Build Coastguard Worker return false;
9725*9880d681SAndroid Build Coastguard Worker
9726*9880d681SAndroid Build Coastguard Worker // Check #2.
9727*9880d681SAndroid Build Coastguard Worker if (!isLoad) {
9728*9880d681SAndroid Build Coastguard Worker SDValue Val = cast<StoreSDNode>(N)->getValue();
9729*9880d681SAndroid Build Coastguard Worker if (Val == BasePtr || BasePtr.getNode()->isPredecessorOf(Val.getNode()))
9730*9880d681SAndroid Build Coastguard Worker return false;
9731*9880d681SAndroid Build Coastguard Worker }
9732*9880d681SAndroid Build Coastguard Worker
9733*9880d681SAndroid Build Coastguard Worker // Caches for hasPredecessorHelper.
9734*9880d681SAndroid Build Coastguard Worker SmallPtrSet<const SDNode *, 32> Visited;
9735*9880d681SAndroid Build Coastguard Worker SmallVector<const SDNode *, 16> Worklist;
9736*9880d681SAndroid Build Coastguard Worker Worklist.push_back(N);
9737*9880d681SAndroid Build Coastguard Worker
9738*9880d681SAndroid Build Coastguard Worker // If the offset is a constant, there may be other adds of constants that
9739*9880d681SAndroid Build Coastguard Worker // can be folded with this one. We should do this to avoid having to keep
9740*9880d681SAndroid Build Coastguard Worker // a copy of the original base pointer.
9741*9880d681SAndroid Build Coastguard Worker SmallVector<SDNode *, 16> OtherUses;
9742*9880d681SAndroid Build Coastguard Worker if (isa<ConstantSDNode>(Offset))
9743*9880d681SAndroid Build Coastguard Worker for (SDNode::use_iterator UI = BasePtr.getNode()->use_begin(),
9744*9880d681SAndroid Build Coastguard Worker UE = BasePtr.getNode()->use_end();
9745*9880d681SAndroid Build Coastguard Worker UI != UE; ++UI) {
9746*9880d681SAndroid Build Coastguard Worker SDUse &Use = UI.getUse();
9747*9880d681SAndroid Build Coastguard Worker // Skip the use that is Ptr and uses of other results from BasePtr's
9748*9880d681SAndroid Build Coastguard Worker // node (important for nodes that return multiple results).
9749*9880d681SAndroid Build Coastguard Worker if (Use.getUser() == Ptr.getNode() || Use != BasePtr)
9750*9880d681SAndroid Build Coastguard Worker continue;
9751*9880d681SAndroid Build Coastguard Worker
9752*9880d681SAndroid Build Coastguard Worker if (SDNode::hasPredecessorHelper(Use.getUser(), Visited, Worklist))
9753*9880d681SAndroid Build Coastguard Worker continue;
9754*9880d681SAndroid Build Coastguard Worker
9755*9880d681SAndroid Build Coastguard Worker if (Use.getUser()->getOpcode() != ISD::ADD &&
9756*9880d681SAndroid Build Coastguard Worker Use.getUser()->getOpcode() != ISD::SUB) {
9757*9880d681SAndroid Build Coastguard Worker OtherUses.clear();
9758*9880d681SAndroid Build Coastguard Worker break;
9759*9880d681SAndroid Build Coastguard Worker }
9760*9880d681SAndroid Build Coastguard Worker
9761*9880d681SAndroid Build Coastguard Worker SDValue Op1 = Use.getUser()->getOperand((UI.getOperandNo() + 1) & 1);
9762*9880d681SAndroid Build Coastguard Worker if (!isa<ConstantSDNode>(Op1)) {
9763*9880d681SAndroid Build Coastguard Worker OtherUses.clear();
9764*9880d681SAndroid Build Coastguard Worker break;
9765*9880d681SAndroid Build Coastguard Worker }
9766*9880d681SAndroid Build Coastguard Worker
9767*9880d681SAndroid Build Coastguard Worker // FIXME: In some cases, we can be smarter about this.
9768*9880d681SAndroid Build Coastguard Worker if (Op1.getValueType() != Offset.getValueType()) {
9769*9880d681SAndroid Build Coastguard Worker OtherUses.clear();
9770*9880d681SAndroid Build Coastguard Worker break;
9771*9880d681SAndroid Build Coastguard Worker }
9772*9880d681SAndroid Build Coastguard Worker
9773*9880d681SAndroid Build Coastguard Worker OtherUses.push_back(Use.getUser());
9774*9880d681SAndroid Build Coastguard Worker }
9775*9880d681SAndroid Build Coastguard Worker
9776*9880d681SAndroid Build Coastguard Worker if (Swapped)
9777*9880d681SAndroid Build Coastguard Worker std::swap(BasePtr, Offset);
9778*9880d681SAndroid Build Coastguard Worker
9779*9880d681SAndroid Build Coastguard Worker // Now check for #3 and #4.
9780*9880d681SAndroid Build Coastguard Worker bool RealUse = false;
9781*9880d681SAndroid Build Coastguard Worker
9782*9880d681SAndroid Build Coastguard Worker for (SDNode *Use : Ptr.getNode()->uses()) {
9783*9880d681SAndroid Build Coastguard Worker if (Use == N)
9784*9880d681SAndroid Build Coastguard Worker continue;
9785*9880d681SAndroid Build Coastguard Worker if (SDNode::hasPredecessorHelper(Use, Visited, Worklist))
9786*9880d681SAndroid Build Coastguard Worker return false;
9787*9880d681SAndroid Build Coastguard Worker
9788*9880d681SAndroid Build Coastguard Worker // If Ptr may be folded in addressing mode of other use, then it's
9789*9880d681SAndroid Build Coastguard Worker // not profitable to do this transformation.
9790*9880d681SAndroid Build Coastguard Worker if (!canFoldInAddressingMode(Ptr.getNode(), Use, DAG, TLI))
9791*9880d681SAndroid Build Coastguard Worker RealUse = true;
9792*9880d681SAndroid Build Coastguard Worker }
9793*9880d681SAndroid Build Coastguard Worker
9794*9880d681SAndroid Build Coastguard Worker if (!RealUse)
9795*9880d681SAndroid Build Coastguard Worker return false;
9796*9880d681SAndroid Build Coastguard Worker
9797*9880d681SAndroid Build Coastguard Worker SDValue Result;
9798*9880d681SAndroid Build Coastguard Worker if (isLoad)
9799*9880d681SAndroid Build Coastguard Worker Result = DAG.getIndexedLoad(SDValue(N,0), SDLoc(N),
9800*9880d681SAndroid Build Coastguard Worker BasePtr, Offset, AM);
9801*9880d681SAndroid Build Coastguard Worker else
9802*9880d681SAndroid Build Coastguard Worker Result = DAG.getIndexedStore(SDValue(N,0), SDLoc(N),
9803*9880d681SAndroid Build Coastguard Worker BasePtr, Offset, AM);
9804*9880d681SAndroid Build Coastguard Worker ++PreIndexedNodes;
9805*9880d681SAndroid Build Coastguard Worker ++NodesCombined;
9806*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "\nReplacing.4 ";
9807*9880d681SAndroid Build Coastguard Worker N->dump(&DAG);
9808*9880d681SAndroid Build Coastguard Worker dbgs() << "\nWith: ";
9809*9880d681SAndroid Build Coastguard Worker Result.getNode()->dump(&DAG);
9810*9880d681SAndroid Build Coastguard Worker dbgs() << '\n');
9811*9880d681SAndroid Build Coastguard Worker WorklistRemover DeadNodes(*this);
9812*9880d681SAndroid Build Coastguard Worker if (isLoad) {
9813*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(N, 0), Result.getValue(0));
9814*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(N, 1), Result.getValue(2));
9815*9880d681SAndroid Build Coastguard Worker } else {
9816*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(N, 0), Result.getValue(1));
9817*9880d681SAndroid Build Coastguard Worker }
9818*9880d681SAndroid Build Coastguard Worker
9819*9880d681SAndroid Build Coastguard Worker // Finally, since the node is now dead, remove it from the graph.
9820*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(N);
9821*9880d681SAndroid Build Coastguard Worker
9822*9880d681SAndroid Build Coastguard Worker if (Swapped)
9823*9880d681SAndroid Build Coastguard Worker std::swap(BasePtr, Offset);
9824*9880d681SAndroid Build Coastguard Worker
9825*9880d681SAndroid Build Coastguard Worker // Replace other uses of BasePtr that can be updated to use Ptr
9826*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0, e = OtherUses.size(); i != e; ++i) {
9827*9880d681SAndroid Build Coastguard Worker unsigned OffsetIdx = 1;
9828*9880d681SAndroid Build Coastguard Worker if (OtherUses[i]->getOperand(OffsetIdx).getNode() == BasePtr.getNode())
9829*9880d681SAndroid Build Coastguard Worker OffsetIdx = 0;
9830*9880d681SAndroid Build Coastguard Worker assert(OtherUses[i]->getOperand(!OffsetIdx).getNode() ==
9831*9880d681SAndroid Build Coastguard Worker BasePtr.getNode() && "Expected BasePtr operand");
9832*9880d681SAndroid Build Coastguard Worker
9833*9880d681SAndroid Build Coastguard Worker // We need to replace ptr0 in the following expression:
9834*9880d681SAndroid Build Coastguard Worker // x0 * offset0 + y0 * ptr0 = t0
9835*9880d681SAndroid Build Coastguard Worker // knowing that
9836*9880d681SAndroid Build Coastguard Worker // x1 * offset1 + y1 * ptr0 = t1 (the indexed load/store)
9837*9880d681SAndroid Build Coastguard Worker //
9838*9880d681SAndroid Build Coastguard Worker // where x0, x1, y0 and y1 in {-1, 1} are given by the types of the
9839*9880d681SAndroid Build Coastguard Worker // indexed load/store and the expresion that needs to be re-written.
9840*9880d681SAndroid Build Coastguard Worker //
9841*9880d681SAndroid Build Coastguard Worker // Therefore, we have:
9842*9880d681SAndroid Build Coastguard Worker // t0 = (x0 * offset0 - x1 * y0 * y1 *offset1) + (y0 * y1) * t1
9843*9880d681SAndroid Build Coastguard Worker
9844*9880d681SAndroid Build Coastguard Worker ConstantSDNode *CN =
9845*9880d681SAndroid Build Coastguard Worker cast<ConstantSDNode>(OtherUses[i]->getOperand(OffsetIdx));
9846*9880d681SAndroid Build Coastguard Worker int X0, X1, Y0, Y1;
9847*9880d681SAndroid Build Coastguard Worker const APInt &Offset0 = CN->getAPIntValue();
9848*9880d681SAndroid Build Coastguard Worker APInt Offset1 = cast<ConstantSDNode>(Offset)->getAPIntValue();
9849*9880d681SAndroid Build Coastguard Worker
9850*9880d681SAndroid Build Coastguard Worker X0 = (OtherUses[i]->getOpcode() == ISD::SUB && OffsetIdx == 1) ? -1 : 1;
9851*9880d681SAndroid Build Coastguard Worker Y0 = (OtherUses[i]->getOpcode() == ISD::SUB && OffsetIdx == 0) ? -1 : 1;
9852*9880d681SAndroid Build Coastguard Worker X1 = (AM == ISD::PRE_DEC && !Swapped) ? -1 : 1;
9853*9880d681SAndroid Build Coastguard Worker Y1 = (AM == ISD::PRE_DEC && Swapped) ? -1 : 1;
9854*9880d681SAndroid Build Coastguard Worker
9855*9880d681SAndroid Build Coastguard Worker unsigned Opcode = (Y0 * Y1 < 0) ? ISD::SUB : ISD::ADD;
9856*9880d681SAndroid Build Coastguard Worker
9857*9880d681SAndroid Build Coastguard Worker APInt CNV = Offset0;
9858*9880d681SAndroid Build Coastguard Worker if (X0 < 0) CNV = -CNV;
9859*9880d681SAndroid Build Coastguard Worker if (X1 * Y0 * Y1 < 0) CNV = CNV + Offset1;
9860*9880d681SAndroid Build Coastguard Worker else CNV = CNV - Offset1;
9861*9880d681SAndroid Build Coastguard Worker
9862*9880d681SAndroid Build Coastguard Worker SDLoc DL(OtherUses[i]);
9863*9880d681SAndroid Build Coastguard Worker
9864*9880d681SAndroid Build Coastguard Worker // We can now generate the new expression.
9865*9880d681SAndroid Build Coastguard Worker SDValue NewOp1 = DAG.getConstant(CNV, DL, CN->getValueType(0));
9866*9880d681SAndroid Build Coastguard Worker SDValue NewOp2 = Result.getValue(isLoad ? 1 : 0);
9867*9880d681SAndroid Build Coastguard Worker
9868*9880d681SAndroid Build Coastguard Worker SDValue NewUse = DAG.getNode(Opcode,
9869*9880d681SAndroid Build Coastguard Worker DL,
9870*9880d681SAndroid Build Coastguard Worker OtherUses[i]->getValueType(0), NewOp1, NewOp2);
9871*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(OtherUses[i], 0), NewUse);
9872*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(OtherUses[i]);
9873*9880d681SAndroid Build Coastguard Worker }
9874*9880d681SAndroid Build Coastguard Worker
9875*9880d681SAndroid Build Coastguard Worker // Replace the uses of Ptr with uses of the updated base value.
9876*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(Ptr, Result.getValue(isLoad ? 1 : 0));
9877*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(Ptr.getNode());
9878*9880d681SAndroid Build Coastguard Worker
9879*9880d681SAndroid Build Coastguard Worker return true;
9880*9880d681SAndroid Build Coastguard Worker }
9881*9880d681SAndroid Build Coastguard Worker
9882*9880d681SAndroid Build Coastguard Worker /// Try to combine a load/store with a add/sub of the base pointer node into a
9883*9880d681SAndroid Build Coastguard Worker /// post-indexed load/store. The transformation folded the add/subtract into the
9884*9880d681SAndroid Build Coastguard Worker /// new indexed load/store effectively and all of its uses are redirected to the
9885*9880d681SAndroid Build Coastguard Worker /// new load/store.
CombineToPostIndexedLoadStore(SDNode * N)9886*9880d681SAndroid Build Coastguard Worker bool DAGCombiner::CombineToPostIndexedLoadStore(SDNode *N) {
9887*9880d681SAndroid Build Coastguard Worker if (Level < AfterLegalizeDAG)
9888*9880d681SAndroid Build Coastguard Worker return false;
9889*9880d681SAndroid Build Coastguard Worker
9890*9880d681SAndroid Build Coastguard Worker bool isLoad = true;
9891*9880d681SAndroid Build Coastguard Worker SDValue Ptr;
9892*9880d681SAndroid Build Coastguard Worker EVT VT;
9893*9880d681SAndroid Build Coastguard Worker if (LoadSDNode *LD = dyn_cast<LoadSDNode>(N)) {
9894*9880d681SAndroid Build Coastguard Worker if (LD->isIndexed())
9895*9880d681SAndroid Build Coastguard Worker return false;
9896*9880d681SAndroid Build Coastguard Worker VT = LD->getMemoryVT();
9897*9880d681SAndroid Build Coastguard Worker if (!TLI.isIndexedLoadLegal(ISD::POST_INC, VT) &&
9898*9880d681SAndroid Build Coastguard Worker !TLI.isIndexedLoadLegal(ISD::POST_DEC, VT))
9899*9880d681SAndroid Build Coastguard Worker return false;
9900*9880d681SAndroid Build Coastguard Worker Ptr = LD->getBasePtr();
9901*9880d681SAndroid Build Coastguard Worker } else if (StoreSDNode *ST = dyn_cast<StoreSDNode>(N)) {
9902*9880d681SAndroid Build Coastguard Worker if (ST->isIndexed())
9903*9880d681SAndroid Build Coastguard Worker return false;
9904*9880d681SAndroid Build Coastguard Worker VT = ST->getMemoryVT();
9905*9880d681SAndroid Build Coastguard Worker if (!TLI.isIndexedStoreLegal(ISD::POST_INC, VT) &&
9906*9880d681SAndroid Build Coastguard Worker !TLI.isIndexedStoreLegal(ISD::POST_DEC, VT))
9907*9880d681SAndroid Build Coastguard Worker return false;
9908*9880d681SAndroid Build Coastguard Worker Ptr = ST->getBasePtr();
9909*9880d681SAndroid Build Coastguard Worker isLoad = false;
9910*9880d681SAndroid Build Coastguard Worker } else {
9911*9880d681SAndroid Build Coastguard Worker return false;
9912*9880d681SAndroid Build Coastguard Worker }
9913*9880d681SAndroid Build Coastguard Worker
9914*9880d681SAndroid Build Coastguard Worker if (Ptr.getNode()->hasOneUse())
9915*9880d681SAndroid Build Coastguard Worker return false;
9916*9880d681SAndroid Build Coastguard Worker
9917*9880d681SAndroid Build Coastguard Worker for (SDNode *Op : Ptr.getNode()->uses()) {
9918*9880d681SAndroid Build Coastguard Worker if (Op == N ||
9919*9880d681SAndroid Build Coastguard Worker (Op->getOpcode() != ISD::ADD && Op->getOpcode() != ISD::SUB))
9920*9880d681SAndroid Build Coastguard Worker continue;
9921*9880d681SAndroid Build Coastguard Worker
9922*9880d681SAndroid Build Coastguard Worker SDValue BasePtr;
9923*9880d681SAndroid Build Coastguard Worker SDValue Offset;
9924*9880d681SAndroid Build Coastguard Worker ISD::MemIndexedMode AM = ISD::UNINDEXED;
9925*9880d681SAndroid Build Coastguard Worker if (TLI.getPostIndexedAddressParts(N, Op, BasePtr, Offset, AM, DAG)) {
9926*9880d681SAndroid Build Coastguard Worker // Don't create a indexed load / store with zero offset.
9927*9880d681SAndroid Build Coastguard Worker if (isNullConstant(Offset))
9928*9880d681SAndroid Build Coastguard Worker continue;
9929*9880d681SAndroid Build Coastguard Worker
9930*9880d681SAndroid Build Coastguard Worker // Try turning it into a post-indexed load / store except when
9931*9880d681SAndroid Build Coastguard Worker // 1) All uses are load / store ops that use it as base ptr (and
9932*9880d681SAndroid Build Coastguard Worker // it may be folded as addressing mmode).
9933*9880d681SAndroid Build Coastguard Worker // 2) Op must be independent of N, i.e. Op is neither a predecessor
9934*9880d681SAndroid Build Coastguard Worker // nor a successor of N. Otherwise, if Op is folded that would
9935*9880d681SAndroid Build Coastguard Worker // create a cycle.
9936*9880d681SAndroid Build Coastguard Worker
9937*9880d681SAndroid Build Coastguard Worker if (isa<FrameIndexSDNode>(BasePtr) || isa<RegisterSDNode>(BasePtr))
9938*9880d681SAndroid Build Coastguard Worker continue;
9939*9880d681SAndroid Build Coastguard Worker
9940*9880d681SAndroid Build Coastguard Worker // Check for #1.
9941*9880d681SAndroid Build Coastguard Worker bool TryNext = false;
9942*9880d681SAndroid Build Coastguard Worker for (SDNode *Use : BasePtr.getNode()->uses()) {
9943*9880d681SAndroid Build Coastguard Worker if (Use == Ptr.getNode())
9944*9880d681SAndroid Build Coastguard Worker continue;
9945*9880d681SAndroid Build Coastguard Worker
9946*9880d681SAndroid Build Coastguard Worker // If all the uses are load / store addresses, then don't do the
9947*9880d681SAndroid Build Coastguard Worker // transformation.
9948*9880d681SAndroid Build Coastguard Worker if (Use->getOpcode() == ISD::ADD || Use->getOpcode() == ISD::SUB){
9949*9880d681SAndroid Build Coastguard Worker bool RealUse = false;
9950*9880d681SAndroid Build Coastguard Worker for (SDNode *UseUse : Use->uses()) {
9951*9880d681SAndroid Build Coastguard Worker if (!canFoldInAddressingMode(Use, UseUse, DAG, TLI))
9952*9880d681SAndroid Build Coastguard Worker RealUse = true;
9953*9880d681SAndroid Build Coastguard Worker }
9954*9880d681SAndroid Build Coastguard Worker
9955*9880d681SAndroid Build Coastguard Worker if (!RealUse) {
9956*9880d681SAndroid Build Coastguard Worker TryNext = true;
9957*9880d681SAndroid Build Coastguard Worker break;
9958*9880d681SAndroid Build Coastguard Worker }
9959*9880d681SAndroid Build Coastguard Worker }
9960*9880d681SAndroid Build Coastguard Worker }
9961*9880d681SAndroid Build Coastguard Worker
9962*9880d681SAndroid Build Coastguard Worker if (TryNext)
9963*9880d681SAndroid Build Coastguard Worker continue;
9964*9880d681SAndroid Build Coastguard Worker
9965*9880d681SAndroid Build Coastguard Worker // Check for #2
9966*9880d681SAndroid Build Coastguard Worker if (!Op->isPredecessorOf(N) && !N->isPredecessorOf(Op)) {
9967*9880d681SAndroid Build Coastguard Worker SDValue Result = isLoad
9968*9880d681SAndroid Build Coastguard Worker ? DAG.getIndexedLoad(SDValue(N,0), SDLoc(N),
9969*9880d681SAndroid Build Coastguard Worker BasePtr, Offset, AM)
9970*9880d681SAndroid Build Coastguard Worker : DAG.getIndexedStore(SDValue(N,0), SDLoc(N),
9971*9880d681SAndroid Build Coastguard Worker BasePtr, Offset, AM);
9972*9880d681SAndroid Build Coastguard Worker ++PostIndexedNodes;
9973*9880d681SAndroid Build Coastguard Worker ++NodesCombined;
9974*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "\nReplacing.5 ";
9975*9880d681SAndroid Build Coastguard Worker N->dump(&DAG);
9976*9880d681SAndroid Build Coastguard Worker dbgs() << "\nWith: ";
9977*9880d681SAndroid Build Coastguard Worker Result.getNode()->dump(&DAG);
9978*9880d681SAndroid Build Coastguard Worker dbgs() << '\n');
9979*9880d681SAndroid Build Coastguard Worker WorklistRemover DeadNodes(*this);
9980*9880d681SAndroid Build Coastguard Worker if (isLoad) {
9981*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(N, 0), Result.getValue(0));
9982*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(N, 1), Result.getValue(2));
9983*9880d681SAndroid Build Coastguard Worker } else {
9984*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(N, 0), Result.getValue(1));
9985*9880d681SAndroid Build Coastguard Worker }
9986*9880d681SAndroid Build Coastguard Worker
9987*9880d681SAndroid Build Coastguard Worker // Finally, since the node is now dead, remove it from the graph.
9988*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(N);
9989*9880d681SAndroid Build Coastguard Worker
9990*9880d681SAndroid Build Coastguard Worker // Replace the uses of Use with uses of the updated base value.
9991*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(Op, 0),
9992*9880d681SAndroid Build Coastguard Worker Result.getValue(isLoad ? 1 : 0));
9993*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(Op);
9994*9880d681SAndroid Build Coastguard Worker return true;
9995*9880d681SAndroid Build Coastguard Worker }
9996*9880d681SAndroid Build Coastguard Worker }
9997*9880d681SAndroid Build Coastguard Worker }
9998*9880d681SAndroid Build Coastguard Worker
9999*9880d681SAndroid Build Coastguard Worker return false;
10000*9880d681SAndroid Build Coastguard Worker }
10001*9880d681SAndroid Build Coastguard Worker
10002*9880d681SAndroid Build Coastguard Worker /// \brief Return the base-pointer arithmetic from an indexed \p LD.
SplitIndexingFromLoad(LoadSDNode * LD)10003*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::SplitIndexingFromLoad(LoadSDNode *LD) {
10004*9880d681SAndroid Build Coastguard Worker ISD::MemIndexedMode AM = LD->getAddressingMode();
10005*9880d681SAndroid Build Coastguard Worker assert(AM != ISD::UNINDEXED);
10006*9880d681SAndroid Build Coastguard Worker SDValue BP = LD->getOperand(1);
10007*9880d681SAndroid Build Coastguard Worker SDValue Inc = LD->getOperand(2);
10008*9880d681SAndroid Build Coastguard Worker
10009*9880d681SAndroid Build Coastguard Worker // Some backends use TargetConstants for load offsets, but don't expect
10010*9880d681SAndroid Build Coastguard Worker // TargetConstants in general ADD nodes. We can convert these constants into
10011*9880d681SAndroid Build Coastguard Worker // regular Constants (if the constant is not opaque).
10012*9880d681SAndroid Build Coastguard Worker assert((Inc.getOpcode() != ISD::TargetConstant ||
10013*9880d681SAndroid Build Coastguard Worker !cast<ConstantSDNode>(Inc)->isOpaque()) &&
10014*9880d681SAndroid Build Coastguard Worker "Cannot split out indexing using opaque target constants");
10015*9880d681SAndroid Build Coastguard Worker if (Inc.getOpcode() == ISD::TargetConstant) {
10016*9880d681SAndroid Build Coastguard Worker ConstantSDNode *ConstInc = cast<ConstantSDNode>(Inc);
10017*9880d681SAndroid Build Coastguard Worker Inc = DAG.getConstant(*ConstInc->getConstantIntValue(), SDLoc(Inc),
10018*9880d681SAndroid Build Coastguard Worker ConstInc->getValueType(0));
10019*9880d681SAndroid Build Coastguard Worker }
10020*9880d681SAndroid Build Coastguard Worker
10021*9880d681SAndroid Build Coastguard Worker unsigned Opc =
10022*9880d681SAndroid Build Coastguard Worker (AM == ISD::PRE_INC || AM == ISD::POST_INC ? ISD::ADD : ISD::SUB);
10023*9880d681SAndroid Build Coastguard Worker return DAG.getNode(Opc, SDLoc(LD), BP.getSimpleValueType(), BP, Inc);
10024*9880d681SAndroid Build Coastguard Worker }
10025*9880d681SAndroid Build Coastguard Worker
visitLOAD(SDNode * N)10026*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitLOAD(SDNode *N) {
10027*9880d681SAndroid Build Coastguard Worker LoadSDNode *LD = cast<LoadSDNode>(N);
10028*9880d681SAndroid Build Coastguard Worker SDValue Chain = LD->getChain();
10029*9880d681SAndroid Build Coastguard Worker SDValue Ptr = LD->getBasePtr();
10030*9880d681SAndroid Build Coastguard Worker
10031*9880d681SAndroid Build Coastguard Worker // If load is not volatile and there are no uses of the loaded value (and
10032*9880d681SAndroid Build Coastguard Worker // the updated indexed value in case of indexed loads), change uses of the
10033*9880d681SAndroid Build Coastguard Worker // chain value into uses of the chain input (i.e. delete the dead load).
10034*9880d681SAndroid Build Coastguard Worker if (!LD->isVolatile()) {
10035*9880d681SAndroid Build Coastguard Worker if (N->getValueType(1) == MVT::Other) {
10036*9880d681SAndroid Build Coastguard Worker // Unindexed loads.
10037*9880d681SAndroid Build Coastguard Worker if (!N->hasAnyUseOfValue(0)) {
10038*9880d681SAndroid Build Coastguard Worker // It's not safe to use the two value CombineTo variant here. e.g.
10039*9880d681SAndroid Build Coastguard Worker // v1, chain2 = load chain1, loc
10040*9880d681SAndroid Build Coastguard Worker // v2, chain3 = load chain2, loc
10041*9880d681SAndroid Build Coastguard Worker // v3 = add v2, c
10042*9880d681SAndroid Build Coastguard Worker // Now we replace use of chain2 with chain1. This makes the second load
10043*9880d681SAndroid Build Coastguard Worker // isomorphic to the one we are deleting, and thus makes this load live.
10044*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "\nReplacing.6 ";
10045*9880d681SAndroid Build Coastguard Worker N->dump(&DAG);
10046*9880d681SAndroid Build Coastguard Worker dbgs() << "\nWith chain: ";
10047*9880d681SAndroid Build Coastguard Worker Chain.getNode()->dump(&DAG);
10048*9880d681SAndroid Build Coastguard Worker dbgs() << "\n");
10049*9880d681SAndroid Build Coastguard Worker WorklistRemover DeadNodes(*this);
10050*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(N, 1), Chain);
10051*9880d681SAndroid Build Coastguard Worker
10052*9880d681SAndroid Build Coastguard Worker if (N->use_empty())
10053*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(N);
10054*9880d681SAndroid Build Coastguard Worker
10055*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
10056*9880d681SAndroid Build Coastguard Worker }
10057*9880d681SAndroid Build Coastguard Worker } else {
10058*9880d681SAndroid Build Coastguard Worker // Indexed loads.
10059*9880d681SAndroid Build Coastguard Worker assert(N->getValueType(2) == MVT::Other && "Malformed indexed loads?");
10060*9880d681SAndroid Build Coastguard Worker
10061*9880d681SAndroid Build Coastguard Worker // If this load has an opaque TargetConstant offset, then we cannot split
10062*9880d681SAndroid Build Coastguard Worker // the indexing into an add/sub directly (that TargetConstant may not be
10063*9880d681SAndroid Build Coastguard Worker // valid for a different type of node, and we cannot convert an opaque
10064*9880d681SAndroid Build Coastguard Worker // target constant into a regular constant).
10065*9880d681SAndroid Build Coastguard Worker bool HasOTCInc = LD->getOperand(2).getOpcode() == ISD::TargetConstant &&
10066*9880d681SAndroid Build Coastguard Worker cast<ConstantSDNode>(LD->getOperand(2))->isOpaque();
10067*9880d681SAndroid Build Coastguard Worker
10068*9880d681SAndroid Build Coastguard Worker if (!N->hasAnyUseOfValue(0) &&
10069*9880d681SAndroid Build Coastguard Worker ((MaySplitLoadIndex && !HasOTCInc) || !N->hasAnyUseOfValue(1))) {
10070*9880d681SAndroid Build Coastguard Worker SDValue Undef = DAG.getUNDEF(N->getValueType(0));
10071*9880d681SAndroid Build Coastguard Worker SDValue Index;
10072*9880d681SAndroid Build Coastguard Worker if (N->hasAnyUseOfValue(1) && MaySplitLoadIndex && !HasOTCInc) {
10073*9880d681SAndroid Build Coastguard Worker Index = SplitIndexingFromLoad(LD);
10074*9880d681SAndroid Build Coastguard Worker // Try to fold the base pointer arithmetic into subsequent loads and
10075*9880d681SAndroid Build Coastguard Worker // stores.
10076*9880d681SAndroid Build Coastguard Worker AddUsersToWorklist(N);
10077*9880d681SAndroid Build Coastguard Worker } else
10078*9880d681SAndroid Build Coastguard Worker Index = DAG.getUNDEF(N->getValueType(1));
10079*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "\nReplacing.7 ";
10080*9880d681SAndroid Build Coastguard Worker N->dump(&DAG);
10081*9880d681SAndroid Build Coastguard Worker dbgs() << "\nWith: ";
10082*9880d681SAndroid Build Coastguard Worker Undef.getNode()->dump(&DAG);
10083*9880d681SAndroid Build Coastguard Worker dbgs() << " and 2 other values\n");
10084*9880d681SAndroid Build Coastguard Worker WorklistRemover DeadNodes(*this);
10085*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(N, 0), Undef);
10086*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(N, 1), Index);
10087*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(N, 2), Chain);
10088*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(N);
10089*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0); // Return N so it doesn't get rechecked!
10090*9880d681SAndroid Build Coastguard Worker }
10091*9880d681SAndroid Build Coastguard Worker }
10092*9880d681SAndroid Build Coastguard Worker }
10093*9880d681SAndroid Build Coastguard Worker
10094*9880d681SAndroid Build Coastguard Worker // If this load is directly stored, replace the load value with the stored
10095*9880d681SAndroid Build Coastguard Worker // value.
10096*9880d681SAndroid Build Coastguard Worker // TODO: Handle store large -> read small portion.
10097*9880d681SAndroid Build Coastguard Worker // TODO: Handle TRUNCSTORE/LOADEXT
10098*9880d681SAndroid Build Coastguard Worker if (ISD::isNormalLoad(N) && !LD->isVolatile()) {
10099*9880d681SAndroid Build Coastguard Worker if (ISD::isNON_TRUNCStore(Chain.getNode())) {
10100*9880d681SAndroid Build Coastguard Worker StoreSDNode *PrevST = cast<StoreSDNode>(Chain);
10101*9880d681SAndroid Build Coastguard Worker if (PrevST->getBasePtr() == Ptr &&
10102*9880d681SAndroid Build Coastguard Worker PrevST->getValue().getValueType() == N->getValueType(0))
10103*9880d681SAndroid Build Coastguard Worker return CombineTo(N, Chain.getOperand(1), Chain);
10104*9880d681SAndroid Build Coastguard Worker }
10105*9880d681SAndroid Build Coastguard Worker }
10106*9880d681SAndroid Build Coastguard Worker
10107*9880d681SAndroid Build Coastguard Worker // Try to infer better alignment information than the load already has.
10108*9880d681SAndroid Build Coastguard Worker if (OptLevel != CodeGenOpt::None && LD->isUnindexed()) {
10109*9880d681SAndroid Build Coastguard Worker if (unsigned Align = DAG.InferPtrAlignment(Ptr)) {
10110*9880d681SAndroid Build Coastguard Worker if (Align > LD->getMemOperand()->getBaseAlignment()) {
10111*9880d681SAndroid Build Coastguard Worker SDValue NewLoad =
10112*9880d681SAndroid Build Coastguard Worker DAG.getExtLoad(LD->getExtensionType(), SDLoc(N),
10113*9880d681SAndroid Build Coastguard Worker LD->getValueType(0),
10114*9880d681SAndroid Build Coastguard Worker Chain, Ptr, LD->getPointerInfo(),
10115*9880d681SAndroid Build Coastguard Worker LD->getMemoryVT(),
10116*9880d681SAndroid Build Coastguard Worker LD->isVolatile(), LD->isNonTemporal(),
10117*9880d681SAndroid Build Coastguard Worker LD->isInvariant(), Align, LD->getAAInfo());
10118*9880d681SAndroid Build Coastguard Worker if (NewLoad.getNode() != N)
10119*9880d681SAndroid Build Coastguard Worker return CombineTo(N, NewLoad, SDValue(NewLoad.getNode(), 1), true);
10120*9880d681SAndroid Build Coastguard Worker }
10121*9880d681SAndroid Build Coastguard Worker }
10122*9880d681SAndroid Build Coastguard Worker }
10123*9880d681SAndroid Build Coastguard Worker
10124*9880d681SAndroid Build Coastguard Worker bool UseAA = CombinerAA.getNumOccurrences() > 0 ? CombinerAA
10125*9880d681SAndroid Build Coastguard Worker : DAG.getSubtarget().useAA();
10126*9880d681SAndroid Build Coastguard Worker #ifndef NDEBUG
10127*9880d681SAndroid Build Coastguard Worker if (CombinerAAOnlyFunc.getNumOccurrences() &&
10128*9880d681SAndroid Build Coastguard Worker CombinerAAOnlyFunc != DAG.getMachineFunction().getName())
10129*9880d681SAndroid Build Coastguard Worker UseAA = false;
10130*9880d681SAndroid Build Coastguard Worker #endif
10131*9880d681SAndroid Build Coastguard Worker if (UseAA && LD->isUnindexed()) {
10132*9880d681SAndroid Build Coastguard Worker // Walk up chain skipping non-aliasing memory nodes.
10133*9880d681SAndroid Build Coastguard Worker SDValue BetterChain = FindBetterChain(N, Chain);
10134*9880d681SAndroid Build Coastguard Worker
10135*9880d681SAndroid Build Coastguard Worker // If there is a better chain.
10136*9880d681SAndroid Build Coastguard Worker if (Chain != BetterChain) {
10137*9880d681SAndroid Build Coastguard Worker SDValue ReplLoad;
10138*9880d681SAndroid Build Coastguard Worker
10139*9880d681SAndroid Build Coastguard Worker // Replace the chain to void dependency.
10140*9880d681SAndroid Build Coastguard Worker if (LD->getExtensionType() == ISD::NON_EXTLOAD) {
10141*9880d681SAndroid Build Coastguard Worker ReplLoad = DAG.getLoad(N->getValueType(0), SDLoc(LD),
10142*9880d681SAndroid Build Coastguard Worker BetterChain, Ptr, LD->getMemOperand());
10143*9880d681SAndroid Build Coastguard Worker } else {
10144*9880d681SAndroid Build Coastguard Worker ReplLoad = DAG.getExtLoad(LD->getExtensionType(), SDLoc(LD),
10145*9880d681SAndroid Build Coastguard Worker LD->getValueType(0),
10146*9880d681SAndroid Build Coastguard Worker BetterChain, Ptr, LD->getMemoryVT(),
10147*9880d681SAndroid Build Coastguard Worker LD->getMemOperand());
10148*9880d681SAndroid Build Coastguard Worker }
10149*9880d681SAndroid Build Coastguard Worker
10150*9880d681SAndroid Build Coastguard Worker // Create token factor to keep old chain connected.
10151*9880d681SAndroid Build Coastguard Worker SDValue Token = DAG.getNode(ISD::TokenFactor, SDLoc(N),
10152*9880d681SAndroid Build Coastguard Worker MVT::Other, Chain, ReplLoad.getValue(1));
10153*9880d681SAndroid Build Coastguard Worker
10154*9880d681SAndroid Build Coastguard Worker // Make sure the new and old chains are cleaned up.
10155*9880d681SAndroid Build Coastguard Worker AddToWorklist(Token.getNode());
10156*9880d681SAndroid Build Coastguard Worker
10157*9880d681SAndroid Build Coastguard Worker // Replace uses with load result and token factor. Don't add users
10158*9880d681SAndroid Build Coastguard Worker // to work list.
10159*9880d681SAndroid Build Coastguard Worker return CombineTo(N, ReplLoad.getValue(0), Token, false);
10160*9880d681SAndroid Build Coastguard Worker }
10161*9880d681SAndroid Build Coastguard Worker }
10162*9880d681SAndroid Build Coastguard Worker
10163*9880d681SAndroid Build Coastguard Worker // Try transforming N to an indexed load.
10164*9880d681SAndroid Build Coastguard Worker if (CombineToPreIndexedLoadStore(N) || CombineToPostIndexedLoadStore(N))
10165*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0);
10166*9880d681SAndroid Build Coastguard Worker
10167*9880d681SAndroid Build Coastguard Worker // Try to slice up N to more direct loads if the slices are mapped to
10168*9880d681SAndroid Build Coastguard Worker // different register banks or pairing can take place.
10169*9880d681SAndroid Build Coastguard Worker if (SliceUpLoad(N))
10170*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0);
10171*9880d681SAndroid Build Coastguard Worker
10172*9880d681SAndroid Build Coastguard Worker return SDValue();
10173*9880d681SAndroid Build Coastguard Worker }
10174*9880d681SAndroid Build Coastguard Worker
10175*9880d681SAndroid Build Coastguard Worker namespace {
10176*9880d681SAndroid Build Coastguard Worker /// \brief Helper structure used to slice a load in smaller loads.
10177*9880d681SAndroid Build Coastguard Worker /// Basically a slice is obtained from the following sequence:
10178*9880d681SAndroid Build Coastguard Worker /// Origin = load Ty1, Base
10179*9880d681SAndroid Build Coastguard Worker /// Shift = srl Ty1 Origin, CstTy Amount
10180*9880d681SAndroid Build Coastguard Worker /// Inst = trunc Shift to Ty2
10181*9880d681SAndroid Build Coastguard Worker ///
10182*9880d681SAndroid Build Coastguard Worker /// Then, it will be rewriten into:
10183*9880d681SAndroid Build Coastguard Worker /// Slice = load SliceTy, Base + SliceOffset
10184*9880d681SAndroid Build Coastguard Worker /// [Inst = zext Slice to Ty2], only if SliceTy <> Ty2
10185*9880d681SAndroid Build Coastguard Worker ///
10186*9880d681SAndroid Build Coastguard Worker /// SliceTy is deduced from the number of bits that are actually used to
10187*9880d681SAndroid Build Coastguard Worker /// build Inst.
10188*9880d681SAndroid Build Coastguard Worker struct LoadedSlice {
10189*9880d681SAndroid Build Coastguard Worker /// \brief Helper structure used to compute the cost of a slice.
10190*9880d681SAndroid Build Coastguard Worker struct Cost {
10191*9880d681SAndroid Build Coastguard Worker /// Are we optimizing for code size.
10192*9880d681SAndroid Build Coastguard Worker bool ForCodeSize;
10193*9880d681SAndroid Build Coastguard Worker /// Various cost.
10194*9880d681SAndroid Build Coastguard Worker unsigned Loads;
10195*9880d681SAndroid Build Coastguard Worker unsigned Truncates;
10196*9880d681SAndroid Build Coastguard Worker unsigned CrossRegisterBanksCopies;
10197*9880d681SAndroid Build Coastguard Worker unsigned ZExts;
10198*9880d681SAndroid Build Coastguard Worker unsigned Shift;
10199*9880d681SAndroid Build Coastguard Worker
Cost__anon093339a80811::LoadedSlice::Cost10200*9880d681SAndroid Build Coastguard Worker Cost(bool ForCodeSize = false)
10201*9880d681SAndroid Build Coastguard Worker : ForCodeSize(ForCodeSize), Loads(0), Truncates(0),
10202*9880d681SAndroid Build Coastguard Worker CrossRegisterBanksCopies(0), ZExts(0), Shift(0) {}
10203*9880d681SAndroid Build Coastguard Worker
10204*9880d681SAndroid Build Coastguard Worker /// \brief Get the cost of one isolated slice.
Cost__anon093339a80811::LoadedSlice::Cost10205*9880d681SAndroid Build Coastguard Worker Cost(const LoadedSlice &LS, bool ForCodeSize = false)
10206*9880d681SAndroid Build Coastguard Worker : ForCodeSize(ForCodeSize), Loads(1), Truncates(0),
10207*9880d681SAndroid Build Coastguard Worker CrossRegisterBanksCopies(0), ZExts(0), Shift(0) {
10208*9880d681SAndroid Build Coastguard Worker EVT TruncType = LS.Inst->getValueType(0);
10209*9880d681SAndroid Build Coastguard Worker EVT LoadedType = LS.getLoadedType();
10210*9880d681SAndroid Build Coastguard Worker if (TruncType != LoadedType &&
10211*9880d681SAndroid Build Coastguard Worker !LS.DAG->getTargetLoweringInfo().isZExtFree(LoadedType, TruncType))
10212*9880d681SAndroid Build Coastguard Worker ZExts = 1;
10213*9880d681SAndroid Build Coastguard Worker }
10214*9880d681SAndroid Build Coastguard Worker
10215*9880d681SAndroid Build Coastguard Worker /// \brief Account for slicing gain in the current cost.
10216*9880d681SAndroid Build Coastguard Worker /// Slicing provide a few gains like removing a shift or a
10217*9880d681SAndroid Build Coastguard Worker /// truncate. This method allows to grow the cost of the original
10218*9880d681SAndroid Build Coastguard Worker /// load with the gain from this slice.
addSliceGain__anon093339a80811::LoadedSlice::Cost10219*9880d681SAndroid Build Coastguard Worker void addSliceGain(const LoadedSlice &LS) {
10220*9880d681SAndroid Build Coastguard Worker // Each slice saves a truncate.
10221*9880d681SAndroid Build Coastguard Worker const TargetLowering &TLI = LS.DAG->getTargetLoweringInfo();
10222*9880d681SAndroid Build Coastguard Worker if (!TLI.isTruncateFree(LS.Inst->getOperand(0).getValueType(),
10223*9880d681SAndroid Build Coastguard Worker LS.Inst->getValueType(0)))
10224*9880d681SAndroid Build Coastguard Worker ++Truncates;
10225*9880d681SAndroid Build Coastguard Worker // If there is a shift amount, this slice gets rid of it.
10226*9880d681SAndroid Build Coastguard Worker if (LS.Shift)
10227*9880d681SAndroid Build Coastguard Worker ++Shift;
10228*9880d681SAndroid Build Coastguard Worker // If this slice can merge a cross register bank copy, account for it.
10229*9880d681SAndroid Build Coastguard Worker if (LS.canMergeExpensiveCrossRegisterBankCopy())
10230*9880d681SAndroid Build Coastguard Worker ++CrossRegisterBanksCopies;
10231*9880d681SAndroid Build Coastguard Worker }
10232*9880d681SAndroid Build Coastguard Worker
operator +=__anon093339a80811::LoadedSlice::Cost10233*9880d681SAndroid Build Coastguard Worker Cost &operator+=(const Cost &RHS) {
10234*9880d681SAndroid Build Coastguard Worker Loads += RHS.Loads;
10235*9880d681SAndroid Build Coastguard Worker Truncates += RHS.Truncates;
10236*9880d681SAndroid Build Coastguard Worker CrossRegisterBanksCopies += RHS.CrossRegisterBanksCopies;
10237*9880d681SAndroid Build Coastguard Worker ZExts += RHS.ZExts;
10238*9880d681SAndroid Build Coastguard Worker Shift += RHS.Shift;
10239*9880d681SAndroid Build Coastguard Worker return *this;
10240*9880d681SAndroid Build Coastguard Worker }
10241*9880d681SAndroid Build Coastguard Worker
operator ==__anon093339a80811::LoadedSlice::Cost10242*9880d681SAndroid Build Coastguard Worker bool operator==(const Cost &RHS) const {
10243*9880d681SAndroid Build Coastguard Worker return Loads == RHS.Loads && Truncates == RHS.Truncates &&
10244*9880d681SAndroid Build Coastguard Worker CrossRegisterBanksCopies == RHS.CrossRegisterBanksCopies &&
10245*9880d681SAndroid Build Coastguard Worker ZExts == RHS.ZExts && Shift == RHS.Shift;
10246*9880d681SAndroid Build Coastguard Worker }
10247*9880d681SAndroid Build Coastguard Worker
operator !=__anon093339a80811::LoadedSlice::Cost10248*9880d681SAndroid Build Coastguard Worker bool operator!=(const Cost &RHS) const { return !(*this == RHS); }
10249*9880d681SAndroid Build Coastguard Worker
operator <__anon093339a80811::LoadedSlice::Cost10250*9880d681SAndroid Build Coastguard Worker bool operator<(const Cost &RHS) const {
10251*9880d681SAndroid Build Coastguard Worker // Assume cross register banks copies are as expensive as loads.
10252*9880d681SAndroid Build Coastguard Worker // FIXME: Do we want some more target hooks?
10253*9880d681SAndroid Build Coastguard Worker unsigned ExpensiveOpsLHS = Loads + CrossRegisterBanksCopies;
10254*9880d681SAndroid Build Coastguard Worker unsigned ExpensiveOpsRHS = RHS.Loads + RHS.CrossRegisterBanksCopies;
10255*9880d681SAndroid Build Coastguard Worker // Unless we are optimizing for code size, consider the
10256*9880d681SAndroid Build Coastguard Worker // expensive operation first.
10257*9880d681SAndroid Build Coastguard Worker if (!ForCodeSize && ExpensiveOpsLHS != ExpensiveOpsRHS)
10258*9880d681SAndroid Build Coastguard Worker return ExpensiveOpsLHS < ExpensiveOpsRHS;
10259*9880d681SAndroid Build Coastguard Worker return (Truncates + ZExts + Shift + ExpensiveOpsLHS) <
10260*9880d681SAndroid Build Coastguard Worker (RHS.Truncates + RHS.ZExts + RHS.Shift + ExpensiveOpsRHS);
10261*9880d681SAndroid Build Coastguard Worker }
10262*9880d681SAndroid Build Coastguard Worker
operator >__anon093339a80811::LoadedSlice::Cost10263*9880d681SAndroid Build Coastguard Worker bool operator>(const Cost &RHS) const { return RHS < *this; }
10264*9880d681SAndroid Build Coastguard Worker
operator <=__anon093339a80811::LoadedSlice::Cost10265*9880d681SAndroid Build Coastguard Worker bool operator<=(const Cost &RHS) const { return !(RHS < *this); }
10266*9880d681SAndroid Build Coastguard Worker
operator >=__anon093339a80811::LoadedSlice::Cost10267*9880d681SAndroid Build Coastguard Worker bool operator>=(const Cost &RHS) const { return !(*this < RHS); }
10268*9880d681SAndroid Build Coastguard Worker };
10269*9880d681SAndroid Build Coastguard Worker // The last instruction that represent the slice. This should be a
10270*9880d681SAndroid Build Coastguard Worker // truncate instruction.
10271*9880d681SAndroid Build Coastguard Worker SDNode *Inst;
10272*9880d681SAndroid Build Coastguard Worker // The original load instruction.
10273*9880d681SAndroid Build Coastguard Worker LoadSDNode *Origin;
10274*9880d681SAndroid Build Coastguard Worker // The right shift amount in bits from the original load.
10275*9880d681SAndroid Build Coastguard Worker unsigned Shift;
10276*9880d681SAndroid Build Coastguard Worker // The DAG from which Origin came from.
10277*9880d681SAndroid Build Coastguard Worker // This is used to get some contextual information about legal types, etc.
10278*9880d681SAndroid Build Coastguard Worker SelectionDAG *DAG;
10279*9880d681SAndroid Build Coastguard Worker
LoadedSlice__anon093339a80811::LoadedSlice10280*9880d681SAndroid Build Coastguard Worker LoadedSlice(SDNode *Inst = nullptr, LoadSDNode *Origin = nullptr,
10281*9880d681SAndroid Build Coastguard Worker unsigned Shift = 0, SelectionDAG *DAG = nullptr)
10282*9880d681SAndroid Build Coastguard Worker : Inst(Inst), Origin(Origin), Shift(Shift), DAG(DAG) {}
10283*9880d681SAndroid Build Coastguard Worker
10284*9880d681SAndroid Build Coastguard Worker /// \brief Get the bits used in a chunk of bits \p BitWidth large.
10285*9880d681SAndroid Build Coastguard Worker /// \return Result is \p BitWidth and has used bits set to 1 and
10286*9880d681SAndroid Build Coastguard Worker /// not used bits set to 0.
getUsedBits__anon093339a80811::LoadedSlice10287*9880d681SAndroid Build Coastguard Worker APInt getUsedBits() const {
10288*9880d681SAndroid Build Coastguard Worker // Reproduce the trunc(lshr) sequence:
10289*9880d681SAndroid Build Coastguard Worker // - Start from the truncated value.
10290*9880d681SAndroid Build Coastguard Worker // - Zero extend to the desired bit width.
10291*9880d681SAndroid Build Coastguard Worker // - Shift left.
10292*9880d681SAndroid Build Coastguard Worker assert(Origin && "No original load to compare against.");
10293*9880d681SAndroid Build Coastguard Worker unsigned BitWidth = Origin->getValueSizeInBits(0);
10294*9880d681SAndroid Build Coastguard Worker assert(Inst && "This slice is not bound to an instruction");
10295*9880d681SAndroid Build Coastguard Worker assert(Inst->getValueSizeInBits(0) <= BitWidth &&
10296*9880d681SAndroid Build Coastguard Worker "Extracted slice is bigger than the whole type!");
10297*9880d681SAndroid Build Coastguard Worker APInt UsedBits(Inst->getValueSizeInBits(0), 0);
10298*9880d681SAndroid Build Coastguard Worker UsedBits.setAllBits();
10299*9880d681SAndroid Build Coastguard Worker UsedBits = UsedBits.zext(BitWidth);
10300*9880d681SAndroid Build Coastguard Worker UsedBits <<= Shift;
10301*9880d681SAndroid Build Coastguard Worker return UsedBits;
10302*9880d681SAndroid Build Coastguard Worker }
10303*9880d681SAndroid Build Coastguard Worker
10304*9880d681SAndroid Build Coastguard Worker /// \brief Get the size of the slice to be loaded in bytes.
getLoadedSize__anon093339a80811::LoadedSlice10305*9880d681SAndroid Build Coastguard Worker unsigned getLoadedSize() const {
10306*9880d681SAndroid Build Coastguard Worker unsigned SliceSize = getUsedBits().countPopulation();
10307*9880d681SAndroid Build Coastguard Worker assert(!(SliceSize & 0x7) && "Size is not a multiple of a byte.");
10308*9880d681SAndroid Build Coastguard Worker return SliceSize / 8;
10309*9880d681SAndroid Build Coastguard Worker }
10310*9880d681SAndroid Build Coastguard Worker
10311*9880d681SAndroid Build Coastguard Worker /// \brief Get the type that will be loaded for this slice.
10312*9880d681SAndroid Build Coastguard Worker /// Note: This may not be the final type for the slice.
getLoadedType__anon093339a80811::LoadedSlice10313*9880d681SAndroid Build Coastguard Worker EVT getLoadedType() const {
10314*9880d681SAndroid Build Coastguard Worker assert(DAG && "Missing context");
10315*9880d681SAndroid Build Coastguard Worker LLVMContext &Ctxt = *DAG->getContext();
10316*9880d681SAndroid Build Coastguard Worker return EVT::getIntegerVT(Ctxt, getLoadedSize() * 8);
10317*9880d681SAndroid Build Coastguard Worker }
10318*9880d681SAndroid Build Coastguard Worker
10319*9880d681SAndroid Build Coastguard Worker /// \brief Get the alignment of the load used for this slice.
getAlignment__anon093339a80811::LoadedSlice10320*9880d681SAndroid Build Coastguard Worker unsigned getAlignment() const {
10321*9880d681SAndroid Build Coastguard Worker unsigned Alignment = Origin->getAlignment();
10322*9880d681SAndroid Build Coastguard Worker unsigned Offset = getOffsetFromBase();
10323*9880d681SAndroid Build Coastguard Worker if (Offset != 0)
10324*9880d681SAndroid Build Coastguard Worker Alignment = MinAlign(Alignment, Alignment + Offset);
10325*9880d681SAndroid Build Coastguard Worker return Alignment;
10326*9880d681SAndroid Build Coastguard Worker }
10327*9880d681SAndroid Build Coastguard Worker
10328*9880d681SAndroid Build Coastguard Worker /// \brief Check if this slice can be rewritten with legal operations.
isLegal__anon093339a80811::LoadedSlice10329*9880d681SAndroid Build Coastguard Worker bool isLegal() const {
10330*9880d681SAndroid Build Coastguard Worker // An invalid slice is not legal.
10331*9880d681SAndroid Build Coastguard Worker if (!Origin || !Inst || !DAG)
10332*9880d681SAndroid Build Coastguard Worker return false;
10333*9880d681SAndroid Build Coastguard Worker
10334*9880d681SAndroid Build Coastguard Worker // Offsets are for indexed load only, we do not handle that.
10335*9880d681SAndroid Build Coastguard Worker if (!Origin->getOffset().isUndef())
10336*9880d681SAndroid Build Coastguard Worker return false;
10337*9880d681SAndroid Build Coastguard Worker
10338*9880d681SAndroid Build Coastguard Worker const TargetLowering &TLI = DAG->getTargetLoweringInfo();
10339*9880d681SAndroid Build Coastguard Worker
10340*9880d681SAndroid Build Coastguard Worker // Check that the type is legal.
10341*9880d681SAndroid Build Coastguard Worker EVT SliceType = getLoadedType();
10342*9880d681SAndroid Build Coastguard Worker if (!TLI.isTypeLegal(SliceType))
10343*9880d681SAndroid Build Coastguard Worker return false;
10344*9880d681SAndroid Build Coastguard Worker
10345*9880d681SAndroid Build Coastguard Worker // Check that the load is legal for this type.
10346*9880d681SAndroid Build Coastguard Worker if (!TLI.isOperationLegal(ISD::LOAD, SliceType))
10347*9880d681SAndroid Build Coastguard Worker return false;
10348*9880d681SAndroid Build Coastguard Worker
10349*9880d681SAndroid Build Coastguard Worker // Check that the offset can be computed.
10350*9880d681SAndroid Build Coastguard Worker // 1. Check its type.
10351*9880d681SAndroid Build Coastguard Worker EVT PtrType = Origin->getBasePtr().getValueType();
10352*9880d681SAndroid Build Coastguard Worker if (PtrType == MVT::Untyped || PtrType.isExtended())
10353*9880d681SAndroid Build Coastguard Worker return false;
10354*9880d681SAndroid Build Coastguard Worker
10355*9880d681SAndroid Build Coastguard Worker // 2. Check that it fits in the immediate.
10356*9880d681SAndroid Build Coastguard Worker if (!TLI.isLegalAddImmediate(getOffsetFromBase()))
10357*9880d681SAndroid Build Coastguard Worker return false;
10358*9880d681SAndroid Build Coastguard Worker
10359*9880d681SAndroid Build Coastguard Worker // 3. Check that the computation is legal.
10360*9880d681SAndroid Build Coastguard Worker if (!TLI.isOperationLegal(ISD::ADD, PtrType))
10361*9880d681SAndroid Build Coastguard Worker return false;
10362*9880d681SAndroid Build Coastguard Worker
10363*9880d681SAndroid Build Coastguard Worker // Check that the zext is legal if it needs one.
10364*9880d681SAndroid Build Coastguard Worker EVT TruncateType = Inst->getValueType(0);
10365*9880d681SAndroid Build Coastguard Worker if (TruncateType != SliceType &&
10366*9880d681SAndroid Build Coastguard Worker !TLI.isOperationLegal(ISD::ZERO_EXTEND, TruncateType))
10367*9880d681SAndroid Build Coastguard Worker return false;
10368*9880d681SAndroid Build Coastguard Worker
10369*9880d681SAndroid Build Coastguard Worker return true;
10370*9880d681SAndroid Build Coastguard Worker }
10371*9880d681SAndroid Build Coastguard Worker
10372*9880d681SAndroid Build Coastguard Worker /// \brief Get the offset in bytes of this slice in the original chunk of
10373*9880d681SAndroid Build Coastguard Worker /// bits.
10374*9880d681SAndroid Build Coastguard Worker /// \pre DAG != nullptr.
getOffsetFromBase__anon093339a80811::LoadedSlice10375*9880d681SAndroid Build Coastguard Worker uint64_t getOffsetFromBase() const {
10376*9880d681SAndroid Build Coastguard Worker assert(DAG && "Missing context.");
10377*9880d681SAndroid Build Coastguard Worker bool IsBigEndian = DAG->getDataLayout().isBigEndian();
10378*9880d681SAndroid Build Coastguard Worker assert(!(Shift & 0x7) && "Shifts not aligned on Bytes are not supported.");
10379*9880d681SAndroid Build Coastguard Worker uint64_t Offset = Shift / 8;
10380*9880d681SAndroid Build Coastguard Worker unsigned TySizeInBytes = Origin->getValueSizeInBits(0) / 8;
10381*9880d681SAndroid Build Coastguard Worker assert(!(Origin->getValueSizeInBits(0) & 0x7) &&
10382*9880d681SAndroid Build Coastguard Worker "The size of the original loaded type is not a multiple of a"
10383*9880d681SAndroid Build Coastguard Worker " byte.");
10384*9880d681SAndroid Build Coastguard Worker // If Offset is bigger than TySizeInBytes, it means we are loading all
10385*9880d681SAndroid Build Coastguard Worker // zeros. This should have been optimized before in the process.
10386*9880d681SAndroid Build Coastguard Worker assert(TySizeInBytes > Offset &&
10387*9880d681SAndroid Build Coastguard Worker "Invalid shift amount for given loaded size");
10388*9880d681SAndroid Build Coastguard Worker if (IsBigEndian)
10389*9880d681SAndroid Build Coastguard Worker Offset = TySizeInBytes - Offset - getLoadedSize();
10390*9880d681SAndroid Build Coastguard Worker return Offset;
10391*9880d681SAndroid Build Coastguard Worker }
10392*9880d681SAndroid Build Coastguard Worker
10393*9880d681SAndroid Build Coastguard Worker /// \brief Generate the sequence of instructions to load the slice
10394*9880d681SAndroid Build Coastguard Worker /// represented by this object and redirect the uses of this slice to
10395*9880d681SAndroid Build Coastguard Worker /// this new sequence of instructions.
10396*9880d681SAndroid Build Coastguard Worker /// \pre this->Inst && this->Origin are valid Instructions and this
10397*9880d681SAndroid Build Coastguard Worker /// object passed the legal check: LoadedSlice::isLegal returned true.
10398*9880d681SAndroid Build Coastguard Worker /// \return The last instruction of the sequence used to load the slice.
loadSlice__anon093339a80811::LoadedSlice10399*9880d681SAndroid Build Coastguard Worker SDValue loadSlice() const {
10400*9880d681SAndroid Build Coastguard Worker assert(Inst && Origin && "Unable to replace a non-existing slice.");
10401*9880d681SAndroid Build Coastguard Worker const SDValue &OldBaseAddr = Origin->getBasePtr();
10402*9880d681SAndroid Build Coastguard Worker SDValue BaseAddr = OldBaseAddr;
10403*9880d681SAndroid Build Coastguard Worker // Get the offset in that chunk of bytes w.r.t. the endianess.
10404*9880d681SAndroid Build Coastguard Worker int64_t Offset = static_cast<int64_t>(getOffsetFromBase());
10405*9880d681SAndroid Build Coastguard Worker assert(Offset >= 0 && "Offset too big to fit in int64_t!");
10406*9880d681SAndroid Build Coastguard Worker if (Offset) {
10407*9880d681SAndroid Build Coastguard Worker // BaseAddr = BaseAddr + Offset.
10408*9880d681SAndroid Build Coastguard Worker EVT ArithType = BaseAddr.getValueType();
10409*9880d681SAndroid Build Coastguard Worker SDLoc DL(Origin);
10410*9880d681SAndroid Build Coastguard Worker BaseAddr = DAG->getNode(ISD::ADD, DL, ArithType, BaseAddr,
10411*9880d681SAndroid Build Coastguard Worker DAG->getConstant(Offset, DL, ArithType));
10412*9880d681SAndroid Build Coastguard Worker }
10413*9880d681SAndroid Build Coastguard Worker
10414*9880d681SAndroid Build Coastguard Worker // Create the type of the loaded slice according to its size.
10415*9880d681SAndroid Build Coastguard Worker EVT SliceType = getLoadedType();
10416*9880d681SAndroid Build Coastguard Worker
10417*9880d681SAndroid Build Coastguard Worker // Create the load for the slice.
10418*9880d681SAndroid Build Coastguard Worker SDValue LastInst = DAG->getLoad(
10419*9880d681SAndroid Build Coastguard Worker SliceType, SDLoc(Origin), Origin->getChain(), BaseAddr,
10420*9880d681SAndroid Build Coastguard Worker Origin->getPointerInfo().getWithOffset(Offset), Origin->isVolatile(),
10421*9880d681SAndroid Build Coastguard Worker Origin->isNonTemporal(), Origin->isInvariant(), getAlignment());
10422*9880d681SAndroid Build Coastguard Worker // If the final type is not the same as the loaded type, this means that
10423*9880d681SAndroid Build Coastguard Worker // we have to pad with zero. Create a zero extend for that.
10424*9880d681SAndroid Build Coastguard Worker EVT FinalType = Inst->getValueType(0);
10425*9880d681SAndroid Build Coastguard Worker if (SliceType != FinalType)
10426*9880d681SAndroid Build Coastguard Worker LastInst =
10427*9880d681SAndroid Build Coastguard Worker DAG->getNode(ISD::ZERO_EXTEND, SDLoc(LastInst), FinalType, LastInst);
10428*9880d681SAndroid Build Coastguard Worker return LastInst;
10429*9880d681SAndroid Build Coastguard Worker }
10430*9880d681SAndroid Build Coastguard Worker
10431*9880d681SAndroid Build Coastguard Worker /// \brief Check if this slice can be merged with an expensive cross register
10432*9880d681SAndroid Build Coastguard Worker /// bank copy. E.g.,
10433*9880d681SAndroid Build Coastguard Worker /// i = load i32
10434*9880d681SAndroid Build Coastguard Worker /// f = bitcast i32 i to float
canMergeExpensiveCrossRegisterBankCopy__anon093339a80811::LoadedSlice10435*9880d681SAndroid Build Coastguard Worker bool canMergeExpensiveCrossRegisterBankCopy() const {
10436*9880d681SAndroid Build Coastguard Worker if (!Inst || !Inst->hasOneUse())
10437*9880d681SAndroid Build Coastguard Worker return false;
10438*9880d681SAndroid Build Coastguard Worker SDNode *Use = *Inst->use_begin();
10439*9880d681SAndroid Build Coastguard Worker if (Use->getOpcode() != ISD::BITCAST)
10440*9880d681SAndroid Build Coastguard Worker return false;
10441*9880d681SAndroid Build Coastguard Worker assert(DAG && "Missing context");
10442*9880d681SAndroid Build Coastguard Worker const TargetLowering &TLI = DAG->getTargetLoweringInfo();
10443*9880d681SAndroid Build Coastguard Worker EVT ResVT = Use->getValueType(0);
10444*9880d681SAndroid Build Coastguard Worker const TargetRegisterClass *ResRC = TLI.getRegClassFor(ResVT.getSimpleVT());
10445*9880d681SAndroid Build Coastguard Worker const TargetRegisterClass *ArgRC =
10446*9880d681SAndroid Build Coastguard Worker TLI.getRegClassFor(Use->getOperand(0).getValueType().getSimpleVT());
10447*9880d681SAndroid Build Coastguard Worker if (ArgRC == ResRC || !TLI.isOperationLegal(ISD::LOAD, ResVT))
10448*9880d681SAndroid Build Coastguard Worker return false;
10449*9880d681SAndroid Build Coastguard Worker
10450*9880d681SAndroid Build Coastguard Worker // At this point, we know that we perform a cross-register-bank copy.
10451*9880d681SAndroid Build Coastguard Worker // Check if it is expensive.
10452*9880d681SAndroid Build Coastguard Worker const TargetRegisterInfo *TRI = DAG->getSubtarget().getRegisterInfo();
10453*9880d681SAndroid Build Coastguard Worker // Assume bitcasts are cheap, unless both register classes do not
10454*9880d681SAndroid Build Coastguard Worker // explicitly share a common sub class.
10455*9880d681SAndroid Build Coastguard Worker if (!TRI || TRI->getCommonSubClass(ArgRC, ResRC))
10456*9880d681SAndroid Build Coastguard Worker return false;
10457*9880d681SAndroid Build Coastguard Worker
10458*9880d681SAndroid Build Coastguard Worker // Check if it will be merged with the load.
10459*9880d681SAndroid Build Coastguard Worker // 1. Check the alignment constraint.
10460*9880d681SAndroid Build Coastguard Worker unsigned RequiredAlignment = DAG->getDataLayout().getABITypeAlignment(
10461*9880d681SAndroid Build Coastguard Worker ResVT.getTypeForEVT(*DAG->getContext()));
10462*9880d681SAndroid Build Coastguard Worker
10463*9880d681SAndroid Build Coastguard Worker if (RequiredAlignment > getAlignment())
10464*9880d681SAndroid Build Coastguard Worker return false;
10465*9880d681SAndroid Build Coastguard Worker
10466*9880d681SAndroid Build Coastguard Worker // 2. Check that the load is a legal operation for that type.
10467*9880d681SAndroid Build Coastguard Worker if (!TLI.isOperationLegal(ISD::LOAD, ResVT))
10468*9880d681SAndroid Build Coastguard Worker return false;
10469*9880d681SAndroid Build Coastguard Worker
10470*9880d681SAndroid Build Coastguard Worker // 3. Check that we do not have a zext in the way.
10471*9880d681SAndroid Build Coastguard Worker if (Inst->getValueType(0) != getLoadedType())
10472*9880d681SAndroid Build Coastguard Worker return false;
10473*9880d681SAndroid Build Coastguard Worker
10474*9880d681SAndroid Build Coastguard Worker return true;
10475*9880d681SAndroid Build Coastguard Worker }
10476*9880d681SAndroid Build Coastguard Worker };
10477*9880d681SAndroid Build Coastguard Worker }
10478*9880d681SAndroid Build Coastguard Worker
10479*9880d681SAndroid Build Coastguard Worker /// \brief Check that all bits set in \p UsedBits form a dense region, i.e.,
10480*9880d681SAndroid Build Coastguard Worker /// \p UsedBits looks like 0..0 1..1 0..0.
areUsedBitsDense(const APInt & UsedBits)10481*9880d681SAndroid Build Coastguard Worker static bool areUsedBitsDense(const APInt &UsedBits) {
10482*9880d681SAndroid Build Coastguard Worker // If all the bits are one, this is dense!
10483*9880d681SAndroid Build Coastguard Worker if (UsedBits.isAllOnesValue())
10484*9880d681SAndroid Build Coastguard Worker return true;
10485*9880d681SAndroid Build Coastguard Worker
10486*9880d681SAndroid Build Coastguard Worker // Get rid of the unused bits on the right.
10487*9880d681SAndroid Build Coastguard Worker APInt NarrowedUsedBits = UsedBits.lshr(UsedBits.countTrailingZeros());
10488*9880d681SAndroid Build Coastguard Worker // Get rid of the unused bits on the left.
10489*9880d681SAndroid Build Coastguard Worker if (NarrowedUsedBits.countLeadingZeros())
10490*9880d681SAndroid Build Coastguard Worker NarrowedUsedBits = NarrowedUsedBits.trunc(NarrowedUsedBits.getActiveBits());
10491*9880d681SAndroid Build Coastguard Worker // Check that the chunk of bits is completely used.
10492*9880d681SAndroid Build Coastguard Worker return NarrowedUsedBits.isAllOnesValue();
10493*9880d681SAndroid Build Coastguard Worker }
10494*9880d681SAndroid Build Coastguard Worker
10495*9880d681SAndroid Build Coastguard Worker /// \brief Check whether or not \p First and \p Second are next to each other
10496*9880d681SAndroid Build Coastguard Worker /// in memory. This means that there is no hole between the bits loaded
10497*9880d681SAndroid Build Coastguard Worker /// by \p First and the bits loaded by \p Second.
areSlicesNextToEachOther(const LoadedSlice & First,const LoadedSlice & Second)10498*9880d681SAndroid Build Coastguard Worker static bool areSlicesNextToEachOther(const LoadedSlice &First,
10499*9880d681SAndroid Build Coastguard Worker const LoadedSlice &Second) {
10500*9880d681SAndroid Build Coastguard Worker assert(First.Origin == Second.Origin && First.Origin &&
10501*9880d681SAndroid Build Coastguard Worker "Unable to match different memory origins.");
10502*9880d681SAndroid Build Coastguard Worker APInt UsedBits = First.getUsedBits();
10503*9880d681SAndroid Build Coastguard Worker assert((UsedBits & Second.getUsedBits()) == 0 &&
10504*9880d681SAndroid Build Coastguard Worker "Slices are not supposed to overlap.");
10505*9880d681SAndroid Build Coastguard Worker UsedBits |= Second.getUsedBits();
10506*9880d681SAndroid Build Coastguard Worker return areUsedBitsDense(UsedBits);
10507*9880d681SAndroid Build Coastguard Worker }
10508*9880d681SAndroid Build Coastguard Worker
10509*9880d681SAndroid Build Coastguard Worker /// \brief Adjust the \p GlobalLSCost according to the target
10510*9880d681SAndroid Build Coastguard Worker /// paring capabilities and the layout of the slices.
10511*9880d681SAndroid Build Coastguard Worker /// \pre \p GlobalLSCost should account for at least as many loads as
10512*9880d681SAndroid Build Coastguard Worker /// there is in the slices in \p LoadedSlices.
adjustCostForPairing(SmallVectorImpl<LoadedSlice> & LoadedSlices,LoadedSlice::Cost & GlobalLSCost)10513*9880d681SAndroid Build Coastguard Worker static void adjustCostForPairing(SmallVectorImpl<LoadedSlice> &LoadedSlices,
10514*9880d681SAndroid Build Coastguard Worker LoadedSlice::Cost &GlobalLSCost) {
10515*9880d681SAndroid Build Coastguard Worker unsigned NumberOfSlices = LoadedSlices.size();
10516*9880d681SAndroid Build Coastguard Worker // If there is less than 2 elements, no pairing is possible.
10517*9880d681SAndroid Build Coastguard Worker if (NumberOfSlices < 2)
10518*9880d681SAndroid Build Coastguard Worker return;
10519*9880d681SAndroid Build Coastguard Worker
10520*9880d681SAndroid Build Coastguard Worker // Sort the slices so that elements that are likely to be next to each
10521*9880d681SAndroid Build Coastguard Worker // other in memory are next to each other in the list.
10522*9880d681SAndroid Build Coastguard Worker std::sort(LoadedSlices.begin(), LoadedSlices.end(),
10523*9880d681SAndroid Build Coastguard Worker [](const LoadedSlice &LHS, const LoadedSlice &RHS) {
10524*9880d681SAndroid Build Coastguard Worker assert(LHS.Origin == RHS.Origin && "Different bases not implemented.");
10525*9880d681SAndroid Build Coastguard Worker return LHS.getOffsetFromBase() < RHS.getOffsetFromBase();
10526*9880d681SAndroid Build Coastguard Worker });
10527*9880d681SAndroid Build Coastguard Worker const TargetLowering &TLI = LoadedSlices[0].DAG->getTargetLoweringInfo();
10528*9880d681SAndroid Build Coastguard Worker // First (resp. Second) is the first (resp. Second) potentially candidate
10529*9880d681SAndroid Build Coastguard Worker // to be placed in a paired load.
10530*9880d681SAndroid Build Coastguard Worker const LoadedSlice *First = nullptr;
10531*9880d681SAndroid Build Coastguard Worker const LoadedSlice *Second = nullptr;
10532*9880d681SAndroid Build Coastguard Worker for (unsigned CurrSlice = 0; CurrSlice < NumberOfSlices; ++CurrSlice,
10533*9880d681SAndroid Build Coastguard Worker // Set the beginning of the pair.
10534*9880d681SAndroid Build Coastguard Worker First = Second) {
10535*9880d681SAndroid Build Coastguard Worker
10536*9880d681SAndroid Build Coastguard Worker Second = &LoadedSlices[CurrSlice];
10537*9880d681SAndroid Build Coastguard Worker
10538*9880d681SAndroid Build Coastguard Worker // If First is NULL, it means we start a new pair.
10539*9880d681SAndroid Build Coastguard Worker // Get to the next slice.
10540*9880d681SAndroid Build Coastguard Worker if (!First)
10541*9880d681SAndroid Build Coastguard Worker continue;
10542*9880d681SAndroid Build Coastguard Worker
10543*9880d681SAndroid Build Coastguard Worker EVT LoadedType = First->getLoadedType();
10544*9880d681SAndroid Build Coastguard Worker
10545*9880d681SAndroid Build Coastguard Worker // If the types of the slices are different, we cannot pair them.
10546*9880d681SAndroid Build Coastguard Worker if (LoadedType != Second->getLoadedType())
10547*9880d681SAndroid Build Coastguard Worker continue;
10548*9880d681SAndroid Build Coastguard Worker
10549*9880d681SAndroid Build Coastguard Worker // Check if the target supplies paired loads for this type.
10550*9880d681SAndroid Build Coastguard Worker unsigned RequiredAlignment = 0;
10551*9880d681SAndroid Build Coastguard Worker if (!TLI.hasPairedLoad(LoadedType, RequiredAlignment)) {
10552*9880d681SAndroid Build Coastguard Worker // move to the next pair, this type is hopeless.
10553*9880d681SAndroid Build Coastguard Worker Second = nullptr;
10554*9880d681SAndroid Build Coastguard Worker continue;
10555*9880d681SAndroid Build Coastguard Worker }
10556*9880d681SAndroid Build Coastguard Worker // Check if we meet the alignment requirement.
10557*9880d681SAndroid Build Coastguard Worker if (RequiredAlignment > First->getAlignment())
10558*9880d681SAndroid Build Coastguard Worker continue;
10559*9880d681SAndroid Build Coastguard Worker
10560*9880d681SAndroid Build Coastguard Worker // Check that both loads are next to each other in memory.
10561*9880d681SAndroid Build Coastguard Worker if (!areSlicesNextToEachOther(*First, *Second))
10562*9880d681SAndroid Build Coastguard Worker continue;
10563*9880d681SAndroid Build Coastguard Worker
10564*9880d681SAndroid Build Coastguard Worker assert(GlobalLSCost.Loads > 0 && "We save more loads than we created!");
10565*9880d681SAndroid Build Coastguard Worker --GlobalLSCost.Loads;
10566*9880d681SAndroid Build Coastguard Worker // Move to the next pair.
10567*9880d681SAndroid Build Coastguard Worker Second = nullptr;
10568*9880d681SAndroid Build Coastguard Worker }
10569*9880d681SAndroid Build Coastguard Worker }
10570*9880d681SAndroid Build Coastguard Worker
10571*9880d681SAndroid Build Coastguard Worker /// \brief Check the profitability of all involved LoadedSlice.
10572*9880d681SAndroid Build Coastguard Worker /// Currently, it is considered profitable if there is exactly two
10573*9880d681SAndroid Build Coastguard Worker /// involved slices (1) which are (2) next to each other in memory, and
10574*9880d681SAndroid Build Coastguard Worker /// whose cost (\see LoadedSlice::Cost) is smaller than the original load (3).
10575*9880d681SAndroid Build Coastguard Worker ///
10576*9880d681SAndroid Build Coastguard Worker /// Note: The order of the elements in \p LoadedSlices may be modified, but not
10577*9880d681SAndroid Build Coastguard Worker /// the elements themselves.
10578*9880d681SAndroid Build Coastguard Worker ///
10579*9880d681SAndroid Build Coastguard Worker /// FIXME: When the cost model will be mature enough, we can relax
10580*9880d681SAndroid Build Coastguard Worker /// constraints (1) and (2).
isSlicingProfitable(SmallVectorImpl<LoadedSlice> & LoadedSlices,const APInt & UsedBits,bool ForCodeSize)10581*9880d681SAndroid Build Coastguard Worker static bool isSlicingProfitable(SmallVectorImpl<LoadedSlice> &LoadedSlices,
10582*9880d681SAndroid Build Coastguard Worker const APInt &UsedBits, bool ForCodeSize) {
10583*9880d681SAndroid Build Coastguard Worker unsigned NumberOfSlices = LoadedSlices.size();
10584*9880d681SAndroid Build Coastguard Worker if (StressLoadSlicing)
10585*9880d681SAndroid Build Coastguard Worker return NumberOfSlices > 1;
10586*9880d681SAndroid Build Coastguard Worker
10587*9880d681SAndroid Build Coastguard Worker // Check (1).
10588*9880d681SAndroid Build Coastguard Worker if (NumberOfSlices != 2)
10589*9880d681SAndroid Build Coastguard Worker return false;
10590*9880d681SAndroid Build Coastguard Worker
10591*9880d681SAndroid Build Coastguard Worker // Check (2).
10592*9880d681SAndroid Build Coastguard Worker if (!areUsedBitsDense(UsedBits))
10593*9880d681SAndroid Build Coastguard Worker return false;
10594*9880d681SAndroid Build Coastguard Worker
10595*9880d681SAndroid Build Coastguard Worker // Check (3).
10596*9880d681SAndroid Build Coastguard Worker LoadedSlice::Cost OrigCost(ForCodeSize), GlobalSlicingCost(ForCodeSize);
10597*9880d681SAndroid Build Coastguard Worker // The original code has one big load.
10598*9880d681SAndroid Build Coastguard Worker OrigCost.Loads = 1;
10599*9880d681SAndroid Build Coastguard Worker for (unsigned CurrSlice = 0; CurrSlice < NumberOfSlices; ++CurrSlice) {
10600*9880d681SAndroid Build Coastguard Worker const LoadedSlice &LS = LoadedSlices[CurrSlice];
10601*9880d681SAndroid Build Coastguard Worker // Accumulate the cost of all the slices.
10602*9880d681SAndroid Build Coastguard Worker LoadedSlice::Cost SliceCost(LS, ForCodeSize);
10603*9880d681SAndroid Build Coastguard Worker GlobalSlicingCost += SliceCost;
10604*9880d681SAndroid Build Coastguard Worker
10605*9880d681SAndroid Build Coastguard Worker // Account as cost in the original configuration the gain obtained
10606*9880d681SAndroid Build Coastguard Worker // with the current slices.
10607*9880d681SAndroid Build Coastguard Worker OrigCost.addSliceGain(LS);
10608*9880d681SAndroid Build Coastguard Worker }
10609*9880d681SAndroid Build Coastguard Worker
10610*9880d681SAndroid Build Coastguard Worker // If the target supports paired load, adjust the cost accordingly.
10611*9880d681SAndroid Build Coastguard Worker adjustCostForPairing(LoadedSlices, GlobalSlicingCost);
10612*9880d681SAndroid Build Coastguard Worker return OrigCost > GlobalSlicingCost;
10613*9880d681SAndroid Build Coastguard Worker }
10614*9880d681SAndroid Build Coastguard Worker
10615*9880d681SAndroid Build Coastguard Worker /// \brief If the given load, \p LI, is used only by trunc or trunc(lshr)
10616*9880d681SAndroid Build Coastguard Worker /// operations, split it in the various pieces being extracted.
10617*9880d681SAndroid Build Coastguard Worker ///
10618*9880d681SAndroid Build Coastguard Worker /// This sort of thing is introduced by SROA.
10619*9880d681SAndroid Build Coastguard Worker /// This slicing takes care not to insert overlapping loads.
10620*9880d681SAndroid Build Coastguard Worker /// \pre LI is a simple load (i.e., not an atomic or volatile load).
SliceUpLoad(SDNode * N)10621*9880d681SAndroid Build Coastguard Worker bool DAGCombiner::SliceUpLoad(SDNode *N) {
10622*9880d681SAndroid Build Coastguard Worker if (Level < AfterLegalizeDAG)
10623*9880d681SAndroid Build Coastguard Worker return false;
10624*9880d681SAndroid Build Coastguard Worker
10625*9880d681SAndroid Build Coastguard Worker LoadSDNode *LD = cast<LoadSDNode>(N);
10626*9880d681SAndroid Build Coastguard Worker if (LD->isVolatile() || !ISD::isNormalLoad(LD) ||
10627*9880d681SAndroid Build Coastguard Worker !LD->getValueType(0).isInteger())
10628*9880d681SAndroid Build Coastguard Worker return false;
10629*9880d681SAndroid Build Coastguard Worker
10630*9880d681SAndroid Build Coastguard Worker // Keep track of already used bits to detect overlapping values.
10631*9880d681SAndroid Build Coastguard Worker // In that case, we will just abort the transformation.
10632*9880d681SAndroid Build Coastguard Worker APInt UsedBits(LD->getValueSizeInBits(0), 0);
10633*9880d681SAndroid Build Coastguard Worker
10634*9880d681SAndroid Build Coastguard Worker SmallVector<LoadedSlice, 4> LoadedSlices;
10635*9880d681SAndroid Build Coastguard Worker
10636*9880d681SAndroid Build Coastguard Worker // Check if this load is used as several smaller chunks of bits.
10637*9880d681SAndroid Build Coastguard Worker // Basically, look for uses in trunc or trunc(lshr) and record a new chain
10638*9880d681SAndroid Build Coastguard Worker // of computation for each trunc.
10639*9880d681SAndroid Build Coastguard Worker for (SDNode::use_iterator UI = LD->use_begin(), UIEnd = LD->use_end();
10640*9880d681SAndroid Build Coastguard Worker UI != UIEnd; ++UI) {
10641*9880d681SAndroid Build Coastguard Worker // Skip the uses of the chain.
10642*9880d681SAndroid Build Coastguard Worker if (UI.getUse().getResNo() != 0)
10643*9880d681SAndroid Build Coastguard Worker continue;
10644*9880d681SAndroid Build Coastguard Worker
10645*9880d681SAndroid Build Coastguard Worker SDNode *User = *UI;
10646*9880d681SAndroid Build Coastguard Worker unsigned Shift = 0;
10647*9880d681SAndroid Build Coastguard Worker
10648*9880d681SAndroid Build Coastguard Worker // Check if this is a trunc(lshr).
10649*9880d681SAndroid Build Coastguard Worker if (User->getOpcode() == ISD::SRL && User->hasOneUse() &&
10650*9880d681SAndroid Build Coastguard Worker isa<ConstantSDNode>(User->getOperand(1))) {
10651*9880d681SAndroid Build Coastguard Worker Shift = cast<ConstantSDNode>(User->getOperand(1))->getZExtValue();
10652*9880d681SAndroid Build Coastguard Worker User = *User->use_begin();
10653*9880d681SAndroid Build Coastguard Worker }
10654*9880d681SAndroid Build Coastguard Worker
10655*9880d681SAndroid Build Coastguard Worker // At this point, User is a Truncate, iff we encountered, trunc or
10656*9880d681SAndroid Build Coastguard Worker // trunc(lshr).
10657*9880d681SAndroid Build Coastguard Worker if (User->getOpcode() != ISD::TRUNCATE)
10658*9880d681SAndroid Build Coastguard Worker return false;
10659*9880d681SAndroid Build Coastguard Worker
10660*9880d681SAndroid Build Coastguard Worker // The width of the type must be a power of 2 and greater than 8-bits.
10661*9880d681SAndroid Build Coastguard Worker // Otherwise the load cannot be represented in LLVM IR.
10662*9880d681SAndroid Build Coastguard Worker // Moreover, if we shifted with a non-8-bits multiple, the slice
10663*9880d681SAndroid Build Coastguard Worker // will be across several bytes. We do not support that.
10664*9880d681SAndroid Build Coastguard Worker unsigned Width = User->getValueSizeInBits(0);
10665*9880d681SAndroid Build Coastguard Worker if (Width < 8 || !isPowerOf2_32(Width) || (Shift & 0x7))
10666*9880d681SAndroid Build Coastguard Worker return 0;
10667*9880d681SAndroid Build Coastguard Worker
10668*9880d681SAndroid Build Coastguard Worker // Build the slice for this chain of computations.
10669*9880d681SAndroid Build Coastguard Worker LoadedSlice LS(User, LD, Shift, &DAG);
10670*9880d681SAndroid Build Coastguard Worker APInt CurrentUsedBits = LS.getUsedBits();
10671*9880d681SAndroid Build Coastguard Worker
10672*9880d681SAndroid Build Coastguard Worker // Check if this slice overlaps with another.
10673*9880d681SAndroid Build Coastguard Worker if ((CurrentUsedBits & UsedBits) != 0)
10674*9880d681SAndroid Build Coastguard Worker return false;
10675*9880d681SAndroid Build Coastguard Worker // Update the bits used globally.
10676*9880d681SAndroid Build Coastguard Worker UsedBits |= CurrentUsedBits;
10677*9880d681SAndroid Build Coastguard Worker
10678*9880d681SAndroid Build Coastguard Worker // Check if the new slice would be legal.
10679*9880d681SAndroid Build Coastguard Worker if (!LS.isLegal())
10680*9880d681SAndroid Build Coastguard Worker return false;
10681*9880d681SAndroid Build Coastguard Worker
10682*9880d681SAndroid Build Coastguard Worker // Record the slice.
10683*9880d681SAndroid Build Coastguard Worker LoadedSlices.push_back(LS);
10684*9880d681SAndroid Build Coastguard Worker }
10685*9880d681SAndroid Build Coastguard Worker
10686*9880d681SAndroid Build Coastguard Worker // Abort slicing if it does not seem to be profitable.
10687*9880d681SAndroid Build Coastguard Worker if (!isSlicingProfitable(LoadedSlices, UsedBits, ForCodeSize))
10688*9880d681SAndroid Build Coastguard Worker return false;
10689*9880d681SAndroid Build Coastguard Worker
10690*9880d681SAndroid Build Coastguard Worker ++SlicedLoads;
10691*9880d681SAndroid Build Coastguard Worker
10692*9880d681SAndroid Build Coastguard Worker // Rewrite each chain to use an independent load.
10693*9880d681SAndroid Build Coastguard Worker // By construction, each chain can be represented by a unique load.
10694*9880d681SAndroid Build Coastguard Worker
10695*9880d681SAndroid Build Coastguard Worker // Prepare the argument for the new token factor for all the slices.
10696*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> ArgChains;
10697*9880d681SAndroid Build Coastguard Worker for (SmallVectorImpl<LoadedSlice>::const_iterator
10698*9880d681SAndroid Build Coastguard Worker LSIt = LoadedSlices.begin(),
10699*9880d681SAndroid Build Coastguard Worker LSItEnd = LoadedSlices.end();
10700*9880d681SAndroid Build Coastguard Worker LSIt != LSItEnd; ++LSIt) {
10701*9880d681SAndroid Build Coastguard Worker SDValue SliceInst = LSIt->loadSlice();
10702*9880d681SAndroid Build Coastguard Worker CombineTo(LSIt->Inst, SliceInst, true);
10703*9880d681SAndroid Build Coastguard Worker if (SliceInst.getNode()->getOpcode() != ISD::LOAD)
10704*9880d681SAndroid Build Coastguard Worker SliceInst = SliceInst.getOperand(0);
10705*9880d681SAndroid Build Coastguard Worker assert(SliceInst->getOpcode() == ISD::LOAD &&
10706*9880d681SAndroid Build Coastguard Worker "It takes more than a zext to get to the loaded slice!!");
10707*9880d681SAndroid Build Coastguard Worker ArgChains.push_back(SliceInst.getValue(1));
10708*9880d681SAndroid Build Coastguard Worker }
10709*9880d681SAndroid Build Coastguard Worker
10710*9880d681SAndroid Build Coastguard Worker SDValue Chain = DAG.getNode(ISD::TokenFactor, SDLoc(LD), MVT::Other,
10711*9880d681SAndroid Build Coastguard Worker ArgChains);
10712*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(N, 1), Chain);
10713*9880d681SAndroid Build Coastguard Worker return true;
10714*9880d681SAndroid Build Coastguard Worker }
10715*9880d681SAndroid Build Coastguard Worker
10716*9880d681SAndroid Build Coastguard Worker /// Check to see if V is (and load (ptr), imm), where the load is having
10717*9880d681SAndroid Build Coastguard Worker /// specific bytes cleared out. If so, return the byte size being masked out
10718*9880d681SAndroid Build Coastguard Worker /// and the shift amount.
10719*9880d681SAndroid Build Coastguard Worker static std::pair<unsigned, unsigned>
CheckForMaskedLoad(SDValue V,SDValue Ptr,SDValue Chain)10720*9880d681SAndroid Build Coastguard Worker CheckForMaskedLoad(SDValue V, SDValue Ptr, SDValue Chain) {
10721*9880d681SAndroid Build Coastguard Worker std::pair<unsigned, unsigned> Result(0, 0);
10722*9880d681SAndroid Build Coastguard Worker
10723*9880d681SAndroid Build Coastguard Worker // Check for the structure we're looking for.
10724*9880d681SAndroid Build Coastguard Worker if (V->getOpcode() != ISD::AND ||
10725*9880d681SAndroid Build Coastguard Worker !isa<ConstantSDNode>(V->getOperand(1)) ||
10726*9880d681SAndroid Build Coastguard Worker !ISD::isNormalLoad(V->getOperand(0).getNode()))
10727*9880d681SAndroid Build Coastguard Worker return Result;
10728*9880d681SAndroid Build Coastguard Worker
10729*9880d681SAndroid Build Coastguard Worker // Check the chain and pointer.
10730*9880d681SAndroid Build Coastguard Worker LoadSDNode *LD = cast<LoadSDNode>(V->getOperand(0));
10731*9880d681SAndroid Build Coastguard Worker if (LD->getBasePtr() != Ptr) return Result; // Not from same pointer.
10732*9880d681SAndroid Build Coastguard Worker
10733*9880d681SAndroid Build Coastguard Worker // The store should be chained directly to the load or be an operand of a
10734*9880d681SAndroid Build Coastguard Worker // tokenfactor.
10735*9880d681SAndroid Build Coastguard Worker if (LD == Chain.getNode())
10736*9880d681SAndroid Build Coastguard Worker ; // ok.
10737*9880d681SAndroid Build Coastguard Worker else if (Chain->getOpcode() != ISD::TokenFactor)
10738*9880d681SAndroid Build Coastguard Worker return Result; // Fail.
10739*9880d681SAndroid Build Coastguard Worker else {
10740*9880d681SAndroid Build Coastguard Worker bool isOk = false;
10741*9880d681SAndroid Build Coastguard Worker for (const SDValue &ChainOp : Chain->op_values())
10742*9880d681SAndroid Build Coastguard Worker if (ChainOp.getNode() == LD) {
10743*9880d681SAndroid Build Coastguard Worker isOk = true;
10744*9880d681SAndroid Build Coastguard Worker break;
10745*9880d681SAndroid Build Coastguard Worker }
10746*9880d681SAndroid Build Coastguard Worker if (!isOk) return Result;
10747*9880d681SAndroid Build Coastguard Worker }
10748*9880d681SAndroid Build Coastguard Worker
10749*9880d681SAndroid Build Coastguard Worker // This only handles simple types.
10750*9880d681SAndroid Build Coastguard Worker if (V.getValueType() != MVT::i16 &&
10751*9880d681SAndroid Build Coastguard Worker V.getValueType() != MVT::i32 &&
10752*9880d681SAndroid Build Coastguard Worker V.getValueType() != MVT::i64)
10753*9880d681SAndroid Build Coastguard Worker return Result;
10754*9880d681SAndroid Build Coastguard Worker
10755*9880d681SAndroid Build Coastguard Worker // Check the constant mask. Invert it so that the bits being masked out are
10756*9880d681SAndroid Build Coastguard Worker // 0 and the bits being kept are 1. Use getSExtValue so that leading bits
10757*9880d681SAndroid Build Coastguard Worker // follow the sign bit for uniformity.
10758*9880d681SAndroid Build Coastguard Worker uint64_t NotMask = ~cast<ConstantSDNode>(V->getOperand(1))->getSExtValue();
10759*9880d681SAndroid Build Coastguard Worker unsigned NotMaskLZ = countLeadingZeros(NotMask);
10760*9880d681SAndroid Build Coastguard Worker if (NotMaskLZ & 7) return Result; // Must be multiple of a byte.
10761*9880d681SAndroid Build Coastguard Worker unsigned NotMaskTZ = countTrailingZeros(NotMask);
10762*9880d681SAndroid Build Coastguard Worker if (NotMaskTZ & 7) return Result; // Must be multiple of a byte.
10763*9880d681SAndroid Build Coastguard Worker if (NotMaskLZ == 64) return Result; // All zero mask.
10764*9880d681SAndroid Build Coastguard Worker
10765*9880d681SAndroid Build Coastguard Worker // See if we have a continuous run of bits. If so, we have 0*1+0*
10766*9880d681SAndroid Build Coastguard Worker if (countTrailingOnes(NotMask >> NotMaskTZ) + NotMaskTZ + NotMaskLZ != 64)
10767*9880d681SAndroid Build Coastguard Worker return Result;
10768*9880d681SAndroid Build Coastguard Worker
10769*9880d681SAndroid Build Coastguard Worker // Adjust NotMaskLZ down to be from the actual size of the int instead of i64.
10770*9880d681SAndroid Build Coastguard Worker if (V.getValueType() != MVT::i64 && NotMaskLZ)
10771*9880d681SAndroid Build Coastguard Worker NotMaskLZ -= 64-V.getValueSizeInBits();
10772*9880d681SAndroid Build Coastguard Worker
10773*9880d681SAndroid Build Coastguard Worker unsigned MaskedBytes = (V.getValueSizeInBits()-NotMaskLZ-NotMaskTZ)/8;
10774*9880d681SAndroid Build Coastguard Worker switch (MaskedBytes) {
10775*9880d681SAndroid Build Coastguard Worker case 1:
10776*9880d681SAndroid Build Coastguard Worker case 2:
10777*9880d681SAndroid Build Coastguard Worker case 4: break;
10778*9880d681SAndroid Build Coastguard Worker default: return Result; // All one mask, or 5-byte mask.
10779*9880d681SAndroid Build Coastguard Worker }
10780*9880d681SAndroid Build Coastguard Worker
10781*9880d681SAndroid Build Coastguard Worker // Verify that the first bit starts at a multiple of mask so that the access
10782*9880d681SAndroid Build Coastguard Worker // is aligned the same as the access width.
10783*9880d681SAndroid Build Coastguard Worker if (NotMaskTZ && NotMaskTZ/8 % MaskedBytes) return Result;
10784*9880d681SAndroid Build Coastguard Worker
10785*9880d681SAndroid Build Coastguard Worker Result.first = MaskedBytes;
10786*9880d681SAndroid Build Coastguard Worker Result.second = NotMaskTZ/8;
10787*9880d681SAndroid Build Coastguard Worker return Result;
10788*9880d681SAndroid Build Coastguard Worker }
10789*9880d681SAndroid Build Coastguard Worker
10790*9880d681SAndroid Build Coastguard Worker
10791*9880d681SAndroid Build Coastguard Worker /// Check to see if IVal is something that provides a value as specified by
10792*9880d681SAndroid Build Coastguard Worker /// MaskInfo. If so, replace the specified store with a narrower store of
10793*9880d681SAndroid Build Coastguard Worker /// truncated IVal.
10794*9880d681SAndroid Build Coastguard Worker static SDNode *
ShrinkLoadReplaceStoreWithStore(const std::pair<unsigned,unsigned> & MaskInfo,SDValue IVal,StoreSDNode * St,DAGCombiner * DC)10795*9880d681SAndroid Build Coastguard Worker ShrinkLoadReplaceStoreWithStore(const std::pair<unsigned, unsigned> &MaskInfo,
10796*9880d681SAndroid Build Coastguard Worker SDValue IVal, StoreSDNode *St,
10797*9880d681SAndroid Build Coastguard Worker DAGCombiner *DC) {
10798*9880d681SAndroid Build Coastguard Worker unsigned NumBytes = MaskInfo.first;
10799*9880d681SAndroid Build Coastguard Worker unsigned ByteShift = MaskInfo.second;
10800*9880d681SAndroid Build Coastguard Worker SelectionDAG &DAG = DC->getDAG();
10801*9880d681SAndroid Build Coastguard Worker
10802*9880d681SAndroid Build Coastguard Worker // Check to see if IVal is all zeros in the part being masked in by the 'or'
10803*9880d681SAndroid Build Coastguard Worker // that uses this. If not, this is not a replacement.
10804*9880d681SAndroid Build Coastguard Worker APInt Mask = ~APInt::getBitsSet(IVal.getValueSizeInBits(),
10805*9880d681SAndroid Build Coastguard Worker ByteShift*8, (ByteShift+NumBytes)*8);
10806*9880d681SAndroid Build Coastguard Worker if (!DAG.MaskedValueIsZero(IVal, Mask)) return nullptr;
10807*9880d681SAndroid Build Coastguard Worker
10808*9880d681SAndroid Build Coastguard Worker // Check that it is legal on the target to do this. It is legal if the new
10809*9880d681SAndroid Build Coastguard Worker // VT we're shrinking to (i8/i16/i32) is legal or we're still before type
10810*9880d681SAndroid Build Coastguard Worker // legalization.
10811*9880d681SAndroid Build Coastguard Worker MVT VT = MVT::getIntegerVT(NumBytes*8);
10812*9880d681SAndroid Build Coastguard Worker if (!DC->isTypeLegal(VT))
10813*9880d681SAndroid Build Coastguard Worker return nullptr;
10814*9880d681SAndroid Build Coastguard Worker
10815*9880d681SAndroid Build Coastguard Worker // Okay, we can do this! Replace the 'St' store with a store of IVal that is
10816*9880d681SAndroid Build Coastguard Worker // shifted by ByteShift and truncated down to NumBytes.
10817*9880d681SAndroid Build Coastguard Worker if (ByteShift) {
10818*9880d681SAndroid Build Coastguard Worker SDLoc DL(IVal);
10819*9880d681SAndroid Build Coastguard Worker IVal = DAG.getNode(ISD::SRL, DL, IVal.getValueType(), IVal,
10820*9880d681SAndroid Build Coastguard Worker DAG.getConstant(ByteShift*8, DL,
10821*9880d681SAndroid Build Coastguard Worker DC->getShiftAmountTy(IVal.getValueType())));
10822*9880d681SAndroid Build Coastguard Worker }
10823*9880d681SAndroid Build Coastguard Worker
10824*9880d681SAndroid Build Coastguard Worker // Figure out the offset for the store and the alignment of the access.
10825*9880d681SAndroid Build Coastguard Worker unsigned StOffset;
10826*9880d681SAndroid Build Coastguard Worker unsigned NewAlign = St->getAlignment();
10827*9880d681SAndroid Build Coastguard Worker
10828*9880d681SAndroid Build Coastguard Worker if (DAG.getDataLayout().isLittleEndian())
10829*9880d681SAndroid Build Coastguard Worker StOffset = ByteShift;
10830*9880d681SAndroid Build Coastguard Worker else
10831*9880d681SAndroid Build Coastguard Worker StOffset = IVal.getValueType().getStoreSize() - ByteShift - NumBytes;
10832*9880d681SAndroid Build Coastguard Worker
10833*9880d681SAndroid Build Coastguard Worker SDValue Ptr = St->getBasePtr();
10834*9880d681SAndroid Build Coastguard Worker if (StOffset) {
10835*9880d681SAndroid Build Coastguard Worker SDLoc DL(IVal);
10836*9880d681SAndroid Build Coastguard Worker Ptr = DAG.getNode(ISD::ADD, DL, Ptr.getValueType(),
10837*9880d681SAndroid Build Coastguard Worker Ptr, DAG.getConstant(StOffset, DL, Ptr.getValueType()));
10838*9880d681SAndroid Build Coastguard Worker NewAlign = MinAlign(NewAlign, StOffset);
10839*9880d681SAndroid Build Coastguard Worker }
10840*9880d681SAndroid Build Coastguard Worker
10841*9880d681SAndroid Build Coastguard Worker // Truncate down to the new size.
10842*9880d681SAndroid Build Coastguard Worker IVal = DAG.getNode(ISD::TRUNCATE, SDLoc(IVal), VT, IVal);
10843*9880d681SAndroid Build Coastguard Worker
10844*9880d681SAndroid Build Coastguard Worker ++OpsNarrowed;
10845*9880d681SAndroid Build Coastguard Worker return DAG.getStore(St->getChain(), SDLoc(St), IVal, Ptr,
10846*9880d681SAndroid Build Coastguard Worker St->getPointerInfo().getWithOffset(StOffset),
10847*9880d681SAndroid Build Coastguard Worker false, false, NewAlign).getNode();
10848*9880d681SAndroid Build Coastguard Worker }
10849*9880d681SAndroid Build Coastguard Worker
10850*9880d681SAndroid Build Coastguard Worker
10851*9880d681SAndroid Build Coastguard Worker /// Look for sequence of load / op / store where op is one of 'or', 'xor', and
10852*9880d681SAndroid Build Coastguard Worker /// 'and' of immediates. If 'op' is only touching some of the loaded bits, try
10853*9880d681SAndroid Build Coastguard Worker /// narrowing the load and store if it would end up being a win for performance
10854*9880d681SAndroid Build Coastguard Worker /// or code size.
ReduceLoadOpStoreWidth(SDNode * N)10855*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::ReduceLoadOpStoreWidth(SDNode *N) {
10856*9880d681SAndroid Build Coastguard Worker StoreSDNode *ST = cast<StoreSDNode>(N);
10857*9880d681SAndroid Build Coastguard Worker if (ST->isVolatile())
10858*9880d681SAndroid Build Coastguard Worker return SDValue();
10859*9880d681SAndroid Build Coastguard Worker
10860*9880d681SAndroid Build Coastguard Worker SDValue Chain = ST->getChain();
10861*9880d681SAndroid Build Coastguard Worker SDValue Value = ST->getValue();
10862*9880d681SAndroid Build Coastguard Worker SDValue Ptr = ST->getBasePtr();
10863*9880d681SAndroid Build Coastguard Worker EVT VT = Value.getValueType();
10864*9880d681SAndroid Build Coastguard Worker
10865*9880d681SAndroid Build Coastguard Worker if (ST->isTruncatingStore() || VT.isVector() || !Value.hasOneUse())
10866*9880d681SAndroid Build Coastguard Worker return SDValue();
10867*9880d681SAndroid Build Coastguard Worker
10868*9880d681SAndroid Build Coastguard Worker unsigned Opc = Value.getOpcode();
10869*9880d681SAndroid Build Coastguard Worker
10870*9880d681SAndroid Build Coastguard Worker // If this is "store (or X, Y), P" and X is "(and (load P), cst)", where cst
10871*9880d681SAndroid Build Coastguard Worker // is a byte mask indicating a consecutive number of bytes, check to see if
10872*9880d681SAndroid Build Coastguard Worker // Y is known to provide just those bytes. If so, we try to replace the
10873*9880d681SAndroid Build Coastguard Worker // load + replace + store sequence with a single (narrower) store, which makes
10874*9880d681SAndroid Build Coastguard Worker // the load dead.
10875*9880d681SAndroid Build Coastguard Worker if (Opc == ISD::OR) {
10876*9880d681SAndroid Build Coastguard Worker std::pair<unsigned, unsigned> MaskedLoad;
10877*9880d681SAndroid Build Coastguard Worker MaskedLoad = CheckForMaskedLoad(Value.getOperand(0), Ptr, Chain);
10878*9880d681SAndroid Build Coastguard Worker if (MaskedLoad.first)
10879*9880d681SAndroid Build Coastguard Worker if (SDNode *NewST = ShrinkLoadReplaceStoreWithStore(MaskedLoad,
10880*9880d681SAndroid Build Coastguard Worker Value.getOperand(1), ST,this))
10881*9880d681SAndroid Build Coastguard Worker return SDValue(NewST, 0);
10882*9880d681SAndroid Build Coastguard Worker
10883*9880d681SAndroid Build Coastguard Worker // Or is commutative, so try swapping X and Y.
10884*9880d681SAndroid Build Coastguard Worker MaskedLoad = CheckForMaskedLoad(Value.getOperand(1), Ptr, Chain);
10885*9880d681SAndroid Build Coastguard Worker if (MaskedLoad.first)
10886*9880d681SAndroid Build Coastguard Worker if (SDNode *NewST = ShrinkLoadReplaceStoreWithStore(MaskedLoad,
10887*9880d681SAndroid Build Coastguard Worker Value.getOperand(0), ST,this))
10888*9880d681SAndroid Build Coastguard Worker return SDValue(NewST, 0);
10889*9880d681SAndroid Build Coastguard Worker }
10890*9880d681SAndroid Build Coastguard Worker
10891*9880d681SAndroid Build Coastguard Worker if ((Opc != ISD::OR && Opc != ISD::XOR && Opc != ISD::AND) ||
10892*9880d681SAndroid Build Coastguard Worker Value.getOperand(1).getOpcode() != ISD::Constant)
10893*9880d681SAndroid Build Coastguard Worker return SDValue();
10894*9880d681SAndroid Build Coastguard Worker
10895*9880d681SAndroid Build Coastguard Worker SDValue N0 = Value.getOperand(0);
10896*9880d681SAndroid Build Coastguard Worker if (ISD::isNormalLoad(N0.getNode()) && N0.hasOneUse() &&
10897*9880d681SAndroid Build Coastguard Worker Chain == SDValue(N0.getNode(), 1)) {
10898*9880d681SAndroid Build Coastguard Worker LoadSDNode *LD = cast<LoadSDNode>(N0);
10899*9880d681SAndroid Build Coastguard Worker if (LD->getBasePtr() != Ptr ||
10900*9880d681SAndroid Build Coastguard Worker LD->getPointerInfo().getAddrSpace() !=
10901*9880d681SAndroid Build Coastguard Worker ST->getPointerInfo().getAddrSpace())
10902*9880d681SAndroid Build Coastguard Worker return SDValue();
10903*9880d681SAndroid Build Coastguard Worker
10904*9880d681SAndroid Build Coastguard Worker // Find the type to narrow it the load / op / store to.
10905*9880d681SAndroid Build Coastguard Worker SDValue N1 = Value.getOperand(1);
10906*9880d681SAndroid Build Coastguard Worker unsigned BitWidth = N1.getValueSizeInBits();
10907*9880d681SAndroid Build Coastguard Worker APInt Imm = cast<ConstantSDNode>(N1)->getAPIntValue();
10908*9880d681SAndroid Build Coastguard Worker if (Opc == ISD::AND)
10909*9880d681SAndroid Build Coastguard Worker Imm ^= APInt::getAllOnesValue(BitWidth);
10910*9880d681SAndroid Build Coastguard Worker if (Imm == 0 || Imm.isAllOnesValue())
10911*9880d681SAndroid Build Coastguard Worker return SDValue();
10912*9880d681SAndroid Build Coastguard Worker unsigned ShAmt = Imm.countTrailingZeros();
10913*9880d681SAndroid Build Coastguard Worker unsigned MSB = BitWidth - Imm.countLeadingZeros() - 1;
10914*9880d681SAndroid Build Coastguard Worker unsigned NewBW = NextPowerOf2(MSB - ShAmt);
10915*9880d681SAndroid Build Coastguard Worker EVT NewVT = EVT::getIntegerVT(*DAG.getContext(), NewBW);
10916*9880d681SAndroid Build Coastguard Worker // The narrowing should be profitable, the load/store operation should be
10917*9880d681SAndroid Build Coastguard Worker // legal (or custom) and the store size should be equal to the NewVT width.
10918*9880d681SAndroid Build Coastguard Worker while (NewBW < BitWidth &&
10919*9880d681SAndroid Build Coastguard Worker (NewVT.getStoreSizeInBits() != NewBW ||
10920*9880d681SAndroid Build Coastguard Worker !TLI.isOperationLegalOrCustom(Opc, NewVT) ||
10921*9880d681SAndroid Build Coastguard Worker !TLI.isNarrowingProfitable(VT, NewVT))) {
10922*9880d681SAndroid Build Coastguard Worker NewBW = NextPowerOf2(NewBW);
10923*9880d681SAndroid Build Coastguard Worker NewVT = EVT::getIntegerVT(*DAG.getContext(), NewBW);
10924*9880d681SAndroid Build Coastguard Worker }
10925*9880d681SAndroid Build Coastguard Worker if (NewBW >= BitWidth)
10926*9880d681SAndroid Build Coastguard Worker return SDValue();
10927*9880d681SAndroid Build Coastguard Worker
10928*9880d681SAndroid Build Coastguard Worker // If the lsb changed does not start at the type bitwidth boundary,
10929*9880d681SAndroid Build Coastguard Worker // start at the previous one.
10930*9880d681SAndroid Build Coastguard Worker if (ShAmt % NewBW)
10931*9880d681SAndroid Build Coastguard Worker ShAmt = (((ShAmt + NewBW - 1) / NewBW) * NewBW) - NewBW;
10932*9880d681SAndroid Build Coastguard Worker APInt Mask = APInt::getBitsSet(BitWidth, ShAmt,
10933*9880d681SAndroid Build Coastguard Worker std::min(BitWidth, ShAmt + NewBW));
10934*9880d681SAndroid Build Coastguard Worker if ((Imm & Mask) == Imm) {
10935*9880d681SAndroid Build Coastguard Worker APInt NewImm = (Imm & Mask).lshr(ShAmt).trunc(NewBW);
10936*9880d681SAndroid Build Coastguard Worker if (Opc == ISD::AND)
10937*9880d681SAndroid Build Coastguard Worker NewImm ^= APInt::getAllOnesValue(NewBW);
10938*9880d681SAndroid Build Coastguard Worker uint64_t PtrOff = ShAmt / 8;
10939*9880d681SAndroid Build Coastguard Worker // For big endian targets, we need to adjust the offset to the pointer to
10940*9880d681SAndroid Build Coastguard Worker // load the correct bytes.
10941*9880d681SAndroid Build Coastguard Worker if (DAG.getDataLayout().isBigEndian())
10942*9880d681SAndroid Build Coastguard Worker PtrOff = (BitWidth + 7 - NewBW) / 8 - PtrOff;
10943*9880d681SAndroid Build Coastguard Worker
10944*9880d681SAndroid Build Coastguard Worker unsigned NewAlign = MinAlign(LD->getAlignment(), PtrOff);
10945*9880d681SAndroid Build Coastguard Worker Type *NewVTTy = NewVT.getTypeForEVT(*DAG.getContext());
10946*9880d681SAndroid Build Coastguard Worker if (NewAlign < DAG.getDataLayout().getABITypeAlignment(NewVTTy))
10947*9880d681SAndroid Build Coastguard Worker return SDValue();
10948*9880d681SAndroid Build Coastguard Worker
10949*9880d681SAndroid Build Coastguard Worker SDValue NewPtr = DAG.getNode(ISD::ADD, SDLoc(LD),
10950*9880d681SAndroid Build Coastguard Worker Ptr.getValueType(), Ptr,
10951*9880d681SAndroid Build Coastguard Worker DAG.getConstant(PtrOff, SDLoc(LD),
10952*9880d681SAndroid Build Coastguard Worker Ptr.getValueType()));
10953*9880d681SAndroid Build Coastguard Worker SDValue NewLD = DAG.getLoad(NewVT, SDLoc(N0),
10954*9880d681SAndroid Build Coastguard Worker LD->getChain(), NewPtr,
10955*9880d681SAndroid Build Coastguard Worker LD->getPointerInfo().getWithOffset(PtrOff),
10956*9880d681SAndroid Build Coastguard Worker LD->isVolatile(), LD->isNonTemporal(),
10957*9880d681SAndroid Build Coastguard Worker LD->isInvariant(), NewAlign,
10958*9880d681SAndroid Build Coastguard Worker LD->getAAInfo());
10959*9880d681SAndroid Build Coastguard Worker SDValue NewVal = DAG.getNode(Opc, SDLoc(Value), NewVT, NewLD,
10960*9880d681SAndroid Build Coastguard Worker DAG.getConstant(NewImm, SDLoc(Value),
10961*9880d681SAndroid Build Coastguard Worker NewVT));
10962*9880d681SAndroid Build Coastguard Worker SDValue NewST = DAG.getStore(Chain, SDLoc(N),
10963*9880d681SAndroid Build Coastguard Worker NewVal, NewPtr,
10964*9880d681SAndroid Build Coastguard Worker ST->getPointerInfo().getWithOffset(PtrOff),
10965*9880d681SAndroid Build Coastguard Worker false, false, NewAlign);
10966*9880d681SAndroid Build Coastguard Worker
10967*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewPtr.getNode());
10968*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewLD.getNode());
10969*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewVal.getNode());
10970*9880d681SAndroid Build Coastguard Worker WorklistRemover DeadNodes(*this);
10971*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(N0.getValue(1), NewLD.getValue(1));
10972*9880d681SAndroid Build Coastguard Worker ++OpsNarrowed;
10973*9880d681SAndroid Build Coastguard Worker return NewST;
10974*9880d681SAndroid Build Coastguard Worker }
10975*9880d681SAndroid Build Coastguard Worker }
10976*9880d681SAndroid Build Coastguard Worker
10977*9880d681SAndroid Build Coastguard Worker return SDValue();
10978*9880d681SAndroid Build Coastguard Worker }
10979*9880d681SAndroid Build Coastguard Worker
10980*9880d681SAndroid Build Coastguard Worker /// For a given floating point load / store pair, if the load value isn't used
10981*9880d681SAndroid Build Coastguard Worker /// by any other operations, then consider transforming the pair to integer
10982*9880d681SAndroid Build Coastguard Worker /// load / store operations if the target deems the transformation profitable.
TransformFPLoadStorePair(SDNode * N)10983*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::TransformFPLoadStorePair(SDNode *N) {
10984*9880d681SAndroid Build Coastguard Worker StoreSDNode *ST = cast<StoreSDNode>(N);
10985*9880d681SAndroid Build Coastguard Worker SDValue Chain = ST->getChain();
10986*9880d681SAndroid Build Coastguard Worker SDValue Value = ST->getValue();
10987*9880d681SAndroid Build Coastguard Worker if (ISD::isNormalStore(ST) && ISD::isNormalLoad(Value.getNode()) &&
10988*9880d681SAndroid Build Coastguard Worker Value.hasOneUse() &&
10989*9880d681SAndroid Build Coastguard Worker Chain == SDValue(Value.getNode(), 1)) {
10990*9880d681SAndroid Build Coastguard Worker LoadSDNode *LD = cast<LoadSDNode>(Value);
10991*9880d681SAndroid Build Coastguard Worker EVT VT = LD->getMemoryVT();
10992*9880d681SAndroid Build Coastguard Worker if (!VT.isFloatingPoint() ||
10993*9880d681SAndroid Build Coastguard Worker VT != ST->getMemoryVT() ||
10994*9880d681SAndroid Build Coastguard Worker LD->isNonTemporal() ||
10995*9880d681SAndroid Build Coastguard Worker ST->isNonTemporal() ||
10996*9880d681SAndroid Build Coastguard Worker LD->getPointerInfo().getAddrSpace() != 0 ||
10997*9880d681SAndroid Build Coastguard Worker ST->getPointerInfo().getAddrSpace() != 0)
10998*9880d681SAndroid Build Coastguard Worker return SDValue();
10999*9880d681SAndroid Build Coastguard Worker
11000*9880d681SAndroid Build Coastguard Worker EVT IntVT = EVT::getIntegerVT(*DAG.getContext(), VT.getSizeInBits());
11001*9880d681SAndroid Build Coastguard Worker if (!TLI.isOperationLegal(ISD::LOAD, IntVT) ||
11002*9880d681SAndroid Build Coastguard Worker !TLI.isOperationLegal(ISD::STORE, IntVT) ||
11003*9880d681SAndroid Build Coastguard Worker !TLI.isDesirableToTransformToIntegerOp(ISD::LOAD, VT) ||
11004*9880d681SAndroid Build Coastguard Worker !TLI.isDesirableToTransformToIntegerOp(ISD::STORE, VT))
11005*9880d681SAndroid Build Coastguard Worker return SDValue();
11006*9880d681SAndroid Build Coastguard Worker
11007*9880d681SAndroid Build Coastguard Worker unsigned LDAlign = LD->getAlignment();
11008*9880d681SAndroid Build Coastguard Worker unsigned STAlign = ST->getAlignment();
11009*9880d681SAndroid Build Coastguard Worker Type *IntVTTy = IntVT.getTypeForEVT(*DAG.getContext());
11010*9880d681SAndroid Build Coastguard Worker unsigned ABIAlign = DAG.getDataLayout().getABITypeAlignment(IntVTTy);
11011*9880d681SAndroid Build Coastguard Worker if (LDAlign < ABIAlign || STAlign < ABIAlign)
11012*9880d681SAndroid Build Coastguard Worker return SDValue();
11013*9880d681SAndroid Build Coastguard Worker
11014*9880d681SAndroid Build Coastguard Worker SDValue NewLD = DAG.getLoad(IntVT, SDLoc(Value),
11015*9880d681SAndroid Build Coastguard Worker LD->getChain(), LD->getBasePtr(),
11016*9880d681SAndroid Build Coastguard Worker LD->getPointerInfo(),
11017*9880d681SAndroid Build Coastguard Worker false, false, false, LDAlign);
11018*9880d681SAndroid Build Coastguard Worker
11019*9880d681SAndroid Build Coastguard Worker SDValue NewST = DAG.getStore(NewLD.getValue(1), SDLoc(N),
11020*9880d681SAndroid Build Coastguard Worker NewLD, ST->getBasePtr(),
11021*9880d681SAndroid Build Coastguard Worker ST->getPointerInfo(),
11022*9880d681SAndroid Build Coastguard Worker false, false, STAlign);
11023*9880d681SAndroid Build Coastguard Worker
11024*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewLD.getNode());
11025*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewST.getNode());
11026*9880d681SAndroid Build Coastguard Worker WorklistRemover DeadNodes(*this);
11027*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(Value.getValue(1), NewLD.getValue(1));
11028*9880d681SAndroid Build Coastguard Worker ++LdStFP2Int;
11029*9880d681SAndroid Build Coastguard Worker return NewST;
11030*9880d681SAndroid Build Coastguard Worker }
11031*9880d681SAndroid Build Coastguard Worker
11032*9880d681SAndroid Build Coastguard Worker return SDValue();
11033*9880d681SAndroid Build Coastguard Worker }
11034*9880d681SAndroid Build Coastguard Worker
11035*9880d681SAndroid Build Coastguard Worker namespace {
11036*9880d681SAndroid Build Coastguard Worker /// Helper struct to parse and store a memory address as base + index + offset.
11037*9880d681SAndroid Build Coastguard Worker /// We ignore sign extensions when it is safe to do so.
11038*9880d681SAndroid Build Coastguard Worker /// The following two expressions are not equivalent. To differentiate we need
11039*9880d681SAndroid Build Coastguard Worker /// to store whether there was a sign extension involved in the index
11040*9880d681SAndroid Build Coastguard Worker /// computation.
11041*9880d681SAndroid Build Coastguard Worker /// (load (i64 add (i64 copyfromreg %c)
11042*9880d681SAndroid Build Coastguard Worker /// (i64 signextend (add (i8 load %index)
11043*9880d681SAndroid Build Coastguard Worker /// (i8 1))))
11044*9880d681SAndroid Build Coastguard Worker /// vs
11045*9880d681SAndroid Build Coastguard Worker ///
11046*9880d681SAndroid Build Coastguard Worker /// (load (i64 add (i64 copyfromreg %c)
11047*9880d681SAndroid Build Coastguard Worker /// (i64 signextend (i32 add (i32 signextend (i8 load %index))
11048*9880d681SAndroid Build Coastguard Worker /// (i32 1)))))
11049*9880d681SAndroid Build Coastguard Worker struct BaseIndexOffset {
11050*9880d681SAndroid Build Coastguard Worker SDValue Base;
11051*9880d681SAndroid Build Coastguard Worker SDValue Index;
11052*9880d681SAndroid Build Coastguard Worker int64_t Offset;
11053*9880d681SAndroid Build Coastguard Worker bool IsIndexSignExt;
11054*9880d681SAndroid Build Coastguard Worker
BaseIndexOffset__anon093339a80a11::BaseIndexOffset11055*9880d681SAndroid Build Coastguard Worker BaseIndexOffset() : Offset(0), IsIndexSignExt(false) {}
11056*9880d681SAndroid Build Coastguard Worker
BaseIndexOffset__anon093339a80a11::BaseIndexOffset11057*9880d681SAndroid Build Coastguard Worker BaseIndexOffset(SDValue Base, SDValue Index, int64_t Offset,
11058*9880d681SAndroid Build Coastguard Worker bool IsIndexSignExt) :
11059*9880d681SAndroid Build Coastguard Worker Base(Base), Index(Index), Offset(Offset), IsIndexSignExt(IsIndexSignExt) {}
11060*9880d681SAndroid Build Coastguard Worker
equalBaseIndex__anon093339a80a11::BaseIndexOffset11061*9880d681SAndroid Build Coastguard Worker bool equalBaseIndex(const BaseIndexOffset &Other) {
11062*9880d681SAndroid Build Coastguard Worker return Other.Base == Base && Other.Index == Index &&
11063*9880d681SAndroid Build Coastguard Worker Other.IsIndexSignExt == IsIndexSignExt;
11064*9880d681SAndroid Build Coastguard Worker }
11065*9880d681SAndroid Build Coastguard Worker
11066*9880d681SAndroid Build Coastguard Worker /// Parses tree in Ptr for base, index, offset addresses.
match__anon093339a80a11::BaseIndexOffset11067*9880d681SAndroid Build Coastguard Worker static BaseIndexOffset match(SDValue Ptr, SelectionDAG &DAG) {
11068*9880d681SAndroid Build Coastguard Worker bool IsIndexSignExt = false;
11069*9880d681SAndroid Build Coastguard Worker
11070*9880d681SAndroid Build Coastguard Worker // Split up a folded GlobalAddress+Offset into its component parts.
11071*9880d681SAndroid Build Coastguard Worker if (GlobalAddressSDNode *GA = dyn_cast<GlobalAddressSDNode>(Ptr))
11072*9880d681SAndroid Build Coastguard Worker if (GA->getOpcode() == ISD::GlobalAddress && GA->getOffset() != 0) {
11073*9880d681SAndroid Build Coastguard Worker return BaseIndexOffset(DAG.getGlobalAddress(GA->getGlobal(),
11074*9880d681SAndroid Build Coastguard Worker SDLoc(GA),
11075*9880d681SAndroid Build Coastguard Worker GA->getValueType(0),
11076*9880d681SAndroid Build Coastguard Worker /*Offset=*/0,
11077*9880d681SAndroid Build Coastguard Worker /*isTargetGA=*/false,
11078*9880d681SAndroid Build Coastguard Worker GA->getTargetFlags()),
11079*9880d681SAndroid Build Coastguard Worker SDValue(),
11080*9880d681SAndroid Build Coastguard Worker GA->getOffset(),
11081*9880d681SAndroid Build Coastguard Worker IsIndexSignExt);
11082*9880d681SAndroid Build Coastguard Worker }
11083*9880d681SAndroid Build Coastguard Worker
11084*9880d681SAndroid Build Coastguard Worker // We only can pattern match BASE + INDEX + OFFSET. If Ptr is not an ADD
11085*9880d681SAndroid Build Coastguard Worker // instruction, then it could be just the BASE or everything else we don't
11086*9880d681SAndroid Build Coastguard Worker // know how to handle. Just use Ptr as BASE and give up.
11087*9880d681SAndroid Build Coastguard Worker if (Ptr->getOpcode() != ISD::ADD)
11088*9880d681SAndroid Build Coastguard Worker return BaseIndexOffset(Ptr, SDValue(), 0, IsIndexSignExt);
11089*9880d681SAndroid Build Coastguard Worker
11090*9880d681SAndroid Build Coastguard Worker // We know that we have at least an ADD instruction. Try to pattern match
11091*9880d681SAndroid Build Coastguard Worker // the simple case of BASE + OFFSET.
11092*9880d681SAndroid Build Coastguard Worker if (isa<ConstantSDNode>(Ptr->getOperand(1))) {
11093*9880d681SAndroid Build Coastguard Worker int64_t Offset = cast<ConstantSDNode>(Ptr->getOperand(1))->getSExtValue();
11094*9880d681SAndroid Build Coastguard Worker return BaseIndexOffset(Ptr->getOperand(0), SDValue(), Offset,
11095*9880d681SAndroid Build Coastguard Worker IsIndexSignExt);
11096*9880d681SAndroid Build Coastguard Worker }
11097*9880d681SAndroid Build Coastguard Worker
11098*9880d681SAndroid Build Coastguard Worker // Inside a loop the current BASE pointer is calculated using an ADD and a
11099*9880d681SAndroid Build Coastguard Worker // MUL instruction. In this case Ptr is the actual BASE pointer.
11100*9880d681SAndroid Build Coastguard Worker // (i64 add (i64 %array_ptr)
11101*9880d681SAndroid Build Coastguard Worker // (i64 mul (i64 %induction_var)
11102*9880d681SAndroid Build Coastguard Worker // (i64 %element_size)))
11103*9880d681SAndroid Build Coastguard Worker if (Ptr->getOperand(1)->getOpcode() == ISD::MUL)
11104*9880d681SAndroid Build Coastguard Worker return BaseIndexOffset(Ptr, SDValue(), 0, IsIndexSignExt);
11105*9880d681SAndroid Build Coastguard Worker
11106*9880d681SAndroid Build Coastguard Worker // Look at Base + Index + Offset cases.
11107*9880d681SAndroid Build Coastguard Worker SDValue Base = Ptr->getOperand(0);
11108*9880d681SAndroid Build Coastguard Worker SDValue IndexOffset = Ptr->getOperand(1);
11109*9880d681SAndroid Build Coastguard Worker
11110*9880d681SAndroid Build Coastguard Worker // Skip signextends.
11111*9880d681SAndroid Build Coastguard Worker if (IndexOffset->getOpcode() == ISD::SIGN_EXTEND) {
11112*9880d681SAndroid Build Coastguard Worker IndexOffset = IndexOffset->getOperand(0);
11113*9880d681SAndroid Build Coastguard Worker IsIndexSignExt = true;
11114*9880d681SAndroid Build Coastguard Worker }
11115*9880d681SAndroid Build Coastguard Worker
11116*9880d681SAndroid Build Coastguard Worker // Either the case of Base + Index (no offset) or something else.
11117*9880d681SAndroid Build Coastguard Worker if (IndexOffset->getOpcode() != ISD::ADD)
11118*9880d681SAndroid Build Coastguard Worker return BaseIndexOffset(Base, IndexOffset, 0, IsIndexSignExt);
11119*9880d681SAndroid Build Coastguard Worker
11120*9880d681SAndroid Build Coastguard Worker // Now we have the case of Base + Index + offset.
11121*9880d681SAndroid Build Coastguard Worker SDValue Index = IndexOffset->getOperand(0);
11122*9880d681SAndroid Build Coastguard Worker SDValue Offset = IndexOffset->getOperand(1);
11123*9880d681SAndroid Build Coastguard Worker
11124*9880d681SAndroid Build Coastguard Worker if (!isa<ConstantSDNode>(Offset))
11125*9880d681SAndroid Build Coastguard Worker return BaseIndexOffset(Ptr, SDValue(), 0, IsIndexSignExt);
11126*9880d681SAndroid Build Coastguard Worker
11127*9880d681SAndroid Build Coastguard Worker // Ignore signextends.
11128*9880d681SAndroid Build Coastguard Worker if (Index->getOpcode() == ISD::SIGN_EXTEND) {
11129*9880d681SAndroid Build Coastguard Worker Index = Index->getOperand(0);
11130*9880d681SAndroid Build Coastguard Worker IsIndexSignExt = true;
11131*9880d681SAndroid Build Coastguard Worker } else IsIndexSignExt = false;
11132*9880d681SAndroid Build Coastguard Worker
11133*9880d681SAndroid Build Coastguard Worker int64_t Off = cast<ConstantSDNode>(Offset)->getSExtValue();
11134*9880d681SAndroid Build Coastguard Worker return BaseIndexOffset(Base, Index, Off, IsIndexSignExt);
11135*9880d681SAndroid Build Coastguard Worker }
11136*9880d681SAndroid Build Coastguard Worker };
11137*9880d681SAndroid Build Coastguard Worker } // namespace
11138*9880d681SAndroid Build Coastguard Worker
11139*9880d681SAndroid Build Coastguard Worker // This is a helper function for visitMUL to check the profitability
11140*9880d681SAndroid Build Coastguard Worker // of folding (mul (add x, c1), c2) -> (add (mul x, c2), c1*c2).
11141*9880d681SAndroid Build Coastguard Worker // MulNode is the original multiply, AddNode is (add x, c1),
11142*9880d681SAndroid Build Coastguard Worker // and ConstNode is c2.
11143*9880d681SAndroid Build Coastguard Worker //
11144*9880d681SAndroid Build Coastguard Worker // If the (add x, c1) has multiple uses, we could increase
11145*9880d681SAndroid Build Coastguard Worker // the number of adds if we make this transformation.
11146*9880d681SAndroid Build Coastguard Worker // It would only be worth doing this if we can remove a
11147*9880d681SAndroid Build Coastguard Worker // multiply in the process. Check for that here.
11148*9880d681SAndroid Build Coastguard Worker // To illustrate:
11149*9880d681SAndroid Build Coastguard Worker // (A + c1) * c3
11150*9880d681SAndroid Build Coastguard Worker // (A + c2) * c3
11151*9880d681SAndroid Build Coastguard Worker // We're checking for cases where we have common "c3 * A" expressions.
isMulAddWithConstProfitable(SDNode * MulNode,SDValue & AddNode,SDValue & ConstNode)11152*9880d681SAndroid Build Coastguard Worker bool DAGCombiner::isMulAddWithConstProfitable(SDNode *MulNode,
11153*9880d681SAndroid Build Coastguard Worker SDValue &AddNode,
11154*9880d681SAndroid Build Coastguard Worker SDValue &ConstNode) {
11155*9880d681SAndroid Build Coastguard Worker APInt Val;
11156*9880d681SAndroid Build Coastguard Worker
11157*9880d681SAndroid Build Coastguard Worker // If the add only has one use, this would be OK to do.
11158*9880d681SAndroid Build Coastguard Worker if (AddNode.getNode()->hasOneUse())
11159*9880d681SAndroid Build Coastguard Worker return true;
11160*9880d681SAndroid Build Coastguard Worker
11161*9880d681SAndroid Build Coastguard Worker // Walk all the users of the constant with which we're multiplying.
11162*9880d681SAndroid Build Coastguard Worker for (SDNode *Use : ConstNode->uses()) {
11163*9880d681SAndroid Build Coastguard Worker
11164*9880d681SAndroid Build Coastguard Worker if (Use == MulNode) // This use is the one we're on right now. Skip it.
11165*9880d681SAndroid Build Coastguard Worker continue;
11166*9880d681SAndroid Build Coastguard Worker
11167*9880d681SAndroid Build Coastguard Worker if (Use->getOpcode() == ISD::MUL) { // We have another multiply use.
11168*9880d681SAndroid Build Coastguard Worker SDNode *OtherOp;
11169*9880d681SAndroid Build Coastguard Worker SDNode *MulVar = AddNode.getOperand(0).getNode();
11170*9880d681SAndroid Build Coastguard Worker
11171*9880d681SAndroid Build Coastguard Worker // OtherOp is what we're multiplying against the constant.
11172*9880d681SAndroid Build Coastguard Worker if (Use->getOperand(0) == ConstNode)
11173*9880d681SAndroid Build Coastguard Worker OtherOp = Use->getOperand(1).getNode();
11174*9880d681SAndroid Build Coastguard Worker else
11175*9880d681SAndroid Build Coastguard Worker OtherOp = Use->getOperand(0).getNode();
11176*9880d681SAndroid Build Coastguard Worker
11177*9880d681SAndroid Build Coastguard Worker // Check to see if multiply is with the same operand of our "add".
11178*9880d681SAndroid Build Coastguard Worker //
11179*9880d681SAndroid Build Coastguard Worker // ConstNode = CONST
11180*9880d681SAndroid Build Coastguard Worker // Use = ConstNode * A <-- visiting Use. OtherOp is A.
11181*9880d681SAndroid Build Coastguard Worker // ...
11182*9880d681SAndroid Build Coastguard Worker // AddNode = (A + c1) <-- MulVar is A.
11183*9880d681SAndroid Build Coastguard Worker // = AddNode * ConstNode <-- current visiting instruction.
11184*9880d681SAndroid Build Coastguard Worker //
11185*9880d681SAndroid Build Coastguard Worker // If we make this transformation, we will have a common
11186*9880d681SAndroid Build Coastguard Worker // multiply (ConstNode * A) that we can save.
11187*9880d681SAndroid Build Coastguard Worker if (OtherOp == MulVar)
11188*9880d681SAndroid Build Coastguard Worker return true;
11189*9880d681SAndroid Build Coastguard Worker
11190*9880d681SAndroid Build Coastguard Worker // Now check to see if a future expansion will give us a common
11191*9880d681SAndroid Build Coastguard Worker // multiply.
11192*9880d681SAndroid Build Coastguard Worker //
11193*9880d681SAndroid Build Coastguard Worker // ConstNode = CONST
11194*9880d681SAndroid Build Coastguard Worker // AddNode = (A + c1)
11195*9880d681SAndroid Build Coastguard Worker // ... = AddNode * ConstNode <-- current visiting instruction.
11196*9880d681SAndroid Build Coastguard Worker // ...
11197*9880d681SAndroid Build Coastguard Worker // OtherOp = (A + c2)
11198*9880d681SAndroid Build Coastguard Worker // Use = OtherOp * ConstNode <-- visiting Use.
11199*9880d681SAndroid Build Coastguard Worker //
11200*9880d681SAndroid Build Coastguard Worker // If we make this transformation, we will have a common
11201*9880d681SAndroid Build Coastguard Worker // multiply (CONST * A) after we also do the same transformation
11202*9880d681SAndroid Build Coastguard Worker // to the "t2" instruction.
11203*9880d681SAndroid Build Coastguard Worker if (OtherOp->getOpcode() == ISD::ADD &&
11204*9880d681SAndroid Build Coastguard Worker DAG.isConstantIntBuildVectorOrConstantInt(OtherOp->getOperand(1)) &&
11205*9880d681SAndroid Build Coastguard Worker OtherOp->getOperand(0).getNode() == MulVar)
11206*9880d681SAndroid Build Coastguard Worker return true;
11207*9880d681SAndroid Build Coastguard Worker }
11208*9880d681SAndroid Build Coastguard Worker }
11209*9880d681SAndroid Build Coastguard Worker
11210*9880d681SAndroid Build Coastguard Worker // Didn't find a case where this would be profitable.
11211*9880d681SAndroid Build Coastguard Worker return false;
11212*9880d681SAndroid Build Coastguard Worker }
11213*9880d681SAndroid Build Coastguard Worker
getMergedConstantVectorStore(SelectionDAG & DAG,const SDLoc & SL,ArrayRef<MemOpLink> Stores,SmallVectorImpl<SDValue> & Chains,EVT Ty) const11214*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::getMergedConstantVectorStore(
11215*9880d681SAndroid Build Coastguard Worker SelectionDAG &DAG, const SDLoc &SL, ArrayRef<MemOpLink> Stores,
11216*9880d681SAndroid Build Coastguard Worker SmallVectorImpl<SDValue> &Chains, EVT Ty) const {
11217*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> BuildVector;
11218*9880d681SAndroid Build Coastguard Worker
11219*9880d681SAndroid Build Coastguard Worker for (unsigned I = 0, E = Ty.getVectorNumElements(); I != E; ++I) {
11220*9880d681SAndroid Build Coastguard Worker StoreSDNode *St = cast<StoreSDNode>(Stores[I].MemNode);
11221*9880d681SAndroid Build Coastguard Worker Chains.push_back(St->getChain());
11222*9880d681SAndroid Build Coastguard Worker BuildVector.push_back(St->getValue());
11223*9880d681SAndroid Build Coastguard Worker }
11224*9880d681SAndroid Build Coastguard Worker
11225*9880d681SAndroid Build Coastguard Worker return DAG.getBuildVector(Ty, SL, BuildVector);
11226*9880d681SAndroid Build Coastguard Worker }
11227*9880d681SAndroid Build Coastguard Worker
MergeStoresOfConstantsOrVecElts(SmallVectorImpl<MemOpLink> & StoreNodes,EVT MemVT,unsigned NumStores,bool IsConstantSrc,bool UseVector)11228*9880d681SAndroid Build Coastguard Worker bool DAGCombiner::MergeStoresOfConstantsOrVecElts(
11229*9880d681SAndroid Build Coastguard Worker SmallVectorImpl<MemOpLink> &StoreNodes, EVT MemVT,
11230*9880d681SAndroid Build Coastguard Worker unsigned NumStores, bool IsConstantSrc, bool UseVector) {
11231*9880d681SAndroid Build Coastguard Worker // Make sure we have something to merge.
11232*9880d681SAndroid Build Coastguard Worker if (NumStores < 2)
11233*9880d681SAndroid Build Coastguard Worker return false;
11234*9880d681SAndroid Build Coastguard Worker
11235*9880d681SAndroid Build Coastguard Worker int64_t ElementSizeBytes = MemVT.getSizeInBits() / 8;
11236*9880d681SAndroid Build Coastguard Worker LSBaseSDNode *FirstInChain = StoreNodes[0].MemNode;
11237*9880d681SAndroid Build Coastguard Worker unsigned LatestNodeUsed = 0;
11238*9880d681SAndroid Build Coastguard Worker
11239*9880d681SAndroid Build Coastguard Worker for (unsigned i=0; i < NumStores; ++i) {
11240*9880d681SAndroid Build Coastguard Worker // Find a chain for the new wide-store operand. Notice that some
11241*9880d681SAndroid Build Coastguard Worker // of the store nodes that we found may not be selected for inclusion
11242*9880d681SAndroid Build Coastguard Worker // in the wide store. The chain we use needs to be the chain of the
11243*9880d681SAndroid Build Coastguard Worker // latest store node which is *used* and replaced by the wide store.
11244*9880d681SAndroid Build Coastguard Worker if (StoreNodes[i].SequenceNum < StoreNodes[LatestNodeUsed].SequenceNum)
11245*9880d681SAndroid Build Coastguard Worker LatestNodeUsed = i;
11246*9880d681SAndroid Build Coastguard Worker }
11247*9880d681SAndroid Build Coastguard Worker
11248*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Chains;
11249*9880d681SAndroid Build Coastguard Worker
11250*9880d681SAndroid Build Coastguard Worker // The latest Node in the DAG.
11251*9880d681SAndroid Build Coastguard Worker LSBaseSDNode *LatestOp = StoreNodes[LatestNodeUsed].MemNode;
11252*9880d681SAndroid Build Coastguard Worker SDLoc DL(StoreNodes[0].MemNode);
11253*9880d681SAndroid Build Coastguard Worker
11254*9880d681SAndroid Build Coastguard Worker SDValue StoredVal;
11255*9880d681SAndroid Build Coastguard Worker if (UseVector) {
11256*9880d681SAndroid Build Coastguard Worker bool IsVec = MemVT.isVector();
11257*9880d681SAndroid Build Coastguard Worker unsigned Elts = NumStores;
11258*9880d681SAndroid Build Coastguard Worker if (IsVec) {
11259*9880d681SAndroid Build Coastguard Worker // When merging vector stores, get the total number of elements.
11260*9880d681SAndroid Build Coastguard Worker Elts *= MemVT.getVectorNumElements();
11261*9880d681SAndroid Build Coastguard Worker }
11262*9880d681SAndroid Build Coastguard Worker // Get the type for the merged vector store.
11263*9880d681SAndroid Build Coastguard Worker EVT Ty = EVT::getVectorVT(*DAG.getContext(), MemVT.getScalarType(), Elts);
11264*9880d681SAndroid Build Coastguard Worker assert(TLI.isTypeLegal(Ty) && "Illegal vector store");
11265*9880d681SAndroid Build Coastguard Worker
11266*9880d681SAndroid Build Coastguard Worker if (IsConstantSrc) {
11267*9880d681SAndroid Build Coastguard Worker StoredVal = getMergedConstantVectorStore(DAG, DL, StoreNodes, Chains, Ty);
11268*9880d681SAndroid Build Coastguard Worker } else {
11269*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Ops;
11270*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < NumStores; ++i) {
11271*9880d681SAndroid Build Coastguard Worker StoreSDNode *St = cast<StoreSDNode>(StoreNodes[i].MemNode);
11272*9880d681SAndroid Build Coastguard Worker SDValue Val = St->getValue();
11273*9880d681SAndroid Build Coastguard Worker // All operands of BUILD_VECTOR / CONCAT_VECTOR must have the same type.
11274*9880d681SAndroid Build Coastguard Worker if (Val.getValueType() != MemVT)
11275*9880d681SAndroid Build Coastguard Worker return false;
11276*9880d681SAndroid Build Coastguard Worker Ops.push_back(Val);
11277*9880d681SAndroid Build Coastguard Worker Chains.push_back(St->getChain());
11278*9880d681SAndroid Build Coastguard Worker }
11279*9880d681SAndroid Build Coastguard Worker
11280*9880d681SAndroid Build Coastguard Worker // Build the extracted vector elements back into a vector.
11281*9880d681SAndroid Build Coastguard Worker StoredVal = DAG.getNode(IsVec ? ISD::CONCAT_VECTORS : ISD::BUILD_VECTOR,
11282*9880d681SAndroid Build Coastguard Worker DL, Ty, Ops); }
11283*9880d681SAndroid Build Coastguard Worker } else {
11284*9880d681SAndroid Build Coastguard Worker // We should always use a vector store when merging extracted vector
11285*9880d681SAndroid Build Coastguard Worker // elements, so this path implies a store of constants.
11286*9880d681SAndroid Build Coastguard Worker assert(IsConstantSrc && "Merged vector elements should use vector store");
11287*9880d681SAndroid Build Coastguard Worker
11288*9880d681SAndroid Build Coastguard Worker unsigned SizeInBits = NumStores * ElementSizeBytes * 8;
11289*9880d681SAndroid Build Coastguard Worker APInt StoreInt(SizeInBits, 0);
11290*9880d681SAndroid Build Coastguard Worker
11291*9880d681SAndroid Build Coastguard Worker // Construct a single integer constant which is made of the smaller
11292*9880d681SAndroid Build Coastguard Worker // constant inputs.
11293*9880d681SAndroid Build Coastguard Worker bool IsLE = DAG.getDataLayout().isLittleEndian();
11294*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < NumStores; ++i) {
11295*9880d681SAndroid Build Coastguard Worker unsigned Idx = IsLE ? (NumStores - 1 - i) : i;
11296*9880d681SAndroid Build Coastguard Worker StoreSDNode *St = cast<StoreSDNode>(StoreNodes[Idx].MemNode);
11297*9880d681SAndroid Build Coastguard Worker Chains.push_back(St->getChain());
11298*9880d681SAndroid Build Coastguard Worker
11299*9880d681SAndroid Build Coastguard Worker SDValue Val = St->getValue();
11300*9880d681SAndroid Build Coastguard Worker StoreInt <<= ElementSizeBytes * 8;
11301*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Val)) {
11302*9880d681SAndroid Build Coastguard Worker StoreInt |= C->getAPIntValue().zext(SizeInBits);
11303*9880d681SAndroid Build Coastguard Worker } else if (ConstantFPSDNode *C = dyn_cast<ConstantFPSDNode>(Val)) {
11304*9880d681SAndroid Build Coastguard Worker StoreInt |= C->getValueAPF().bitcastToAPInt().zext(SizeInBits);
11305*9880d681SAndroid Build Coastguard Worker } else {
11306*9880d681SAndroid Build Coastguard Worker llvm_unreachable("Invalid constant element type");
11307*9880d681SAndroid Build Coastguard Worker }
11308*9880d681SAndroid Build Coastguard Worker }
11309*9880d681SAndroid Build Coastguard Worker
11310*9880d681SAndroid Build Coastguard Worker // Create the new Load and Store operations.
11311*9880d681SAndroid Build Coastguard Worker EVT StoreTy = EVT::getIntegerVT(*DAG.getContext(), SizeInBits);
11312*9880d681SAndroid Build Coastguard Worker StoredVal = DAG.getConstant(StoreInt, DL, StoreTy);
11313*9880d681SAndroid Build Coastguard Worker }
11314*9880d681SAndroid Build Coastguard Worker
11315*9880d681SAndroid Build Coastguard Worker assert(!Chains.empty());
11316*9880d681SAndroid Build Coastguard Worker
11317*9880d681SAndroid Build Coastguard Worker SDValue NewChain = DAG.getNode(ISD::TokenFactor, DL, MVT::Other, Chains);
11318*9880d681SAndroid Build Coastguard Worker SDValue NewStore = DAG.getStore(NewChain, DL, StoredVal,
11319*9880d681SAndroid Build Coastguard Worker FirstInChain->getBasePtr(),
11320*9880d681SAndroid Build Coastguard Worker FirstInChain->getPointerInfo(),
11321*9880d681SAndroid Build Coastguard Worker false, false,
11322*9880d681SAndroid Build Coastguard Worker FirstInChain->getAlignment());
11323*9880d681SAndroid Build Coastguard Worker
11324*9880d681SAndroid Build Coastguard Worker bool UseAA = CombinerAA.getNumOccurrences() > 0 ? CombinerAA
11325*9880d681SAndroid Build Coastguard Worker : DAG.getSubtarget().useAA();
11326*9880d681SAndroid Build Coastguard Worker if (UseAA) {
11327*9880d681SAndroid Build Coastguard Worker // Replace all merged stores with the new store.
11328*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < NumStores; ++i)
11329*9880d681SAndroid Build Coastguard Worker CombineTo(StoreNodes[i].MemNode, NewStore);
11330*9880d681SAndroid Build Coastguard Worker } else {
11331*9880d681SAndroid Build Coastguard Worker // Replace the last store with the new store.
11332*9880d681SAndroid Build Coastguard Worker CombineTo(LatestOp, NewStore);
11333*9880d681SAndroid Build Coastguard Worker // Erase all other stores.
11334*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < NumStores; ++i) {
11335*9880d681SAndroid Build Coastguard Worker if (StoreNodes[i].MemNode == LatestOp)
11336*9880d681SAndroid Build Coastguard Worker continue;
11337*9880d681SAndroid Build Coastguard Worker StoreSDNode *St = cast<StoreSDNode>(StoreNodes[i].MemNode);
11338*9880d681SAndroid Build Coastguard Worker // ReplaceAllUsesWith will replace all uses that existed when it was
11339*9880d681SAndroid Build Coastguard Worker // called, but graph optimizations may cause new ones to appear. For
11340*9880d681SAndroid Build Coastguard Worker // example, the case in pr14333 looks like
11341*9880d681SAndroid Build Coastguard Worker //
11342*9880d681SAndroid Build Coastguard Worker // St's chain -> St -> another store -> X
11343*9880d681SAndroid Build Coastguard Worker //
11344*9880d681SAndroid Build Coastguard Worker // And the only difference from St to the other store is the chain.
11345*9880d681SAndroid Build Coastguard Worker // When we change it's chain to be St's chain they become identical,
11346*9880d681SAndroid Build Coastguard Worker // get CSEed and the net result is that X is now a use of St.
11347*9880d681SAndroid Build Coastguard Worker // Since we know that St is redundant, just iterate.
11348*9880d681SAndroid Build Coastguard Worker while (!St->use_empty())
11349*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesWith(SDValue(St, 0), St->getChain());
11350*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(St);
11351*9880d681SAndroid Build Coastguard Worker }
11352*9880d681SAndroid Build Coastguard Worker }
11353*9880d681SAndroid Build Coastguard Worker
11354*9880d681SAndroid Build Coastguard Worker return true;
11355*9880d681SAndroid Build Coastguard Worker }
11356*9880d681SAndroid Build Coastguard Worker
getStoreMergeAndAliasCandidates(StoreSDNode * St,SmallVectorImpl<MemOpLink> & StoreNodes,SmallVectorImpl<LSBaseSDNode * > & AliasLoadNodes)11357*9880d681SAndroid Build Coastguard Worker void DAGCombiner::getStoreMergeAndAliasCandidates(
11358*9880d681SAndroid Build Coastguard Worker StoreSDNode* St, SmallVectorImpl<MemOpLink> &StoreNodes,
11359*9880d681SAndroid Build Coastguard Worker SmallVectorImpl<LSBaseSDNode*> &AliasLoadNodes) {
11360*9880d681SAndroid Build Coastguard Worker // This holds the base pointer, index, and the offset in bytes from the base
11361*9880d681SAndroid Build Coastguard Worker // pointer.
11362*9880d681SAndroid Build Coastguard Worker BaseIndexOffset BasePtr = BaseIndexOffset::match(St->getBasePtr(), DAG);
11363*9880d681SAndroid Build Coastguard Worker
11364*9880d681SAndroid Build Coastguard Worker // We must have a base and an offset.
11365*9880d681SAndroid Build Coastguard Worker if (!BasePtr.Base.getNode())
11366*9880d681SAndroid Build Coastguard Worker return;
11367*9880d681SAndroid Build Coastguard Worker
11368*9880d681SAndroid Build Coastguard Worker // Do not handle stores to undef base pointers.
11369*9880d681SAndroid Build Coastguard Worker if (BasePtr.Base.isUndef())
11370*9880d681SAndroid Build Coastguard Worker return;
11371*9880d681SAndroid Build Coastguard Worker
11372*9880d681SAndroid Build Coastguard Worker // Walk up the chain and look for nodes with offsets from the same
11373*9880d681SAndroid Build Coastguard Worker // base pointer. Stop when reaching an instruction with a different kind
11374*9880d681SAndroid Build Coastguard Worker // or instruction which has a different base pointer.
11375*9880d681SAndroid Build Coastguard Worker EVT MemVT = St->getMemoryVT();
11376*9880d681SAndroid Build Coastguard Worker unsigned Seq = 0;
11377*9880d681SAndroid Build Coastguard Worker StoreSDNode *Index = St;
11378*9880d681SAndroid Build Coastguard Worker
11379*9880d681SAndroid Build Coastguard Worker
11380*9880d681SAndroid Build Coastguard Worker bool UseAA = CombinerAA.getNumOccurrences() > 0 ? CombinerAA
11381*9880d681SAndroid Build Coastguard Worker : DAG.getSubtarget().useAA();
11382*9880d681SAndroid Build Coastguard Worker
11383*9880d681SAndroid Build Coastguard Worker if (UseAA) {
11384*9880d681SAndroid Build Coastguard Worker // Look at other users of the same chain. Stores on the same chain do not
11385*9880d681SAndroid Build Coastguard Worker // alias. If combiner-aa is enabled, non-aliasing stores are canonicalized
11386*9880d681SAndroid Build Coastguard Worker // to be on the same chain, so don't bother looking at adjacent chains.
11387*9880d681SAndroid Build Coastguard Worker
11388*9880d681SAndroid Build Coastguard Worker SDValue Chain = St->getChain();
11389*9880d681SAndroid Build Coastguard Worker for (auto I = Chain->use_begin(), E = Chain->use_end(); I != E; ++I) {
11390*9880d681SAndroid Build Coastguard Worker if (StoreSDNode *OtherST = dyn_cast<StoreSDNode>(*I)) {
11391*9880d681SAndroid Build Coastguard Worker if (I.getOperandNo() != 0)
11392*9880d681SAndroid Build Coastguard Worker continue;
11393*9880d681SAndroid Build Coastguard Worker
11394*9880d681SAndroid Build Coastguard Worker if (OtherST->isVolatile() || OtherST->isIndexed())
11395*9880d681SAndroid Build Coastguard Worker continue;
11396*9880d681SAndroid Build Coastguard Worker
11397*9880d681SAndroid Build Coastguard Worker if (OtherST->getMemoryVT() != MemVT)
11398*9880d681SAndroid Build Coastguard Worker continue;
11399*9880d681SAndroid Build Coastguard Worker
11400*9880d681SAndroid Build Coastguard Worker BaseIndexOffset Ptr = BaseIndexOffset::match(OtherST->getBasePtr(), DAG);
11401*9880d681SAndroid Build Coastguard Worker
11402*9880d681SAndroid Build Coastguard Worker if (Ptr.equalBaseIndex(BasePtr))
11403*9880d681SAndroid Build Coastguard Worker StoreNodes.push_back(MemOpLink(OtherST, Ptr.Offset, Seq++));
11404*9880d681SAndroid Build Coastguard Worker }
11405*9880d681SAndroid Build Coastguard Worker }
11406*9880d681SAndroid Build Coastguard Worker
11407*9880d681SAndroid Build Coastguard Worker return;
11408*9880d681SAndroid Build Coastguard Worker }
11409*9880d681SAndroid Build Coastguard Worker
11410*9880d681SAndroid Build Coastguard Worker while (Index) {
11411*9880d681SAndroid Build Coastguard Worker // If the chain has more than one use, then we can't reorder the mem ops.
11412*9880d681SAndroid Build Coastguard Worker if (Index != St && !SDValue(Index, 0)->hasOneUse())
11413*9880d681SAndroid Build Coastguard Worker break;
11414*9880d681SAndroid Build Coastguard Worker
11415*9880d681SAndroid Build Coastguard Worker // Find the base pointer and offset for this memory node.
11416*9880d681SAndroid Build Coastguard Worker BaseIndexOffset Ptr = BaseIndexOffset::match(Index->getBasePtr(), DAG);
11417*9880d681SAndroid Build Coastguard Worker
11418*9880d681SAndroid Build Coastguard Worker // Check that the base pointer is the same as the original one.
11419*9880d681SAndroid Build Coastguard Worker if (!Ptr.equalBaseIndex(BasePtr))
11420*9880d681SAndroid Build Coastguard Worker break;
11421*9880d681SAndroid Build Coastguard Worker
11422*9880d681SAndroid Build Coastguard Worker // The memory operands must not be volatile.
11423*9880d681SAndroid Build Coastguard Worker if (Index->isVolatile() || Index->isIndexed())
11424*9880d681SAndroid Build Coastguard Worker break;
11425*9880d681SAndroid Build Coastguard Worker
11426*9880d681SAndroid Build Coastguard Worker // No truncation.
11427*9880d681SAndroid Build Coastguard Worker if (Index->isTruncatingStore())
11428*9880d681SAndroid Build Coastguard Worker break;
11429*9880d681SAndroid Build Coastguard Worker
11430*9880d681SAndroid Build Coastguard Worker // The stored memory type must be the same.
11431*9880d681SAndroid Build Coastguard Worker if (Index->getMemoryVT() != MemVT)
11432*9880d681SAndroid Build Coastguard Worker break;
11433*9880d681SAndroid Build Coastguard Worker
11434*9880d681SAndroid Build Coastguard Worker // We do not allow under-aligned stores in order to prevent
11435*9880d681SAndroid Build Coastguard Worker // overriding stores. NOTE: this is a bad hack. Alignment SHOULD
11436*9880d681SAndroid Build Coastguard Worker // be irrelevant here; what MATTERS is that we not move memory
11437*9880d681SAndroid Build Coastguard Worker // operations that potentially overlap past each-other.
11438*9880d681SAndroid Build Coastguard Worker if (Index->getAlignment() < MemVT.getStoreSize())
11439*9880d681SAndroid Build Coastguard Worker break;
11440*9880d681SAndroid Build Coastguard Worker
11441*9880d681SAndroid Build Coastguard Worker // We found a potential memory operand to merge.
11442*9880d681SAndroid Build Coastguard Worker StoreNodes.push_back(MemOpLink(Index, Ptr.Offset, Seq++));
11443*9880d681SAndroid Build Coastguard Worker
11444*9880d681SAndroid Build Coastguard Worker // Find the next memory operand in the chain. If the next operand in the
11445*9880d681SAndroid Build Coastguard Worker // chain is a store then move up and continue the scan with the next
11446*9880d681SAndroid Build Coastguard Worker // memory operand. If the next operand is a load save it and use alias
11447*9880d681SAndroid Build Coastguard Worker // information to check if it interferes with anything.
11448*9880d681SAndroid Build Coastguard Worker SDNode *NextInChain = Index->getChain().getNode();
11449*9880d681SAndroid Build Coastguard Worker while (1) {
11450*9880d681SAndroid Build Coastguard Worker if (StoreSDNode *STn = dyn_cast<StoreSDNode>(NextInChain)) {
11451*9880d681SAndroid Build Coastguard Worker // We found a store node. Use it for the next iteration.
11452*9880d681SAndroid Build Coastguard Worker Index = STn;
11453*9880d681SAndroid Build Coastguard Worker break;
11454*9880d681SAndroid Build Coastguard Worker } else if (LoadSDNode *Ldn = dyn_cast<LoadSDNode>(NextInChain)) {
11455*9880d681SAndroid Build Coastguard Worker if (Ldn->isVolatile()) {
11456*9880d681SAndroid Build Coastguard Worker Index = nullptr;
11457*9880d681SAndroid Build Coastguard Worker break;
11458*9880d681SAndroid Build Coastguard Worker }
11459*9880d681SAndroid Build Coastguard Worker
11460*9880d681SAndroid Build Coastguard Worker // Save the load node for later. Continue the scan.
11461*9880d681SAndroid Build Coastguard Worker AliasLoadNodes.push_back(Ldn);
11462*9880d681SAndroid Build Coastguard Worker NextInChain = Ldn->getChain().getNode();
11463*9880d681SAndroid Build Coastguard Worker continue;
11464*9880d681SAndroid Build Coastguard Worker } else {
11465*9880d681SAndroid Build Coastguard Worker Index = nullptr;
11466*9880d681SAndroid Build Coastguard Worker break;
11467*9880d681SAndroid Build Coastguard Worker }
11468*9880d681SAndroid Build Coastguard Worker }
11469*9880d681SAndroid Build Coastguard Worker }
11470*9880d681SAndroid Build Coastguard Worker }
11471*9880d681SAndroid Build Coastguard Worker
11472*9880d681SAndroid Build Coastguard Worker // We need to check that merging these stores does not cause a loop
11473*9880d681SAndroid Build Coastguard Worker // in the DAG. Any store candidate may depend on another candidate
11474*9880d681SAndroid Build Coastguard Worker // indirectly through its operand (we already consider dependencies
11475*9880d681SAndroid Build Coastguard Worker // through the chain). Check in parallel by searching up from
11476*9880d681SAndroid Build Coastguard Worker // non-chain operands of candidates.
checkMergeStoreCandidatesForDependencies(SmallVectorImpl<MemOpLink> & StoreNodes)11477*9880d681SAndroid Build Coastguard Worker bool DAGCombiner::checkMergeStoreCandidatesForDependencies(
11478*9880d681SAndroid Build Coastguard Worker SmallVectorImpl<MemOpLink> &StoreNodes) {
11479*9880d681SAndroid Build Coastguard Worker SmallPtrSet<const SDNode *, 16> Visited;
11480*9880d681SAndroid Build Coastguard Worker SmallVector<const SDNode *, 8> Worklist;
11481*9880d681SAndroid Build Coastguard Worker // search ops of store candidates
11482*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < StoreNodes.size(); ++i) {
11483*9880d681SAndroid Build Coastguard Worker SDNode *n = StoreNodes[i].MemNode;
11484*9880d681SAndroid Build Coastguard Worker // Potential loops may happen only through non-chain operands
11485*9880d681SAndroid Build Coastguard Worker for (unsigned j = 1; j < n->getNumOperands(); ++j)
11486*9880d681SAndroid Build Coastguard Worker Worklist.push_back(n->getOperand(j).getNode());
11487*9880d681SAndroid Build Coastguard Worker }
11488*9880d681SAndroid Build Coastguard Worker // search through DAG. We can stop early if we find a storenode
11489*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < StoreNodes.size(); ++i) {
11490*9880d681SAndroid Build Coastguard Worker if (SDNode::hasPredecessorHelper(StoreNodes[i].MemNode, Visited, Worklist))
11491*9880d681SAndroid Build Coastguard Worker return false;
11492*9880d681SAndroid Build Coastguard Worker }
11493*9880d681SAndroid Build Coastguard Worker return true;
11494*9880d681SAndroid Build Coastguard Worker }
11495*9880d681SAndroid Build Coastguard Worker
MergeConsecutiveStores(StoreSDNode * St)11496*9880d681SAndroid Build Coastguard Worker bool DAGCombiner::MergeConsecutiveStores(StoreSDNode* St) {
11497*9880d681SAndroid Build Coastguard Worker if (OptLevel == CodeGenOpt::None)
11498*9880d681SAndroid Build Coastguard Worker return false;
11499*9880d681SAndroid Build Coastguard Worker
11500*9880d681SAndroid Build Coastguard Worker EVT MemVT = St->getMemoryVT();
11501*9880d681SAndroid Build Coastguard Worker int64_t ElementSizeBytes = MemVT.getSizeInBits() / 8;
11502*9880d681SAndroid Build Coastguard Worker bool NoVectors = DAG.getMachineFunction().getFunction()->hasFnAttribute(
11503*9880d681SAndroid Build Coastguard Worker Attribute::NoImplicitFloat);
11504*9880d681SAndroid Build Coastguard Worker
11505*9880d681SAndroid Build Coastguard Worker // This function cannot currently deal with non-byte-sized memory sizes.
11506*9880d681SAndroid Build Coastguard Worker if (ElementSizeBytes * 8 != MemVT.getSizeInBits())
11507*9880d681SAndroid Build Coastguard Worker return false;
11508*9880d681SAndroid Build Coastguard Worker
11509*9880d681SAndroid Build Coastguard Worker if (!MemVT.isSimple())
11510*9880d681SAndroid Build Coastguard Worker return false;
11511*9880d681SAndroid Build Coastguard Worker
11512*9880d681SAndroid Build Coastguard Worker // Perform an early exit check. Do not bother looking at stored values that
11513*9880d681SAndroid Build Coastguard Worker // are not constants, loads, or extracted vector elements.
11514*9880d681SAndroid Build Coastguard Worker SDValue StoredVal = St->getValue();
11515*9880d681SAndroid Build Coastguard Worker bool IsLoadSrc = isa<LoadSDNode>(StoredVal);
11516*9880d681SAndroid Build Coastguard Worker bool IsConstantSrc = isa<ConstantSDNode>(StoredVal) ||
11517*9880d681SAndroid Build Coastguard Worker isa<ConstantFPSDNode>(StoredVal);
11518*9880d681SAndroid Build Coastguard Worker bool IsExtractVecSrc = (StoredVal.getOpcode() == ISD::EXTRACT_VECTOR_ELT ||
11519*9880d681SAndroid Build Coastguard Worker StoredVal.getOpcode() == ISD::EXTRACT_SUBVECTOR);
11520*9880d681SAndroid Build Coastguard Worker
11521*9880d681SAndroid Build Coastguard Worker if (!IsConstantSrc && !IsLoadSrc && !IsExtractVecSrc)
11522*9880d681SAndroid Build Coastguard Worker return false;
11523*9880d681SAndroid Build Coastguard Worker
11524*9880d681SAndroid Build Coastguard Worker // Don't merge vectors into wider vectors if the source data comes from loads.
11525*9880d681SAndroid Build Coastguard Worker // TODO: This restriction can be lifted by using logic similar to the
11526*9880d681SAndroid Build Coastguard Worker // ExtractVecSrc case.
11527*9880d681SAndroid Build Coastguard Worker if (MemVT.isVector() && IsLoadSrc)
11528*9880d681SAndroid Build Coastguard Worker return false;
11529*9880d681SAndroid Build Coastguard Worker
11530*9880d681SAndroid Build Coastguard Worker // Only look at ends of store sequences.
11531*9880d681SAndroid Build Coastguard Worker SDValue Chain = SDValue(St, 0);
11532*9880d681SAndroid Build Coastguard Worker if (Chain->hasOneUse() && Chain->use_begin()->getOpcode() == ISD::STORE)
11533*9880d681SAndroid Build Coastguard Worker return false;
11534*9880d681SAndroid Build Coastguard Worker
11535*9880d681SAndroid Build Coastguard Worker // Save the LoadSDNodes that we find in the chain.
11536*9880d681SAndroid Build Coastguard Worker // We need to make sure that these nodes do not interfere with
11537*9880d681SAndroid Build Coastguard Worker // any of the store nodes.
11538*9880d681SAndroid Build Coastguard Worker SmallVector<LSBaseSDNode*, 8> AliasLoadNodes;
11539*9880d681SAndroid Build Coastguard Worker
11540*9880d681SAndroid Build Coastguard Worker // Save the StoreSDNodes that we find in the chain.
11541*9880d681SAndroid Build Coastguard Worker SmallVector<MemOpLink, 8> StoreNodes;
11542*9880d681SAndroid Build Coastguard Worker
11543*9880d681SAndroid Build Coastguard Worker getStoreMergeAndAliasCandidates(St, StoreNodes, AliasLoadNodes);
11544*9880d681SAndroid Build Coastguard Worker
11545*9880d681SAndroid Build Coastguard Worker // Check if there is anything to merge.
11546*9880d681SAndroid Build Coastguard Worker if (StoreNodes.size() < 2)
11547*9880d681SAndroid Build Coastguard Worker return false;
11548*9880d681SAndroid Build Coastguard Worker
11549*9880d681SAndroid Build Coastguard Worker // only do dep endence check in AA case
11550*9880d681SAndroid Build Coastguard Worker bool UseAA = CombinerAA.getNumOccurrences() > 0 ? CombinerAA
11551*9880d681SAndroid Build Coastguard Worker : DAG.getSubtarget().useAA();
11552*9880d681SAndroid Build Coastguard Worker if (UseAA && !checkMergeStoreCandidatesForDependencies(StoreNodes))
11553*9880d681SAndroid Build Coastguard Worker return false;
11554*9880d681SAndroid Build Coastguard Worker
11555*9880d681SAndroid Build Coastguard Worker // Sort the memory operands according to their distance from the
11556*9880d681SAndroid Build Coastguard Worker // base pointer. As a secondary criteria: make sure stores coming
11557*9880d681SAndroid Build Coastguard Worker // later in the code come first in the list. This is important for
11558*9880d681SAndroid Build Coastguard Worker // the non-UseAA case, because we're merging stores into the FINAL
11559*9880d681SAndroid Build Coastguard Worker // store along a chain which potentially contains aliasing stores.
11560*9880d681SAndroid Build Coastguard Worker // Thus, if there are multiple stores to the same address, the last
11561*9880d681SAndroid Build Coastguard Worker // one can be considered for merging but not the others.
11562*9880d681SAndroid Build Coastguard Worker std::sort(StoreNodes.begin(), StoreNodes.end(),
11563*9880d681SAndroid Build Coastguard Worker [](MemOpLink LHS, MemOpLink RHS) {
11564*9880d681SAndroid Build Coastguard Worker return LHS.OffsetFromBase < RHS.OffsetFromBase ||
11565*9880d681SAndroid Build Coastguard Worker (LHS.OffsetFromBase == RHS.OffsetFromBase &&
11566*9880d681SAndroid Build Coastguard Worker LHS.SequenceNum < RHS.SequenceNum);
11567*9880d681SAndroid Build Coastguard Worker });
11568*9880d681SAndroid Build Coastguard Worker
11569*9880d681SAndroid Build Coastguard Worker // Scan the memory operations on the chain and find the first non-consecutive
11570*9880d681SAndroid Build Coastguard Worker // store memory address.
11571*9880d681SAndroid Build Coastguard Worker unsigned LastConsecutiveStore = 0;
11572*9880d681SAndroid Build Coastguard Worker int64_t StartAddress = StoreNodes[0].OffsetFromBase;
11573*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0, e = StoreNodes.size(); i < e; ++i) {
11574*9880d681SAndroid Build Coastguard Worker
11575*9880d681SAndroid Build Coastguard Worker // Check that the addresses are consecutive starting from the second
11576*9880d681SAndroid Build Coastguard Worker // element in the list of stores.
11577*9880d681SAndroid Build Coastguard Worker if (i > 0) {
11578*9880d681SAndroid Build Coastguard Worker int64_t CurrAddress = StoreNodes[i].OffsetFromBase;
11579*9880d681SAndroid Build Coastguard Worker if (CurrAddress - StartAddress != (ElementSizeBytes * i))
11580*9880d681SAndroid Build Coastguard Worker break;
11581*9880d681SAndroid Build Coastguard Worker }
11582*9880d681SAndroid Build Coastguard Worker
11583*9880d681SAndroid Build Coastguard Worker // Check if this store interferes with any of the loads that we found.
11584*9880d681SAndroid Build Coastguard Worker // If we find a load that alias with this store. Stop the sequence.
11585*9880d681SAndroid Build Coastguard Worker if (std::any_of(AliasLoadNodes.begin(), AliasLoadNodes.end(),
11586*9880d681SAndroid Build Coastguard Worker [&](LSBaseSDNode* Ldn) {
11587*9880d681SAndroid Build Coastguard Worker return isAlias(Ldn, StoreNodes[i].MemNode);
11588*9880d681SAndroid Build Coastguard Worker }))
11589*9880d681SAndroid Build Coastguard Worker break;
11590*9880d681SAndroid Build Coastguard Worker
11591*9880d681SAndroid Build Coastguard Worker // Mark this node as useful.
11592*9880d681SAndroid Build Coastguard Worker LastConsecutiveStore = i;
11593*9880d681SAndroid Build Coastguard Worker }
11594*9880d681SAndroid Build Coastguard Worker
11595*9880d681SAndroid Build Coastguard Worker // The node with the lowest store address.
11596*9880d681SAndroid Build Coastguard Worker LSBaseSDNode *FirstInChain = StoreNodes[0].MemNode;
11597*9880d681SAndroid Build Coastguard Worker unsigned FirstStoreAS = FirstInChain->getAddressSpace();
11598*9880d681SAndroid Build Coastguard Worker unsigned FirstStoreAlign = FirstInChain->getAlignment();
11599*9880d681SAndroid Build Coastguard Worker LLVMContext &Context = *DAG.getContext();
11600*9880d681SAndroid Build Coastguard Worker const DataLayout &DL = DAG.getDataLayout();
11601*9880d681SAndroid Build Coastguard Worker
11602*9880d681SAndroid Build Coastguard Worker // Store the constants into memory as one consecutive store.
11603*9880d681SAndroid Build Coastguard Worker if (IsConstantSrc) {
11604*9880d681SAndroid Build Coastguard Worker unsigned LastLegalType = 0;
11605*9880d681SAndroid Build Coastguard Worker unsigned LastLegalVectorType = 0;
11606*9880d681SAndroid Build Coastguard Worker bool NonZero = false;
11607*9880d681SAndroid Build Coastguard Worker for (unsigned i=0; i<LastConsecutiveStore+1; ++i) {
11608*9880d681SAndroid Build Coastguard Worker StoreSDNode *St = cast<StoreSDNode>(StoreNodes[i].MemNode);
11609*9880d681SAndroid Build Coastguard Worker SDValue StoredVal = St->getValue();
11610*9880d681SAndroid Build Coastguard Worker
11611*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(StoredVal)) {
11612*9880d681SAndroid Build Coastguard Worker NonZero |= !C->isNullValue();
11613*9880d681SAndroid Build Coastguard Worker } else if (ConstantFPSDNode *C = dyn_cast<ConstantFPSDNode>(StoredVal)) {
11614*9880d681SAndroid Build Coastguard Worker NonZero |= !C->getConstantFPValue()->isNullValue();
11615*9880d681SAndroid Build Coastguard Worker } else {
11616*9880d681SAndroid Build Coastguard Worker // Non-constant.
11617*9880d681SAndroid Build Coastguard Worker break;
11618*9880d681SAndroid Build Coastguard Worker }
11619*9880d681SAndroid Build Coastguard Worker
11620*9880d681SAndroid Build Coastguard Worker // Find a legal type for the constant store.
11621*9880d681SAndroid Build Coastguard Worker unsigned SizeInBits = (i+1) * ElementSizeBytes * 8;
11622*9880d681SAndroid Build Coastguard Worker EVT StoreTy = EVT::getIntegerVT(Context, SizeInBits);
11623*9880d681SAndroid Build Coastguard Worker bool IsFast;
11624*9880d681SAndroid Build Coastguard Worker if (TLI.isTypeLegal(StoreTy) &&
11625*9880d681SAndroid Build Coastguard Worker TLI.allowsMemoryAccess(Context, DL, StoreTy, FirstStoreAS,
11626*9880d681SAndroid Build Coastguard Worker FirstStoreAlign, &IsFast) && IsFast) {
11627*9880d681SAndroid Build Coastguard Worker LastLegalType = i+1;
11628*9880d681SAndroid Build Coastguard Worker // Or check whether a truncstore is legal.
11629*9880d681SAndroid Build Coastguard Worker } else if (TLI.getTypeAction(Context, StoreTy) ==
11630*9880d681SAndroid Build Coastguard Worker TargetLowering::TypePromoteInteger) {
11631*9880d681SAndroid Build Coastguard Worker EVT LegalizedStoredValueTy =
11632*9880d681SAndroid Build Coastguard Worker TLI.getTypeToTransformTo(Context, StoredVal.getValueType());
11633*9880d681SAndroid Build Coastguard Worker if (TLI.isTruncStoreLegal(LegalizedStoredValueTy, StoreTy) &&
11634*9880d681SAndroid Build Coastguard Worker TLI.allowsMemoryAccess(Context, DL, LegalizedStoredValueTy,
11635*9880d681SAndroid Build Coastguard Worker FirstStoreAS, FirstStoreAlign, &IsFast) &&
11636*9880d681SAndroid Build Coastguard Worker IsFast) {
11637*9880d681SAndroid Build Coastguard Worker LastLegalType = i + 1;
11638*9880d681SAndroid Build Coastguard Worker }
11639*9880d681SAndroid Build Coastguard Worker }
11640*9880d681SAndroid Build Coastguard Worker
11641*9880d681SAndroid Build Coastguard Worker // We only use vectors if the constant is known to be zero or the target
11642*9880d681SAndroid Build Coastguard Worker // allows it and the function is not marked with the noimplicitfloat
11643*9880d681SAndroid Build Coastguard Worker // attribute.
11644*9880d681SAndroid Build Coastguard Worker if ((!NonZero || TLI.storeOfVectorConstantIsCheap(MemVT, i+1,
11645*9880d681SAndroid Build Coastguard Worker FirstStoreAS)) &&
11646*9880d681SAndroid Build Coastguard Worker !NoVectors) {
11647*9880d681SAndroid Build Coastguard Worker // Find a legal type for the vector store.
11648*9880d681SAndroid Build Coastguard Worker EVT Ty = EVT::getVectorVT(Context, MemVT, i+1);
11649*9880d681SAndroid Build Coastguard Worker if (TLI.isTypeLegal(Ty) &&
11650*9880d681SAndroid Build Coastguard Worker TLI.allowsMemoryAccess(Context, DL, Ty, FirstStoreAS,
11651*9880d681SAndroid Build Coastguard Worker FirstStoreAlign, &IsFast) && IsFast)
11652*9880d681SAndroid Build Coastguard Worker LastLegalVectorType = i + 1;
11653*9880d681SAndroid Build Coastguard Worker }
11654*9880d681SAndroid Build Coastguard Worker }
11655*9880d681SAndroid Build Coastguard Worker
11656*9880d681SAndroid Build Coastguard Worker // Check if we found a legal integer type to store.
11657*9880d681SAndroid Build Coastguard Worker if (LastLegalType == 0 && LastLegalVectorType == 0)
11658*9880d681SAndroid Build Coastguard Worker return false;
11659*9880d681SAndroid Build Coastguard Worker
11660*9880d681SAndroid Build Coastguard Worker bool UseVector = (LastLegalVectorType > LastLegalType) && !NoVectors;
11661*9880d681SAndroid Build Coastguard Worker unsigned NumElem = UseVector ? LastLegalVectorType : LastLegalType;
11662*9880d681SAndroid Build Coastguard Worker
11663*9880d681SAndroid Build Coastguard Worker return MergeStoresOfConstantsOrVecElts(StoreNodes, MemVT, NumElem,
11664*9880d681SAndroid Build Coastguard Worker true, UseVector);
11665*9880d681SAndroid Build Coastguard Worker }
11666*9880d681SAndroid Build Coastguard Worker
11667*9880d681SAndroid Build Coastguard Worker // When extracting multiple vector elements, try to store them
11668*9880d681SAndroid Build Coastguard Worker // in one vector store rather than a sequence of scalar stores.
11669*9880d681SAndroid Build Coastguard Worker if (IsExtractVecSrc) {
11670*9880d681SAndroid Build Coastguard Worker unsigned NumStoresToMerge = 0;
11671*9880d681SAndroid Build Coastguard Worker bool IsVec = MemVT.isVector();
11672*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < LastConsecutiveStore + 1; ++i) {
11673*9880d681SAndroid Build Coastguard Worker StoreSDNode *St = cast<StoreSDNode>(StoreNodes[i].MemNode);
11674*9880d681SAndroid Build Coastguard Worker unsigned StoreValOpcode = St->getValue().getOpcode();
11675*9880d681SAndroid Build Coastguard Worker // This restriction could be loosened.
11676*9880d681SAndroid Build Coastguard Worker // Bail out if any stored values are not elements extracted from a vector.
11677*9880d681SAndroid Build Coastguard Worker // It should be possible to handle mixed sources, but load sources need
11678*9880d681SAndroid Build Coastguard Worker // more careful handling (see the block of code below that handles
11679*9880d681SAndroid Build Coastguard Worker // consecutive loads).
11680*9880d681SAndroid Build Coastguard Worker if (StoreValOpcode != ISD::EXTRACT_VECTOR_ELT &&
11681*9880d681SAndroid Build Coastguard Worker StoreValOpcode != ISD::EXTRACT_SUBVECTOR)
11682*9880d681SAndroid Build Coastguard Worker return false;
11683*9880d681SAndroid Build Coastguard Worker
11684*9880d681SAndroid Build Coastguard Worker // Find a legal type for the vector store.
11685*9880d681SAndroid Build Coastguard Worker unsigned Elts = i + 1;
11686*9880d681SAndroid Build Coastguard Worker if (IsVec) {
11687*9880d681SAndroid Build Coastguard Worker // When merging vector stores, get the total number of elements.
11688*9880d681SAndroid Build Coastguard Worker Elts *= MemVT.getVectorNumElements();
11689*9880d681SAndroid Build Coastguard Worker }
11690*9880d681SAndroid Build Coastguard Worker EVT Ty = EVT::getVectorVT(*DAG.getContext(), MemVT.getScalarType(), Elts);
11691*9880d681SAndroid Build Coastguard Worker bool IsFast;
11692*9880d681SAndroid Build Coastguard Worker if (TLI.isTypeLegal(Ty) &&
11693*9880d681SAndroid Build Coastguard Worker TLI.allowsMemoryAccess(Context, DL, Ty, FirstStoreAS,
11694*9880d681SAndroid Build Coastguard Worker FirstStoreAlign, &IsFast) && IsFast)
11695*9880d681SAndroid Build Coastguard Worker NumStoresToMerge = i + 1;
11696*9880d681SAndroid Build Coastguard Worker }
11697*9880d681SAndroid Build Coastguard Worker
11698*9880d681SAndroid Build Coastguard Worker return MergeStoresOfConstantsOrVecElts(StoreNodes, MemVT, NumStoresToMerge,
11699*9880d681SAndroid Build Coastguard Worker false, true);
11700*9880d681SAndroid Build Coastguard Worker }
11701*9880d681SAndroid Build Coastguard Worker
11702*9880d681SAndroid Build Coastguard Worker // Below we handle the case of multiple consecutive stores that
11703*9880d681SAndroid Build Coastguard Worker // come from multiple consecutive loads. We merge them into a single
11704*9880d681SAndroid Build Coastguard Worker // wide load and a single wide store.
11705*9880d681SAndroid Build Coastguard Worker
11706*9880d681SAndroid Build Coastguard Worker // Look for load nodes which are used by the stored values.
11707*9880d681SAndroid Build Coastguard Worker SmallVector<MemOpLink, 8> LoadNodes;
11708*9880d681SAndroid Build Coastguard Worker
11709*9880d681SAndroid Build Coastguard Worker // Find acceptable loads. Loads need to have the same chain (token factor),
11710*9880d681SAndroid Build Coastguard Worker // must not be zext, volatile, indexed, and they must be consecutive.
11711*9880d681SAndroid Build Coastguard Worker BaseIndexOffset LdBasePtr;
11712*9880d681SAndroid Build Coastguard Worker for (unsigned i=0; i<LastConsecutiveStore+1; ++i) {
11713*9880d681SAndroid Build Coastguard Worker StoreSDNode *St = cast<StoreSDNode>(StoreNodes[i].MemNode);
11714*9880d681SAndroid Build Coastguard Worker LoadSDNode *Ld = dyn_cast<LoadSDNode>(St->getValue());
11715*9880d681SAndroid Build Coastguard Worker if (!Ld) break;
11716*9880d681SAndroid Build Coastguard Worker
11717*9880d681SAndroid Build Coastguard Worker // Loads must only have one use.
11718*9880d681SAndroid Build Coastguard Worker if (!Ld->hasNUsesOfValue(1, 0))
11719*9880d681SAndroid Build Coastguard Worker break;
11720*9880d681SAndroid Build Coastguard Worker
11721*9880d681SAndroid Build Coastguard Worker // The memory operands must not be volatile.
11722*9880d681SAndroid Build Coastguard Worker if (Ld->isVolatile() || Ld->isIndexed())
11723*9880d681SAndroid Build Coastguard Worker break;
11724*9880d681SAndroid Build Coastguard Worker
11725*9880d681SAndroid Build Coastguard Worker // We do not accept ext loads.
11726*9880d681SAndroid Build Coastguard Worker if (Ld->getExtensionType() != ISD::NON_EXTLOAD)
11727*9880d681SAndroid Build Coastguard Worker break;
11728*9880d681SAndroid Build Coastguard Worker
11729*9880d681SAndroid Build Coastguard Worker // The stored memory type must be the same.
11730*9880d681SAndroid Build Coastguard Worker if (Ld->getMemoryVT() != MemVT)
11731*9880d681SAndroid Build Coastguard Worker break;
11732*9880d681SAndroid Build Coastguard Worker
11733*9880d681SAndroid Build Coastguard Worker BaseIndexOffset LdPtr = BaseIndexOffset::match(Ld->getBasePtr(), DAG);
11734*9880d681SAndroid Build Coastguard Worker // If this is not the first ptr that we check.
11735*9880d681SAndroid Build Coastguard Worker if (LdBasePtr.Base.getNode()) {
11736*9880d681SAndroid Build Coastguard Worker // The base ptr must be the same.
11737*9880d681SAndroid Build Coastguard Worker if (!LdPtr.equalBaseIndex(LdBasePtr))
11738*9880d681SAndroid Build Coastguard Worker break;
11739*9880d681SAndroid Build Coastguard Worker } else {
11740*9880d681SAndroid Build Coastguard Worker // Check that all other base pointers are the same as this one.
11741*9880d681SAndroid Build Coastguard Worker LdBasePtr = LdPtr;
11742*9880d681SAndroid Build Coastguard Worker }
11743*9880d681SAndroid Build Coastguard Worker
11744*9880d681SAndroid Build Coastguard Worker // We found a potential memory operand to merge.
11745*9880d681SAndroid Build Coastguard Worker LoadNodes.push_back(MemOpLink(Ld, LdPtr.Offset, 0));
11746*9880d681SAndroid Build Coastguard Worker }
11747*9880d681SAndroid Build Coastguard Worker
11748*9880d681SAndroid Build Coastguard Worker if (LoadNodes.size() < 2)
11749*9880d681SAndroid Build Coastguard Worker return false;
11750*9880d681SAndroid Build Coastguard Worker
11751*9880d681SAndroid Build Coastguard Worker // If we have load/store pair instructions and we only have two values,
11752*9880d681SAndroid Build Coastguard Worker // don't bother.
11753*9880d681SAndroid Build Coastguard Worker unsigned RequiredAlignment;
11754*9880d681SAndroid Build Coastguard Worker if (LoadNodes.size() == 2 && TLI.hasPairedLoad(MemVT, RequiredAlignment) &&
11755*9880d681SAndroid Build Coastguard Worker St->getAlignment() >= RequiredAlignment)
11756*9880d681SAndroid Build Coastguard Worker return false;
11757*9880d681SAndroid Build Coastguard Worker
11758*9880d681SAndroid Build Coastguard Worker LoadSDNode *FirstLoad = cast<LoadSDNode>(LoadNodes[0].MemNode);
11759*9880d681SAndroid Build Coastguard Worker unsigned FirstLoadAS = FirstLoad->getAddressSpace();
11760*9880d681SAndroid Build Coastguard Worker unsigned FirstLoadAlign = FirstLoad->getAlignment();
11761*9880d681SAndroid Build Coastguard Worker
11762*9880d681SAndroid Build Coastguard Worker // Scan the memory operations on the chain and find the first non-consecutive
11763*9880d681SAndroid Build Coastguard Worker // load memory address. These variables hold the index in the store node
11764*9880d681SAndroid Build Coastguard Worker // array.
11765*9880d681SAndroid Build Coastguard Worker unsigned LastConsecutiveLoad = 0;
11766*9880d681SAndroid Build Coastguard Worker // This variable refers to the size and not index in the array.
11767*9880d681SAndroid Build Coastguard Worker unsigned LastLegalVectorType = 0;
11768*9880d681SAndroid Build Coastguard Worker unsigned LastLegalIntegerType = 0;
11769*9880d681SAndroid Build Coastguard Worker StartAddress = LoadNodes[0].OffsetFromBase;
11770*9880d681SAndroid Build Coastguard Worker SDValue FirstChain = FirstLoad->getChain();
11771*9880d681SAndroid Build Coastguard Worker for (unsigned i = 1; i < LoadNodes.size(); ++i) {
11772*9880d681SAndroid Build Coastguard Worker // All loads must share the same chain.
11773*9880d681SAndroid Build Coastguard Worker if (LoadNodes[i].MemNode->getChain() != FirstChain)
11774*9880d681SAndroid Build Coastguard Worker break;
11775*9880d681SAndroid Build Coastguard Worker
11776*9880d681SAndroid Build Coastguard Worker int64_t CurrAddress = LoadNodes[i].OffsetFromBase;
11777*9880d681SAndroid Build Coastguard Worker if (CurrAddress - StartAddress != (ElementSizeBytes * i))
11778*9880d681SAndroid Build Coastguard Worker break;
11779*9880d681SAndroid Build Coastguard Worker LastConsecutiveLoad = i;
11780*9880d681SAndroid Build Coastguard Worker // Find a legal type for the vector store.
11781*9880d681SAndroid Build Coastguard Worker EVT StoreTy = EVT::getVectorVT(Context, MemVT, i+1);
11782*9880d681SAndroid Build Coastguard Worker bool IsFastSt, IsFastLd;
11783*9880d681SAndroid Build Coastguard Worker if (TLI.isTypeLegal(StoreTy) &&
11784*9880d681SAndroid Build Coastguard Worker TLI.allowsMemoryAccess(Context, DL, StoreTy, FirstStoreAS,
11785*9880d681SAndroid Build Coastguard Worker FirstStoreAlign, &IsFastSt) && IsFastSt &&
11786*9880d681SAndroid Build Coastguard Worker TLI.allowsMemoryAccess(Context, DL, StoreTy, FirstLoadAS,
11787*9880d681SAndroid Build Coastguard Worker FirstLoadAlign, &IsFastLd) && IsFastLd) {
11788*9880d681SAndroid Build Coastguard Worker LastLegalVectorType = i + 1;
11789*9880d681SAndroid Build Coastguard Worker }
11790*9880d681SAndroid Build Coastguard Worker
11791*9880d681SAndroid Build Coastguard Worker // Find a legal type for the integer store.
11792*9880d681SAndroid Build Coastguard Worker unsigned SizeInBits = (i+1) * ElementSizeBytes * 8;
11793*9880d681SAndroid Build Coastguard Worker StoreTy = EVT::getIntegerVT(Context, SizeInBits);
11794*9880d681SAndroid Build Coastguard Worker if (TLI.isTypeLegal(StoreTy) &&
11795*9880d681SAndroid Build Coastguard Worker TLI.allowsMemoryAccess(Context, DL, StoreTy, FirstStoreAS,
11796*9880d681SAndroid Build Coastguard Worker FirstStoreAlign, &IsFastSt) && IsFastSt &&
11797*9880d681SAndroid Build Coastguard Worker TLI.allowsMemoryAccess(Context, DL, StoreTy, FirstLoadAS,
11798*9880d681SAndroid Build Coastguard Worker FirstLoadAlign, &IsFastLd) && IsFastLd)
11799*9880d681SAndroid Build Coastguard Worker LastLegalIntegerType = i + 1;
11800*9880d681SAndroid Build Coastguard Worker // Or check whether a truncstore and extload is legal.
11801*9880d681SAndroid Build Coastguard Worker else if (TLI.getTypeAction(Context, StoreTy) ==
11802*9880d681SAndroid Build Coastguard Worker TargetLowering::TypePromoteInteger) {
11803*9880d681SAndroid Build Coastguard Worker EVT LegalizedStoredValueTy =
11804*9880d681SAndroid Build Coastguard Worker TLI.getTypeToTransformTo(Context, StoreTy);
11805*9880d681SAndroid Build Coastguard Worker if (TLI.isTruncStoreLegal(LegalizedStoredValueTy, StoreTy) &&
11806*9880d681SAndroid Build Coastguard Worker TLI.isLoadExtLegal(ISD::ZEXTLOAD, LegalizedStoredValueTy, StoreTy) &&
11807*9880d681SAndroid Build Coastguard Worker TLI.isLoadExtLegal(ISD::SEXTLOAD, LegalizedStoredValueTy, StoreTy) &&
11808*9880d681SAndroid Build Coastguard Worker TLI.isLoadExtLegal(ISD::EXTLOAD, LegalizedStoredValueTy, StoreTy) &&
11809*9880d681SAndroid Build Coastguard Worker TLI.allowsMemoryAccess(Context, DL, LegalizedStoredValueTy,
11810*9880d681SAndroid Build Coastguard Worker FirstStoreAS, FirstStoreAlign, &IsFastSt) &&
11811*9880d681SAndroid Build Coastguard Worker IsFastSt &&
11812*9880d681SAndroid Build Coastguard Worker TLI.allowsMemoryAccess(Context, DL, LegalizedStoredValueTy,
11813*9880d681SAndroid Build Coastguard Worker FirstLoadAS, FirstLoadAlign, &IsFastLd) &&
11814*9880d681SAndroid Build Coastguard Worker IsFastLd)
11815*9880d681SAndroid Build Coastguard Worker LastLegalIntegerType = i+1;
11816*9880d681SAndroid Build Coastguard Worker }
11817*9880d681SAndroid Build Coastguard Worker }
11818*9880d681SAndroid Build Coastguard Worker
11819*9880d681SAndroid Build Coastguard Worker // Only use vector types if the vector type is larger than the integer type.
11820*9880d681SAndroid Build Coastguard Worker // If they are the same, use integers.
11821*9880d681SAndroid Build Coastguard Worker bool UseVectorTy = LastLegalVectorType > LastLegalIntegerType && !NoVectors;
11822*9880d681SAndroid Build Coastguard Worker unsigned LastLegalType = std::max(LastLegalVectorType, LastLegalIntegerType);
11823*9880d681SAndroid Build Coastguard Worker
11824*9880d681SAndroid Build Coastguard Worker // We add +1 here because the LastXXX variables refer to location while
11825*9880d681SAndroid Build Coastguard Worker // the NumElem refers to array/index size.
11826*9880d681SAndroid Build Coastguard Worker unsigned NumElem = std::min(LastConsecutiveStore, LastConsecutiveLoad) + 1;
11827*9880d681SAndroid Build Coastguard Worker NumElem = std::min(LastLegalType, NumElem);
11828*9880d681SAndroid Build Coastguard Worker
11829*9880d681SAndroid Build Coastguard Worker if (NumElem < 2)
11830*9880d681SAndroid Build Coastguard Worker return false;
11831*9880d681SAndroid Build Coastguard Worker
11832*9880d681SAndroid Build Coastguard Worker // Collect the chains from all merged stores.
11833*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> MergeStoreChains;
11834*9880d681SAndroid Build Coastguard Worker MergeStoreChains.push_back(StoreNodes[0].MemNode->getChain());
11835*9880d681SAndroid Build Coastguard Worker
11836*9880d681SAndroid Build Coastguard Worker // The latest Node in the DAG.
11837*9880d681SAndroid Build Coastguard Worker unsigned LatestNodeUsed = 0;
11838*9880d681SAndroid Build Coastguard Worker for (unsigned i=1; i<NumElem; ++i) {
11839*9880d681SAndroid Build Coastguard Worker // Find a chain for the new wide-store operand. Notice that some
11840*9880d681SAndroid Build Coastguard Worker // of the store nodes that we found may not be selected for inclusion
11841*9880d681SAndroid Build Coastguard Worker // in the wide store. The chain we use needs to be the chain of the
11842*9880d681SAndroid Build Coastguard Worker // latest store node which is *used* and replaced by the wide store.
11843*9880d681SAndroid Build Coastguard Worker if (StoreNodes[i].SequenceNum < StoreNodes[LatestNodeUsed].SequenceNum)
11844*9880d681SAndroid Build Coastguard Worker LatestNodeUsed = i;
11845*9880d681SAndroid Build Coastguard Worker
11846*9880d681SAndroid Build Coastguard Worker MergeStoreChains.push_back(StoreNodes[i].MemNode->getChain());
11847*9880d681SAndroid Build Coastguard Worker }
11848*9880d681SAndroid Build Coastguard Worker
11849*9880d681SAndroid Build Coastguard Worker LSBaseSDNode *LatestOp = StoreNodes[LatestNodeUsed].MemNode;
11850*9880d681SAndroid Build Coastguard Worker
11851*9880d681SAndroid Build Coastguard Worker // Find if it is better to use vectors or integers to load and store
11852*9880d681SAndroid Build Coastguard Worker // to memory.
11853*9880d681SAndroid Build Coastguard Worker EVT JointMemOpVT;
11854*9880d681SAndroid Build Coastguard Worker if (UseVectorTy) {
11855*9880d681SAndroid Build Coastguard Worker JointMemOpVT = EVT::getVectorVT(Context, MemVT, NumElem);
11856*9880d681SAndroid Build Coastguard Worker } else {
11857*9880d681SAndroid Build Coastguard Worker unsigned SizeInBits = NumElem * ElementSizeBytes * 8;
11858*9880d681SAndroid Build Coastguard Worker JointMemOpVT = EVT::getIntegerVT(Context, SizeInBits);
11859*9880d681SAndroid Build Coastguard Worker }
11860*9880d681SAndroid Build Coastguard Worker
11861*9880d681SAndroid Build Coastguard Worker SDLoc LoadDL(LoadNodes[0].MemNode);
11862*9880d681SAndroid Build Coastguard Worker SDLoc StoreDL(StoreNodes[0].MemNode);
11863*9880d681SAndroid Build Coastguard Worker
11864*9880d681SAndroid Build Coastguard Worker // The merged loads are required to have the same incoming chain, so
11865*9880d681SAndroid Build Coastguard Worker // using the first's chain is acceptable.
11866*9880d681SAndroid Build Coastguard Worker SDValue NewLoad = DAG.getLoad(
11867*9880d681SAndroid Build Coastguard Worker JointMemOpVT, LoadDL, FirstLoad->getChain(), FirstLoad->getBasePtr(),
11868*9880d681SAndroid Build Coastguard Worker FirstLoad->getPointerInfo(), false, false, false, FirstLoadAlign);
11869*9880d681SAndroid Build Coastguard Worker
11870*9880d681SAndroid Build Coastguard Worker SDValue NewStoreChain =
11871*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::TokenFactor, StoreDL, MVT::Other, MergeStoreChains);
11872*9880d681SAndroid Build Coastguard Worker
11873*9880d681SAndroid Build Coastguard Worker SDValue NewStore = DAG.getStore(
11874*9880d681SAndroid Build Coastguard Worker NewStoreChain, StoreDL, NewLoad, FirstInChain->getBasePtr(),
11875*9880d681SAndroid Build Coastguard Worker FirstInChain->getPointerInfo(), false, false, FirstStoreAlign);
11876*9880d681SAndroid Build Coastguard Worker
11877*9880d681SAndroid Build Coastguard Worker // Transfer chain users from old loads to the new load.
11878*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < NumElem; ++i) {
11879*9880d681SAndroid Build Coastguard Worker LoadSDNode *Ld = cast<LoadSDNode>(LoadNodes[i].MemNode);
11880*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(Ld, 1),
11881*9880d681SAndroid Build Coastguard Worker SDValue(NewLoad.getNode(), 1));
11882*9880d681SAndroid Build Coastguard Worker }
11883*9880d681SAndroid Build Coastguard Worker
11884*9880d681SAndroid Build Coastguard Worker if (UseAA) {
11885*9880d681SAndroid Build Coastguard Worker // Replace the all stores with the new store.
11886*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < NumElem; ++i)
11887*9880d681SAndroid Build Coastguard Worker CombineTo(StoreNodes[i].MemNode, NewStore);
11888*9880d681SAndroid Build Coastguard Worker } else {
11889*9880d681SAndroid Build Coastguard Worker // Replace the last store with the new store.
11890*9880d681SAndroid Build Coastguard Worker CombineTo(LatestOp, NewStore);
11891*9880d681SAndroid Build Coastguard Worker // Erase all other stores.
11892*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < NumElem; ++i) {
11893*9880d681SAndroid Build Coastguard Worker // Remove all Store nodes.
11894*9880d681SAndroid Build Coastguard Worker if (StoreNodes[i].MemNode == LatestOp)
11895*9880d681SAndroid Build Coastguard Worker continue;
11896*9880d681SAndroid Build Coastguard Worker StoreSDNode *St = cast<StoreSDNode>(StoreNodes[i].MemNode);
11897*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValueWith(SDValue(St, 0), St->getChain());
11898*9880d681SAndroid Build Coastguard Worker deleteAndRecombine(St);
11899*9880d681SAndroid Build Coastguard Worker }
11900*9880d681SAndroid Build Coastguard Worker }
11901*9880d681SAndroid Build Coastguard Worker
11902*9880d681SAndroid Build Coastguard Worker return true;
11903*9880d681SAndroid Build Coastguard Worker }
11904*9880d681SAndroid Build Coastguard Worker
replaceStoreChain(StoreSDNode * ST,SDValue BetterChain)11905*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::replaceStoreChain(StoreSDNode *ST, SDValue BetterChain) {
11906*9880d681SAndroid Build Coastguard Worker SDLoc SL(ST);
11907*9880d681SAndroid Build Coastguard Worker SDValue ReplStore;
11908*9880d681SAndroid Build Coastguard Worker
11909*9880d681SAndroid Build Coastguard Worker // Replace the chain to avoid dependency.
11910*9880d681SAndroid Build Coastguard Worker if (ST->isTruncatingStore()) {
11911*9880d681SAndroid Build Coastguard Worker ReplStore = DAG.getTruncStore(BetterChain, SL, ST->getValue(),
11912*9880d681SAndroid Build Coastguard Worker ST->getBasePtr(), ST->getMemoryVT(),
11913*9880d681SAndroid Build Coastguard Worker ST->getMemOperand());
11914*9880d681SAndroid Build Coastguard Worker } else {
11915*9880d681SAndroid Build Coastguard Worker ReplStore = DAG.getStore(BetterChain, SL, ST->getValue(), ST->getBasePtr(),
11916*9880d681SAndroid Build Coastguard Worker ST->getMemOperand());
11917*9880d681SAndroid Build Coastguard Worker }
11918*9880d681SAndroid Build Coastguard Worker
11919*9880d681SAndroid Build Coastguard Worker // Create token to keep both nodes around.
11920*9880d681SAndroid Build Coastguard Worker SDValue Token = DAG.getNode(ISD::TokenFactor, SL,
11921*9880d681SAndroid Build Coastguard Worker MVT::Other, ST->getChain(), ReplStore);
11922*9880d681SAndroid Build Coastguard Worker
11923*9880d681SAndroid Build Coastguard Worker // Make sure the new and old chains are cleaned up.
11924*9880d681SAndroid Build Coastguard Worker AddToWorklist(Token.getNode());
11925*9880d681SAndroid Build Coastguard Worker
11926*9880d681SAndroid Build Coastguard Worker // Don't add users to work list.
11927*9880d681SAndroid Build Coastguard Worker return CombineTo(ST, Token, false);
11928*9880d681SAndroid Build Coastguard Worker }
11929*9880d681SAndroid Build Coastguard Worker
replaceStoreOfFPConstant(StoreSDNode * ST)11930*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::replaceStoreOfFPConstant(StoreSDNode *ST) {
11931*9880d681SAndroid Build Coastguard Worker SDValue Value = ST->getValue();
11932*9880d681SAndroid Build Coastguard Worker if (Value.getOpcode() == ISD::TargetConstantFP)
11933*9880d681SAndroid Build Coastguard Worker return SDValue();
11934*9880d681SAndroid Build Coastguard Worker
11935*9880d681SAndroid Build Coastguard Worker SDLoc DL(ST);
11936*9880d681SAndroid Build Coastguard Worker
11937*9880d681SAndroid Build Coastguard Worker SDValue Chain = ST->getChain();
11938*9880d681SAndroid Build Coastguard Worker SDValue Ptr = ST->getBasePtr();
11939*9880d681SAndroid Build Coastguard Worker
11940*9880d681SAndroid Build Coastguard Worker const ConstantFPSDNode *CFP = cast<ConstantFPSDNode>(Value);
11941*9880d681SAndroid Build Coastguard Worker
11942*9880d681SAndroid Build Coastguard Worker // NOTE: If the original store is volatile, this transform must not increase
11943*9880d681SAndroid Build Coastguard Worker // the number of stores. For example, on x86-32 an f64 can be stored in one
11944*9880d681SAndroid Build Coastguard Worker // processor operation but an i64 (which is not legal) requires two. So the
11945*9880d681SAndroid Build Coastguard Worker // transform should not be done in this case.
11946*9880d681SAndroid Build Coastguard Worker
11947*9880d681SAndroid Build Coastguard Worker SDValue Tmp;
11948*9880d681SAndroid Build Coastguard Worker switch (CFP->getSimpleValueType(0).SimpleTy) {
11949*9880d681SAndroid Build Coastguard Worker default:
11950*9880d681SAndroid Build Coastguard Worker llvm_unreachable("Unknown FP type");
11951*9880d681SAndroid Build Coastguard Worker case MVT::f16: // We don't do this for these yet.
11952*9880d681SAndroid Build Coastguard Worker case MVT::f80:
11953*9880d681SAndroid Build Coastguard Worker case MVT::f128:
11954*9880d681SAndroid Build Coastguard Worker case MVT::ppcf128:
11955*9880d681SAndroid Build Coastguard Worker return SDValue();
11956*9880d681SAndroid Build Coastguard Worker case MVT::f32:
11957*9880d681SAndroid Build Coastguard Worker if ((isTypeLegal(MVT::i32) && !LegalOperations && !ST->isVolatile()) ||
11958*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegalOrCustom(ISD::STORE, MVT::i32)) {
11959*9880d681SAndroid Build Coastguard Worker ;
11960*9880d681SAndroid Build Coastguard Worker Tmp = DAG.getConstant((uint32_t)CFP->getValueAPF().
11961*9880d681SAndroid Build Coastguard Worker bitcastToAPInt().getZExtValue(), SDLoc(CFP),
11962*9880d681SAndroid Build Coastguard Worker MVT::i32);
11963*9880d681SAndroid Build Coastguard Worker return DAG.getStore(Chain, DL, Tmp, Ptr, ST->getMemOperand());
11964*9880d681SAndroid Build Coastguard Worker }
11965*9880d681SAndroid Build Coastguard Worker
11966*9880d681SAndroid Build Coastguard Worker return SDValue();
11967*9880d681SAndroid Build Coastguard Worker case MVT::f64:
11968*9880d681SAndroid Build Coastguard Worker if ((TLI.isTypeLegal(MVT::i64) && !LegalOperations &&
11969*9880d681SAndroid Build Coastguard Worker !ST->isVolatile()) ||
11970*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegalOrCustom(ISD::STORE, MVT::i64)) {
11971*9880d681SAndroid Build Coastguard Worker ;
11972*9880d681SAndroid Build Coastguard Worker Tmp = DAG.getConstant(CFP->getValueAPF().bitcastToAPInt().
11973*9880d681SAndroid Build Coastguard Worker getZExtValue(), SDLoc(CFP), MVT::i64);
11974*9880d681SAndroid Build Coastguard Worker return DAG.getStore(Chain, DL, Tmp,
11975*9880d681SAndroid Build Coastguard Worker Ptr, ST->getMemOperand());
11976*9880d681SAndroid Build Coastguard Worker }
11977*9880d681SAndroid Build Coastguard Worker
11978*9880d681SAndroid Build Coastguard Worker if (!ST->isVolatile() &&
11979*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegalOrCustom(ISD::STORE, MVT::i32)) {
11980*9880d681SAndroid Build Coastguard Worker // Many FP stores are not made apparent until after legalize, e.g. for
11981*9880d681SAndroid Build Coastguard Worker // argument passing. Since this is so common, custom legalize the
11982*9880d681SAndroid Build Coastguard Worker // 64-bit integer store into two 32-bit stores.
11983*9880d681SAndroid Build Coastguard Worker uint64_t Val = CFP->getValueAPF().bitcastToAPInt().getZExtValue();
11984*9880d681SAndroid Build Coastguard Worker SDValue Lo = DAG.getConstant(Val & 0xFFFFFFFF, SDLoc(CFP), MVT::i32);
11985*9880d681SAndroid Build Coastguard Worker SDValue Hi = DAG.getConstant(Val >> 32, SDLoc(CFP), MVT::i32);
11986*9880d681SAndroid Build Coastguard Worker if (DAG.getDataLayout().isBigEndian())
11987*9880d681SAndroid Build Coastguard Worker std::swap(Lo, Hi);
11988*9880d681SAndroid Build Coastguard Worker
11989*9880d681SAndroid Build Coastguard Worker unsigned Alignment = ST->getAlignment();
11990*9880d681SAndroid Build Coastguard Worker bool isVolatile = ST->isVolatile();
11991*9880d681SAndroid Build Coastguard Worker bool isNonTemporal = ST->isNonTemporal();
11992*9880d681SAndroid Build Coastguard Worker AAMDNodes AAInfo = ST->getAAInfo();
11993*9880d681SAndroid Build Coastguard Worker
11994*9880d681SAndroid Build Coastguard Worker SDValue St0 = DAG.getStore(Chain, DL, Lo,
11995*9880d681SAndroid Build Coastguard Worker Ptr, ST->getPointerInfo(),
11996*9880d681SAndroid Build Coastguard Worker isVolatile, isNonTemporal,
11997*9880d681SAndroid Build Coastguard Worker ST->getAlignment(), AAInfo);
11998*9880d681SAndroid Build Coastguard Worker Ptr = DAG.getNode(ISD::ADD, DL, Ptr.getValueType(), Ptr,
11999*9880d681SAndroid Build Coastguard Worker DAG.getConstant(4, DL, Ptr.getValueType()));
12000*9880d681SAndroid Build Coastguard Worker Alignment = MinAlign(Alignment, 4U);
12001*9880d681SAndroid Build Coastguard Worker SDValue St1 = DAG.getStore(Chain, DL, Hi,
12002*9880d681SAndroid Build Coastguard Worker Ptr, ST->getPointerInfo().getWithOffset(4),
12003*9880d681SAndroid Build Coastguard Worker isVolatile, isNonTemporal,
12004*9880d681SAndroid Build Coastguard Worker Alignment, AAInfo);
12005*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TokenFactor, DL, MVT::Other,
12006*9880d681SAndroid Build Coastguard Worker St0, St1);
12007*9880d681SAndroid Build Coastguard Worker }
12008*9880d681SAndroid Build Coastguard Worker
12009*9880d681SAndroid Build Coastguard Worker return SDValue();
12010*9880d681SAndroid Build Coastguard Worker }
12011*9880d681SAndroid Build Coastguard Worker }
12012*9880d681SAndroid Build Coastguard Worker
visitSTORE(SDNode * N)12013*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSTORE(SDNode *N) {
12014*9880d681SAndroid Build Coastguard Worker StoreSDNode *ST = cast<StoreSDNode>(N);
12015*9880d681SAndroid Build Coastguard Worker SDValue Chain = ST->getChain();
12016*9880d681SAndroid Build Coastguard Worker SDValue Value = ST->getValue();
12017*9880d681SAndroid Build Coastguard Worker SDValue Ptr = ST->getBasePtr();
12018*9880d681SAndroid Build Coastguard Worker
12019*9880d681SAndroid Build Coastguard Worker // If this is a store of a bit convert, store the input value if the
12020*9880d681SAndroid Build Coastguard Worker // resultant store does not need a higher alignment than the original.
12021*9880d681SAndroid Build Coastguard Worker if (Value.getOpcode() == ISD::BITCAST && !ST->isTruncatingStore() &&
12022*9880d681SAndroid Build Coastguard Worker ST->isUnindexed()) {
12023*9880d681SAndroid Build Coastguard Worker EVT SVT = Value.getOperand(0).getValueType();
12024*9880d681SAndroid Build Coastguard Worker if (((!LegalOperations && !ST->isVolatile()) ||
12025*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegalOrCustom(ISD::STORE, SVT)) &&
12026*9880d681SAndroid Build Coastguard Worker TLI.isStoreBitCastBeneficial(Value.getValueType(), SVT)) {
12027*9880d681SAndroid Build Coastguard Worker unsigned OrigAlign = ST->getAlignment();
12028*9880d681SAndroid Build Coastguard Worker bool Fast = false;
12029*9880d681SAndroid Build Coastguard Worker if (TLI.allowsMemoryAccess(*DAG.getContext(), DAG.getDataLayout(), SVT,
12030*9880d681SAndroid Build Coastguard Worker ST->getAddressSpace(), OrigAlign, &Fast) &&
12031*9880d681SAndroid Build Coastguard Worker Fast) {
12032*9880d681SAndroid Build Coastguard Worker return DAG.getStore(Chain, SDLoc(N), Value.getOperand(0),
12033*9880d681SAndroid Build Coastguard Worker Ptr, ST->getPointerInfo(), ST->isVolatile(),
12034*9880d681SAndroid Build Coastguard Worker ST->isNonTemporal(), OrigAlign,
12035*9880d681SAndroid Build Coastguard Worker ST->getAAInfo());
12036*9880d681SAndroid Build Coastguard Worker }
12037*9880d681SAndroid Build Coastguard Worker }
12038*9880d681SAndroid Build Coastguard Worker }
12039*9880d681SAndroid Build Coastguard Worker
12040*9880d681SAndroid Build Coastguard Worker // Turn 'store undef, Ptr' -> nothing.
12041*9880d681SAndroid Build Coastguard Worker if (Value.isUndef() && ST->isUnindexed())
12042*9880d681SAndroid Build Coastguard Worker return Chain;
12043*9880d681SAndroid Build Coastguard Worker
12044*9880d681SAndroid Build Coastguard Worker // Try to infer better alignment information than the store already has.
12045*9880d681SAndroid Build Coastguard Worker if (OptLevel != CodeGenOpt::None && ST->isUnindexed()) {
12046*9880d681SAndroid Build Coastguard Worker if (unsigned Align = DAG.InferPtrAlignment(Ptr)) {
12047*9880d681SAndroid Build Coastguard Worker if (Align > ST->getAlignment()) {
12048*9880d681SAndroid Build Coastguard Worker SDValue NewStore =
12049*9880d681SAndroid Build Coastguard Worker DAG.getTruncStore(Chain, SDLoc(N), Value,
12050*9880d681SAndroid Build Coastguard Worker Ptr, ST->getPointerInfo(), ST->getMemoryVT(),
12051*9880d681SAndroid Build Coastguard Worker ST->isVolatile(), ST->isNonTemporal(), Align,
12052*9880d681SAndroid Build Coastguard Worker ST->getAAInfo());
12053*9880d681SAndroid Build Coastguard Worker if (NewStore.getNode() != N)
12054*9880d681SAndroid Build Coastguard Worker return CombineTo(ST, NewStore, true);
12055*9880d681SAndroid Build Coastguard Worker }
12056*9880d681SAndroid Build Coastguard Worker }
12057*9880d681SAndroid Build Coastguard Worker }
12058*9880d681SAndroid Build Coastguard Worker
12059*9880d681SAndroid Build Coastguard Worker // Try transforming a pair floating point load / store ops to integer
12060*9880d681SAndroid Build Coastguard Worker // load / store ops.
12061*9880d681SAndroid Build Coastguard Worker if (SDValue NewST = TransformFPLoadStorePair(N))
12062*9880d681SAndroid Build Coastguard Worker return NewST;
12063*9880d681SAndroid Build Coastguard Worker
12064*9880d681SAndroid Build Coastguard Worker bool UseAA = CombinerAA.getNumOccurrences() > 0 ? CombinerAA
12065*9880d681SAndroid Build Coastguard Worker : DAG.getSubtarget().useAA();
12066*9880d681SAndroid Build Coastguard Worker #ifndef NDEBUG
12067*9880d681SAndroid Build Coastguard Worker if (CombinerAAOnlyFunc.getNumOccurrences() &&
12068*9880d681SAndroid Build Coastguard Worker CombinerAAOnlyFunc != DAG.getMachineFunction().getName())
12069*9880d681SAndroid Build Coastguard Worker UseAA = false;
12070*9880d681SAndroid Build Coastguard Worker #endif
12071*9880d681SAndroid Build Coastguard Worker if (UseAA && ST->isUnindexed()) {
12072*9880d681SAndroid Build Coastguard Worker // FIXME: We should do this even without AA enabled. AA will just allow
12073*9880d681SAndroid Build Coastguard Worker // FindBetterChain to work in more situations. The problem with this is that
12074*9880d681SAndroid Build Coastguard Worker // any combine that expects memory operations to be on consecutive chains
12075*9880d681SAndroid Build Coastguard Worker // first needs to be updated to look for users of the same chain.
12076*9880d681SAndroid Build Coastguard Worker
12077*9880d681SAndroid Build Coastguard Worker // Walk up chain skipping non-aliasing memory nodes, on this store and any
12078*9880d681SAndroid Build Coastguard Worker // adjacent stores.
12079*9880d681SAndroid Build Coastguard Worker if (findBetterNeighborChains(ST)) {
12080*9880d681SAndroid Build Coastguard Worker // replaceStoreChain uses CombineTo, which handled all of the worklist
12081*9880d681SAndroid Build Coastguard Worker // manipulation. Return the original node to not do anything else.
12082*9880d681SAndroid Build Coastguard Worker return SDValue(ST, 0);
12083*9880d681SAndroid Build Coastguard Worker }
12084*9880d681SAndroid Build Coastguard Worker Chain = ST->getChain();
12085*9880d681SAndroid Build Coastguard Worker }
12086*9880d681SAndroid Build Coastguard Worker
12087*9880d681SAndroid Build Coastguard Worker // Try transforming N to an indexed store.
12088*9880d681SAndroid Build Coastguard Worker if (CombineToPreIndexedLoadStore(N) || CombineToPostIndexedLoadStore(N))
12089*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0);
12090*9880d681SAndroid Build Coastguard Worker
12091*9880d681SAndroid Build Coastguard Worker // FIXME: is there such a thing as a truncating indexed store?
12092*9880d681SAndroid Build Coastguard Worker if (ST->isTruncatingStore() && ST->isUnindexed() &&
12093*9880d681SAndroid Build Coastguard Worker Value.getValueType().isInteger()) {
12094*9880d681SAndroid Build Coastguard Worker // See if we can simplify the input to this truncstore with knowledge that
12095*9880d681SAndroid Build Coastguard Worker // only the low bits are being used. For example:
12096*9880d681SAndroid Build Coastguard Worker // "truncstore (or (shl x, 8), y), i8" -> "truncstore y, i8"
12097*9880d681SAndroid Build Coastguard Worker SDValue Shorter =
12098*9880d681SAndroid Build Coastguard Worker GetDemandedBits(Value,
12099*9880d681SAndroid Build Coastguard Worker APInt::getLowBitsSet(
12100*9880d681SAndroid Build Coastguard Worker Value.getValueType().getScalarType().getSizeInBits(),
12101*9880d681SAndroid Build Coastguard Worker ST->getMemoryVT().getScalarType().getSizeInBits()));
12102*9880d681SAndroid Build Coastguard Worker AddToWorklist(Value.getNode());
12103*9880d681SAndroid Build Coastguard Worker if (Shorter.getNode())
12104*9880d681SAndroid Build Coastguard Worker return DAG.getTruncStore(Chain, SDLoc(N), Shorter,
12105*9880d681SAndroid Build Coastguard Worker Ptr, ST->getMemoryVT(), ST->getMemOperand());
12106*9880d681SAndroid Build Coastguard Worker
12107*9880d681SAndroid Build Coastguard Worker // Otherwise, see if we can simplify the operation with
12108*9880d681SAndroid Build Coastguard Worker // SimplifyDemandedBits, which only works if the value has a single use.
12109*9880d681SAndroid Build Coastguard Worker if (SimplifyDemandedBits(Value,
12110*9880d681SAndroid Build Coastguard Worker APInt::getLowBitsSet(
12111*9880d681SAndroid Build Coastguard Worker Value.getValueType().getScalarType().getSizeInBits(),
12112*9880d681SAndroid Build Coastguard Worker ST->getMemoryVT().getScalarType().getSizeInBits())))
12113*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0);
12114*9880d681SAndroid Build Coastguard Worker }
12115*9880d681SAndroid Build Coastguard Worker
12116*9880d681SAndroid Build Coastguard Worker // If this is a load followed by a store to the same location, then the store
12117*9880d681SAndroid Build Coastguard Worker // is dead/noop.
12118*9880d681SAndroid Build Coastguard Worker if (LoadSDNode *Ld = dyn_cast<LoadSDNode>(Value)) {
12119*9880d681SAndroid Build Coastguard Worker if (Ld->getBasePtr() == Ptr && ST->getMemoryVT() == Ld->getMemoryVT() &&
12120*9880d681SAndroid Build Coastguard Worker ST->isUnindexed() && !ST->isVolatile() &&
12121*9880d681SAndroid Build Coastguard Worker // There can't be any side effects between the load and store, such as
12122*9880d681SAndroid Build Coastguard Worker // a call or store.
12123*9880d681SAndroid Build Coastguard Worker Chain.reachesChainWithoutSideEffects(SDValue(Ld, 1))) {
12124*9880d681SAndroid Build Coastguard Worker // The store is dead, remove it.
12125*9880d681SAndroid Build Coastguard Worker return Chain;
12126*9880d681SAndroid Build Coastguard Worker }
12127*9880d681SAndroid Build Coastguard Worker }
12128*9880d681SAndroid Build Coastguard Worker
12129*9880d681SAndroid Build Coastguard Worker // If this is a store followed by a store with the same value to the same
12130*9880d681SAndroid Build Coastguard Worker // location, then the store is dead/noop.
12131*9880d681SAndroid Build Coastguard Worker if (StoreSDNode *ST1 = dyn_cast<StoreSDNode>(Chain)) {
12132*9880d681SAndroid Build Coastguard Worker if (ST1->getBasePtr() == Ptr && ST->getMemoryVT() == ST1->getMemoryVT() &&
12133*9880d681SAndroid Build Coastguard Worker ST1->getValue() == Value && ST->isUnindexed() && !ST->isVolatile() &&
12134*9880d681SAndroid Build Coastguard Worker ST1->isUnindexed() && !ST1->isVolatile()) {
12135*9880d681SAndroid Build Coastguard Worker // The store is dead, remove it.
12136*9880d681SAndroid Build Coastguard Worker return Chain;
12137*9880d681SAndroid Build Coastguard Worker }
12138*9880d681SAndroid Build Coastguard Worker }
12139*9880d681SAndroid Build Coastguard Worker
12140*9880d681SAndroid Build Coastguard Worker // If this is an FP_ROUND or TRUNC followed by a store, fold this into a
12141*9880d681SAndroid Build Coastguard Worker // truncating store. We can do this even if this is already a truncstore.
12142*9880d681SAndroid Build Coastguard Worker if ((Value.getOpcode() == ISD::FP_ROUND || Value.getOpcode() == ISD::TRUNCATE)
12143*9880d681SAndroid Build Coastguard Worker && Value.getNode()->hasOneUse() && ST->isUnindexed() &&
12144*9880d681SAndroid Build Coastguard Worker TLI.isTruncStoreLegal(Value.getOperand(0).getValueType(),
12145*9880d681SAndroid Build Coastguard Worker ST->getMemoryVT())) {
12146*9880d681SAndroid Build Coastguard Worker return DAG.getTruncStore(Chain, SDLoc(N), Value.getOperand(0),
12147*9880d681SAndroid Build Coastguard Worker Ptr, ST->getMemoryVT(), ST->getMemOperand());
12148*9880d681SAndroid Build Coastguard Worker }
12149*9880d681SAndroid Build Coastguard Worker
12150*9880d681SAndroid Build Coastguard Worker // Only perform this optimization before the types are legal, because we
12151*9880d681SAndroid Build Coastguard Worker // don't want to perform this optimization on every DAGCombine invocation.
12152*9880d681SAndroid Build Coastguard Worker if (!LegalTypes) {
12153*9880d681SAndroid Build Coastguard Worker bool EverChanged = false;
12154*9880d681SAndroid Build Coastguard Worker
12155*9880d681SAndroid Build Coastguard Worker do {
12156*9880d681SAndroid Build Coastguard Worker // There can be multiple store sequences on the same chain.
12157*9880d681SAndroid Build Coastguard Worker // Keep trying to merge store sequences until we are unable to do so
12158*9880d681SAndroid Build Coastguard Worker // or until we merge the last store on the chain.
12159*9880d681SAndroid Build Coastguard Worker bool Changed = MergeConsecutiveStores(ST);
12160*9880d681SAndroid Build Coastguard Worker EverChanged |= Changed;
12161*9880d681SAndroid Build Coastguard Worker if (!Changed) break;
12162*9880d681SAndroid Build Coastguard Worker } while (ST->getOpcode() != ISD::DELETED_NODE);
12163*9880d681SAndroid Build Coastguard Worker
12164*9880d681SAndroid Build Coastguard Worker if (EverChanged)
12165*9880d681SAndroid Build Coastguard Worker return SDValue(N, 0);
12166*9880d681SAndroid Build Coastguard Worker }
12167*9880d681SAndroid Build Coastguard Worker
12168*9880d681SAndroid Build Coastguard Worker // Turn 'store float 1.0, Ptr' -> 'store int 0x12345678, Ptr'
12169*9880d681SAndroid Build Coastguard Worker //
12170*9880d681SAndroid Build Coastguard Worker // Make sure to do this only after attempting to merge stores in order to
12171*9880d681SAndroid Build Coastguard Worker // avoid changing the types of some subset of stores due to visit order,
12172*9880d681SAndroid Build Coastguard Worker // preventing their merging.
12173*9880d681SAndroid Build Coastguard Worker if (isa<ConstantFPSDNode>(Value)) {
12174*9880d681SAndroid Build Coastguard Worker if (SDValue NewSt = replaceStoreOfFPConstant(ST))
12175*9880d681SAndroid Build Coastguard Worker return NewSt;
12176*9880d681SAndroid Build Coastguard Worker }
12177*9880d681SAndroid Build Coastguard Worker
12178*9880d681SAndroid Build Coastguard Worker return ReduceLoadOpStoreWidth(N);
12179*9880d681SAndroid Build Coastguard Worker }
12180*9880d681SAndroid Build Coastguard Worker
visitINSERT_VECTOR_ELT(SDNode * N)12181*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitINSERT_VECTOR_ELT(SDNode *N) {
12182*9880d681SAndroid Build Coastguard Worker SDValue InVec = N->getOperand(0);
12183*9880d681SAndroid Build Coastguard Worker SDValue InVal = N->getOperand(1);
12184*9880d681SAndroid Build Coastguard Worker SDValue EltNo = N->getOperand(2);
12185*9880d681SAndroid Build Coastguard Worker SDLoc dl(N);
12186*9880d681SAndroid Build Coastguard Worker
12187*9880d681SAndroid Build Coastguard Worker // If the inserted element is an UNDEF, just use the input vector.
12188*9880d681SAndroid Build Coastguard Worker if (InVal.isUndef())
12189*9880d681SAndroid Build Coastguard Worker return InVec;
12190*9880d681SAndroid Build Coastguard Worker
12191*9880d681SAndroid Build Coastguard Worker EVT VT = InVec.getValueType();
12192*9880d681SAndroid Build Coastguard Worker
12193*9880d681SAndroid Build Coastguard Worker // If we can't generate a legal BUILD_VECTOR, exit
12194*9880d681SAndroid Build Coastguard Worker if (LegalOperations && !TLI.isOperationLegal(ISD::BUILD_VECTOR, VT))
12195*9880d681SAndroid Build Coastguard Worker return SDValue();
12196*9880d681SAndroid Build Coastguard Worker
12197*9880d681SAndroid Build Coastguard Worker // Check that we know which element is being inserted
12198*9880d681SAndroid Build Coastguard Worker if (!isa<ConstantSDNode>(EltNo))
12199*9880d681SAndroid Build Coastguard Worker return SDValue();
12200*9880d681SAndroid Build Coastguard Worker unsigned Elt = cast<ConstantSDNode>(EltNo)->getZExtValue();
12201*9880d681SAndroid Build Coastguard Worker
12202*9880d681SAndroid Build Coastguard Worker // Canonicalize insert_vector_elt dag nodes.
12203*9880d681SAndroid Build Coastguard Worker // Example:
12204*9880d681SAndroid Build Coastguard Worker // (insert_vector_elt (insert_vector_elt A, Idx0), Idx1)
12205*9880d681SAndroid Build Coastguard Worker // -> (insert_vector_elt (insert_vector_elt A, Idx1), Idx0)
12206*9880d681SAndroid Build Coastguard Worker //
12207*9880d681SAndroid Build Coastguard Worker // Do this only if the child insert_vector node has one use; also
12208*9880d681SAndroid Build Coastguard Worker // do this only if indices are both constants and Idx1 < Idx0.
12209*9880d681SAndroid Build Coastguard Worker if (InVec.getOpcode() == ISD::INSERT_VECTOR_ELT && InVec.hasOneUse()
12210*9880d681SAndroid Build Coastguard Worker && isa<ConstantSDNode>(InVec.getOperand(2))) {
12211*9880d681SAndroid Build Coastguard Worker unsigned OtherElt =
12212*9880d681SAndroid Build Coastguard Worker cast<ConstantSDNode>(InVec.getOperand(2))->getZExtValue();
12213*9880d681SAndroid Build Coastguard Worker if (Elt < OtherElt) {
12214*9880d681SAndroid Build Coastguard Worker // Swap nodes.
12215*9880d681SAndroid Build Coastguard Worker SDValue NewOp = DAG.getNode(ISD::INSERT_VECTOR_ELT, SDLoc(N), VT,
12216*9880d681SAndroid Build Coastguard Worker InVec.getOperand(0), InVal, EltNo);
12217*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewOp.getNode());
12218*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::INSERT_VECTOR_ELT, SDLoc(InVec.getNode()),
12219*9880d681SAndroid Build Coastguard Worker VT, NewOp, InVec.getOperand(1), InVec.getOperand(2));
12220*9880d681SAndroid Build Coastguard Worker }
12221*9880d681SAndroid Build Coastguard Worker }
12222*9880d681SAndroid Build Coastguard Worker
12223*9880d681SAndroid Build Coastguard Worker // Check that the operand is a BUILD_VECTOR (or UNDEF, which can essentially
12224*9880d681SAndroid Build Coastguard Worker // be converted to a BUILD_VECTOR). Fill in the Ops vector with the
12225*9880d681SAndroid Build Coastguard Worker // vector elements.
12226*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Ops;
12227*9880d681SAndroid Build Coastguard Worker // Do not combine these two vectors if the output vector will not replace
12228*9880d681SAndroid Build Coastguard Worker // the input vector.
12229*9880d681SAndroid Build Coastguard Worker if (InVec.getOpcode() == ISD::BUILD_VECTOR && InVec.hasOneUse()) {
12230*9880d681SAndroid Build Coastguard Worker Ops.append(InVec.getNode()->op_begin(),
12231*9880d681SAndroid Build Coastguard Worker InVec.getNode()->op_end());
12232*9880d681SAndroid Build Coastguard Worker } else if (InVec.isUndef()) {
12233*9880d681SAndroid Build Coastguard Worker unsigned NElts = VT.getVectorNumElements();
12234*9880d681SAndroid Build Coastguard Worker Ops.append(NElts, DAG.getUNDEF(InVal.getValueType()));
12235*9880d681SAndroid Build Coastguard Worker } else {
12236*9880d681SAndroid Build Coastguard Worker return SDValue();
12237*9880d681SAndroid Build Coastguard Worker }
12238*9880d681SAndroid Build Coastguard Worker
12239*9880d681SAndroid Build Coastguard Worker // Insert the element
12240*9880d681SAndroid Build Coastguard Worker if (Elt < Ops.size()) {
12241*9880d681SAndroid Build Coastguard Worker // All the operands of BUILD_VECTOR must have the same type;
12242*9880d681SAndroid Build Coastguard Worker // we enforce that here.
12243*9880d681SAndroid Build Coastguard Worker EVT OpVT = Ops[0].getValueType();
12244*9880d681SAndroid Build Coastguard Worker if (InVal.getValueType() != OpVT)
12245*9880d681SAndroid Build Coastguard Worker InVal = OpVT.bitsGT(InVal.getValueType()) ?
12246*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::ANY_EXTEND, dl, OpVT, InVal) :
12247*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::TRUNCATE, dl, OpVT, InVal);
12248*9880d681SAndroid Build Coastguard Worker Ops[Elt] = InVal;
12249*9880d681SAndroid Build Coastguard Worker }
12250*9880d681SAndroid Build Coastguard Worker
12251*9880d681SAndroid Build Coastguard Worker // Return the new vector
12252*9880d681SAndroid Build Coastguard Worker return DAG.getBuildVector(VT, dl, Ops);
12253*9880d681SAndroid Build Coastguard Worker }
12254*9880d681SAndroid Build Coastguard Worker
ReplaceExtractVectorEltOfLoadWithNarrowedLoad(SDNode * EVE,EVT InVecVT,SDValue EltNo,LoadSDNode * OriginalLoad)12255*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::ReplaceExtractVectorEltOfLoadWithNarrowedLoad(
12256*9880d681SAndroid Build Coastguard Worker SDNode *EVE, EVT InVecVT, SDValue EltNo, LoadSDNode *OriginalLoad) {
12257*9880d681SAndroid Build Coastguard Worker assert(!OriginalLoad->isVolatile());
12258*9880d681SAndroid Build Coastguard Worker
12259*9880d681SAndroid Build Coastguard Worker EVT ResultVT = EVE->getValueType(0);
12260*9880d681SAndroid Build Coastguard Worker EVT VecEltVT = InVecVT.getVectorElementType();
12261*9880d681SAndroid Build Coastguard Worker unsigned Align = OriginalLoad->getAlignment();
12262*9880d681SAndroid Build Coastguard Worker unsigned NewAlign = DAG.getDataLayout().getABITypeAlignment(
12263*9880d681SAndroid Build Coastguard Worker VecEltVT.getTypeForEVT(*DAG.getContext()));
12264*9880d681SAndroid Build Coastguard Worker
12265*9880d681SAndroid Build Coastguard Worker if (NewAlign > Align || !TLI.isOperationLegalOrCustom(ISD::LOAD, VecEltVT))
12266*9880d681SAndroid Build Coastguard Worker return SDValue();
12267*9880d681SAndroid Build Coastguard Worker
12268*9880d681SAndroid Build Coastguard Worker Align = NewAlign;
12269*9880d681SAndroid Build Coastguard Worker
12270*9880d681SAndroid Build Coastguard Worker SDValue NewPtr = OriginalLoad->getBasePtr();
12271*9880d681SAndroid Build Coastguard Worker SDValue Offset;
12272*9880d681SAndroid Build Coastguard Worker EVT PtrType = NewPtr.getValueType();
12273*9880d681SAndroid Build Coastguard Worker MachinePointerInfo MPI;
12274*9880d681SAndroid Build Coastguard Worker SDLoc DL(EVE);
12275*9880d681SAndroid Build Coastguard Worker if (auto *ConstEltNo = dyn_cast<ConstantSDNode>(EltNo)) {
12276*9880d681SAndroid Build Coastguard Worker int Elt = ConstEltNo->getZExtValue();
12277*9880d681SAndroid Build Coastguard Worker unsigned PtrOff = VecEltVT.getSizeInBits() * Elt / 8;
12278*9880d681SAndroid Build Coastguard Worker Offset = DAG.getConstant(PtrOff, DL, PtrType);
12279*9880d681SAndroid Build Coastguard Worker MPI = OriginalLoad->getPointerInfo().getWithOffset(PtrOff);
12280*9880d681SAndroid Build Coastguard Worker } else {
12281*9880d681SAndroid Build Coastguard Worker Offset = DAG.getZExtOrTrunc(EltNo, DL, PtrType);
12282*9880d681SAndroid Build Coastguard Worker Offset = DAG.getNode(
12283*9880d681SAndroid Build Coastguard Worker ISD::MUL, DL, PtrType, Offset,
12284*9880d681SAndroid Build Coastguard Worker DAG.getConstant(VecEltVT.getStoreSize(), DL, PtrType));
12285*9880d681SAndroid Build Coastguard Worker MPI = OriginalLoad->getPointerInfo();
12286*9880d681SAndroid Build Coastguard Worker }
12287*9880d681SAndroid Build Coastguard Worker NewPtr = DAG.getNode(ISD::ADD, DL, PtrType, NewPtr, Offset);
12288*9880d681SAndroid Build Coastguard Worker
12289*9880d681SAndroid Build Coastguard Worker // The replacement we need to do here is a little tricky: we need to
12290*9880d681SAndroid Build Coastguard Worker // replace an extractelement of a load with a load.
12291*9880d681SAndroid Build Coastguard Worker // Use ReplaceAllUsesOfValuesWith to do the replacement.
12292*9880d681SAndroid Build Coastguard Worker // Note that this replacement assumes that the extractvalue is the only
12293*9880d681SAndroid Build Coastguard Worker // use of the load; that's okay because we don't want to perform this
12294*9880d681SAndroid Build Coastguard Worker // transformation in other cases anyway.
12295*9880d681SAndroid Build Coastguard Worker SDValue Load;
12296*9880d681SAndroid Build Coastguard Worker SDValue Chain;
12297*9880d681SAndroid Build Coastguard Worker if (ResultVT.bitsGT(VecEltVT)) {
12298*9880d681SAndroid Build Coastguard Worker // If the result type of vextract is wider than the load, then issue an
12299*9880d681SAndroid Build Coastguard Worker // extending load instead.
12300*9880d681SAndroid Build Coastguard Worker ISD::LoadExtType ExtType = TLI.isLoadExtLegal(ISD::ZEXTLOAD, ResultVT,
12301*9880d681SAndroid Build Coastguard Worker VecEltVT)
12302*9880d681SAndroid Build Coastguard Worker ? ISD::ZEXTLOAD
12303*9880d681SAndroid Build Coastguard Worker : ISD::EXTLOAD;
12304*9880d681SAndroid Build Coastguard Worker Load = DAG.getExtLoad(
12305*9880d681SAndroid Build Coastguard Worker ExtType, SDLoc(EVE), ResultVT, OriginalLoad->getChain(), NewPtr, MPI,
12306*9880d681SAndroid Build Coastguard Worker VecEltVT, OriginalLoad->isVolatile(), OriginalLoad->isNonTemporal(),
12307*9880d681SAndroid Build Coastguard Worker OriginalLoad->isInvariant(), Align, OriginalLoad->getAAInfo());
12308*9880d681SAndroid Build Coastguard Worker Chain = Load.getValue(1);
12309*9880d681SAndroid Build Coastguard Worker } else {
12310*9880d681SAndroid Build Coastguard Worker Load = DAG.getLoad(
12311*9880d681SAndroid Build Coastguard Worker VecEltVT, SDLoc(EVE), OriginalLoad->getChain(), NewPtr, MPI,
12312*9880d681SAndroid Build Coastguard Worker OriginalLoad->isVolatile(), OriginalLoad->isNonTemporal(),
12313*9880d681SAndroid Build Coastguard Worker OriginalLoad->isInvariant(), Align, OriginalLoad->getAAInfo());
12314*9880d681SAndroid Build Coastguard Worker Chain = Load.getValue(1);
12315*9880d681SAndroid Build Coastguard Worker if (ResultVT.bitsLT(VecEltVT))
12316*9880d681SAndroid Build Coastguard Worker Load = DAG.getNode(ISD::TRUNCATE, SDLoc(EVE), ResultVT, Load);
12317*9880d681SAndroid Build Coastguard Worker else
12318*9880d681SAndroid Build Coastguard Worker Load = DAG.getBitcast(ResultVT, Load);
12319*9880d681SAndroid Build Coastguard Worker }
12320*9880d681SAndroid Build Coastguard Worker WorklistRemover DeadNodes(*this);
12321*9880d681SAndroid Build Coastguard Worker SDValue From[] = { SDValue(EVE, 0), SDValue(OriginalLoad, 1) };
12322*9880d681SAndroid Build Coastguard Worker SDValue To[] = { Load, Chain };
12323*9880d681SAndroid Build Coastguard Worker DAG.ReplaceAllUsesOfValuesWith(From, To, 2);
12324*9880d681SAndroid Build Coastguard Worker // Since we're explicitly calling ReplaceAllUses, add the new node to the
12325*9880d681SAndroid Build Coastguard Worker // worklist explicitly as well.
12326*9880d681SAndroid Build Coastguard Worker AddToWorklist(Load.getNode());
12327*9880d681SAndroid Build Coastguard Worker AddUsersToWorklist(Load.getNode()); // Add users too
12328*9880d681SAndroid Build Coastguard Worker // Make sure to revisit this node to clean it up; it will usually be dead.
12329*9880d681SAndroid Build Coastguard Worker AddToWorklist(EVE);
12330*9880d681SAndroid Build Coastguard Worker ++OpsNarrowed;
12331*9880d681SAndroid Build Coastguard Worker return SDValue(EVE, 0);
12332*9880d681SAndroid Build Coastguard Worker }
12333*9880d681SAndroid Build Coastguard Worker
visitEXTRACT_VECTOR_ELT(SDNode * N)12334*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitEXTRACT_VECTOR_ELT(SDNode *N) {
12335*9880d681SAndroid Build Coastguard Worker // (vextract (scalar_to_vector val, 0) -> val
12336*9880d681SAndroid Build Coastguard Worker SDValue InVec = N->getOperand(0);
12337*9880d681SAndroid Build Coastguard Worker EVT VT = InVec.getValueType();
12338*9880d681SAndroid Build Coastguard Worker EVT NVT = N->getValueType(0);
12339*9880d681SAndroid Build Coastguard Worker
12340*9880d681SAndroid Build Coastguard Worker if (InVec.getOpcode() == ISD::SCALAR_TO_VECTOR) {
12341*9880d681SAndroid Build Coastguard Worker // Check if the result type doesn't match the inserted element type. A
12342*9880d681SAndroid Build Coastguard Worker // SCALAR_TO_VECTOR may truncate the inserted element and the
12343*9880d681SAndroid Build Coastguard Worker // EXTRACT_VECTOR_ELT may widen the extracted vector.
12344*9880d681SAndroid Build Coastguard Worker SDValue InOp = InVec.getOperand(0);
12345*9880d681SAndroid Build Coastguard Worker if (InOp.getValueType() != NVT) {
12346*9880d681SAndroid Build Coastguard Worker assert(InOp.getValueType().isInteger() && NVT.isInteger());
12347*9880d681SAndroid Build Coastguard Worker return DAG.getSExtOrTrunc(InOp, SDLoc(InVec), NVT);
12348*9880d681SAndroid Build Coastguard Worker }
12349*9880d681SAndroid Build Coastguard Worker return InOp;
12350*9880d681SAndroid Build Coastguard Worker }
12351*9880d681SAndroid Build Coastguard Worker
12352*9880d681SAndroid Build Coastguard Worker SDValue EltNo = N->getOperand(1);
12353*9880d681SAndroid Build Coastguard Worker ConstantSDNode *ConstEltNo = dyn_cast<ConstantSDNode>(EltNo);
12354*9880d681SAndroid Build Coastguard Worker
12355*9880d681SAndroid Build Coastguard Worker // extract_vector_elt (build_vector x, y), 1 -> y
12356*9880d681SAndroid Build Coastguard Worker if (ConstEltNo &&
12357*9880d681SAndroid Build Coastguard Worker InVec.getOpcode() == ISD::BUILD_VECTOR &&
12358*9880d681SAndroid Build Coastguard Worker TLI.isTypeLegal(VT) &&
12359*9880d681SAndroid Build Coastguard Worker (InVec.hasOneUse() ||
12360*9880d681SAndroid Build Coastguard Worker TLI.aggressivelyPreferBuildVectorSources(VT))) {
12361*9880d681SAndroid Build Coastguard Worker SDValue Elt = InVec.getOperand(ConstEltNo->getZExtValue());
12362*9880d681SAndroid Build Coastguard Worker EVT InEltVT = Elt.getValueType();
12363*9880d681SAndroid Build Coastguard Worker
12364*9880d681SAndroid Build Coastguard Worker // Sometimes build_vector's scalar input types do not match result type.
12365*9880d681SAndroid Build Coastguard Worker if (NVT == InEltVT)
12366*9880d681SAndroid Build Coastguard Worker return Elt;
12367*9880d681SAndroid Build Coastguard Worker
12368*9880d681SAndroid Build Coastguard Worker // TODO: It may be useful to truncate if free if the build_vector implicitly
12369*9880d681SAndroid Build Coastguard Worker // converts.
12370*9880d681SAndroid Build Coastguard Worker }
12371*9880d681SAndroid Build Coastguard Worker
12372*9880d681SAndroid Build Coastguard Worker // extract_vector_elt (v2i32 (bitcast i64:x)), 0 -> i32 (trunc i64:x)
12373*9880d681SAndroid Build Coastguard Worker if (ConstEltNo && InVec.getOpcode() == ISD::BITCAST && InVec.hasOneUse() &&
12374*9880d681SAndroid Build Coastguard Worker ConstEltNo->isNullValue() && VT.isInteger()) {
12375*9880d681SAndroid Build Coastguard Worker SDValue BCSrc = InVec.getOperand(0);
12376*9880d681SAndroid Build Coastguard Worker if (BCSrc.getValueType().isScalarInteger())
12377*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TRUNCATE, SDLoc(N), NVT, BCSrc);
12378*9880d681SAndroid Build Coastguard Worker }
12379*9880d681SAndroid Build Coastguard Worker
12380*9880d681SAndroid Build Coastguard Worker // extract_vector_elt (insert_vector_elt vec, val, idx), idx) -> val
12381*9880d681SAndroid Build Coastguard Worker //
12382*9880d681SAndroid Build Coastguard Worker // This only really matters if the index is non-constant since other combines
12383*9880d681SAndroid Build Coastguard Worker // on the constant elements already work.
12384*9880d681SAndroid Build Coastguard Worker if (InVec.getOpcode() == ISD::INSERT_VECTOR_ELT &&
12385*9880d681SAndroid Build Coastguard Worker EltNo == InVec.getOperand(2)) {
12386*9880d681SAndroid Build Coastguard Worker SDValue Elt = InVec.getOperand(1);
12387*9880d681SAndroid Build Coastguard Worker return VT.isInteger() ? DAG.getAnyExtOrTrunc(Elt, SDLoc(N), NVT) : Elt;
12388*9880d681SAndroid Build Coastguard Worker }
12389*9880d681SAndroid Build Coastguard Worker
12390*9880d681SAndroid Build Coastguard Worker // Transform: (EXTRACT_VECTOR_ELT( VECTOR_SHUFFLE )) -> EXTRACT_VECTOR_ELT.
12391*9880d681SAndroid Build Coastguard Worker // We only perform this optimization before the op legalization phase because
12392*9880d681SAndroid Build Coastguard Worker // we may introduce new vector instructions which are not backed by TD
12393*9880d681SAndroid Build Coastguard Worker // patterns. For example on AVX, extracting elements from a wide vector
12394*9880d681SAndroid Build Coastguard Worker // without using extract_subvector. However, if we can find an underlying
12395*9880d681SAndroid Build Coastguard Worker // scalar value, then we can always use that.
12396*9880d681SAndroid Build Coastguard Worker if (ConstEltNo && InVec.getOpcode() == ISD::VECTOR_SHUFFLE) {
12397*9880d681SAndroid Build Coastguard Worker int NumElem = VT.getVectorNumElements();
12398*9880d681SAndroid Build Coastguard Worker ShuffleVectorSDNode *SVOp = cast<ShuffleVectorSDNode>(InVec);
12399*9880d681SAndroid Build Coastguard Worker // Find the new index to extract from.
12400*9880d681SAndroid Build Coastguard Worker int OrigElt = SVOp->getMaskElt(ConstEltNo->getZExtValue());
12401*9880d681SAndroid Build Coastguard Worker
12402*9880d681SAndroid Build Coastguard Worker // Extracting an undef index is undef.
12403*9880d681SAndroid Build Coastguard Worker if (OrigElt == -1)
12404*9880d681SAndroid Build Coastguard Worker return DAG.getUNDEF(NVT);
12405*9880d681SAndroid Build Coastguard Worker
12406*9880d681SAndroid Build Coastguard Worker // Select the right vector half to extract from.
12407*9880d681SAndroid Build Coastguard Worker SDValue SVInVec;
12408*9880d681SAndroid Build Coastguard Worker if (OrigElt < NumElem) {
12409*9880d681SAndroid Build Coastguard Worker SVInVec = InVec->getOperand(0);
12410*9880d681SAndroid Build Coastguard Worker } else {
12411*9880d681SAndroid Build Coastguard Worker SVInVec = InVec->getOperand(1);
12412*9880d681SAndroid Build Coastguard Worker OrigElt -= NumElem;
12413*9880d681SAndroid Build Coastguard Worker }
12414*9880d681SAndroid Build Coastguard Worker
12415*9880d681SAndroid Build Coastguard Worker if (SVInVec.getOpcode() == ISD::BUILD_VECTOR) {
12416*9880d681SAndroid Build Coastguard Worker SDValue InOp = SVInVec.getOperand(OrigElt);
12417*9880d681SAndroid Build Coastguard Worker if (InOp.getValueType() != NVT) {
12418*9880d681SAndroid Build Coastguard Worker assert(InOp.getValueType().isInteger() && NVT.isInteger());
12419*9880d681SAndroid Build Coastguard Worker InOp = DAG.getSExtOrTrunc(InOp, SDLoc(SVInVec), NVT);
12420*9880d681SAndroid Build Coastguard Worker }
12421*9880d681SAndroid Build Coastguard Worker
12422*9880d681SAndroid Build Coastguard Worker return InOp;
12423*9880d681SAndroid Build Coastguard Worker }
12424*9880d681SAndroid Build Coastguard Worker
12425*9880d681SAndroid Build Coastguard Worker // FIXME: We should handle recursing on other vector shuffles and
12426*9880d681SAndroid Build Coastguard Worker // scalar_to_vector here as well.
12427*9880d681SAndroid Build Coastguard Worker
12428*9880d681SAndroid Build Coastguard Worker if (!LegalOperations) {
12429*9880d681SAndroid Build Coastguard Worker EVT IndexTy = TLI.getVectorIdxTy(DAG.getDataLayout());
12430*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SDLoc(N), NVT, SVInVec,
12431*9880d681SAndroid Build Coastguard Worker DAG.getConstant(OrigElt, SDLoc(SVOp), IndexTy));
12432*9880d681SAndroid Build Coastguard Worker }
12433*9880d681SAndroid Build Coastguard Worker }
12434*9880d681SAndroid Build Coastguard Worker
12435*9880d681SAndroid Build Coastguard Worker bool BCNumEltsChanged = false;
12436*9880d681SAndroid Build Coastguard Worker EVT ExtVT = VT.getVectorElementType();
12437*9880d681SAndroid Build Coastguard Worker EVT LVT = ExtVT;
12438*9880d681SAndroid Build Coastguard Worker
12439*9880d681SAndroid Build Coastguard Worker // If the result of load has to be truncated, then it's not necessarily
12440*9880d681SAndroid Build Coastguard Worker // profitable.
12441*9880d681SAndroid Build Coastguard Worker if (NVT.bitsLT(LVT) && !TLI.isTruncateFree(LVT, NVT))
12442*9880d681SAndroid Build Coastguard Worker return SDValue();
12443*9880d681SAndroid Build Coastguard Worker
12444*9880d681SAndroid Build Coastguard Worker if (InVec.getOpcode() == ISD::BITCAST) {
12445*9880d681SAndroid Build Coastguard Worker // Don't duplicate a load with other uses.
12446*9880d681SAndroid Build Coastguard Worker if (!InVec.hasOneUse())
12447*9880d681SAndroid Build Coastguard Worker return SDValue();
12448*9880d681SAndroid Build Coastguard Worker
12449*9880d681SAndroid Build Coastguard Worker EVT BCVT = InVec.getOperand(0).getValueType();
12450*9880d681SAndroid Build Coastguard Worker if (!BCVT.isVector() || ExtVT.bitsGT(BCVT.getVectorElementType()))
12451*9880d681SAndroid Build Coastguard Worker return SDValue();
12452*9880d681SAndroid Build Coastguard Worker if (VT.getVectorNumElements() != BCVT.getVectorNumElements())
12453*9880d681SAndroid Build Coastguard Worker BCNumEltsChanged = true;
12454*9880d681SAndroid Build Coastguard Worker InVec = InVec.getOperand(0);
12455*9880d681SAndroid Build Coastguard Worker ExtVT = BCVT.getVectorElementType();
12456*9880d681SAndroid Build Coastguard Worker }
12457*9880d681SAndroid Build Coastguard Worker
12458*9880d681SAndroid Build Coastguard Worker // (vextract (vN[if]M load $addr), i) -> ([if]M load $addr + i * size)
12459*9880d681SAndroid Build Coastguard Worker if (!LegalOperations && !ConstEltNo && InVec.hasOneUse() &&
12460*9880d681SAndroid Build Coastguard Worker ISD::isNormalLoad(InVec.getNode()) &&
12461*9880d681SAndroid Build Coastguard Worker !N->getOperand(1)->hasPredecessor(InVec.getNode())) {
12462*9880d681SAndroid Build Coastguard Worker SDValue Index = N->getOperand(1);
12463*9880d681SAndroid Build Coastguard Worker if (LoadSDNode *OrigLoad = dyn_cast<LoadSDNode>(InVec)) {
12464*9880d681SAndroid Build Coastguard Worker if (!OrigLoad->isVolatile()) {
12465*9880d681SAndroid Build Coastguard Worker return ReplaceExtractVectorEltOfLoadWithNarrowedLoad(N, VT, Index,
12466*9880d681SAndroid Build Coastguard Worker OrigLoad);
12467*9880d681SAndroid Build Coastguard Worker }
12468*9880d681SAndroid Build Coastguard Worker }
12469*9880d681SAndroid Build Coastguard Worker }
12470*9880d681SAndroid Build Coastguard Worker
12471*9880d681SAndroid Build Coastguard Worker // Perform only after legalization to ensure build_vector / vector_shuffle
12472*9880d681SAndroid Build Coastguard Worker // optimizations have already been done.
12473*9880d681SAndroid Build Coastguard Worker if (!LegalOperations) return SDValue();
12474*9880d681SAndroid Build Coastguard Worker
12475*9880d681SAndroid Build Coastguard Worker // (vextract (v4f32 load $addr), c) -> (f32 load $addr+c*size)
12476*9880d681SAndroid Build Coastguard Worker // (vextract (v4f32 s2v (f32 load $addr)), c) -> (f32 load $addr+c*size)
12477*9880d681SAndroid Build Coastguard Worker // (vextract (v4f32 shuffle (load $addr), <1,u,u,u>), 0) -> (f32 load $addr)
12478*9880d681SAndroid Build Coastguard Worker
12479*9880d681SAndroid Build Coastguard Worker if (ConstEltNo) {
12480*9880d681SAndroid Build Coastguard Worker int Elt = cast<ConstantSDNode>(EltNo)->getZExtValue();
12481*9880d681SAndroid Build Coastguard Worker
12482*9880d681SAndroid Build Coastguard Worker LoadSDNode *LN0 = nullptr;
12483*9880d681SAndroid Build Coastguard Worker const ShuffleVectorSDNode *SVN = nullptr;
12484*9880d681SAndroid Build Coastguard Worker if (ISD::isNormalLoad(InVec.getNode())) {
12485*9880d681SAndroid Build Coastguard Worker LN0 = cast<LoadSDNode>(InVec);
12486*9880d681SAndroid Build Coastguard Worker } else if (InVec.getOpcode() == ISD::SCALAR_TO_VECTOR &&
12487*9880d681SAndroid Build Coastguard Worker InVec.getOperand(0).getValueType() == ExtVT &&
12488*9880d681SAndroid Build Coastguard Worker ISD::isNormalLoad(InVec.getOperand(0).getNode())) {
12489*9880d681SAndroid Build Coastguard Worker // Don't duplicate a load with other uses.
12490*9880d681SAndroid Build Coastguard Worker if (!InVec.hasOneUse())
12491*9880d681SAndroid Build Coastguard Worker return SDValue();
12492*9880d681SAndroid Build Coastguard Worker
12493*9880d681SAndroid Build Coastguard Worker LN0 = cast<LoadSDNode>(InVec.getOperand(0));
12494*9880d681SAndroid Build Coastguard Worker } else if ((SVN = dyn_cast<ShuffleVectorSDNode>(InVec))) {
12495*9880d681SAndroid Build Coastguard Worker // (vextract (vector_shuffle (load $addr), v2, <1, u, u, u>), 1)
12496*9880d681SAndroid Build Coastguard Worker // =>
12497*9880d681SAndroid Build Coastguard Worker // (load $addr+1*size)
12498*9880d681SAndroid Build Coastguard Worker
12499*9880d681SAndroid Build Coastguard Worker // Don't duplicate a load with other uses.
12500*9880d681SAndroid Build Coastguard Worker if (!InVec.hasOneUse())
12501*9880d681SAndroid Build Coastguard Worker return SDValue();
12502*9880d681SAndroid Build Coastguard Worker
12503*9880d681SAndroid Build Coastguard Worker // If the bit convert changed the number of elements, it is unsafe
12504*9880d681SAndroid Build Coastguard Worker // to examine the mask.
12505*9880d681SAndroid Build Coastguard Worker if (BCNumEltsChanged)
12506*9880d681SAndroid Build Coastguard Worker return SDValue();
12507*9880d681SAndroid Build Coastguard Worker
12508*9880d681SAndroid Build Coastguard Worker // Select the input vector, guarding against out of range extract vector.
12509*9880d681SAndroid Build Coastguard Worker unsigned NumElems = VT.getVectorNumElements();
12510*9880d681SAndroid Build Coastguard Worker int Idx = (Elt > (int)NumElems) ? -1 : SVN->getMaskElt(Elt);
12511*9880d681SAndroid Build Coastguard Worker InVec = (Idx < (int)NumElems) ? InVec.getOperand(0) : InVec.getOperand(1);
12512*9880d681SAndroid Build Coastguard Worker
12513*9880d681SAndroid Build Coastguard Worker if (InVec.getOpcode() == ISD::BITCAST) {
12514*9880d681SAndroid Build Coastguard Worker // Don't duplicate a load with other uses.
12515*9880d681SAndroid Build Coastguard Worker if (!InVec.hasOneUse())
12516*9880d681SAndroid Build Coastguard Worker return SDValue();
12517*9880d681SAndroid Build Coastguard Worker
12518*9880d681SAndroid Build Coastguard Worker InVec = InVec.getOperand(0);
12519*9880d681SAndroid Build Coastguard Worker }
12520*9880d681SAndroid Build Coastguard Worker if (ISD::isNormalLoad(InVec.getNode())) {
12521*9880d681SAndroid Build Coastguard Worker LN0 = cast<LoadSDNode>(InVec);
12522*9880d681SAndroid Build Coastguard Worker Elt = (Idx < (int)NumElems) ? Idx : Idx - (int)NumElems;
12523*9880d681SAndroid Build Coastguard Worker EltNo = DAG.getConstant(Elt, SDLoc(EltNo), EltNo.getValueType());
12524*9880d681SAndroid Build Coastguard Worker }
12525*9880d681SAndroid Build Coastguard Worker }
12526*9880d681SAndroid Build Coastguard Worker
12527*9880d681SAndroid Build Coastguard Worker // Make sure we found a non-volatile load and the extractelement is
12528*9880d681SAndroid Build Coastguard Worker // the only use.
12529*9880d681SAndroid Build Coastguard Worker if (!LN0 || !LN0->hasNUsesOfValue(1,0) || LN0->isVolatile())
12530*9880d681SAndroid Build Coastguard Worker return SDValue();
12531*9880d681SAndroid Build Coastguard Worker
12532*9880d681SAndroid Build Coastguard Worker // If Idx was -1 above, Elt is going to be -1, so just return undef.
12533*9880d681SAndroid Build Coastguard Worker if (Elt == -1)
12534*9880d681SAndroid Build Coastguard Worker return DAG.getUNDEF(LVT);
12535*9880d681SAndroid Build Coastguard Worker
12536*9880d681SAndroid Build Coastguard Worker return ReplaceExtractVectorEltOfLoadWithNarrowedLoad(N, VT, EltNo, LN0);
12537*9880d681SAndroid Build Coastguard Worker }
12538*9880d681SAndroid Build Coastguard Worker
12539*9880d681SAndroid Build Coastguard Worker return SDValue();
12540*9880d681SAndroid Build Coastguard Worker }
12541*9880d681SAndroid Build Coastguard Worker
12542*9880d681SAndroid Build Coastguard Worker // Simplify (build_vec (ext )) to (bitcast (build_vec ))
reduceBuildVecExtToExtBuildVec(SDNode * N)12543*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::reduceBuildVecExtToExtBuildVec(SDNode *N) {
12544*9880d681SAndroid Build Coastguard Worker // We perform this optimization post type-legalization because
12545*9880d681SAndroid Build Coastguard Worker // the type-legalizer often scalarizes integer-promoted vectors.
12546*9880d681SAndroid Build Coastguard Worker // Performing this optimization before may create bit-casts which
12547*9880d681SAndroid Build Coastguard Worker // will be type-legalized to complex code sequences.
12548*9880d681SAndroid Build Coastguard Worker // We perform this optimization only before the operation legalizer because we
12549*9880d681SAndroid Build Coastguard Worker // may introduce illegal operations.
12550*9880d681SAndroid Build Coastguard Worker if (Level != AfterLegalizeVectorOps && Level != AfterLegalizeTypes)
12551*9880d681SAndroid Build Coastguard Worker return SDValue();
12552*9880d681SAndroid Build Coastguard Worker
12553*9880d681SAndroid Build Coastguard Worker unsigned NumInScalars = N->getNumOperands();
12554*9880d681SAndroid Build Coastguard Worker SDLoc dl(N);
12555*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
12556*9880d681SAndroid Build Coastguard Worker
12557*9880d681SAndroid Build Coastguard Worker // Check to see if this is a BUILD_VECTOR of a bunch of values
12558*9880d681SAndroid Build Coastguard Worker // which come from any_extend or zero_extend nodes. If so, we can create
12559*9880d681SAndroid Build Coastguard Worker // a new BUILD_VECTOR using bit-casts which may enable other BUILD_VECTOR
12560*9880d681SAndroid Build Coastguard Worker // optimizations. We do not handle sign-extend because we can't fill the sign
12561*9880d681SAndroid Build Coastguard Worker // using shuffles.
12562*9880d681SAndroid Build Coastguard Worker EVT SourceType = MVT::Other;
12563*9880d681SAndroid Build Coastguard Worker bool AllAnyExt = true;
12564*9880d681SAndroid Build Coastguard Worker
12565*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i != NumInScalars; ++i) {
12566*9880d681SAndroid Build Coastguard Worker SDValue In = N->getOperand(i);
12567*9880d681SAndroid Build Coastguard Worker // Ignore undef inputs.
12568*9880d681SAndroid Build Coastguard Worker if (In.isUndef()) continue;
12569*9880d681SAndroid Build Coastguard Worker
12570*9880d681SAndroid Build Coastguard Worker bool AnyExt = In.getOpcode() == ISD::ANY_EXTEND;
12571*9880d681SAndroid Build Coastguard Worker bool ZeroExt = In.getOpcode() == ISD::ZERO_EXTEND;
12572*9880d681SAndroid Build Coastguard Worker
12573*9880d681SAndroid Build Coastguard Worker // Abort if the element is not an extension.
12574*9880d681SAndroid Build Coastguard Worker if (!ZeroExt && !AnyExt) {
12575*9880d681SAndroid Build Coastguard Worker SourceType = MVT::Other;
12576*9880d681SAndroid Build Coastguard Worker break;
12577*9880d681SAndroid Build Coastguard Worker }
12578*9880d681SAndroid Build Coastguard Worker
12579*9880d681SAndroid Build Coastguard Worker // The input is a ZeroExt or AnyExt. Check the original type.
12580*9880d681SAndroid Build Coastguard Worker EVT InTy = In.getOperand(0).getValueType();
12581*9880d681SAndroid Build Coastguard Worker
12582*9880d681SAndroid Build Coastguard Worker // Check that all of the widened source types are the same.
12583*9880d681SAndroid Build Coastguard Worker if (SourceType == MVT::Other)
12584*9880d681SAndroid Build Coastguard Worker // First time.
12585*9880d681SAndroid Build Coastguard Worker SourceType = InTy;
12586*9880d681SAndroid Build Coastguard Worker else if (InTy != SourceType) {
12587*9880d681SAndroid Build Coastguard Worker // Multiple income types. Abort.
12588*9880d681SAndroid Build Coastguard Worker SourceType = MVT::Other;
12589*9880d681SAndroid Build Coastguard Worker break;
12590*9880d681SAndroid Build Coastguard Worker }
12591*9880d681SAndroid Build Coastguard Worker
12592*9880d681SAndroid Build Coastguard Worker // Check if all of the extends are ANY_EXTENDs.
12593*9880d681SAndroid Build Coastguard Worker AllAnyExt &= AnyExt;
12594*9880d681SAndroid Build Coastguard Worker }
12595*9880d681SAndroid Build Coastguard Worker
12596*9880d681SAndroid Build Coastguard Worker // In order to have valid types, all of the inputs must be extended from the
12597*9880d681SAndroid Build Coastguard Worker // same source type and all of the inputs must be any or zero extend.
12598*9880d681SAndroid Build Coastguard Worker // Scalar sizes must be a power of two.
12599*9880d681SAndroid Build Coastguard Worker EVT OutScalarTy = VT.getScalarType();
12600*9880d681SAndroid Build Coastguard Worker bool ValidTypes = SourceType != MVT::Other &&
12601*9880d681SAndroid Build Coastguard Worker isPowerOf2_32(OutScalarTy.getSizeInBits()) &&
12602*9880d681SAndroid Build Coastguard Worker isPowerOf2_32(SourceType.getSizeInBits());
12603*9880d681SAndroid Build Coastguard Worker
12604*9880d681SAndroid Build Coastguard Worker // Create a new simpler BUILD_VECTOR sequence which other optimizations can
12605*9880d681SAndroid Build Coastguard Worker // turn into a single shuffle instruction.
12606*9880d681SAndroid Build Coastguard Worker if (!ValidTypes)
12607*9880d681SAndroid Build Coastguard Worker return SDValue();
12608*9880d681SAndroid Build Coastguard Worker
12609*9880d681SAndroid Build Coastguard Worker bool isLE = DAG.getDataLayout().isLittleEndian();
12610*9880d681SAndroid Build Coastguard Worker unsigned ElemRatio = OutScalarTy.getSizeInBits()/SourceType.getSizeInBits();
12611*9880d681SAndroid Build Coastguard Worker assert(ElemRatio > 1 && "Invalid element size ratio");
12612*9880d681SAndroid Build Coastguard Worker SDValue Filler = AllAnyExt ? DAG.getUNDEF(SourceType):
12613*9880d681SAndroid Build Coastguard Worker DAG.getConstant(0, SDLoc(N), SourceType);
12614*9880d681SAndroid Build Coastguard Worker
12615*9880d681SAndroid Build Coastguard Worker unsigned NewBVElems = ElemRatio * VT.getVectorNumElements();
12616*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Ops(NewBVElems, Filler);
12617*9880d681SAndroid Build Coastguard Worker
12618*9880d681SAndroid Build Coastguard Worker // Populate the new build_vector
12619*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i) {
12620*9880d681SAndroid Build Coastguard Worker SDValue Cast = N->getOperand(i);
12621*9880d681SAndroid Build Coastguard Worker assert((Cast.getOpcode() == ISD::ANY_EXTEND ||
12622*9880d681SAndroid Build Coastguard Worker Cast.getOpcode() == ISD::ZERO_EXTEND ||
12623*9880d681SAndroid Build Coastguard Worker Cast.isUndef()) && "Invalid cast opcode");
12624*9880d681SAndroid Build Coastguard Worker SDValue In;
12625*9880d681SAndroid Build Coastguard Worker if (Cast.isUndef())
12626*9880d681SAndroid Build Coastguard Worker In = DAG.getUNDEF(SourceType);
12627*9880d681SAndroid Build Coastguard Worker else
12628*9880d681SAndroid Build Coastguard Worker In = Cast->getOperand(0);
12629*9880d681SAndroid Build Coastguard Worker unsigned Index = isLE ? (i * ElemRatio) :
12630*9880d681SAndroid Build Coastguard Worker (i * ElemRatio + (ElemRatio - 1));
12631*9880d681SAndroid Build Coastguard Worker
12632*9880d681SAndroid Build Coastguard Worker assert(Index < Ops.size() && "Invalid index");
12633*9880d681SAndroid Build Coastguard Worker Ops[Index] = In;
12634*9880d681SAndroid Build Coastguard Worker }
12635*9880d681SAndroid Build Coastguard Worker
12636*9880d681SAndroid Build Coastguard Worker // The type of the new BUILD_VECTOR node.
12637*9880d681SAndroid Build Coastguard Worker EVT VecVT = EVT::getVectorVT(*DAG.getContext(), SourceType, NewBVElems);
12638*9880d681SAndroid Build Coastguard Worker assert(VecVT.getSizeInBits() == VT.getSizeInBits() &&
12639*9880d681SAndroid Build Coastguard Worker "Invalid vector size");
12640*9880d681SAndroid Build Coastguard Worker // Check if the new vector type is legal.
12641*9880d681SAndroid Build Coastguard Worker if (!isTypeLegal(VecVT)) return SDValue();
12642*9880d681SAndroid Build Coastguard Worker
12643*9880d681SAndroid Build Coastguard Worker // Make the new BUILD_VECTOR.
12644*9880d681SAndroid Build Coastguard Worker SDValue BV = DAG.getBuildVector(VecVT, dl, Ops);
12645*9880d681SAndroid Build Coastguard Worker
12646*9880d681SAndroid Build Coastguard Worker // The new BUILD_VECTOR node has the potential to be further optimized.
12647*9880d681SAndroid Build Coastguard Worker AddToWorklist(BV.getNode());
12648*9880d681SAndroid Build Coastguard Worker // Bitcast to the desired type.
12649*9880d681SAndroid Build Coastguard Worker return DAG.getBitcast(VT, BV);
12650*9880d681SAndroid Build Coastguard Worker }
12651*9880d681SAndroid Build Coastguard Worker
reduceBuildVecConvertToConvertBuildVec(SDNode * N)12652*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::reduceBuildVecConvertToConvertBuildVec(SDNode *N) {
12653*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
12654*9880d681SAndroid Build Coastguard Worker
12655*9880d681SAndroid Build Coastguard Worker unsigned NumInScalars = N->getNumOperands();
12656*9880d681SAndroid Build Coastguard Worker SDLoc dl(N);
12657*9880d681SAndroid Build Coastguard Worker
12658*9880d681SAndroid Build Coastguard Worker EVT SrcVT = MVT::Other;
12659*9880d681SAndroid Build Coastguard Worker unsigned Opcode = ISD::DELETED_NODE;
12660*9880d681SAndroid Build Coastguard Worker unsigned NumDefs = 0;
12661*9880d681SAndroid Build Coastguard Worker
12662*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i != NumInScalars; ++i) {
12663*9880d681SAndroid Build Coastguard Worker SDValue In = N->getOperand(i);
12664*9880d681SAndroid Build Coastguard Worker unsigned Opc = In.getOpcode();
12665*9880d681SAndroid Build Coastguard Worker
12666*9880d681SAndroid Build Coastguard Worker if (Opc == ISD::UNDEF)
12667*9880d681SAndroid Build Coastguard Worker continue;
12668*9880d681SAndroid Build Coastguard Worker
12669*9880d681SAndroid Build Coastguard Worker // If all scalar values are floats and converted from integers.
12670*9880d681SAndroid Build Coastguard Worker if (Opcode == ISD::DELETED_NODE &&
12671*9880d681SAndroid Build Coastguard Worker (Opc == ISD::UINT_TO_FP || Opc == ISD::SINT_TO_FP)) {
12672*9880d681SAndroid Build Coastguard Worker Opcode = Opc;
12673*9880d681SAndroid Build Coastguard Worker }
12674*9880d681SAndroid Build Coastguard Worker
12675*9880d681SAndroid Build Coastguard Worker if (Opc != Opcode)
12676*9880d681SAndroid Build Coastguard Worker return SDValue();
12677*9880d681SAndroid Build Coastguard Worker
12678*9880d681SAndroid Build Coastguard Worker EVT InVT = In.getOperand(0).getValueType();
12679*9880d681SAndroid Build Coastguard Worker
12680*9880d681SAndroid Build Coastguard Worker // If all scalar values are typed differently, bail out. It's chosen to
12681*9880d681SAndroid Build Coastguard Worker // simplify BUILD_VECTOR of integer types.
12682*9880d681SAndroid Build Coastguard Worker if (SrcVT == MVT::Other)
12683*9880d681SAndroid Build Coastguard Worker SrcVT = InVT;
12684*9880d681SAndroid Build Coastguard Worker if (SrcVT != InVT)
12685*9880d681SAndroid Build Coastguard Worker return SDValue();
12686*9880d681SAndroid Build Coastguard Worker NumDefs++;
12687*9880d681SAndroid Build Coastguard Worker }
12688*9880d681SAndroid Build Coastguard Worker
12689*9880d681SAndroid Build Coastguard Worker // If the vector has just one element defined, it's not worth to fold it into
12690*9880d681SAndroid Build Coastguard Worker // a vectorized one.
12691*9880d681SAndroid Build Coastguard Worker if (NumDefs < 2)
12692*9880d681SAndroid Build Coastguard Worker return SDValue();
12693*9880d681SAndroid Build Coastguard Worker
12694*9880d681SAndroid Build Coastguard Worker assert((Opcode == ISD::UINT_TO_FP || Opcode == ISD::SINT_TO_FP)
12695*9880d681SAndroid Build Coastguard Worker && "Should only handle conversion from integer to float.");
12696*9880d681SAndroid Build Coastguard Worker assert(SrcVT != MVT::Other && "Cannot determine source type!");
12697*9880d681SAndroid Build Coastguard Worker
12698*9880d681SAndroid Build Coastguard Worker EVT NVT = EVT::getVectorVT(*DAG.getContext(), SrcVT, NumInScalars);
12699*9880d681SAndroid Build Coastguard Worker
12700*9880d681SAndroid Build Coastguard Worker if (!TLI.isOperationLegalOrCustom(Opcode, NVT))
12701*9880d681SAndroid Build Coastguard Worker return SDValue();
12702*9880d681SAndroid Build Coastguard Worker
12703*9880d681SAndroid Build Coastguard Worker // Just because the floating-point vector type is legal does not necessarily
12704*9880d681SAndroid Build Coastguard Worker // mean that the corresponding integer vector type is.
12705*9880d681SAndroid Build Coastguard Worker if (!isTypeLegal(NVT))
12706*9880d681SAndroid Build Coastguard Worker return SDValue();
12707*9880d681SAndroid Build Coastguard Worker
12708*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Opnds;
12709*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i != NumInScalars; ++i) {
12710*9880d681SAndroid Build Coastguard Worker SDValue In = N->getOperand(i);
12711*9880d681SAndroid Build Coastguard Worker
12712*9880d681SAndroid Build Coastguard Worker if (In.isUndef())
12713*9880d681SAndroid Build Coastguard Worker Opnds.push_back(DAG.getUNDEF(SrcVT));
12714*9880d681SAndroid Build Coastguard Worker else
12715*9880d681SAndroid Build Coastguard Worker Opnds.push_back(In.getOperand(0));
12716*9880d681SAndroid Build Coastguard Worker }
12717*9880d681SAndroid Build Coastguard Worker SDValue BV = DAG.getBuildVector(NVT, dl, Opnds);
12718*9880d681SAndroid Build Coastguard Worker AddToWorklist(BV.getNode());
12719*9880d681SAndroid Build Coastguard Worker
12720*9880d681SAndroid Build Coastguard Worker return DAG.getNode(Opcode, dl, VT, BV);
12721*9880d681SAndroid Build Coastguard Worker }
12722*9880d681SAndroid Build Coastguard Worker
visitBUILD_VECTOR(SDNode * N)12723*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitBUILD_VECTOR(SDNode *N) {
12724*9880d681SAndroid Build Coastguard Worker unsigned NumInScalars = N->getNumOperands();
12725*9880d681SAndroid Build Coastguard Worker SDLoc dl(N);
12726*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
12727*9880d681SAndroid Build Coastguard Worker
12728*9880d681SAndroid Build Coastguard Worker // A vector built entirely of undefs is undef.
12729*9880d681SAndroid Build Coastguard Worker if (ISD::allOperandsUndef(N))
12730*9880d681SAndroid Build Coastguard Worker return DAG.getUNDEF(VT);
12731*9880d681SAndroid Build Coastguard Worker
12732*9880d681SAndroid Build Coastguard Worker if (SDValue V = reduceBuildVecExtToExtBuildVec(N))
12733*9880d681SAndroid Build Coastguard Worker return V;
12734*9880d681SAndroid Build Coastguard Worker
12735*9880d681SAndroid Build Coastguard Worker if (SDValue V = reduceBuildVecConvertToConvertBuildVec(N))
12736*9880d681SAndroid Build Coastguard Worker return V;
12737*9880d681SAndroid Build Coastguard Worker
12738*9880d681SAndroid Build Coastguard Worker // Check to see if this is a BUILD_VECTOR of a bunch of EXTRACT_VECTOR_ELT
12739*9880d681SAndroid Build Coastguard Worker // operations. If so, and if the EXTRACT_VECTOR_ELT vector inputs come from
12740*9880d681SAndroid Build Coastguard Worker // at most two distinct vectors, turn this into a shuffle node.
12741*9880d681SAndroid Build Coastguard Worker
12742*9880d681SAndroid Build Coastguard Worker // Only type-legal BUILD_VECTOR nodes are converted to shuffle nodes.
12743*9880d681SAndroid Build Coastguard Worker if (!isTypeLegal(VT))
12744*9880d681SAndroid Build Coastguard Worker return SDValue();
12745*9880d681SAndroid Build Coastguard Worker
12746*9880d681SAndroid Build Coastguard Worker // May only combine to shuffle after legalize if shuffle is legal.
12747*9880d681SAndroid Build Coastguard Worker if (LegalOperations && !TLI.isOperationLegal(ISD::VECTOR_SHUFFLE, VT))
12748*9880d681SAndroid Build Coastguard Worker return SDValue();
12749*9880d681SAndroid Build Coastguard Worker
12750*9880d681SAndroid Build Coastguard Worker SDValue VecIn1, VecIn2;
12751*9880d681SAndroid Build Coastguard Worker bool UsesZeroVector = false;
12752*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i != NumInScalars; ++i) {
12753*9880d681SAndroid Build Coastguard Worker SDValue Op = N->getOperand(i);
12754*9880d681SAndroid Build Coastguard Worker // Ignore undef inputs.
12755*9880d681SAndroid Build Coastguard Worker if (Op.isUndef()) continue;
12756*9880d681SAndroid Build Coastguard Worker
12757*9880d681SAndroid Build Coastguard Worker // See if we can combine this build_vector into a blend with a zero vector.
12758*9880d681SAndroid Build Coastguard Worker if (!VecIn2.getNode() && (isNullConstant(Op) || isNullFPConstant(Op))) {
12759*9880d681SAndroid Build Coastguard Worker UsesZeroVector = true;
12760*9880d681SAndroid Build Coastguard Worker continue;
12761*9880d681SAndroid Build Coastguard Worker }
12762*9880d681SAndroid Build Coastguard Worker
12763*9880d681SAndroid Build Coastguard Worker // If this input is something other than a EXTRACT_VECTOR_ELT with a
12764*9880d681SAndroid Build Coastguard Worker // constant index, bail out.
12765*9880d681SAndroid Build Coastguard Worker if (Op.getOpcode() != ISD::EXTRACT_VECTOR_ELT ||
12766*9880d681SAndroid Build Coastguard Worker !isa<ConstantSDNode>(Op.getOperand(1))) {
12767*9880d681SAndroid Build Coastguard Worker VecIn1 = VecIn2 = SDValue(nullptr, 0);
12768*9880d681SAndroid Build Coastguard Worker break;
12769*9880d681SAndroid Build Coastguard Worker }
12770*9880d681SAndroid Build Coastguard Worker
12771*9880d681SAndroid Build Coastguard Worker // We allow up to two distinct input vectors.
12772*9880d681SAndroid Build Coastguard Worker SDValue ExtractedFromVec = Op.getOperand(0);
12773*9880d681SAndroid Build Coastguard Worker if (ExtractedFromVec == VecIn1 || ExtractedFromVec == VecIn2)
12774*9880d681SAndroid Build Coastguard Worker continue;
12775*9880d681SAndroid Build Coastguard Worker
12776*9880d681SAndroid Build Coastguard Worker if (!VecIn1.getNode()) {
12777*9880d681SAndroid Build Coastguard Worker VecIn1 = ExtractedFromVec;
12778*9880d681SAndroid Build Coastguard Worker } else if (!VecIn2.getNode() && !UsesZeroVector) {
12779*9880d681SAndroid Build Coastguard Worker VecIn2 = ExtractedFromVec;
12780*9880d681SAndroid Build Coastguard Worker } else {
12781*9880d681SAndroid Build Coastguard Worker // Too many inputs.
12782*9880d681SAndroid Build Coastguard Worker VecIn1 = VecIn2 = SDValue(nullptr, 0);
12783*9880d681SAndroid Build Coastguard Worker break;
12784*9880d681SAndroid Build Coastguard Worker }
12785*9880d681SAndroid Build Coastguard Worker }
12786*9880d681SAndroid Build Coastguard Worker
12787*9880d681SAndroid Build Coastguard Worker // If everything is good, we can make a shuffle operation.
12788*9880d681SAndroid Build Coastguard Worker if (VecIn1.getNode()) {
12789*9880d681SAndroid Build Coastguard Worker unsigned InNumElements = VecIn1.getValueType().getVectorNumElements();
12790*9880d681SAndroid Build Coastguard Worker SmallVector<int, 8> Mask;
12791*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i != NumInScalars; ++i) {
12792*9880d681SAndroid Build Coastguard Worker unsigned Opcode = N->getOperand(i).getOpcode();
12793*9880d681SAndroid Build Coastguard Worker if (Opcode == ISD::UNDEF) {
12794*9880d681SAndroid Build Coastguard Worker Mask.push_back(-1);
12795*9880d681SAndroid Build Coastguard Worker continue;
12796*9880d681SAndroid Build Coastguard Worker }
12797*9880d681SAndroid Build Coastguard Worker
12798*9880d681SAndroid Build Coastguard Worker // Operands can also be zero.
12799*9880d681SAndroid Build Coastguard Worker if (Opcode != ISD::EXTRACT_VECTOR_ELT) {
12800*9880d681SAndroid Build Coastguard Worker assert(UsesZeroVector &&
12801*9880d681SAndroid Build Coastguard Worker (Opcode == ISD::Constant || Opcode == ISD::ConstantFP) &&
12802*9880d681SAndroid Build Coastguard Worker "Unexpected node found!");
12803*9880d681SAndroid Build Coastguard Worker Mask.push_back(NumInScalars+i);
12804*9880d681SAndroid Build Coastguard Worker continue;
12805*9880d681SAndroid Build Coastguard Worker }
12806*9880d681SAndroid Build Coastguard Worker
12807*9880d681SAndroid Build Coastguard Worker // If extracting from the first vector, just use the index directly.
12808*9880d681SAndroid Build Coastguard Worker SDValue Extract = N->getOperand(i);
12809*9880d681SAndroid Build Coastguard Worker SDValue ExtVal = Extract.getOperand(1);
12810*9880d681SAndroid Build Coastguard Worker unsigned ExtIndex = cast<ConstantSDNode>(ExtVal)->getZExtValue();
12811*9880d681SAndroid Build Coastguard Worker if (Extract.getOperand(0) == VecIn1) {
12812*9880d681SAndroid Build Coastguard Worker Mask.push_back(ExtIndex);
12813*9880d681SAndroid Build Coastguard Worker continue;
12814*9880d681SAndroid Build Coastguard Worker }
12815*9880d681SAndroid Build Coastguard Worker
12816*9880d681SAndroid Build Coastguard Worker // Otherwise, use InIdx + InputVecSize
12817*9880d681SAndroid Build Coastguard Worker Mask.push_back(InNumElements + ExtIndex);
12818*9880d681SAndroid Build Coastguard Worker }
12819*9880d681SAndroid Build Coastguard Worker
12820*9880d681SAndroid Build Coastguard Worker // Avoid introducing illegal shuffles with zero.
12821*9880d681SAndroid Build Coastguard Worker if (UsesZeroVector && !TLI.isVectorClearMaskLegal(Mask, VT))
12822*9880d681SAndroid Build Coastguard Worker return SDValue();
12823*9880d681SAndroid Build Coastguard Worker
12824*9880d681SAndroid Build Coastguard Worker // We can't generate a shuffle node with mismatched input and output types.
12825*9880d681SAndroid Build Coastguard Worker // Attempt to transform a single input vector to the correct type.
12826*9880d681SAndroid Build Coastguard Worker if ((VT != VecIn1.getValueType())) {
12827*9880d681SAndroid Build Coastguard Worker // If the input vector type has a different base type to the output
12828*9880d681SAndroid Build Coastguard Worker // vector type, bail out.
12829*9880d681SAndroid Build Coastguard Worker EVT VTElemType = VT.getVectorElementType();
12830*9880d681SAndroid Build Coastguard Worker if ((VecIn1.getValueType().getVectorElementType() != VTElemType) ||
12831*9880d681SAndroid Build Coastguard Worker (VecIn2.getNode() &&
12832*9880d681SAndroid Build Coastguard Worker (VecIn2.getValueType().getVectorElementType() != VTElemType)))
12833*9880d681SAndroid Build Coastguard Worker return SDValue();
12834*9880d681SAndroid Build Coastguard Worker
12835*9880d681SAndroid Build Coastguard Worker // If the input vector is too small, widen it.
12836*9880d681SAndroid Build Coastguard Worker // We only support widening of vectors which are half the size of the
12837*9880d681SAndroid Build Coastguard Worker // output registers. For example XMM->YMM widening on X86 with AVX.
12838*9880d681SAndroid Build Coastguard Worker EVT VecInT = VecIn1.getValueType();
12839*9880d681SAndroid Build Coastguard Worker if (VecInT.getSizeInBits() * 2 == VT.getSizeInBits()) {
12840*9880d681SAndroid Build Coastguard Worker // If we only have one small input, widen it by adding undef values.
12841*9880d681SAndroid Build Coastguard Worker if (!VecIn2.getNode())
12842*9880d681SAndroid Build Coastguard Worker VecIn1 = DAG.getNode(ISD::CONCAT_VECTORS, dl, VT, VecIn1,
12843*9880d681SAndroid Build Coastguard Worker DAG.getUNDEF(VecIn1.getValueType()));
12844*9880d681SAndroid Build Coastguard Worker else if (VecIn1.getValueType() == VecIn2.getValueType()) {
12845*9880d681SAndroid Build Coastguard Worker // If we have two small inputs of the same type, try to concat them.
12846*9880d681SAndroid Build Coastguard Worker VecIn1 = DAG.getNode(ISD::CONCAT_VECTORS, dl, VT, VecIn1, VecIn2);
12847*9880d681SAndroid Build Coastguard Worker VecIn2 = SDValue(nullptr, 0);
12848*9880d681SAndroid Build Coastguard Worker } else
12849*9880d681SAndroid Build Coastguard Worker return SDValue();
12850*9880d681SAndroid Build Coastguard Worker } else if (VecInT.getSizeInBits() == VT.getSizeInBits() * 2) {
12851*9880d681SAndroid Build Coastguard Worker // If the input vector is too large, try to split it.
12852*9880d681SAndroid Build Coastguard Worker // We don't support having two input vectors that are too large.
12853*9880d681SAndroid Build Coastguard Worker // If the zero vector was used, we can not split the vector,
12854*9880d681SAndroid Build Coastguard Worker // since we'd need 3 inputs.
12855*9880d681SAndroid Build Coastguard Worker if (UsesZeroVector || VecIn2.getNode())
12856*9880d681SAndroid Build Coastguard Worker return SDValue();
12857*9880d681SAndroid Build Coastguard Worker
12858*9880d681SAndroid Build Coastguard Worker if (!TLI.isExtractSubvectorCheap(VT, VT.getVectorNumElements()))
12859*9880d681SAndroid Build Coastguard Worker return SDValue();
12860*9880d681SAndroid Build Coastguard Worker
12861*9880d681SAndroid Build Coastguard Worker // Try to replace VecIn1 with two extract_subvectors
12862*9880d681SAndroid Build Coastguard Worker // No need to update the masks, they should still be correct.
12863*9880d681SAndroid Build Coastguard Worker VecIn2 = DAG.getNode(
12864*9880d681SAndroid Build Coastguard Worker ISD::EXTRACT_SUBVECTOR, dl, VT, VecIn1,
12865*9880d681SAndroid Build Coastguard Worker DAG.getConstant(VT.getVectorNumElements(), dl,
12866*9880d681SAndroid Build Coastguard Worker TLI.getVectorIdxTy(DAG.getDataLayout())));
12867*9880d681SAndroid Build Coastguard Worker VecIn1 = DAG.getNode(
12868*9880d681SAndroid Build Coastguard Worker ISD::EXTRACT_SUBVECTOR, dl, VT, VecIn1,
12869*9880d681SAndroid Build Coastguard Worker DAG.getConstant(0, dl, TLI.getVectorIdxTy(DAG.getDataLayout())));
12870*9880d681SAndroid Build Coastguard Worker } else
12871*9880d681SAndroid Build Coastguard Worker return SDValue();
12872*9880d681SAndroid Build Coastguard Worker }
12873*9880d681SAndroid Build Coastguard Worker
12874*9880d681SAndroid Build Coastguard Worker if (UsesZeroVector)
12875*9880d681SAndroid Build Coastguard Worker VecIn2 = VT.isInteger() ? DAG.getConstant(0, dl, VT) :
12876*9880d681SAndroid Build Coastguard Worker DAG.getConstantFP(0.0, dl, VT);
12877*9880d681SAndroid Build Coastguard Worker else
12878*9880d681SAndroid Build Coastguard Worker // If VecIn2 is unused then change it to undef.
12879*9880d681SAndroid Build Coastguard Worker VecIn2 = VecIn2.getNode() ? VecIn2 : DAG.getUNDEF(VT);
12880*9880d681SAndroid Build Coastguard Worker
12881*9880d681SAndroid Build Coastguard Worker // Check that we were able to transform all incoming values to the same
12882*9880d681SAndroid Build Coastguard Worker // type.
12883*9880d681SAndroid Build Coastguard Worker if (VecIn2.getValueType() != VecIn1.getValueType() ||
12884*9880d681SAndroid Build Coastguard Worker VecIn1.getValueType() != VT)
12885*9880d681SAndroid Build Coastguard Worker return SDValue();
12886*9880d681SAndroid Build Coastguard Worker
12887*9880d681SAndroid Build Coastguard Worker // Return the new VECTOR_SHUFFLE node.
12888*9880d681SAndroid Build Coastguard Worker SDValue Ops[2];
12889*9880d681SAndroid Build Coastguard Worker Ops[0] = VecIn1;
12890*9880d681SAndroid Build Coastguard Worker Ops[1] = VecIn2;
12891*9880d681SAndroid Build Coastguard Worker return DAG.getVectorShuffle(VT, dl, Ops[0], Ops[1], Mask);
12892*9880d681SAndroid Build Coastguard Worker }
12893*9880d681SAndroid Build Coastguard Worker
12894*9880d681SAndroid Build Coastguard Worker return SDValue();
12895*9880d681SAndroid Build Coastguard Worker }
12896*9880d681SAndroid Build Coastguard Worker
combineConcatVectorOfScalars(SDNode * N,SelectionDAG & DAG)12897*9880d681SAndroid Build Coastguard Worker static SDValue combineConcatVectorOfScalars(SDNode *N, SelectionDAG &DAG) {
12898*9880d681SAndroid Build Coastguard Worker const TargetLowering &TLI = DAG.getTargetLoweringInfo();
12899*9880d681SAndroid Build Coastguard Worker EVT OpVT = N->getOperand(0).getValueType();
12900*9880d681SAndroid Build Coastguard Worker
12901*9880d681SAndroid Build Coastguard Worker // If the operands are legal vectors, leave them alone.
12902*9880d681SAndroid Build Coastguard Worker if (TLI.isTypeLegal(OpVT))
12903*9880d681SAndroid Build Coastguard Worker return SDValue();
12904*9880d681SAndroid Build Coastguard Worker
12905*9880d681SAndroid Build Coastguard Worker SDLoc DL(N);
12906*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
12907*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Ops;
12908*9880d681SAndroid Build Coastguard Worker
12909*9880d681SAndroid Build Coastguard Worker EVT SVT = EVT::getIntegerVT(*DAG.getContext(), OpVT.getSizeInBits());
12910*9880d681SAndroid Build Coastguard Worker SDValue ScalarUndef = DAG.getNode(ISD::UNDEF, DL, SVT);
12911*9880d681SAndroid Build Coastguard Worker
12912*9880d681SAndroid Build Coastguard Worker // Keep track of what we encounter.
12913*9880d681SAndroid Build Coastguard Worker bool AnyInteger = false;
12914*9880d681SAndroid Build Coastguard Worker bool AnyFP = false;
12915*9880d681SAndroid Build Coastguard Worker for (const SDValue &Op : N->ops()) {
12916*9880d681SAndroid Build Coastguard Worker if (ISD::BITCAST == Op.getOpcode() &&
12917*9880d681SAndroid Build Coastguard Worker !Op.getOperand(0).getValueType().isVector())
12918*9880d681SAndroid Build Coastguard Worker Ops.push_back(Op.getOperand(0));
12919*9880d681SAndroid Build Coastguard Worker else if (ISD::UNDEF == Op.getOpcode())
12920*9880d681SAndroid Build Coastguard Worker Ops.push_back(ScalarUndef);
12921*9880d681SAndroid Build Coastguard Worker else
12922*9880d681SAndroid Build Coastguard Worker return SDValue();
12923*9880d681SAndroid Build Coastguard Worker
12924*9880d681SAndroid Build Coastguard Worker // Note whether we encounter an integer or floating point scalar.
12925*9880d681SAndroid Build Coastguard Worker // If it's neither, bail out, it could be something weird like x86mmx.
12926*9880d681SAndroid Build Coastguard Worker EVT LastOpVT = Ops.back().getValueType();
12927*9880d681SAndroid Build Coastguard Worker if (LastOpVT.isFloatingPoint())
12928*9880d681SAndroid Build Coastguard Worker AnyFP = true;
12929*9880d681SAndroid Build Coastguard Worker else if (LastOpVT.isInteger())
12930*9880d681SAndroid Build Coastguard Worker AnyInteger = true;
12931*9880d681SAndroid Build Coastguard Worker else
12932*9880d681SAndroid Build Coastguard Worker return SDValue();
12933*9880d681SAndroid Build Coastguard Worker }
12934*9880d681SAndroid Build Coastguard Worker
12935*9880d681SAndroid Build Coastguard Worker // If any of the operands is a floating point scalar bitcast to a vector,
12936*9880d681SAndroid Build Coastguard Worker // use floating point types throughout, and bitcast everything.
12937*9880d681SAndroid Build Coastguard Worker // Replace UNDEFs by another scalar UNDEF node, of the final desired type.
12938*9880d681SAndroid Build Coastguard Worker if (AnyFP) {
12939*9880d681SAndroid Build Coastguard Worker SVT = EVT::getFloatingPointVT(OpVT.getSizeInBits());
12940*9880d681SAndroid Build Coastguard Worker ScalarUndef = DAG.getNode(ISD::UNDEF, DL, SVT);
12941*9880d681SAndroid Build Coastguard Worker if (AnyInteger) {
12942*9880d681SAndroid Build Coastguard Worker for (SDValue &Op : Ops) {
12943*9880d681SAndroid Build Coastguard Worker if (Op.getValueType() == SVT)
12944*9880d681SAndroid Build Coastguard Worker continue;
12945*9880d681SAndroid Build Coastguard Worker if (Op.isUndef())
12946*9880d681SAndroid Build Coastguard Worker Op = ScalarUndef;
12947*9880d681SAndroid Build Coastguard Worker else
12948*9880d681SAndroid Build Coastguard Worker Op = DAG.getBitcast(SVT, Op);
12949*9880d681SAndroid Build Coastguard Worker }
12950*9880d681SAndroid Build Coastguard Worker }
12951*9880d681SAndroid Build Coastguard Worker }
12952*9880d681SAndroid Build Coastguard Worker
12953*9880d681SAndroid Build Coastguard Worker EVT VecVT = EVT::getVectorVT(*DAG.getContext(), SVT,
12954*9880d681SAndroid Build Coastguard Worker VT.getSizeInBits() / SVT.getSizeInBits());
12955*9880d681SAndroid Build Coastguard Worker return DAG.getBitcast(VT, DAG.getBuildVector(VecVT, DL, Ops));
12956*9880d681SAndroid Build Coastguard Worker }
12957*9880d681SAndroid Build Coastguard Worker
12958*9880d681SAndroid Build Coastguard Worker // Check to see if this is a CONCAT_VECTORS of a bunch of EXTRACT_SUBVECTOR
12959*9880d681SAndroid Build Coastguard Worker // operations. If so, and if the EXTRACT_SUBVECTOR vector inputs come from at
12960*9880d681SAndroid Build Coastguard Worker // most two distinct vectors the same size as the result, attempt to turn this
12961*9880d681SAndroid Build Coastguard Worker // into a legal shuffle.
combineConcatVectorOfExtracts(SDNode * N,SelectionDAG & DAG)12962*9880d681SAndroid Build Coastguard Worker static SDValue combineConcatVectorOfExtracts(SDNode *N, SelectionDAG &DAG) {
12963*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
12964*9880d681SAndroid Build Coastguard Worker EVT OpVT = N->getOperand(0).getValueType();
12965*9880d681SAndroid Build Coastguard Worker int NumElts = VT.getVectorNumElements();
12966*9880d681SAndroid Build Coastguard Worker int NumOpElts = OpVT.getVectorNumElements();
12967*9880d681SAndroid Build Coastguard Worker
12968*9880d681SAndroid Build Coastguard Worker SDValue SV0 = DAG.getUNDEF(VT), SV1 = DAG.getUNDEF(VT);
12969*9880d681SAndroid Build Coastguard Worker SmallVector<int, 8> Mask;
12970*9880d681SAndroid Build Coastguard Worker
12971*9880d681SAndroid Build Coastguard Worker for (SDValue Op : N->ops()) {
12972*9880d681SAndroid Build Coastguard Worker // Peek through any bitcast.
12973*9880d681SAndroid Build Coastguard Worker while (Op.getOpcode() == ISD::BITCAST)
12974*9880d681SAndroid Build Coastguard Worker Op = Op.getOperand(0);
12975*9880d681SAndroid Build Coastguard Worker
12976*9880d681SAndroid Build Coastguard Worker // UNDEF nodes convert to UNDEF shuffle mask values.
12977*9880d681SAndroid Build Coastguard Worker if (Op.isUndef()) {
12978*9880d681SAndroid Build Coastguard Worker Mask.append((unsigned)NumOpElts, -1);
12979*9880d681SAndroid Build Coastguard Worker continue;
12980*9880d681SAndroid Build Coastguard Worker }
12981*9880d681SAndroid Build Coastguard Worker
12982*9880d681SAndroid Build Coastguard Worker if (Op.getOpcode() != ISD::EXTRACT_SUBVECTOR)
12983*9880d681SAndroid Build Coastguard Worker return SDValue();
12984*9880d681SAndroid Build Coastguard Worker
12985*9880d681SAndroid Build Coastguard Worker // What vector are we extracting the subvector from and at what index?
12986*9880d681SAndroid Build Coastguard Worker SDValue ExtVec = Op.getOperand(0);
12987*9880d681SAndroid Build Coastguard Worker
12988*9880d681SAndroid Build Coastguard Worker // We want the EVT of the original extraction to correctly scale the
12989*9880d681SAndroid Build Coastguard Worker // extraction index.
12990*9880d681SAndroid Build Coastguard Worker EVT ExtVT = ExtVec.getValueType();
12991*9880d681SAndroid Build Coastguard Worker
12992*9880d681SAndroid Build Coastguard Worker // Peek through any bitcast.
12993*9880d681SAndroid Build Coastguard Worker while (ExtVec.getOpcode() == ISD::BITCAST)
12994*9880d681SAndroid Build Coastguard Worker ExtVec = ExtVec.getOperand(0);
12995*9880d681SAndroid Build Coastguard Worker
12996*9880d681SAndroid Build Coastguard Worker // UNDEF nodes convert to UNDEF shuffle mask values.
12997*9880d681SAndroid Build Coastguard Worker if (ExtVec.isUndef()) {
12998*9880d681SAndroid Build Coastguard Worker Mask.append((unsigned)NumOpElts, -1);
12999*9880d681SAndroid Build Coastguard Worker continue;
13000*9880d681SAndroid Build Coastguard Worker }
13001*9880d681SAndroid Build Coastguard Worker
13002*9880d681SAndroid Build Coastguard Worker if (!isa<ConstantSDNode>(Op.getOperand(1)))
13003*9880d681SAndroid Build Coastguard Worker return SDValue();
13004*9880d681SAndroid Build Coastguard Worker int ExtIdx = cast<ConstantSDNode>(Op.getOperand(1))->getZExtValue();
13005*9880d681SAndroid Build Coastguard Worker
13006*9880d681SAndroid Build Coastguard Worker // Ensure that we are extracting a subvector from a vector the same
13007*9880d681SAndroid Build Coastguard Worker // size as the result.
13008*9880d681SAndroid Build Coastguard Worker if (ExtVT.getSizeInBits() != VT.getSizeInBits())
13009*9880d681SAndroid Build Coastguard Worker return SDValue();
13010*9880d681SAndroid Build Coastguard Worker
13011*9880d681SAndroid Build Coastguard Worker // Scale the subvector index to account for any bitcast.
13012*9880d681SAndroid Build Coastguard Worker int NumExtElts = ExtVT.getVectorNumElements();
13013*9880d681SAndroid Build Coastguard Worker if (0 == (NumExtElts % NumElts))
13014*9880d681SAndroid Build Coastguard Worker ExtIdx /= (NumExtElts / NumElts);
13015*9880d681SAndroid Build Coastguard Worker else if (0 == (NumElts % NumExtElts))
13016*9880d681SAndroid Build Coastguard Worker ExtIdx *= (NumElts / NumExtElts);
13017*9880d681SAndroid Build Coastguard Worker else
13018*9880d681SAndroid Build Coastguard Worker return SDValue();
13019*9880d681SAndroid Build Coastguard Worker
13020*9880d681SAndroid Build Coastguard Worker // At most we can reference 2 inputs in the final shuffle.
13021*9880d681SAndroid Build Coastguard Worker if (SV0.isUndef() || SV0 == ExtVec) {
13022*9880d681SAndroid Build Coastguard Worker SV0 = ExtVec;
13023*9880d681SAndroid Build Coastguard Worker for (int i = 0; i != NumOpElts; ++i)
13024*9880d681SAndroid Build Coastguard Worker Mask.push_back(i + ExtIdx);
13025*9880d681SAndroid Build Coastguard Worker } else if (SV1.isUndef() || SV1 == ExtVec) {
13026*9880d681SAndroid Build Coastguard Worker SV1 = ExtVec;
13027*9880d681SAndroid Build Coastguard Worker for (int i = 0; i != NumOpElts; ++i)
13028*9880d681SAndroid Build Coastguard Worker Mask.push_back(i + ExtIdx + NumElts);
13029*9880d681SAndroid Build Coastguard Worker } else {
13030*9880d681SAndroid Build Coastguard Worker return SDValue();
13031*9880d681SAndroid Build Coastguard Worker }
13032*9880d681SAndroid Build Coastguard Worker }
13033*9880d681SAndroid Build Coastguard Worker
13034*9880d681SAndroid Build Coastguard Worker if (!DAG.getTargetLoweringInfo().isShuffleMaskLegal(Mask, VT))
13035*9880d681SAndroid Build Coastguard Worker return SDValue();
13036*9880d681SAndroid Build Coastguard Worker
13037*9880d681SAndroid Build Coastguard Worker return DAG.getVectorShuffle(VT, SDLoc(N), DAG.getBitcast(VT, SV0),
13038*9880d681SAndroid Build Coastguard Worker DAG.getBitcast(VT, SV1), Mask);
13039*9880d681SAndroid Build Coastguard Worker }
13040*9880d681SAndroid Build Coastguard Worker
visitCONCAT_VECTORS(SDNode * N)13041*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitCONCAT_VECTORS(SDNode *N) {
13042*9880d681SAndroid Build Coastguard Worker // If we only have one input vector, we don't need to do any concatenation.
13043*9880d681SAndroid Build Coastguard Worker if (N->getNumOperands() == 1)
13044*9880d681SAndroid Build Coastguard Worker return N->getOperand(0);
13045*9880d681SAndroid Build Coastguard Worker
13046*9880d681SAndroid Build Coastguard Worker // Check if all of the operands are undefs.
13047*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
13048*9880d681SAndroid Build Coastguard Worker if (ISD::allOperandsUndef(N))
13049*9880d681SAndroid Build Coastguard Worker return DAG.getUNDEF(VT);
13050*9880d681SAndroid Build Coastguard Worker
13051*9880d681SAndroid Build Coastguard Worker // Optimize concat_vectors where all but the first of the vectors are undef.
13052*9880d681SAndroid Build Coastguard Worker if (std::all_of(std::next(N->op_begin()), N->op_end(), [](const SDValue &Op) {
13053*9880d681SAndroid Build Coastguard Worker return Op.isUndef();
13054*9880d681SAndroid Build Coastguard Worker })) {
13055*9880d681SAndroid Build Coastguard Worker SDValue In = N->getOperand(0);
13056*9880d681SAndroid Build Coastguard Worker assert(In.getValueType().isVector() && "Must concat vectors");
13057*9880d681SAndroid Build Coastguard Worker
13058*9880d681SAndroid Build Coastguard Worker // Transform: concat_vectors(scalar, undef) -> scalar_to_vector(sclr).
13059*9880d681SAndroid Build Coastguard Worker if (In->getOpcode() == ISD::BITCAST &&
13060*9880d681SAndroid Build Coastguard Worker !In->getOperand(0)->getValueType(0).isVector()) {
13061*9880d681SAndroid Build Coastguard Worker SDValue Scalar = In->getOperand(0);
13062*9880d681SAndroid Build Coastguard Worker
13063*9880d681SAndroid Build Coastguard Worker // If the bitcast type isn't legal, it might be a trunc of a legal type;
13064*9880d681SAndroid Build Coastguard Worker // look through the trunc so we can still do the transform:
13065*9880d681SAndroid Build Coastguard Worker // concat_vectors(trunc(scalar), undef) -> scalar_to_vector(scalar)
13066*9880d681SAndroid Build Coastguard Worker if (Scalar->getOpcode() == ISD::TRUNCATE &&
13067*9880d681SAndroid Build Coastguard Worker !TLI.isTypeLegal(Scalar.getValueType()) &&
13068*9880d681SAndroid Build Coastguard Worker TLI.isTypeLegal(Scalar->getOperand(0).getValueType()))
13069*9880d681SAndroid Build Coastguard Worker Scalar = Scalar->getOperand(0);
13070*9880d681SAndroid Build Coastguard Worker
13071*9880d681SAndroid Build Coastguard Worker EVT SclTy = Scalar->getValueType(0);
13072*9880d681SAndroid Build Coastguard Worker
13073*9880d681SAndroid Build Coastguard Worker if (!SclTy.isFloatingPoint() && !SclTy.isInteger())
13074*9880d681SAndroid Build Coastguard Worker return SDValue();
13075*9880d681SAndroid Build Coastguard Worker
13076*9880d681SAndroid Build Coastguard Worker EVT NVT = EVT::getVectorVT(*DAG.getContext(), SclTy,
13077*9880d681SAndroid Build Coastguard Worker VT.getSizeInBits() / SclTy.getSizeInBits());
13078*9880d681SAndroid Build Coastguard Worker if (!TLI.isTypeLegal(NVT) || !TLI.isTypeLegal(Scalar.getValueType()))
13079*9880d681SAndroid Build Coastguard Worker return SDValue();
13080*9880d681SAndroid Build Coastguard Worker
13081*9880d681SAndroid Build Coastguard Worker SDLoc dl = SDLoc(N);
13082*9880d681SAndroid Build Coastguard Worker SDValue Res = DAG.getNode(ISD::SCALAR_TO_VECTOR, dl, NVT, Scalar);
13083*9880d681SAndroid Build Coastguard Worker return DAG.getBitcast(VT, Res);
13084*9880d681SAndroid Build Coastguard Worker }
13085*9880d681SAndroid Build Coastguard Worker }
13086*9880d681SAndroid Build Coastguard Worker
13087*9880d681SAndroid Build Coastguard Worker // Fold any combination of BUILD_VECTOR or UNDEF nodes into one BUILD_VECTOR.
13088*9880d681SAndroid Build Coastguard Worker // We have already tested above for an UNDEF only concatenation.
13089*9880d681SAndroid Build Coastguard Worker // fold (concat_vectors (BUILD_VECTOR A, B, ...), (BUILD_VECTOR C, D, ...))
13090*9880d681SAndroid Build Coastguard Worker // -> (BUILD_VECTOR A, B, ..., C, D, ...)
13091*9880d681SAndroid Build Coastguard Worker auto IsBuildVectorOrUndef = [](const SDValue &Op) {
13092*9880d681SAndroid Build Coastguard Worker return ISD::UNDEF == Op.getOpcode() || ISD::BUILD_VECTOR == Op.getOpcode();
13093*9880d681SAndroid Build Coastguard Worker };
13094*9880d681SAndroid Build Coastguard Worker if (llvm::all_of(N->ops(), IsBuildVectorOrUndef)) {
13095*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Opnds;
13096*9880d681SAndroid Build Coastguard Worker EVT SVT = VT.getScalarType();
13097*9880d681SAndroid Build Coastguard Worker
13098*9880d681SAndroid Build Coastguard Worker EVT MinVT = SVT;
13099*9880d681SAndroid Build Coastguard Worker if (!SVT.isFloatingPoint()) {
13100*9880d681SAndroid Build Coastguard Worker // If BUILD_VECTOR are from built from integer, they may have different
13101*9880d681SAndroid Build Coastguard Worker // operand types. Get the smallest type and truncate all operands to it.
13102*9880d681SAndroid Build Coastguard Worker bool FoundMinVT = false;
13103*9880d681SAndroid Build Coastguard Worker for (const SDValue &Op : N->ops())
13104*9880d681SAndroid Build Coastguard Worker if (ISD::BUILD_VECTOR == Op.getOpcode()) {
13105*9880d681SAndroid Build Coastguard Worker EVT OpSVT = Op.getOperand(0)->getValueType(0);
13106*9880d681SAndroid Build Coastguard Worker MinVT = (!FoundMinVT || OpSVT.bitsLE(MinVT)) ? OpSVT : MinVT;
13107*9880d681SAndroid Build Coastguard Worker FoundMinVT = true;
13108*9880d681SAndroid Build Coastguard Worker }
13109*9880d681SAndroid Build Coastguard Worker assert(FoundMinVT && "Concat vector type mismatch");
13110*9880d681SAndroid Build Coastguard Worker }
13111*9880d681SAndroid Build Coastguard Worker
13112*9880d681SAndroid Build Coastguard Worker for (const SDValue &Op : N->ops()) {
13113*9880d681SAndroid Build Coastguard Worker EVT OpVT = Op.getValueType();
13114*9880d681SAndroid Build Coastguard Worker unsigned NumElts = OpVT.getVectorNumElements();
13115*9880d681SAndroid Build Coastguard Worker
13116*9880d681SAndroid Build Coastguard Worker if (ISD::UNDEF == Op.getOpcode())
13117*9880d681SAndroid Build Coastguard Worker Opnds.append(NumElts, DAG.getUNDEF(MinVT));
13118*9880d681SAndroid Build Coastguard Worker
13119*9880d681SAndroid Build Coastguard Worker if (ISD::BUILD_VECTOR == Op.getOpcode()) {
13120*9880d681SAndroid Build Coastguard Worker if (SVT.isFloatingPoint()) {
13121*9880d681SAndroid Build Coastguard Worker assert(SVT == OpVT.getScalarType() && "Concat vector type mismatch");
13122*9880d681SAndroid Build Coastguard Worker Opnds.append(Op->op_begin(), Op->op_begin() + NumElts);
13123*9880d681SAndroid Build Coastguard Worker } else {
13124*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i != NumElts; ++i)
13125*9880d681SAndroid Build Coastguard Worker Opnds.push_back(
13126*9880d681SAndroid Build Coastguard Worker DAG.getNode(ISD::TRUNCATE, SDLoc(N), MinVT, Op.getOperand(i)));
13127*9880d681SAndroid Build Coastguard Worker }
13128*9880d681SAndroid Build Coastguard Worker }
13129*9880d681SAndroid Build Coastguard Worker }
13130*9880d681SAndroid Build Coastguard Worker
13131*9880d681SAndroid Build Coastguard Worker assert(VT.getVectorNumElements() == Opnds.size() &&
13132*9880d681SAndroid Build Coastguard Worker "Concat vector type mismatch");
13133*9880d681SAndroid Build Coastguard Worker return DAG.getBuildVector(VT, SDLoc(N), Opnds);
13134*9880d681SAndroid Build Coastguard Worker }
13135*9880d681SAndroid Build Coastguard Worker
13136*9880d681SAndroid Build Coastguard Worker // Fold CONCAT_VECTORS of only bitcast scalars (or undef) to BUILD_VECTOR.
13137*9880d681SAndroid Build Coastguard Worker if (SDValue V = combineConcatVectorOfScalars(N, DAG))
13138*9880d681SAndroid Build Coastguard Worker return V;
13139*9880d681SAndroid Build Coastguard Worker
13140*9880d681SAndroid Build Coastguard Worker // Fold CONCAT_VECTORS of EXTRACT_SUBVECTOR (or undef) to VECTOR_SHUFFLE.
13141*9880d681SAndroid Build Coastguard Worker if (Level < AfterLegalizeVectorOps && TLI.isTypeLegal(VT))
13142*9880d681SAndroid Build Coastguard Worker if (SDValue V = combineConcatVectorOfExtracts(N, DAG))
13143*9880d681SAndroid Build Coastguard Worker return V;
13144*9880d681SAndroid Build Coastguard Worker
13145*9880d681SAndroid Build Coastguard Worker // Type legalization of vectors and DAG canonicalization of SHUFFLE_VECTOR
13146*9880d681SAndroid Build Coastguard Worker // nodes often generate nop CONCAT_VECTOR nodes.
13147*9880d681SAndroid Build Coastguard Worker // Scan the CONCAT_VECTOR operands and look for a CONCAT operations that
13148*9880d681SAndroid Build Coastguard Worker // place the incoming vectors at the exact same location.
13149*9880d681SAndroid Build Coastguard Worker SDValue SingleSource = SDValue();
13150*9880d681SAndroid Build Coastguard Worker unsigned PartNumElem = N->getOperand(0).getValueType().getVectorNumElements();
13151*9880d681SAndroid Build Coastguard Worker
13152*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i) {
13153*9880d681SAndroid Build Coastguard Worker SDValue Op = N->getOperand(i);
13154*9880d681SAndroid Build Coastguard Worker
13155*9880d681SAndroid Build Coastguard Worker if (Op.isUndef())
13156*9880d681SAndroid Build Coastguard Worker continue;
13157*9880d681SAndroid Build Coastguard Worker
13158*9880d681SAndroid Build Coastguard Worker // Check if this is the identity extract:
13159*9880d681SAndroid Build Coastguard Worker if (Op.getOpcode() != ISD::EXTRACT_SUBVECTOR)
13160*9880d681SAndroid Build Coastguard Worker return SDValue();
13161*9880d681SAndroid Build Coastguard Worker
13162*9880d681SAndroid Build Coastguard Worker // Find the single incoming vector for the extract_subvector.
13163*9880d681SAndroid Build Coastguard Worker if (SingleSource.getNode()) {
13164*9880d681SAndroid Build Coastguard Worker if (Op.getOperand(0) != SingleSource)
13165*9880d681SAndroid Build Coastguard Worker return SDValue();
13166*9880d681SAndroid Build Coastguard Worker } else {
13167*9880d681SAndroid Build Coastguard Worker SingleSource = Op.getOperand(0);
13168*9880d681SAndroid Build Coastguard Worker
13169*9880d681SAndroid Build Coastguard Worker // Check the source type is the same as the type of the result.
13170*9880d681SAndroid Build Coastguard Worker // If not, this concat may extend the vector, so we can not
13171*9880d681SAndroid Build Coastguard Worker // optimize it away.
13172*9880d681SAndroid Build Coastguard Worker if (SingleSource.getValueType() != N->getValueType(0))
13173*9880d681SAndroid Build Coastguard Worker return SDValue();
13174*9880d681SAndroid Build Coastguard Worker }
13175*9880d681SAndroid Build Coastguard Worker
13176*9880d681SAndroid Build Coastguard Worker unsigned IdentityIndex = i * PartNumElem;
13177*9880d681SAndroid Build Coastguard Worker ConstantSDNode *CS = dyn_cast<ConstantSDNode>(Op.getOperand(1));
13178*9880d681SAndroid Build Coastguard Worker // The extract index must be constant.
13179*9880d681SAndroid Build Coastguard Worker if (!CS)
13180*9880d681SAndroid Build Coastguard Worker return SDValue();
13181*9880d681SAndroid Build Coastguard Worker
13182*9880d681SAndroid Build Coastguard Worker // Check that we are reading from the identity index.
13183*9880d681SAndroid Build Coastguard Worker if (CS->getZExtValue() != IdentityIndex)
13184*9880d681SAndroid Build Coastguard Worker return SDValue();
13185*9880d681SAndroid Build Coastguard Worker }
13186*9880d681SAndroid Build Coastguard Worker
13187*9880d681SAndroid Build Coastguard Worker if (SingleSource.getNode())
13188*9880d681SAndroid Build Coastguard Worker return SingleSource;
13189*9880d681SAndroid Build Coastguard Worker
13190*9880d681SAndroid Build Coastguard Worker return SDValue();
13191*9880d681SAndroid Build Coastguard Worker }
13192*9880d681SAndroid Build Coastguard Worker
visitEXTRACT_SUBVECTOR(SDNode * N)13193*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitEXTRACT_SUBVECTOR(SDNode* N) {
13194*9880d681SAndroid Build Coastguard Worker EVT NVT = N->getValueType(0);
13195*9880d681SAndroid Build Coastguard Worker SDValue V = N->getOperand(0);
13196*9880d681SAndroid Build Coastguard Worker
13197*9880d681SAndroid Build Coastguard Worker if (V->getOpcode() == ISD::CONCAT_VECTORS) {
13198*9880d681SAndroid Build Coastguard Worker // Combine:
13199*9880d681SAndroid Build Coastguard Worker // (extract_subvec (concat V1, V2, ...), i)
13200*9880d681SAndroid Build Coastguard Worker // Into:
13201*9880d681SAndroid Build Coastguard Worker // Vi if possible
13202*9880d681SAndroid Build Coastguard Worker // Only operand 0 is checked as 'concat' assumes all inputs of the same
13203*9880d681SAndroid Build Coastguard Worker // type.
13204*9880d681SAndroid Build Coastguard Worker if (V->getOperand(0).getValueType() != NVT)
13205*9880d681SAndroid Build Coastguard Worker return SDValue();
13206*9880d681SAndroid Build Coastguard Worker unsigned Idx = N->getConstantOperandVal(1);
13207*9880d681SAndroid Build Coastguard Worker unsigned NumElems = NVT.getVectorNumElements();
13208*9880d681SAndroid Build Coastguard Worker assert((Idx % NumElems) == 0 &&
13209*9880d681SAndroid Build Coastguard Worker "IDX in concat is not a multiple of the result vector length.");
13210*9880d681SAndroid Build Coastguard Worker return V->getOperand(Idx / NumElems);
13211*9880d681SAndroid Build Coastguard Worker }
13212*9880d681SAndroid Build Coastguard Worker
13213*9880d681SAndroid Build Coastguard Worker // Skip bitcasting
13214*9880d681SAndroid Build Coastguard Worker if (V->getOpcode() == ISD::BITCAST)
13215*9880d681SAndroid Build Coastguard Worker V = V.getOperand(0);
13216*9880d681SAndroid Build Coastguard Worker
13217*9880d681SAndroid Build Coastguard Worker if (V->getOpcode() == ISD::INSERT_SUBVECTOR) {
13218*9880d681SAndroid Build Coastguard Worker SDLoc dl(N);
13219*9880d681SAndroid Build Coastguard Worker // Handle only simple case where vector being inserted and vector
13220*9880d681SAndroid Build Coastguard Worker // being extracted are of same type, and are half size of larger vectors.
13221*9880d681SAndroid Build Coastguard Worker EVT BigVT = V->getOperand(0).getValueType();
13222*9880d681SAndroid Build Coastguard Worker EVT SmallVT = V->getOperand(1).getValueType();
13223*9880d681SAndroid Build Coastguard Worker if (!NVT.bitsEq(SmallVT) || NVT.getSizeInBits()*2 != BigVT.getSizeInBits())
13224*9880d681SAndroid Build Coastguard Worker return SDValue();
13225*9880d681SAndroid Build Coastguard Worker
13226*9880d681SAndroid Build Coastguard Worker // Only handle cases where both indexes are constants with the same type.
13227*9880d681SAndroid Build Coastguard Worker ConstantSDNode *ExtIdx = dyn_cast<ConstantSDNode>(N->getOperand(1));
13228*9880d681SAndroid Build Coastguard Worker ConstantSDNode *InsIdx = dyn_cast<ConstantSDNode>(V->getOperand(2));
13229*9880d681SAndroid Build Coastguard Worker
13230*9880d681SAndroid Build Coastguard Worker if (InsIdx && ExtIdx &&
13231*9880d681SAndroid Build Coastguard Worker InsIdx->getValueType(0).getSizeInBits() <= 64 &&
13232*9880d681SAndroid Build Coastguard Worker ExtIdx->getValueType(0).getSizeInBits() <= 64) {
13233*9880d681SAndroid Build Coastguard Worker // Combine:
13234*9880d681SAndroid Build Coastguard Worker // (extract_subvec (insert_subvec V1, V2, InsIdx), ExtIdx)
13235*9880d681SAndroid Build Coastguard Worker // Into:
13236*9880d681SAndroid Build Coastguard Worker // indices are equal or bit offsets are equal => V1
13237*9880d681SAndroid Build Coastguard Worker // otherwise => (extract_subvec V1, ExtIdx)
13238*9880d681SAndroid Build Coastguard Worker if (InsIdx->getZExtValue() * SmallVT.getScalarType().getSizeInBits() ==
13239*9880d681SAndroid Build Coastguard Worker ExtIdx->getZExtValue() * NVT.getScalarType().getSizeInBits())
13240*9880d681SAndroid Build Coastguard Worker return DAG.getBitcast(NVT, V->getOperand(1));
13241*9880d681SAndroid Build Coastguard Worker return DAG.getNode(
13242*9880d681SAndroid Build Coastguard Worker ISD::EXTRACT_SUBVECTOR, dl, NVT,
13243*9880d681SAndroid Build Coastguard Worker DAG.getBitcast(N->getOperand(0).getValueType(), V->getOperand(0)),
13244*9880d681SAndroid Build Coastguard Worker N->getOperand(1));
13245*9880d681SAndroid Build Coastguard Worker }
13246*9880d681SAndroid Build Coastguard Worker }
13247*9880d681SAndroid Build Coastguard Worker
13248*9880d681SAndroid Build Coastguard Worker return SDValue();
13249*9880d681SAndroid Build Coastguard Worker }
13250*9880d681SAndroid Build Coastguard Worker
simplifyShuffleOperandRecursively(SmallBitVector & UsedElements,SDValue V,SelectionDAG & DAG)13251*9880d681SAndroid Build Coastguard Worker static SDValue simplifyShuffleOperandRecursively(SmallBitVector &UsedElements,
13252*9880d681SAndroid Build Coastguard Worker SDValue V, SelectionDAG &DAG) {
13253*9880d681SAndroid Build Coastguard Worker SDLoc DL(V);
13254*9880d681SAndroid Build Coastguard Worker EVT VT = V.getValueType();
13255*9880d681SAndroid Build Coastguard Worker
13256*9880d681SAndroid Build Coastguard Worker switch (V.getOpcode()) {
13257*9880d681SAndroid Build Coastguard Worker default:
13258*9880d681SAndroid Build Coastguard Worker return V;
13259*9880d681SAndroid Build Coastguard Worker
13260*9880d681SAndroid Build Coastguard Worker case ISD::CONCAT_VECTORS: {
13261*9880d681SAndroid Build Coastguard Worker EVT OpVT = V->getOperand(0).getValueType();
13262*9880d681SAndroid Build Coastguard Worker int OpSize = OpVT.getVectorNumElements();
13263*9880d681SAndroid Build Coastguard Worker SmallBitVector OpUsedElements(OpSize, false);
13264*9880d681SAndroid Build Coastguard Worker bool FoundSimplification = false;
13265*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 4> NewOps;
13266*9880d681SAndroid Build Coastguard Worker NewOps.reserve(V->getNumOperands());
13267*9880d681SAndroid Build Coastguard Worker for (int i = 0, NumOps = V->getNumOperands(); i < NumOps; ++i) {
13268*9880d681SAndroid Build Coastguard Worker SDValue Op = V->getOperand(i);
13269*9880d681SAndroid Build Coastguard Worker bool OpUsed = false;
13270*9880d681SAndroid Build Coastguard Worker for (int j = 0; j < OpSize; ++j)
13271*9880d681SAndroid Build Coastguard Worker if (UsedElements[i * OpSize + j]) {
13272*9880d681SAndroid Build Coastguard Worker OpUsedElements[j] = true;
13273*9880d681SAndroid Build Coastguard Worker OpUsed = true;
13274*9880d681SAndroid Build Coastguard Worker }
13275*9880d681SAndroid Build Coastguard Worker NewOps.push_back(
13276*9880d681SAndroid Build Coastguard Worker OpUsed ? simplifyShuffleOperandRecursively(OpUsedElements, Op, DAG)
13277*9880d681SAndroid Build Coastguard Worker : DAG.getUNDEF(OpVT));
13278*9880d681SAndroid Build Coastguard Worker FoundSimplification |= Op == NewOps.back();
13279*9880d681SAndroid Build Coastguard Worker OpUsedElements.reset();
13280*9880d681SAndroid Build Coastguard Worker }
13281*9880d681SAndroid Build Coastguard Worker if (FoundSimplification)
13282*9880d681SAndroid Build Coastguard Worker V = DAG.getNode(ISD::CONCAT_VECTORS, DL, VT, NewOps);
13283*9880d681SAndroid Build Coastguard Worker return V;
13284*9880d681SAndroid Build Coastguard Worker }
13285*9880d681SAndroid Build Coastguard Worker
13286*9880d681SAndroid Build Coastguard Worker case ISD::INSERT_SUBVECTOR: {
13287*9880d681SAndroid Build Coastguard Worker SDValue BaseV = V->getOperand(0);
13288*9880d681SAndroid Build Coastguard Worker SDValue SubV = V->getOperand(1);
13289*9880d681SAndroid Build Coastguard Worker auto *IdxN = dyn_cast<ConstantSDNode>(V->getOperand(2));
13290*9880d681SAndroid Build Coastguard Worker if (!IdxN)
13291*9880d681SAndroid Build Coastguard Worker return V;
13292*9880d681SAndroid Build Coastguard Worker
13293*9880d681SAndroid Build Coastguard Worker int SubSize = SubV.getValueType().getVectorNumElements();
13294*9880d681SAndroid Build Coastguard Worker int Idx = IdxN->getZExtValue();
13295*9880d681SAndroid Build Coastguard Worker bool SubVectorUsed = false;
13296*9880d681SAndroid Build Coastguard Worker SmallBitVector SubUsedElements(SubSize, false);
13297*9880d681SAndroid Build Coastguard Worker for (int i = 0; i < SubSize; ++i)
13298*9880d681SAndroid Build Coastguard Worker if (UsedElements[i + Idx]) {
13299*9880d681SAndroid Build Coastguard Worker SubVectorUsed = true;
13300*9880d681SAndroid Build Coastguard Worker SubUsedElements[i] = true;
13301*9880d681SAndroid Build Coastguard Worker UsedElements[i + Idx] = false;
13302*9880d681SAndroid Build Coastguard Worker }
13303*9880d681SAndroid Build Coastguard Worker
13304*9880d681SAndroid Build Coastguard Worker // Now recurse on both the base and sub vectors.
13305*9880d681SAndroid Build Coastguard Worker SDValue SimplifiedSubV =
13306*9880d681SAndroid Build Coastguard Worker SubVectorUsed
13307*9880d681SAndroid Build Coastguard Worker ? simplifyShuffleOperandRecursively(SubUsedElements, SubV, DAG)
13308*9880d681SAndroid Build Coastguard Worker : DAG.getUNDEF(SubV.getValueType());
13309*9880d681SAndroid Build Coastguard Worker SDValue SimplifiedBaseV = simplifyShuffleOperandRecursively(UsedElements, BaseV, DAG);
13310*9880d681SAndroid Build Coastguard Worker if (SimplifiedSubV != SubV || SimplifiedBaseV != BaseV)
13311*9880d681SAndroid Build Coastguard Worker V = DAG.getNode(ISD::INSERT_SUBVECTOR, DL, VT,
13312*9880d681SAndroid Build Coastguard Worker SimplifiedBaseV, SimplifiedSubV, V->getOperand(2));
13313*9880d681SAndroid Build Coastguard Worker return V;
13314*9880d681SAndroid Build Coastguard Worker }
13315*9880d681SAndroid Build Coastguard Worker }
13316*9880d681SAndroid Build Coastguard Worker }
13317*9880d681SAndroid Build Coastguard Worker
simplifyShuffleOperands(ShuffleVectorSDNode * SVN,SDValue N0,SDValue N1,SelectionDAG & DAG)13318*9880d681SAndroid Build Coastguard Worker static SDValue simplifyShuffleOperands(ShuffleVectorSDNode *SVN, SDValue N0,
13319*9880d681SAndroid Build Coastguard Worker SDValue N1, SelectionDAG &DAG) {
13320*9880d681SAndroid Build Coastguard Worker EVT VT = SVN->getValueType(0);
13321*9880d681SAndroid Build Coastguard Worker int NumElts = VT.getVectorNumElements();
13322*9880d681SAndroid Build Coastguard Worker SmallBitVector N0UsedElements(NumElts, false), N1UsedElements(NumElts, false);
13323*9880d681SAndroid Build Coastguard Worker for (int M : SVN->getMask())
13324*9880d681SAndroid Build Coastguard Worker if (M >= 0 && M < NumElts)
13325*9880d681SAndroid Build Coastguard Worker N0UsedElements[M] = true;
13326*9880d681SAndroid Build Coastguard Worker else if (M >= NumElts)
13327*9880d681SAndroid Build Coastguard Worker N1UsedElements[M - NumElts] = true;
13328*9880d681SAndroid Build Coastguard Worker
13329*9880d681SAndroid Build Coastguard Worker SDValue S0 = simplifyShuffleOperandRecursively(N0UsedElements, N0, DAG);
13330*9880d681SAndroid Build Coastguard Worker SDValue S1 = simplifyShuffleOperandRecursively(N1UsedElements, N1, DAG);
13331*9880d681SAndroid Build Coastguard Worker if (S0 == N0 && S1 == N1)
13332*9880d681SAndroid Build Coastguard Worker return SDValue();
13333*9880d681SAndroid Build Coastguard Worker
13334*9880d681SAndroid Build Coastguard Worker return DAG.getVectorShuffle(VT, SDLoc(SVN), S0, S1, SVN->getMask());
13335*9880d681SAndroid Build Coastguard Worker }
13336*9880d681SAndroid Build Coastguard Worker
13337*9880d681SAndroid Build Coastguard Worker // Tries to turn a shuffle of two CONCAT_VECTORS into a single concat,
13338*9880d681SAndroid Build Coastguard Worker // or turn a shuffle of a single concat into simpler shuffle then concat.
partitionShuffleOfConcats(SDNode * N,SelectionDAG & DAG)13339*9880d681SAndroid Build Coastguard Worker static SDValue partitionShuffleOfConcats(SDNode *N, SelectionDAG &DAG) {
13340*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
13341*9880d681SAndroid Build Coastguard Worker unsigned NumElts = VT.getVectorNumElements();
13342*9880d681SAndroid Build Coastguard Worker
13343*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
13344*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
13345*9880d681SAndroid Build Coastguard Worker ShuffleVectorSDNode *SVN = cast<ShuffleVectorSDNode>(N);
13346*9880d681SAndroid Build Coastguard Worker
13347*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 4> Ops;
13348*9880d681SAndroid Build Coastguard Worker EVT ConcatVT = N0.getOperand(0).getValueType();
13349*9880d681SAndroid Build Coastguard Worker unsigned NumElemsPerConcat = ConcatVT.getVectorNumElements();
13350*9880d681SAndroid Build Coastguard Worker unsigned NumConcats = NumElts / NumElemsPerConcat;
13351*9880d681SAndroid Build Coastguard Worker
13352*9880d681SAndroid Build Coastguard Worker // Special case: shuffle(concat(A,B)) can be more efficiently represented
13353*9880d681SAndroid Build Coastguard Worker // as concat(shuffle(A,B),UNDEF) if the shuffle doesn't set any of the high
13354*9880d681SAndroid Build Coastguard Worker // half vector elements.
13355*9880d681SAndroid Build Coastguard Worker if (NumElemsPerConcat * 2 == NumElts && N1.isUndef() &&
13356*9880d681SAndroid Build Coastguard Worker std::all_of(SVN->getMask().begin() + NumElemsPerConcat,
13357*9880d681SAndroid Build Coastguard Worker SVN->getMask().end(), [](int i) { return i == -1; })) {
13358*9880d681SAndroid Build Coastguard Worker N0 = DAG.getVectorShuffle(ConcatVT, SDLoc(N), N0.getOperand(0), N0.getOperand(1),
13359*9880d681SAndroid Build Coastguard Worker makeArrayRef(SVN->getMask().begin(), NumElemsPerConcat));
13360*9880d681SAndroid Build Coastguard Worker N1 = DAG.getUNDEF(ConcatVT);
13361*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::CONCAT_VECTORS, SDLoc(N), VT, N0, N1);
13362*9880d681SAndroid Build Coastguard Worker }
13363*9880d681SAndroid Build Coastguard Worker
13364*9880d681SAndroid Build Coastguard Worker // Look at every vector that's inserted. We're looking for exact
13365*9880d681SAndroid Build Coastguard Worker // subvector-sized copies from a concatenated vector
13366*9880d681SAndroid Build Coastguard Worker for (unsigned I = 0; I != NumConcats; ++I) {
13367*9880d681SAndroid Build Coastguard Worker // Make sure we're dealing with a copy.
13368*9880d681SAndroid Build Coastguard Worker unsigned Begin = I * NumElemsPerConcat;
13369*9880d681SAndroid Build Coastguard Worker bool AllUndef = true, NoUndef = true;
13370*9880d681SAndroid Build Coastguard Worker for (unsigned J = Begin; J != Begin + NumElemsPerConcat; ++J) {
13371*9880d681SAndroid Build Coastguard Worker if (SVN->getMaskElt(J) >= 0)
13372*9880d681SAndroid Build Coastguard Worker AllUndef = false;
13373*9880d681SAndroid Build Coastguard Worker else
13374*9880d681SAndroid Build Coastguard Worker NoUndef = false;
13375*9880d681SAndroid Build Coastguard Worker }
13376*9880d681SAndroid Build Coastguard Worker
13377*9880d681SAndroid Build Coastguard Worker if (NoUndef) {
13378*9880d681SAndroid Build Coastguard Worker if (SVN->getMaskElt(Begin) % NumElemsPerConcat != 0)
13379*9880d681SAndroid Build Coastguard Worker return SDValue();
13380*9880d681SAndroid Build Coastguard Worker
13381*9880d681SAndroid Build Coastguard Worker for (unsigned J = 1; J != NumElemsPerConcat; ++J)
13382*9880d681SAndroid Build Coastguard Worker if (SVN->getMaskElt(Begin + J - 1) + 1 != SVN->getMaskElt(Begin + J))
13383*9880d681SAndroid Build Coastguard Worker return SDValue();
13384*9880d681SAndroid Build Coastguard Worker
13385*9880d681SAndroid Build Coastguard Worker unsigned FirstElt = SVN->getMaskElt(Begin) / NumElemsPerConcat;
13386*9880d681SAndroid Build Coastguard Worker if (FirstElt < N0.getNumOperands())
13387*9880d681SAndroid Build Coastguard Worker Ops.push_back(N0.getOperand(FirstElt));
13388*9880d681SAndroid Build Coastguard Worker else
13389*9880d681SAndroid Build Coastguard Worker Ops.push_back(N1.getOperand(FirstElt - N0.getNumOperands()));
13390*9880d681SAndroid Build Coastguard Worker
13391*9880d681SAndroid Build Coastguard Worker } else if (AllUndef) {
13392*9880d681SAndroid Build Coastguard Worker Ops.push_back(DAG.getUNDEF(N0.getOperand(0).getValueType()));
13393*9880d681SAndroid Build Coastguard Worker } else { // Mixed with general masks and undefs, can't do optimization.
13394*9880d681SAndroid Build Coastguard Worker return SDValue();
13395*9880d681SAndroid Build Coastguard Worker }
13396*9880d681SAndroid Build Coastguard Worker }
13397*9880d681SAndroid Build Coastguard Worker
13398*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::CONCAT_VECTORS, SDLoc(N), VT, Ops);
13399*9880d681SAndroid Build Coastguard Worker }
13400*9880d681SAndroid Build Coastguard Worker
visitVECTOR_SHUFFLE(SDNode * N)13401*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitVECTOR_SHUFFLE(SDNode *N) {
13402*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
13403*9880d681SAndroid Build Coastguard Worker unsigned NumElts = VT.getVectorNumElements();
13404*9880d681SAndroid Build Coastguard Worker
13405*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
13406*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
13407*9880d681SAndroid Build Coastguard Worker
13408*9880d681SAndroid Build Coastguard Worker assert(N0.getValueType() == VT && "Vector shuffle must be normalized in DAG");
13409*9880d681SAndroid Build Coastguard Worker
13410*9880d681SAndroid Build Coastguard Worker // Canonicalize shuffle undef, undef -> undef
13411*9880d681SAndroid Build Coastguard Worker if (N0.isUndef() && N1.isUndef())
13412*9880d681SAndroid Build Coastguard Worker return DAG.getUNDEF(VT);
13413*9880d681SAndroid Build Coastguard Worker
13414*9880d681SAndroid Build Coastguard Worker ShuffleVectorSDNode *SVN = cast<ShuffleVectorSDNode>(N);
13415*9880d681SAndroid Build Coastguard Worker
13416*9880d681SAndroid Build Coastguard Worker // Canonicalize shuffle v, v -> v, undef
13417*9880d681SAndroid Build Coastguard Worker if (N0 == N1) {
13418*9880d681SAndroid Build Coastguard Worker SmallVector<int, 8> NewMask;
13419*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i != NumElts; ++i) {
13420*9880d681SAndroid Build Coastguard Worker int Idx = SVN->getMaskElt(i);
13421*9880d681SAndroid Build Coastguard Worker if (Idx >= (int)NumElts) Idx -= NumElts;
13422*9880d681SAndroid Build Coastguard Worker NewMask.push_back(Idx);
13423*9880d681SAndroid Build Coastguard Worker }
13424*9880d681SAndroid Build Coastguard Worker return DAG.getVectorShuffle(VT, SDLoc(N), N0, DAG.getUNDEF(VT), NewMask);
13425*9880d681SAndroid Build Coastguard Worker }
13426*9880d681SAndroid Build Coastguard Worker
13427*9880d681SAndroid Build Coastguard Worker // Canonicalize shuffle undef, v -> v, undef. Commute the shuffle mask.
13428*9880d681SAndroid Build Coastguard Worker if (N0.isUndef())
13429*9880d681SAndroid Build Coastguard Worker return DAG.getCommutedVectorShuffle(*SVN);
13430*9880d681SAndroid Build Coastguard Worker
13431*9880d681SAndroid Build Coastguard Worker // Remove references to rhs if it is undef
13432*9880d681SAndroid Build Coastguard Worker if (N1.isUndef()) {
13433*9880d681SAndroid Build Coastguard Worker bool Changed = false;
13434*9880d681SAndroid Build Coastguard Worker SmallVector<int, 8> NewMask;
13435*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i != NumElts; ++i) {
13436*9880d681SAndroid Build Coastguard Worker int Idx = SVN->getMaskElt(i);
13437*9880d681SAndroid Build Coastguard Worker if (Idx >= (int)NumElts) {
13438*9880d681SAndroid Build Coastguard Worker Idx = -1;
13439*9880d681SAndroid Build Coastguard Worker Changed = true;
13440*9880d681SAndroid Build Coastguard Worker }
13441*9880d681SAndroid Build Coastguard Worker NewMask.push_back(Idx);
13442*9880d681SAndroid Build Coastguard Worker }
13443*9880d681SAndroid Build Coastguard Worker if (Changed)
13444*9880d681SAndroid Build Coastguard Worker return DAG.getVectorShuffle(VT, SDLoc(N), N0, N1, NewMask);
13445*9880d681SAndroid Build Coastguard Worker }
13446*9880d681SAndroid Build Coastguard Worker
13447*9880d681SAndroid Build Coastguard Worker // If it is a splat, check if the argument vector is another splat or a
13448*9880d681SAndroid Build Coastguard Worker // build_vector.
13449*9880d681SAndroid Build Coastguard Worker if (SVN->isSplat() && SVN->getSplatIndex() < (int)NumElts) {
13450*9880d681SAndroid Build Coastguard Worker SDNode *V = N0.getNode();
13451*9880d681SAndroid Build Coastguard Worker
13452*9880d681SAndroid Build Coastguard Worker // If this is a bit convert that changes the element type of the vector but
13453*9880d681SAndroid Build Coastguard Worker // not the number of vector elements, look through it. Be careful not to
13454*9880d681SAndroid Build Coastguard Worker // look though conversions that change things like v4f32 to v2f64.
13455*9880d681SAndroid Build Coastguard Worker if (V->getOpcode() == ISD::BITCAST) {
13456*9880d681SAndroid Build Coastguard Worker SDValue ConvInput = V->getOperand(0);
13457*9880d681SAndroid Build Coastguard Worker if (ConvInput.getValueType().isVector() &&
13458*9880d681SAndroid Build Coastguard Worker ConvInput.getValueType().getVectorNumElements() == NumElts)
13459*9880d681SAndroid Build Coastguard Worker V = ConvInput.getNode();
13460*9880d681SAndroid Build Coastguard Worker }
13461*9880d681SAndroid Build Coastguard Worker
13462*9880d681SAndroid Build Coastguard Worker if (V->getOpcode() == ISD::BUILD_VECTOR) {
13463*9880d681SAndroid Build Coastguard Worker assert(V->getNumOperands() == NumElts &&
13464*9880d681SAndroid Build Coastguard Worker "BUILD_VECTOR has wrong number of operands");
13465*9880d681SAndroid Build Coastguard Worker SDValue Base;
13466*9880d681SAndroid Build Coastguard Worker bool AllSame = true;
13467*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i != NumElts; ++i) {
13468*9880d681SAndroid Build Coastguard Worker if (!V->getOperand(i).isUndef()) {
13469*9880d681SAndroid Build Coastguard Worker Base = V->getOperand(i);
13470*9880d681SAndroid Build Coastguard Worker break;
13471*9880d681SAndroid Build Coastguard Worker }
13472*9880d681SAndroid Build Coastguard Worker }
13473*9880d681SAndroid Build Coastguard Worker // Splat of <u, u, u, u>, return <u, u, u, u>
13474*9880d681SAndroid Build Coastguard Worker if (!Base.getNode())
13475*9880d681SAndroid Build Coastguard Worker return N0;
13476*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i != NumElts; ++i) {
13477*9880d681SAndroid Build Coastguard Worker if (V->getOperand(i) != Base) {
13478*9880d681SAndroid Build Coastguard Worker AllSame = false;
13479*9880d681SAndroid Build Coastguard Worker break;
13480*9880d681SAndroid Build Coastguard Worker }
13481*9880d681SAndroid Build Coastguard Worker }
13482*9880d681SAndroid Build Coastguard Worker // Splat of <x, x, x, x>, return <x, x, x, x>
13483*9880d681SAndroid Build Coastguard Worker if (AllSame)
13484*9880d681SAndroid Build Coastguard Worker return N0;
13485*9880d681SAndroid Build Coastguard Worker
13486*9880d681SAndroid Build Coastguard Worker // Canonicalize any other splat as a build_vector.
13487*9880d681SAndroid Build Coastguard Worker const SDValue &Splatted = V->getOperand(SVN->getSplatIndex());
13488*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Ops(NumElts, Splatted);
13489*9880d681SAndroid Build Coastguard Worker SDValue NewBV = DAG.getBuildVector(V->getValueType(0), SDLoc(N), Ops);
13490*9880d681SAndroid Build Coastguard Worker
13491*9880d681SAndroid Build Coastguard Worker // We may have jumped through bitcasts, so the type of the
13492*9880d681SAndroid Build Coastguard Worker // BUILD_VECTOR may not match the type of the shuffle.
13493*9880d681SAndroid Build Coastguard Worker if (V->getValueType(0) != VT)
13494*9880d681SAndroid Build Coastguard Worker NewBV = DAG.getBitcast(VT, NewBV);
13495*9880d681SAndroid Build Coastguard Worker return NewBV;
13496*9880d681SAndroid Build Coastguard Worker }
13497*9880d681SAndroid Build Coastguard Worker }
13498*9880d681SAndroid Build Coastguard Worker
13499*9880d681SAndroid Build Coastguard Worker // There are various patterns used to build up a vector from smaller vectors,
13500*9880d681SAndroid Build Coastguard Worker // subvectors, or elements. Scan chains of these and replace unused insertions
13501*9880d681SAndroid Build Coastguard Worker // or components with undef.
13502*9880d681SAndroid Build Coastguard Worker if (SDValue S = simplifyShuffleOperands(SVN, N0, N1, DAG))
13503*9880d681SAndroid Build Coastguard Worker return S;
13504*9880d681SAndroid Build Coastguard Worker
13505*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::CONCAT_VECTORS &&
13506*9880d681SAndroid Build Coastguard Worker Level < AfterLegalizeVectorOps &&
13507*9880d681SAndroid Build Coastguard Worker (N1.isUndef() ||
13508*9880d681SAndroid Build Coastguard Worker (N1.getOpcode() == ISD::CONCAT_VECTORS &&
13509*9880d681SAndroid Build Coastguard Worker N0.getOperand(0).getValueType() == N1.getOperand(0).getValueType()))) {
13510*9880d681SAndroid Build Coastguard Worker if (SDValue V = partitionShuffleOfConcats(N, DAG))
13511*9880d681SAndroid Build Coastguard Worker return V;
13512*9880d681SAndroid Build Coastguard Worker }
13513*9880d681SAndroid Build Coastguard Worker
13514*9880d681SAndroid Build Coastguard Worker // Attempt to combine a shuffle of 2 inputs of 'scalar sources' -
13515*9880d681SAndroid Build Coastguard Worker // BUILD_VECTOR or SCALAR_TO_VECTOR into a single BUILD_VECTOR.
13516*9880d681SAndroid Build Coastguard Worker if (Level < AfterLegalizeVectorOps && TLI.isTypeLegal(VT)) {
13517*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Ops;
13518*9880d681SAndroid Build Coastguard Worker for (int M : SVN->getMask()) {
13519*9880d681SAndroid Build Coastguard Worker SDValue Op = DAG.getUNDEF(VT.getScalarType());
13520*9880d681SAndroid Build Coastguard Worker if (M >= 0) {
13521*9880d681SAndroid Build Coastguard Worker int Idx = M % NumElts;
13522*9880d681SAndroid Build Coastguard Worker SDValue &S = (M < (int)NumElts ? N0 : N1);
13523*9880d681SAndroid Build Coastguard Worker if (S.getOpcode() == ISD::BUILD_VECTOR && S.hasOneUse()) {
13524*9880d681SAndroid Build Coastguard Worker Op = S.getOperand(Idx);
13525*9880d681SAndroid Build Coastguard Worker } else if (S.getOpcode() == ISD::SCALAR_TO_VECTOR && S.hasOneUse()) {
13526*9880d681SAndroid Build Coastguard Worker if (Idx == 0)
13527*9880d681SAndroid Build Coastguard Worker Op = S.getOperand(0);
13528*9880d681SAndroid Build Coastguard Worker } else {
13529*9880d681SAndroid Build Coastguard Worker // Operand can't be combined - bail out.
13530*9880d681SAndroid Build Coastguard Worker break;
13531*9880d681SAndroid Build Coastguard Worker }
13532*9880d681SAndroid Build Coastguard Worker }
13533*9880d681SAndroid Build Coastguard Worker Ops.push_back(Op);
13534*9880d681SAndroid Build Coastguard Worker }
13535*9880d681SAndroid Build Coastguard Worker if (Ops.size() == VT.getVectorNumElements()) {
13536*9880d681SAndroid Build Coastguard Worker // BUILD_VECTOR requires all inputs to be of the same type, find the
13537*9880d681SAndroid Build Coastguard Worker // maximum type and extend them all.
13538*9880d681SAndroid Build Coastguard Worker EVT SVT = VT.getScalarType();
13539*9880d681SAndroid Build Coastguard Worker if (SVT.isInteger())
13540*9880d681SAndroid Build Coastguard Worker for (SDValue &Op : Ops)
13541*9880d681SAndroid Build Coastguard Worker SVT = (SVT.bitsLT(Op.getValueType()) ? Op.getValueType() : SVT);
13542*9880d681SAndroid Build Coastguard Worker if (SVT != VT.getScalarType())
13543*9880d681SAndroid Build Coastguard Worker for (SDValue &Op : Ops)
13544*9880d681SAndroid Build Coastguard Worker Op = TLI.isZExtFree(Op.getValueType(), SVT)
13545*9880d681SAndroid Build Coastguard Worker ? DAG.getZExtOrTrunc(Op, SDLoc(N), SVT)
13546*9880d681SAndroid Build Coastguard Worker : DAG.getSExtOrTrunc(Op, SDLoc(N), SVT);
13547*9880d681SAndroid Build Coastguard Worker return DAG.getBuildVector(VT, SDLoc(N), Ops);
13548*9880d681SAndroid Build Coastguard Worker }
13549*9880d681SAndroid Build Coastguard Worker }
13550*9880d681SAndroid Build Coastguard Worker
13551*9880d681SAndroid Build Coastguard Worker // If this shuffle only has a single input that is a bitcasted shuffle,
13552*9880d681SAndroid Build Coastguard Worker // attempt to merge the 2 shuffles and suitably bitcast the inputs/output
13553*9880d681SAndroid Build Coastguard Worker // back to their original types.
13554*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::BITCAST && N0.hasOneUse() &&
13555*9880d681SAndroid Build Coastguard Worker N1.isUndef() && Level < AfterLegalizeVectorOps &&
13556*9880d681SAndroid Build Coastguard Worker TLI.isTypeLegal(VT)) {
13557*9880d681SAndroid Build Coastguard Worker
13558*9880d681SAndroid Build Coastguard Worker // Peek through the bitcast only if there is one user.
13559*9880d681SAndroid Build Coastguard Worker SDValue BC0 = N0;
13560*9880d681SAndroid Build Coastguard Worker while (BC0.getOpcode() == ISD::BITCAST) {
13561*9880d681SAndroid Build Coastguard Worker if (!BC0.hasOneUse())
13562*9880d681SAndroid Build Coastguard Worker break;
13563*9880d681SAndroid Build Coastguard Worker BC0 = BC0.getOperand(0);
13564*9880d681SAndroid Build Coastguard Worker }
13565*9880d681SAndroid Build Coastguard Worker
13566*9880d681SAndroid Build Coastguard Worker auto ScaleShuffleMask = [](ArrayRef<int> Mask, int Scale) {
13567*9880d681SAndroid Build Coastguard Worker if (Scale == 1)
13568*9880d681SAndroid Build Coastguard Worker return SmallVector<int, 8>(Mask.begin(), Mask.end());
13569*9880d681SAndroid Build Coastguard Worker
13570*9880d681SAndroid Build Coastguard Worker SmallVector<int, 8> NewMask;
13571*9880d681SAndroid Build Coastguard Worker for (int M : Mask)
13572*9880d681SAndroid Build Coastguard Worker for (int s = 0; s != Scale; ++s)
13573*9880d681SAndroid Build Coastguard Worker NewMask.push_back(M < 0 ? -1 : Scale * M + s);
13574*9880d681SAndroid Build Coastguard Worker return NewMask;
13575*9880d681SAndroid Build Coastguard Worker };
13576*9880d681SAndroid Build Coastguard Worker
13577*9880d681SAndroid Build Coastguard Worker if (BC0.getOpcode() == ISD::VECTOR_SHUFFLE && BC0.hasOneUse()) {
13578*9880d681SAndroid Build Coastguard Worker EVT SVT = VT.getScalarType();
13579*9880d681SAndroid Build Coastguard Worker EVT InnerVT = BC0->getValueType(0);
13580*9880d681SAndroid Build Coastguard Worker EVT InnerSVT = InnerVT.getScalarType();
13581*9880d681SAndroid Build Coastguard Worker
13582*9880d681SAndroid Build Coastguard Worker // Determine which shuffle works with the smaller scalar type.
13583*9880d681SAndroid Build Coastguard Worker EVT ScaleVT = SVT.bitsLT(InnerSVT) ? VT : InnerVT;
13584*9880d681SAndroid Build Coastguard Worker EVT ScaleSVT = ScaleVT.getScalarType();
13585*9880d681SAndroid Build Coastguard Worker
13586*9880d681SAndroid Build Coastguard Worker if (TLI.isTypeLegal(ScaleVT) &&
13587*9880d681SAndroid Build Coastguard Worker 0 == (InnerSVT.getSizeInBits() % ScaleSVT.getSizeInBits()) &&
13588*9880d681SAndroid Build Coastguard Worker 0 == (SVT.getSizeInBits() % ScaleSVT.getSizeInBits())) {
13589*9880d681SAndroid Build Coastguard Worker
13590*9880d681SAndroid Build Coastguard Worker int InnerScale = InnerSVT.getSizeInBits() / ScaleSVT.getSizeInBits();
13591*9880d681SAndroid Build Coastguard Worker int OuterScale = SVT.getSizeInBits() / ScaleSVT.getSizeInBits();
13592*9880d681SAndroid Build Coastguard Worker
13593*9880d681SAndroid Build Coastguard Worker // Scale the shuffle masks to the smaller scalar type.
13594*9880d681SAndroid Build Coastguard Worker ShuffleVectorSDNode *InnerSVN = cast<ShuffleVectorSDNode>(BC0);
13595*9880d681SAndroid Build Coastguard Worker SmallVector<int, 8> InnerMask =
13596*9880d681SAndroid Build Coastguard Worker ScaleShuffleMask(InnerSVN->getMask(), InnerScale);
13597*9880d681SAndroid Build Coastguard Worker SmallVector<int, 8> OuterMask =
13598*9880d681SAndroid Build Coastguard Worker ScaleShuffleMask(SVN->getMask(), OuterScale);
13599*9880d681SAndroid Build Coastguard Worker
13600*9880d681SAndroid Build Coastguard Worker // Merge the shuffle masks.
13601*9880d681SAndroid Build Coastguard Worker SmallVector<int, 8> NewMask;
13602*9880d681SAndroid Build Coastguard Worker for (int M : OuterMask)
13603*9880d681SAndroid Build Coastguard Worker NewMask.push_back(M < 0 ? -1 : InnerMask[M]);
13604*9880d681SAndroid Build Coastguard Worker
13605*9880d681SAndroid Build Coastguard Worker // Test for shuffle mask legality over both commutations.
13606*9880d681SAndroid Build Coastguard Worker SDValue SV0 = BC0->getOperand(0);
13607*9880d681SAndroid Build Coastguard Worker SDValue SV1 = BC0->getOperand(1);
13608*9880d681SAndroid Build Coastguard Worker bool LegalMask = TLI.isShuffleMaskLegal(NewMask, ScaleVT);
13609*9880d681SAndroid Build Coastguard Worker if (!LegalMask) {
13610*9880d681SAndroid Build Coastguard Worker std::swap(SV0, SV1);
13611*9880d681SAndroid Build Coastguard Worker ShuffleVectorSDNode::commuteMask(NewMask);
13612*9880d681SAndroid Build Coastguard Worker LegalMask = TLI.isShuffleMaskLegal(NewMask, ScaleVT);
13613*9880d681SAndroid Build Coastguard Worker }
13614*9880d681SAndroid Build Coastguard Worker
13615*9880d681SAndroid Build Coastguard Worker if (LegalMask) {
13616*9880d681SAndroid Build Coastguard Worker SV0 = DAG.getBitcast(ScaleVT, SV0);
13617*9880d681SAndroid Build Coastguard Worker SV1 = DAG.getBitcast(ScaleVT, SV1);
13618*9880d681SAndroid Build Coastguard Worker return DAG.getBitcast(
13619*9880d681SAndroid Build Coastguard Worker VT, DAG.getVectorShuffle(ScaleVT, SDLoc(N), SV0, SV1, NewMask));
13620*9880d681SAndroid Build Coastguard Worker }
13621*9880d681SAndroid Build Coastguard Worker }
13622*9880d681SAndroid Build Coastguard Worker }
13623*9880d681SAndroid Build Coastguard Worker }
13624*9880d681SAndroid Build Coastguard Worker
13625*9880d681SAndroid Build Coastguard Worker // Canonicalize shuffles according to rules:
13626*9880d681SAndroid Build Coastguard Worker // shuffle(A, shuffle(A, B)) -> shuffle(shuffle(A,B), A)
13627*9880d681SAndroid Build Coastguard Worker // shuffle(B, shuffle(A, B)) -> shuffle(shuffle(A,B), B)
13628*9880d681SAndroid Build Coastguard Worker // shuffle(B, shuffle(A, Undef)) -> shuffle(shuffle(A, Undef), B)
13629*9880d681SAndroid Build Coastguard Worker if (N1.getOpcode() == ISD::VECTOR_SHUFFLE &&
13630*9880d681SAndroid Build Coastguard Worker N0.getOpcode() != ISD::VECTOR_SHUFFLE && Level < AfterLegalizeDAG &&
13631*9880d681SAndroid Build Coastguard Worker TLI.isTypeLegal(VT)) {
13632*9880d681SAndroid Build Coastguard Worker // The incoming shuffle must be of the same type as the result of the
13633*9880d681SAndroid Build Coastguard Worker // current shuffle.
13634*9880d681SAndroid Build Coastguard Worker assert(N1->getOperand(0).getValueType() == VT &&
13635*9880d681SAndroid Build Coastguard Worker "Shuffle types don't match");
13636*9880d681SAndroid Build Coastguard Worker
13637*9880d681SAndroid Build Coastguard Worker SDValue SV0 = N1->getOperand(0);
13638*9880d681SAndroid Build Coastguard Worker SDValue SV1 = N1->getOperand(1);
13639*9880d681SAndroid Build Coastguard Worker bool HasSameOp0 = N0 == SV0;
13640*9880d681SAndroid Build Coastguard Worker bool IsSV1Undef = SV1.isUndef();
13641*9880d681SAndroid Build Coastguard Worker if (HasSameOp0 || IsSV1Undef || N0 == SV1)
13642*9880d681SAndroid Build Coastguard Worker // Commute the operands of this shuffle so that next rule
13643*9880d681SAndroid Build Coastguard Worker // will trigger.
13644*9880d681SAndroid Build Coastguard Worker return DAG.getCommutedVectorShuffle(*SVN);
13645*9880d681SAndroid Build Coastguard Worker }
13646*9880d681SAndroid Build Coastguard Worker
13647*9880d681SAndroid Build Coastguard Worker // Try to fold according to rules:
13648*9880d681SAndroid Build Coastguard Worker // shuffle(shuffle(A, B, M0), C, M1) -> shuffle(A, B, M2)
13649*9880d681SAndroid Build Coastguard Worker // shuffle(shuffle(A, B, M0), C, M1) -> shuffle(A, C, M2)
13650*9880d681SAndroid Build Coastguard Worker // shuffle(shuffle(A, B, M0), C, M1) -> shuffle(B, C, M2)
13651*9880d681SAndroid Build Coastguard Worker // Don't try to fold shuffles with illegal type.
13652*9880d681SAndroid Build Coastguard Worker // Only fold if this shuffle is the only user of the other shuffle.
13653*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::VECTOR_SHUFFLE && N->isOnlyUserOf(N0.getNode()) &&
13654*9880d681SAndroid Build Coastguard Worker Level < AfterLegalizeDAG && TLI.isTypeLegal(VT)) {
13655*9880d681SAndroid Build Coastguard Worker ShuffleVectorSDNode *OtherSV = cast<ShuffleVectorSDNode>(N0);
13656*9880d681SAndroid Build Coastguard Worker
13657*9880d681SAndroid Build Coastguard Worker // The incoming shuffle must be of the same type as the result of the
13658*9880d681SAndroid Build Coastguard Worker // current shuffle.
13659*9880d681SAndroid Build Coastguard Worker assert(OtherSV->getOperand(0).getValueType() == VT &&
13660*9880d681SAndroid Build Coastguard Worker "Shuffle types don't match");
13661*9880d681SAndroid Build Coastguard Worker
13662*9880d681SAndroid Build Coastguard Worker SDValue SV0, SV1;
13663*9880d681SAndroid Build Coastguard Worker SmallVector<int, 4> Mask;
13664*9880d681SAndroid Build Coastguard Worker // Compute the combined shuffle mask for a shuffle with SV0 as the first
13665*9880d681SAndroid Build Coastguard Worker // operand, and SV1 as the second operand.
13666*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i != NumElts; ++i) {
13667*9880d681SAndroid Build Coastguard Worker int Idx = SVN->getMaskElt(i);
13668*9880d681SAndroid Build Coastguard Worker if (Idx < 0) {
13669*9880d681SAndroid Build Coastguard Worker // Propagate Undef.
13670*9880d681SAndroid Build Coastguard Worker Mask.push_back(Idx);
13671*9880d681SAndroid Build Coastguard Worker continue;
13672*9880d681SAndroid Build Coastguard Worker }
13673*9880d681SAndroid Build Coastguard Worker
13674*9880d681SAndroid Build Coastguard Worker SDValue CurrentVec;
13675*9880d681SAndroid Build Coastguard Worker if (Idx < (int)NumElts) {
13676*9880d681SAndroid Build Coastguard Worker // This shuffle index refers to the inner shuffle N0. Lookup the inner
13677*9880d681SAndroid Build Coastguard Worker // shuffle mask to identify which vector is actually referenced.
13678*9880d681SAndroid Build Coastguard Worker Idx = OtherSV->getMaskElt(Idx);
13679*9880d681SAndroid Build Coastguard Worker if (Idx < 0) {
13680*9880d681SAndroid Build Coastguard Worker // Propagate Undef.
13681*9880d681SAndroid Build Coastguard Worker Mask.push_back(Idx);
13682*9880d681SAndroid Build Coastguard Worker continue;
13683*9880d681SAndroid Build Coastguard Worker }
13684*9880d681SAndroid Build Coastguard Worker
13685*9880d681SAndroid Build Coastguard Worker CurrentVec = (Idx < (int) NumElts) ? OtherSV->getOperand(0)
13686*9880d681SAndroid Build Coastguard Worker : OtherSV->getOperand(1);
13687*9880d681SAndroid Build Coastguard Worker } else {
13688*9880d681SAndroid Build Coastguard Worker // This shuffle index references an element within N1.
13689*9880d681SAndroid Build Coastguard Worker CurrentVec = N1;
13690*9880d681SAndroid Build Coastguard Worker }
13691*9880d681SAndroid Build Coastguard Worker
13692*9880d681SAndroid Build Coastguard Worker // Simple case where 'CurrentVec' is UNDEF.
13693*9880d681SAndroid Build Coastguard Worker if (CurrentVec.isUndef()) {
13694*9880d681SAndroid Build Coastguard Worker Mask.push_back(-1);
13695*9880d681SAndroid Build Coastguard Worker continue;
13696*9880d681SAndroid Build Coastguard Worker }
13697*9880d681SAndroid Build Coastguard Worker
13698*9880d681SAndroid Build Coastguard Worker // Canonicalize the shuffle index. We don't know yet if CurrentVec
13699*9880d681SAndroid Build Coastguard Worker // will be the first or second operand of the combined shuffle.
13700*9880d681SAndroid Build Coastguard Worker Idx = Idx % NumElts;
13701*9880d681SAndroid Build Coastguard Worker if (!SV0.getNode() || SV0 == CurrentVec) {
13702*9880d681SAndroid Build Coastguard Worker // Ok. CurrentVec is the left hand side.
13703*9880d681SAndroid Build Coastguard Worker // Update the mask accordingly.
13704*9880d681SAndroid Build Coastguard Worker SV0 = CurrentVec;
13705*9880d681SAndroid Build Coastguard Worker Mask.push_back(Idx);
13706*9880d681SAndroid Build Coastguard Worker continue;
13707*9880d681SAndroid Build Coastguard Worker }
13708*9880d681SAndroid Build Coastguard Worker
13709*9880d681SAndroid Build Coastguard Worker // Bail out if we cannot convert the shuffle pair into a single shuffle.
13710*9880d681SAndroid Build Coastguard Worker if (SV1.getNode() && SV1 != CurrentVec)
13711*9880d681SAndroid Build Coastguard Worker return SDValue();
13712*9880d681SAndroid Build Coastguard Worker
13713*9880d681SAndroid Build Coastguard Worker // Ok. CurrentVec is the right hand side.
13714*9880d681SAndroid Build Coastguard Worker // Update the mask accordingly.
13715*9880d681SAndroid Build Coastguard Worker SV1 = CurrentVec;
13716*9880d681SAndroid Build Coastguard Worker Mask.push_back(Idx + NumElts);
13717*9880d681SAndroid Build Coastguard Worker }
13718*9880d681SAndroid Build Coastguard Worker
13719*9880d681SAndroid Build Coastguard Worker // Check if all indices in Mask are Undef. In case, propagate Undef.
13720*9880d681SAndroid Build Coastguard Worker bool isUndefMask = true;
13721*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i != NumElts && isUndefMask; ++i)
13722*9880d681SAndroid Build Coastguard Worker isUndefMask &= Mask[i] < 0;
13723*9880d681SAndroid Build Coastguard Worker
13724*9880d681SAndroid Build Coastguard Worker if (isUndefMask)
13725*9880d681SAndroid Build Coastguard Worker return DAG.getUNDEF(VT);
13726*9880d681SAndroid Build Coastguard Worker
13727*9880d681SAndroid Build Coastguard Worker if (!SV0.getNode())
13728*9880d681SAndroid Build Coastguard Worker SV0 = DAG.getUNDEF(VT);
13729*9880d681SAndroid Build Coastguard Worker if (!SV1.getNode())
13730*9880d681SAndroid Build Coastguard Worker SV1 = DAG.getUNDEF(VT);
13731*9880d681SAndroid Build Coastguard Worker
13732*9880d681SAndroid Build Coastguard Worker // Avoid introducing shuffles with illegal mask.
13733*9880d681SAndroid Build Coastguard Worker if (!TLI.isShuffleMaskLegal(Mask, VT)) {
13734*9880d681SAndroid Build Coastguard Worker ShuffleVectorSDNode::commuteMask(Mask);
13735*9880d681SAndroid Build Coastguard Worker
13736*9880d681SAndroid Build Coastguard Worker if (!TLI.isShuffleMaskLegal(Mask, VT))
13737*9880d681SAndroid Build Coastguard Worker return SDValue();
13738*9880d681SAndroid Build Coastguard Worker
13739*9880d681SAndroid Build Coastguard Worker // shuffle(shuffle(A, B, M0), C, M1) -> shuffle(B, A, M2)
13740*9880d681SAndroid Build Coastguard Worker // shuffle(shuffle(A, B, M0), C, M1) -> shuffle(C, A, M2)
13741*9880d681SAndroid Build Coastguard Worker // shuffle(shuffle(A, B, M0), C, M1) -> shuffle(C, B, M2)
13742*9880d681SAndroid Build Coastguard Worker std::swap(SV0, SV1);
13743*9880d681SAndroid Build Coastguard Worker }
13744*9880d681SAndroid Build Coastguard Worker
13745*9880d681SAndroid Build Coastguard Worker // shuffle(shuffle(A, B, M0), C, M1) -> shuffle(A, B, M2)
13746*9880d681SAndroid Build Coastguard Worker // shuffle(shuffle(A, B, M0), C, M1) -> shuffle(A, C, M2)
13747*9880d681SAndroid Build Coastguard Worker // shuffle(shuffle(A, B, M0), C, M1) -> shuffle(B, C, M2)
13748*9880d681SAndroid Build Coastguard Worker return DAG.getVectorShuffle(VT, SDLoc(N), SV0, SV1, Mask);
13749*9880d681SAndroid Build Coastguard Worker }
13750*9880d681SAndroid Build Coastguard Worker
13751*9880d681SAndroid Build Coastguard Worker return SDValue();
13752*9880d681SAndroid Build Coastguard Worker }
13753*9880d681SAndroid Build Coastguard Worker
visitSCALAR_TO_VECTOR(SDNode * N)13754*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitSCALAR_TO_VECTOR(SDNode *N) {
13755*9880d681SAndroid Build Coastguard Worker SDValue InVal = N->getOperand(0);
13756*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
13757*9880d681SAndroid Build Coastguard Worker
13758*9880d681SAndroid Build Coastguard Worker // Replace a SCALAR_TO_VECTOR(EXTRACT_VECTOR_ELT(V,C0)) pattern
13759*9880d681SAndroid Build Coastguard Worker // with a VECTOR_SHUFFLE.
13760*9880d681SAndroid Build Coastguard Worker if (InVal.getOpcode() == ISD::EXTRACT_VECTOR_ELT) {
13761*9880d681SAndroid Build Coastguard Worker SDValue InVec = InVal->getOperand(0);
13762*9880d681SAndroid Build Coastguard Worker SDValue EltNo = InVal->getOperand(1);
13763*9880d681SAndroid Build Coastguard Worker
13764*9880d681SAndroid Build Coastguard Worker // FIXME: We could support implicit truncation if the shuffle can be
13765*9880d681SAndroid Build Coastguard Worker // scaled to a smaller vector scalar type.
13766*9880d681SAndroid Build Coastguard Worker ConstantSDNode *C0 = dyn_cast<ConstantSDNode>(EltNo);
13767*9880d681SAndroid Build Coastguard Worker if (C0 && VT == InVec.getValueType() &&
13768*9880d681SAndroid Build Coastguard Worker VT.getScalarType() == InVal.getValueType()) {
13769*9880d681SAndroid Build Coastguard Worker SmallVector<int, 8> NewMask(VT.getVectorNumElements(), -1);
13770*9880d681SAndroid Build Coastguard Worker int Elt = C0->getZExtValue();
13771*9880d681SAndroid Build Coastguard Worker NewMask[0] = Elt;
13772*9880d681SAndroid Build Coastguard Worker
13773*9880d681SAndroid Build Coastguard Worker if (TLI.isShuffleMaskLegal(NewMask, VT))
13774*9880d681SAndroid Build Coastguard Worker return DAG.getVectorShuffle(VT, SDLoc(N), InVec, DAG.getUNDEF(VT),
13775*9880d681SAndroid Build Coastguard Worker NewMask);
13776*9880d681SAndroid Build Coastguard Worker }
13777*9880d681SAndroid Build Coastguard Worker }
13778*9880d681SAndroid Build Coastguard Worker
13779*9880d681SAndroid Build Coastguard Worker return SDValue();
13780*9880d681SAndroid Build Coastguard Worker }
13781*9880d681SAndroid Build Coastguard Worker
visitINSERT_SUBVECTOR(SDNode * N)13782*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitINSERT_SUBVECTOR(SDNode *N) {
13783*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
13784*9880d681SAndroid Build Coastguard Worker SDValue N1 = N->getOperand(1);
13785*9880d681SAndroid Build Coastguard Worker SDValue N2 = N->getOperand(2);
13786*9880d681SAndroid Build Coastguard Worker
13787*9880d681SAndroid Build Coastguard Worker if (N0.getValueType() != N1.getValueType())
13788*9880d681SAndroid Build Coastguard Worker return SDValue();
13789*9880d681SAndroid Build Coastguard Worker
13790*9880d681SAndroid Build Coastguard Worker // If the input vector is a concatenation, and the insert replaces
13791*9880d681SAndroid Build Coastguard Worker // one of the halves, we can optimize into a single concat_vectors.
13792*9880d681SAndroid Build Coastguard Worker if (N0.getOpcode() == ISD::CONCAT_VECTORS && N0->getNumOperands() == 2 &&
13793*9880d681SAndroid Build Coastguard Worker N2.getOpcode() == ISD::Constant) {
13794*9880d681SAndroid Build Coastguard Worker APInt InsIdx = cast<ConstantSDNode>(N2)->getAPIntValue();
13795*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
13796*9880d681SAndroid Build Coastguard Worker
13797*9880d681SAndroid Build Coastguard Worker // Lower half: fold (insert_subvector (concat_vectors X, Y), Z) ->
13798*9880d681SAndroid Build Coastguard Worker // (concat_vectors Z, Y)
13799*9880d681SAndroid Build Coastguard Worker if (InsIdx == 0)
13800*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::CONCAT_VECTORS, SDLoc(N), VT, N1,
13801*9880d681SAndroid Build Coastguard Worker N0.getOperand(1));
13802*9880d681SAndroid Build Coastguard Worker
13803*9880d681SAndroid Build Coastguard Worker // Upper half: fold (insert_subvector (concat_vectors X, Y), Z) ->
13804*9880d681SAndroid Build Coastguard Worker // (concat_vectors X, Z)
13805*9880d681SAndroid Build Coastguard Worker if (InsIdx == VT.getVectorNumElements() / 2)
13806*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::CONCAT_VECTORS, SDLoc(N), VT, N0.getOperand(0),
13807*9880d681SAndroid Build Coastguard Worker N1);
13808*9880d681SAndroid Build Coastguard Worker }
13809*9880d681SAndroid Build Coastguard Worker
13810*9880d681SAndroid Build Coastguard Worker return SDValue();
13811*9880d681SAndroid Build Coastguard Worker }
13812*9880d681SAndroid Build Coastguard Worker
visitFP_TO_FP16(SDNode * N)13813*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFP_TO_FP16(SDNode *N) {
13814*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
13815*9880d681SAndroid Build Coastguard Worker
13816*9880d681SAndroid Build Coastguard Worker // fold (fp_to_fp16 (fp16_to_fp op)) -> op
13817*9880d681SAndroid Build Coastguard Worker if (N0->getOpcode() == ISD::FP16_TO_FP)
13818*9880d681SAndroid Build Coastguard Worker return N0->getOperand(0);
13819*9880d681SAndroid Build Coastguard Worker
13820*9880d681SAndroid Build Coastguard Worker return SDValue();
13821*9880d681SAndroid Build Coastguard Worker }
13822*9880d681SAndroid Build Coastguard Worker
visitFP16_TO_FP(SDNode * N)13823*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::visitFP16_TO_FP(SDNode *N) {
13824*9880d681SAndroid Build Coastguard Worker SDValue N0 = N->getOperand(0);
13825*9880d681SAndroid Build Coastguard Worker
13826*9880d681SAndroid Build Coastguard Worker // fold fp16_to_fp(op & 0xffff) -> fp16_to_fp(op)
13827*9880d681SAndroid Build Coastguard Worker if (N0->getOpcode() == ISD::AND) {
13828*9880d681SAndroid Build Coastguard Worker ConstantSDNode *AndConst = getAsNonOpaqueConstant(N0.getOperand(1));
13829*9880d681SAndroid Build Coastguard Worker if (AndConst && AndConst->getAPIntValue() == 0xffff) {
13830*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FP16_TO_FP, SDLoc(N), N->getValueType(0),
13831*9880d681SAndroid Build Coastguard Worker N0.getOperand(0));
13832*9880d681SAndroid Build Coastguard Worker }
13833*9880d681SAndroid Build Coastguard Worker }
13834*9880d681SAndroid Build Coastguard Worker
13835*9880d681SAndroid Build Coastguard Worker return SDValue();
13836*9880d681SAndroid Build Coastguard Worker }
13837*9880d681SAndroid Build Coastguard Worker
13838*9880d681SAndroid Build Coastguard Worker /// Returns a vector_shuffle if it able to transform an AND to a vector_shuffle
13839*9880d681SAndroid Build Coastguard Worker /// with the destination vector and a zero vector.
13840*9880d681SAndroid Build Coastguard Worker /// e.g. AND V, <0xffffffff, 0, 0xffffffff, 0>. ==>
13841*9880d681SAndroid Build Coastguard Worker /// vector_shuffle V, Zero, <0, 4, 2, 4>
XformToShuffleWithZero(SDNode * N)13842*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::XformToShuffleWithZero(SDNode *N) {
13843*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
13844*9880d681SAndroid Build Coastguard Worker SDValue LHS = N->getOperand(0);
13845*9880d681SAndroid Build Coastguard Worker SDValue RHS = N->getOperand(1);
13846*9880d681SAndroid Build Coastguard Worker SDLoc dl(N);
13847*9880d681SAndroid Build Coastguard Worker
13848*9880d681SAndroid Build Coastguard Worker // Make sure we're not running after operation legalization where it
13849*9880d681SAndroid Build Coastguard Worker // may have custom lowered the vector shuffles.
13850*9880d681SAndroid Build Coastguard Worker if (LegalOperations)
13851*9880d681SAndroid Build Coastguard Worker return SDValue();
13852*9880d681SAndroid Build Coastguard Worker
13853*9880d681SAndroid Build Coastguard Worker if (N->getOpcode() != ISD::AND)
13854*9880d681SAndroid Build Coastguard Worker return SDValue();
13855*9880d681SAndroid Build Coastguard Worker
13856*9880d681SAndroid Build Coastguard Worker if (RHS.getOpcode() == ISD::BITCAST)
13857*9880d681SAndroid Build Coastguard Worker RHS = RHS.getOperand(0);
13858*9880d681SAndroid Build Coastguard Worker
13859*9880d681SAndroid Build Coastguard Worker if (RHS.getOpcode() != ISD::BUILD_VECTOR)
13860*9880d681SAndroid Build Coastguard Worker return SDValue();
13861*9880d681SAndroid Build Coastguard Worker
13862*9880d681SAndroid Build Coastguard Worker EVT RVT = RHS.getValueType();
13863*9880d681SAndroid Build Coastguard Worker unsigned NumElts = RHS.getNumOperands();
13864*9880d681SAndroid Build Coastguard Worker
13865*9880d681SAndroid Build Coastguard Worker // Attempt to create a valid clear mask, splitting the mask into
13866*9880d681SAndroid Build Coastguard Worker // sub elements and checking to see if each is
13867*9880d681SAndroid Build Coastguard Worker // all zeros or all ones - suitable for shuffle masking.
13868*9880d681SAndroid Build Coastguard Worker auto BuildClearMask = [&](int Split) {
13869*9880d681SAndroid Build Coastguard Worker int NumSubElts = NumElts * Split;
13870*9880d681SAndroid Build Coastguard Worker int NumSubBits = RVT.getScalarSizeInBits() / Split;
13871*9880d681SAndroid Build Coastguard Worker
13872*9880d681SAndroid Build Coastguard Worker SmallVector<int, 8> Indices;
13873*9880d681SAndroid Build Coastguard Worker for (int i = 0; i != NumSubElts; ++i) {
13874*9880d681SAndroid Build Coastguard Worker int EltIdx = i / Split;
13875*9880d681SAndroid Build Coastguard Worker int SubIdx = i % Split;
13876*9880d681SAndroid Build Coastguard Worker SDValue Elt = RHS.getOperand(EltIdx);
13877*9880d681SAndroid Build Coastguard Worker if (Elt.isUndef()) {
13878*9880d681SAndroid Build Coastguard Worker Indices.push_back(-1);
13879*9880d681SAndroid Build Coastguard Worker continue;
13880*9880d681SAndroid Build Coastguard Worker }
13881*9880d681SAndroid Build Coastguard Worker
13882*9880d681SAndroid Build Coastguard Worker APInt Bits;
13883*9880d681SAndroid Build Coastguard Worker if (isa<ConstantSDNode>(Elt))
13884*9880d681SAndroid Build Coastguard Worker Bits = cast<ConstantSDNode>(Elt)->getAPIntValue();
13885*9880d681SAndroid Build Coastguard Worker else if (isa<ConstantFPSDNode>(Elt))
13886*9880d681SAndroid Build Coastguard Worker Bits = cast<ConstantFPSDNode>(Elt)->getValueAPF().bitcastToAPInt();
13887*9880d681SAndroid Build Coastguard Worker else
13888*9880d681SAndroid Build Coastguard Worker return SDValue();
13889*9880d681SAndroid Build Coastguard Worker
13890*9880d681SAndroid Build Coastguard Worker // Extract the sub element from the constant bit mask.
13891*9880d681SAndroid Build Coastguard Worker if (DAG.getDataLayout().isBigEndian()) {
13892*9880d681SAndroid Build Coastguard Worker Bits = Bits.lshr((Split - SubIdx - 1) * NumSubBits);
13893*9880d681SAndroid Build Coastguard Worker } else {
13894*9880d681SAndroid Build Coastguard Worker Bits = Bits.lshr(SubIdx * NumSubBits);
13895*9880d681SAndroid Build Coastguard Worker }
13896*9880d681SAndroid Build Coastguard Worker
13897*9880d681SAndroid Build Coastguard Worker if (Split > 1)
13898*9880d681SAndroid Build Coastguard Worker Bits = Bits.trunc(NumSubBits);
13899*9880d681SAndroid Build Coastguard Worker
13900*9880d681SAndroid Build Coastguard Worker if (Bits.isAllOnesValue())
13901*9880d681SAndroid Build Coastguard Worker Indices.push_back(i);
13902*9880d681SAndroid Build Coastguard Worker else if (Bits == 0)
13903*9880d681SAndroid Build Coastguard Worker Indices.push_back(i + NumSubElts);
13904*9880d681SAndroid Build Coastguard Worker else
13905*9880d681SAndroid Build Coastguard Worker return SDValue();
13906*9880d681SAndroid Build Coastguard Worker }
13907*9880d681SAndroid Build Coastguard Worker
13908*9880d681SAndroid Build Coastguard Worker // Let's see if the target supports this vector_shuffle.
13909*9880d681SAndroid Build Coastguard Worker EVT ClearSVT = EVT::getIntegerVT(*DAG.getContext(), NumSubBits);
13910*9880d681SAndroid Build Coastguard Worker EVT ClearVT = EVT::getVectorVT(*DAG.getContext(), ClearSVT, NumSubElts);
13911*9880d681SAndroid Build Coastguard Worker if (!TLI.isVectorClearMaskLegal(Indices, ClearVT))
13912*9880d681SAndroid Build Coastguard Worker return SDValue();
13913*9880d681SAndroid Build Coastguard Worker
13914*9880d681SAndroid Build Coastguard Worker SDValue Zero = DAG.getConstant(0, dl, ClearVT);
13915*9880d681SAndroid Build Coastguard Worker return DAG.getBitcast(VT, DAG.getVectorShuffle(ClearVT, dl,
13916*9880d681SAndroid Build Coastguard Worker DAG.getBitcast(ClearVT, LHS),
13917*9880d681SAndroid Build Coastguard Worker Zero, Indices));
13918*9880d681SAndroid Build Coastguard Worker };
13919*9880d681SAndroid Build Coastguard Worker
13920*9880d681SAndroid Build Coastguard Worker // Determine maximum split level (byte level masking).
13921*9880d681SAndroid Build Coastguard Worker int MaxSplit = 1;
13922*9880d681SAndroid Build Coastguard Worker if (RVT.getScalarSizeInBits() % 8 == 0)
13923*9880d681SAndroid Build Coastguard Worker MaxSplit = RVT.getScalarSizeInBits() / 8;
13924*9880d681SAndroid Build Coastguard Worker
13925*9880d681SAndroid Build Coastguard Worker for (int Split = 1; Split <= MaxSplit; ++Split)
13926*9880d681SAndroid Build Coastguard Worker if (RVT.getScalarSizeInBits() % Split == 0)
13927*9880d681SAndroid Build Coastguard Worker if (SDValue S = BuildClearMask(Split))
13928*9880d681SAndroid Build Coastguard Worker return S;
13929*9880d681SAndroid Build Coastguard Worker
13930*9880d681SAndroid Build Coastguard Worker return SDValue();
13931*9880d681SAndroid Build Coastguard Worker }
13932*9880d681SAndroid Build Coastguard Worker
13933*9880d681SAndroid Build Coastguard Worker /// Visit a binary vector operation, like ADD.
SimplifyVBinOp(SDNode * N)13934*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::SimplifyVBinOp(SDNode *N) {
13935*9880d681SAndroid Build Coastguard Worker assert(N->getValueType(0).isVector() &&
13936*9880d681SAndroid Build Coastguard Worker "SimplifyVBinOp only works on vectors!");
13937*9880d681SAndroid Build Coastguard Worker
13938*9880d681SAndroid Build Coastguard Worker SDValue LHS = N->getOperand(0);
13939*9880d681SAndroid Build Coastguard Worker SDValue RHS = N->getOperand(1);
13940*9880d681SAndroid Build Coastguard Worker SDValue Ops[] = {LHS, RHS};
13941*9880d681SAndroid Build Coastguard Worker
13942*9880d681SAndroid Build Coastguard Worker // See if we can constant fold the vector operation.
13943*9880d681SAndroid Build Coastguard Worker if (SDValue Fold = DAG.FoldConstantVectorArithmetic(
13944*9880d681SAndroid Build Coastguard Worker N->getOpcode(), SDLoc(LHS), LHS.getValueType(), Ops, N->getFlags()))
13945*9880d681SAndroid Build Coastguard Worker return Fold;
13946*9880d681SAndroid Build Coastguard Worker
13947*9880d681SAndroid Build Coastguard Worker // Try to convert a constant mask AND into a shuffle clear mask.
13948*9880d681SAndroid Build Coastguard Worker if (SDValue Shuffle = XformToShuffleWithZero(N))
13949*9880d681SAndroid Build Coastguard Worker return Shuffle;
13950*9880d681SAndroid Build Coastguard Worker
13951*9880d681SAndroid Build Coastguard Worker // Type legalization might introduce new shuffles in the DAG.
13952*9880d681SAndroid Build Coastguard Worker // Fold (VBinOp (shuffle (A, Undef, Mask)), (shuffle (B, Undef, Mask)))
13953*9880d681SAndroid Build Coastguard Worker // -> (shuffle (VBinOp (A, B)), Undef, Mask).
13954*9880d681SAndroid Build Coastguard Worker if (LegalTypes && isa<ShuffleVectorSDNode>(LHS) &&
13955*9880d681SAndroid Build Coastguard Worker isa<ShuffleVectorSDNode>(RHS) && LHS.hasOneUse() && RHS.hasOneUse() &&
13956*9880d681SAndroid Build Coastguard Worker LHS.getOperand(1).isUndef() &&
13957*9880d681SAndroid Build Coastguard Worker RHS.getOperand(1).isUndef()) {
13958*9880d681SAndroid Build Coastguard Worker ShuffleVectorSDNode *SVN0 = cast<ShuffleVectorSDNode>(LHS);
13959*9880d681SAndroid Build Coastguard Worker ShuffleVectorSDNode *SVN1 = cast<ShuffleVectorSDNode>(RHS);
13960*9880d681SAndroid Build Coastguard Worker
13961*9880d681SAndroid Build Coastguard Worker if (SVN0->getMask().equals(SVN1->getMask())) {
13962*9880d681SAndroid Build Coastguard Worker EVT VT = N->getValueType(0);
13963*9880d681SAndroid Build Coastguard Worker SDValue UndefVector = LHS.getOperand(1);
13964*9880d681SAndroid Build Coastguard Worker SDValue NewBinOp = DAG.getNode(N->getOpcode(), SDLoc(N), VT,
13965*9880d681SAndroid Build Coastguard Worker LHS.getOperand(0), RHS.getOperand(0),
13966*9880d681SAndroid Build Coastguard Worker N->getFlags());
13967*9880d681SAndroid Build Coastguard Worker AddUsersToWorklist(N);
13968*9880d681SAndroid Build Coastguard Worker return DAG.getVectorShuffle(VT, SDLoc(N), NewBinOp, UndefVector,
13969*9880d681SAndroid Build Coastguard Worker SVN0->getMask());
13970*9880d681SAndroid Build Coastguard Worker }
13971*9880d681SAndroid Build Coastguard Worker }
13972*9880d681SAndroid Build Coastguard Worker
13973*9880d681SAndroid Build Coastguard Worker return SDValue();
13974*9880d681SAndroid Build Coastguard Worker }
13975*9880d681SAndroid Build Coastguard Worker
SimplifySelect(const SDLoc & DL,SDValue N0,SDValue N1,SDValue N2)13976*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::SimplifySelect(const SDLoc &DL, SDValue N0, SDValue N1,
13977*9880d681SAndroid Build Coastguard Worker SDValue N2) {
13978*9880d681SAndroid Build Coastguard Worker assert(N0.getOpcode() ==ISD::SETCC && "First argument must be a SetCC node!");
13979*9880d681SAndroid Build Coastguard Worker
13980*9880d681SAndroid Build Coastguard Worker SDValue SCC = SimplifySelectCC(DL, N0.getOperand(0), N0.getOperand(1), N1, N2,
13981*9880d681SAndroid Build Coastguard Worker cast<CondCodeSDNode>(N0.getOperand(2))->get());
13982*9880d681SAndroid Build Coastguard Worker
13983*9880d681SAndroid Build Coastguard Worker // If we got a simplified select_cc node back from SimplifySelectCC, then
13984*9880d681SAndroid Build Coastguard Worker // break it down into a new SETCC node, and a new SELECT node, and then return
13985*9880d681SAndroid Build Coastguard Worker // the SELECT node, since we were called with a SELECT node.
13986*9880d681SAndroid Build Coastguard Worker if (SCC.getNode()) {
13987*9880d681SAndroid Build Coastguard Worker // Check to see if we got a select_cc back (to turn into setcc/select).
13988*9880d681SAndroid Build Coastguard Worker // Otherwise, just return whatever node we got back, like fabs.
13989*9880d681SAndroid Build Coastguard Worker if (SCC.getOpcode() == ISD::SELECT_CC) {
13990*9880d681SAndroid Build Coastguard Worker SDValue SETCC = DAG.getNode(ISD::SETCC, SDLoc(N0),
13991*9880d681SAndroid Build Coastguard Worker N0.getValueType(),
13992*9880d681SAndroid Build Coastguard Worker SCC.getOperand(0), SCC.getOperand(1),
13993*9880d681SAndroid Build Coastguard Worker SCC.getOperand(4));
13994*9880d681SAndroid Build Coastguard Worker AddToWorklist(SETCC.getNode());
13995*9880d681SAndroid Build Coastguard Worker return DAG.getSelect(SDLoc(SCC), SCC.getValueType(), SETCC,
13996*9880d681SAndroid Build Coastguard Worker SCC.getOperand(2), SCC.getOperand(3));
13997*9880d681SAndroid Build Coastguard Worker }
13998*9880d681SAndroid Build Coastguard Worker
13999*9880d681SAndroid Build Coastguard Worker return SCC;
14000*9880d681SAndroid Build Coastguard Worker }
14001*9880d681SAndroid Build Coastguard Worker return SDValue();
14002*9880d681SAndroid Build Coastguard Worker }
14003*9880d681SAndroid Build Coastguard Worker
14004*9880d681SAndroid Build Coastguard Worker /// Given a SELECT or a SELECT_CC node, where LHS and RHS are the two values
14005*9880d681SAndroid Build Coastguard Worker /// being selected between, see if we can simplify the select. Callers of this
14006*9880d681SAndroid Build Coastguard Worker /// should assume that TheSelect is deleted if this returns true. As such, they
14007*9880d681SAndroid Build Coastguard Worker /// should return the appropriate thing (e.g. the node) back to the top-level of
14008*9880d681SAndroid Build Coastguard Worker /// the DAG combiner loop to avoid it being looked at.
SimplifySelectOps(SDNode * TheSelect,SDValue LHS,SDValue RHS)14009*9880d681SAndroid Build Coastguard Worker bool DAGCombiner::SimplifySelectOps(SDNode *TheSelect, SDValue LHS,
14010*9880d681SAndroid Build Coastguard Worker SDValue RHS) {
14011*9880d681SAndroid Build Coastguard Worker
14012*9880d681SAndroid Build Coastguard Worker // fold (select (setcc x, [+-]0.0, *lt), NaN, (fsqrt x))
14013*9880d681SAndroid Build Coastguard Worker // The select + setcc is redundant, because fsqrt returns NaN for X < 0.
14014*9880d681SAndroid Build Coastguard Worker if (const ConstantFPSDNode *NaN = isConstOrConstSplatFP(LHS)) {
14015*9880d681SAndroid Build Coastguard Worker if (NaN->isNaN() && RHS.getOpcode() == ISD::FSQRT) {
14016*9880d681SAndroid Build Coastguard Worker // We have: (select (setcc ?, ?, ?), NaN, (fsqrt ?))
14017*9880d681SAndroid Build Coastguard Worker SDValue Sqrt = RHS;
14018*9880d681SAndroid Build Coastguard Worker ISD::CondCode CC;
14019*9880d681SAndroid Build Coastguard Worker SDValue CmpLHS;
14020*9880d681SAndroid Build Coastguard Worker const ConstantFPSDNode *Zero = nullptr;
14021*9880d681SAndroid Build Coastguard Worker
14022*9880d681SAndroid Build Coastguard Worker if (TheSelect->getOpcode() == ISD::SELECT_CC) {
14023*9880d681SAndroid Build Coastguard Worker CC = dyn_cast<CondCodeSDNode>(TheSelect->getOperand(4))->get();
14024*9880d681SAndroid Build Coastguard Worker CmpLHS = TheSelect->getOperand(0);
14025*9880d681SAndroid Build Coastguard Worker Zero = isConstOrConstSplatFP(TheSelect->getOperand(1));
14026*9880d681SAndroid Build Coastguard Worker } else {
14027*9880d681SAndroid Build Coastguard Worker // SELECT or VSELECT
14028*9880d681SAndroid Build Coastguard Worker SDValue Cmp = TheSelect->getOperand(0);
14029*9880d681SAndroid Build Coastguard Worker if (Cmp.getOpcode() == ISD::SETCC) {
14030*9880d681SAndroid Build Coastguard Worker CC = dyn_cast<CondCodeSDNode>(Cmp.getOperand(2))->get();
14031*9880d681SAndroid Build Coastguard Worker CmpLHS = Cmp.getOperand(0);
14032*9880d681SAndroid Build Coastguard Worker Zero = isConstOrConstSplatFP(Cmp.getOperand(1));
14033*9880d681SAndroid Build Coastguard Worker }
14034*9880d681SAndroid Build Coastguard Worker }
14035*9880d681SAndroid Build Coastguard Worker if (Zero && Zero->isZero() &&
14036*9880d681SAndroid Build Coastguard Worker Sqrt.getOperand(0) == CmpLHS && (CC == ISD::SETOLT ||
14037*9880d681SAndroid Build Coastguard Worker CC == ISD::SETULT || CC == ISD::SETLT)) {
14038*9880d681SAndroid Build Coastguard Worker // We have: (select (setcc x, [+-]0.0, *lt), NaN, (fsqrt x))
14039*9880d681SAndroid Build Coastguard Worker CombineTo(TheSelect, Sqrt);
14040*9880d681SAndroid Build Coastguard Worker return true;
14041*9880d681SAndroid Build Coastguard Worker }
14042*9880d681SAndroid Build Coastguard Worker }
14043*9880d681SAndroid Build Coastguard Worker }
14044*9880d681SAndroid Build Coastguard Worker // Cannot simplify select with vector condition
14045*9880d681SAndroid Build Coastguard Worker if (TheSelect->getOperand(0).getValueType().isVector()) return false;
14046*9880d681SAndroid Build Coastguard Worker
14047*9880d681SAndroid Build Coastguard Worker // If this is a select from two identical things, try to pull the operation
14048*9880d681SAndroid Build Coastguard Worker // through the select.
14049*9880d681SAndroid Build Coastguard Worker if (LHS.getOpcode() != RHS.getOpcode() ||
14050*9880d681SAndroid Build Coastguard Worker !LHS.hasOneUse() || !RHS.hasOneUse())
14051*9880d681SAndroid Build Coastguard Worker return false;
14052*9880d681SAndroid Build Coastguard Worker
14053*9880d681SAndroid Build Coastguard Worker // If this is a load and the token chain is identical, replace the select
14054*9880d681SAndroid Build Coastguard Worker // of two loads with a load through a select of the address to load from.
14055*9880d681SAndroid Build Coastguard Worker // This triggers in things like "select bool X, 10.0, 123.0" after the FP
14056*9880d681SAndroid Build Coastguard Worker // constants have been dropped into the constant pool.
14057*9880d681SAndroid Build Coastguard Worker if (LHS.getOpcode() == ISD::LOAD) {
14058*9880d681SAndroid Build Coastguard Worker LoadSDNode *LLD = cast<LoadSDNode>(LHS);
14059*9880d681SAndroid Build Coastguard Worker LoadSDNode *RLD = cast<LoadSDNode>(RHS);
14060*9880d681SAndroid Build Coastguard Worker
14061*9880d681SAndroid Build Coastguard Worker // Token chains must be identical.
14062*9880d681SAndroid Build Coastguard Worker if (LHS.getOperand(0) != RHS.getOperand(0) ||
14063*9880d681SAndroid Build Coastguard Worker // Do not let this transformation reduce the number of volatile loads.
14064*9880d681SAndroid Build Coastguard Worker LLD->isVolatile() || RLD->isVolatile() ||
14065*9880d681SAndroid Build Coastguard Worker // FIXME: If either is a pre/post inc/dec load,
14066*9880d681SAndroid Build Coastguard Worker // we'd need to split out the address adjustment.
14067*9880d681SAndroid Build Coastguard Worker LLD->isIndexed() || RLD->isIndexed() ||
14068*9880d681SAndroid Build Coastguard Worker // If this is an EXTLOAD, the VT's must match.
14069*9880d681SAndroid Build Coastguard Worker LLD->getMemoryVT() != RLD->getMemoryVT() ||
14070*9880d681SAndroid Build Coastguard Worker // If this is an EXTLOAD, the kind of extension must match.
14071*9880d681SAndroid Build Coastguard Worker (LLD->getExtensionType() != RLD->getExtensionType() &&
14072*9880d681SAndroid Build Coastguard Worker // The only exception is if one of the extensions is anyext.
14073*9880d681SAndroid Build Coastguard Worker LLD->getExtensionType() != ISD::EXTLOAD &&
14074*9880d681SAndroid Build Coastguard Worker RLD->getExtensionType() != ISD::EXTLOAD) ||
14075*9880d681SAndroid Build Coastguard Worker // FIXME: this discards src value information. This is
14076*9880d681SAndroid Build Coastguard Worker // over-conservative. It would be beneficial to be able to remember
14077*9880d681SAndroid Build Coastguard Worker // both potential memory locations. Since we are discarding
14078*9880d681SAndroid Build Coastguard Worker // src value info, don't do the transformation if the memory
14079*9880d681SAndroid Build Coastguard Worker // locations are not in the default address space.
14080*9880d681SAndroid Build Coastguard Worker LLD->getPointerInfo().getAddrSpace() != 0 ||
14081*9880d681SAndroid Build Coastguard Worker RLD->getPointerInfo().getAddrSpace() != 0 ||
14082*9880d681SAndroid Build Coastguard Worker !TLI.isOperationLegalOrCustom(TheSelect->getOpcode(),
14083*9880d681SAndroid Build Coastguard Worker LLD->getBasePtr().getValueType()))
14084*9880d681SAndroid Build Coastguard Worker return false;
14085*9880d681SAndroid Build Coastguard Worker
14086*9880d681SAndroid Build Coastguard Worker // Check that the select condition doesn't reach either load. If so,
14087*9880d681SAndroid Build Coastguard Worker // folding this will induce a cycle into the DAG. If not, this is safe to
14088*9880d681SAndroid Build Coastguard Worker // xform, so create a select of the addresses.
14089*9880d681SAndroid Build Coastguard Worker SDValue Addr;
14090*9880d681SAndroid Build Coastguard Worker if (TheSelect->getOpcode() == ISD::SELECT) {
14091*9880d681SAndroid Build Coastguard Worker SDNode *CondNode = TheSelect->getOperand(0).getNode();
14092*9880d681SAndroid Build Coastguard Worker if ((LLD->hasAnyUseOfValue(1) && LLD->isPredecessorOf(CondNode)) ||
14093*9880d681SAndroid Build Coastguard Worker (RLD->hasAnyUseOfValue(1) && RLD->isPredecessorOf(CondNode)))
14094*9880d681SAndroid Build Coastguard Worker return false;
14095*9880d681SAndroid Build Coastguard Worker // The loads must not depend on one another.
14096*9880d681SAndroid Build Coastguard Worker if (LLD->isPredecessorOf(RLD) ||
14097*9880d681SAndroid Build Coastguard Worker RLD->isPredecessorOf(LLD))
14098*9880d681SAndroid Build Coastguard Worker return false;
14099*9880d681SAndroid Build Coastguard Worker Addr = DAG.getSelect(SDLoc(TheSelect),
14100*9880d681SAndroid Build Coastguard Worker LLD->getBasePtr().getValueType(),
14101*9880d681SAndroid Build Coastguard Worker TheSelect->getOperand(0), LLD->getBasePtr(),
14102*9880d681SAndroid Build Coastguard Worker RLD->getBasePtr());
14103*9880d681SAndroid Build Coastguard Worker } else { // Otherwise SELECT_CC
14104*9880d681SAndroid Build Coastguard Worker SDNode *CondLHS = TheSelect->getOperand(0).getNode();
14105*9880d681SAndroid Build Coastguard Worker SDNode *CondRHS = TheSelect->getOperand(1).getNode();
14106*9880d681SAndroid Build Coastguard Worker
14107*9880d681SAndroid Build Coastguard Worker if ((LLD->hasAnyUseOfValue(1) &&
14108*9880d681SAndroid Build Coastguard Worker (LLD->isPredecessorOf(CondLHS) || LLD->isPredecessorOf(CondRHS))) ||
14109*9880d681SAndroid Build Coastguard Worker (RLD->hasAnyUseOfValue(1) &&
14110*9880d681SAndroid Build Coastguard Worker (RLD->isPredecessorOf(CondLHS) || RLD->isPredecessorOf(CondRHS))))
14111*9880d681SAndroid Build Coastguard Worker return false;
14112*9880d681SAndroid Build Coastguard Worker
14113*9880d681SAndroid Build Coastguard Worker Addr = DAG.getNode(ISD::SELECT_CC, SDLoc(TheSelect),
14114*9880d681SAndroid Build Coastguard Worker LLD->getBasePtr().getValueType(),
14115*9880d681SAndroid Build Coastguard Worker TheSelect->getOperand(0),
14116*9880d681SAndroid Build Coastguard Worker TheSelect->getOperand(1),
14117*9880d681SAndroid Build Coastguard Worker LLD->getBasePtr(), RLD->getBasePtr(),
14118*9880d681SAndroid Build Coastguard Worker TheSelect->getOperand(4));
14119*9880d681SAndroid Build Coastguard Worker }
14120*9880d681SAndroid Build Coastguard Worker
14121*9880d681SAndroid Build Coastguard Worker SDValue Load;
14122*9880d681SAndroid Build Coastguard Worker // It is safe to replace the two loads if they have different alignments,
14123*9880d681SAndroid Build Coastguard Worker // but the new load must be the minimum (most restrictive) alignment of the
14124*9880d681SAndroid Build Coastguard Worker // inputs.
14125*9880d681SAndroid Build Coastguard Worker bool isInvariant = LLD->isInvariant() & RLD->isInvariant();
14126*9880d681SAndroid Build Coastguard Worker unsigned Alignment = std::min(LLD->getAlignment(), RLD->getAlignment());
14127*9880d681SAndroid Build Coastguard Worker if (LLD->getExtensionType() == ISD::NON_EXTLOAD) {
14128*9880d681SAndroid Build Coastguard Worker Load = DAG.getLoad(TheSelect->getValueType(0),
14129*9880d681SAndroid Build Coastguard Worker SDLoc(TheSelect),
14130*9880d681SAndroid Build Coastguard Worker // FIXME: Discards pointer and AA info.
14131*9880d681SAndroid Build Coastguard Worker LLD->getChain(), Addr, MachinePointerInfo(),
14132*9880d681SAndroid Build Coastguard Worker LLD->isVolatile(), LLD->isNonTemporal(),
14133*9880d681SAndroid Build Coastguard Worker isInvariant, Alignment);
14134*9880d681SAndroid Build Coastguard Worker } else {
14135*9880d681SAndroid Build Coastguard Worker Load = DAG.getExtLoad(LLD->getExtensionType() == ISD::EXTLOAD ?
14136*9880d681SAndroid Build Coastguard Worker RLD->getExtensionType() : LLD->getExtensionType(),
14137*9880d681SAndroid Build Coastguard Worker SDLoc(TheSelect),
14138*9880d681SAndroid Build Coastguard Worker TheSelect->getValueType(0),
14139*9880d681SAndroid Build Coastguard Worker // FIXME: Discards pointer and AA info.
14140*9880d681SAndroid Build Coastguard Worker LLD->getChain(), Addr, MachinePointerInfo(),
14141*9880d681SAndroid Build Coastguard Worker LLD->getMemoryVT(), LLD->isVolatile(),
14142*9880d681SAndroid Build Coastguard Worker LLD->isNonTemporal(), isInvariant, Alignment);
14143*9880d681SAndroid Build Coastguard Worker }
14144*9880d681SAndroid Build Coastguard Worker
14145*9880d681SAndroid Build Coastguard Worker // Users of the select now use the result of the load.
14146*9880d681SAndroid Build Coastguard Worker CombineTo(TheSelect, Load);
14147*9880d681SAndroid Build Coastguard Worker
14148*9880d681SAndroid Build Coastguard Worker // Users of the old loads now use the new load's chain. We know the
14149*9880d681SAndroid Build Coastguard Worker // old-load value is dead now.
14150*9880d681SAndroid Build Coastguard Worker CombineTo(LHS.getNode(), Load.getValue(0), Load.getValue(1));
14151*9880d681SAndroid Build Coastguard Worker CombineTo(RHS.getNode(), Load.getValue(0), Load.getValue(1));
14152*9880d681SAndroid Build Coastguard Worker return true;
14153*9880d681SAndroid Build Coastguard Worker }
14154*9880d681SAndroid Build Coastguard Worker
14155*9880d681SAndroid Build Coastguard Worker return false;
14156*9880d681SAndroid Build Coastguard Worker }
14157*9880d681SAndroid Build Coastguard Worker
14158*9880d681SAndroid Build Coastguard Worker /// Simplify an expression of the form (N0 cond N1) ? N2 : N3
14159*9880d681SAndroid Build Coastguard Worker /// where 'cond' is the comparison specified by CC.
SimplifySelectCC(const SDLoc & DL,SDValue N0,SDValue N1,SDValue N2,SDValue N3,ISD::CondCode CC,bool NotExtCompare)14160*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::SimplifySelectCC(const SDLoc &DL, SDValue N0, SDValue N1,
14161*9880d681SAndroid Build Coastguard Worker SDValue N2, SDValue N3, ISD::CondCode CC,
14162*9880d681SAndroid Build Coastguard Worker bool NotExtCompare) {
14163*9880d681SAndroid Build Coastguard Worker // (x ? y : y) -> y.
14164*9880d681SAndroid Build Coastguard Worker if (N2 == N3) return N2;
14165*9880d681SAndroid Build Coastguard Worker
14166*9880d681SAndroid Build Coastguard Worker EVT VT = N2.getValueType();
14167*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1.getNode());
14168*9880d681SAndroid Build Coastguard Worker ConstantSDNode *N2C = dyn_cast<ConstantSDNode>(N2.getNode());
14169*9880d681SAndroid Build Coastguard Worker
14170*9880d681SAndroid Build Coastguard Worker // Determine if the condition we're dealing with is constant
14171*9880d681SAndroid Build Coastguard Worker SDValue SCC = SimplifySetCC(getSetCCResultType(N0.getValueType()),
14172*9880d681SAndroid Build Coastguard Worker N0, N1, CC, DL, false);
14173*9880d681SAndroid Build Coastguard Worker if (SCC.getNode()) AddToWorklist(SCC.getNode());
14174*9880d681SAndroid Build Coastguard Worker
14175*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *SCCC = dyn_cast_or_null<ConstantSDNode>(SCC.getNode())) {
14176*9880d681SAndroid Build Coastguard Worker // fold select_cc true, x, y -> x
14177*9880d681SAndroid Build Coastguard Worker // fold select_cc false, x, y -> y
14178*9880d681SAndroid Build Coastguard Worker return !SCCC->isNullValue() ? N2 : N3;
14179*9880d681SAndroid Build Coastguard Worker }
14180*9880d681SAndroid Build Coastguard Worker
14181*9880d681SAndroid Build Coastguard Worker // Check to see if we can simplify the select into an fabs node
14182*9880d681SAndroid Build Coastguard Worker if (ConstantFPSDNode *CFP = dyn_cast<ConstantFPSDNode>(N1)) {
14183*9880d681SAndroid Build Coastguard Worker // Allow either -0.0 or 0.0
14184*9880d681SAndroid Build Coastguard Worker if (CFP->isZero()) {
14185*9880d681SAndroid Build Coastguard Worker // select (setg[te] X, +/-0.0), X, fneg(X) -> fabs
14186*9880d681SAndroid Build Coastguard Worker if ((CC == ISD::SETGE || CC == ISD::SETGT) &&
14187*9880d681SAndroid Build Coastguard Worker N0 == N2 && N3.getOpcode() == ISD::FNEG &&
14188*9880d681SAndroid Build Coastguard Worker N2 == N3.getOperand(0))
14189*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FABS, DL, VT, N0);
14190*9880d681SAndroid Build Coastguard Worker
14191*9880d681SAndroid Build Coastguard Worker // select (setl[te] X, +/-0.0), fneg(X), X -> fabs
14192*9880d681SAndroid Build Coastguard Worker if ((CC == ISD::SETLT || CC == ISD::SETLE) &&
14193*9880d681SAndroid Build Coastguard Worker N0 == N3 && N2.getOpcode() == ISD::FNEG &&
14194*9880d681SAndroid Build Coastguard Worker N2.getOperand(0) == N3)
14195*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::FABS, DL, VT, N3);
14196*9880d681SAndroid Build Coastguard Worker }
14197*9880d681SAndroid Build Coastguard Worker }
14198*9880d681SAndroid Build Coastguard Worker
14199*9880d681SAndroid Build Coastguard Worker // Turn "(a cond b) ? 1.0f : 2.0f" into "load (tmp + ((a cond b) ? 0 : 4)"
14200*9880d681SAndroid Build Coastguard Worker // where "tmp" is a constant pool entry containing an array with 1.0 and 2.0
14201*9880d681SAndroid Build Coastguard Worker // in it. This is a win when the constant is not otherwise available because
14202*9880d681SAndroid Build Coastguard Worker // it replaces two constant pool loads with one. We only do this if the FP
14203*9880d681SAndroid Build Coastguard Worker // type is known to be legal, because if it isn't, then we are before legalize
14204*9880d681SAndroid Build Coastguard Worker // types an we want the other legalization to happen first (e.g. to avoid
14205*9880d681SAndroid Build Coastguard Worker // messing with soft float) and if the ConstantFP is not legal, because if
14206*9880d681SAndroid Build Coastguard Worker // it is legal, we may not need to store the FP constant in a constant pool.
14207*9880d681SAndroid Build Coastguard Worker if (ConstantFPSDNode *TV = dyn_cast<ConstantFPSDNode>(N2))
14208*9880d681SAndroid Build Coastguard Worker if (ConstantFPSDNode *FV = dyn_cast<ConstantFPSDNode>(N3)) {
14209*9880d681SAndroid Build Coastguard Worker if (TLI.isTypeLegal(N2.getValueType()) &&
14210*9880d681SAndroid Build Coastguard Worker (TLI.getOperationAction(ISD::ConstantFP, N2.getValueType()) !=
14211*9880d681SAndroid Build Coastguard Worker TargetLowering::Legal &&
14212*9880d681SAndroid Build Coastguard Worker !TLI.isFPImmLegal(TV->getValueAPF(), TV->getValueType(0)) &&
14213*9880d681SAndroid Build Coastguard Worker !TLI.isFPImmLegal(FV->getValueAPF(), FV->getValueType(0))) &&
14214*9880d681SAndroid Build Coastguard Worker // If both constants have multiple uses, then we won't need to do an
14215*9880d681SAndroid Build Coastguard Worker // extra load, they are likely around in registers for other users.
14216*9880d681SAndroid Build Coastguard Worker (TV->hasOneUse() || FV->hasOneUse())) {
14217*9880d681SAndroid Build Coastguard Worker Constant *Elts[] = {
14218*9880d681SAndroid Build Coastguard Worker const_cast<ConstantFP*>(FV->getConstantFPValue()),
14219*9880d681SAndroid Build Coastguard Worker const_cast<ConstantFP*>(TV->getConstantFPValue())
14220*9880d681SAndroid Build Coastguard Worker };
14221*9880d681SAndroid Build Coastguard Worker Type *FPTy = Elts[0]->getType();
14222*9880d681SAndroid Build Coastguard Worker const DataLayout &TD = DAG.getDataLayout();
14223*9880d681SAndroid Build Coastguard Worker
14224*9880d681SAndroid Build Coastguard Worker // Create a ConstantArray of the two constants.
14225*9880d681SAndroid Build Coastguard Worker Constant *CA = ConstantArray::get(ArrayType::get(FPTy, 2), Elts);
14226*9880d681SAndroid Build Coastguard Worker SDValue CPIdx =
14227*9880d681SAndroid Build Coastguard Worker DAG.getConstantPool(CA, TLI.getPointerTy(DAG.getDataLayout()),
14228*9880d681SAndroid Build Coastguard Worker TD.getPrefTypeAlignment(FPTy));
14229*9880d681SAndroid Build Coastguard Worker unsigned Alignment = cast<ConstantPoolSDNode>(CPIdx)->getAlignment();
14230*9880d681SAndroid Build Coastguard Worker
14231*9880d681SAndroid Build Coastguard Worker // Get the offsets to the 0 and 1 element of the array so that we can
14232*9880d681SAndroid Build Coastguard Worker // select between them.
14233*9880d681SAndroid Build Coastguard Worker SDValue Zero = DAG.getIntPtrConstant(0, DL);
14234*9880d681SAndroid Build Coastguard Worker unsigned EltSize = (unsigned)TD.getTypeAllocSize(Elts[0]->getType());
14235*9880d681SAndroid Build Coastguard Worker SDValue One = DAG.getIntPtrConstant(EltSize, SDLoc(FV));
14236*9880d681SAndroid Build Coastguard Worker
14237*9880d681SAndroid Build Coastguard Worker SDValue Cond = DAG.getSetCC(DL,
14238*9880d681SAndroid Build Coastguard Worker getSetCCResultType(N0.getValueType()),
14239*9880d681SAndroid Build Coastguard Worker N0, N1, CC);
14240*9880d681SAndroid Build Coastguard Worker AddToWorklist(Cond.getNode());
14241*9880d681SAndroid Build Coastguard Worker SDValue CstOffset = DAG.getSelect(DL, Zero.getValueType(),
14242*9880d681SAndroid Build Coastguard Worker Cond, One, Zero);
14243*9880d681SAndroid Build Coastguard Worker AddToWorklist(CstOffset.getNode());
14244*9880d681SAndroid Build Coastguard Worker CPIdx = DAG.getNode(ISD::ADD, DL, CPIdx.getValueType(), CPIdx,
14245*9880d681SAndroid Build Coastguard Worker CstOffset);
14246*9880d681SAndroid Build Coastguard Worker AddToWorklist(CPIdx.getNode());
14247*9880d681SAndroid Build Coastguard Worker return DAG.getLoad(
14248*9880d681SAndroid Build Coastguard Worker TV->getValueType(0), DL, DAG.getEntryNode(), CPIdx,
14249*9880d681SAndroid Build Coastguard Worker MachinePointerInfo::getConstantPool(DAG.getMachineFunction()),
14250*9880d681SAndroid Build Coastguard Worker false, false, false, Alignment);
14251*9880d681SAndroid Build Coastguard Worker }
14252*9880d681SAndroid Build Coastguard Worker }
14253*9880d681SAndroid Build Coastguard Worker
14254*9880d681SAndroid Build Coastguard Worker // Check to see if we can perform the "gzip trick", transforming
14255*9880d681SAndroid Build Coastguard Worker // (select_cc setlt X, 0, A, 0) -> (and (sra X, (sub size(X), 1), A)
14256*9880d681SAndroid Build Coastguard Worker if (isNullConstant(N3) && CC == ISD::SETLT &&
14257*9880d681SAndroid Build Coastguard Worker (isNullConstant(N1) || // (a < 0) ? b : 0
14258*9880d681SAndroid Build Coastguard Worker (isOneConstant(N1) && N0 == N2))) { // (a < 1) ? a : 0
14259*9880d681SAndroid Build Coastguard Worker EVT XType = N0.getValueType();
14260*9880d681SAndroid Build Coastguard Worker EVT AType = N2.getValueType();
14261*9880d681SAndroid Build Coastguard Worker if (XType.bitsGE(AType)) {
14262*9880d681SAndroid Build Coastguard Worker // and (sra X, size(X)-1, A) -> "and (srl X, C2), A" iff A is a
14263*9880d681SAndroid Build Coastguard Worker // single-bit constant.
14264*9880d681SAndroid Build Coastguard Worker if (N2C && ((N2C->getAPIntValue() & (N2C->getAPIntValue() - 1)) == 0)) {
14265*9880d681SAndroid Build Coastguard Worker unsigned ShCtV = N2C->getAPIntValue().logBase2();
14266*9880d681SAndroid Build Coastguard Worker ShCtV = XType.getSizeInBits() - ShCtV - 1;
14267*9880d681SAndroid Build Coastguard Worker SDValue ShCt = DAG.getConstant(ShCtV, SDLoc(N0),
14268*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(N0.getValueType()));
14269*9880d681SAndroid Build Coastguard Worker SDValue Shift = DAG.getNode(ISD::SRL, SDLoc(N0),
14270*9880d681SAndroid Build Coastguard Worker XType, N0, ShCt);
14271*9880d681SAndroid Build Coastguard Worker AddToWorklist(Shift.getNode());
14272*9880d681SAndroid Build Coastguard Worker
14273*9880d681SAndroid Build Coastguard Worker if (XType.bitsGT(AType)) {
14274*9880d681SAndroid Build Coastguard Worker Shift = DAG.getNode(ISD::TRUNCATE, DL, AType, Shift);
14275*9880d681SAndroid Build Coastguard Worker AddToWorklist(Shift.getNode());
14276*9880d681SAndroid Build Coastguard Worker }
14277*9880d681SAndroid Build Coastguard Worker
14278*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, DL, AType, Shift, N2);
14279*9880d681SAndroid Build Coastguard Worker }
14280*9880d681SAndroid Build Coastguard Worker
14281*9880d681SAndroid Build Coastguard Worker SDValue Shift = DAG.getNode(ISD::SRA, SDLoc(N0),
14282*9880d681SAndroid Build Coastguard Worker XType, N0,
14283*9880d681SAndroid Build Coastguard Worker DAG.getConstant(XType.getSizeInBits() - 1,
14284*9880d681SAndroid Build Coastguard Worker SDLoc(N0),
14285*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(N0.getValueType())));
14286*9880d681SAndroid Build Coastguard Worker AddToWorklist(Shift.getNode());
14287*9880d681SAndroid Build Coastguard Worker
14288*9880d681SAndroid Build Coastguard Worker if (XType.bitsGT(AType)) {
14289*9880d681SAndroid Build Coastguard Worker Shift = DAG.getNode(ISD::TRUNCATE, DL, AType, Shift);
14290*9880d681SAndroid Build Coastguard Worker AddToWorklist(Shift.getNode());
14291*9880d681SAndroid Build Coastguard Worker }
14292*9880d681SAndroid Build Coastguard Worker
14293*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, DL, AType, Shift, N2);
14294*9880d681SAndroid Build Coastguard Worker }
14295*9880d681SAndroid Build Coastguard Worker }
14296*9880d681SAndroid Build Coastguard Worker
14297*9880d681SAndroid Build Coastguard Worker // fold (select_cc seteq (and x, y), 0, 0, A) -> (and (shr (shl x)) A)
14298*9880d681SAndroid Build Coastguard Worker // where y is has a single bit set.
14299*9880d681SAndroid Build Coastguard Worker // A plaintext description would be, we can turn the SELECT_CC into an AND
14300*9880d681SAndroid Build Coastguard Worker // when the condition can be materialized as an all-ones register. Any
14301*9880d681SAndroid Build Coastguard Worker // single bit-test can be materialized as an all-ones register with
14302*9880d681SAndroid Build Coastguard Worker // shift-left and shift-right-arith.
14303*9880d681SAndroid Build Coastguard Worker if (CC == ISD::SETEQ && N0->getOpcode() == ISD::AND &&
14304*9880d681SAndroid Build Coastguard Worker N0->getValueType(0) == VT && isNullConstant(N1) && isNullConstant(N2)) {
14305*9880d681SAndroid Build Coastguard Worker SDValue AndLHS = N0->getOperand(0);
14306*9880d681SAndroid Build Coastguard Worker ConstantSDNode *ConstAndRHS = dyn_cast<ConstantSDNode>(N0->getOperand(1));
14307*9880d681SAndroid Build Coastguard Worker if (ConstAndRHS && ConstAndRHS->getAPIntValue().countPopulation() == 1) {
14308*9880d681SAndroid Build Coastguard Worker // Shift the tested bit over the sign bit.
14309*9880d681SAndroid Build Coastguard Worker const APInt &AndMask = ConstAndRHS->getAPIntValue();
14310*9880d681SAndroid Build Coastguard Worker SDValue ShlAmt =
14311*9880d681SAndroid Build Coastguard Worker DAG.getConstant(AndMask.countLeadingZeros(), SDLoc(AndLHS),
14312*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(AndLHS.getValueType()));
14313*9880d681SAndroid Build Coastguard Worker SDValue Shl = DAG.getNode(ISD::SHL, SDLoc(N0), VT, AndLHS, ShlAmt);
14314*9880d681SAndroid Build Coastguard Worker
14315*9880d681SAndroid Build Coastguard Worker // Now arithmetic right shift it all the way over, so the result is either
14316*9880d681SAndroid Build Coastguard Worker // all-ones, or zero.
14317*9880d681SAndroid Build Coastguard Worker SDValue ShrAmt =
14318*9880d681SAndroid Build Coastguard Worker DAG.getConstant(AndMask.getBitWidth() - 1, SDLoc(Shl),
14319*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(Shl.getValueType()));
14320*9880d681SAndroid Build Coastguard Worker SDValue Shr = DAG.getNode(ISD::SRA, SDLoc(N0), VT, Shl, ShrAmt);
14321*9880d681SAndroid Build Coastguard Worker
14322*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::AND, DL, VT, Shr, N3);
14323*9880d681SAndroid Build Coastguard Worker }
14324*9880d681SAndroid Build Coastguard Worker }
14325*9880d681SAndroid Build Coastguard Worker
14326*9880d681SAndroid Build Coastguard Worker // fold select C, 16, 0 -> shl C, 4
14327*9880d681SAndroid Build Coastguard Worker if (N2C && isNullConstant(N3) && N2C->getAPIntValue().isPowerOf2() &&
14328*9880d681SAndroid Build Coastguard Worker TLI.getBooleanContents(N0.getValueType()) ==
14329*9880d681SAndroid Build Coastguard Worker TargetLowering::ZeroOrOneBooleanContent) {
14330*9880d681SAndroid Build Coastguard Worker
14331*9880d681SAndroid Build Coastguard Worker // If the caller doesn't want us to simplify this into a zext of a compare,
14332*9880d681SAndroid Build Coastguard Worker // don't do it.
14333*9880d681SAndroid Build Coastguard Worker if (NotExtCompare && N2C->isOne())
14334*9880d681SAndroid Build Coastguard Worker return SDValue();
14335*9880d681SAndroid Build Coastguard Worker
14336*9880d681SAndroid Build Coastguard Worker // Get a SetCC of the condition
14337*9880d681SAndroid Build Coastguard Worker // NOTE: Don't create a SETCC if it's not legal on this target.
14338*9880d681SAndroid Build Coastguard Worker if (!LegalOperations ||
14339*9880d681SAndroid Build Coastguard Worker TLI.isOperationLegal(ISD::SETCC, N0.getValueType())) {
14340*9880d681SAndroid Build Coastguard Worker SDValue Temp, SCC;
14341*9880d681SAndroid Build Coastguard Worker // cast from setcc result type to select result type
14342*9880d681SAndroid Build Coastguard Worker if (LegalTypes) {
14343*9880d681SAndroid Build Coastguard Worker SCC = DAG.getSetCC(DL, getSetCCResultType(N0.getValueType()),
14344*9880d681SAndroid Build Coastguard Worker N0, N1, CC);
14345*9880d681SAndroid Build Coastguard Worker if (N2.getValueType().bitsLT(SCC.getValueType()))
14346*9880d681SAndroid Build Coastguard Worker Temp = DAG.getZeroExtendInReg(SCC, SDLoc(N2),
14347*9880d681SAndroid Build Coastguard Worker N2.getValueType());
14348*9880d681SAndroid Build Coastguard Worker else
14349*9880d681SAndroid Build Coastguard Worker Temp = DAG.getNode(ISD::ZERO_EXTEND, SDLoc(N2),
14350*9880d681SAndroid Build Coastguard Worker N2.getValueType(), SCC);
14351*9880d681SAndroid Build Coastguard Worker } else {
14352*9880d681SAndroid Build Coastguard Worker SCC = DAG.getSetCC(SDLoc(N0), MVT::i1, N0, N1, CC);
14353*9880d681SAndroid Build Coastguard Worker Temp = DAG.getNode(ISD::ZERO_EXTEND, SDLoc(N2),
14354*9880d681SAndroid Build Coastguard Worker N2.getValueType(), SCC);
14355*9880d681SAndroid Build Coastguard Worker }
14356*9880d681SAndroid Build Coastguard Worker
14357*9880d681SAndroid Build Coastguard Worker AddToWorklist(SCC.getNode());
14358*9880d681SAndroid Build Coastguard Worker AddToWorklist(Temp.getNode());
14359*9880d681SAndroid Build Coastguard Worker
14360*9880d681SAndroid Build Coastguard Worker if (N2C->isOne())
14361*9880d681SAndroid Build Coastguard Worker return Temp;
14362*9880d681SAndroid Build Coastguard Worker
14363*9880d681SAndroid Build Coastguard Worker // shl setcc result by log2 n2c
14364*9880d681SAndroid Build Coastguard Worker return DAG.getNode(
14365*9880d681SAndroid Build Coastguard Worker ISD::SHL, DL, N2.getValueType(), Temp,
14366*9880d681SAndroid Build Coastguard Worker DAG.getConstant(N2C->getAPIntValue().logBase2(), SDLoc(Temp),
14367*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(Temp.getValueType())));
14368*9880d681SAndroid Build Coastguard Worker }
14369*9880d681SAndroid Build Coastguard Worker }
14370*9880d681SAndroid Build Coastguard Worker
14371*9880d681SAndroid Build Coastguard Worker // Check to see if this is an integer abs.
14372*9880d681SAndroid Build Coastguard Worker // select_cc setg[te] X, 0, X, -X ->
14373*9880d681SAndroid Build Coastguard Worker // select_cc setgt X, -1, X, -X ->
14374*9880d681SAndroid Build Coastguard Worker // select_cc setl[te] X, 0, -X, X ->
14375*9880d681SAndroid Build Coastguard Worker // select_cc setlt X, 1, -X, X ->
14376*9880d681SAndroid Build Coastguard Worker // Y = sra (X, size(X)-1); xor (add (X, Y), Y)
14377*9880d681SAndroid Build Coastguard Worker if (N1C) {
14378*9880d681SAndroid Build Coastguard Worker ConstantSDNode *SubC = nullptr;
14379*9880d681SAndroid Build Coastguard Worker if (((N1C->isNullValue() && (CC == ISD::SETGT || CC == ISD::SETGE)) ||
14380*9880d681SAndroid Build Coastguard Worker (N1C->isAllOnesValue() && CC == ISD::SETGT)) &&
14381*9880d681SAndroid Build Coastguard Worker N0 == N2 && N3.getOpcode() == ISD::SUB && N0 == N3.getOperand(1))
14382*9880d681SAndroid Build Coastguard Worker SubC = dyn_cast<ConstantSDNode>(N3.getOperand(0));
14383*9880d681SAndroid Build Coastguard Worker else if (((N1C->isNullValue() && (CC == ISD::SETLT || CC == ISD::SETLE)) ||
14384*9880d681SAndroid Build Coastguard Worker (N1C->isOne() && CC == ISD::SETLT)) &&
14385*9880d681SAndroid Build Coastguard Worker N0 == N3 && N2.getOpcode() == ISD::SUB && N0 == N2.getOperand(1))
14386*9880d681SAndroid Build Coastguard Worker SubC = dyn_cast<ConstantSDNode>(N2.getOperand(0));
14387*9880d681SAndroid Build Coastguard Worker
14388*9880d681SAndroid Build Coastguard Worker EVT XType = N0.getValueType();
14389*9880d681SAndroid Build Coastguard Worker if (SubC && SubC->isNullValue() && XType.isInteger()) {
14390*9880d681SAndroid Build Coastguard Worker SDLoc DL(N0);
14391*9880d681SAndroid Build Coastguard Worker SDValue Shift = DAG.getNode(ISD::SRA, DL, XType,
14392*9880d681SAndroid Build Coastguard Worker N0,
14393*9880d681SAndroid Build Coastguard Worker DAG.getConstant(XType.getSizeInBits() - 1, DL,
14394*9880d681SAndroid Build Coastguard Worker getShiftAmountTy(N0.getValueType())));
14395*9880d681SAndroid Build Coastguard Worker SDValue Add = DAG.getNode(ISD::ADD, DL,
14396*9880d681SAndroid Build Coastguard Worker XType, N0, Shift);
14397*9880d681SAndroid Build Coastguard Worker AddToWorklist(Shift.getNode());
14398*9880d681SAndroid Build Coastguard Worker AddToWorklist(Add.getNode());
14399*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::XOR, DL, XType, Add, Shift);
14400*9880d681SAndroid Build Coastguard Worker }
14401*9880d681SAndroid Build Coastguard Worker }
14402*9880d681SAndroid Build Coastguard Worker
14403*9880d681SAndroid Build Coastguard Worker // select_cc seteq X, 0, sizeof(X), ctlz(X) -> ctlz(X)
14404*9880d681SAndroid Build Coastguard Worker // select_cc seteq X, 0, sizeof(X), ctlz_zero_undef(X) -> ctlz(X)
14405*9880d681SAndroid Build Coastguard Worker // select_cc seteq X, 0, sizeof(X), cttz(X) -> cttz(X)
14406*9880d681SAndroid Build Coastguard Worker // select_cc seteq X, 0, sizeof(X), cttz_zero_undef(X) -> cttz(X)
14407*9880d681SAndroid Build Coastguard Worker // select_cc setne X, 0, ctlz(X), sizeof(X) -> ctlz(X)
14408*9880d681SAndroid Build Coastguard Worker // select_cc setne X, 0, ctlz_zero_undef(X), sizeof(X) -> ctlz(X)
14409*9880d681SAndroid Build Coastguard Worker // select_cc setne X, 0, cttz(X), sizeof(X) -> cttz(X)
14410*9880d681SAndroid Build Coastguard Worker // select_cc setne X, 0, cttz_zero_undef(X), sizeof(X) -> cttz(X)
14411*9880d681SAndroid Build Coastguard Worker if (N1C && N1C->isNullValue() && (CC == ISD::SETEQ || CC == ISD::SETNE)) {
14412*9880d681SAndroid Build Coastguard Worker SDValue ValueOnZero = N2;
14413*9880d681SAndroid Build Coastguard Worker SDValue Count = N3;
14414*9880d681SAndroid Build Coastguard Worker // If the condition is NE instead of E, swap the operands.
14415*9880d681SAndroid Build Coastguard Worker if (CC == ISD::SETNE)
14416*9880d681SAndroid Build Coastguard Worker std::swap(ValueOnZero, Count);
14417*9880d681SAndroid Build Coastguard Worker // Check if the value on zero is a constant equal to the bits in the type.
14418*9880d681SAndroid Build Coastguard Worker if (auto *ValueOnZeroC = dyn_cast<ConstantSDNode>(ValueOnZero)) {
14419*9880d681SAndroid Build Coastguard Worker if (ValueOnZeroC->getAPIntValue() == VT.getSizeInBits()) {
14420*9880d681SAndroid Build Coastguard Worker // If the other operand is cttz/cttz_zero_undef of N0, and cttz is
14421*9880d681SAndroid Build Coastguard Worker // legal, combine to just cttz.
14422*9880d681SAndroid Build Coastguard Worker if ((Count.getOpcode() == ISD::CTTZ ||
14423*9880d681SAndroid Build Coastguard Worker Count.getOpcode() == ISD::CTTZ_ZERO_UNDEF) &&
14424*9880d681SAndroid Build Coastguard Worker N0 == Count.getOperand(0) &&
14425*9880d681SAndroid Build Coastguard Worker (!LegalOperations || TLI.isOperationLegal(ISD::CTTZ, VT)))
14426*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::CTTZ, DL, VT, N0);
14427*9880d681SAndroid Build Coastguard Worker // If the other operand is ctlz/ctlz_zero_undef of N0, and ctlz is
14428*9880d681SAndroid Build Coastguard Worker // legal, combine to just ctlz.
14429*9880d681SAndroid Build Coastguard Worker if ((Count.getOpcode() == ISD::CTLZ ||
14430*9880d681SAndroid Build Coastguard Worker Count.getOpcode() == ISD::CTLZ_ZERO_UNDEF) &&
14431*9880d681SAndroid Build Coastguard Worker N0 == Count.getOperand(0) &&
14432*9880d681SAndroid Build Coastguard Worker (!LegalOperations || TLI.isOperationLegal(ISD::CTLZ, VT)))
14433*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::CTLZ, DL, VT, N0);
14434*9880d681SAndroid Build Coastguard Worker }
14435*9880d681SAndroid Build Coastguard Worker }
14436*9880d681SAndroid Build Coastguard Worker }
14437*9880d681SAndroid Build Coastguard Worker
14438*9880d681SAndroid Build Coastguard Worker return SDValue();
14439*9880d681SAndroid Build Coastguard Worker }
14440*9880d681SAndroid Build Coastguard Worker
14441*9880d681SAndroid Build Coastguard Worker /// This is a stub for TargetLowering::SimplifySetCC.
SimplifySetCC(EVT VT,SDValue N0,SDValue N1,ISD::CondCode Cond,const SDLoc & DL,bool foldBooleans)14442*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::SimplifySetCC(EVT VT, SDValue N0, SDValue N1,
14443*9880d681SAndroid Build Coastguard Worker ISD::CondCode Cond, const SDLoc &DL,
14444*9880d681SAndroid Build Coastguard Worker bool foldBooleans) {
14445*9880d681SAndroid Build Coastguard Worker TargetLowering::DAGCombinerInfo
14446*9880d681SAndroid Build Coastguard Worker DagCombineInfo(DAG, Level, false, this);
14447*9880d681SAndroid Build Coastguard Worker return TLI.SimplifySetCC(VT, N0, N1, Cond, foldBooleans, DagCombineInfo, DL);
14448*9880d681SAndroid Build Coastguard Worker }
14449*9880d681SAndroid Build Coastguard Worker
14450*9880d681SAndroid Build Coastguard Worker /// Given an ISD::SDIV node expressing a divide by constant, return
14451*9880d681SAndroid Build Coastguard Worker /// a DAG expression to select that will generate the same value by multiplying
14452*9880d681SAndroid Build Coastguard Worker /// by a magic number.
14453*9880d681SAndroid Build Coastguard Worker /// Ref: "Hacker's Delight" or "The PowerPC Compiler Writer's Guide".
BuildSDIV(SDNode * N)14454*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::BuildSDIV(SDNode *N) {
14455*9880d681SAndroid Build Coastguard Worker // when optimising for minimum size, we don't want to expand a div to a mul
14456*9880d681SAndroid Build Coastguard Worker // and a shift.
14457*9880d681SAndroid Build Coastguard Worker if (DAG.getMachineFunction().getFunction()->optForMinSize())
14458*9880d681SAndroid Build Coastguard Worker return SDValue();
14459*9880d681SAndroid Build Coastguard Worker
14460*9880d681SAndroid Build Coastguard Worker ConstantSDNode *C = isConstOrConstSplat(N->getOperand(1));
14461*9880d681SAndroid Build Coastguard Worker if (!C)
14462*9880d681SAndroid Build Coastguard Worker return SDValue();
14463*9880d681SAndroid Build Coastguard Worker
14464*9880d681SAndroid Build Coastguard Worker // Avoid division by zero.
14465*9880d681SAndroid Build Coastguard Worker if (C->isNullValue())
14466*9880d681SAndroid Build Coastguard Worker return SDValue();
14467*9880d681SAndroid Build Coastguard Worker
14468*9880d681SAndroid Build Coastguard Worker std::vector<SDNode*> Built;
14469*9880d681SAndroid Build Coastguard Worker SDValue S =
14470*9880d681SAndroid Build Coastguard Worker TLI.BuildSDIV(N, C->getAPIntValue(), DAG, LegalOperations, &Built);
14471*9880d681SAndroid Build Coastguard Worker
14472*9880d681SAndroid Build Coastguard Worker for (SDNode *N : Built)
14473*9880d681SAndroid Build Coastguard Worker AddToWorklist(N);
14474*9880d681SAndroid Build Coastguard Worker return S;
14475*9880d681SAndroid Build Coastguard Worker }
14476*9880d681SAndroid Build Coastguard Worker
14477*9880d681SAndroid Build Coastguard Worker /// Given an ISD::SDIV node expressing a divide by constant power of 2, return a
14478*9880d681SAndroid Build Coastguard Worker /// DAG expression that will generate the same value by right shifting.
BuildSDIVPow2(SDNode * N)14479*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::BuildSDIVPow2(SDNode *N) {
14480*9880d681SAndroid Build Coastguard Worker ConstantSDNode *C = isConstOrConstSplat(N->getOperand(1));
14481*9880d681SAndroid Build Coastguard Worker if (!C)
14482*9880d681SAndroid Build Coastguard Worker return SDValue();
14483*9880d681SAndroid Build Coastguard Worker
14484*9880d681SAndroid Build Coastguard Worker // Avoid division by zero.
14485*9880d681SAndroid Build Coastguard Worker if (C->isNullValue())
14486*9880d681SAndroid Build Coastguard Worker return SDValue();
14487*9880d681SAndroid Build Coastguard Worker
14488*9880d681SAndroid Build Coastguard Worker std::vector<SDNode *> Built;
14489*9880d681SAndroid Build Coastguard Worker SDValue S = TLI.BuildSDIVPow2(N, C->getAPIntValue(), DAG, &Built);
14490*9880d681SAndroid Build Coastguard Worker
14491*9880d681SAndroid Build Coastguard Worker for (SDNode *N : Built)
14492*9880d681SAndroid Build Coastguard Worker AddToWorklist(N);
14493*9880d681SAndroid Build Coastguard Worker return S;
14494*9880d681SAndroid Build Coastguard Worker }
14495*9880d681SAndroid Build Coastguard Worker
14496*9880d681SAndroid Build Coastguard Worker /// Given an ISD::UDIV node expressing a divide by constant, return a DAG
14497*9880d681SAndroid Build Coastguard Worker /// expression that will generate the same value by multiplying by a magic
14498*9880d681SAndroid Build Coastguard Worker /// number.
14499*9880d681SAndroid Build Coastguard Worker /// Ref: "Hacker's Delight" or "The PowerPC Compiler Writer's Guide".
BuildUDIV(SDNode * N)14500*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::BuildUDIV(SDNode *N) {
14501*9880d681SAndroid Build Coastguard Worker // when optimising for minimum size, we don't want to expand a div to a mul
14502*9880d681SAndroid Build Coastguard Worker // and a shift.
14503*9880d681SAndroid Build Coastguard Worker if (DAG.getMachineFunction().getFunction()->optForMinSize())
14504*9880d681SAndroid Build Coastguard Worker return SDValue();
14505*9880d681SAndroid Build Coastguard Worker
14506*9880d681SAndroid Build Coastguard Worker ConstantSDNode *C = isConstOrConstSplat(N->getOperand(1));
14507*9880d681SAndroid Build Coastguard Worker if (!C)
14508*9880d681SAndroid Build Coastguard Worker return SDValue();
14509*9880d681SAndroid Build Coastguard Worker
14510*9880d681SAndroid Build Coastguard Worker // Avoid division by zero.
14511*9880d681SAndroid Build Coastguard Worker if (C->isNullValue())
14512*9880d681SAndroid Build Coastguard Worker return SDValue();
14513*9880d681SAndroid Build Coastguard Worker
14514*9880d681SAndroid Build Coastguard Worker std::vector<SDNode*> Built;
14515*9880d681SAndroid Build Coastguard Worker SDValue S =
14516*9880d681SAndroid Build Coastguard Worker TLI.BuildUDIV(N, C->getAPIntValue(), DAG, LegalOperations, &Built);
14517*9880d681SAndroid Build Coastguard Worker
14518*9880d681SAndroid Build Coastguard Worker for (SDNode *N : Built)
14519*9880d681SAndroid Build Coastguard Worker AddToWorklist(N);
14520*9880d681SAndroid Build Coastguard Worker return S;
14521*9880d681SAndroid Build Coastguard Worker }
14522*9880d681SAndroid Build Coastguard Worker
BuildReciprocalEstimate(SDValue Op,SDNodeFlags * Flags)14523*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::BuildReciprocalEstimate(SDValue Op, SDNodeFlags *Flags) {
14524*9880d681SAndroid Build Coastguard Worker if (Level >= AfterLegalizeDAG)
14525*9880d681SAndroid Build Coastguard Worker return SDValue();
14526*9880d681SAndroid Build Coastguard Worker
14527*9880d681SAndroid Build Coastguard Worker // Expose the DAG combiner to the target combiner implementations.
14528*9880d681SAndroid Build Coastguard Worker TargetLowering::DAGCombinerInfo DCI(DAG, Level, false, this);
14529*9880d681SAndroid Build Coastguard Worker
14530*9880d681SAndroid Build Coastguard Worker unsigned Iterations = 0;
14531*9880d681SAndroid Build Coastguard Worker if (SDValue Est = TLI.getRecipEstimate(Op, DCI, Iterations)) {
14532*9880d681SAndroid Build Coastguard Worker if (Iterations) {
14533*9880d681SAndroid Build Coastguard Worker // Newton iteration for a function: F(X) is X_{i+1} = X_i - F(X_i)/F'(X_i)
14534*9880d681SAndroid Build Coastguard Worker // For the reciprocal, we need to find the zero of the function:
14535*9880d681SAndroid Build Coastguard Worker // F(X) = A X - 1 [which has a zero at X = 1/A]
14536*9880d681SAndroid Build Coastguard Worker // =>
14537*9880d681SAndroid Build Coastguard Worker // X_{i+1} = X_i (2 - A X_i) = X_i + X_i (1 - A X_i) [this second form
14538*9880d681SAndroid Build Coastguard Worker // does not require additional intermediate precision]
14539*9880d681SAndroid Build Coastguard Worker EVT VT = Op.getValueType();
14540*9880d681SAndroid Build Coastguard Worker SDLoc DL(Op);
14541*9880d681SAndroid Build Coastguard Worker SDValue FPOne = DAG.getConstantFP(1.0, DL, VT);
14542*9880d681SAndroid Build Coastguard Worker
14543*9880d681SAndroid Build Coastguard Worker AddToWorklist(Est.getNode());
14544*9880d681SAndroid Build Coastguard Worker
14545*9880d681SAndroid Build Coastguard Worker // Newton iterations: Est = Est + Est (1 - Arg * Est)
14546*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < Iterations; ++i) {
14547*9880d681SAndroid Build Coastguard Worker SDValue NewEst = DAG.getNode(ISD::FMUL, DL, VT, Op, Est, Flags);
14548*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewEst.getNode());
14549*9880d681SAndroid Build Coastguard Worker
14550*9880d681SAndroid Build Coastguard Worker NewEst = DAG.getNode(ISD::FSUB, DL, VT, FPOne, NewEst, Flags);
14551*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewEst.getNode());
14552*9880d681SAndroid Build Coastguard Worker
14553*9880d681SAndroid Build Coastguard Worker NewEst = DAG.getNode(ISD::FMUL, DL, VT, Est, NewEst, Flags);
14554*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewEst.getNode());
14555*9880d681SAndroid Build Coastguard Worker
14556*9880d681SAndroid Build Coastguard Worker Est = DAG.getNode(ISD::FADD, DL, VT, Est, NewEst, Flags);
14557*9880d681SAndroid Build Coastguard Worker AddToWorklist(Est.getNode());
14558*9880d681SAndroid Build Coastguard Worker }
14559*9880d681SAndroid Build Coastguard Worker }
14560*9880d681SAndroid Build Coastguard Worker return Est;
14561*9880d681SAndroid Build Coastguard Worker }
14562*9880d681SAndroid Build Coastguard Worker
14563*9880d681SAndroid Build Coastguard Worker return SDValue();
14564*9880d681SAndroid Build Coastguard Worker }
14565*9880d681SAndroid Build Coastguard Worker
14566*9880d681SAndroid Build Coastguard Worker /// Newton iteration for a function: F(X) is X_{i+1} = X_i - F(X_i)/F'(X_i)
14567*9880d681SAndroid Build Coastguard Worker /// For the reciprocal sqrt, we need to find the zero of the function:
14568*9880d681SAndroid Build Coastguard Worker /// F(X) = 1/X^2 - A [which has a zero at X = 1/sqrt(A)]
14569*9880d681SAndroid Build Coastguard Worker /// =>
14570*9880d681SAndroid Build Coastguard Worker /// X_{i+1} = X_i (1.5 - A X_i^2 / 2)
14571*9880d681SAndroid Build Coastguard Worker /// As a result, we precompute A/2 prior to the iteration loop.
buildSqrtNROneConst(SDValue Arg,SDValue Est,unsigned Iterations,SDNodeFlags * Flags,bool Reciprocal)14572*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::buildSqrtNROneConst(SDValue Arg, SDValue Est,
14573*9880d681SAndroid Build Coastguard Worker unsigned Iterations,
14574*9880d681SAndroid Build Coastguard Worker SDNodeFlags *Flags, bool Reciprocal) {
14575*9880d681SAndroid Build Coastguard Worker EVT VT = Arg.getValueType();
14576*9880d681SAndroid Build Coastguard Worker SDLoc DL(Arg);
14577*9880d681SAndroid Build Coastguard Worker SDValue ThreeHalves = DAG.getConstantFP(1.5, DL, VT);
14578*9880d681SAndroid Build Coastguard Worker
14579*9880d681SAndroid Build Coastguard Worker // We now need 0.5 * Arg which we can write as (1.5 * Arg - Arg) so that
14580*9880d681SAndroid Build Coastguard Worker // this entire sequence requires only one FP constant.
14581*9880d681SAndroid Build Coastguard Worker SDValue HalfArg = DAG.getNode(ISD::FMUL, DL, VT, ThreeHalves, Arg, Flags);
14582*9880d681SAndroid Build Coastguard Worker AddToWorklist(HalfArg.getNode());
14583*9880d681SAndroid Build Coastguard Worker
14584*9880d681SAndroid Build Coastguard Worker HalfArg = DAG.getNode(ISD::FSUB, DL, VT, HalfArg, Arg, Flags);
14585*9880d681SAndroid Build Coastguard Worker AddToWorklist(HalfArg.getNode());
14586*9880d681SAndroid Build Coastguard Worker
14587*9880d681SAndroid Build Coastguard Worker // Newton iterations: Est = Est * (1.5 - HalfArg * Est * Est)
14588*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < Iterations; ++i) {
14589*9880d681SAndroid Build Coastguard Worker SDValue NewEst = DAG.getNode(ISD::FMUL, DL, VT, Est, Est, Flags);
14590*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewEst.getNode());
14591*9880d681SAndroid Build Coastguard Worker
14592*9880d681SAndroid Build Coastguard Worker NewEst = DAG.getNode(ISD::FMUL, DL, VT, HalfArg, NewEst, Flags);
14593*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewEst.getNode());
14594*9880d681SAndroid Build Coastguard Worker
14595*9880d681SAndroid Build Coastguard Worker NewEst = DAG.getNode(ISD::FSUB, DL, VT, ThreeHalves, NewEst, Flags);
14596*9880d681SAndroid Build Coastguard Worker AddToWorklist(NewEst.getNode());
14597*9880d681SAndroid Build Coastguard Worker
14598*9880d681SAndroid Build Coastguard Worker Est = DAG.getNode(ISD::FMUL, DL, VT, Est, NewEst, Flags);
14599*9880d681SAndroid Build Coastguard Worker AddToWorklist(Est.getNode());
14600*9880d681SAndroid Build Coastguard Worker }
14601*9880d681SAndroid Build Coastguard Worker
14602*9880d681SAndroid Build Coastguard Worker // If non-reciprocal square root is requested, multiply the result by Arg.
14603*9880d681SAndroid Build Coastguard Worker if (!Reciprocal) {
14604*9880d681SAndroid Build Coastguard Worker Est = DAG.getNode(ISD::FMUL, DL, VT, Est, Arg, Flags);
14605*9880d681SAndroid Build Coastguard Worker AddToWorklist(Est.getNode());
14606*9880d681SAndroid Build Coastguard Worker }
14607*9880d681SAndroid Build Coastguard Worker
14608*9880d681SAndroid Build Coastguard Worker return Est;
14609*9880d681SAndroid Build Coastguard Worker }
14610*9880d681SAndroid Build Coastguard Worker
14611*9880d681SAndroid Build Coastguard Worker /// Newton iteration for a function: F(X) is X_{i+1} = X_i - F(X_i)/F'(X_i)
14612*9880d681SAndroid Build Coastguard Worker /// For the reciprocal sqrt, we need to find the zero of the function:
14613*9880d681SAndroid Build Coastguard Worker /// F(X) = 1/X^2 - A [which has a zero at X = 1/sqrt(A)]
14614*9880d681SAndroid Build Coastguard Worker /// =>
14615*9880d681SAndroid Build Coastguard Worker /// X_{i+1} = (-0.5 * X_i) * (A * X_i * X_i + (-3.0))
buildSqrtNRTwoConst(SDValue Arg,SDValue Est,unsigned Iterations,SDNodeFlags * Flags,bool Reciprocal)14616*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::buildSqrtNRTwoConst(SDValue Arg, SDValue Est,
14617*9880d681SAndroid Build Coastguard Worker unsigned Iterations,
14618*9880d681SAndroid Build Coastguard Worker SDNodeFlags *Flags, bool Reciprocal) {
14619*9880d681SAndroid Build Coastguard Worker EVT VT = Arg.getValueType();
14620*9880d681SAndroid Build Coastguard Worker SDLoc DL(Arg);
14621*9880d681SAndroid Build Coastguard Worker SDValue MinusThree = DAG.getConstantFP(-3.0, DL, VT);
14622*9880d681SAndroid Build Coastguard Worker SDValue MinusHalf = DAG.getConstantFP(-0.5, DL, VT);
14623*9880d681SAndroid Build Coastguard Worker
14624*9880d681SAndroid Build Coastguard Worker // This routine must enter the loop below to work correctly
14625*9880d681SAndroid Build Coastguard Worker // when (Reciprocal == false).
14626*9880d681SAndroid Build Coastguard Worker assert(Iterations > 0);
14627*9880d681SAndroid Build Coastguard Worker
14628*9880d681SAndroid Build Coastguard Worker // Newton iterations for reciprocal square root:
14629*9880d681SAndroid Build Coastguard Worker // E = (E * -0.5) * ((A * E) * E + -3.0)
14630*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < Iterations; ++i) {
14631*9880d681SAndroid Build Coastguard Worker SDValue AE = DAG.getNode(ISD::FMUL, DL, VT, Arg, Est, Flags);
14632*9880d681SAndroid Build Coastguard Worker AddToWorklist(AE.getNode());
14633*9880d681SAndroid Build Coastguard Worker
14634*9880d681SAndroid Build Coastguard Worker SDValue AEE = DAG.getNode(ISD::FMUL, DL, VT, AE, Est, Flags);
14635*9880d681SAndroid Build Coastguard Worker AddToWorklist(AEE.getNode());
14636*9880d681SAndroid Build Coastguard Worker
14637*9880d681SAndroid Build Coastguard Worker SDValue RHS = DAG.getNode(ISD::FADD, DL, VT, AEE, MinusThree, Flags);
14638*9880d681SAndroid Build Coastguard Worker AddToWorklist(RHS.getNode());
14639*9880d681SAndroid Build Coastguard Worker
14640*9880d681SAndroid Build Coastguard Worker // When calculating a square root at the last iteration build:
14641*9880d681SAndroid Build Coastguard Worker // S = ((A * E) * -0.5) * ((A * E) * E + -3.0)
14642*9880d681SAndroid Build Coastguard Worker // (notice a common subexpression)
14643*9880d681SAndroid Build Coastguard Worker SDValue LHS;
14644*9880d681SAndroid Build Coastguard Worker if (Reciprocal || (i + 1) < Iterations) {
14645*9880d681SAndroid Build Coastguard Worker // RSQRT: LHS = (E * -0.5)
14646*9880d681SAndroid Build Coastguard Worker LHS = DAG.getNode(ISD::FMUL, DL, VT, Est, MinusHalf, Flags);
14647*9880d681SAndroid Build Coastguard Worker } else {
14648*9880d681SAndroid Build Coastguard Worker // SQRT: LHS = (A * E) * -0.5
14649*9880d681SAndroid Build Coastguard Worker LHS = DAG.getNode(ISD::FMUL, DL, VT, AE, MinusHalf, Flags);
14650*9880d681SAndroid Build Coastguard Worker }
14651*9880d681SAndroid Build Coastguard Worker AddToWorklist(LHS.getNode());
14652*9880d681SAndroid Build Coastguard Worker
14653*9880d681SAndroid Build Coastguard Worker Est = DAG.getNode(ISD::FMUL, DL, VT, LHS, RHS, Flags);
14654*9880d681SAndroid Build Coastguard Worker AddToWorklist(Est.getNode());
14655*9880d681SAndroid Build Coastguard Worker }
14656*9880d681SAndroid Build Coastguard Worker
14657*9880d681SAndroid Build Coastguard Worker return Est;
14658*9880d681SAndroid Build Coastguard Worker }
14659*9880d681SAndroid Build Coastguard Worker
14660*9880d681SAndroid Build Coastguard Worker /// Build code to calculate either rsqrt(Op) or sqrt(Op). In the latter case
14661*9880d681SAndroid Build Coastguard Worker /// Op*rsqrt(Op) is actually computed, so additional postprocessing is needed if
14662*9880d681SAndroid Build Coastguard Worker /// Op can be zero.
buildSqrtEstimateImpl(SDValue Op,SDNodeFlags * Flags,bool Reciprocal)14663*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::buildSqrtEstimateImpl(SDValue Op, SDNodeFlags *Flags,
14664*9880d681SAndroid Build Coastguard Worker bool Reciprocal) {
14665*9880d681SAndroid Build Coastguard Worker if (Level >= AfterLegalizeDAG)
14666*9880d681SAndroid Build Coastguard Worker return SDValue();
14667*9880d681SAndroid Build Coastguard Worker
14668*9880d681SAndroid Build Coastguard Worker // Expose the DAG combiner to the target combiner implementations.
14669*9880d681SAndroid Build Coastguard Worker TargetLowering::DAGCombinerInfo DCI(DAG, Level, false, this);
14670*9880d681SAndroid Build Coastguard Worker unsigned Iterations = 0;
14671*9880d681SAndroid Build Coastguard Worker bool UseOneConstNR = false;
14672*9880d681SAndroid Build Coastguard Worker if (SDValue Est = TLI.getRsqrtEstimate(Op, DCI, Iterations, UseOneConstNR)) {
14673*9880d681SAndroid Build Coastguard Worker AddToWorklist(Est.getNode());
14674*9880d681SAndroid Build Coastguard Worker if (Iterations) {
14675*9880d681SAndroid Build Coastguard Worker Est = UseOneConstNR
14676*9880d681SAndroid Build Coastguard Worker ? buildSqrtNROneConst(Op, Est, Iterations, Flags, Reciprocal)
14677*9880d681SAndroid Build Coastguard Worker : buildSqrtNRTwoConst(Op, Est, Iterations, Flags, Reciprocal);
14678*9880d681SAndroid Build Coastguard Worker }
14679*9880d681SAndroid Build Coastguard Worker return Est;
14680*9880d681SAndroid Build Coastguard Worker }
14681*9880d681SAndroid Build Coastguard Worker
14682*9880d681SAndroid Build Coastguard Worker return SDValue();
14683*9880d681SAndroid Build Coastguard Worker }
14684*9880d681SAndroid Build Coastguard Worker
buildRsqrtEstimate(SDValue Op,SDNodeFlags * Flags)14685*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::buildRsqrtEstimate(SDValue Op, SDNodeFlags *Flags) {
14686*9880d681SAndroid Build Coastguard Worker return buildSqrtEstimateImpl(Op, Flags, true);
14687*9880d681SAndroid Build Coastguard Worker }
14688*9880d681SAndroid Build Coastguard Worker
buildSqrtEstimate(SDValue Op,SDNodeFlags * Flags)14689*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::buildSqrtEstimate(SDValue Op, SDNodeFlags *Flags) {
14690*9880d681SAndroid Build Coastguard Worker SDValue Est = buildSqrtEstimateImpl(Op, Flags, false);
14691*9880d681SAndroid Build Coastguard Worker if (!Est)
14692*9880d681SAndroid Build Coastguard Worker return SDValue();
14693*9880d681SAndroid Build Coastguard Worker
14694*9880d681SAndroid Build Coastguard Worker // Unfortunately, Est is now NaN if the input was exactly 0.
14695*9880d681SAndroid Build Coastguard Worker // Select out this case and force the answer to 0.
14696*9880d681SAndroid Build Coastguard Worker EVT VT = Est.getValueType();
14697*9880d681SAndroid Build Coastguard Worker SDLoc DL(Op);
14698*9880d681SAndroid Build Coastguard Worker SDValue Zero = DAG.getConstantFP(0.0, DL, VT);
14699*9880d681SAndroid Build Coastguard Worker EVT CCVT = getSetCCResultType(VT);
14700*9880d681SAndroid Build Coastguard Worker SDValue ZeroCmp = DAG.getSetCC(DL, CCVT, Op, Zero, ISD::SETEQ);
14701*9880d681SAndroid Build Coastguard Worker AddToWorklist(ZeroCmp.getNode());
14702*9880d681SAndroid Build Coastguard Worker
14703*9880d681SAndroid Build Coastguard Worker Est = DAG.getNode(VT.isVector() ? ISD::VSELECT : ISD::SELECT, DL, VT, ZeroCmp,
14704*9880d681SAndroid Build Coastguard Worker Zero, Est);
14705*9880d681SAndroid Build Coastguard Worker AddToWorklist(Est.getNode());
14706*9880d681SAndroid Build Coastguard Worker return Est;
14707*9880d681SAndroid Build Coastguard Worker }
14708*9880d681SAndroid Build Coastguard Worker
14709*9880d681SAndroid Build Coastguard Worker /// Return true if base is a frame index, which is known not to alias with
14710*9880d681SAndroid Build Coastguard Worker /// anything but itself. Provides base object and offset as results.
FindBaseOffset(SDValue Ptr,SDValue & Base,int64_t & Offset,const GlobalValue * & GV,const void * & CV)14711*9880d681SAndroid Build Coastguard Worker static bool FindBaseOffset(SDValue Ptr, SDValue &Base, int64_t &Offset,
14712*9880d681SAndroid Build Coastguard Worker const GlobalValue *&GV, const void *&CV) {
14713*9880d681SAndroid Build Coastguard Worker // Assume it is a primitive operation.
14714*9880d681SAndroid Build Coastguard Worker Base = Ptr; Offset = 0; GV = nullptr; CV = nullptr;
14715*9880d681SAndroid Build Coastguard Worker
14716*9880d681SAndroid Build Coastguard Worker // If it's an adding a simple constant then integrate the offset.
14717*9880d681SAndroid Build Coastguard Worker if (Base.getOpcode() == ISD::ADD) {
14718*9880d681SAndroid Build Coastguard Worker if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Base.getOperand(1))) {
14719*9880d681SAndroid Build Coastguard Worker Base = Base.getOperand(0);
14720*9880d681SAndroid Build Coastguard Worker Offset += C->getZExtValue();
14721*9880d681SAndroid Build Coastguard Worker }
14722*9880d681SAndroid Build Coastguard Worker }
14723*9880d681SAndroid Build Coastguard Worker
14724*9880d681SAndroid Build Coastguard Worker // Return the underlying GlobalValue, and update the Offset. Return false
14725*9880d681SAndroid Build Coastguard Worker // for GlobalAddressSDNode since the same GlobalAddress may be represented
14726*9880d681SAndroid Build Coastguard Worker // by multiple nodes with different offsets.
14727*9880d681SAndroid Build Coastguard Worker if (GlobalAddressSDNode *G = dyn_cast<GlobalAddressSDNode>(Base)) {
14728*9880d681SAndroid Build Coastguard Worker GV = G->getGlobal();
14729*9880d681SAndroid Build Coastguard Worker Offset += G->getOffset();
14730*9880d681SAndroid Build Coastguard Worker return false;
14731*9880d681SAndroid Build Coastguard Worker }
14732*9880d681SAndroid Build Coastguard Worker
14733*9880d681SAndroid Build Coastguard Worker // Return the underlying Constant value, and update the Offset. Return false
14734*9880d681SAndroid Build Coastguard Worker // for ConstantSDNodes since the same constant pool entry may be represented
14735*9880d681SAndroid Build Coastguard Worker // by multiple nodes with different offsets.
14736*9880d681SAndroid Build Coastguard Worker if (ConstantPoolSDNode *C = dyn_cast<ConstantPoolSDNode>(Base)) {
14737*9880d681SAndroid Build Coastguard Worker CV = C->isMachineConstantPoolEntry() ? (const void *)C->getMachineCPVal()
14738*9880d681SAndroid Build Coastguard Worker : (const void *)C->getConstVal();
14739*9880d681SAndroid Build Coastguard Worker Offset += C->getOffset();
14740*9880d681SAndroid Build Coastguard Worker return false;
14741*9880d681SAndroid Build Coastguard Worker }
14742*9880d681SAndroid Build Coastguard Worker // If it's any of the following then it can't alias with anything but itself.
14743*9880d681SAndroid Build Coastguard Worker return isa<FrameIndexSDNode>(Base);
14744*9880d681SAndroid Build Coastguard Worker }
14745*9880d681SAndroid Build Coastguard Worker
14746*9880d681SAndroid Build Coastguard Worker /// Return true if there is any possibility that the two addresses overlap.
isAlias(LSBaseSDNode * Op0,LSBaseSDNode * Op1) const14747*9880d681SAndroid Build Coastguard Worker bool DAGCombiner::isAlias(LSBaseSDNode *Op0, LSBaseSDNode *Op1) const {
14748*9880d681SAndroid Build Coastguard Worker // If they are the same then they must be aliases.
14749*9880d681SAndroid Build Coastguard Worker if (Op0->getBasePtr() == Op1->getBasePtr()) return true;
14750*9880d681SAndroid Build Coastguard Worker
14751*9880d681SAndroid Build Coastguard Worker // If they are both volatile then they cannot be reordered.
14752*9880d681SAndroid Build Coastguard Worker if (Op0->isVolatile() && Op1->isVolatile()) return true;
14753*9880d681SAndroid Build Coastguard Worker
14754*9880d681SAndroid Build Coastguard Worker // If one operation reads from invariant memory, and the other may store, they
14755*9880d681SAndroid Build Coastguard Worker // cannot alias. These should really be checking the equivalent of mayWrite,
14756*9880d681SAndroid Build Coastguard Worker // but it only matters for memory nodes other than load /store.
14757*9880d681SAndroid Build Coastguard Worker if (Op0->isInvariant() && Op1->writeMem())
14758*9880d681SAndroid Build Coastguard Worker return false;
14759*9880d681SAndroid Build Coastguard Worker
14760*9880d681SAndroid Build Coastguard Worker if (Op1->isInvariant() && Op0->writeMem())
14761*9880d681SAndroid Build Coastguard Worker return false;
14762*9880d681SAndroid Build Coastguard Worker
14763*9880d681SAndroid Build Coastguard Worker // Gather base node and offset information.
14764*9880d681SAndroid Build Coastguard Worker SDValue Base1, Base2;
14765*9880d681SAndroid Build Coastguard Worker int64_t Offset1, Offset2;
14766*9880d681SAndroid Build Coastguard Worker const GlobalValue *GV1, *GV2;
14767*9880d681SAndroid Build Coastguard Worker const void *CV1, *CV2;
14768*9880d681SAndroid Build Coastguard Worker bool isFrameIndex1 = FindBaseOffset(Op0->getBasePtr(),
14769*9880d681SAndroid Build Coastguard Worker Base1, Offset1, GV1, CV1);
14770*9880d681SAndroid Build Coastguard Worker bool isFrameIndex2 = FindBaseOffset(Op1->getBasePtr(),
14771*9880d681SAndroid Build Coastguard Worker Base2, Offset2, GV2, CV2);
14772*9880d681SAndroid Build Coastguard Worker
14773*9880d681SAndroid Build Coastguard Worker // If they have a same base address then check to see if they overlap.
14774*9880d681SAndroid Build Coastguard Worker if (Base1 == Base2 || (GV1 && (GV1 == GV2)) || (CV1 && (CV1 == CV2)))
14775*9880d681SAndroid Build Coastguard Worker return !((Offset1 + (Op0->getMemoryVT().getSizeInBits() >> 3)) <= Offset2 ||
14776*9880d681SAndroid Build Coastguard Worker (Offset2 + (Op1->getMemoryVT().getSizeInBits() >> 3)) <= Offset1);
14777*9880d681SAndroid Build Coastguard Worker
14778*9880d681SAndroid Build Coastguard Worker // It is possible for different frame indices to alias each other, mostly
14779*9880d681SAndroid Build Coastguard Worker // when tail call optimization reuses return address slots for arguments.
14780*9880d681SAndroid Build Coastguard Worker // To catch this case, look up the actual index of frame indices to compute
14781*9880d681SAndroid Build Coastguard Worker // the real alias relationship.
14782*9880d681SAndroid Build Coastguard Worker if (isFrameIndex1 && isFrameIndex2) {
14783*9880d681SAndroid Build Coastguard Worker MachineFrameInfo *MFI = DAG.getMachineFunction().getFrameInfo();
14784*9880d681SAndroid Build Coastguard Worker Offset1 += MFI->getObjectOffset(cast<FrameIndexSDNode>(Base1)->getIndex());
14785*9880d681SAndroid Build Coastguard Worker Offset2 += MFI->getObjectOffset(cast<FrameIndexSDNode>(Base2)->getIndex());
14786*9880d681SAndroid Build Coastguard Worker return !((Offset1 + (Op0->getMemoryVT().getSizeInBits() >> 3)) <= Offset2 ||
14787*9880d681SAndroid Build Coastguard Worker (Offset2 + (Op1->getMemoryVT().getSizeInBits() >> 3)) <= Offset1);
14788*9880d681SAndroid Build Coastguard Worker }
14789*9880d681SAndroid Build Coastguard Worker
14790*9880d681SAndroid Build Coastguard Worker // Otherwise, if we know what the bases are, and they aren't identical, then
14791*9880d681SAndroid Build Coastguard Worker // we know they cannot alias.
14792*9880d681SAndroid Build Coastguard Worker if ((isFrameIndex1 || CV1 || GV1) && (isFrameIndex2 || CV2 || GV2))
14793*9880d681SAndroid Build Coastguard Worker return false;
14794*9880d681SAndroid Build Coastguard Worker
14795*9880d681SAndroid Build Coastguard Worker // If we know required SrcValue1 and SrcValue2 have relatively large alignment
14796*9880d681SAndroid Build Coastguard Worker // compared to the size and offset of the access, we may be able to prove they
14797*9880d681SAndroid Build Coastguard Worker // do not alias. This check is conservative for now to catch cases created by
14798*9880d681SAndroid Build Coastguard Worker // splitting vector types.
14799*9880d681SAndroid Build Coastguard Worker if ((Op0->getOriginalAlignment() == Op1->getOriginalAlignment()) &&
14800*9880d681SAndroid Build Coastguard Worker (Op0->getSrcValueOffset() != Op1->getSrcValueOffset()) &&
14801*9880d681SAndroid Build Coastguard Worker (Op0->getMemoryVT().getSizeInBits() >> 3 ==
14802*9880d681SAndroid Build Coastguard Worker Op1->getMemoryVT().getSizeInBits() >> 3) &&
14803*9880d681SAndroid Build Coastguard Worker (Op0->getOriginalAlignment() > (Op0->getMemoryVT().getSizeInBits() >> 3))) {
14804*9880d681SAndroid Build Coastguard Worker int64_t OffAlign1 = Op0->getSrcValueOffset() % Op0->getOriginalAlignment();
14805*9880d681SAndroid Build Coastguard Worker int64_t OffAlign2 = Op1->getSrcValueOffset() % Op1->getOriginalAlignment();
14806*9880d681SAndroid Build Coastguard Worker
14807*9880d681SAndroid Build Coastguard Worker // There is no overlap between these relatively aligned accesses of similar
14808*9880d681SAndroid Build Coastguard Worker // size, return no alias.
14809*9880d681SAndroid Build Coastguard Worker if ((OffAlign1 + (Op0->getMemoryVT().getSizeInBits() >> 3)) <= OffAlign2 ||
14810*9880d681SAndroid Build Coastguard Worker (OffAlign2 + (Op1->getMemoryVT().getSizeInBits() >> 3)) <= OffAlign1)
14811*9880d681SAndroid Build Coastguard Worker return false;
14812*9880d681SAndroid Build Coastguard Worker }
14813*9880d681SAndroid Build Coastguard Worker
14814*9880d681SAndroid Build Coastguard Worker bool UseAA = CombinerGlobalAA.getNumOccurrences() > 0
14815*9880d681SAndroid Build Coastguard Worker ? CombinerGlobalAA
14816*9880d681SAndroid Build Coastguard Worker : DAG.getSubtarget().useAA();
14817*9880d681SAndroid Build Coastguard Worker #ifndef NDEBUG
14818*9880d681SAndroid Build Coastguard Worker if (CombinerAAOnlyFunc.getNumOccurrences() &&
14819*9880d681SAndroid Build Coastguard Worker CombinerAAOnlyFunc != DAG.getMachineFunction().getName())
14820*9880d681SAndroid Build Coastguard Worker UseAA = false;
14821*9880d681SAndroid Build Coastguard Worker #endif
14822*9880d681SAndroid Build Coastguard Worker if (UseAA &&
14823*9880d681SAndroid Build Coastguard Worker Op0->getMemOperand()->getValue() && Op1->getMemOperand()->getValue()) {
14824*9880d681SAndroid Build Coastguard Worker // Use alias analysis information.
14825*9880d681SAndroid Build Coastguard Worker int64_t MinOffset = std::min(Op0->getSrcValueOffset(),
14826*9880d681SAndroid Build Coastguard Worker Op1->getSrcValueOffset());
14827*9880d681SAndroid Build Coastguard Worker int64_t Overlap1 = (Op0->getMemoryVT().getSizeInBits() >> 3) +
14828*9880d681SAndroid Build Coastguard Worker Op0->getSrcValueOffset() - MinOffset;
14829*9880d681SAndroid Build Coastguard Worker int64_t Overlap2 = (Op1->getMemoryVT().getSizeInBits() >> 3) +
14830*9880d681SAndroid Build Coastguard Worker Op1->getSrcValueOffset() - MinOffset;
14831*9880d681SAndroid Build Coastguard Worker AliasResult AAResult =
14832*9880d681SAndroid Build Coastguard Worker AA.alias(MemoryLocation(Op0->getMemOperand()->getValue(), Overlap1,
14833*9880d681SAndroid Build Coastguard Worker UseTBAA ? Op0->getAAInfo() : AAMDNodes()),
14834*9880d681SAndroid Build Coastguard Worker MemoryLocation(Op1->getMemOperand()->getValue(), Overlap2,
14835*9880d681SAndroid Build Coastguard Worker UseTBAA ? Op1->getAAInfo() : AAMDNodes()));
14836*9880d681SAndroid Build Coastguard Worker if (AAResult == NoAlias)
14837*9880d681SAndroid Build Coastguard Worker return false;
14838*9880d681SAndroid Build Coastguard Worker }
14839*9880d681SAndroid Build Coastguard Worker
14840*9880d681SAndroid Build Coastguard Worker // Otherwise we have to assume they alias.
14841*9880d681SAndroid Build Coastguard Worker return true;
14842*9880d681SAndroid Build Coastguard Worker }
14843*9880d681SAndroid Build Coastguard Worker
14844*9880d681SAndroid Build Coastguard Worker /// Walk up chain skipping non-aliasing memory nodes,
14845*9880d681SAndroid Build Coastguard Worker /// looking for aliasing nodes and adding them to the Aliases vector.
GatherAllAliases(SDNode * N,SDValue OriginalChain,SmallVectorImpl<SDValue> & Aliases)14846*9880d681SAndroid Build Coastguard Worker void DAGCombiner::GatherAllAliases(SDNode *N, SDValue OriginalChain,
14847*9880d681SAndroid Build Coastguard Worker SmallVectorImpl<SDValue> &Aliases) {
14848*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Chains; // List of chains to visit.
14849*9880d681SAndroid Build Coastguard Worker SmallPtrSet<SDNode *, 16> Visited; // Visited node set.
14850*9880d681SAndroid Build Coastguard Worker
14851*9880d681SAndroid Build Coastguard Worker // Get alias information for node.
14852*9880d681SAndroid Build Coastguard Worker bool IsLoad = isa<LoadSDNode>(N) && !cast<LSBaseSDNode>(N)->isVolatile();
14853*9880d681SAndroid Build Coastguard Worker
14854*9880d681SAndroid Build Coastguard Worker // Starting off.
14855*9880d681SAndroid Build Coastguard Worker Chains.push_back(OriginalChain);
14856*9880d681SAndroid Build Coastguard Worker unsigned Depth = 0;
14857*9880d681SAndroid Build Coastguard Worker
14858*9880d681SAndroid Build Coastguard Worker // Look at each chain and determine if it is an alias. If so, add it to the
14859*9880d681SAndroid Build Coastguard Worker // aliases list. If not, then continue up the chain looking for the next
14860*9880d681SAndroid Build Coastguard Worker // candidate.
14861*9880d681SAndroid Build Coastguard Worker while (!Chains.empty()) {
14862*9880d681SAndroid Build Coastguard Worker SDValue Chain = Chains.pop_back_val();
14863*9880d681SAndroid Build Coastguard Worker
14864*9880d681SAndroid Build Coastguard Worker // For TokenFactor nodes, look at each operand and only continue up the
14865*9880d681SAndroid Build Coastguard Worker // chain until we reach the depth limit.
14866*9880d681SAndroid Build Coastguard Worker //
14867*9880d681SAndroid Build Coastguard Worker // FIXME: The depth check could be made to return the last non-aliasing
14868*9880d681SAndroid Build Coastguard Worker // chain we found before we hit a tokenfactor rather than the original
14869*9880d681SAndroid Build Coastguard Worker // chain.
14870*9880d681SAndroid Build Coastguard Worker if (Depth > TLI.getGatherAllAliasesMaxDepth()) {
14871*9880d681SAndroid Build Coastguard Worker Aliases.clear();
14872*9880d681SAndroid Build Coastguard Worker Aliases.push_back(OriginalChain);
14873*9880d681SAndroid Build Coastguard Worker return;
14874*9880d681SAndroid Build Coastguard Worker }
14875*9880d681SAndroid Build Coastguard Worker
14876*9880d681SAndroid Build Coastguard Worker // Don't bother if we've been before.
14877*9880d681SAndroid Build Coastguard Worker if (!Visited.insert(Chain.getNode()).second)
14878*9880d681SAndroid Build Coastguard Worker continue;
14879*9880d681SAndroid Build Coastguard Worker
14880*9880d681SAndroid Build Coastguard Worker switch (Chain.getOpcode()) {
14881*9880d681SAndroid Build Coastguard Worker case ISD::EntryToken:
14882*9880d681SAndroid Build Coastguard Worker // Entry token is ideal chain operand, but handled in FindBetterChain.
14883*9880d681SAndroid Build Coastguard Worker break;
14884*9880d681SAndroid Build Coastguard Worker
14885*9880d681SAndroid Build Coastguard Worker case ISD::LOAD:
14886*9880d681SAndroid Build Coastguard Worker case ISD::STORE: {
14887*9880d681SAndroid Build Coastguard Worker // Get alias information for Chain.
14888*9880d681SAndroid Build Coastguard Worker bool IsOpLoad = isa<LoadSDNode>(Chain.getNode()) &&
14889*9880d681SAndroid Build Coastguard Worker !cast<LSBaseSDNode>(Chain.getNode())->isVolatile();
14890*9880d681SAndroid Build Coastguard Worker
14891*9880d681SAndroid Build Coastguard Worker // If chain is alias then stop here.
14892*9880d681SAndroid Build Coastguard Worker if (!(IsLoad && IsOpLoad) &&
14893*9880d681SAndroid Build Coastguard Worker isAlias(cast<LSBaseSDNode>(N), cast<LSBaseSDNode>(Chain.getNode()))) {
14894*9880d681SAndroid Build Coastguard Worker Aliases.push_back(Chain);
14895*9880d681SAndroid Build Coastguard Worker } else {
14896*9880d681SAndroid Build Coastguard Worker // Look further up the chain.
14897*9880d681SAndroid Build Coastguard Worker Chains.push_back(Chain.getOperand(0));
14898*9880d681SAndroid Build Coastguard Worker ++Depth;
14899*9880d681SAndroid Build Coastguard Worker }
14900*9880d681SAndroid Build Coastguard Worker break;
14901*9880d681SAndroid Build Coastguard Worker }
14902*9880d681SAndroid Build Coastguard Worker
14903*9880d681SAndroid Build Coastguard Worker case ISD::TokenFactor:
14904*9880d681SAndroid Build Coastguard Worker // We have to check each of the operands of the token factor for "small"
14905*9880d681SAndroid Build Coastguard Worker // token factors, so we queue them up. Adding the operands to the queue
14906*9880d681SAndroid Build Coastguard Worker // (stack) in reverse order maintains the original order and increases the
14907*9880d681SAndroid Build Coastguard Worker // likelihood that getNode will find a matching token factor (CSE.)
14908*9880d681SAndroid Build Coastguard Worker if (Chain.getNumOperands() > 16) {
14909*9880d681SAndroid Build Coastguard Worker Aliases.push_back(Chain);
14910*9880d681SAndroid Build Coastguard Worker break;
14911*9880d681SAndroid Build Coastguard Worker }
14912*9880d681SAndroid Build Coastguard Worker for (unsigned n = Chain.getNumOperands(); n;)
14913*9880d681SAndroid Build Coastguard Worker Chains.push_back(Chain.getOperand(--n));
14914*9880d681SAndroid Build Coastguard Worker ++Depth;
14915*9880d681SAndroid Build Coastguard Worker break;
14916*9880d681SAndroid Build Coastguard Worker
14917*9880d681SAndroid Build Coastguard Worker default:
14918*9880d681SAndroid Build Coastguard Worker // For all other instructions we will just have to take what we can get.
14919*9880d681SAndroid Build Coastguard Worker Aliases.push_back(Chain);
14920*9880d681SAndroid Build Coastguard Worker break;
14921*9880d681SAndroid Build Coastguard Worker }
14922*9880d681SAndroid Build Coastguard Worker }
14923*9880d681SAndroid Build Coastguard Worker }
14924*9880d681SAndroid Build Coastguard Worker
14925*9880d681SAndroid Build Coastguard Worker /// Walk up chain skipping non-aliasing memory nodes, looking for a better chain
14926*9880d681SAndroid Build Coastguard Worker /// (aliasing node.)
FindBetterChain(SDNode * N,SDValue OldChain)14927*9880d681SAndroid Build Coastguard Worker SDValue DAGCombiner::FindBetterChain(SDNode *N, SDValue OldChain) {
14928*9880d681SAndroid Build Coastguard Worker SmallVector<SDValue, 8> Aliases; // Ops for replacing token factor.
14929*9880d681SAndroid Build Coastguard Worker
14930*9880d681SAndroid Build Coastguard Worker // Accumulate all the aliases to this node.
14931*9880d681SAndroid Build Coastguard Worker GatherAllAliases(N, OldChain, Aliases);
14932*9880d681SAndroid Build Coastguard Worker
14933*9880d681SAndroid Build Coastguard Worker // If no operands then chain to entry token.
14934*9880d681SAndroid Build Coastguard Worker if (Aliases.size() == 0)
14935*9880d681SAndroid Build Coastguard Worker return DAG.getEntryNode();
14936*9880d681SAndroid Build Coastguard Worker
14937*9880d681SAndroid Build Coastguard Worker // If a single operand then chain to it. We don't need to revisit it.
14938*9880d681SAndroid Build Coastguard Worker if (Aliases.size() == 1)
14939*9880d681SAndroid Build Coastguard Worker return Aliases[0];
14940*9880d681SAndroid Build Coastguard Worker
14941*9880d681SAndroid Build Coastguard Worker // Construct a custom tailored token factor.
14942*9880d681SAndroid Build Coastguard Worker return DAG.getNode(ISD::TokenFactor, SDLoc(N), MVT::Other, Aliases);
14943*9880d681SAndroid Build Coastguard Worker }
14944*9880d681SAndroid Build Coastguard Worker
findBetterNeighborChains(StoreSDNode * St)14945*9880d681SAndroid Build Coastguard Worker bool DAGCombiner::findBetterNeighborChains(StoreSDNode *St) {
14946*9880d681SAndroid Build Coastguard Worker // This holds the base pointer, index, and the offset in bytes from the base
14947*9880d681SAndroid Build Coastguard Worker // pointer.
14948*9880d681SAndroid Build Coastguard Worker BaseIndexOffset BasePtr = BaseIndexOffset::match(St->getBasePtr(), DAG);
14949*9880d681SAndroid Build Coastguard Worker
14950*9880d681SAndroid Build Coastguard Worker // We must have a base and an offset.
14951*9880d681SAndroid Build Coastguard Worker if (!BasePtr.Base.getNode())
14952*9880d681SAndroid Build Coastguard Worker return false;
14953*9880d681SAndroid Build Coastguard Worker
14954*9880d681SAndroid Build Coastguard Worker // Do not handle stores to undef base pointers.
14955*9880d681SAndroid Build Coastguard Worker if (BasePtr.Base.isUndef())
14956*9880d681SAndroid Build Coastguard Worker return false;
14957*9880d681SAndroid Build Coastguard Worker
14958*9880d681SAndroid Build Coastguard Worker SmallVector<StoreSDNode *, 8> ChainedStores;
14959*9880d681SAndroid Build Coastguard Worker ChainedStores.push_back(St);
14960*9880d681SAndroid Build Coastguard Worker
14961*9880d681SAndroid Build Coastguard Worker // Walk up the chain and look for nodes with offsets from the same
14962*9880d681SAndroid Build Coastguard Worker // base pointer. Stop when reaching an instruction with a different kind
14963*9880d681SAndroid Build Coastguard Worker // or instruction which has a different base pointer.
14964*9880d681SAndroid Build Coastguard Worker StoreSDNode *Index = St;
14965*9880d681SAndroid Build Coastguard Worker while (Index) {
14966*9880d681SAndroid Build Coastguard Worker // If the chain has more than one use, then we can't reorder the mem ops.
14967*9880d681SAndroid Build Coastguard Worker if (Index != St && !SDValue(Index, 0)->hasOneUse())
14968*9880d681SAndroid Build Coastguard Worker break;
14969*9880d681SAndroid Build Coastguard Worker
14970*9880d681SAndroid Build Coastguard Worker if (Index->isVolatile() || Index->isIndexed())
14971*9880d681SAndroid Build Coastguard Worker break;
14972*9880d681SAndroid Build Coastguard Worker
14973*9880d681SAndroid Build Coastguard Worker // Find the base pointer and offset for this memory node.
14974*9880d681SAndroid Build Coastguard Worker BaseIndexOffset Ptr = BaseIndexOffset::match(Index->getBasePtr(), DAG);
14975*9880d681SAndroid Build Coastguard Worker
14976*9880d681SAndroid Build Coastguard Worker // Check that the base pointer is the same as the original one.
14977*9880d681SAndroid Build Coastguard Worker if (!Ptr.equalBaseIndex(BasePtr))
14978*9880d681SAndroid Build Coastguard Worker break;
14979*9880d681SAndroid Build Coastguard Worker
14980*9880d681SAndroid Build Coastguard Worker // Find the next memory operand in the chain. If the next operand in the
14981*9880d681SAndroid Build Coastguard Worker // chain is a store then move up and continue the scan with the next
14982*9880d681SAndroid Build Coastguard Worker // memory operand. If the next operand is a load save it and use alias
14983*9880d681SAndroid Build Coastguard Worker // information to check if it interferes with anything.
14984*9880d681SAndroid Build Coastguard Worker SDNode *NextInChain = Index->getChain().getNode();
14985*9880d681SAndroid Build Coastguard Worker while (true) {
14986*9880d681SAndroid Build Coastguard Worker if (StoreSDNode *STn = dyn_cast<StoreSDNode>(NextInChain)) {
14987*9880d681SAndroid Build Coastguard Worker // We found a store node. Use it for the next iteration.
14988*9880d681SAndroid Build Coastguard Worker if (STn->isVolatile() || STn->isIndexed()) {
14989*9880d681SAndroid Build Coastguard Worker Index = nullptr;
14990*9880d681SAndroid Build Coastguard Worker break;
14991*9880d681SAndroid Build Coastguard Worker }
14992*9880d681SAndroid Build Coastguard Worker ChainedStores.push_back(STn);
14993*9880d681SAndroid Build Coastguard Worker Index = STn;
14994*9880d681SAndroid Build Coastguard Worker break;
14995*9880d681SAndroid Build Coastguard Worker } else if (LoadSDNode *Ldn = dyn_cast<LoadSDNode>(NextInChain)) {
14996*9880d681SAndroid Build Coastguard Worker NextInChain = Ldn->getChain().getNode();
14997*9880d681SAndroid Build Coastguard Worker continue;
14998*9880d681SAndroid Build Coastguard Worker } else {
14999*9880d681SAndroid Build Coastguard Worker Index = nullptr;
15000*9880d681SAndroid Build Coastguard Worker break;
15001*9880d681SAndroid Build Coastguard Worker }
15002*9880d681SAndroid Build Coastguard Worker }
15003*9880d681SAndroid Build Coastguard Worker }
15004*9880d681SAndroid Build Coastguard Worker
15005*9880d681SAndroid Build Coastguard Worker bool MadeChangeToSt = false;
15006*9880d681SAndroid Build Coastguard Worker SmallVector<std::pair<StoreSDNode *, SDValue>, 8> BetterChains;
15007*9880d681SAndroid Build Coastguard Worker
15008*9880d681SAndroid Build Coastguard Worker for (StoreSDNode *ChainedStore : ChainedStores) {
15009*9880d681SAndroid Build Coastguard Worker SDValue Chain = ChainedStore->getChain();
15010*9880d681SAndroid Build Coastguard Worker SDValue BetterChain = FindBetterChain(ChainedStore, Chain);
15011*9880d681SAndroid Build Coastguard Worker
15012*9880d681SAndroid Build Coastguard Worker if (Chain != BetterChain) {
15013*9880d681SAndroid Build Coastguard Worker if (ChainedStore == St)
15014*9880d681SAndroid Build Coastguard Worker MadeChangeToSt = true;
15015*9880d681SAndroid Build Coastguard Worker BetterChains.push_back(std::make_pair(ChainedStore, BetterChain));
15016*9880d681SAndroid Build Coastguard Worker }
15017*9880d681SAndroid Build Coastguard Worker }
15018*9880d681SAndroid Build Coastguard Worker
15019*9880d681SAndroid Build Coastguard Worker // Do all replacements after finding the replacements to make to avoid making
15020*9880d681SAndroid Build Coastguard Worker // the chains more complicated by introducing new TokenFactors.
15021*9880d681SAndroid Build Coastguard Worker for (auto Replacement : BetterChains)
15022*9880d681SAndroid Build Coastguard Worker replaceStoreChain(Replacement.first, Replacement.second);
15023*9880d681SAndroid Build Coastguard Worker
15024*9880d681SAndroid Build Coastguard Worker return MadeChangeToSt;
15025*9880d681SAndroid Build Coastguard Worker }
15026*9880d681SAndroid Build Coastguard Worker
15027*9880d681SAndroid Build Coastguard Worker /// This is the entry point for the file.
Combine(CombineLevel Level,AliasAnalysis & AA,CodeGenOpt::Level OptLevel)15028*9880d681SAndroid Build Coastguard Worker void SelectionDAG::Combine(CombineLevel Level, AliasAnalysis &AA,
15029*9880d681SAndroid Build Coastguard Worker CodeGenOpt::Level OptLevel) {
15030*9880d681SAndroid Build Coastguard Worker /// This is the main entry point to this class.
15031*9880d681SAndroid Build Coastguard Worker DAGCombiner(*this, AA, OptLevel).Run(Level);
15032*9880d681SAndroid Build Coastguard Worker }
15033