xref: /aosp_15_r20/external/llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp (revision 9880d6810fe72a1726cb53787c6711e909410d58)
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