xref: /freebsd/contrib/llvm-project/llvm/lib/Target/AMDGPU/AMDGPUISelLowering.cpp (revision 8a4dda33d67586ca2624f2a38417baa03a533a7f)
10b57cec5SDimitry Andric //===-- AMDGPUISelLowering.cpp - AMDGPU Common DAG lowering functions -----===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
80b57cec5SDimitry Andric //
90b57cec5SDimitry Andric /// \file
100b57cec5SDimitry Andric /// This is the parent TargetLowering class for hardware code gen
110b57cec5SDimitry Andric /// targets.
120b57cec5SDimitry Andric //
130b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
140b57cec5SDimitry Andric 
150b57cec5SDimitry Andric #include "AMDGPUISelLowering.h"
160b57cec5SDimitry Andric #include "AMDGPU.h"
17e8d8bef9SDimitry Andric #include "AMDGPUInstrInfo.h"
18e8d8bef9SDimitry Andric #include "AMDGPUMachineFunction.h"
190b57cec5SDimitry Andric #include "SIMachineFunctionInfo.h"
200b57cec5SDimitry Andric #include "llvm/CodeGen/Analysis.h"
2106c3fb27SDimitry Andric #include "llvm/CodeGen/GlobalISel/GISelKnownBits.h"
2281ad6265SDimitry Andric #include "llvm/CodeGen/MachineFrameInfo.h"
230b57cec5SDimitry Andric #include "llvm/IR/DiagnosticInfo.h"
24e8d8bef9SDimitry Andric #include "llvm/IR/IntrinsicsAMDGPU.h"
2506c3fb27SDimitry Andric #include "llvm/IR/PatternMatch.h"
26e8d8bef9SDimitry Andric #include "llvm/Support/CommandLine.h"
270b57cec5SDimitry Andric #include "llvm/Support/KnownBits.h"
28e8d8bef9SDimitry Andric #include "llvm/Target/TargetMachine.h"
29e8d8bef9SDimitry Andric 
300b57cec5SDimitry Andric using namespace llvm;
310b57cec5SDimitry Andric 
320b57cec5SDimitry Andric #include "AMDGPUGenCallingConv.inc"
330b57cec5SDimitry Andric 
345ffd83dbSDimitry Andric static cl::opt<bool> AMDGPUBypassSlowDiv(
355ffd83dbSDimitry Andric   "amdgpu-bypass-slow-div",
365ffd83dbSDimitry Andric   cl::desc("Skip 64-bit divide for dynamic 32-bit values"),
375ffd83dbSDimitry Andric   cl::init(true));
385ffd83dbSDimitry Andric 
390b57cec5SDimitry Andric // Find a larger type to do a load / store of a vector with.
400b57cec5SDimitry Andric EVT AMDGPUTargetLowering::getEquivalentMemType(LLVMContext &Ctx, EVT VT) {
410b57cec5SDimitry Andric   unsigned StoreSize = VT.getStoreSizeInBits();
420b57cec5SDimitry Andric   if (StoreSize <= 32)
430b57cec5SDimitry Andric     return EVT::getIntegerVT(Ctx, StoreSize);
440b57cec5SDimitry Andric 
450b57cec5SDimitry Andric   assert(StoreSize % 32 == 0 && "Store size not a multiple of 32");
460b57cec5SDimitry Andric   return EVT::getVectorVT(Ctx, MVT::i32, StoreSize / 32);
470b57cec5SDimitry Andric }
480b57cec5SDimitry Andric 
490b57cec5SDimitry Andric unsigned AMDGPUTargetLowering::numBitsUnsigned(SDValue Op, SelectionDAG &DAG) {
50349cc55cSDimitry Andric   return DAG.computeKnownBits(Op).countMaxActiveBits();
510b57cec5SDimitry Andric }
520b57cec5SDimitry Andric 
530b57cec5SDimitry Andric unsigned AMDGPUTargetLowering::numBitsSigned(SDValue Op, SelectionDAG &DAG) {
540b57cec5SDimitry Andric   // In order for this to be a signed 24-bit value, bit 23, must
550b57cec5SDimitry Andric   // be a sign bit.
5604eeddc0SDimitry Andric   return DAG.ComputeMaxSignificantBits(Op);
570b57cec5SDimitry Andric }
580b57cec5SDimitry Andric 
590b57cec5SDimitry Andric AMDGPUTargetLowering::AMDGPUTargetLowering(const TargetMachine &TM,
600b57cec5SDimitry Andric                                            const AMDGPUSubtarget &STI)
610b57cec5SDimitry Andric     : TargetLowering(TM), Subtarget(&STI) {
620b57cec5SDimitry Andric   // Lower floating point store/load to integer store/load to reduce the number
630b57cec5SDimitry Andric   // of patterns in tablegen.
640b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::f32, Promote);
650b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::f32, MVT::i32);
660b57cec5SDimitry Andric 
670b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v2f32, Promote);
680b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v2f32, MVT::v2i32);
690b57cec5SDimitry Andric 
700b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v3f32, Promote);
710b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v3f32, MVT::v3i32);
720b57cec5SDimitry Andric 
730b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v4f32, Promote);
740b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v4f32, MVT::v4i32);
750b57cec5SDimitry Andric 
760b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v5f32, Promote);
770b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v5f32, MVT::v5i32);
780b57cec5SDimitry Andric 
79fe6060f1SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v6f32, Promote);
80fe6060f1SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v6f32, MVT::v6i32);
81fe6060f1SDimitry Andric 
82fe6060f1SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v7f32, Promote);
83fe6060f1SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v7f32, MVT::v7i32);
84fe6060f1SDimitry Andric 
850b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v8f32, Promote);
860b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v8f32, MVT::v8i32);
870b57cec5SDimitry Andric 
88bdd1243dSDimitry Andric   setOperationAction(ISD::LOAD, MVT::v9f32, Promote);
89bdd1243dSDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v9f32, MVT::v9i32);
90bdd1243dSDimitry Andric 
91bdd1243dSDimitry Andric   setOperationAction(ISD::LOAD, MVT::v10f32, Promote);
92bdd1243dSDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v10f32, MVT::v10i32);
93bdd1243dSDimitry Andric 
94bdd1243dSDimitry Andric   setOperationAction(ISD::LOAD, MVT::v11f32, Promote);
95bdd1243dSDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v11f32, MVT::v11i32);
96bdd1243dSDimitry Andric 
97bdd1243dSDimitry Andric   setOperationAction(ISD::LOAD, MVT::v12f32, Promote);
98bdd1243dSDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v12f32, MVT::v12i32);
99bdd1243dSDimitry Andric 
1000b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v16f32, Promote);
1010b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v16f32, MVT::v16i32);
1020b57cec5SDimitry Andric 
1030b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v32f32, Promote);
1040b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v32f32, MVT::v32i32);
1050b57cec5SDimitry Andric 
1060b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::i64, Promote);
1070b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::i64, MVT::v2i32);
1080b57cec5SDimitry Andric 
1090b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v2i64, Promote);
1100b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v2i64, MVT::v4i32);
1110b57cec5SDimitry Andric 
1120b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::f64, Promote);
1130b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::f64, MVT::v2i32);
1140b57cec5SDimitry Andric 
1150b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v2f64, Promote);
1160b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v2f64, MVT::v4i32);
1170b57cec5SDimitry Andric 
118fe6060f1SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v3i64, Promote);
119fe6060f1SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v3i64, MVT::v6i32);
120fe6060f1SDimitry Andric 
1215ffd83dbSDimitry Andric   setOperationAction(ISD::LOAD, MVT::v4i64, Promote);
1225ffd83dbSDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v4i64, MVT::v8i32);
1235ffd83dbSDimitry Andric 
124fe6060f1SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v3f64, Promote);
125fe6060f1SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v3f64, MVT::v6i32);
126fe6060f1SDimitry Andric 
1275ffd83dbSDimitry Andric   setOperationAction(ISD::LOAD, MVT::v4f64, Promote);
1285ffd83dbSDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v4f64, MVT::v8i32);
1295ffd83dbSDimitry Andric 
1305ffd83dbSDimitry Andric   setOperationAction(ISD::LOAD, MVT::v8i64, Promote);
1315ffd83dbSDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v8i64, MVT::v16i32);
1325ffd83dbSDimitry Andric 
1335ffd83dbSDimitry Andric   setOperationAction(ISD::LOAD, MVT::v8f64, Promote);
1345ffd83dbSDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v8f64, MVT::v16i32);
1355ffd83dbSDimitry Andric 
1365ffd83dbSDimitry Andric   setOperationAction(ISD::LOAD, MVT::v16i64, Promote);
1375ffd83dbSDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v16i64, MVT::v32i32);
1385ffd83dbSDimitry Andric 
1395ffd83dbSDimitry Andric   setOperationAction(ISD::LOAD, MVT::v16f64, Promote);
1405ffd83dbSDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v16f64, MVT::v32i32);
1415ffd83dbSDimitry Andric 
14206c3fb27SDimitry Andric   setOperationAction(ISD::LOAD, MVT::i128, Promote);
14306c3fb27SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::i128, MVT::v4i32);
14406c3fb27SDimitry Andric 
1450b57cec5SDimitry Andric   // There are no 64-bit extloads. These should be done as a 32-bit extload and
1460b57cec5SDimitry Andric   // an extension to 64-bit.
14781ad6265SDimitry Andric   for (MVT VT : MVT::integer_valuetypes())
14881ad6265SDimitry Andric     setLoadExtAction({ISD::EXTLOAD, ISD::SEXTLOAD, ISD::ZEXTLOAD}, MVT::i64, VT,
14981ad6265SDimitry Andric                      Expand);
1500b57cec5SDimitry Andric 
1510b57cec5SDimitry Andric   for (MVT VT : MVT::integer_valuetypes()) {
1520b57cec5SDimitry Andric     if (VT == MVT::i64)
1530b57cec5SDimitry Andric       continue;
1540b57cec5SDimitry Andric 
15581ad6265SDimitry Andric     for (auto Op : {ISD::SEXTLOAD, ISD::ZEXTLOAD, ISD::EXTLOAD}) {
15681ad6265SDimitry Andric       setLoadExtAction(Op, VT, MVT::i1, Promote);
15781ad6265SDimitry Andric       setLoadExtAction(Op, VT, MVT::i8, Legal);
15881ad6265SDimitry Andric       setLoadExtAction(Op, VT, MVT::i16, Legal);
15981ad6265SDimitry Andric       setLoadExtAction(Op, VT, MVT::i32, Expand);
16081ad6265SDimitry Andric     }
1610b57cec5SDimitry Andric   }
1620b57cec5SDimitry Andric 
16381ad6265SDimitry Andric   for (MVT VT : MVT::integer_fixedlen_vector_valuetypes())
16481ad6265SDimitry Andric     for (auto MemVT :
16581ad6265SDimitry Andric          {MVT::v2i8, MVT::v4i8, MVT::v2i16, MVT::v3i16, MVT::v4i16})
16681ad6265SDimitry Andric       setLoadExtAction({ISD::SEXTLOAD, ISD::ZEXTLOAD, ISD::EXTLOAD}, VT, MemVT,
16781ad6265SDimitry Andric                        Expand);
1680b57cec5SDimitry Andric 
1690b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::f32, MVT::f16, Expand);
170bdd1243dSDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::f32, MVT::bf16, Expand);
1710b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v2f32, MVT::v2f16, Expand);
1728bcb0991SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v3f32, MVT::v3f16, Expand);
1730b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v4f32, MVT::v4f16, Expand);
1740b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v8f32, MVT::v8f16, Expand);
1758bcb0991SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v16f32, MVT::v16f16, Expand);
1768bcb0991SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v32f32, MVT::v32f16, Expand);
1770b57cec5SDimitry Andric 
1780b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::f64, MVT::f32, Expand);
1790b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v2f64, MVT::v2f32, Expand);
180fe6060f1SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v3f64, MVT::v3f32, Expand);
1810b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v4f64, MVT::v4f32, Expand);
1820b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v8f64, MVT::v8f32, Expand);
1835ffd83dbSDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v16f64, MVT::v16f32, Expand);
1840b57cec5SDimitry Andric 
1850b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::f64, MVT::f16, Expand);
186bdd1243dSDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::f64, MVT::bf16, Expand);
1870b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v2f64, MVT::v2f16, Expand);
188fe6060f1SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v3f64, MVT::v3f16, Expand);
1890b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v4f64, MVT::v4f16, Expand);
1900b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v8f64, MVT::v8f16, Expand);
1915ffd83dbSDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v16f64, MVT::v16f16, Expand);
1920b57cec5SDimitry Andric 
1930b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::f32, Promote);
1940b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::f32, MVT::i32);
1950b57cec5SDimitry Andric 
1960b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v2f32, Promote);
1970b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v2f32, MVT::v2i32);
1980b57cec5SDimitry Andric 
1990b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v3f32, Promote);
2000b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v3f32, MVT::v3i32);
2010b57cec5SDimitry Andric 
2020b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v4f32, Promote);
2030b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v4f32, MVT::v4i32);
2040b57cec5SDimitry Andric 
2050b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v5f32, Promote);
2060b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v5f32, MVT::v5i32);
2070b57cec5SDimitry Andric 
208fe6060f1SDimitry Andric   setOperationAction(ISD::STORE, MVT::v6f32, Promote);
209fe6060f1SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v6f32, MVT::v6i32);
210fe6060f1SDimitry Andric 
211fe6060f1SDimitry Andric   setOperationAction(ISD::STORE, MVT::v7f32, Promote);
212fe6060f1SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v7f32, MVT::v7i32);
213fe6060f1SDimitry Andric 
2140b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v8f32, Promote);
2150b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v8f32, MVT::v8i32);
2160b57cec5SDimitry Andric 
217bdd1243dSDimitry Andric   setOperationAction(ISD::STORE, MVT::v9f32, Promote);
218bdd1243dSDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v9f32, MVT::v9i32);
219bdd1243dSDimitry Andric 
220bdd1243dSDimitry Andric   setOperationAction(ISD::STORE, MVT::v10f32, Promote);
221bdd1243dSDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v10f32, MVT::v10i32);
222bdd1243dSDimitry Andric 
223bdd1243dSDimitry Andric   setOperationAction(ISD::STORE, MVT::v11f32, Promote);
224bdd1243dSDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v11f32, MVT::v11i32);
225bdd1243dSDimitry Andric 
226bdd1243dSDimitry Andric   setOperationAction(ISD::STORE, MVT::v12f32, Promote);
227bdd1243dSDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v12f32, MVT::v12i32);
228bdd1243dSDimitry Andric 
2290b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v16f32, Promote);
2300b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v16f32, MVT::v16i32);
2310b57cec5SDimitry Andric 
2320b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v32f32, Promote);
2330b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v32f32, MVT::v32i32);
2340b57cec5SDimitry Andric 
2350b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::i64, Promote);
2360b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::i64, MVT::v2i32);
2370b57cec5SDimitry Andric 
2380b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v2i64, Promote);
2390b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v2i64, MVT::v4i32);
2400b57cec5SDimitry Andric 
2410b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::f64, Promote);
2420b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::f64, MVT::v2i32);
2430b57cec5SDimitry Andric 
2440b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v2f64, Promote);
2450b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v2f64, MVT::v4i32);
2460b57cec5SDimitry Andric 
247fe6060f1SDimitry Andric   setOperationAction(ISD::STORE, MVT::v3i64, Promote);
248fe6060f1SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v3i64, MVT::v6i32);
249fe6060f1SDimitry Andric 
250fe6060f1SDimitry Andric   setOperationAction(ISD::STORE, MVT::v3f64, Promote);
251fe6060f1SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v3f64, MVT::v6i32);
252fe6060f1SDimitry Andric 
2535ffd83dbSDimitry Andric   setOperationAction(ISD::STORE, MVT::v4i64, Promote);
2545ffd83dbSDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v4i64, MVT::v8i32);
2555ffd83dbSDimitry Andric 
2565ffd83dbSDimitry Andric   setOperationAction(ISD::STORE, MVT::v4f64, Promote);
2575ffd83dbSDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v4f64, MVT::v8i32);
2585ffd83dbSDimitry Andric 
2595ffd83dbSDimitry Andric   setOperationAction(ISD::STORE, MVT::v8i64, Promote);
2605ffd83dbSDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v8i64, MVT::v16i32);
2615ffd83dbSDimitry Andric 
2625ffd83dbSDimitry Andric   setOperationAction(ISD::STORE, MVT::v8f64, Promote);
2635ffd83dbSDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v8f64, MVT::v16i32);
2645ffd83dbSDimitry Andric 
2655ffd83dbSDimitry Andric   setOperationAction(ISD::STORE, MVT::v16i64, Promote);
2665ffd83dbSDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v16i64, MVT::v32i32);
2675ffd83dbSDimitry Andric 
2685ffd83dbSDimitry Andric   setOperationAction(ISD::STORE, MVT::v16f64, Promote);
2695ffd83dbSDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v16f64, MVT::v32i32);
2705ffd83dbSDimitry Andric 
27106c3fb27SDimitry Andric   setOperationAction(ISD::STORE, MVT::i128, Promote);
27206c3fb27SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::i128, MVT::v4i32);
27306c3fb27SDimitry Andric 
2740b57cec5SDimitry Andric   setTruncStoreAction(MVT::i64, MVT::i1, Expand);
2750b57cec5SDimitry Andric   setTruncStoreAction(MVT::i64, MVT::i8, Expand);
2760b57cec5SDimitry Andric   setTruncStoreAction(MVT::i64, MVT::i16, Expand);
2770b57cec5SDimitry Andric   setTruncStoreAction(MVT::i64, MVT::i32, Expand);
2780b57cec5SDimitry Andric 
2790b57cec5SDimitry Andric   setTruncStoreAction(MVT::v2i64, MVT::v2i1, Expand);
2800b57cec5SDimitry Andric   setTruncStoreAction(MVT::v2i64, MVT::v2i8, Expand);
2810b57cec5SDimitry Andric   setTruncStoreAction(MVT::v2i64, MVT::v2i16, Expand);
2820b57cec5SDimitry Andric   setTruncStoreAction(MVT::v2i64, MVT::v2i32, Expand);
2830b57cec5SDimitry Andric 
284bdd1243dSDimitry Andric   setTruncStoreAction(MVT::f32, MVT::bf16, Expand);
2850b57cec5SDimitry Andric   setTruncStoreAction(MVT::f32, MVT::f16, Expand);
2860b57cec5SDimitry Andric   setTruncStoreAction(MVT::v2f32, MVT::v2f16, Expand);
2878bcb0991SDimitry Andric   setTruncStoreAction(MVT::v3f32, MVT::v3f16, Expand);
2880b57cec5SDimitry Andric   setTruncStoreAction(MVT::v4f32, MVT::v4f16, Expand);
2890b57cec5SDimitry Andric   setTruncStoreAction(MVT::v8f32, MVT::v8f16, Expand);
2908bcb0991SDimitry Andric   setTruncStoreAction(MVT::v16f32, MVT::v16f16, Expand);
2918bcb0991SDimitry Andric   setTruncStoreAction(MVT::v32f32, MVT::v32f16, Expand);
2920b57cec5SDimitry Andric 
293bdd1243dSDimitry Andric   setTruncStoreAction(MVT::f64, MVT::bf16, Expand);
2940b57cec5SDimitry Andric   setTruncStoreAction(MVT::f64, MVT::f16, Expand);
2950b57cec5SDimitry Andric   setTruncStoreAction(MVT::f64, MVT::f32, Expand);
2960b57cec5SDimitry Andric 
2970b57cec5SDimitry Andric   setTruncStoreAction(MVT::v2f64, MVT::v2f32, Expand);
2980b57cec5SDimitry Andric   setTruncStoreAction(MVT::v2f64, MVT::v2f16, Expand);
2990b57cec5SDimitry Andric 
300fe6060f1SDimitry Andric   setTruncStoreAction(MVT::v3i64, MVT::v3i32, Expand);
301fe6060f1SDimitry Andric   setTruncStoreAction(MVT::v3i64, MVT::v3i16, Expand);
302fe6060f1SDimitry Andric   setTruncStoreAction(MVT::v3f64, MVT::v3f32, Expand);
303fe6060f1SDimitry Andric   setTruncStoreAction(MVT::v3f64, MVT::v3f16, Expand);
304fe6060f1SDimitry Andric 
3055ffd83dbSDimitry Andric   setTruncStoreAction(MVT::v4i64, MVT::v4i32, Expand);
3065ffd83dbSDimitry Andric   setTruncStoreAction(MVT::v4i64, MVT::v4i16, Expand);
3070b57cec5SDimitry Andric   setTruncStoreAction(MVT::v4f64, MVT::v4f32, Expand);
3080b57cec5SDimitry Andric   setTruncStoreAction(MVT::v4f64, MVT::v4f16, Expand);
3090b57cec5SDimitry Andric 
3100b57cec5SDimitry Andric   setTruncStoreAction(MVT::v8f64, MVT::v8f32, Expand);
3110b57cec5SDimitry Andric   setTruncStoreAction(MVT::v8f64, MVT::v8f16, Expand);
3120b57cec5SDimitry Andric 
3135ffd83dbSDimitry Andric   setTruncStoreAction(MVT::v16f64, MVT::v16f32, Expand);
3145ffd83dbSDimitry Andric   setTruncStoreAction(MVT::v16f64, MVT::v16f16, Expand);
3155ffd83dbSDimitry Andric   setTruncStoreAction(MVT::v16i64, MVT::v16i16, Expand);
3165ffd83dbSDimitry Andric   setTruncStoreAction(MVT::v16i64, MVT::v16i16, Expand);
3175ffd83dbSDimitry Andric   setTruncStoreAction(MVT::v16i64, MVT::v16i8, Expand);
3185ffd83dbSDimitry Andric   setTruncStoreAction(MVT::v16i64, MVT::v16i8, Expand);
3195ffd83dbSDimitry Andric   setTruncStoreAction(MVT::v16i64, MVT::v16i1, Expand);
3200b57cec5SDimitry Andric 
32181ad6265SDimitry Andric   setOperationAction(ISD::Constant, {MVT::i32, MVT::i64}, Legal);
32281ad6265SDimitry Andric   setOperationAction(ISD::ConstantFP, {MVT::f32, MVT::f64}, Legal);
3230b57cec5SDimitry Andric 
32481ad6265SDimitry Andric   setOperationAction({ISD::BR_JT, ISD::BRIND}, MVT::Other, Expand);
3250b57cec5SDimitry Andric 
3260b57cec5SDimitry Andric   // This is totally unsupported, just custom lower to produce an error.
3270b57cec5SDimitry Andric   setOperationAction(ISD::DYNAMIC_STACKALLOC, MVT::i32, Custom);
3280b57cec5SDimitry Andric 
3290b57cec5SDimitry Andric   // Library functions.  These default to Expand, but we have instructions
3300b57cec5SDimitry Andric   // for them.
33106c3fb27SDimitry Andric   setOperationAction({ISD::FCEIL, ISD::FPOW, ISD::FABS, ISD::FFLOOR, ISD::FRINT,
33206c3fb27SDimitry Andric                       ISD::FTRUNC, ISD::FMINNUM, ISD::FMAXNUM},
33381ad6265SDimitry Andric                      MVT::f32, Legal);
3340b57cec5SDimitry Andric 
33506c3fb27SDimitry Andric   setOperationAction(ISD::FLOG2, MVT::f32, Custom);
33681ad6265SDimitry Andric   setOperationAction(ISD::FROUND, {MVT::f32, MVT::f64}, Custom);
3370b57cec5SDimitry Andric 
33806c3fb27SDimitry Andric   setOperationAction({ISD::FLOG, ISD::FLOG10, ISD::FEXP, ISD::FEXP2}, MVT::f32,
33906c3fb27SDimitry Andric                      Custom);
3400b57cec5SDimitry Andric 
341bdd1243dSDimitry Andric   setOperationAction(ISD::FNEARBYINT, {MVT::f16, MVT::f32, MVT::f64}, Custom);
342bdd1243dSDimitry Andric 
343bdd1243dSDimitry Andric   setOperationAction(ISD::FROUNDEVEN, {MVT::f16, MVT::f32, MVT::f64}, Custom);
3440b57cec5SDimitry Andric 
34581ad6265SDimitry Andric   setOperationAction(ISD::FREM, {MVT::f16, MVT::f32, MVT::f64}, Custom);
3460b57cec5SDimitry Andric 
347bdd1243dSDimitry Andric   if (Subtarget->has16BitInsts())
348bdd1243dSDimitry Andric     setOperationAction(ISD::IS_FPCLASS, {MVT::f16, MVT::f32, MVT::f64}, Legal);
34906c3fb27SDimitry Andric   else {
350bdd1243dSDimitry Andric     setOperationAction(ISD::IS_FPCLASS, {MVT::f32, MVT::f64}, Legal);
35106c3fb27SDimitry Andric     setOperationAction({ISD::FLOG2, ISD::FEXP2}, MVT::f16, Custom);
35206c3fb27SDimitry Andric   }
35306c3fb27SDimitry Andric 
35406c3fb27SDimitry Andric   setOperationAction({ISD::FLOG10, ISD::FLOG, ISD::FEXP}, MVT::f16, Custom);
355bdd1243dSDimitry Andric 
356bdd1243dSDimitry Andric   // FIXME: These IS_FPCLASS vector fp types are marked custom so it reaches
357bdd1243dSDimitry Andric   // scalarization code. Can be removed when IS_FPCLASS expand isn't called by
358bdd1243dSDimitry Andric   // default unless marked custom/legal.
359bdd1243dSDimitry Andric   setOperationAction(
360bdd1243dSDimitry Andric       ISD::IS_FPCLASS,
361bdd1243dSDimitry Andric       {MVT::v2f16, MVT::v3f16, MVT::v4f16, MVT::v16f16, MVT::v2f32, MVT::v3f32,
362bdd1243dSDimitry Andric        MVT::v4f32, MVT::v5f32, MVT::v6f32, MVT::v7f32, MVT::v8f32, MVT::v16f32,
363bdd1243dSDimitry Andric        MVT::v2f64, MVT::v3f64, MVT::v4f64, MVT::v8f64, MVT::v16f64},
364bdd1243dSDimitry Andric       Custom);
365bdd1243dSDimitry Andric 
3660b57cec5SDimitry Andric   // Expand to fneg + fadd.
3670b57cec5SDimitry Andric   setOperationAction(ISD::FSUB, MVT::f64, Expand);
3680b57cec5SDimitry Andric 
36981ad6265SDimitry Andric   setOperationAction(ISD::CONCAT_VECTORS,
37081ad6265SDimitry Andric                      {MVT::v3i32,  MVT::v3f32,  MVT::v4i32,  MVT::v4f32,
37181ad6265SDimitry Andric                       MVT::v5i32,  MVT::v5f32,  MVT::v6i32,  MVT::v6f32,
372bdd1243dSDimitry Andric                       MVT::v7i32,  MVT::v7f32,  MVT::v8i32,  MVT::v8f32,
373bdd1243dSDimitry Andric                       MVT::v9i32,  MVT::v9f32,  MVT::v10i32, MVT::v10f32,
374bdd1243dSDimitry Andric                       MVT::v11i32, MVT::v11f32, MVT::v12i32, MVT::v12f32},
37581ad6265SDimitry Andric                      Custom);
37681ad6265SDimitry Andric   setOperationAction(
37781ad6265SDimitry Andric       ISD::EXTRACT_SUBVECTOR,
37881ad6265SDimitry Andric       {MVT::v2f16,  MVT::v2i16,  MVT::v4f16,  MVT::v4i16,  MVT::v2f32,
37981ad6265SDimitry Andric        MVT::v2i32,  MVT::v3f32,  MVT::v3i32,  MVT::v4f32,  MVT::v4i32,
38081ad6265SDimitry Andric        MVT::v5f32,  MVT::v5i32,  MVT::v6f32,  MVT::v6i32,  MVT::v7f32,
381bdd1243dSDimitry Andric        MVT::v7i32,  MVT::v8f32,  MVT::v8i32,  MVT::v9f32,  MVT::v9i32,
382bdd1243dSDimitry Andric        MVT::v10i32, MVT::v10f32, MVT::v11i32, MVT::v11f32, MVT::v12i32,
383bdd1243dSDimitry Andric        MVT::v12f32, MVT::v16f16, MVT::v16i16, MVT::v16f32, MVT::v16i32,
384bdd1243dSDimitry Andric        MVT::v32f32, MVT::v32i32, MVT::v2f64,  MVT::v2i64,  MVT::v3f64,
385bdd1243dSDimitry Andric        MVT::v3i64,  MVT::v4f64,  MVT::v4i64,  MVT::v8f64,  MVT::v8i64,
386bdd1243dSDimitry Andric        MVT::v16f64, MVT::v16i64},
38781ad6265SDimitry Andric       Custom);
3880b57cec5SDimitry Andric 
3890b57cec5SDimitry Andric   setOperationAction(ISD::FP16_TO_FP, MVT::f64, Expand);
39081ad6265SDimitry Andric   setOperationAction(ISD::FP_TO_FP16, {MVT::f64, MVT::f32}, Custom);
3910b57cec5SDimitry Andric 
3920b57cec5SDimitry Andric   const MVT ScalarIntVTs[] = { MVT::i32, MVT::i64 };
3930b57cec5SDimitry Andric   for (MVT VT : ScalarIntVTs) {
3940b57cec5SDimitry Andric     // These should use [SU]DIVREM, so set them to expand
39581ad6265SDimitry Andric     setOperationAction({ISD::SDIV, ISD::UDIV, ISD::SREM, ISD::UREM}, VT,
39681ad6265SDimitry Andric                        Expand);
3970b57cec5SDimitry Andric 
3980b57cec5SDimitry Andric     // GPU does not have divrem function for signed or unsigned.
39981ad6265SDimitry Andric     setOperationAction({ISD::SDIVREM, ISD::UDIVREM}, VT, Custom);
4000b57cec5SDimitry Andric 
4010b57cec5SDimitry Andric     // GPU does not have [S|U]MUL_LOHI functions as a single instruction.
40281ad6265SDimitry Andric     setOperationAction({ISD::SMUL_LOHI, ISD::UMUL_LOHI}, VT, Expand);
4030b57cec5SDimitry Andric 
40481ad6265SDimitry Andric     setOperationAction({ISD::BSWAP, ISD::CTTZ, ISD::CTLZ}, VT, Expand);
4050b57cec5SDimitry Andric 
4060b57cec5SDimitry Andric     // AMDGPU uses ADDC/SUBC/ADDE/SUBE
40781ad6265SDimitry Andric     setOperationAction({ISD::ADDC, ISD::SUBC, ISD::ADDE, ISD::SUBE}, VT, Legal);
4080b57cec5SDimitry Andric   }
4090b57cec5SDimitry Andric 
4105ffd83dbSDimitry Andric   // The hardware supports 32-bit FSHR, but not FSHL.
4115ffd83dbSDimitry Andric   setOperationAction(ISD::FSHR, MVT::i32, Legal);
4125ffd83dbSDimitry Andric 
4130b57cec5SDimitry Andric   // The hardware supports 32-bit ROTR, but not ROTL.
41481ad6265SDimitry Andric   setOperationAction(ISD::ROTL, {MVT::i32, MVT::i64}, Expand);
4150b57cec5SDimitry Andric   setOperationAction(ISD::ROTR, MVT::i64, Expand);
4160b57cec5SDimitry Andric 
41781ad6265SDimitry Andric   setOperationAction({ISD::MULHU, ISD::MULHS}, MVT::i16, Expand);
418e8d8bef9SDimitry Andric 
41981ad6265SDimitry Andric   setOperationAction({ISD::MUL, ISD::MULHU, ISD::MULHS}, MVT::i64, Expand);
42081ad6265SDimitry Andric   setOperationAction(
42181ad6265SDimitry Andric       {ISD::UINT_TO_FP, ISD::SINT_TO_FP, ISD::FP_TO_SINT, ISD::FP_TO_UINT},
42281ad6265SDimitry Andric       MVT::i64, Custom);
4230b57cec5SDimitry Andric   setOperationAction(ISD::SELECT_CC, MVT::i64, Expand);
4240b57cec5SDimitry Andric 
42581ad6265SDimitry Andric   setOperationAction({ISD::SMIN, ISD::UMIN, ISD::SMAX, ISD::UMAX}, MVT::i32,
42681ad6265SDimitry Andric                      Legal);
4270b57cec5SDimitry Andric 
42881ad6265SDimitry Andric   setOperationAction(
42981ad6265SDimitry Andric       {ISD::CTTZ, ISD::CTTZ_ZERO_UNDEF, ISD::CTLZ, ISD::CTLZ_ZERO_UNDEF},
43081ad6265SDimitry Andric       MVT::i64, Custom);
4310b57cec5SDimitry Andric 
4320b57cec5SDimitry Andric   static const MVT::SimpleValueType VectorIntTypes[] = {
433bdd1243dSDimitry Andric       MVT::v2i32, MVT::v3i32, MVT::v4i32, MVT::v5i32, MVT::v6i32, MVT::v7i32,
434bdd1243dSDimitry Andric       MVT::v9i32, MVT::v10i32, MVT::v11i32, MVT::v12i32};
4350b57cec5SDimitry Andric 
4360b57cec5SDimitry Andric   for (MVT VT : VectorIntTypes) {
4370b57cec5SDimitry Andric     // Expand the following operations for the current type by default.
43881ad6265SDimitry Andric     setOperationAction({ISD::ADD,        ISD::AND,     ISD::FP_TO_SINT,
43981ad6265SDimitry Andric                         ISD::FP_TO_UINT, ISD::MUL,     ISD::MULHU,
44081ad6265SDimitry Andric                         ISD::MULHS,      ISD::OR,      ISD::SHL,
44181ad6265SDimitry Andric                         ISD::SRA,        ISD::SRL,     ISD::ROTL,
44281ad6265SDimitry Andric                         ISD::ROTR,       ISD::SUB,     ISD::SINT_TO_FP,
44381ad6265SDimitry Andric                         ISD::UINT_TO_FP, ISD::SDIV,    ISD::UDIV,
44481ad6265SDimitry Andric                         ISD::SREM,       ISD::UREM,    ISD::SMUL_LOHI,
44581ad6265SDimitry Andric                         ISD::UMUL_LOHI,  ISD::SDIVREM, ISD::UDIVREM,
44681ad6265SDimitry Andric                         ISD::SELECT,     ISD::VSELECT, ISD::SELECT_CC,
44781ad6265SDimitry Andric                         ISD::XOR,        ISD::BSWAP,   ISD::CTPOP,
44881ad6265SDimitry Andric                         ISD::CTTZ,       ISD::CTLZ,    ISD::VECTOR_SHUFFLE,
44981ad6265SDimitry Andric                         ISD::SETCC},
45081ad6265SDimitry Andric                        VT, Expand);
4510b57cec5SDimitry Andric   }
4520b57cec5SDimitry Andric 
4530b57cec5SDimitry Andric   static const MVT::SimpleValueType FloatVectorTypes[] = {
454bdd1243dSDimitry Andric       MVT::v2f32, MVT::v3f32,  MVT::v4f32, MVT::v5f32, MVT::v6f32, MVT::v7f32,
455bdd1243dSDimitry Andric       MVT::v9f32, MVT::v10f32, MVT::v11f32, MVT::v12f32};
4560b57cec5SDimitry Andric 
4570b57cec5SDimitry Andric   for (MVT VT : FloatVectorTypes) {
45881ad6265SDimitry Andric     setOperationAction(
45981ad6265SDimitry Andric         {ISD::FABS,    ISD::FMINNUM,      ISD::FMAXNUM,   ISD::FADD,
46081ad6265SDimitry Andric          ISD::FCEIL,   ISD::FCOS,         ISD::FDIV,      ISD::FEXP2,
46181ad6265SDimitry Andric          ISD::FEXP,    ISD::FLOG2,        ISD::FREM,      ISD::FLOG,
46281ad6265SDimitry Andric          ISD::FLOG10,  ISD::FPOW,         ISD::FFLOOR,    ISD::FTRUNC,
46381ad6265SDimitry Andric          ISD::FMUL,    ISD::FMA,          ISD::FRINT,     ISD::FNEARBYINT,
46481ad6265SDimitry Andric          ISD::FSQRT,   ISD::FSIN,         ISD::FSUB,      ISD::FNEG,
46581ad6265SDimitry Andric          ISD::VSELECT, ISD::SELECT_CC,    ISD::FCOPYSIGN, ISD::VECTOR_SHUFFLE,
46681ad6265SDimitry Andric          ISD::SETCC,   ISD::FCANONICALIZE},
46781ad6265SDimitry Andric         VT, Expand);
4680b57cec5SDimitry Andric   }
4690b57cec5SDimitry Andric 
4700b57cec5SDimitry Andric   // This causes using an unrolled select operation rather than expansion with
4710b57cec5SDimitry Andric   // bit operations. This is in general better, but the alternative using BFI
4720b57cec5SDimitry Andric   // instructions may be better if the select sources are SGPRs.
4730b57cec5SDimitry Andric   setOperationAction(ISD::SELECT, MVT::v2f32, Promote);
4740b57cec5SDimitry Andric   AddPromotedToType(ISD::SELECT, MVT::v2f32, MVT::v2i32);
4750b57cec5SDimitry Andric 
4760b57cec5SDimitry Andric   setOperationAction(ISD::SELECT, MVT::v3f32, Promote);
4770b57cec5SDimitry Andric   AddPromotedToType(ISD::SELECT, MVT::v3f32, MVT::v3i32);
4780b57cec5SDimitry Andric 
4790b57cec5SDimitry Andric   setOperationAction(ISD::SELECT, MVT::v4f32, Promote);
4800b57cec5SDimitry Andric   AddPromotedToType(ISD::SELECT, MVT::v4f32, MVT::v4i32);
4810b57cec5SDimitry Andric 
4820b57cec5SDimitry Andric   setOperationAction(ISD::SELECT, MVT::v5f32, Promote);
4830b57cec5SDimitry Andric   AddPromotedToType(ISD::SELECT, MVT::v5f32, MVT::v5i32);
4840b57cec5SDimitry Andric 
485fe6060f1SDimitry Andric   setOperationAction(ISD::SELECT, MVT::v6f32, Promote);
486fe6060f1SDimitry Andric   AddPromotedToType(ISD::SELECT, MVT::v6f32, MVT::v6i32);
487fe6060f1SDimitry Andric 
488fe6060f1SDimitry Andric   setOperationAction(ISD::SELECT, MVT::v7f32, Promote);
489fe6060f1SDimitry Andric   AddPromotedToType(ISD::SELECT, MVT::v7f32, MVT::v7i32);
490fe6060f1SDimitry Andric 
491bdd1243dSDimitry Andric   setOperationAction(ISD::SELECT, MVT::v9f32, Promote);
492bdd1243dSDimitry Andric   AddPromotedToType(ISD::SELECT, MVT::v9f32, MVT::v9i32);
493bdd1243dSDimitry Andric 
494bdd1243dSDimitry Andric   setOperationAction(ISD::SELECT, MVT::v10f32, Promote);
495bdd1243dSDimitry Andric   AddPromotedToType(ISD::SELECT, MVT::v10f32, MVT::v10i32);
496bdd1243dSDimitry Andric 
497bdd1243dSDimitry Andric   setOperationAction(ISD::SELECT, MVT::v11f32, Promote);
498bdd1243dSDimitry Andric   AddPromotedToType(ISD::SELECT, MVT::v11f32, MVT::v11i32);
499bdd1243dSDimitry Andric 
500bdd1243dSDimitry Andric   setOperationAction(ISD::SELECT, MVT::v12f32, Promote);
501bdd1243dSDimitry Andric   AddPromotedToType(ISD::SELECT, MVT::v12f32, MVT::v12i32);
502bdd1243dSDimitry Andric 
5030b57cec5SDimitry Andric   // There are no libcalls of any kind.
5040b57cec5SDimitry Andric   for (int I = 0; I < RTLIB::UNKNOWN_LIBCALL; ++I)
5050b57cec5SDimitry Andric     setLibcallName(static_cast<RTLIB::Libcall>(I), nullptr);
5060b57cec5SDimitry Andric 
5070b57cec5SDimitry Andric   setSchedulingPreference(Sched::RegPressure);
5080b57cec5SDimitry Andric   setJumpIsExpensive(true);
5090b57cec5SDimitry Andric 
5100b57cec5SDimitry Andric   // FIXME: This is only partially true. If we have to do vector compares, any
5110b57cec5SDimitry Andric   // SGPR pair can be a condition register. If we have a uniform condition, we
5120b57cec5SDimitry Andric   // are better off doing SALU operations, where there is only one SCC. For now,
5130b57cec5SDimitry Andric   // we don't have a way of knowing during instruction selection if a condition
5140b57cec5SDimitry Andric   // will be uniform and we always use vector compares. Assume we are using
5150b57cec5SDimitry Andric   // vector compares until that is fixed.
5160b57cec5SDimitry Andric   setHasMultipleConditionRegisters(true);
5170b57cec5SDimitry Andric 
5180b57cec5SDimitry Andric   setMinCmpXchgSizeInBits(32);
5190b57cec5SDimitry Andric   setSupportsUnalignedAtomics(false);
5200b57cec5SDimitry Andric 
5210b57cec5SDimitry Andric   PredictableSelectIsExpensive = false;
5220b57cec5SDimitry Andric 
5230b57cec5SDimitry Andric   // We want to find all load dependencies for long chains of stores to enable
5240b57cec5SDimitry Andric   // merging into very wide vectors. The problem is with vectors with > 4
5250b57cec5SDimitry Andric   // elements. MergeConsecutiveStores will attempt to merge these because x8/x16
5260b57cec5SDimitry Andric   // vectors are a legal type, even though we have to split the loads
5270b57cec5SDimitry Andric   // usually. When we can more precisely specify load legality per address
5280b57cec5SDimitry Andric   // space, we should be able to make FindBetterChain/MergeConsecutiveStores
5290b57cec5SDimitry Andric   // smarter so that they can figure out what to do in 2 iterations without all
5300b57cec5SDimitry Andric   // N > 4 stores on the same chain.
5310b57cec5SDimitry Andric   GatherAllAliasesMaxDepth = 16;
5320b57cec5SDimitry Andric 
5330b57cec5SDimitry Andric   // memcpy/memmove/memset are expanded in the IR, so we shouldn't need to worry
5340b57cec5SDimitry Andric   // about these during lowering.
5350b57cec5SDimitry Andric   MaxStoresPerMemcpy  = 0xffffffff;
5360b57cec5SDimitry Andric   MaxStoresPerMemmove = 0xffffffff;
5370b57cec5SDimitry Andric   MaxStoresPerMemset  = 0xffffffff;
5380b57cec5SDimitry Andric 
5395ffd83dbSDimitry Andric   // The expansion for 64-bit division is enormous.
5405ffd83dbSDimitry Andric   if (AMDGPUBypassSlowDiv)
5415ffd83dbSDimitry Andric     addBypassSlowDiv(64, 32);
5425ffd83dbSDimitry Andric 
54381ad6265SDimitry Andric   setTargetDAGCombine({ISD::BITCAST,    ISD::SHL,
54481ad6265SDimitry Andric                        ISD::SRA,        ISD::SRL,
54581ad6265SDimitry Andric                        ISD::TRUNCATE,   ISD::MUL,
54681ad6265SDimitry Andric                        ISD::SMUL_LOHI,  ISD::UMUL_LOHI,
54781ad6265SDimitry Andric                        ISD::MULHU,      ISD::MULHS,
54881ad6265SDimitry Andric                        ISD::SELECT,     ISD::SELECT_CC,
54981ad6265SDimitry Andric                        ISD::STORE,      ISD::FADD,
55081ad6265SDimitry Andric                        ISD::FSUB,       ISD::FNEG,
55181ad6265SDimitry Andric                        ISD::FABS,       ISD::AssertZext,
55281ad6265SDimitry Andric                        ISD::AssertSext, ISD::INTRINSIC_WO_CHAIN});
5530b57cec5SDimitry Andric }
5540b57cec5SDimitry Andric 
555e8d8bef9SDimitry Andric bool AMDGPUTargetLowering::mayIgnoreSignedZero(SDValue Op) const {
556e8d8bef9SDimitry Andric   if (getTargetMachine().Options.NoSignedZerosFPMath)
557e8d8bef9SDimitry Andric     return true;
558e8d8bef9SDimitry Andric 
559e8d8bef9SDimitry Andric   const auto Flags = Op.getNode()->getFlags();
560e8d8bef9SDimitry Andric   if (Flags.hasNoSignedZeros())
561e8d8bef9SDimitry Andric     return true;
562e8d8bef9SDimitry Andric 
563e8d8bef9SDimitry Andric   return false;
564e8d8bef9SDimitry Andric }
565e8d8bef9SDimitry Andric 
5660b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
5670b57cec5SDimitry Andric // Target Information
5680b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
5690b57cec5SDimitry Andric 
5700b57cec5SDimitry Andric LLVM_READNONE
57106c3fb27SDimitry Andric static bool fnegFoldsIntoOpcode(unsigned Opc) {
5720b57cec5SDimitry Andric   switch (Opc) {
5730b57cec5SDimitry Andric   case ISD::FADD:
5740b57cec5SDimitry Andric   case ISD::FSUB:
5750b57cec5SDimitry Andric   case ISD::FMUL:
5760b57cec5SDimitry Andric   case ISD::FMA:
5770b57cec5SDimitry Andric   case ISD::FMAD:
5780b57cec5SDimitry Andric   case ISD::FMINNUM:
5790b57cec5SDimitry Andric   case ISD::FMAXNUM:
5800b57cec5SDimitry Andric   case ISD::FMINNUM_IEEE:
5810b57cec5SDimitry Andric   case ISD::FMAXNUM_IEEE:
58206c3fb27SDimitry Andric   case ISD::SELECT:
5830b57cec5SDimitry Andric   case ISD::FSIN:
5840b57cec5SDimitry Andric   case ISD::FTRUNC:
5850b57cec5SDimitry Andric   case ISD::FRINT:
5860b57cec5SDimitry Andric   case ISD::FNEARBYINT:
5870b57cec5SDimitry Andric   case ISD::FCANONICALIZE:
5880b57cec5SDimitry Andric   case AMDGPUISD::RCP:
5890b57cec5SDimitry Andric   case AMDGPUISD::RCP_LEGACY:
5900b57cec5SDimitry Andric   case AMDGPUISD::RCP_IFLAG:
5910b57cec5SDimitry Andric   case AMDGPUISD::SIN_HW:
5920b57cec5SDimitry Andric   case AMDGPUISD::FMUL_LEGACY:
5930b57cec5SDimitry Andric   case AMDGPUISD::FMIN_LEGACY:
5940b57cec5SDimitry Andric   case AMDGPUISD::FMAX_LEGACY:
5950b57cec5SDimitry Andric   case AMDGPUISD::FMED3:
596e8d8bef9SDimitry Andric     // TODO: handle llvm.amdgcn.fma.legacy
5970b57cec5SDimitry Andric     return true;
59806c3fb27SDimitry Andric   case ISD::BITCAST:
59906c3fb27SDimitry Andric     llvm_unreachable("bitcast is special cased");
6000b57cec5SDimitry Andric   default:
6010b57cec5SDimitry Andric     return false;
6020b57cec5SDimitry Andric   }
6030b57cec5SDimitry Andric }
6040b57cec5SDimitry Andric 
60506c3fb27SDimitry Andric static bool fnegFoldsIntoOp(const SDNode *N) {
60606c3fb27SDimitry Andric   unsigned Opc = N->getOpcode();
60706c3fb27SDimitry Andric   if (Opc == ISD::BITCAST) {
60806c3fb27SDimitry Andric     // TODO: Is there a benefit to checking the conditions performFNegCombine
60906c3fb27SDimitry Andric     // does? We don't for the other cases.
61006c3fb27SDimitry Andric     SDValue BCSrc = N->getOperand(0);
61106c3fb27SDimitry Andric     if (BCSrc.getOpcode() == ISD::BUILD_VECTOR) {
61206c3fb27SDimitry Andric       return BCSrc.getNumOperands() == 2 &&
61306c3fb27SDimitry Andric              BCSrc.getOperand(1).getValueSizeInBits() == 32;
61406c3fb27SDimitry Andric     }
61506c3fb27SDimitry Andric 
61606c3fb27SDimitry Andric     return BCSrc.getOpcode() == ISD::SELECT && BCSrc.getValueType() == MVT::f32;
61706c3fb27SDimitry Andric   }
61806c3fb27SDimitry Andric 
61906c3fb27SDimitry Andric   return fnegFoldsIntoOpcode(Opc);
62006c3fb27SDimitry Andric }
62106c3fb27SDimitry Andric 
6220b57cec5SDimitry Andric /// \p returns true if the operation will definitely need to use a 64-bit
6230b57cec5SDimitry Andric /// encoding, and thus will use a VOP3 encoding regardless of the source
6240b57cec5SDimitry Andric /// modifiers.
6250b57cec5SDimitry Andric LLVM_READONLY
6260b57cec5SDimitry Andric static bool opMustUseVOP3Encoding(const SDNode *N, MVT VT) {
62706c3fb27SDimitry Andric   return (N->getNumOperands() > 2 && N->getOpcode() != ISD::SELECT) ||
62806c3fb27SDimitry Andric          VT == MVT::f64;
62906c3fb27SDimitry Andric }
63006c3fb27SDimitry Andric 
63106c3fb27SDimitry Andric /// Return true if v_cndmask_b32 will support fabs/fneg source modifiers for the
63206c3fb27SDimitry Andric /// type for ISD::SELECT.
63306c3fb27SDimitry Andric LLVM_READONLY
63406c3fb27SDimitry Andric static bool selectSupportsSourceMods(const SDNode *N) {
63506c3fb27SDimitry Andric   // TODO: Only applies if select will be vector
63606c3fb27SDimitry Andric   return N->getValueType(0) == MVT::f32;
6370b57cec5SDimitry Andric }
6380b57cec5SDimitry Andric 
6390b57cec5SDimitry Andric // Most FP instructions support source modifiers, but this could be refined
6400b57cec5SDimitry Andric // slightly.
6410b57cec5SDimitry Andric LLVM_READONLY
6420b57cec5SDimitry Andric static bool hasSourceMods(const SDNode *N) {
6430b57cec5SDimitry Andric   if (isa<MemSDNode>(N))
6440b57cec5SDimitry Andric     return false;
6450b57cec5SDimitry Andric 
6460b57cec5SDimitry Andric   switch (N->getOpcode()) {
6470b57cec5SDimitry Andric   case ISD::CopyToReg:
6480b57cec5SDimitry Andric   case ISD::FDIV:
6490b57cec5SDimitry Andric   case ISD::FREM:
6500b57cec5SDimitry Andric   case ISD::INLINEASM:
6510b57cec5SDimitry Andric   case ISD::INLINEASM_BR:
6520b57cec5SDimitry Andric   case AMDGPUISD::DIV_SCALE:
6538bcb0991SDimitry Andric   case ISD::INTRINSIC_W_CHAIN:
6540b57cec5SDimitry Andric 
6550b57cec5SDimitry Andric   // TODO: Should really be looking at the users of the bitcast. These are
6560b57cec5SDimitry Andric   // problematic because bitcasts are used to legalize all stores to integer
6570b57cec5SDimitry Andric   // types.
6580b57cec5SDimitry Andric   case ISD::BITCAST:
6590b57cec5SDimitry Andric     return false;
6608bcb0991SDimitry Andric   case ISD::INTRINSIC_WO_CHAIN: {
6618bcb0991SDimitry Andric     switch (cast<ConstantSDNode>(N->getOperand(0))->getZExtValue()) {
6628bcb0991SDimitry Andric     case Intrinsic::amdgcn_interp_p1:
6638bcb0991SDimitry Andric     case Intrinsic::amdgcn_interp_p2:
6648bcb0991SDimitry Andric     case Intrinsic::amdgcn_interp_mov:
6658bcb0991SDimitry Andric     case Intrinsic::amdgcn_interp_p1_f16:
6668bcb0991SDimitry Andric     case Intrinsic::amdgcn_interp_p2_f16:
6678bcb0991SDimitry Andric       return false;
6688bcb0991SDimitry Andric     default:
6698bcb0991SDimitry Andric       return true;
6708bcb0991SDimitry Andric     }
6718bcb0991SDimitry Andric   }
67206c3fb27SDimitry Andric   case ISD::SELECT:
67306c3fb27SDimitry Andric     return selectSupportsSourceMods(N);
6740b57cec5SDimitry Andric   default:
6750b57cec5SDimitry Andric     return true;
6760b57cec5SDimitry Andric   }
6770b57cec5SDimitry Andric }
6780b57cec5SDimitry Andric 
6790b57cec5SDimitry Andric bool AMDGPUTargetLowering::allUsesHaveSourceMods(const SDNode *N,
6800b57cec5SDimitry Andric                                                  unsigned CostThreshold) {
6810b57cec5SDimitry Andric   // Some users (such as 3-operand FMA/MAD) must use a VOP3 encoding, and thus
6820b57cec5SDimitry Andric   // it is truly free to use a source modifier in all cases. If there are
6830b57cec5SDimitry Andric   // multiple users but for each one will necessitate using VOP3, there will be
6840b57cec5SDimitry Andric   // a code size increase. Try to avoid increasing code size unless we know it
6850b57cec5SDimitry Andric   // will save on the instruction count.
6860b57cec5SDimitry Andric   unsigned NumMayIncreaseSize = 0;
6870b57cec5SDimitry Andric   MVT VT = N->getValueType(0).getScalarType().getSimpleVT();
6880b57cec5SDimitry Andric 
68906c3fb27SDimitry Andric   assert(!N->use_empty());
69006c3fb27SDimitry Andric 
6910b57cec5SDimitry Andric   // XXX - Should this limit number of uses to check?
6920b57cec5SDimitry Andric   for (const SDNode *U : N->uses()) {
6930b57cec5SDimitry Andric     if (!hasSourceMods(U))
6940b57cec5SDimitry Andric       return false;
6950b57cec5SDimitry Andric 
6960b57cec5SDimitry Andric     if (!opMustUseVOP3Encoding(U, VT)) {
6970b57cec5SDimitry Andric       if (++NumMayIncreaseSize > CostThreshold)
6980b57cec5SDimitry Andric         return false;
6990b57cec5SDimitry Andric     }
7000b57cec5SDimitry Andric   }
7010b57cec5SDimitry Andric 
7020b57cec5SDimitry Andric   return true;
7030b57cec5SDimitry Andric }
7040b57cec5SDimitry Andric 
7055ffd83dbSDimitry Andric EVT AMDGPUTargetLowering::getTypeForExtReturn(LLVMContext &Context, EVT VT,
7065ffd83dbSDimitry Andric                                               ISD::NodeType ExtendKind) const {
7075ffd83dbSDimitry Andric   assert(!VT.isVector() && "only scalar expected");
7085ffd83dbSDimitry Andric 
7095ffd83dbSDimitry Andric   // Round to the next multiple of 32-bits.
7105ffd83dbSDimitry Andric   unsigned Size = VT.getSizeInBits();
7115ffd83dbSDimitry Andric   if (Size <= 32)
7125ffd83dbSDimitry Andric     return MVT::i32;
7135ffd83dbSDimitry Andric   return EVT::getIntegerVT(Context, 32 * ((Size + 31) / 32));
7145ffd83dbSDimitry Andric }
7155ffd83dbSDimitry Andric 
7160b57cec5SDimitry Andric MVT AMDGPUTargetLowering::getVectorIdxTy(const DataLayout &) const {
7170b57cec5SDimitry Andric   return MVT::i32;
7180b57cec5SDimitry Andric }
7190b57cec5SDimitry Andric 
7200b57cec5SDimitry Andric bool AMDGPUTargetLowering::isSelectSupported(SelectSupportKind SelType) const {
7210b57cec5SDimitry Andric   return true;
7220b57cec5SDimitry Andric }
7230b57cec5SDimitry Andric 
7240b57cec5SDimitry Andric // The backend supports 32 and 64 bit floating point immediates.
7250b57cec5SDimitry Andric // FIXME: Why are we reporting vectors of FP immediates as legal?
7260b57cec5SDimitry Andric bool AMDGPUTargetLowering::isFPImmLegal(const APFloat &Imm, EVT VT,
7270b57cec5SDimitry Andric                                         bool ForCodeSize) const {
7280b57cec5SDimitry Andric   EVT ScalarVT = VT.getScalarType();
7290b57cec5SDimitry Andric   return (ScalarVT == MVT::f32 || ScalarVT == MVT::f64 ||
7300b57cec5SDimitry Andric          (ScalarVT == MVT::f16 && Subtarget->has16BitInsts()));
7310b57cec5SDimitry Andric }
7320b57cec5SDimitry Andric 
7330b57cec5SDimitry Andric // We don't want to shrink f64 / f32 constants.
7340b57cec5SDimitry Andric bool AMDGPUTargetLowering::ShouldShrinkFPConstant(EVT VT) const {
7350b57cec5SDimitry Andric   EVT ScalarVT = VT.getScalarType();
7360b57cec5SDimitry Andric   return (ScalarVT != MVT::f32 && ScalarVT != MVT::f64);
7370b57cec5SDimitry Andric }
7380b57cec5SDimitry Andric 
7390b57cec5SDimitry Andric bool AMDGPUTargetLowering::shouldReduceLoadWidth(SDNode *N,
7400b57cec5SDimitry Andric                                                  ISD::LoadExtType ExtTy,
7410b57cec5SDimitry Andric                                                  EVT NewVT) const {
7420b57cec5SDimitry Andric   // TODO: This may be worth removing. Check regression tests for diffs.
7430b57cec5SDimitry Andric   if (!TargetLoweringBase::shouldReduceLoadWidth(N, ExtTy, NewVT))
7440b57cec5SDimitry Andric     return false;
7450b57cec5SDimitry Andric 
7460b57cec5SDimitry Andric   unsigned NewSize = NewVT.getStoreSizeInBits();
7470b57cec5SDimitry Andric 
7485ffd83dbSDimitry Andric   // If we are reducing to a 32-bit load or a smaller multi-dword load,
7495ffd83dbSDimitry Andric   // this is always better.
7505ffd83dbSDimitry Andric   if (NewSize >= 32)
7510b57cec5SDimitry Andric     return true;
7520b57cec5SDimitry Andric 
7530b57cec5SDimitry Andric   EVT OldVT = N->getValueType(0);
7540b57cec5SDimitry Andric   unsigned OldSize = OldVT.getStoreSizeInBits();
7550b57cec5SDimitry Andric 
7560b57cec5SDimitry Andric   MemSDNode *MN = cast<MemSDNode>(N);
7570b57cec5SDimitry Andric   unsigned AS = MN->getAddressSpace();
7580b57cec5SDimitry Andric   // Do not shrink an aligned scalar load to sub-dword.
7590b57cec5SDimitry Andric   // Scalar engine cannot do sub-dword loads.
76081ad6265SDimitry Andric   if (OldSize >= 32 && NewSize < 32 && MN->getAlign() >= Align(4) &&
7610b57cec5SDimitry Andric       (AS == AMDGPUAS::CONSTANT_ADDRESS ||
7620b57cec5SDimitry Andric        AS == AMDGPUAS::CONSTANT_ADDRESS_32BIT ||
76381ad6265SDimitry Andric        (isa<LoadSDNode>(N) && AS == AMDGPUAS::GLOBAL_ADDRESS &&
76481ad6265SDimitry Andric         MN->isInvariant())) &&
7650b57cec5SDimitry Andric       AMDGPUInstrInfo::isUniformMMO(MN->getMemOperand()))
7660b57cec5SDimitry Andric     return false;
7670b57cec5SDimitry Andric 
7680b57cec5SDimitry Andric   // Don't produce extloads from sub 32-bit types. SI doesn't have scalar
7690b57cec5SDimitry Andric   // extloads, so doing one requires using a buffer_load. In cases where we
7700b57cec5SDimitry Andric   // still couldn't use a scalar load, using the wider load shouldn't really
7710b57cec5SDimitry Andric   // hurt anything.
7720b57cec5SDimitry Andric 
7730b57cec5SDimitry Andric   // If the old size already had to be an extload, there's no harm in continuing
7740b57cec5SDimitry Andric   // to reduce the width.
7750b57cec5SDimitry Andric   return (OldSize < 32);
7760b57cec5SDimitry Andric }
7770b57cec5SDimitry Andric 
7780b57cec5SDimitry Andric bool AMDGPUTargetLowering::isLoadBitCastBeneficial(EVT LoadTy, EVT CastTy,
7790b57cec5SDimitry Andric                                                    const SelectionDAG &DAG,
7800b57cec5SDimitry Andric                                                    const MachineMemOperand &MMO) const {
7810b57cec5SDimitry Andric 
7820b57cec5SDimitry Andric   assert(LoadTy.getSizeInBits() == CastTy.getSizeInBits());
7830b57cec5SDimitry Andric 
7840b57cec5SDimitry Andric   if (LoadTy.getScalarType() == MVT::i32)
7850b57cec5SDimitry Andric     return false;
7860b57cec5SDimitry Andric 
7870b57cec5SDimitry Andric   unsigned LScalarSize = LoadTy.getScalarSizeInBits();
7880b57cec5SDimitry Andric   unsigned CastScalarSize = CastTy.getScalarSizeInBits();
7890b57cec5SDimitry Andric 
7900b57cec5SDimitry Andric   if ((LScalarSize >= CastScalarSize) && (CastScalarSize < 32))
7910b57cec5SDimitry Andric     return false;
7920b57cec5SDimitry Andric 
793bdd1243dSDimitry Andric   unsigned Fast = 0;
7948bcb0991SDimitry Andric   return allowsMemoryAccessForAlignment(*DAG.getContext(), DAG.getDataLayout(),
7958bcb0991SDimitry Andric                                         CastTy, MMO, &Fast) &&
7968bcb0991SDimitry Andric          Fast;
7970b57cec5SDimitry Andric }
7980b57cec5SDimitry Andric 
7990b57cec5SDimitry Andric // SI+ has instructions for cttz / ctlz for 32-bit values. This is probably also
8000b57cec5SDimitry Andric // profitable with the expansion for 64-bit since it's generally good to
8010b57cec5SDimitry Andric // speculate things.
802bdd1243dSDimitry Andric bool AMDGPUTargetLowering::isCheapToSpeculateCttz(Type *Ty) const {
8030b57cec5SDimitry Andric   return true;
8040b57cec5SDimitry Andric }
8050b57cec5SDimitry Andric 
806bdd1243dSDimitry Andric bool AMDGPUTargetLowering::isCheapToSpeculateCtlz(Type *Ty) const {
8070b57cec5SDimitry Andric   return true;
8080b57cec5SDimitry Andric }
8090b57cec5SDimitry Andric 
8100b57cec5SDimitry Andric bool AMDGPUTargetLowering::isSDNodeAlwaysUniform(const SDNode *N) const {
8110b57cec5SDimitry Andric   switch (N->getOpcode()) {
8120b57cec5SDimitry Andric   case ISD::EntryToken:
8130b57cec5SDimitry Andric   case ISD::TokenFactor:
8140b57cec5SDimitry Andric     return true;
815e8d8bef9SDimitry Andric   case ISD::INTRINSIC_WO_CHAIN: {
8160b57cec5SDimitry Andric     unsigned IntrID = cast<ConstantSDNode>(N->getOperand(0))->getZExtValue();
8170b57cec5SDimitry Andric     switch (IntrID) {
8180b57cec5SDimitry Andric     case Intrinsic::amdgcn_readfirstlane:
8190b57cec5SDimitry Andric     case Intrinsic::amdgcn_readlane:
8200b57cec5SDimitry Andric       return true;
8210b57cec5SDimitry Andric     }
822e8d8bef9SDimitry Andric     return false;
8230b57cec5SDimitry Andric   }
8240b57cec5SDimitry Andric   case ISD::LOAD:
8258bcb0991SDimitry Andric     if (cast<LoadSDNode>(N)->getMemOperand()->getAddrSpace() ==
8268bcb0991SDimitry Andric         AMDGPUAS::CONSTANT_ADDRESS_32BIT)
8270b57cec5SDimitry Andric       return true;
8280b57cec5SDimitry Andric     return false;
82981ad6265SDimitry Andric   case AMDGPUISD::SETCC: // ballot-style instruction
83081ad6265SDimitry Andric     return true;
8310b57cec5SDimitry Andric   }
832e8d8bef9SDimitry Andric   return false;
8330b57cec5SDimitry Andric }
8340b57cec5SDimitry Andric 
8355ffd83dbSDimitry Andric SDValue AMDGPUTargetLowering::getNegatedExpression(
8365ffd83dbSDimitry Andric     SDValue Op, SelectionDAG &DAG, bool LegalOperations, bool ForCodeSize,
8375ffd83dbSDimitry Andric     NegatibleCost &Cost, unsigned Depth) const {
8385ffd83dbSDimitry Andric 
8395ffd83dbSDimitry Andric   switch (Op.getOpcode()) {
8405ffd83dbSDimitry Andric   case ISD::FMA:
8415ffd83dbSDimitry Andric   case ISD::FMAD: {
8425ffd83dbSDimitry Andric     // Negating a fma is not free if it has users without source mods.
8435ffd83dbSDimitry Andric     if (!allUsesHaveSourceMods(Op.getNode()))
8445ffd83dbSDimitry Andric       return SDValue();
8455ffd83dbSDimitry Andric     break;
8465ffd83dbSDimitry Andric   }
84706c3fb27SDimitry Andric   case AMDGPUISD::RCP: {
84806c3fb27SDimitry Andric     SDValue Src = Op.getOperand(0);
84906c3fb27SDimitry Andric     EVT VT = Op.getValueType();
85006c3fb27SDimitry Andric     SDLoc SL(Op);
85106c3fb27SDimitry Andric 
85206c3fb27SDimitry Andric     SDValue NegSrc = getNegatedExpression(Src, DAG, LegalOperations,
85306c3fb27SDimitry Andric                                           ForCodeSize, Cost, Depth + 1);
85406c3fb27SDimitry Andric     if (NegSrc)
85506c3fb27SDimitry Andric       return DAG.getNode(AMDGPUISD::RCP, SL, VT, NegSrc, Op->getFlags());
85606c3fb27SDimitry Andric     return SDValue();
85706c3fb27SDimitry Andric   }
8585ffd83dbSDimitry Andric   default:
8595ffd83dbSDimitry Andric     break;
8605ffd83dbSDimitry Andric   }
8615ffd83dbSDimitry Andric 
8625ffd83dbSDimitry Andric   return TargetLowering::getNegatedExpression(Op, DAG, LegalOperations,
8635ffd83dbSDimitry Andric                                               ForCodeSize, Cost, Depth);
8645ffd83dbSDimitry Andric }
8655ffd83dbSDimitry Andric 
8660b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
8670b57cec5SDimitry Andric // Target Properties
8680b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
8690b57cec5SDimitry Andric 
8700b57cec5SDimitry Andric bool AMDGPUTargetLowering::isFAbsFree(EVT VT) const {
8710b57cec5SDimitry Andric   assert(VT.isFloatingPoint());
8720b57cec5SDimitry Andric 
8730b57cec5SDimitry Andric   // Packed operations do not have a fabs modifier.
8740b57cec5SDimitry Andric   return VT == MVT::f32 || VT == MVT::f64 ||
8750b57cec5SDimitry Andric          (Subtarget->has16BitInsts() && VT == MVT::f16);
8760b57cec5SDimitry Andric }
8770b57cec5SDimitry Andric 
8780b57cec5SDimitry Andric bool AMDGPUTargetLowering::isFNegFree(EVT VT) const {
8790b57cec5SDimitry Andric   assert(VT.isFloatingPoint());
880fe6060f1SDimitry Andric   // Report this based on the end legalized type.
881fe6060f1SDimitry Andric   VT = VT.getScalarType();
882fe6060f1SDimitry Andric   return VT == MVT::f32 || VT == MVT::f64 || VT == MVT::f16;
8830b57cec5SDimitry Andric }
8840b57cec5SDimitry Andric 
88506c3fb27SDimitry Andric bool AMDGPUTargetLowering:: storeOfVectorConstantIsCheap(bool IsZero, EVT MemVT,
8860b57cec5SDimitry Andric                                                          unsigned NumElem,
8870b57cec5SDimitry Andric                                                          unsigned AS) const {
8880b57cec5SDimitry Andric   return true;
8890b57cec5SDimitry Andric }
8900b57cec5SDimitry Andric 
8910b57cec5SDimitry Andric bool AMDGPUTargetLowering::aggressivelyPreferBuildVectorSources(EVT VecVT) const {
8920b57cec5SDimitry Andric   // There are few operations which truly have vector input operands. Any vector
8930b57cec5SDimitry Andric   // operation is going to involve operations on each component, and a
8940b57cec5SDimitry Andric   // build_vector will be a copy per element, so it always makes sense to use a
8950b57cec5SDimitry Andric   // build_vector input in place of the extracted element to avoid a copy into a
8960b57cec5SDimitry Andric   // super register.
8970b57cec5SDimitry Andric   //
8980b57cec5SDimitry Andric   // We should probably only do this if all users are extracts only, but this
8990b57cec5SDimitry Andric   // should be the common case.
9000b57cec5SDimitry Andric   return true;
9010b57cec5SDimitry Andric }
9020b57cec5SDimitry Andric 
9030b57cec5SDimitry Andric bool AMDGPUTargetLowering::isTruncateFree(EVT Source, EVT Dest) const {
9040b57cec5SDimitry Andric   // Truncate is just accessing a subregister.
9050b57cec5SDimitry Andric 
9060b57cec5SDimitry Andric   unsigned SrcSize = Source.getSizeInBits();
9070b57cec5SDimitry Andric   unsigned DestSize = Dest.getSizeInBits();
9080b57cec5SDimitry Andric 
9090b57cec5SDimitry Andric   return DestSize < SrcSize && DestSize % 32 == 0 ;
9100b57cec5SDimitry Andric }
9110b57cec5SDimitry Andric 
9120b57cec5SDimitry Andric bool AMDGPUTargetLowering::isTruncateFree(Type *Source, Type *Dest) const {
9130b57cec5SDimitry Andric   // Truncate is just accessing a subregister.
9140b57cec5SDimitry Andric 
9150b57cec5SDimitry Andric   unsigned SrcSize = Source->getScalarSizeInBits();
9160b57cec5SDimitry Andric   unsigned DestSize = Dest->getScalarSizeInBits();
9170b57cec5SDimitry Andric 
9180b57cec5SDimitry Andric   if (DestSize== 16 && Subtarget->has16BitInsts())
9190b57cec5SDimitry Andric     return SrcSize >= 32;
9200b57cec5SDimitry Andric 
9210b57cec5SDimitry Andric   return DestSize < SrcSize && DestSize % 32 == 0;
9220b57cec5SDimitry Andric }
9230b57cec5SDimitry Andric 
9240b57cec5SDimitry Andric bool AMDGPUTargetLowering::isZExtFree(Type *Src, Type *Dest) const {
9250b57cec5SDimitry Andric   unsigned SrcSize = Src->getScalarSizeInBits();
9260b57cec5SDimitry Andric   unsigned DestSize = Dest->getScalarSizeInBits();
9270b57cec5SDimitry Andric 
9280b57cec5SDimitry Andric   if (SrcSize == 16 && Subtarget->has16BitInsts())
9290b57cec5SDimitry Andric     return DestSize >= 32;
9300b57cec5SDimitry Andric 
9310b57cec5SDimitry Andric   return SrcSize == 32 && DestSize == 64;
9320b57cec5SDimitry Andric }
9330b57cec5SDimitry Andric 
9340b57cec5SDimitry Andric bool AMDGPUTargetLowering::isZExtFree(EVT Src, EVT Dest) const {
9350b57cec5SDimitry Andric   // Any register load of a 64-bit value really requires 2 32-bit moves. For all
9360b57cec5SDimitry Andric   // practical purposes, the extra mov 0 to load a 64-bit is free.  As used,
9370b57cec5SDimitry Andric   // this will enable reducing 64-bit operations the 32-bit, which is always
9380b57cec5SDimitry Andric   // good.
9390b57cec5SDimitry Andric 
9400b57cec5SDimitry Andric   if (Src == MVT::i16)
9410b57cec5SDimitry Andric     return Dest == MVT::i32 ||Dest == MVT::i64 ;
9420b57cec5SDimitry Andric 
9430b57cec5SDimitry Andric   return Src == MVT::i32 && Dest == MVT::i64;
9440b57cec5SDimitry Andric }
9450b57cec5SDimitry Andric 
9460b57cec5SDimitry Andric bool AMDGPUTargetLowering::isNarrowingProfitable(EVT SrcVT, EVT DestVT) const {
9470b57cec5SDimitry Andric   // There aren't really 64-bit registers, but pairs of 32-bit ones and only a
9480b57cec5SDimitry Andric   // limited number of native 64-bit operations. Shrinking an operation to fit
9490b57cec5SDimitry Andric   // in a single 32-bit register should always be helpful. As currently used,
9500b57cec5SDimitry Andric   // this is much less general than the name suggests, and is only used in
9510b57cec5SDimitry Andric   // places trying to reduce the sizes of loads. Shrinking loads to < 32-bits is
9520b57cec5SDimitry Andric   // not profitable, and may actually be harmful.
9530b57cec5SDimitry Andric   return SrcVT.getSizeInBits() > 32 && DestVT.getSizeInBits() == 32;
9540b57cec5SDimitry Andric }
9550b57cec5SDimitry Andric 
956bdd1243dSDimitry Andric bool AMDGPUTargetLowering::isDesirableToCommuteWithShift(
957bdd1243dSDimitry Andric     const SDNode* N, CombineLevel Level) const {
958bdd1243dSDimitry Andric   assert((N->getOpcode() == ISD::SHL || N->getOpcode() == ISD::SRA ||
959bdd1243dSDimitry Andric           N->getOpcode() == ISD::SRL) &&
960bdd1243dSDimitry Andric          "Expected shift op");
961bdd1243dSDimitry Andric   // Always commute pre-type legalization and right shifts.
962bdd1243dSDimitry Andric   // We're looking for shl(or(x,y),z) patterns.
963bdd1243dSDimitry Andric   if (Level < CombineLevel::AfterLegalizeTypes ||
964bdd1243dSDimitry Andric       N->getOpcode() != ISD::SHL || N->getOperand(0).getOpcode() != ISD::OR)
965bdd1243dSDimitry Andric     return true;
966bdd1243dSDimitry Andric 
967bdd1243dSDimitry Andric   // If only user is a i32 right-shift, then don't destroy a BFE pattern.
968bdd1243dSDimitry Andric   if (N->getValueType(0) == MVT::i32 && N->use_size() == 1 &&
969bdd1243dSDimitry Andric       (N->use_begin()->getOpcode() == ISD::SRA ||
970bdd1243dSDimitry Andric        N->use_begin()->getOpcode() == ISD::SRL))
971bdd1243dSDimitry Andric     return false;
972bdd1243dSDimitry Andric 
973bdd1243dSDimitry Andric   // Don't destroy or(shl(load_zext(),c), load_zext()) patterns.
974bdd1243dSDimitry Andric   auto IsShiftAndLoad = [](SDValue LHS, SDValue RHS) {
975bdd1243dSDimitry Andric     if (LHS.getOpcode() != ISD::SHL)
976bdd1243dSDimitry Andric       return false;
977bdd1243dSDimitry Andric     auto *RHSLd = dyn_cast<LoadSDNode>(RHS);
978bdd1243dSDimitry Andric     auto *LHS0 = dyn_cast<LoadSDNode>(LHS.getOperand(0));
979bdd1243dSDimitry Andric     auto *LHS1 = dyn_cast<ConstantSDNode>(LHS.getOperand(1));
980bdd1243dSDimitry Andric     return LHS0 && LHS1 && RHSLd && LHS0->getExtensionType() == ISD::ZEXTLOAD &&
981bdd1243dSDimitry Andric            LHS1->getAPIntValue() == LHS0->getMemoryVT().getScalarSizeInBits() &&
982bdd1243dSDimitry Andric            RHSLd->getExtensionType() == ISD::ZEXTLOAD;
983bdd1243dSDimitry Andric   };
984bdd1243dSDimitry Andric   SDValue LHS = N->getOperand(0).getOperand(0);
985bdd1243dSDimitry Andric   SDValue RHS = N->getOperand(0).getOperand(1);
986bdd1243dSDimitry Andric   return !(IsShiftAndLoad(LHS, RHS) || IsShiftAndLoad(RHS, LHS));
987bdd1243dSDimitry Andric }
988bdd1243dSDimitry Andric 
9890b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
9900b57cec5SDimitry Andric // TargetLowering Callbacks
9910b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
9920b57cec5SDimitry Andric 
9930b57cec5SDimitry Andric CCAssignFn *AMDGPUCallLowering::CCAssignFnForCall(CallingConv::ID CC,
9940b57cec5SDimitry Andric                                                   bool IsVarArg) {
9950b57cec5SDimitry Andric   switch (CC) {
9960b57cec5SDimitry Andric   case CallingConv::AMDGPU_VS:
9970b57cec5SDimitry Andric   case CallingConv::AMDGPU_GS:
9980b57cec5SDimitry Andric   case CallingConv::AMDGPU_PS:
9990b57cec5SDimitry Andric   case CallingConv::AMDGPU_CS:
10000b57cec5SDimitry Andric   case CallingConv::AMDGPU_HS:
10010b57cec5SDimitry Andric   case CallingConv::AMDGPU_ES:
10020b57cec5SDimitry Andric   case CallingConv::AMDGPU_LS:
10030b57cec5SDimitry Andric     return CC_AMDGPU;
10040b57cec5SDimitry Andric   case CallingConv::C:
10050b57cec5SDimitry Andric   case CallingConv::Fast:
10060b57cec5SDimitry Andric   case CallingConv::Cold:
10070b57cec5SDimitry Andric     return CC_AMDGPU_Func;
1008e8d8bef9SDimitry Andric   case CallingConv::AMDGPU_Gfx:
1009e8d8bef9SDimitry Andric     return CC_SI_Gfx;
10100b57cec5SDimitry Andric   case CallingConv::AMDGPU_KERNEL:
10110b57cec5SDimitry Andric   case CallingConv::SPIR_KERNEL:
10120b57cec5SDimitry Andric   default:
10130b57cec5SDimitry Andric     report_fatal_error("Unsupported calling convention for call");
10140b57cec5SDimitry Andric   }
10150b57cec5SDimitry Andric }
10160b57cec5SDimitry Andric 
10170b57cec5SDimitry Andric CCAssignFn *AMDGPUCallLowering::CCAssignFnForReturn(CallingConv::ID CC,
10180b57cec5SDimitry Andric                                                     bool IsVarArg) {
10190b57cec5SDimitry Andric   switch (CC) {
10200b57cec5SDimitry Andric   case CallingConv::AMDGPU_KERNEL:
10210b57cec5SDimitry Andric   case CallingConv::SPIR_KERNEL:
10220b57cec5SDimitry Andric     llvm_unreachable("kernels should not be handled here");
10230b57cec5SDimitry Andric   case CallingConv::AMDGPU_VS:
10240b57cec5SDimitry Andric   case CallingConv::AMDGPU_GS:
10250b57cec5SDimitry Andric   case CallingConv::AMDGPU_PS:
10260b57cec5SDimitry Andric   case CallingConv::AMDGPU_CS:
10270b57cec5SDimitry Andric   case CallingConv::AMDGPU_HS:
10280b57cec5SDimitry Andric   case CallingConv::AMDGPU_ES:
10290b57cec5SDimitry Andric   case CallingConv::AMDGPU_LS:
10300b57cec5SDimitry Andric     return RetCC_SI_Shader;
1031e8d8bef9SDimitry Andric   case CallingConv::AMDGPU_Gfx:
1032e8d8bef9SDimitry Andric     return RetCC_SI_Gfx;
10330b57cec5SDimitry Andric   case CallingConv::C:
10340b57cec5SDimitry Andric   case CallingConv::Fast:
10350b57cec5SDimitry Andric   case CallingConv::Cold:
10360b57cec5SDimitry Andric     return RetCC_AMDGPU_Func;
10370b57cec5SDimitry Andric   default:
10380b57cec5SDimitry Andric     report_fatal_error("Unsupported calling convention.");
10390b57cec5SDimitry Andric   }
10400b57cec5SDimitry Andric }
10410b57cec5SDimitry Andric 
10420b57cec5SDimitry Andric /// The SelectionDAGBuilder will automatically promote function arguments
10430b57cec5SDimitry Andric /// with illegal types.  However, this does not work for the AMDGPU targets
10440b57cec5SDimitry Andric /// since the function arguments are stored in memory as these illegal types.
10450b57cec5SDimitry Andric /// In order to handle this properly we need to get the original types sizes
10460b57cec5SDimitry Andric /// from the LLVM IR Function and fixup the ISD:InputArg values before
10470b57cec5SDimitry Andric /// passing them to AnalyzeFormalArguments()
10480b57cec5SDimitry Andric 
10490b57cec5SDimitry Andric /// When the SelectionDAGBuilder computes the Ins, it takes care of splitting
10500b57cec5SDimitry Andric /// input values across multiple registers.  Each item in the Ins array
10510b57cec5SDimitry Andric /// represents a single value that will be stored in registers.  Ins[x].VT is
10520b57cec5SDimitry Andric /// the value type of the value that will be stored in the register, so
10530b57cec5SDimitry Andric /// whatever SDNode we lower the argument to needs to be this type.
10540b57cec5SDimitry Andric ///
10550b57cec5SDimitry Andric /// In order to correctly lower the arguments we need to know the size of each
10560b57cec5SDimitry Andric /// argument.  Since Ins[x].VT gives us the size of the register that will
10570b57cec5SDimitry Andric /// hold the value, we need to look at Ins[x].ArgVT to see the 'real' type
1058349cc55cSDimitry Andric /// for the original function argument so that we can deduce the correct memory
10590b57cec5SDimitry Andric /// type to use for Ins[x].  In most cases the correct memory type will be
10600b57cec5SDimitry Andric /// Ins[x].ArgVT.  However, this will not always be the case.  If, for example,
10610b57cec5SDimitry Andric /// we have a kernel argument of type v8i8, this argument will be split into
10620b57cec5SDimitry Andric /// 8 parts and each part will be represented by its own item in the Ins array.
10630b57cec5SDimitry Andric /// For each part the Ins[x].ArgVT will be the v8i8, which is the full type of
10640b57cec5SDimitry Andric /// the argument before it was split.  From this, we deduce that the memory type
10650b57cec5SDimitry Andric /// for each individual part is i8.  We pass the memory type as LocVT to the
10660b57cec5SDimitry Andric /// calling convention analysis function and the register type (Ins[x].VT) as
10670b57cec5SDimitry Andric /// the ValVT.
10680b57cec5SDimitry Andric void AMDGPUTargetLowering::analyzeFormalArgumentsCompute(
10690b57cec5SDimitry Andric   CCState &State,
10700b57cec5SDimitry Andric   const SmallVectorImpl<ISD::InputArg> &Ins) const {
10710b57cec5SDimitry Andric   const MachineFunction &MF = State.getMachineFunction();
10720b57cec5SDimitry Andric   const Function &Fn = MF.getFunction();
10730b57cec5SDimitry Andric   LLVMContext &Ctx = Fn.getParent()->getContext();
10740b57cec5SDimitry Andric   const AMDGPUSubtarget &ST = AMDGPUSubtarget::get(MF);
107506c3fb27SDimitry Andric   const unsigned ExplicitOffset = ST.getExplicitKernelArgOffset();
10760b57cec5SDimitry Andric   CallingConv::ID CC = Fn.getCallingConv();
10770b57cec5SDimitry Andric 
10785ffd83dbSDimitry Andric   Align MaxAlign = Align(1);
10790b57cec5SDimitry Andric   uint64_t ExplicitArgOffset = 0;
10800b57cec5SDimitry Andric   const DataLayout &DL = Fn.getParent()->getDataLayout();
10810b57cec5SDimitry Andric 
10820b57cec5SDimitry Andric   unsigned InIndex = 0;
10830b57cec5SDimitry Andric 
10840b57cec5SDimitry Andric   for (const Argument &Arg : Fn.args()) {
1085e8d8bef9SDimitry Andric     const bool IsByRef = Arg.hasByRefAttr();
10860b57cec5SDimitry Andric     Type *BaseArgTy = Arg.getType();
1087e8d8bef9SDimitry Andric     Type *MemArgTy = IsByRef ? Arg.getParamByRefType() : BaseArgTy;
108881ad6265SDimitry Andric     Align Alignment = DL.getValueOrABITypeAlignment(
1089bdd1243dSDimitry Andric         IsByRef ? Arg.getParamAlign() : std::nullopt, MemArgTy);
109081ad6265SDimitry Andric     MaxAlign = std::max(Alignment, MaxAlign);
1091e8d8bef9SDimitry Andric     uint64_t AllocSize = DL.getTypeAllocSize(MemArgTy);
10920b57cec5SDimitry Andric 
10935ffd83dbSDimitry Andric     uint64_t ArgOffset = alignTo(ExplicitArgOffset, Alignment) + ExplicitOffset;
10945ffd83dbSDimitry Andric     ExplicitArgOffset = alignTo(ExplicitArgOffset, Alignment) + AllocSize;
10950b57cec5SDimitry Andric 
10960b57cec5SDimitry Andric     // We're basically throwing away everything passed into us and starting over
10970b57cec5SDimitry Andric     // to get accurate in-memory offsets. The "PartOffset" is completely useless
10980b57cec5SDimitry Andric     // to us as computed in Ins.
10990b57cec5SDimitry Andric     //
11000b57cec5SDimitry Andric     // We also need to figure out what type legalization is trying to do to get
11010b57cec5SDimitry Andric     // the correct memory offsets.
11020b57cec5SDimitry Andric 
11030b57cec5SDimitry Andric     SmallVector<EVT, 16> ValueVTs;
11040b57cec5SDimitry Andric     SmallVector<uint64_t, 16> Offsets;
11050b57cec5SDimitry Andric     ComputeValueVTs(*this, DL, BaseArgTy, ValueVTs, &Offsets, ArgOffset);
11060b57cec5SDimitry Andric 
11070b57cec5SDimitry Andric     for (unsigned Value = 0, NumValues = ValueVTs.size();
11080b57cec5SDimitry Andric          Value != NumValues; ++Value) {
11090b57cec5SDimitry Andric       uint64_t BasePartOffset = Offsets[Value];
11100b57cec5SDimitry Andric 
11110b57cec5SDimitry Andric       EVT ArgVT = ValueVTs[Value];
11120b57cec5SDimitry Andric       EVT MemVT = ArgVT;
11130b57cec5SDimitry Andric       MVT RegisterVT = getRegisterTypeForCallingConv(Ctx, CC, ArgVT);
11140b57cec5SDimitry Andric       unsigned NumRegs = getNumRegistersForCallingConv(Ctx, CC, ArgVT);
11150b57cec5SDimitry Andric 
11160b57cec5SDimitry Andric       if (NumRegs == 1) {
11170b57cec5SDimitry Andric         // This argument is not split, so the IR type is the memory type.
11180b57cec5SDimitry Andric         if (ArgVT.isExtended()) {
11190b57cec5SDimitry Andric           // We have an extended type, like i24, so we should just use the
11200b57cec5SDimitry Andric           // register type.
11210b57cec5SDimitry Andric           MemVT = RegisterVT;
11220b57cec5SDimitry Andric         } else {
11230b57cec5SDimitry Andric           MemVT = ArgVT;
11240b57cec5SDimitry Andric         }
11250b57cec5SDimitry Andric       } else if (ArgVT.isVector() && RegisterVT.isVector() &&
11260b57cec5SDimitry Andric                  ArgVT.getScalarType() == RegisterVT.getScalarType()) {
11270b57cec5SDimitry Andric         assert(ArgVT.getVectorNumElements() > RegisterVT.getVectorNumElements());
11280b57cec5SDimitry Andric         // We have a vector value which has been split into a vector with
11290b57cec5SDimitry Andric         // the same scalar type, but fewer elements.  This should handle
11300b57cec5SDimitry Andric         // all the floating-point vector types.
11310b57cec5SDimitry Andric         MemVT = RegisterVT;
11320b57cec5SDimitry Andric       } else if (ArgVT.isVector() &&
11330b57cec5SDimitry Andric                  ArgVT.getVectorNumElements() == NumRegs) {
11340b57cec5SDimitry Andric         // This arg has been split so that each element is stored in a separate
11350b57cec5SDimitry Andric         // register.
11360b57cec5SDimitry Andric         MemVT = ArgVT.getScalarType();
11370b57cec5SDimitry Andric       } else if (ArgVT.isExtended()) {
11380b57cec5SDimitry Andric         // We have an extended type, like i65.
11390b57cec5SDimitry Andric         MemVT = RegisterVT;
11400b57cec5SDimitry Andric       } else {
11410b57cec5SDimitry Andric         unsigned MemoryBits = ArgVT.getStoreSizeInBits() / NumRegs;
11420b57cec5SDimitry Andric         assert(ArgVT.getStoreSizeInBits() % NumRegs == 0);
11430b57cec5SDimitry Andric         if (RegisterVT.isInteger()) {
11440b57cec5SDimitry Andric           MemVT = EVT::getIntegerVT(State.getContext(), MemoryBits);
11450b57cec5SDimitry Andric         } else if (RegisterVT.isVector()) {
11460b57cec5SDimitry Andric           assert(!RegisterVT.getScalarType().isFloatingPoint());
11470b57cec5SDimitry Andric           unsigned NumElements = RegisterVT.getVectorNumElements();
11480b57cec5SDimitry Andric           assert(MemoryBits % NumElements == 0);
11490b57cec5SDimitry Andric           // This vector type has been split into another vector type with
11500b57cec5SDimitry Andric           // a different elements size.
11510b57cec5SDimitry Andric           EVT ScalarVT = EVT::getIntegerVT(State.getContext(),
11520b57cec5SDimitry Andric                                            MemoryBits / NumElements);
11530b57cec5SDimitry Andric           MemVT = EVT::getVectorVT(State.getContext(), ScalarVT, NumElements);
11540b57cec5SDimitry Andric         } else {
11550b57cec5SDimitry Andric           llvm_unreachable("cannot deduce memory type.");
11560b57cec5SDimitry Andric         }
11570b57cec5SDimitry Andric       }
11580b57cec5SDimitry Andric 
11590b57cec5SDimitry Andric       // Convert one element vectors to scalar.
11600b57cec5SDimitry Andric       if (MemVT.isVector() && MemVT.getVectorNumElements() == 1)
11610b57cec5SDimitry Andric         MemVT = MemVT.getScalarType();
11620b57cec5SDimitry Andric 
11630b57cec5SDimitry Andric       // Round up vec3/vec5 argument.
11640b57cec5SDimitry Andric       if (MemVT.isVector() && !MemVT.isPow2VectorType()) {
11650b57cec5SDimitry Andric         assert(MemVT.getVectorNumElements() == 3 ||
1166bdd1243dSDimitry Andric                MemVT.getVectorNumElements() == 5 ||
1167bdd1243dSDimitry Andric                (MemVT.getVectorNumElements() >= 9 &&
1168bdd1243dSDimitry Andric                 MemVT.getVectorNumElements() <= 12));
11690b57cec5SDimitry Andric         MemVT = MemVT.getPow2VectorType(State.getContext());
11705ffd83dbSDimitry Andric       } else if (!MemVT.isSimple() && !MemVT.isVector()) {
11715ffd83dbSDimitry Andric         MemVT = MemVT.getRoundIntegerType(State.getContext());
11720b57cec5SDimitry Andric       }
11730b57cec5SDimitry Andric 
11740b57cec5SDimitry Andric       unsigned PartOffset = 0;
11750b57cec5SDimitry Andric       for (unsigned i = 0; i != NumRegs; ++i) {
11760b57cec5SDimitry Andric         State.addLoc(CCValAssign::getCustomMem(InIndex++, RegisterVT,
11770b57cec5SDimitry Andric                                                BasePartOffset + PartOffset,
11780b57cec5SDimitry Andric                                                MemVT.getSimpleVT(),
11790b57cec5SDimitry Andric                                                CCValAssign::Full));
11800b57cec5SDimitry Andric         PartOffset += MemVT.getStoreSize();
11810b57cec5SDimitry Andric       }
11820b57cec5SDimitry Andric     }
11830b57cec5SDimitry Andric   }
11840b57cec5SDimitry Andric }
11850b57cec5SDimitry Andric 
11860b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerReturn(
11870b57cec5SDimitry Andric   SDValue Chain, CallingConv::ID CallConv,
11880b57cec5SDimitry Andric   bool isVarArg,
11890b57cec5SDimitry Andric   const SmallVectorImpl<ISD::OutputArg> &Outs,
11900b57cec5SDimitry Andric   const SmallVectorImpl<SDValue> &OutVals,
11910b57cec5SDimitry Andric   const SDLoc &DL, SelectionDAG &DAG) const {
11920b57cec5SDimitry Andric   // FIXME: Fails for r600 tests
11930b57cec5SDimitry Andric   //assert(!isVarArg && Outs.empty() && OutVals.empty() &&
11940b57cec5SDimitry Andric   // "wave terminate should not have return values");
11950b57cec5SDimitry Andric   return DAG.getNode(AMDGPUISD::ENDPGM, DL, MVT::Other, Chain);
11960b57cec5SDimitry Andric }
11970b57cec5SDimitry Andric 
11980b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
11990b57cec5SDimitry Andric // Target specific lowering
12000b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
12010b57cec5SDimitry Andric 
12020b57cec5SDimitry Andric /// Selects the correct CCAssignFn for a given CallingConvention value.
12030b57cec5SDimitry Andric CCAssignFn *AMDGPUTargetLowering::CCAssignFnForCall(CallingConv::ID CC,
12040b57cec5SDimitry Andric                                                     bool IsVarArg) {
12050b57cec5SDimitry Andric   return AMDGPUCallLowering::CCAssignFnForCall(CC, IsVarArg);
12060b57cec5SDimitry Andric }
12070b57cec5SDimitry Andric 
12080b57cec5SDimitry Andric CCAssignFn *AMDGPUTargetLowering::CCAssignFnForReturn(CallingConv::ID CC,
12090b57cec5SDimitry Andric                                                       bool IsVarArg) {
12100b57cec5SDimitry Andric   return AMDGPUCallLowering::CCAssignFnForReturn(CC, IsVarArg);
12110b57cec5SDimitry Andric }
12120b57cec5SDimitry Andric 
12130b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::addTokenForArgument(SDValue Chain,
12140b57cec5SDimitry Andric                                                   SelectionDAG &DAG,
12150b57cec5SDimitry Andric                                                   MachineFrameInfo &MFI,
12160b57cec5SDimitry Andric                                                   int ClobberedFI) const {
12170b57cec5SDimitry Andric   SmallVector<SDValue, 8> ArgChains;
12180b57cec5SDimitry Andric   int64_t FirstByte = MFI.getObjectOffset(ClobberedFI);
12190b57cec5SDimitry Andric   int64_t LastByte = FirstByte + MFI.getObjectSize(ClobberedFI) - 1;
12200b57cec5SDimitry Andric 
12210b57cec5SDimitry Andric   // Include the original chain at the beginning of the list. When this is
12220b57cec5SDimitry Andric   // used by target LowerCall hooks, this helps legalize find the
12230b57cec5SDimitry Andric   // CALLSEQ_BEGIN node.
12240b57cec5SDimitry Andric   ArgChains.push_back(Chain);
12250b57cec5SDimitry Andric 
12260b57cec5SDimitry Andric   // Add a chain value for each stack argument corresponding
1227349cc55cSDimitry Andric   for (SDNode *U : DAG.getEntryNode().getNode()->uses()) {
1228349cc55cSDimitry Andric     if (LoadSDNode *L = dyn_cast<LoadSDNode>(U)) {
12290b57cec5SDimitry Andric       if (FrameIndexSDNode *FI = dyn_cast<FrameIndexSDNode>(L->getBasePtr())) {
12300b57cec5SDimitry Andric         if (FI->getIndex() < 0) {
12310b57cec5SDimitry Andric           int64_t InFirstByte = MFI.getObjectOffset(FI->getIndex());
12320b57cec5SDimitry Andric           int64_t InLastByte = InFirstByte;
12330b57cec5SDimitry Andric           InLastByte += MFI.getObjectSize(FI->getIndex()) - 1;
12340b57cec5SDimitry Andric 
12350b57cec5SDimitry Andric           if ((InFirstByte <= FirstByte && FirstByte <= InLastByte) ||
12360b57cec5SDimitry Andric               (FirstByte <= InFirstByte && InFirstByte <= LastByte))
12370b57cec5SDimitry Andric             ArgChains.push_back(SDValue(L, 1));
12380b57cec5SDimitry Andric         }
12390b57cec5SDimitry Andric       }
12400b57cec5SDimitry Andric     }
12410b57cec5SDimitry Andric   }
12420b57cec5SDimitry Andric 
12430b57cec5SDimitry Andric   // Build a tokenfactor for all the chains.
12440b57cec5SDimitry Andric   return DAG.getNode(ISD::TokenFactor, SDLoc(Chain), MVT::Other, ArgChains);
12450b57cec5SDimitry Andric }
12460b57cec5SDimitry Andric 
12470b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::lowerUnhandledCall(CallLoweringInfo &CLI,
12480b57cec5SDimitry Andric                                                  SmallVectorImpl<SDValue> &InVals,
12490b57cec5SDimitry Andric                                                  StringRef Reason) const {
12500b57cec5SDimitry Andric   SDValue Callee = CLI.Callee;
12510b57cec5SDimitry Andric   SelectionDAG &DAG = CLI.DAG;
12520b57cec5SDimitry Andric 
12530b57cec5SDimitry Andric   const Function &Fn = DAG.getMachineFunction().getFunction();
12540b57cec5SDimitry Andric 
12550b57cec5SDimitry Andric   StringRef FuncName("<unknown>");
12560b57cec5SDimitry Andric 
12570b57cec5SDimitry Andric   if (const ExternalSymbolSDNode *G = dyn_cast<ExternalSymbolSDNode>(Callee))
12580b57cec5SDimitry Andric     FuncName = G->getSymbol();
12590b57cec5SDimitry Andric   else if (const GlobalAddressSDNode *G = dyn_cast<GlobalAddressSDNode>(Callee))
12600b57cec5SDimitry Andric     FuncName = G->getGlobal()->getName();
12610b57cec5SDimitry Andric 
12620b57cec5SDimitry Andric   DiagnosticInfoUnsupported NoCalls(
12630b57cec5SDimitry Andric     Fn, Reason + FuncName, CLI.DL.getDebugLoc());
12640b57cec5SDimitry Andric   DAG.getContext()->diagnose(NoCalls);
12650b57cec5SDimitry Andric 
12660b57cec5SDimitry Andric   if (!CLI.IsTailCall) {
12670b57cec5SDimitry Andric     for (unsigned I = 0, E = CLI.Ins.size(); I != E; ++I)
12680b57cec5SDimitry Andric       InVals.push_back(DAG.getUNDEF(CLI.Ins[I].VT));
12690b57cec5SDimitry Andric   }
12700b57cec5SDimitry Andric 
12710b57cec5SDimitry Andric   return DAG.getEntryNode();
12720b57cec5SDimitry Andric }
12730b57cec5SDimitry Andric 
12740b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerCall(CallLoweringInfo &CLI,
12750b57cec5SDimitry Andric                                         SmallVectorImpl<SDValue> &InVals) const {
12760b57cec5SDimitry Andric   return lowerUnhandledCall(CLI, InVals, "unsupported call to function ");
12770b57cec5SDimitry Andric }
12780b57cec5SDimitry Andric 
12790b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerDYNAMIC_STACKALLOC(SDValue Op,
12800b57cec5SDimitry Andric                                                       SelectionDAG &DAG) const {
12810b57cec5SDimitry Andric   const Function &Fn = DAG.getMachineFunction().getFunction();
12820b57cec5SDimitry Andric 
12830b57cec5SDimitry Andric   DiagnosticInfoUnsupported NoDynamicAlloca(Fn, "unsupported dynamic alloca",
12840b57cec5SDimitry Andric                                             SDLoc(Op).getDebugLoc());
12850b57cec5SDimitry Andric   DAG.getContext()->diagnose(NoDynamicAlloca);
12860b57cec5SDimitry Andric   auto Ops = {DAG.getConstant(0, SDLoc(), Op.getValueType()), Op.getOperand(0)};
12870b57cec5SDimitry Andric   return DAG.getMergeValues(Ops, SDLoc());
12880b57cec5SDimitry Andric }
12890b57cec5SDimitry Andric 
12900b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerOperation(SDValue Op,
12910b57cec5SDimitry Andric                                              SelectionDAG &DAG) const {
12920b57cec5SDimitry Andric   switch (Op.getOpcode()) {
12930b57cec5SDimitry Andric   default:
12940b57cec5SDimitry Andric     Op->print(errs(), &DAG);
12950b57cec5SDimitry Andric     llvm_unreachable("Custom lowering code for this "
12960b57cec5SDimitry Andric                      "instruction is not implemented yet!");
12970b57cec5SDimitry Andric     break;
12980b57cec5SDimitry Andric   case ISD::SIGN_EXTEND_INREG: return LowerSIGN_EXTEND_INREG(Op, DAG);
12990b57cec5SDimitry Andric   case ISD::CONCAT_VECTORS: return LowerCONCAT_VECTORS(Op, DAG);
13000b57cec5SDimitry Andric   case ISD::EXTRACT_SUBVECTOR: return LowerEXTRACT_SUBVECTOR(Op, DAG);
13010b57cec5SDimitry Andric   case ISD::UDIVREM: return LowerUDIVREM(Op, DAG);
13020b57cec5SDimitry Andric   case ISD::SDIVREM: return LowerSDIVREM(Op, DAG);
13030b57cec5SDimitry Andric   case ISD::FREM: return LowerFREM(Op, DAG);
13040b57cec5SDimitry Andric   case ISD::FCEIL: return LowerFCEIL(Op, DAG);
13050b57cec5SDimitry Andric   case ISD::FTRUNC: return LowerFTRUNC(Op, DAG);
13060b57cec5SDimitry Andric   case ISD::FRINT: return LowerFRINT(Op, DAG);
13070b57cec5SDimitry Andric   case ISD::FNEARBYINT: return LowerFNEARBYINT(Op, DAG);
1308bdd1243dSDimitry Andric   case ISD::FROUNDEVEN:
1309bdd1243dSDimitry Andric     return LowerFROUNDEVEN(Op, DAG);
13100b57cec5SDimitry Andric   case ISD::FROUND: return LowerFROUND(Op, DAG);
13110b57cec5SDimitry Andric   case ISD::FFLOOR: return LowerFFLOOR(Op, DAG);
131206c3fb27SDimitry Andric   case ISD::FLOG2:
131306c3fb27SDimitry Andric     return LowerFLOG2(Op, DAG);
13140b57cec5SDimitry Andric   case ISD::FLOG:
13150b57cec5SDimitry Andric   case ISD::FLOG10:
131606c3fb27SDimitry Andric     return LowerFLOGCommon(Op, DAG);
13170b57cec5SDimitry Andric   case ISD::FEXP:
13180b57cec5SDimitry Andric     return lowerFEXP(Op, DAG);
131906c3fb27SDimitry Andric   case ISD::FEXP2:
132006c3fb27SDimitry Andric     return lowerFEXP2(Op, DAG);
13210b57cec5SDimitry Andric   case ISD::SINT_TO_FP: return LowerSINT_TO_FP(Op, DAG);
13220b57cec5SDimitry Andric   case ISD::UINT_TO_FP: return LowerUINT_TO_FP(Op, DAG);
13230b57cec5SDimitry Andric   case ISD::FP_TO_FP16: return LowerFP_TO_FP16(Op, DAG);
1324fe6060f1SDimitry Andric   case ISD::FP_TO_SINT:
1325fe6060f1SDimitry Andric   case ISD::FP_TO_UINT:
1326fe6060f1SDimitry Andric     return LowerFP_TO_INT(Op, DAG);
13270b57cec5SDimitry Andric   case ISD::CTTZ:
13280b57cec5SDimitry Andric   case ISD::CTTZ_ZERO_UNDEF:
13290b57cec5SDimitry Andric   case ISD::CTLZ:
13300b57cec5SDimitry Andric   case ISD::CTLZ_ZERO_UNDEF:
13310b57cec5SDimitry Andric     return LowerCTLZ_CTTZ(Op, DAG);
13320b57cec5SDimitry Andric   case ISD::DYNAMIC_STACKALLOC: return LowerDYNAMIC_STACKALLOC(Op, DAG);
13330b57cec5SDimitry Andric   }
13340b57cec5SDimitry Andric   return Op;
13350b57cec5SDimitry Andric }
13360b57cec5SDimitry Andric 
13370b57cec5SDimitry Andric void AMDGPUTargetLowering::ReplaceNodeResults(SDNode *N,
13380b57cec5SDimitry Andric                                               SmallVectorImpl<SDValue> &Results,
13390b57cec5SDimitry Andric                                               SelectionDAG &DAG) const {
13400b57cec5SDimitry Andric   switch (N->getOpcode()) {
13410b57cec5SDimitry Andric   case ISD::SIGN_EXTEND_INREG:
13420b57cec5SDimitry Andric     // Different parts of legalization seem to interpret which type of
13430b57cec5SDimitry Andric     // sign_extend_inreg is the one to check for custom lowering. The extended
13440b57cec5SDimitry Andric     // from type is what really matters, but some places check for custom
13450b57cec5SDimitry Andric     // lowering of the result type. This results in trying to use
13460b57cec5SDimitry Andric     // ReplaceNodeResults to sext_in_reg to an illegal type, so we'll just do
13470b57cec5SDimitry Andric     // nothing here and let the illegal result integer be handled normally.
13480b57cec5SDimitry Andric     return;
134906c3fb27SDimitry Andric   case ISD::FLOG2:
135006c3fb27SDimitry Andric     if (SDValue Lowered = LowerFLOG2(SDValue(N, 0), DAG))
135106c3fb27SDimitry Andric       Results.push_back(Lowered);
135206c3fb27SDimitry Andric     return;
135306c3fb27SDimitry Andric   case ISD::FLOG:
135406c3fb27SDimitry Andric   case ISD::FLOG10:
135506c3fb27SDimitry Andric     if (SDValue Lowered = LowerFLOGCommon(SDValue(N, 0), DAG))
135606c3fb27SDimitry Andric       Results.push_back(Lowered);
135706c3fb27SDimitry Andric     return;
135806c3fb27SDimitry Andric   case ISD::FEXP2:
135906c3fb27SDimitry Andric     if (SDValue Lowered = lowerFEXP2(SDValue(N, 0), DAG))
136006c3fb27SDimitry Andric       Results.push_back(Lowered);
136106c3fb27SDimitry Andric     return;
136206c3fb27SDimitry Andric   case ISD::FEXP:
136306c3fb27SDimitry Andric     if (SDValue Lowered = lowerFEXP(SDValue(N, 0), DAG))
136406c3fb27SDimitry Andric       Results.push_back(Lowered);
136506c3fb27SDimitry Andric     return;
13660b57cec5SDimitry Andric   default:
13670b57cec5SDimitry Andric     return;
13680b57cec5SDimitry Andric   }
13690b57cec5SDimitry Andric }
13700b57cec5SDimitry Andric 
13710b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerGlobalAddress(AMDGPUMachineFunction* MFI,
13720b57cec5SDimitry Andric                                                  SDValue Op,
13730b57cec5SDimitry Andric                                                  SelectionDAG &DAG) const {
13740b57cec5SDimitry Andric 
13750b57cec5SDimitry Andric   const DataLayout &DL = DAG.getDataLayout();
13760b57cec5SDimitry Andric   GlobalAddressSDNode *G = cast<GlobalAddressSDNode>(Op);
13770b57cec5SDimitry Andric   const GlobalValue *GV = G->getGlobal();
13780b57cec5SDimitry Andric 
137906c3fb27SDimitry Andric   if (!MFI->isModuleEntryFunction()) {
138006c3fb27SDimitry Andric     if (std::optional<uint32_t> Address =
138106c3fb27SDimitry Andric             AMDGPUMachineFunction::getLDSAbsoluteAddress(*GV)) {
138206c3fb27SDimitry Andric       return DAG.getConstant(*Address, SDLoc(Op), Op.getValueType());
138306c3fb27SDimitry Andric     }
138406c3fb27SDimitry Andric   }
138506c3fb27SDimitry Andric 
13860b57cec5SDimitry Andric   if (G->getAddressSpace() == AMDGPUAS::LOCAL_ADDRESS ||
13870b57cec5SDimitry Andric       G->getAddressSpace() == AMDGPUAS::REGION_ADDRESS) {
1388fe6060f1SDimitry Andric     if (!MFI->isModuleEntryFunction() &&
1389fe6060f1SDimitry Andric         !GV->getName().equals("llvm.amdgcn.module.lds")) {
13905ffd83dbSDimitry Andric       SDLoc DL(Op);
13910b57cec5SDimitry Andric       const Function &Fn = DAG.getMachineFunction().getFunction();
13920b57cec5SDimitry Andric       DiagnosticInfoUnsupported BadLDSDecl(
13935ffd83dbSDimitry Andric         Fn, "local memory global used by non-kernel function",
13945ffd83dbSDimitry Andric         DL.getDebugLoc(), DS_Warning);
13950b57cec5SDimitry Andric       DAG.getContext()->diagnose(BadLDSDecl);
13965ffd83dbSDimitry Andric 
13975ffd83dbSDimitry Andric       // We currently don't have a way to correctly allocate LDS objects that
13985ffd83dbSDimitry Andric       // aren't directly associated with a kernel. We do force inlining of
13995ffd83dbSDimitry Andric       // functions that use local objects. However, if these dead functions are
14005ffd83dbSDimitry Andric       // not eliminated, we don't want a compile time error. Just emit a warning
14015ffd83dbSDimitry Andric       // and a trap, since there should be no callable path here.
14025ffd83dbSDimitry Andric       SDValue Trap = DAG.getNode(ISD::TRAP, DL, MVT::Other, DAG.getEntryNode());
14035ffd83dbSDimitry Andric       SDValue OutputChain = DAG.getNode(ISD::TokenFactor, DL, MVT::Other,
14045ffd83dbSDimitry Andric                                         Trap, DAG.getRoot());
14055ffd83dbSDimitry Andric       DAG.setRoot(OutputChain);
14065ffd83dbSDimitry Andric       return DAG.getUNDEF(Op.getValueType());
14070b57cec5SDimitry Andric     }
14080b57cec5SDimitry Andric 
14090b57cec5SDimitry Andric     // XXX: What does the value of G->getOffset() mean?
14100b57cec5SDimitry Andric     assert(G->getOffset() == 0 &&
14110b57cec5SDimitry Andric          "Do not know what to do with an non-zero offset");
14120b57cec5SDimitry Andric 
14130b57cec5SDimitry Andric     // TODO: We could emit code to handle the initialization somewhere.
1414349cc55cSDimitry Andric     // We ignore the initializer for now and legalize it to allow selection.
1415349cc55cSDimitry Andric     // The initializer will anyway get errored out during assembly emission.
14165ffd83dbSDimitry Andric     unsigned Offset = MFI->allocateLDSGlobal(DL, *cast<GlobalVariable>(GV));
14170b57cec5SDimitry Andric     return DAG.getConstant(Offset, SDLoc(Op), Op.getValueType());
14180b57cec5SDimitry Andric   }
14190b57cec5SDimitry Andric   return SDValue();
14200b57cec5SDimitry Andric }
14210b57cec5SDimitry Andric 
14220b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerCONCAT_VECTORS(SDValue Op,
14230b57cec5SDimitry Andric                                                   SelectionDAG &DAG) const {
14240b57cec5SDimitry Andric   SmallVector<SDValue, 8> Args;
1425bdd1243dSDimitry Andric   SDLoc SL(Op);
14260b57cec5SDimitry Andric 
14270b57cec5SDimitry Andric   EVT VT = Op.getValueType();
1428bdd1243dSDimitry Andric   if (VT.getVectorElementType().getSizeInBits() < 32) {
1429bdd1243dSDimitry Andric     unsigned OpBitSize = Op.getOperand(0).getValueType().getSizeInBits();
1430bdd1243dSDimitry Andric     if (OpBitSize >= 32 && OpBitSize % 32 == 0) {
1431bdd1243dSDimitry Andric       unsigned NewNumElt = OpBitSize / 32;
1432bdd1243dSDimitry Andric       EVT NewEltVT = (NewNumElt == 1) ? MVT::i32
1433bdd1243dSDimitry Andric                                       : EVT::getVectorVT(*DAG.getContext(),
1434bdd1243dSDimitry Andric                                                          MVT::i32, NewNumElt);
1435bdd1243dSDimitry Andric       for (const SDUse &U : Op->ops()) {
1436bdd1243dSDimitry Andric         SDValue In = U.get();
1437bdd1243dSDimitry Andric         SDValue NewIn = DAG.getNode(ISD::BITCAST, SL, NewEltVT, In);
1438bdd1243dSDimitry Andric         if (NewNumElt > 1)
1439bdd1243dSDimitry Andric           DAG.ExtractVectorElements(NewIn, Args);
1440bdd1243dSDimitry Andric         else
1441bdd1243dSDimitry Andric           Args.push_back(NewIn);
1442bdd1243dSDimitry Andric       }
14430b57cec5SDimitry Andric 
1444bdd1243dSDimitry Andric       EVT NewVT = EVT::getVectorVT(*DAG.getContext(), MVT::i32,
1445bdd1243dSDimitry Andric                                    NewNumElt * Op.getNumOperands());
1446bdd1243dSDimitry Andric       SDValue BV = DAG.getBuildVector(NewVT, SL, Args);
14470b57cec5SDimitry Andric       return DAG.getNode(ISD::BITCAST, SL, VT, BV);
14480b57cec5SDimitry Andric     }
1449bdd1243dSDimitry Andric   }
14500b57cec5SDimitry Andric 
14510b57cec5SDimitry Andric   for (const SDUse &U : Op->ops())
14520b57cec5SDimitry Andric     DAG.ExtractVectorElements(U.get(), Args);
14530b57cec5SDimitry Andric 
1454bdd1243dSDimitry Andric   return DAG.getBuildVector(Op.getValueType(), SL, Args);
14550b57cec5SDimitry Andric }
14560b57cec5SDimitry Andric 
14570b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerEXTRACT_SUBVECTOR(SDValue Op,
14580b57cec5SDimitry Andric                                                      SelectionDAG &DAG) const {
145906c3fb27SDimitry Andric   SDLoc SL(Op);
14600b57cec5SDimitry Andric   SmallVector<SDValue, 8> Args;
14610b57cec5SDimitry Andric   unsigned Start = cast<ConstantSDNode>(Op.getOperand(1))->getZExtValue();
14620b57cec5SDimitry Andric   EVT VT = Op.getValueType();
1463fe6060f1SDimitry Andric   EVT SrcVT = Op.getOperand(0).getValueType();
1464fe6060f1SDimitry Andric 
146506c3fb27SDimitry Andric   if (VT.getScalarSizeInBits() == 16 && Start % 2 == 0) {
146606c3fb27SDimitry Andric     unsigned NumElt = VT.getVectorNumElements();
146706c3fb27SDimitry Andric     unsigned NumSrcElt = SrcVT.getVectorNumElements();
146806c3fb27SDimitry Andric     assert(NumElt % 2 == 0 && NumSrcElt % 2 == 0 && "expect legal types");
1469fe6060f1SDimitry Andric 
147006c3fb27SDimitry Andric     // Extract 32-bit registers at a time.
147106c3fb27SDimitry Andric     EVT NewSrcVT = EVT::getVectorVT(*DAG.getContext(), MVT::i32, NumSrcElt / 2);
147206c3fb27SDimitry Andric     EVT NewVT = NumElt == 2
147306c3fb27SDimitry Andric                     ? MVT::i32
147406c3fb27SDimitry Andric                     : EVT::getVectorVT(*DAG.getContext(), MVT::i32, NumElt / 2);
147506c3fb27SDimitry Andric     SDValue Tmp = DAG.getNode(ISD::BITCAST, SL, NewSrcVT, Op.getOperand(0));
147604eeddc0SDimitry Andric 
147706c3fb27SDimitry Andric     DAG.ExtractVectorElements(Tmp, Args, Start / 2, NumElt / 2);
147806c3fb27SDimitry Andric     if (NumElt == 2)
147906c3fb27SDimitry Andric       Tmp = Args[0];
148006c3fb27SDimitry Andric     else
148106c3fb27SDimitry Andric       Tmp = DAG.getBuildVector(NewVT, SL, Args);
148206c3fb27SDimitry Andric 
148306c3fb27SDimitry Andric     return DAG.getNode(ISD::BITCAST, SL, VT, Tmp);
148406c3fb27SDimitry Andric   }
148581ad6265SDimitry Andric 
14860b57cec5SDimitry Andric   DAG.ExtractVectorElements(Op.getOperand(0), Args, Start,
14870b57cec5SDimitry Andric                             VT.getVectorNumElements());
14880b57cec5SDimitry Andric 
148906c3fb27SDimitry Andric   return DAG.getBuildVector(Op.getValueType(), SL, Args);
14900b57cec5SDimitry Andric }
14910b57cec5SDimitry Andric 
149206c3fb27SDimitry Andric // TODO: Handle fabs too
149306c3fb27SDimitry Andric static SDValue peekFNeg(SDValue Val) {
149406c3fb27SDimitry Andric   if (Val.getOpcode() == ISD::FNEG)
149506c3fb27SDimitry Andric     return Val.getOperand(0);
14960b57cec5SDimitry Andric 
149706c3fb27SDimitry Andric   return Val;
149806c3fb27SDimitry Andric }
149906c3fb27SDimitry Andric 
150006c3fb27SDimitry Andric static SDValue peekFPSignOps(SDValue Val) {
150106c3fb27SDimitry Andric   if (Val.getOpcode() == ISD::FNEG)
150206c3fb27SDimitry Andric     Val = Val.getOperand(0);
150306c3fb27SDimitry Andric   if (Val.getOpcode() == ISD::FABS)
150406c3fb27SDimitry Andric     Val = Val.getOperand(0);
150506c3fb27SDimitry Andric   if (Val.getOpcode() == ISD::FCOPYSIGN)
150606c3fb27SDimitry Andric     Val = Val.getOperand(0);
150706c3fb27SDimitry Andric   return Val;
150806c3fb27SDimitry Andric }
150906c3fb27SDimitry Andric 
151006c3fb27SDimitry Andric SDValue AMDGPUTargetLowering::combineFMinMaxLegacyImpl(
151106c3fb27SDimitry Andric     const SDLoc &DL, EVT VT, SDValue LHS, SDValue RHS, SDValue True,
151206c3fb27SDimitry Andric     SDValue False, SDValue CC, DAGCombinerInfo &DCI) const {
15130b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
15140b57cec5SDimitry Andric   ISD::CondCode CCOpcode = cast<CondCodeSDNode>(CC)->get();
15150b57cec5SDimitry Andric   switch (CCOpcode) {
15160b57cec5SDimitry Andric   case ISD::SETOEQ:
15170b57cec5SDimitry Andric   case ISD::SETONE:
15180b57cec5SDimitry Andric   case ISD::SETUNE:
15190b57cec5SDimitry Andric   case ISD::SETNE:
15200b57cec5SDimitry Andric   case ISD::SETUEQ:
15210b57cec5SDimitry Andric   case ISD::SETEQ:
15220b57cec5SDimitry Andric   case ISD::SETFALSE:
15230b57cec5SDimitry Andric   case ISD::SETFALSE2:
15240b57cec5SDimitry Andric   case ISD::SETTRUE:
15250b57cec5SDimitry Andric   case ISD::SETTRUE2:
15260b57cec5SDimitry Andric   case ISD::SETUO:
15270b57cec5SDimitry Andric   case ISD::SETO:
15280b57cec5SDimitry Andric     break;
15290b57cec5SDimitry Andric   case ISD::SETULE:
15300b57cec5SDimitry Andric   case ISD::SETULT: {
15310b57cec5SDimitry Andric     if (LHS == True)
15320b57cec5SDimitry Andric       return DAG.getNode(AMDGPUISD::FMIN_LEGACY, DL, VT, RHS, LHS);
15330b57cec5SDimitry Andric     return DAG.getNode(AMDGPUISD::FMAX_LEGACY, DL, VT, LHS, RHS);
15340b57cec5SDimitry Andric   }
15350b57cec5SDimitry Andric   case ISD::SETOLE:
15360b57cec5SDimitry Andric   case ISD::SETOLT:
15370b57cec5SDimitry Andric   case ISD::SETLE:
15380b57cec5SDimitry Andric   case ISD::SETLT: {
15390b57cec5SDimitry Andric     // Ordered. Assume ordered for undefined.
15400b57cec5SDimitry Andric 
15410b57cec5SDimitry Andric     // Only do this after legalization to avoid interfering with other combines
15420b57cec5SDimitry Andric     // which might occur.
15430b57cec5SDimitry Andric     if (DCI.getDAGCombineLevel() < AfterLegalizeDAG &&
15440b57cec5SDimitry Andric         !DCI.isCalledByLegalizer())
15450b57cec5SDimitry Andric       return SDValue();
15460b57cec5SDimitry Andric 
15470b57cec5SDimitry Andric     // We need to permute the operands to get the correct NaN behavior. The
15480b57cec5SDimitry Andric     // selected operand is the second one based on the failing compare with NaN,
15490b57cec5SDimitry Andric     // so permute it based on the compare type the hardware uses.
15500b57cec5SDimitry Andric     if (LHS == True)
15510b57cec5SDimitry Andric       return DAG.getNode(AMDGPUISD::FMIN_LEGACY, DL, VT, LHS, RHS);
15520b57cec5SDimitry Andric     return DAG.getNode(AMDGPUISD::FMAX_LEGACY, DL, VT, RHS, LHS);
15530b57cec5SDimitry Andric   }
15540b57cec5SDimitry Andric   case ISD::SETUGE:
15550b57cec5SDimitry Andric   case ISD::SETUGT: {
15560b57cec5SDimitry Andric     if (LHS == True)
15570b57cec5SDimitry Andric       return DAG.getNode(AMDGPUISD::FMAX_LEGACY, DL, VT, RHS, LHS);
15580b57cec5SDimitry Andric     return DAG.getNode(AMDGPUISD::FMIN_LEGACY, DL, VT, LHS, RHS);
15590b57cec5SDimitry Andric   }
15600b57cec5SDimitry Andric   case ISD::SETGT:
15610b57cec5SDimitry Andric   case ISD::SETGE:
15620b57cec5SDimitry Andric   case ISD::SETOGE:
15630b57cec5SDimitry Andric   case ISD::SETOGT: {
15640b57cec5SDimitry Andric     if (DCI.getDAGCombineLevel() < AfterLegalizeDAG &&
15650b57cec5SDimitry Andric         !DCI.isCalledByLegalizer())
15660b57cec5SDimitry Andric       return SDValue();
15670b57cec5SDimitry Andric 
15680b57cec5SDimitry Andric     if (LHS == True)
15690b57cec5SDimitry Andric       return DAG.getNode(AMDGPUISD::FMAX_LEGACY, DL, VT, LHS, RHS);
15700b57cec5SDimitry Andric     return DAG.getNode(AMDGPUISD::FMIN_LEGACY, DL, VT, RHS, LHS);
15710b57cec5SDimitry Andric   }
15720b57cec5SDimitry Andric   case ISD::SETCC_INVALID:
15730b57cec5SDimitry Andric     llvm_unreachable("Invalid setcc condcode!");
15740b57cec5SDimitry Andric   }
15750b57cec5SDimitry Andric   return SDValue();
15760b57cec5SDimitry Andric }
15770b57cec5SDimitry Andric 
157806c3fb27SDimitry Andric /// Generate Min/Max node
157906c3fb27SDimitry Andric SDValue AMDGPUTargetLowering::combineFMinMaxLegacy(const SDLoc &DL, EVT VT,
158006c3fb27SDimitry Andric                                                    SDValue LHS, SDValue RHS,
158106c3fb27SDimitry Andric                                                    SDValue True, SDValue False,
158206c3fb27SDimitry Andric                                                    SDValue CC,
158306c3fb27SDimitry Andric                                                    DAGCombinerInfo &DCI) const {
158406c3fb27SDimitry Andric   if ((LHS == True && RHS == False) || (LHS == False && RHS == True))
158506c3fb27SDimitry Andric     return combineFMinMaxLegacyImpl(DL, VT, LHS, RHS, True, False, CC, DCI);
158606c3fb27SDimitry Andric 
158706c3fb27SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
158806c3fb27SDimitry Andric 
158906c3fb27SDimitry Andric   // If we can't directly match this, try to see if we can fold an fneg to
159006c3fb27SDimitry Andric   // match.
159106c3fb27SDimitry Andric 
159206c3fb27SDimitry Andric   ConstantFPSDNode *CRHS = dyn_cast<ConstantFPSDNode>(RHS);
159306c3fb27SDimitry Andric   ConstantFPSDNode *CFalse = dyn_cast<ConstantFPSDNode>(False);
159406c3fb27SDimitry Andric   SDValue NegTrue = peekFNeg(True);
159506c3fb27SDimitry Andric 
159606c3fb27SDimitry Andric   // Undo the combine foldFreeOpFromSelect does if it helps us match the
159706c3fb27SDimitry Andric   // fmin/fmax.
159806c3fb27SDimitry Andric   //
159906c3fb27SDimitry Andric   // select (fcmp olt (lhs, K)), (fneg lhs), -K
160006c3fb27SDimitry Andric   // -> fneg (fmin_legacy lhs, K)
160106c3fb27SDimitry Andric   //
160206c3fb27SDimitry Andric   // TODO: Use getNegatedExpression
160306c3fb27SDimitry Andric   if (LHS == NegTrue && CFalse && CRHS) {
160406c3fb27SDimitry Andric     APFloat NegRHS = neg(CRHS->getValueAPF());
160506c3fb27SDimitry Andric     if (NegRHS == CFalse->getValueAPF()) {
160606c3fb27SDimitry Andric       SDValue Combined =
160706c3fb27SDimitry Andric           combineFMinMaxLegacyImpl(DL, VT, LHS, RHS, NegTrue, False, CC, DCI);
160806c3fb27SDimitry Andric       if (Combined)
160906c3fb27SDimitry Andric         return DAG.getNode(ISD::FNEG, DL, VT, Combined);
161006c3fb27SDimitry Andric       return SDValue();
161106c3fb27SDimitry Andric     }
161206c3fb27SDimitry Andric   }
161306c3fb27SDimitry Andric 
161406c3fb27SDimitry Andric   return SDValue();
161506c3fb27SDimitry Andric }
161606c3fb27SDimitry Andric 
16170b57cec5SDimitry Andric std::pair<SDValue, SDValue>
16180b57cec5SDimitry Andric AMDGPUTargetLowering::split64BitValue(SDValue Op, SelectionDAG &DAG) const {
16190b57cec5SDimitry Andric   SDLoc SL(Op);
16200b57cec5SDimitry Andric 
16210b57cec5SDimitry Andric   SDValue Vec = DAG.getNode(ISD::BITCAST, SL, MVT::v2i32, Op);
16220b57cec5SDimitry Andric 
16230b57cec5SDimitry Andric   const SDValue Zero = DAG.getConstant(0, SL, MVT::i32);
16240b57cec5SDimitry Andric   const SDValue One = DAG.getConstant(1, SL, MVT::i32);
16250b57cec5SDimitry Andric 
16260b57cec5SDimitry Andric   SDValue Lo = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, MVT::i32, Vec, Zero);
16270b57cec5SDimitry Andric   SDValue Hi = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, MVT::i32, Vec, One);
16280b57cec5SDimitry Andric 
1629bdd1243dSDimitry Andric   return std::pair(Lo, Hi);
16300b57cec5SDimitry Andric }
16310b57cec5SDimitry Andric 
16320b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::getLoHalf64(SDValue Op, SelectionDAG &DAG) const {
16330b57cec5SDimitry Andric   SDLoc SL(Op);
16340b57cec5SDimitry Andric 
16350b57cec5SDimitry Andric   SDValue Vec = DAG.getNode(ISD::BITCAST, SL, MVT::v2i32, Op);
16360b57cec5SDimitry Andric   const SDValue Zero = DAG.getConstant(0, SL, MVT::i32);
16370b57cec5SDimitry Andric   return DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, MVT::i32, Vec, Zero);
16380b57cec5SDimitry Andric }
16390b57cec5SDimitry Andric 
16400b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::getHiHalf64(SDValue Op, SelectionDAG &DAG) const {
16410b57cec5SDimitry Andric   SDLoc SL(Op);
16420b57cec5SDimitry Andric 
16430b57cec5SDimitry Andric   SDValue Vec = DAG.getNode(ISD::BITCAST, SL, MVT::v2i32, Op);
16440b57cec5SDimitry Andric   const SDValue One = DAG.getConstant(1, SL, MVT::i32);
16450b57cec5SDimitry Andric   return DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, MVT::i32, Vec, One);
16460b57cec5SDimitry Andric }
16470b57cec5SDimitry Andric 
16480b57cec5SDimitry Andric // Split a vector type into two parts. The first part is a power of two vector.
16490b57cec5SDimitry Andric // The second part is whatever is left over, and is a scalar if it would
16500b57cec5SDimitry Andric // otherwise be a 1-vector.
16510b57cec5SDimitry Andric std::pair<EVT, EVT>
16520b57cec5SDimitry Andric AMDGPUTargetLowering::getSplitDestVTs(const EVT &VT, SelectionDAG &DAG) const {
16530b57cec5SDimitry Andric   EVT LoVT, HiVT;
16540b57cec5SDimitry Andric   EVT EltVT = VT.getVectorElementType();
16550b57cec5SDimitry Andric   unsigned NumElts = VT.getVectorNumElements();
16560b57cec5SDimitry Andric   unsigned LoNumElts = PowerOf2Ceil((NumElts + 1) / 2);
16570b57cec5SDimitry Andric   LoVT = EVT::getVectorVT(*DAG.getContext(), EltVT, LoNumElts);
16580b57cec5SDimitry Andric   HiVT = NumElts - LoNumElts == 1
16590b57cec5SDimitry Andric              ? EltVT
16600b57cec5SDimitry Andric              : EVT::getVectorVT(*DAG.getContext(), EltVT, NumElts - LoNumElts);
1661bdd1243dSDimitry Andric   return std::pair(LoVT, HiVT);
16620b57cec5SDimitry Andric }
16630b57cec5SDimitry Andric 
16640b57cec5SDimitry Andric // Split a vector value into two parts of types LoVT and HiVT. HiVT could be
16650b57cec5SDimitry Andric // scalar.
16660b57cec5SDimitry Andric std::pair<SDValue, SDValue>
16670b57cec5SDimitry Andric AMDGPUTargetLowering::splitVector(const SDValue &N, const SDLoc &DL,
16680b57cec5SDimitry Andric                                   const EVT &LoVT, const EVT &HiVT,
16690b57cec5SDimitry Andric                                   SelectionDAG &DAG) const {
16700b57cec5SDimitry Andric   assert(LoVT.getVectorNumElements() +
16710b57cec5SDimitry Andric                  (HiVT.isVector() ? HiVT.getVectorNumElements() : 1) <=
16720b57cec5SDimitry Andric              N.getValueType().getVectorNumElements() &&
16730b57cec5SDimitry Andric          "More vector elements requested than available!");
16740b57cec5SDimitry Andric   SDValue Lo = DAG.getNode(ISD::EXTRACT_SUBVECTOR, DL, LoVT, N,
16755ffd83dbSDimitry Andric                            DAG.getVectorIdxConstant(0, DL));
16760b57cec5SDimitry Andric   SDValue Hi = DAG.getNode(
16770b57cec5SDimitry Andric       HiVT.isVector() ? ISD::EXTRACT_SUBVECTOR : ISD::EXTRACT_VECTOR_ELT, DL,
16785ffd83dbSDimitry Andric       HiVT, N, DAG.getVectorIdxConstant(LoVT.getVectorNumElements(), DL));
1679bdd1243dSDimitry Andric   return std::pair(Lo, Hi);
16800b57cec5SDimitry Andric }
16810b57cec5SDimitry Andric 
16820b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::SplitVectorLoad(const SDValue Op,
16830b57cec5SDimitry Andric                                               SelectionDAG &DAG) const {
16840b57cec5SDimitry Andric   LoadSDNode *Load = cast<LoadSDNode>(Op);
16850b57cec5SDimitry Andric   EVT VT = Op.getValueType();
1686480093f4SDimitry Andric   SDLoc SL(Op);
16870b57cec5SDimitry Andric 
16880b57cec5SDimitry Andric 
16890b57cec5SDimitry Andric   // If this is a 2 element vector, we really want to scalarize and not create
16900b57cec5SDimitry Andric   // weird 1 element vectors.
1691480093f4SDimitry Andric   if (VT.getVectorNumElements() == 2) {
1692480093f4SDimitry Andric     SDValue Ops[2];
1693480093f4SDimitry Andric     std::tie(Ops[0], Ops[1]) = scalarizeVectorLoad(Load, DAG);
1694480093f4SDimitry Andric     return DAG.getMergeValues(Ops, SL);
1695480093f4SDimitry Andric   }
16960b57cec5SDimitry Andric 
16970b57cec5SDimitry Andric   SDValue BasePtr = Load->getBasePtr();
16980b57cec5SDimitry Andric   EVT MemVT = Load->getMemoryVT();
16990b57cec5SDimitry Andric 
17000b57cec5SDimitry Andric   const MachinePointerInfo &SrcValue = Load->getMemOperand()->getPointerInfo();
17010b57cec5SDimitry Andric 
17020b57cec5SDimitry Andric   EVT LoVT, HiVT;
17030b57cec5SDimitry Andric   EVT LoMemVT, HiMemVT;
17040b57cec5SDimitry Andric   SDValue Lo, Hi;
17050b57cec5SDimitry Andric 
17060b57cec5SDimitry Andric   std::tie(LoVT, HiVT) = getSplitDestVTs(VT, DAG);
17070b57cec5SDimitry Andric   std::tie(LoMemVT, HiMemVT) = getSplitDestVTs(MemVT, DAG);
17080b57cec5SDimitry Andric   std::tie(Lo, Hi) = splitVector(Op, SL, LoVT, HiVT, DAG);
17090b57cec5SDimitry Andric 
17100b57cec5SDimitry Andric   unsigned Size = LoMemVT.getStoreSize();
171181ad6265SDimitry Andric   Align BaseAlign = Load->getAlign();
171281ad6265SDimitry Andric   Align HiAlign = commonAlignment(BaseAlign, Size);
17130b57cec5SDimitry Andric 
17140b57cec5SDimitry Andric   SDValue LoLoad = DAG.getExtLoad(Load->getExtensionType(), SL, LoVT,
17150b57cec5SDimitry Andric                                   Load->getChain(), BasePtr, SrcValue, LoMemVT,
17160b57cec5SDimitry Andric                                   BaseAlign, Load->getMemOperand()->getFlags());
1717e8d8bef9SDimitry Andric   SDValue HiPtr = DAG.getObjectPtrOffset(SL, BasePtr, TypeSize::Fixed(Size));
17180b57cec5SDimitry Andric   SDValue HiLoad =
17190b57cec5SDimitry Andric       DAG.getExtLoad(Load->getExtensionType(), SL, HiVT, Load->getChain(),
17200b57cec5SDimitry Andric                      HiPtr, SrcValue.getWithOffset(LoMemVT.getStoreSize()),
17210b57cec5SDimitry Andric                      HiMemVT, HiAlign, Load->getMemOperand()->getFlags());
17220b57cec5SDimitry Andric 
17230b57cec5SDimitry Andric   SDValue Join;
17240b57cec5SDimitry Andric   if (LoVT == HiVT) {
17250b57cec5SDimitry Andric     // This is the case that the vector is power of two so was evenly split.
17260b57cec5SDimitry Andric     Join = DAG.getNode(ISD::CONCAT_VECTORS, SL, VT, LoLoad, HiLoad);
17270b57cec5SDimitry Andric   } else {
17280b57cec5SDimitry Andric     Join = DAG.getNode(ISD::INSERT_SUBVECTOR, SL, VT, DAG.getUNDEF(VT), LoLoad,
17295ffd83dbSDimitry Andric                        DAG.getVectorIdxConstant(0, SL));
17305ffd83dbSDimitry Andric     Join = DAG.getNode(
17315ffd83dbSDimitry Andric         HiVT.isVector() ? ISD::INSERT_SUBVECTOR : ISD::INSERT_VECTOR_ELT, SL,
17325ffd83dbSDimitry Andric         VT, Join, HiLoad,
17335ffd83dbSDimitry Andric         DAG.getVectorIdxConstant(LoVT.getVectorNumElements(), SL));
17340b57cec5SDimitry Andric   }
17350b57cec5SDimitry Andric 
17360b57cec5SDimitry Andric   SDValue Ops[] = {Join, DAG.getNode(ISD::TokenFactor, SL, MVT::Other,
17370b57cec5SDimitry Andric                                      LoLoad.getValue(1), HiLoad.getValue(1))};
17380b57cec5SDimitry Andric 
17390b57cec5SDimitry Andric   return DAG.getMergeValues(Ops, SL);
17400b57cec5SDimitry Andric }
17410b57cec5SDimitry Andric 
1742e8d8bef9SDimitry Andric SDValue AMDGPUTargetLowering::WidenOrSplitVectorLoad(SDValue Op,
17430b57cec5SDimitry Andric                                                      SelectionDAG &DAG) const {
17440b57cec5SDimitry Andric   LoadSDNode *Load = cast<LoadSDNode>(Op);
17450b57cec5SDimitry Andric   EVT VT = Op.getValueType();
17460b57cec5SDimitry Andric   SDValue BasePtr = Load->getBasePtr();
17470b57cec5SDimitry Andric   EVT MemVT = Load->getMemoryVT();
17480b57cec5SDimitry Andric   SDLoc SL(Op);
17490b57cec5SDimitry Andric   const MachinePointerInfo &SrcValue = Load->getMemOperand()->getPointerInfo();
175081ad6265SDimitry Andric   Align BaseAlign = Load->getAlign();
1751e8d8bef9SDimitry Andric   unsigned NumElements = MemVT.getVectorNumElements();
1752e8d8bef9SDimitry Andric 
1753e8d8bef9SDimitry Andric   // Widen from vec3 to vec4 when the load is at least 8-byte aligned
1754e8d8bef9SDimitry Andric   // or 16-byte fully dereferenceable. Otherwise, split the vector load.
1755e8d8bef9SDimitry Andric   if (NumElements != 3 ||
175681ad6265SDimitry Andric       (BaseAlign < Align(8) &&
1757e8d8bef9SDimitry Andric        !SrcValue.isDereferenceable(16, *DAG.getContext(), DAG.getDataLayout())))
1758e8d8bef9SDimitry Andric     return SplitVectorLoad(Op, DAG);
1759e8d8bef9SDimitry Andric 
1760e8d8bef9SDimitry Andric   assert(NumElements == 3);
17610b57cec5SDimitry Andric 
17620b57cec5SDimitry Andric   EVT WideVT =
17630b57cec5SDimitry Andric       EVT::getVectorVT(*DAG.getContext(), VT.getVectorElementType(), 4);
17640b57cec5SDimitry Andric   EVT WideMemVT =
17650b57cec5SDimitry Andric       EVT::getVectorVT(*DAG.getContext(), MemVT.getVectorElementType(), 4);
17660b57cec5SDimitry Andric   SDValue WideLoad = DAG.getExtLoad(
17670b57cec5SDimitry Andric       Load->getExtensionType(), SL, WideVT, Load->getChain(), BasePtr, SrcValue,
17680b57cec5SDimitry Andric       WideMemVT, BaseAlign, Load->getMemOperand()->getFlags());
17690b57cec5SDimitry Andric   return DAG.getMergeValues(
17700b57cec5SDimitry Andric       {DAG.getNode(ISD::EXTRACT_SUBVECTOR, SL, VT, WideLoad,
17715ffd83dbSDimitry Andric                    DAG.getVectorIdxConstant(0, SL)),
17720b57cec5SDimitry Andric        WideLoad.getValue(1)},
17730b57cec5SDimitry Andric       SL);
17740b57cec5SDimitry Andric }
17750b57cec5SDimitry Andric 
17760b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::SplitVectorStore(SDValue Op,
17770b57cec5SDimitry Andric                                                SelectionDAG &DAG) const {
17780b57cec5SDimitry Andric   StoreSDNode *Store = cast<StoreSDNode>(Op);
17790b57cec5SDimitry Andric   SDValue Val = Store->getValue();
17800b57cec5SDimitry Andric   EVT VT = Val.getValueType();
17810b57cec5SDimitry Andric 
17820b57cec5SDimitry Andric   // If this is a 2 element vector, we really want to scalarize and not create
17830b57cec5SDimitry Andric   // weird 1 element vectors.
17840b57cec5SDimitry Andric   if (VT.getVectorNumElements() == 2)
17850b57cec5SDimitry Andric     return scalarizeVectorStore(Store, DAG);
17860b57cec5SDimitry Andric 
17870b57cec5SDimitry Andric   EVT MemVT = Store->getMemoryVT();
17880b57cec5SDimitry Andric   SDValue Chain = Store->getChain();
17890b57cec5SDimitry Andric   SDValue BasePtr = Store->getBasePtr();
17900b57cec5SDimitry Andric   SDLoc SL(Op);
17910b57cec5SDimitry Andric 
17920b57cec5SDimitry Andric   EVT LoVT, HiVT;
17930b57cec5SDimitry Andric   EVT LoMemVT, HiMemVT;
17940b57cec5SDimitry Andric   SDValue Lo, Hi;
17950b57cec5SDimitry Andric 
17960b57cec5SDimitry Andric   std::tie(LoVT, HiVT) = getSplitDestVTs(VT, DAG);
17970b57cec5SDimitry Andric   std::tie(LoMemVT, HiMemVT) = getSplitDestVTs(MemVT, DAG);
17980b57cec5SDimitry Andric   std::tie(Lo, Hi) = splitVector(Val, SL, LoVT, HiVT, DAG);
17990b57cec5SDimitry Andric 
18000b57cec5SDimitry Andric   SDValue HiPtr = DAG.getObjectPtrOffset(SL, BasePtr, LoMemVT.getStoreSize());
18010b57cec5SDimitry Andric 
18020b57cec5SDimitry Andric   const MachinePointerInfo &SrcValue = Store->getMemOperand()->getPointerInfo();
180381ad6265SDimitry Andric   Align BaseAlign = Store->getAlign();
18040b57cec5SDimitry Andric   unsigned Size = LoMemVT.getStoreSize();
180581ad6265SDimitry Andric   Align HiAlign = commonAlignment(BaseAlign, Size);
18060b57cec5SDimitry Andric 
18070b57cec5SDimitry Andric   SDValue LoStore =
18080b57cec5SDimitry Andric       DAG.getTruncStore(Chain, SL, Lo, BasePtr, SrcValue, LoMemVT, BaseAlign,
18090b57cec5SDimitry Andric                         Store->getMemOperand()->getFlags());
18100b57cec5SDimitry Andric   SDValue HiStore =
18110b57cec5SDimitry Andric       DAG.getTruncStore(Chain, SL, Hi, HiPtr, SrcValue.getWithOffset(Size),
18120b57cec5SDimitry Andric                         HiMemVT, HiAlign, Store->getMemOperand()->getFlags());
18130b57cec5SDimitry Andric 
18140b57cec5SDimitry Andric   return DAG.getNode(ISD::TokenFactor, SL, MVT::Other, LoStore, HiStore);
18150b57cec5SDimitry Andric }
18160b57cec5SDimitry Andric 
18170b57cec5SDimitry Andric // This is a shortcut for integer division because we have fast i32<->f32
18180b57cec5SDimitry Andric // conversions, and fast f32 reciprocal instructions. The fractional part of a
18190b57cec5SDimitry Andric // float is enough to accurately represent up to a 24-bit signed integer.
18200b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerDIVREM24(SDValue Op, SelectionDAG &DAG,
18210b57cec5SDimitry Andric                                             bool Sign) const {
18220b57cec5SDimitry Andric   SDLoc DL(Op);
18230b57cec5SDimitry Andric   EVT VT = Op.getValueType();
18240b57cec5SDimitry Andric   SDValue LHS = Op.getOperand(0);
18250b57cec5SDimitry Andric   SDValue RHS = Op.getOperand(1);
18260b57cec5SDimitry Andric   MVT IntVT = MVT::i32;
18270b57cec5SDimitry Andric   MVT FltVT = MVT::f32;
18280b57cec5SDimitry Andric 
18290b57cec5SDimitry Andric   unsigned LHSSignBits = DAG.ComputeNumSignBits(LHS);
18300b57cec5SDimitry Andric   if (LHSSignBits < 9)
18310b57cec5SDimitry Andric     return SDValue();
18320b57cec5SDimitry Andric 
18330b57cec5SDimitry Andric   unsigned RHSSignBits = DAG.ComputeNumSignBits(RHS);
18340b57cec5SDimitry Andric   if (RHSSignBits < 9)
18350b57cec5SDimitry Andric     return SDValue();
18360b57cec5SDimitry Andric 
18370b57cec5SDimitry Andric   unsigned BitSize = VT.getSizeInBits();
18380b57cec5SDimitry Andric   unsigned SignBits = std::min(LHSSignBits, RHSSignBits);
18390b57cec5SDimitry Andric   unsigned DivBits = BitSize - SignBits;
18400b57cec5SDimitry Andric   if (Sign)
18410b57cec5SDimitry Andric     ++DivBits;
18420b57cec5SDimitry Andric 
18430b57cec5SDimitry Andric   ISD::NodeType ToFp = Sign ? ISD::SINT_TO_FP : ISD::UINT_TO_FP;
18440b57cec5SDimitry Andric   ISD::NodeType ToInt = Sign ? ISD::FP_TO_SINT : ISD::FP_TO_UINT;
18450b57cec5SDimitry Andric 
18460b57cec5SDimitry Andric   SDValue jq = DAG.getConstant(1, DL, IntVT);
18470b57cec5SDimitry Andric 
18480b57cec5SDimitry Andric   if (Sign) {
18490b57cec5SDimitry Andric     // char|short jq = ia ^ ib;
18500b57cec5SDimitry Andric     jq = DAG.getNode(ISD::XOR, DL, VT, LHS, RHS);
18510b57cec5SDimitry Andric 
18520b57cec5SDimitry Andric     // jq = jq >> (bitsize - 2)
18530b57cec5SDimitry Andric     jq = DAG.getNode(ISD::SRA, DL, VT, jq,
18540b57cec5SDimitry Andric                      DAG.getConstant(BitSize - 2, DL, VT));
18550b57cec5SDimitry Andric 
18560b57cec5SDimitry Andric     // jq = jq | 0x1
18570b57cec5SDimitry Andric     jq = DAG.getNode(ISD::OR, DL, VT, jq, DAG.getConstant(1, DL, VT));
18580b57cec5SDimitry Andric   }
18590b57cec5SDimitry Andric 
18600b57cec5SDimitry Andric   // int ia = (int)LHS;
18610b57cec5SDimitry Andric   SDValue ia = LHS;
18620b57cec5SDimitry Andric 
18630b57cec5SDimitry Andric   // int ib, (int)RHS;
18640b57cec5SDimitry Andric   SDValue ib = RHS;
18650b57cec5SDimitry Andric 
18660b57cec5SDimitry Andric   // float fa = (float)ia;
18670b57cec5SDimitry Andric   SDValue fa = DAG.getNode(ToFp, DL, FltVT, ia);
18680b57cec5SDimitry Andric 
18690b57cec5SDimitry Andric   // float fb = (float)ib;
18700b57cec5SDimitry Andric   SDValue fb = DAG.getNode(ToFp, DL, FltVT, ib);
18710b57cec5SDimitry Andric 
18720b57cec5SDimitry Andric   SDValue fq = DAG.getNode(ISD::FMUL, DL, FltVT,
18730b57cec5SDimitry Andric                            fa, DAG.getNode(AMDGPUISD::RCP, DL, FltVT, fb));
18740b57cec5SDimitry Andric 
18750b57cec5SDimitry Andric   // fq = trunc(fq);
18760b57cec5SDimitry Andric   fq = DAG.getNode(ISD::FTRUNC, DL, FltVT, fq);
18770b57cec5SDimitry Andric 
18780b57cec5SDimitry Andric   // float fqneg = -fq;
18790b57cec5SDimitry Andric   SDValue fqneg = DAG.getNode(ISD::FNEG, DL, FltVT, fq);
18800b57cec5SDimitry Andric 
1881480093f4SDimitry Andric   MachineFunction &MF = DAG.getMachineFunction();
1882bdd1243dSDimitry Andric 
1883bdd1243dSDimitry Andric   bool UseFmadFtz = false;
1884bdd1243dSDimitry Andric   if (Subtarget->isGCN()) {
1885bdd1243dSDimitry Andric     const SIMachineFunctionInfo *MFI = MF.getInfo<SIMachineFunctionInfo>();
188606c3fb27SDimitry Andric     UseFmadFtz =
188706c3fb27SDimitry Andric         MFI->getMode().FP32Denormals != DenormalMode::getPreserveSign();
1888bdd1243dSDimitry Andric   }
1889480093f4SDimitry Andric 
18900b57cec5SDimitry Andric   // float fr = mad(fqneg, fb, fa);
1891bdd1243dSDimitry Andric   unsigned OpCode = !Subtarget->hasMadMacF32Insts() ? (unsigned)ISD::FMA
1892bdd1243dSDimitry Andric                     : UseFmadFtz ? (unsigned)AMDGPUISD::FMAD_FTZ
1893bdd1243dSDimitry Andric                                  : (unsigned)ISD::FMAD;
18940b57cec5SDimitry Andric   SDValue fr = DAG.getNode(OpCode, DL, FltVT, fqneg, fb, fa);
18950b57cec5SDimitry Andric 
18960b57cec5SDimitry Andric   // int iq = (int)fq;
18970b57cec5SDimitry Andric   SDValue iq = DAG.getNode(ToInt, DL, IntVT, fq);
18980b57cec5SDimitry Andric 
18990b57cec5SDimitry Andric   // fr = fabs(fr);
19000b57cec5SDimitry Andric   fr = DAG.getNode(ISD::FABS, DL, FltVT, fr);
19010b57cec5SDimitry Andric 
19020b57cec5SDimitry Andric   // fb = fabs(fb);
19030b57cec5SDimitry Andric   fb = DAG.getNode(ISD::FABS, DL, FltVT, fb);
19040b57cec5SDimitry Andric 
19050b57cec5SDimitry Andric   EVT SetCCVT = getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), VT);
19060b57cec5SDimitry Andric 
19070b57cec5SDimitry Andric   // int cv = fr >= fb;
19080b57cec5SDimitry Andric   SDValue cv = DAG.getSetCC(DL, SetCCVT, fr, fb, ISD::SETOGE);
19090b57cec5SDimitry Andric 
19100b57cec5SDimitry Andric   // jq = (cv ? jq : 0);
19110b57cec5SDimitry Andric   jq = DAG.getNode(ISD::SELECT, DL, VT, cv, jq, DAG.getConstant(0, DL, VT));
19120b57cec5SDimitry Andric 
19130b57cec5SDimitry Andric   // dst = iq + jq;
19140b57cec5SDimitry Andric   SDValue Div = DAG.getNode(ISD::ADD, DL, VT, iq, jq);
19150b57cec5SDimitry Andric 
19160b57cec5SDimitry Andric   // Rem needs compensation, it's easier to recompute it
19170b57cec5SDimitry Andric   SDValue Rem = DAG.getNode(ISD::MUL, DL, VT, Div, RHS);
19180b57cec5SDimitry Andric   Rem = DAG.getNode(ISD::SUB, DL, VT, LHS, Rem);
19190b57cec5SDimitry Andric 
19200b57cec5SDimitry Andric   // Truncate to number of bits this divide really is.
19210b57cec5SDimitry Andric   if (Sign) {
19220b57cec5SDimitry Andric     SDValue InRegSize
19230b57cec5SDimitry Andric       = DAG.getValueType(EVT::getIntegerVT(*DAG.getContext(), DivBits));
19240b57cec5SDimitry Andric     Div = DAG.getNode(ISD::SIGN_EXTEND_INREG, DL, VT, Div, InRegSize);
19250b57cec5SDimitry Andric     Rem = DAG.getNode(ISD::SIGN_EXTEND_INREG, DL, VT, Rem, InRegSize);
19260b57cec5SDimitry Andric   } else {
19270b57cec5SDimitry Andric     SDValue TruncMask = DAG.getConstant((UINT64_C(1) << DivBits) - 1, DL, VT);
19280b57cec5SDimitry Andric     Div = DAG.getNode(ISD::AND, DL, VT, Div, TruncMask);
19290b57cec5SDimitry Andric     Rem = DAG.getNode(ISD::AND, DL, VT, Rem, TruncMask);
19300b57cec5SDimitry Andric   }
19310b57cec5SDimitry Andric 
19320b57cec5SDimitry Andric   return DAG.getMergeValues({ Div, Rem }, DL);
19330b57cec5SDimitry Andric }
19340b57cec5SDimitry Andric 
19350b57cec5SDimitry Andric void AMDGPUTargetLowering::LowerUDIVREM64(SDValue Op,
19360b57cec5SDimitry Andric                                       SelectionDAG &DAG,
19370b57cec5SDimitry Andric                                       SmallVectorImpl<SDValue> &Results) const {
19380b57cec5SDimitry Andric   SDLoc DL(Op);
19390b57cec5SDimitry Andric   EVT VT = Op.getValueType();
19400b57cec5SDimitry Andric 
19410b57cec5SDimitry Andric   assert(VT == MVT::i64 && "LowerUDIVREM64 expects an i64");
19420b57cec5SDimitry Andric 
19430b57cec5SDimitry Andric   EVT HalfVT = VT.getHalfSizedIntegerVT(*DAG.getContext());
19440b57cec5SDimitry Andric 
19450b57cec5SDimitry Andric   SDValue One = DAG.getConstant(1, DL, HalfVT);
19460b57cec5SDimitry Andric   SDValue Zero = DAG.getConstant(0, DL, HalfVT);
19470b57cec5SDimitry Andric 
19480b57cec5SDimitry Andric   //HiLo split
194906c3fb27SDimitry Andric   SDValue LHS_Lo, LHS_Hi;
19500b57cec5SDimitry Andric   SDValue LHS = Op.getOperand(0);
195106c3fb27SDimitry Andric   std::tie(LHS_Lo, LHS_Hi) = DAG.SplitScalar(LHS, DL, HalfVT, HalfVT);
19520b57cec5SDimitry Andric 
195306c3fb27SDimitry Andric   SDValue RHS_Lo, RHS_Hi;
19540b57cec5SDimitry Andric   SDValue RHS = Op.getOperand(1);
195506c3fb27SDimitry Andric   std::tie(RHS_Lo, RHS_Hi) = DAG.SplitScalar(RHS, DL, HalfVT, HalfVT);
19560b57cec5SDimitry Andric 
19570b57cec5SDimitry Andric   if (DAG.MaskedValueIsZero(RHS, APInt::getHighBitsSet(64, 32)) &&
19580b57cec5SDimitry Andric       DAG.MaskedValueIsZero(LHS, APInt::getHighBitsSet(64, 32))) {
19590b57cec5SDimitry Andric 
19600b57cec5SDimitry Andric     SDValue Res = DAG.getNode(ISD::UDIVREM, DL, DAG.getVTList(HalfVT, HalfVT),
19610b57cec5SDimitry Andric                               LHS_Lo, RHS_Lo);
19620b57cec5SDimitry Andric 
19630b57cec5SDimitry Andric     SDValue DIV = DAG.getBuildVector(MVT::v2i32, DL, {Res.getValue(0), Zero});
19640b57cec5SDimitry Andric     SDValue REM = DAG.getBuildVector(MVT::v2i32, DL, {Res.getValue(1), Zero});
19650b57cec5SDimitry Andric 
19660b57cec5SDimitry Andric     Results.push_back(DAG.getNode(ISD::BITCAST, DL, MVT::i64, DIV));
19670b57cec5SDimitry Andric     Results.push_back(DAG.getNode(ISD::BITCAST, DL, MVT::i64, REM));
19680b57cec5SDimitry Andric     return;
19690b57cec5SDimitry Andric   }
19700b57cec5SDimitry Andric 
19710b57cec5SDimitry Andric   if (isTypeLegal(MVT::i64)) {
1972349cc55cSDimitry Andric     // The algorithm here is based on ideas from "Software Integer Division",
1973349cc55cSDimitry Andric     // Tom Rodeheffer, August 2008.
1974349cc55cSDimitry Andric 
1975480093f4SDimitry Andric     MachineFunction &MF = DAG.getMachineFunction();
1976480093f4SDimitry Andric     const SIMachineFunctionInfo *MFI = MF.getInfo<SIMachineFunctionInfo>();
1977480093f4SDimitry Andric 
19780b57cec5SDimitry Andric     // Compute denominator reciprocal.
197906c3fb27SDimitry Andric     unsigned FMAD =
198006c3fb27SDimitry Andric         !Subtarget->hasMadMacF32Insts() ? (unsigned)ISD::FMA
198106c3fb27SDimitry Andric         : MFI->getMode().FP32Denormals == DenormalMode::getPreserveSign()
198206c3fb27SDimitry Andric             ? (unsigned)ISD::FMAD
198306c3fb27SDimitry Andric             : (unsigned)AMDGPUISD::FMAD_FTZ;
19840b57cec5SDimitry Andric 
19850b57cec5SDimitry Andric     SDValue Cvt_Lo = DAG.getNode(ISD::UINT_TO_FP, DL, MVT::f32, RHS_Lo);
19860b57cec5SDimitry Andric     SDValue Cvt_Hi = DAG.getNode(ISD::UINT_TO_FP, DL, MVT::f32, RHS_Hi);
19870b57cec5SDimitry Andric     SDValue Mad1 = DAG.getNode(FMAD, DL, MVT::f32, Cvt_Hi,
19880b57cec5SDimitry Andric       DAG.getConstantFP(APInt(32, 0x4f800000).bitsToFloat(), DL, MVT::f32),
19890b57cec5SDimitry Andric       Cvt_Lo);
19900b57cec5SDimitry Andric     SDValue Rcp = DAG.getNode(AMDGPUISD::RCP, DL, MVT::f32, Mad1);
19910b57cec5SDimitry Andric     SDValue Mul1 = DAG.getNode(ISD::FMUL, DL, MVT::f32, Rcp,
19920b57cec5SDimitry Andric       DAG.getConstantFP(APInt(32, 0x5f7ffffc).bitsToFloat(), DL, MVT::f32));
19930b57cec5SDimitry Andric     SDValue Mul2 = DAG.getNode(ISD::FMUL, DL, MVT::f32, Mul1,
19940b57cec5SDimitry Andric       DAG.getConstantFP(APInt(32, 0x2f800000).bitsToFloat(), DL, MVT::f32));
19950b57cec5SDimitry Andric     SDValue Trunc = DAG.getNode(ISD::FTRUNC, DL, MVT::f32, Mul2);
19960b57cec5SDimitry Andric     SDValue Mad2 = DAG.getNode(FMAD, DL, MVT::f32, Trunc,
19970b57cec5SDimitry Andric       DAG.getConstantFP(APInt(32, 0xcf800000).bitsToFloat(), DL, MVT::f32),
19980b57cec5SDimitry Andric       Mul1);
19990b57cec5SDimitry Andric     SDValue Rcp_Lo = DAG.getNode(ISD::FP_TO_UINT, DL, HalfVT, Mad2);
20000b57cec5SDimitry Andric     SDValue Rcp_Hi = DAG.getNode(ISD::FP_TO_UINT, DL, HalfVT, Trunc);
20010b57cec5SDimitry Andric     SDValue Rcp64 = DAG.getBitcast(VT,
20020b57cec5SDimitry Andric                         DAG.getBuildVector(MVT::v2i32, DL, {Rcp_Lo, Rcp_Hi}));
20030b57cec5SDimitry Andric 
20040b57cec5SDimitry Andric     SDValue Zero64 = DAG.getConstant(0, DL, VT);
20050b57cec5SDimitry Andric     SDValue One64  = DAG.getConstant(1, DL, VT);
20060b57cec5SDimitry Andric     SDValue Zero1 = DAG.getConstant(0, DL, MVT::i1);
20070b57cec5SDimitry Andric     SDVTList HalfCarryVT = DAG.getVTList(HalfVT, MVT::i1);
20080b57cec5SDimitry Andric 
2009349cc55cSDimitry Andric     // First round of UNR (Unsigned integer Newton-Raphson).
20100b57cec5SDimitry Andric     SDValue Neg_RHS = DAG.getNode(ISD::SUB, DL, VT, Zero64, RHS);
20110b57cec5SDimitry Andric     SDValue Mullo1 = DAG.getNode(ISD::MUL, DL, VT, Neg_RHS, Rcp64);
20120b57cec5SDimitry Andric     SDValue Mulhi1 = DAG.getNode(ISD::MULHU, DL, VT, Rcp64, Mullo1);
201306c3fb27SDimitry Andric     SDValue Mulhi1_Lo, Mulhi1_Hi;
201406c3fb27SDimitry Andric     std::tie(Mulhi1_Lo, Mulhi1_Hi) =
201506c3fb27SDimitry Andric         DAG.SplitScalar(Mulhi1, DL, HalfVT, HalfVT);
201606c3fb27SDimitry Andric     SDValue Add1_Lo = DAG.getNode(ISD::UADDO_CARRY, DL, HalfCarryVT, Rcp_Lo,
20170b57cec5SDimitry Andric                                   Mulhi1_Lo, Zero1);
201806c3fb27SDimitry Andric     SDValue Add1_Hi = DAG.getNode(ISD::UADDO_CARRY, DL, HalfCarryVT, Rcp_Hi,
20190b57cec5SDimitry Andric                                   Mulhi1_Hi, Add1_Lo.getValue(1));
20200b57cec5SDimitry Andric     SDValue Add1 = DAG.getBitcast(VT,
20210b57cec5SDimitry Andric                         DAG.getBuildVector(MVT::v2i32, DL, {Add1_Lo, Add1_Hi}));
20220b57cec5SDimitry Andric 
2023349cc55cSDimitry Andric     // Second round of UNR.
20240b57cec5SDimitry Andric     SDValue Mullo2 = DAG.getNode(ISD::MUL, DL, VT, Neg_RHS, Add1);
20250b57cec5SDimitry Andric     SDValue Mulhi2 = DAG.getNode(ISD::MULHU, DL, VT, Add1, Mullo2);
202606c3fb27SDimitry Andric     SDValue Mulhi2_Lo, Mulhi2_Hi;
202706c3fb27SDimitry Andric     std::tie(Mulhi2_Lo, Mulhi2_Hi) =
202806c3fb27SDimitry Andric         DAG.SplitScalar(Mulhi2, DL, HalfVT, HalfVT);
202906c3fb27SDimitry Andric     SDValue Add2_Lo = DAG.getNode(ISD::UADDO_CARRY, DL, HalfCarryVT, Add1_Lo,
20300b57cec5SDimitry Andric                                   Mulhi2_Lo, Zero1);
203106c3fb27SDimitry Andric     SDValue Add2_Hi = DAG.getNode(ISD::UADDO_CARRY, DL, HalfCarryVT, Add1_Hi,
2032349cc55cSDimitry Andric                                   Mulhi2_Hi, Add2_Lo.getValue(1));
20330b57cec5SDimitry Andric     SDValue Add2 = DAG.getBitcast(VT,
20340b57cec5SDimitry Andric                         DAG.getBuildVector(MVT::v2i32, DL, {Add2_Lo, Add2_Hi}));
2035349cc55cSDimitry Andric 
20360b57cec5SDimitry Andric     SDValue Mulhi3 = DAG.getNode(ISD::MULHU, DL, VT, LHS, Add2);
20370b57cec5SDimitry Andric 
20380b57cec5SDimitry Andric     SDValue Mul3 = DAG.getNode(ISD::MUL, DL, VT, RHS, Mulhi3);
20390b57cec5SDimitry Andric 
204006c3fb27SDimitry Andric     SDValue Mul3_Lo, Mul3_Hi;
204106c3fb27SDimitry Andric     std::tie(Mul3_Lo, Mul3_Hi) = DAG.SplitScalar(Mul3, DL, HalfVT, HalfVT);
204206c3fb27SDimitry Andric     SDValue Sub1_Lo = DAG.getNode(ISD::USUBO_CARRY, DL, HalfCarryVT, LHS_Lo,
20430b57cec5SDimitry Andric                                   Mul3_Lo, Zero1);
204406c3fb27SDimitry Andric     SDValue Sub1_Hi = DAG.getNode(ISD::USUBO_CARRY, DL, HalfCarryVT, LHS_Hi,
20450b57cec5SDimitry Andric                                   Mul3_Hi, Sub1_Lo.getValue(1));
20460b57cec5SDimitry Andric     SDValue Sub1_Mi = DAG.getNode(ISD::SUB, DL, HalfVT, LHS_Hi, Mul3_Hi);
20470b57cec5SDimitry Andric     SDValue Sub1 = DAG.getBitcast(VT,
20480b57cec5SDimitry Andric                         DAG.getBuildVector(MVT::v2i32, DL, {Sub1_Lo, Sub1_Hi}));
20490b57cec5SDimitry Andric 
20500b57cec5SDimitry Andric     SDValue MinusOne = DAG.getConstant(0xffffffffu, DL, HalfVT);
20510b57cec5SDimitry Andric     SDValue C1 = DAG.getSelectCC(DL, Sub1_Hi, RHS_Hi, MinusOne, Zero,
20520b57cec5SDimitry Andric                                  ISD::SETUGE);
20530b57cec5SDimitry Andric     SDValue C2 = DAG.getSelectCC(DL, Sub1_Lo, RHS_Lo, MinusOne, Zero,
20540b57cec5SDimitry Andric                                  ISD::SETUGE);
20550b57cec5SDimitry Andric     SDValue C3 = DAG.getSelectCC(DL, Sub1_Hi, RHS_Hi, C2, C1, ISD::SETEQ);
20560b57cec5SDimitry Andric 
20570b57cec5SDimitry Andric     // TODO: Here and below portions of the code can be enclosed into if/endif.
20580b57cec5SDimitry Andric     // Currently control flow is unconditional and we have 4 selects after
20590b57cec5SDimitry Andric     // potential endif to substitute PHIs.
20600b57cec5SDimitry Andric 
20610b57cec5SDimitry Andric     // if C3 != 0 ...
206206c3fb27SDimitry Andric     SDValue Sub2_Lo = DAG.getNode(ISD::USUBO_CARRY, DL, HalfCarryVT, Sub1_Lo,
20630b57cec5SDimitry Andric                                   RHS_Lo, Zero1);
206406c3fb27SDimitry Andric     SDValue Sub2_Mi = DAG.getNode(ISD::USUBO_CARRY, DL, HalfCarryVT, Sub1_Mi,
20650b57cec5SDimitry Andric                                   RHS_Hi, Sub1_Lo.getValue(1));
206606c3fb27SDimitry Andric     SDValue Sub2_Hi = DAG.getNode(ISD::USUBO_CARRY, DL, HalfCarryVT, Sub2_Mi,
20670b57cec5SDimitry Andric                                   Zero, Sub2_Lo.getValue(1));
20680b57cec5SDimitry Andric     SDValue Sub2 = DAG.getBitcast(VT,
20690b57cec5SDimitry Andric                         DAG.getBuildVector(MVT::v2i32, DL, {Sub2_Lo, Sub2_Hi}));
20700b57cec5SDimitry Andric 
20710b57cec5SDimitry Andric     SDValue Add3 = DAG.getNode(ISD::ADD, DL, VT, Mulhi3, One64);
20720b57cec5SDimitry Andric 
20730b57cec5SDimitry Andric     SDValue C4 = DAG.getSelectCC(DL, Sub2_Hi, RHS_Hi, MinusOne, Zero,
20740b57cec5SDimitry Andric                                  ISD::SETUGE);
20750b57cec5SDimitry Andric     SDValue C5 = DAG.getSelectCC(DL, Sub2_Lo, RHS_Lo, MinusOne, Zero,
20760b57cec5SDimitry Andric                                  ISD::SETUGE);
20770b57cec5SDimitry Andric     SDValue C6 = DAG.getSelectCC(DL, Sub2_Hi, RHS_Hi, C5, C4, ISD::SETEQ);
20780b57cec5SDimitry Andric 
20790b57cec5SDimitry Andric     // if (C6 != 0)
20800b57cec5SDimitry Andric     SDValue Add4 = DAG.getNode(ISD::ADD, DL, VT, Add3, One64);
20810b57cec5SDimitry Andric 
208206c3fb27SDimitry Andric     SDValue Sub3_Lo = DAG.getNode(ISD::USUBO_CARRY, DL, HalfCarryVT, Sub2_Lo,
20830b57cec5SDimitry Andric                                   RHS_Lo, Zero1);
208406c3fb27SDimitry Andric     SDValue Sub3_Mi = DAG.getNode(ISD::USUBO_CARRY, DL, HalfCarryVT, Sub2_Mi,
20850b57cec5SDimitry Andric                                   RHS_Hi, Sub2_Lo.getValue(1));
208606c3fb27SDimitry Andric     SDValue Sub3_Hi = DAG.getNode(ISD::USUBO_CARRY, DL, HalfCarryVT, Sub3_Mi,
20870b57cec5SDimitry Andric                                   Zero, Sub3_Lo.getValue(1));
20880b57cec5SDimitry Andric     SDValue Sub3 = DAG.getBitcast(VT,
20890b57cec5SDimitry Andric                         DAG.getBuildVector(MVT::v2i32, DL, {Sub3_Lo, Sub3_Hi}));
20900b57cec5SDimitry Andric 
20910b57cec5SDimitry Andric     // endif C6
20920b57cec5SDimitry Andric     // endif C3
20930b57cec5SDimitry Andric 
20940b57cec5SDimitry Andric     SDValue Sel1 = DAG.getSelectCC(DL, C6, Zero, Add4, Add3, ISD::SETNE);
20950b57cec5SDimitry Andric     SDValue Div  = DAG.getSelectCC(DL, C3, Zero, Sel1, Mulhi3, ISD::SETNE);
20960b57cec5SDimitry Andric 
20970b57cec5SDimitry Andric     SDValue Sel2 = DAG.getSelectCC(DL, C6, Zero, Sub3, Sub2, ISD::SETNE);
20980b57cec5SDimitry Andric     SDValue Rem  = DAG.getSelectCC(DL, C3, Zero, Sel2, Sub1, ISD::SETNE);
20990b57cec5SDimitry Andric 
21000b57cec5SDimitry Andric     Results.push_back(Div);
21010b57cec5SDimitry Andric     Results.push_back(Rem);
21020b57cec5SDimitry Andric 
21030b57cec5SDimitry Andric     return;
21040b57cec5SDimitry Andric   }
21050b57cec5SDimitry Andric 
21060b57cec5SDimitry Andric   // r600 expandion.
21070b57cec5SDimitry Andric   // Get Speculative values
21080b57cec5SDimitry Andric   SDValue DIV_Part = DAG.getNode(ISD::UDIV, DL, HalfVT, LHS_Hi, RHS_Lo);
21090b57cec5SDimitry Andric   SDValue REM_Part = DAG.getNode(ISD::UREM, DL, HalfVT, LHS_Hi, RHS_Lo);
21100b57cec5SDimitry Andric 
21110b57cec5SDimitry Andric   SDValue REM_Lo = DAG.getSelectCC(DL, RHS_Hi, Zero, REM_Part, LHS_Hi, ISD::SETEQ);
21120b57cec5SDimitry Andric   SDValue REM = DAG.getBuildVector(MVT::v2i32, DL, {REM_Lo, Zero});
21130b57cec5SDimitry Andric   REM = DAG.getNode(ISD::BITCAST, DL, MVT::i64, REM);
21140b57cec5SDimitry Andric 
21150b57cec5SDimitry Andric   SDValue DIV_Hi = DAG.getSelectCC(DL, RHS_Hi, Zero, DIV_Part, Zero, ISD::SETEQ);
21160b57cec5SDimitry Andric   SDValue DIV_Lo = Zero;
21170b57cec5SDimitry Andric 
21180b57cec5SDimitry Andric   const unsigned halfBitWidth = HalfVT.getSizeInBits();
21190b57cec5SDimitry Andric 
21200b57cec5SDimitry Andric   for (unsigned i = 0; i < halfBitWidth; ++i) {
21210b57cec5SDimitry Andric     const unsigned bitPos = halfBitWidth - i - 1;
21220b57cec5SDimitry Andric     SDValue POS = DAG.getConstant(bitPos, DL, HalfVT);
21230b57cec5SDimitry Andric     // Get value of high bit
21240b57cec5SDimitry Andric     SDValue HBit = DAG.getNode(ISD::SRL, DL, HalfVT, LHS_Lo, POS);
21250b57cec5SDimitry Andric     HBit = DAG.getNode(ISD::AND, DL, HalfVT, HBit, One);
21260b57cec5SDimitry Andric     HBit = DAG.getNode(ISD::ZERO_EXTEND, DL, VT, HBit);
21270b57cec5SDimitry Andric 
21280b57cec5SDimitry Andric     // Shift
21290b57cec5SDimitry Andric     REM = DAG.getNode(ISD::SHL, DL, VT, REM, DAG.getConstant(1, DL, VT));
21300b57cec5SDimitry Andric     // Add LHS high bit
21310b57cec5SDimitry Andric     REM = DAG.getNode(ISD::OR, DL, VT, REM, HBit);
21320b57cec5SDimitry Andric 
21330b57cec5SDimitry Andric     SDValue BIT = DAG.getConstant(1ULL << bitPos, DL, HalfVT);
21340b57cec5SDimitry Andric     SDValue realBIT = DAG.getSelectCC(DL, REM, RHS, BIT, Zero, ISD::SETUGE);
21350b57cec5SDimitry Andric 
21360b57cec5SDimitry Andric     DIV_Lo = DAG.getNode(ISD::OR, DL, HalfVT, DIV_Lo, realBIT);
21370b57cec5SDimitry Andric 
21380b57cec5SDimitry Andric     // Update REM
21390b57cec5SDimitry Andric     SDValue REM_sub = DAG.getNode(ISD::SUB, DL, VT, REM, RHS);
21400b57cec5SDimitry Andric     REM = DAG.getSelectCC(DL, REM, RHS, REM_sub, REM, ISD::SETUGE);
21410b57cec5SDimitry Andric   }
21420b57cec5SDimitry Andric 
21430b57cec5SDimitry Andric   SDValue DIV = DAG.getBuildVector(MVT::v2i32, DL, {DIV_Lo, DIV_Hi});
21440b57cec5SDimitry Andric   DIV = DAG.getNode(ISD::BITCAST, DL, MVT::i64, DIV);
21450b57cec5SDimitry Andric   Results.push_back(DIV);
21460b57cec5SDimitry Andric   Results.push_back(REM);
21470b57cec5SDimitry Andric }
21480b57cec5SDimitry Andric 
21490b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerUDIVREM(SDValue Op,
21500b57cec5SDimitry Andric                                            SelectionDAG &DAG) const {
21510b57cec5SDimitry Andric   SDLoc DL(Op);
21520b57cec5SDimitry Andric   EVT VT = Op.getValueType();
21530b57cec5SDimitry Andric 
21540b57cec5SDimitry Andric   if (VT == MVT::i64) {
21550b57cec5SDimitry Andric     SmallVector<SDValue, 2> Results;
21560b57cec5SDimitry Andric     LowerUDIVREM64(Op, DAG, Results);
21570b57cec5SDimitry Andric     return DAG.getMergeValues(Results, DL);
21580b57cec5SDimitry Andric   }
21590b57cec5SDimitry Andric 
21600b57cec5SDimitry Andric   if (VT == MVT::i32) {
21610b57cec5SDimitry Andric     if (SDValue Res = LowerDIVREM24(Op, DAG, false))
21620b57cec5SDimitry Andric       return Res;
21630b57cec5SDimitry Andric   }
21640b57cec5SDimitry Andric 
21655ffd83dbSDimitry Andric   SDValue X = Op.getOperand(0);
21665ffd83dbSDimitry Andric   SDValue Y = Op.getOperand(1);
21670b57cec5SDimitry Andric 
21685ffd83dbSDimitry Andric   // See AMDGPUCodeGenPrepare::expandDivRem32 for a description of the
21695ffd83dbSDimitry Andric   // algorithm used here.
21700b57cec5SDimitry Andric 
21715ffd83dbSDimitry Andric   // Initial estimate of inv(y).
21725ffd83dbSDimitry Andric   SDValue Z = DAG.getNode(AMDGPUISD::URECIP, DL, VT, Y);
21730b57cec5SDimitry Andric 
21745ffd83dbSDimitry Andric   // One round of UNR.
21755ffd83dbSDimitry Andric   SDValue NegY = DAG.getNode(ISD::SUB, DL, VT, DAG.getConstant(0, DL, VT), Y);
21765ffd83dbSDimitry Andric   SDValue NegYZ = DAG.getNode(ISD::MUL, DL, VT, NegY, Z);
21775ffd83dbSDimitry Andric   Z = DAG.getNode(ISD::ADD, DL, VT, Z,
21785ffd83dbSDimitry Andric                   DAG.getNode(ISD::MULHU, DL, VT, Z, NegYZ));
21790b57cec5SDimitry Andric 
21805ffd83dbSDimitry Andric   // Quotient/remainder estimate.
21815ffd83dbSDimitry Andric   SDValue Q = DAG.getNode(ISD::MULHU, DL, VT, X, Z);
21825ffd83dbSDimitry Andric   SDValue R =
21835ffd83dbSDimitry Andric       DAG.getNode(ISD::SUB, DL, VT, X, DAG.getNode(ISD::MUL, DL, VT, Q, Y));
21840b57cec5SDimitry Andric 
21855ffd83dbSDimitry Andric   // First quotient/remainder refinement.
21865ffd83dbSDimitry Andric   EVT CCVT = getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), VT);
21875ffd83dbSDimitry Andric   SDValue One = DAG.getConstant(1, DL, VT);
21885ffd83dbSDimitry Andric   SDValue Cond = DAG.getSetCC(DL, CCVT, R, Y, ISD::SETUGE);
21895ffd83dbSDimitry Andric   Q = DAG.getNode(ISD::SELECT, DL, VT, Cond,
21905ffd83dbSDimitry Andric                   DAG.getNode(ISD::ADD, DL, VT, Q, One), Q);
21915ffd83dbSDimitry Andric   R = DAG.getNode(ISD::SELECT, DL, VT, Cond,
21925ffd83dbSDimitry Andric                   DAG.getNode(ISD::SUB, DL, VT, R, Y), R);
21930b57cec5SDimitry Andric 
21945ffd83dbSDimitry Andric   // Second quotient/remainder refinement.
21955ffd83dbSDimitry Andric   Cond = DAG.getSetCC(DL, CCVT, R, Y, ISD::SETUGE);
21965ffd83dbSDimitry Andric   Q = DAG.getNode(ISD::SELECT, DL, VT, Cond,
21975ffd83dbSDimitry Andric                   DAG.getNode(ISD::ADD, DL, VT, Q, One), Q);
21985ffd83dbSDimitry Andric   R = DAG.getNode(ISD::SELECT, DL, VT, Cond,
21995ffd83dbSDimitry Andric                   DAG.getNode(ISD::SUB, DL, VT, R, Y), R);
22000b57cec5SDimitry Andric 
22015ffd83dbSDimitry Andric   return DAG.getMergeValues({Q, R}, DL);
22020b57cec5SDimitry Andric }
22030b57cec5SDimitry Andric 
22040b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerSDIVREM(SDValue Op,
22050b57cec5SDimitry Andric                                            SelectionDAG &DAG) const {
22060b57cec5SDimitry Andric   SDLoc DL(Op);
22070b57cec5SDimitry Andric   EVT VT = Op.getValueType();
22080b57cec5SDimitry Andric 
22090b57cec5SDimitry Andric   SDValue LHS = Op.getOperand(0);
22100b57cec5SDimitry Andric   SDValue RHS = Op.getOperand(1);
22110b57cec5SDimitry Andric 
22120b57cec5SDimitry Andric   SDValue Zero = DAG.getConstant(0, DL, VT);
22130b57cec5SDimitry Andric   SDValue NegOne = DAG.getConstant(-1, DL, VT);
22140b57cec5SDimitry Andric 
22150b57cec5SDimitry Andric   if (VT == MVT::i32) {
22160b57cec5SDimitry Andric     if (SDValue Res = LowerDIVREM24(Op, DAG, true))
22170b57cec5SDimitry Andric       return Res;
22180b57cec5SDimitry Andric   }
22190b57cec5SDimitry Andric 
22200b57cec5SDimitry Andric   if (VT == MVT::i64 &&
22210b57cec5SDimitry Andric       DAG.ComputeNumSignBits(LHS) > 32 &&
22220b57cec5SDimitry Andric       DAG.ComputeNumSignBits(RHS) > 32) {
22230b57cec5SDimitry Andric     EVT HalfVT = VT.getHalfSizedIntegerVT(*DAG.getContext());
22240b57cec5SDimitry Andric 
22250b57cec5SDimitry Andric     //HiLo split
22260b57cec5SDimitry Andric     SDValue LHS_Lo = DAG.getNode(ISD::EXTRACT_ELEMENT, DL, HalfVT, LHS, Zero);
22270b57cec5SDimitry Andric     SDValue RHS_Lo = DAG.getNode(ISD::EXTRACT_ELEMENT, DL, HalfVT, RHS, Zero);
22280b57cec5SDimitry Andric     SDValue DIVREM = DAG.getNode(ISD::SDIVREM, DL, DAG.getVTList(HalfVT, HalfVT),
22290b57cec5SDimitry Andric                                  LHS_Lo, RHS_Lo);
22300b57cec5SDimitry Andric     SDValue Res[2] = {
22310b57cec5SDimitry Andric       DAG.getNode(ISD::SIGN_EXTEND, DL, VT, DIVREM.getValue(0)),
22320b57cec5SDimitry Andric       DAG.getNode(ISD::SIGN_EXTEND, DL, VT, DIVREM.getValue(1))
22330b57cec5SDimitry Andric     };
22340b57cec5SDimitry Andric     return DAG.getMergeValues(Res, DL);
22350b57cec5SDimitry Andric   }
22360b57cec5SDimitry Andric 
22370b57cec5SDimitry Andric   SDValue LHSign = DAG.getSelectCC(DL, LHS, Zero, NegOne, Zero, ISD::SETLT);
22380b57cec5SDimitry Andric   SDValue RHSign = DAG.getSelectCC(DL, RHS, Zero, NegOne, Zero, ISD::SETLT);
22390b57cec5SDimitry Andric   SDValue DSign = DAG.getNode(ISD::XOR, DL, VT, LHSign, RHSign);
22400b57cec5SDimitry Andric   SDValue RSign = LHSign; // Remainder sign is the same as LHS
22410b57cec5SDimitry Andric 
22420b57cec5SDimitry Andric   LHS = DAG.getNode(ISD::ADD, DL, VT, LHS, LHSign);
22430b57cec5SDimitry Andric   RHS = DAG.getNode(ISD::ADD, DL, VT, RHS, RHSign);
22440b57cec5SDimitry Andric 
22450b57cec5SDimitry Andric   LHS = DAG.getNode(ISD::XOR, DL, VT, LHS, LHSign);
22460b57cec5SDimitry Andric   RHS = DAG.getNode(ISD::XOR, DL, VT, RHS, RHSign);
22470b57cec5SDimitry Andric 
22480b57cec5SDimitry Andric   SDValue Div = DAG.getNode(ISD::UDIVREM, DL, DAG.getVTList(VT, VT), LHS, RHS);
22490b57cec5SDimitry Andric   SDValue Rem = Div.getValue(1);
22500b57cec5SDimitry Andric 
22510b57cec5SDimitry Andric   Div = DAG.getNode(ISD::XOR, DL, VT, Div, DSign);
22520b57cec5SDimitry Andric   Rem = DAG.getNode(ISD::XOR, DL, VT, Rem, RSign);
22530b57cec5SDimitry Andric 
22540b57cec5SDimitry Andric   Div = DAG.getNode(ISD::SUB, DL, VT, Div, DSign);
22550b57cec5SDimitry Andric   Rem = DAG.getNode(ISD::SUB, DL, VT, Rem, RSign);
22560b57cec5SDimitry Andric 
22570b57cec5SDimitry Andric   SDValue Res[2] = {
22580b57cec5SDimitry Andric     Div,
22590b57cec5SDimitry Andric     Rem
22600b57cec5SDimitry Andric   };
22610b57cec5SDimitry Andric   return DAG.getMergeValues(Res, DL);
22620b57cec5SDimitry Andric }
22630b57cec5SDimitry Andric 
2264e8d8bef9SDimitry Andric // (frem x, y) -> (fma (fneg (ftrunc (fdiv x, y))), y, x)
22650b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFREM(SDValue Op, SelectionDAG &DAG) const {
22660b57cec5SDimitry Andric   SDLoc SL(Op);
22670b57cec5SDimitry Andric   EVT VT = Op.getValueType();
2268e8d8bef9SDimitry Andric   auto Flags = Op->getFlags();
22690b57cec5SDimitry Andric   SDValue X = Op.getOperand(0);
22700b57cec5SDimitry Andric   SDValue Y = Op.getOperand(1);
22710b57cec5SDimitry Andric 
2272e8d8bef9SDimitry Andric   SDValue Div = DAG.getNode(ISD::FDIV, SL, VT, X, Y, Flags);
2273e8d8bef9SDimitry Andric   SDValue Trunc = DAG.getNode(ISD::FTRUNC, SL, VT, Div, Flags);
2274e8d8bef9SDimitry Andric   SDValue Neg = DAG.getNode(ISD::FNEG, SL, VT, Trunc, Flags);
2275e8d8bef9SDimitry Andric   // TODO: For f32 use FMAD instead if !hasFastFMA32?
2276e8d8bef9SDimitry Andric   return DAG.getNode(ISD::FMA, SL, VT, Neg, Y, X, Flags);
22770b57cec5SDimitry Andric }
22780b57cec5SDimitry Andric 
22790b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFCEIL(SDValue Op, SelectionDAG &DAG) const {
22800b57cec5SDimitry Andric   SDLoc SL(Op);
22810b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
22820b57cec5SDimitry Andric 
22830b57cec5SDimitry Andric   // result = trunc(src)
22840b57cec5SDimitry Andric   // if (src > 0.0 && src != result)
22850b57cec5SDimitry Andric   //   result += 1.0
22860b57cec5SDimitry Andric 
22870b57cec5SDimitry Andric   SDValue Trunc = DAG.getNode(ISD::FTRUNC, SL, MVT::f64, Src);
22880b57cec5SDimitry Andric 
22890b57cec5SDimitry Andric   const SDValue Zero = DAG.getConstantFP(0.0, SL, MVT::f64);
22900b57cec5SDimitry Andric   const SDValue One = DAG.getConstantFP(1.0, SL, MVT::f64);
22910b57cec5SDimitry Andric 
22920b57cec5SDimitry Andric   EVT SetCCVT =
22930b57cec5SDimitry Andric       getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), MVT::f64);
22940b57cec5SDimitry Andric 
22950b57cec5SDimitry Andric   SDValue Lt0 = DAG.getSetCC(SL, SetCCVT, Src, Zero, ISD::SETOGT);
22960b57cec5SDimitry Andric   SDValue NeTrunc = DAG.getSetCC(SL, SetCCVT, Src, Trunc, ISD::SETONE);
22970b57cec5SDimitry Andric   SDValue And = DAG.getNode(ISD::AND, SL, SetCCVT, Lt0, NeTrunc);
22980b57cec5SDimitry Andric 
22990b57cec5SDimitry Andric   SDValue Add = DAG.getNode(ISD::SELECT, SL, MVT::f64, And, One, Zero);
23000b57cec5SDimitry Andric   // TODO: Should this propagate fast-math-flags?
23010b57cec5SDimitry Andric   return DAG.getNode(ISD::FADD, SL, MVT::f64, Trunc, Add);
23020b57cec5SDimitry Andric }
23030b57cec5SDimitry Andric 
23040b57cec5SDimitry Andric static SDValue extractF64Exponent(SDValue Hi, const SDLoc &SL,
23050b57cec5SDimitry Andric                                   SelectionDAG &DAG) {
23060b57cec5SDimitry Andric   const unsigned FractBits = 52;
23070b57cec5SDimitry Andric   const unsigned ExpBits = 11;
23080b57cec5SDimitry Andric 
23090b57cec5SDimitry Andric   SDValue ExpPart = DAG.getNode(AMDGPUISD::BFE_U32, SL, MVT::i32,
23100b57cec5SDimitry Andric                                 Hi,
23110b57cec5SDimitry Andric                                 DAG.getConstant(FractBits - 32, SL, MVT::i32),
23120b57cec5SDimitry Andric                                 DAG.getConstant(ExpBits, SL, MVT::i32));
23130b57cec5SDimitry Andric   SDValue Exp = DAG.getNode(ISD::SUB, SL, MVT::i32, ExpPart,
23140b57cec5SDimitry Andric                             DAG.getConstant(1023, SL, MVT::i32));
23150b57cec5SDimitry Andric 
23160b57cec5SDimitry Andric   return Exp;
23170b57cec5SDimitry Andric }
23180b57cec5SDimitry Andric 
23190b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFTRUNC(SDValue Op, SelectionDAG &DAG) const {
23200b57cec5SDimitry Andric   SDLoc SL(Op);
23210b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
23220b57cec5SDimitry Andric 
23230b57cec5SDimitry Andric   assert(Op.getValueType() == MVT::f64);
23240b57cec5SDimitry Andric 
23250b57cec5SDimitry Andric   const SDValue Zero = DAG.getConstant(0, SL, MVT::i32);
23260b57cec5SDimitry Andric 
23270b57cec5SDimitry Andric   // Extract the upper half, since this is where we will find the sign and
23280b57cec5SDimitry Andric   // exponent.
2329349cc55cSDimitry Andric   SDValue Hi = getHiHalf64(Src, DAG);
23300b57cec5SDimitry Andric 
23310b57cec5SDimitry Andric   SDValue Exp = extractF64Exponent(Hi, SL, DAG);
23320b57cec5SDimitry Andric 
23330b57cec5SDimitry Andric   const unsigned FractBits = 52;
23340b57cec5SDimitry Andric 
23350b57cec5SDimitry Andric   // Extract the sign bit.
23360b57cec5SDimitry Andric   const SDValue SignBitMask = DAG.getConstant(UINT32_C(1) << 31, SL, MVT::i32);
23370b57cec5SDimitry Andric   SDValue SignBit = DAG.getNode(ISD::AND, SL, MVT::i32, Hi, SignBitMask);
23380b57cec5SDimitry Andric 
23390b57cec5SDimitry Andric   // Extend back to 64-bits.
23400b57cec5SDimitry Andric   SDValue SignBit64 = DAG.getBuildVector(MVT::v2i32, SL, {Zero, SignBit});
23410b57cec5SDimitry Andric   SignBit64 = DAG.getNode(ISD::BITCAST, SL, MVT::i64, SignBit64);
23420b57cec5SDimitry Andric 
23430b57cec5SDimitry Andric   SDValue BcInt = DAG.getNode(ISD::BITCAST, SL, MVT::i64, Src);
23440b57cec5SDimitry Andric   const SDValue FractMask
23450b57cec5SDimitry Andric     = DAG.getConstant((UINT64_C(1) << FractBits) - 1, SL, MVT::i64);
23460b57cec5SDimitry Andric 
23470b57cec5SDimitry Andric   SDValue Shr = DAG.getNode(ISD::SRA, SL, MVT::i64, FractMask, Exp);
23480b57cec5SDimitry Andric   SDValue Not = DAG.getNOT(SL, Shr, MVT::i64);
23490b57cec5SDimitry Andric   SDValue Tmp0 = DAG.getNode(ISD::AND, SL, MVT::i64, BcInt, Not);
23500b57cec5SDimitry Andric 
23510b57cec5SDimitry Andric   EVT SetCCVT =
23520b57cec5SDimitry Andric       getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), MVT::i32);
23530b57cec5SDimitry Andric 
23540b57cec5SDimitry Andric   const SDValue FiftyOne = DAG.getConstant(FractBits - 1, SL, MVT::i32);
23550b57cec5SDimitry Andric 
23560b57cec5SDimitry Andric   SDValue ExpLt0 = DAG.getSetCC(SL, SetCCVT, Exp, Zero, ISD::SETLT);
23570b57cec5SDimitry Andric   SDValue ExpGt51 = DAG.getSetCC(SL, SetCCVT, Exp, FiftyOne, ISD::SETGT);
23580b57cec5SDimitry Andric 
23590b57cec5SDimitry Andric   SDValue Tmp1 = DAG.getNode(ISD::SELECT, SL, MVT::i64, ExpLt0, SignBit64, Tmp0);
23600b57cec5SDimitry Andric   SDValue Tmp2 = DAG.getNode(ISD::SELECT, SL, MVT::i64, ExpGt51, BcInt, Tmp1);
23610b57cec5SDimitry Andric 
23620b57cec5SDimitry Andric   return DAG.getNode(ISD::BITCAST, SL, MVT::f64, Tmp2);
23630b57cec5SDimitry Andric }
23640b57cec5SDimitry Andric 
23650b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFRINT(SDValue Op, SelectionDAG &DAG) const {
23660b57cec5SDimitry Andric   SDLoc SL(Op);
23670b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
23680b57cec5SDimitry Andric 
23690b57cec5SDimitry Andric   assert(Op.getValueType() == MVT::f64);
23700b57cec5SDimitry Andric 
23710b57cec5SDimitry Andric   APFloat C1Val(APFloat::IEEEdouble(), "0x1.0p+52");
23720b57cec5SDimitry Andric   SDValue C1 = DAG.getConstantFP(C1Val, SL, MVT::f64);
23730b57cec5SDimitry Andric   SDValue CopySign = DAG.getNode(ISD::FCOPYSIGN, SL, MVT::f64, C1, Src);
23740b57cec5SDimitry Andric 
23750b57cec5SDimitry Andric   // TODO: Should this propagate fast-math-flags?
23760b57cec5SDimitry Andric 
23770b57cec5SDimitry Andric   SDValue Tmp1 = DAG.getNode(ISD::FADD, SL, MVT::f64, Src, CopySign);
23780b57cec5SDimitry Andric   SDValue Tmp2 = DAG.getNode(ISD::FSUB, SL, MVT::f64, Tmp1, CopySign);
23790b57cec5SDimitry Andric 
23800b57cec5SDimitry Andric   SDValue Fabs = DAG.getNode(ISD::FABS, SL, MVT::f64, Src);
23810b57cec5SDimitry Andric 
23820b57cec5SDimitry Andric   APFloat C2Val(APFloat::IEEEdouble(), "0x1.fffffffffffffp+51");
23830b57cec5SDimitry Andric   SDValue C2 = DAG.getConstantFP(C2Val, SL, MVT::f64);
23840b57cec5SDimitry Andric 
23850b57cec5SDimitry Andric   EVT SetCCVT =
23860b57cec5SDimitry Andric       getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), MVT::f64);
23870b57cec5SDimitry Andric   SDValue Cond = DAG.getSetCC(SL, SetCCVT, Fabs, C2, ISD::SETOGT);
23880b57cec5SDimitry Andric 
23890b57cec5SDimitry Andric   return DAG.getSelect(SL, MVT::f64, Cond, Src, Tmp2);
23900b57cec5SDimitry Andric }
23910b57cec5SDimitry Andric 
23920b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFNEARBYINT(SDValue Op, SelectionDAG &DAG) const {
23930b57cec5SDimitry Andric   // FNEARBYINT and FRINT are the same, except in their handling of FP
23940b57cec5SDimitry Andric   // exceptions. Those aren't really meaningful for us, and OpenCL only has
23950b57cec5SDimitry Andric   // rint, so just treat them as equivalent.
23960b57cec5SDimitry Andric   return DAG.getNode(ISD::FRINT, SDLoc(Op), Op.getValueType(), Op.getOperand(0));
23970b57cec5SDimitry Andric }
23980b57cec5SDimitry Andric 
2399bdd1243dSDimitry Andric SDValue AMDGPUTargetLowering::LowerFROUNDEVEN(SDValue Op,
2400bdd1243dSDimitry Andric                                               SelectionDAG &DAG) const {
2401bdd1243dSDimitry Andric   auto VT = Op.getValueType();
2402bdd1243dSDimitry Andric   auto Arg = Op.getOperand(0u);
2403bdd1243dSDimitry Andric   return DAG.getNode(ISD::FRINT, SDLoc(Op), VT, Arg);
2404bdd1243dSDimitry Andric }
2405bdd1243dSDimitry Andric 
24060b57cec5SDimitry Andric // XXX - May require not supporting f32 denormals?
24070b57cec5SDimitry Andric 
24080b57cec5SDimitry Andric // Don't handle v2f16. The extra instructions to scalarize and repack around the
24090b57cec5SDimitry Andric // compare and vselect end up producing worse code than scalarizing the whole
24100b57cec5SDimitry Andric // operation.
24115ffd83dbSDimitry Andric SDValue AMDGPUTargetLowering::LowerFROUND(SDValue Op, SelectionDAG &DAG) const {
24120b57cec5SDimitry Andric   SDLoc SL(Op);
24130b57cec5SDimitry Andric   SDValue X = Op.getOperand(0);
24140b57cec5SDimitry Andric   EVT VT = Op.getValueType();
24150b57cec5SDimitry Andric 
24160b57cec5SDimitry Andric   SDValue T = DAG.getNode(ISD::FTRUNC, SL, VT, X);
24170b57cec5SDimitry Andric 
24180b57cec5SDimitry Andric   // TODO: Should this propagate fast-math-flags?
24190b57cec5SDimitry Andric 
24200b57cec5SDimitry Andric   SDValue Diff = DAG.getNode(ISD::FSUB, SL, VT, X, T);
24210b57cec5SDimitry Andric 
24220b57cec5SDimitry Andric   SDValue AbsDiff = DAG.getNode(ISD::FABS, SL, VT, Diff);
24230b57cec5SDimitry Andric 
24240b57cec5SDimitry Andric   const SDValue Zero = DAG.getConstantFP(0.0, SL, VT);
24250b57cec5SDimitry Andric   const SDValue One = DAG.getConstantFP(1.0, SL, VT);
24260b57cec5SDimitry Andric   const SDValue Half = DAG.getConstantFP(0.5, SL, VT);
24270b57cec5SDimitry Andric 
24280b57cec5SDimitry Andric   SDValue SignOne = DAG.getNode(ISD::FCOPYSIGN, SL, VT, One, X);
24290b57cec5SDimitry Andric 
24300b57cec5SDimitry Andric   EVT SetCCVT =
24310b57cec5SDimitry Andric       getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), VT);
24320b57cec5SDimitry Andric 
24330b57cec5SDimitry Andric   SDValue Cmp = DAG.getSetCC(SL, SetCCVT, AbsDiff, Half, ISD::SETOGE);
24340b57cec5SDimitry Andric 
24350b57cec5SDimitry Andric   SDValue Sel = DAG.getNode(ISD::SELECT, SL, VT, Cmp, SignOne, Zero);
24360b57cec5SDimitry Andric 
24370b57cec5SDimitry Andric   return DAG.getNode(ISD::FADD, SL, VT, T, Sel);
24380b57cec5SDimitry Andric }
24390b57cec5SDimitry Andric 
24400b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFFLOOR(SDValue Op, SelectionDAG &DAG) const {
24410b57cec5SDimitry Andric   SDLoc SL(Op);
24420b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
24430b57cec5SDimitry Andric 
24440b57cec5SDimitry Andric   // result = trunc(src);
24450b57cec5SDimitry Andric   // if (src < 0.0 && src != result)
24460b57cec5SDimitry Andric   //   result += -1.0.
24470b57cec5SDimitry Andric 
24480b57cec5SDimitry Andric   SDValue Trunc = DAG.getNode(ISD::FTRUNC, SL, MVT::f64, Src);
24490b57cec5SDimitry Andric 
24500b57cec5SDimitry Andric   const SDValue Zero = DAG.getConstantFP(0.0, SL, MVT::f64);
24510b57cec5SDimitry Andric   const SDValue NegOne = DAG.getConstantFP(-1.0, SL, MVT::f64);
24520b57cec5SDimitry Andric 
24530b57cec5SDimitry Andric   EVT SetCCVT =
24540b57cec5SDimitry Andric       getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), MVT::f64);
24550b57cec5SDimitry Andric 
24560b57cec5SDimitry Andric   SDValue Lt0 = DAG.getSetCC(SL, SetCCVT, Src, Zero, ISD::SETOLT);
24570b57cec5SDimitry Andric   SDValue NeTrunc = DAG.getSetCC(SL, SetCCVT, Src, Trunc, ISD::SETONE);
24580b57cec5SDimitry Andric   SDValue And = DAG.getNode(ISD::AND, SL, SetCCVT, Lt0, NeTrunc);
24590b57cec5SDimitry Andric 
24600b57cec5SDimitry Andric   SDValue Add = DAG.getNode(ISD::SELECT, SL, MVT::f64, And, NegOne, Zero);
24610b57cec5SDimitry Andric   // TODO: Should this propagate fast-math-flags?
24620b57cec5SDimitry Andric   return DAG.getNode(ISD::FADD, SL, MVT::f64, Trunc, Add);
24630b57cec5SDimitry Andric }
24640b57cec5SDimitry Andric 
246506c3fb27SDimitry Andric /// Return true if it's known that \p Src can never be an f32 denormal value.
246606c3fb27SDimitry Andric static bool valueIsKnownNeverF32Denorm(SDValue Src) {
246706c3fb27SDimitry Andric   switch (Src.getOpcode()) {
246806c3fb27SDimitry Andric   case ISD::FP_EXTEND:
246906c3fb27SDimitry Andric     return Src.getOperand(0).getValueType() == MVT::f16;
247006c3fb27SDimitry Andric   case ISD::FP16_TO_FP:
247106c3fb27SDimitry Andric     return true;
247206c3fb27SDimitry Andric   default:
247306c3fb27SDimitry Andric     return false;
24740b57cec5SDimitry Andric   }
24750b57cec5SDimitry Andric 
247606c3fb27SDimitry Andric   llvm_unreachable("covered opcode switch");
247706c3fb27SDimitry Andric }
247806c3fb27SDimitry Andric 
247906c3fb27SDimitry Andric static bool allowApproxFunc(const SelectionDAG &DAG, SDNodeFlags Flags) {
248006c3fb27SDimitry Andric   if (Flags.hasApproximateFuncs())
248106c3fb27SDimitry Andric     return true;
248206c3fb27SDimitry Andric   auto &Options = DAG.getTarget().Options;
248306c3fb27SDimitry Andric   return Options.UnsafeFPMath || Options.ApproxFuncFPMath;
248406c3fb27SDimitry Andric }
248506c3fb27SDimitry Andric 
248606c3fb27SDimitry Andric static bool needsDenormHandlingF32(const SelectionDAG &DAG, SDValue Src,
248706c3fb27SDimitry Andric                                    SDNodeFlags Flags) {
248806c3fb27SDimitry Andric   return !valueIsKnownNeverF32Denorm(Src) &&
248906c3fb27SDimitry Andric          DAG.getMachineFunction()
249006c3fb27SDimitry Andric                  .getDenormalMode(APFloat::IEEEsingle())
249106c3fb27SDimitry Andric                  .Input != DenormalMode::PreserveSign;
249206c3fb27SDimitry Andric }
249306c3fb27SDimitry Andric 
249406c3fb27SDimitry Andric SDValue AMDGPUTargetLowering::getIsLtSmallestNormal(SelectionDAG &DAG,
249506c3fb27SDimitry Andric                                                     SDValue Src,
249606c3fb27SDimitry Andric                                                     SDNodeFlags Flags) const {
249706c3fb27SDimitry Andric   SDLoc SL(Src);
249806c3fb27SDimitry Andric   EVT VT = Src.getValueType();
249906c3fb27SDimitry Andric   const fltSemantics &Semantics = SelectionDAG::EVTToAPFloatSemantics(VT);
250006c3fb27SDimitry Andric   SDValue SmallestNormal =
250106c3fb27SDimitry Andric       DAG.getConstantFP(APFloat::getSmallestNormalized(Semantics), SL, VT);
250206c3fb27SDimitry Andric 
250306c3fb27SDimitry Andric   // Want to scale denormals up, but negatives and 0 work just as well on the
250406c3fb27SDimitry Andric   // scaled path.
250506c3fb27SDimitry Andric   SDValue IsLtSmallestNormal = DAG.getSetCC(
250606c3fb27SDimitry Andric       SL, getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), VT), Src,
250706c3fb27SDimitry Andric       SmallestNormal, ISD::SETOLT);
250806c3fb27SDimitry Andric 
250906c3fb27SDimitry Andric   return IsLtSmallestNormal;
251006c3fb27SDimitry Andric }
251106c3fb27SDimitry Andric 
251206c3fb27SDimitry Andric SDValue AMDGPUTargetLowering::getIsFinite(SelectionDAG &DAG, SDValue Src,
251306c3fb27SDimitry Andric                                           SDNodeFlags Flags) const {
251406c3fb27SDimitry Andric   SDLoc SL(Src);
251506c3fb27SDimitry Andric   EVT VT = Src.getValueType();
251606c3fb27SDimitry Andric   const fltSemantics &Semantics = SelectionDAG::EVTToAPFloatSemantics(VT);
251706c3fb27SDimitry Andric   SDValue Inf = DAG.getConstantFP(APFloat::getInf(Semantics), SL, VT);
251806c3fb27SDimitry Andric 
251906c3fb27SDimitry Andric   SDValue Fabs = DAG.getNode(ISD::FABS, SL, VT, Src, Flags);
252006c3fb27SDimitry Andric   SDValue IsFinite = DAG.getSetCC(
252106c3fb27SDimitry Andric       SL, getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), VT), Fabs,
252206c3fb27SDimitry Andric       Inf, ISD::SETOLT);
252306c3fb27SDimitry Andric   return IsFinite;
252406c3fb27SDimitry Andric }
252506c3fb27SDimitry Andric 
252606c3fb27SDimitry Andric /// If denormal handling is required return the scaled input to FLOG2, and the
252706c3fb27SDimitry Andric /// check for denormal range. Otherwise, return null values.
252806c3fb27SDimitry Andric std::pair<SDValue, SDValue>
252906c3fb27SDimitry Andric AMDGPUTargetLowering::getScaledLogInput(SelectionDAG &DAG, const SDLoc SL,
253006c3fb27SDimitry Andric                                         SDValue Src, SDNodeFlags Flags) const {
2531*8a4dda33SDimitry Andric   if (!needsDenormHandlingF32(DAG, Src, Flags))
253206c3fb27SDimitry Andric     return {};
253306c3fb27SDimitry Andric 
253406c3fb27SDimitry Andric   MVT VT = MVT::f32;
253506c3fb27SDimitry Andric   const fltSemantics &Semantics = APFloat::IEEEsingle();
253606c3fb27SDimitry Andric   SDValue SmallestNormal =
253706c3fb27SDimitry Andric       DAG.getConstantFP(APFloat::getSmallestNormalized(Semantics), SL, VT);
253806c3fb27SDimitry Andric 
253906c3fb27SDimitry Andric   SDValue IsLtSmallestNormal = DAG.getSetCC(
254006c3fb27SDimitry Andric       SL, getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), VT), Src,
254106c3fb27SDimitry Andric       SmallestNormal, ISD::SETOLT);
254206c3fb27SDimitry Andric 
254306c3fb27SDimitry Andric   SDValue Scale32 = DAG.getConstantFP(0x1.0p+32, SL, VT);
254406c3fb27SDimitry Andric   SDValue One = DAG.getConstantFP(1.0, SL, VT);
254506c3fb27SDimitry Andric   SDValue ScaleFactor =
254606c3fb27SDimitry Andric       DAG.getNode(ISD::SELECT, SL, VT, IsLtSmallestNormal, Scale32, One, Flags);
254706c3fb27SDimitry Andric 
254806c3fb27SDimitry Andric   SDValue ScaledInput = DAG.getNode(ISD::FMUL, SL, VT, Src, ScaleFactor, Flags);
254906c3fb27SDimitry Andric   return {ScaledInput, IsLtSmallestNormal};
255006c3fb27SDimitry Andric }
255106c3fb27SDimitry Andric 
255206c3fb27SDimitry Andric SDValue AMDGPUTargetLowering::LowerFLOG2(SDValue Op, SelectionDAG &DAG) const {
255306c3fb27SDimitry Andric   // v_log_f32 is good enough for OpenCL, except it doesn't handle denormals.
255406c3fb27SDimitry Andric   // If we have to handle denormals, scale up the input and adjust the result.
255506c3fb27SDimitry Andric 
255606c3fb27SDimitry Andric   // scaled = x * (is_denormal ? 0x1.0p+32 : 1.0)
255706c3fb27SDimitry Andric   // log2 = amdgpu_log2 - (is_denormal ? 32.0 : 0.0)
255806c3fb27SDimitry Andric 
255906c3fb27SDimitry Andric   SDLoc SL(Op);
256006c3fb27SDimitry Andric   EVT VT = Op.getValueType();
256106c3fb27SDimitry Andric   SDValue Src = Op.getOperand(0);
256206c3fb27SDimitry Andric   SDNodeFlags Flags = Op->getFlags();
256306c3fb27SDimitry Andric 
256406c3fb27SDimitry Andric   if (VT == MVT::f16) {
256506c3fb27SDimitry Andric     // Nothing in half is a denormal when promoted to f32.
256606c3fb27SDimitry Andric     assert(!Subtarget->has16BitInsts());
256706c3fb27SDimitry Andric     SDValue Ext = DAG.getNode(ISD::FP_EXTEND, SL, MVT::f32, Src, Flags);
256806c3fb27SDimitry Andric     SDValue Log = DAG.getNode(AMDGPUISD::LOG, SL, MVT::f32, Ext, Flags);
256906c3fb27SDimitry Andric     return DAG.getNode(ISD::FP_ROUND, SL, VT, Log,
257006c3fb27SDimitry Andric                        DAG.getTargetConstant(0, SL, MVT::i32), Flags);
257106c3fb27SDimitry Andric   }
257206c3fb27SDimitry Andric 
257306c3fb27SDimitry Andric   auto [ScaledInput, IsLtSmallestNormal] =
257406c3fb27SDimitry Andric       getScaledLogInput(DAG, SL, Src, Flags);
257506c3fb27SDimitry Andric   if (!ScaledInput)
257606c3fb27SDimitry Andric     return DAG.getNode(AMDGPUISD::LOG, SL, VT, Src, Flags);
257706c3fb27SDimitry Andric 
257806c3fb27SDimitry Andric   SDValue Log2 = DAG.getNode(AMDGPUISD::LOG, SL, VT, ScaledInput, Flags);
257906c3fb27SDimitry Andric 
258006c3fb27SDimitry Andric   SDValue ThirtyTwo = DAG.getConstantFP(32.0, SL, VT);
258106c3fb27SDimitry Andric   SDValue Zero = DAG.getConstantFP(0.0, SL, VT);
258206c3fb27SDimitry Andric   SDValue ResultOffset =
258306c3fb27SDimitry Andric       DAG.getNode(ISD::SELECT, SL, VT, IsLtSmallestNormal, ThirtyTwo, Zero);
258406c3fb27SDimitry Andric   return DAG.getNode(ISD::FSUB, SL, VT, Log2, ResultOffset, Flags);
258506c3fb27SDimitry Andric }
258606c3fb27SDimitry Andric 
258706c3fb27SDimitry Andric static SDValue getMad(SelectionDAG &DAG, const SDLoc &SL, EVT VT, SDValue X,
258806c3fb27SDimitry Andric                       SDValue Y, SDValue C, SDNodeFlags Flags = SDNodeFlags()) {
258906c3fb27SDimitry Andric   SDValue Mul = DAG.getNode(ISD::FMUL, SL, VT, X, Y, Flags);
259006c3fb27SDimitry Andric   return DAG.getNode(ISD::FADD, SL, VT, Mul, C, Flags);
259106c3fb27SDimitry Andric }
259206c3fb27SDimitry Andric 
259306c3fb27SDimitry Andric SDValue AMDGPUTargetLowering::LowerFLOGCommon(SDValue Op,
259406c3fb27SDimitry Andric                                               SelectionDAG &DAG) const {
259506c3fb27SDimitry Andric   SDValue X = Op.getOperand(0);
259606c3fb27SDimitry Andric   EVT VT = Op.getValueType();
259706c3fb27SDimitry Andric   SDNodeFlags Flags = Op->getFlags();
259806c3fb27SDimitry Andric   SDLoc DL(Op);
259906c3fb27SDimitry Andric 
260006c3fb27SDimitry Andric   const bool IsLog10 = Op.getOpcode() == ISD::FLOG10;
260106c3fb27SDimitry Andric   assert(IsLog10 || Op.getOpcode() == ISD::FLOG);
260206c3fb27SDimitry Andric 
260306c3fb27SDimitry Andric   const auto &Options = getTargetMachine().Options;
260406c3fb27SDimitry Andric   if (VT == MVT::f16 || Flags.hasApproximateFuncs() ||
260506c3fb27SDimitry Andric       Options.ApproxFuncFPMath || Options.UnsafeFPMath) {
260606c3fb27SDimitry Andric 
260706c3fb27SDimitry Andric     if (VT == MVT::f16 && !Subtarget->has16BitInsts()) {
260806c3fb27SDimitry Andric       // Log and multiply in f32 is good enough for f16.
260906c3fb27SDimitry Andric       X = DAG.getNode(ISD::FP_EXTEND, DL, MVT::f32, X, Flags);
261006c3fb27SDimitry Andric     }
261106c3fb27SDimitry Andric 
2612*8a4dda33SDimitry Andric     SDValue Lowered = LowerFLOGUnsafe(X, DL, DAG, IsLog10, Flags);
261306c3fb27SDimitry Andric     if (VT == MVT::f16 && !Subtarget->has16BitInsts()) {
261406c3fb27SDimitry Andric       return DAG.getNode(ISD::FP_ROUND, DL, VT, Lowered,
261506c3fb27SDimitry Andric                          DAG.getTargetConstant(0, DL, MVT::i32), Flags);
261606c3fb27SDimitry Andric     }
261706c3fb27SDimitry Andric 
261806c3fb27SDimitry Andric     return Lowered;
261906c3fb27SDimitry Andric   }
262006c3fb27SDimitry Andric 
262106c3fb27SDimitry Andric   auto [ScaledInput, IsScaled] = getScaledLogInput(DAG, DL, X, Flags);
262206c3fb27SDimitry Andric   if (ScaledInput)
262306c3fb27SDimitry Andric     X = ScaledInput;
262406c3fb27SDimitry Andric 
262506c3fb27SDimitry Andric   SDValue Y = DAG.getNode(AMDGPUISD::LOG, DL, VT, X, Flags);
262606c3fb27SDimitry Andric 
262706c3fb27SDimitry Andric   SDValue R;
262806c3fb27SDimitry Andric   if (Subtarget->hasFastFMAF32()) {
262906c3fb27SDimitry Andric     // c+cc are ln(2)/ln(10) to more than 49 bits
263006c3fb27SDimitry Andric     const float c_log10 = 0x1.344134p-2f;
263106c3fb27SDimitry Andric     const float cc_log10 = 0x1.09f79ep-26f;
263206c3fb27SDimitry Andric 
263306c3fb27SDimitry Andric     // c + cc is ln(2) to more than 49 bits
263406c3fb27SDimitry Andric     const float c_log = 0x1.62e42ep-1f;
263506c3fb27SDimitry Andric     const float cc_log = 0x1.efa39ep-25f;
263606c3fb27SDimitry Andric 
263706c3fb27SDimitry Andric     SDValue C = DAG.getConstantFP(IsLog10 ? c_log10 : c_log, DL, VT);
263806c3fb27SDimitry Andric     SDValue CC = DAG.getConstantFP(IsLog10 ? cc_log10 : cc_log, DL, VT);
263906c3fb27SDimitry Andric 
264006c3fb27SDimitry Andric     R = DAG.getNode(ISD::FMUL, DL, VT, Y, C, Flags);
264106c3fb27SDimitry Andric     SDValue NegR = DAG.getNode(ISD::FNEG, DL, VT, R, Flags);
264206c3fb27SDimitry Andric     SDValue FMA0 = DAG.getNode(ISD::FMA, DL, VT, Y, C, NegR, Flags);
264306c3fb27SDimitry Andric     SDValue FMA1 = DAG.getNode(ISD::FMA, DL, VT, Y, CC, FMA0, Flags);
264406c3fb27SDimitry Andric     R = DAG.getNode(ISD::FADD, DL, VT, R, FMA1, Flags);
264506c3fb27SDimitry Andric   } else {
264606c3fb27SDimitry Andric     // ch+ct is ln(2)/ln(10) to more than 36 bits
264706c3fb27SDimitry Andric     const float ch_log10 = 0x1.344000p-2f;
264806c3fb27SDimitry Andric     const float ct_log10 = 0x1.3509f6p-18f;
264906c3fb27SDimitry Andric 
265006c3fb27SDimitry Andric     // ch + ct is ln(2) to more than 36 bits
265106c3fb27SDimitry Andric     const float ch_log = 0x1.62e000p-1f;
265206c3fb27SDimitry Andric     const float ct_log = 0x1.0bfbe8p-15f;
265306c3fb27SDimitry Andric 
265406c3fb27SDimitry Andric     SDValue CH = DAG.getConstantFP(IsLog10 ? ch_log10 : ch_log, DL, VT);
265506c3fb27SDimitry Andric     SDValue CT = DAG.getConstantFP(IsLog10 ? ct_log10 : ct_log, DL, VT);
265606c3fb27SDimitry Andric 
265706c3fb27SDimitry Andric     SDValue YAsInt = DAG.getNode(ISD::BITCAST, DL, MVT::i32, Y);
265806c3fb27SDimitry Andric     SDValue MaskConst = DAG.getConstant(0xfffff000, DL, MVT::i32);
265906c3fb27SDimitry Andric     SDValue YHInt = DAG.getNode(ISD::AND, DL, MVT::i32, YAsInt, MaskConst);
266006c3fb27SDimitry Andric     SDValue YH = DAG.getNode(ISD::BITCAST, DL, MVT::f32, YHInt);
266106c3fb27SDimitry Andric     SDValue YT = DAG.getNode(ISD::FSUB, DL, VT, Y, YH, Flags);
266206c3fb27SDimitry Andric 
266306c3fb27SDimitry Andric     SDValue YTCT = DAG.getNode(ISD::FMUL, DL, VT, YT, CT, Flags);
266406c3fb27SDimitry Andric     SDValue Mad0 = getMad(DAG, DL, VT, YH, CT, YTCT, Flags);
266506c3fb27SDimitry Andric     SDValue Mad1 = getMad(DAG, DL, VT, YT, CH, Mad0, Flags);
266606c3fb27SDimitry Andric     R = getMad(DAG, DL, VT, YH, CH, Mad1);
266706c3fb27SDimitry Andric   }
266806c3fb27SDimitry Andric 
266906c3fb27SDimitry Andric   const bool IsFiniteOnly = (Flags.hasNoNaNs() || Options.NoNaNsFPMath) &&
267006c3fb27SDimitry Andric                             (Flags.hasNoInfs() || Options.NoInfsFPMath);
267106c3fb27SDimitry Andric 
267206c3fb27SDimitry Andric   // TODO: Check if known finite from source value.
267306c3fb27SDimitry Andric   if (!IsFiniteOnly) {
267406c3fb27SDimitry Andric     SDValue IsFinite = getIsFinite(DAG, Y, Flags);
267506c3fb27SDimitry Andric     R = DAG.getNode(ISD::SELECT, DL, VT, IsFinite, R, Y, Flags);
267606c3fb27SDimitry Andric   }
267706c3fb27SDimitry Andric 
267806c3fb27SDimitry Andric   if (IsScaled) {
267906c3fb27SDimitry Andric     SDValue Zero = DAG.getConstantFP(0.0f, DL, VT);
268006c3fb27SDimitry Andric     SDValue ShiftK =
268106c3fb27SDimitry Andric         DAG.getConstantFP(IsLog10 ? 0x1.344136p+3f : 0x1.62e430p+4f, DL, VT);
268206c3fb27SDimitry Andric     SDValue Shift =
268306c3fb27SDimitry Andric         DAG.getNode(ISD::SELECT, DL, VT, IsScaled, ShiftK, Zero, Flags);
268406c3fb27SDimitry Andric     R = DAG.getNode(ISD::FSUB, DL, VT, R, Shift, Flags);
268506c3fb27SDimitry Andric   }
268606c3fb27SDimitry Andric 
268706c3fb27SDimitry Andric   return R;
268806c3fb27SDimitry Andric }
268906c3fb27SDimitry Andric 
269006c3fb27SDimitry Andric SDValue AMDGPUTargetLowering::LowerFLOG10(SDValue Op, SelectionDAG &DAG) const {
269106c3fb27SDimitry Andric   return LowerFLOGCommon(Op, DAG);
269206c3fb27SDimitry Andric }
269306c3fb27SDimitry Andric 
269406c3fb27SDimitry Andric // Do f32 fast math expansion for flog2 or flog10. This is accurate enough for a
269506c3fb27SDimitry Andric // promote f16 operation.
269606c3fb27SDimitry Andric SDValue AMDGPUTargetLowering::LowerFLOGUnsafe(SDValue Src, const SDLoc &SL,
2697*8a4dda33SDimitry Andric                                               SelectionDAG &DAG, bool IsLog10,
269806c3fb27SDimitry Andric                                               SDNodeFlags Flags) const {
269906c3fb27SDimitry Andric   EVT VT = Src.getValueType();
270006c3fb27SDimitry Andric   unsigned LogOp = VT == MVT::f32 ? AMDGPUISD::LOG : ISD::FLOG2;
2701*8a4dda33SDimitry Andric 
2702*8a4dda33SDimitry Andric   double Log2BaseInverted =
2703*8a4dda33SDimitry Andric       IsLog10 ? numbers::ln2 / numbers::ln10 : numbers::ln2;
2704*8a4dda33SDimitry Andric 
2705*8a4dda33SDimitry Andric   if (VT == MVT::f32) {
2706*8a4dda33SDimitry Andric     auto [ScaledInput, IsScaled] = getScaledLogInput(DAG, SL, Src, Flags);
2707*8a4dda33SDimitry Andric     if (ScaledInput) {
2708*8a4dda33SDimitry Andric       SDValue LogSrc = DAG.getNode(AMDGPUISD::LOG, SL, VT, ScaledInput, Flags);
2709*8a4dda33SDimitry Andric       SDValue ScaledResultOffset =
2710*8a4dda33SDimitry Andric           DAG.getConstantFP(-32.0 * Log2BaseInverted, SL, VT);
2711*8a4dda33SDimitry Andric 
2712*8a4dda33SDimitry Andric       SDValue Zero = DAG.getConstantFP(0.0f, SL, VT);
2713*8a4dda33SDimitry Andric 
2714*8a4dda33SDimitry Andric       SDValue ResultOffset = DAG.getNode(ISD::SELECT, SL, VT, IsScaled,
2715*8a4dda33SDimitry Andric                                          ScaledResultOffset, Zero, Flags);
2716*8a4dda33SDimitry Andric 
2717*8a4dda33SDimitry Andric       SDValue Log2Inv = DAG.getConstantFP(Log2BaseInverted, SL, VT);
2718*8a4dda33SDimitry Andric 
2719*8a4dda33SDimitry Andric       if (Subtarget->hasFastFMAF32())
2720*8a4dda33SDimitry Andric         return DAG.getNode(ISD::FMA, SL, VT, LogSrc, Log2Inv, ResultOffset,
2721*8a4dda33SDimitry Andric                            Flags);
2722*8a4dda33SDimitry Andric       SDValue Mul = DAG.getNode(ISD::FMUL, SL, VT, LogSrc, Log2Inv, Flags);
2723*8a4dda33SDimitry Andric       return DAG.getNode(ISD::FADD, SL, VT, Mul, ResultOffset);
2724*8a4dda33SDimitry Andric     }
2725*8a4dda33SDimitry Andric   }
2726*8a4dda33SDimitry Andric 
272706c3fb27SDimitry Andric   SDValue Log2Operand = DAG.getNode(LogOp, SL, VT, Src, Flags);
272806c3fb27SDimitry Andric   SDValue Log2BaseInvertedOperand = DAG.getConstantFP(Log2BaseInverted, SL, VT);
272906c3fb27SDimitry Andric 
273006c3fb27SDimitry Andric   return DAG.getNode(ISD::FMUL, SL, VT, Log2Operand, Log2BaseInvertedOperand,
273106c3fb27SDimitry Andric                      Flags);
273206c3fb27SDimitry Andric }
273306c3fb27SDimitry Andric 
273406c3fb27SDimitry Andric SDValue AMDGPUTargetLowering::lowerFEXP2(SDValue Op, SelectionDAG &DAG) const {
273506c3fb27SDimitry Andric   // v_exp_f32 is good enough for OpenCL, except it doesn't handle denormals.
273606c3fb27SDimitry Andric   // If we have to handle denormals, scale up the input and adjust the result.
273706c3fb27SDimitry Andric 
273806c3fb27SDimitry Andric   SDLoc SL(Op);
273906c3fb27SDimitry Andric   EVT VT = Op.getValueType();
274006c3fb27SDimitry Andric   SDValue Src = Op.getOperand(0);
274106c3fb27SDimitry Andric   SDNodeFlags Flags = Op->getFlags();
274206c3fb27SDimitry Andric 
274306c3fb27SDimitry Andric   if (VT == MVT::f16) {
274406c3fb27SDimitry Andric     // Nothing in half is a denormal when promoted to f32.
274506c3fb27SDimitry Andric     assert(!Subtarget->has16BitInsts());
274606c3fb27SDimitry Andric     SDValue Ext = DAG.getNode(ISD::FP_EXTEND, SL, MVT::f32, Src, Flags);
274706c3fb27SDimitry Andric     SDValue Log = DAG.getNode(AMDGPUISD::EXP, SL, MVT::f32, Ext, Flags);
274806c3fb27SDimitry Andric     return DAG.getNode(ISD::FP_ROUND, SL, VT, Log,
274906c3fb27SDimitry Andric                        DAG.getTargetConstant(0, SL, MVT::i32), Flags);
275006c3fb27SDimitry Andric   }
275106c3fb27SDimitry Andric 
275206c3fb27SDimitry Andric   assert(VT == MVT::f32);
275306c3fb27SDimitry Andric 
2754*8a4dda33SDimitry Andric   if (!needsDenormHandlingF32(DAG, Src, Flags))
275506c3fb27SDimitry Andric     return DAG.getNode(AMDGPUISD::EXP, SL, MVT::f32, Src, Flags);
275606c3fb27SDimitry Andric 
275706c3fb27SDimitry Andric   // bool needs_scaling = x < -0x1.f80000p+6f;
275806c3fb27SDimitry Andric   // v_exp_f32(x + (s ? 0x1.0p+6f : 0.0f)) * (s ? 0x1.0p-64f : 1.0f);
275906c3fb27SDimitry Andric 
276006c3fb27SDimitry Andric   // -nextafter(128.0, -1)
276106c3fb27SDimitry Andric   SDValue RangeCheckConst = DAG.getConstantFP(-0x1.f80000p+6f, SL, VT);
276206c3fb27SDimitry Andric 
276306c3fb27SDimitry Andric   EVT SetCCVT = getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), VT);
276406c3fb27SDimitry Andric 
276506c3fb27SDimitry Andric   SDValue NeedsScaling =
276606c3fb27SDimitry Andric       DAG.getSetCC(SL, SetCCVT, Src, RangeCheckConst, ISD::SETOLT);
276706c3fb27SDimitry Andric 
276806c3fb27SDimitry Andric   SDValue SixtyFour = DAG.getConstantFP(0x1.0p+6f, SL, VT);
276906c3fb27SDimitry Andric   SDValue Zero = DAG.getConstantFP(0.0, SL, VT);
277006c3fb27SDimitry Andric 
277106c3fb27SDimitry Andric   SDValue AddOffset =
277206c3fb27SDimitry Andric       DAG.getNode(ISD::SELECT, SL, VT, NeedsScaling, SixtyFour, Zero);
277306c3fb27SDimitry Andric 
277406c3fb27SDimitry Andric   SDValue AddInput = DAG.getNode(ISD::FADD, SL, VT, Src, AddOffset, Flags);
277506c3fb27SDimitry Andric   SDValue Exp2 = DAG.getNode(AMDGPUISD::EXP, SL, VT, AddInput, Flags);
277606c3fb27SDimitry Andric 
277706c3fb27SDimitry Andric   SDValue TwoExpNeg64 = DAG.getConstantFP(0x1.0p-64f, SL, VT);
277806c3fb27SDimitry Andric   SDValue One = DAG.getConstantFP(1.0, SL, VT);
277906c3fb27SDimitry Andric   SDValue ResultScale =
278006c3fb27SDimitry Andric       DAG.getNode(ISD::SELECT, SL, VT, NeedsScaling, TwoExpNeg64, One);
278106c3fb27SDimitry Andric 
278206c3fb27SDimitry Andric   return DAG.getNode(ISD::FMUL, SL, VT, Exp2, ResultScale, Flags);
278306c3fb27SDimitry Andric }
278406c3fb27SDimitry Andric 
278506c3fb27SDimitry Andric SDValue AMDGPUTargetLowering::lowerFEXPUnsafe(SDValue Op, const SDLoc &SL,
278606c3fb27SDimitry Andric                                               SelectionDAG &DAG,
278706c3fb27SDimitry Andric                                               SDNodeFlags Flags) const {
27880b57cec5SDimitry Andric   // exp2(M_LOG2E_F * f);
278906c3fb27SDimitry Andric   EVT VT = Op.getValueType();
279006c3fb27SDimitry Andric   const SDValue K = DAG.getConstantFP(numbers::log2e, SL, VT);
279106c3fb27SDimitry Andric   SDValue Mul = DAG.getNode(ISD::FMUL, SL, VT, Op, K, Flags);
279206c3fb27SDimitry Andric   return DAG.getNode(VT == MVT::f32 ? AMDGPUISD::EXP : ISD::FEXP2, SL, VT, Mul,
279306c3fb27SDimitry Andric                      Flags);
279406c3fb27SDimitry Andric }
279506c3fb27SDimitry Andric 
27960b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::lowerFEXP(SDValue Op, SelectionDAG &DAG) const {
27970b57cec5SDimitry Andric   EVT VT = Op.getValueType();
27980b57cec5SDimitry Andric   SDLoc SL(Op);
279906c3fb27SDimitry Andric   SDValue X = Op.getOperand(0);
280006c3fb27SDimitry Andric   SDNodeFlags Flags = Op->getFlags();
280106c3fb27SDimitry Andric   const bool IsExp10 = false; // TODO: For some reason exp10 is missing
28020b57cec5SDimitry Andric 
280306c3fb27SDimitry Andric   if (VT.getScalarType() == MVT::f16) {
280406c3fb27SDimitry Andric     // v_exp_f16 (fmul x, log2e)
280506c3fb27SDimitry Andric     if (allowApproxFunc(DAG, Flags)) // TODO: Does this really require fast?
280606c3fb27SDimitry Andric       return lowerFEXPUnsafe(X, SL, DAG, Flags);
280706c3fb27SDimitry Andric 
280806c3fb27SDimitry Andric     if (VT.isVector())
280906c3fb27SDimitry Andric       return SDValue();
281006c3fb27SDimitry Andric 
281106c3fb27SDimitry Andric     // exp(f16 x) ->
281206c3fb27SDimitry Andric     //   fptrunc (v_exp_f32 (fmul (fpext x), log2e))
281306c3fb27SDimitry Andric 
281406c3fb27SDimitry Andric     // Nothing in half is a denormal when promoted to f32.
281506c3fb27SDimitry Andric     SDValue Ext = DAG.getNode(ISD::FP_EXTEND, SL, MVT::f32, X, Flags);
281606c3fb27SDimitry Andric     SDValue Lowered = lowerFEXPUnsafe(Ext, SL, DAG, Flags);
281706c3fb27SDimitry Andric     return DAG.getNode(ISD::FP_ROUND, SL, VT, Lowered,
281806c3fb27SDimitry Andric                        DAG.getTargetConstant(0, SL, MVT::i32), Flags);
281906c3fb27SDimitry Andric   }
282006c3fb27SDimitry Andric 
282106c3fb27SDimitry Andric   assert(VT == MVT::f32);
282206c3fb27SDimitry Andric 
282306c3fb27SDimitry Andric   // TODO: Interpret allowApproxFunc as ignoring DAZ. This is currently copying
282406c3fb27SDimitry Andric   // library behavior. Also, is known-not-daz source sufficient?
282506c3fb27SDimitry Andric   if (allowApproxFunc(DAG, Flags) && !needsDenormHandlingF32(DAG, X, Flags)) {
282606c3fb27SDimitry Andric     assert(!IsExp10 && "todo exp10 support");
282706c3fb27SDimitry Andric     return lowerFEXPUnsafe(X, SL, DAG, Flags);
282806c3fb27SDimitry Andric   }
282906c3fb27SDimitry Andric 
283006c3fb27SDimitry Andric   //    Algorithm:
283106c3fb27SDimitry Andric   //
283206c3fb27SDimitry Andric   //    e^x = 2^(x/ln(2)) = 2^(x*(64/ln(2))/64)
283306c3fb27SDimitry Andric   //
283406c3fb27SDimitry Andric   //    x*(64/ln(2)) = n + f, |f| <= 0.5, n is integer
283506c3fb27SDimitry Andric   //    n = 64*m + j,   0 <= j < 64
283606c3fb27SDimitry Andric   //
283706c3fb27SDimitry Andric   //    e^x = 2^((64*m + j + f)/64)
283806c3fb27SDimitry Andric   //        = (2^m) * (2^(j/64)) * 2^(f/64)
283906c3fb27SDimitry Andric   //        = (2^m) * (2^(j/64)) * e^(f*(ln(2)/64))
284006c3fb27SDimitry Andric   //
284106c3fb27SDimitry Andric   //    f = x*(64/ln(2)) - n
284206c3fb27SDimitry Andric   //    r = f*(ln(2)/64) = x - n*(ln(2)/64)
284306c3fb27SDimitry Andric   //
284406c3fb27SDimitry Andric   //    e^x = (2^m) * (2^(j/64)) * e^r
284506c3fb27SDimitry Andric   //
284606c3fb27SDimitry Andric   //    (2^(j/64)) is precomputed
284706c3fb27SDimitry Andric   //
284806c3fb27SDimitry Andric   //    e^r = 1 + r + (r^2)/2! + (r^3)/3! + (r^4)/4! + (r^5)/5!
284906c3fb27SDimitry Andric   //    e^r = 1 + q
285006c3fb27SDimitry Andric   //
285106c3fb27SDimitry Andric   //    q = r + (r^2)/2! + (r^3)/3! + (r^4)/4! + (r^5)/5!
285206c3fb27SDimitry Andric   //
285306c3fb27SDimitry Andric   //    e^x = (2^m) * ( (2^(j/64)) + q*(2^(j/64)) )
285406c3fb27SDimitry Andric   SDNodeFlags FlagsNoContract = Flags;
285506c3fb27SDimitry Andric   FlagsNoContract.setAllowContract(false);
285606c3fb27SDimitry Andric 
285706c3fb27SDimitry Andric   SDValue PH, PL;
285806c3fb27SDimitry Andric   if (Subtarget->hasFastFMAF32()) {
285906c3fb27SDimitry Andric     const float c_exp = numbers::log2ef;
286006c3fb27SDimitry Andric     const float cc_exp = 0x1.4ae0bep-26f; // c+cc are 49 bits
286106c3fb27SDimitry Andric     const float c_exp10 = 0x1.a934f0p+1f;
286206c3fb27SDimitry Andric     const float cc_exp10 = 0x1.2f346ep-24f;
286306c3fb27SDimitry Andric 
286406c3fb27SDimitry Andric     SDValue C = DAG.getConstantFP(IsExp10 ? c_exp10 : c_exp, SL, VT);
286506c3fb27SDimitry Andric     SDValue CC = DAG.getConstantFP(IsExp10 ? cc_exp10 : cc_exp, SL, VT);
286606c3fb27SDimitry Andric 
286706c3fb27SDimitry Andric     PH = DAG.getNode(ISD::FMUL, SL, VT, X, C, Flags);
286806c3fb27SDimitry Andric     SDValue NegPH = DAG.getNode(ISD::FNEG, SL, VT, PH, Flags);
286906c3fb27SDimitry Andric     SDValue FMA0 = DAG.getNode(ISD::FMA, SL, VT, X, C, NegPH, Flags);
287006c3fb27SDimitry Andric     PL = DAG.getNode(ISD::FMA, SL, VT, X, CC, FMA0, Flags);
287106c3fb27SDimitry Andric   } else {
287206c3fb27SDimitry Andric     const float ch_exp = 0x1.714000p+0f;
287306c3fb27SDimitry Andric     const float cl_exp = 0x1.47652ap-12f; // ch + cl are 36 bits
287406c3fb27SDimitry Andric 
287506c3fb27SDimitry Andric     const float ch_exp10 = 0x1.a92000p+1f;
287606c3fb27SDimitry Andric     const float cl_exp10 = 0x1.4f0978p-11f;
287706c3fb27SDimitry Andric 
287806c3fb27SDimitry Andric     SDValue CH = DAG.getConstantFP(IsExp10 ? ch_exp10 : ch_exp, SL, VT);
287906c3fb27SDimitry Andric     SDValue CL = DAG.getConstantFP(IsExp10 ? cl_exp10 : cl_exp, SL, VT);
288006c3fb27SDimitry Andric 
288106c3fb27SDimitry Andric     SDValue XAsInt = DAG.getNode(ISD::BITCAST, SL, MVT::i32, X);
288206c3fb27SDimitry Andric     SDValue MaskConst = DAG.getConstant(0xfffff000, SL, MVT::i32);
288306c3fb27SDimitry Andric     SDValue XHAsInt = DAG.getNode(ISD::AND, SL, MVT::i32, XAsInt, MaskConst);
288406c3fb27SDimitry Andric     SDValue XH = DAG.getNode(ISD::BITCAST, SL, VT, XHAsInt);
288506c3fb27SDimitry Andric     SDValue XL = DAG.getNode(ISD::FSUB, SL, VT, X, XH, Flags);
288606c3fb27SDimitry Andric 
288706c3fb27SDimitry Andric     PH = DAG.getNode(ISD::FMUL, SL, VT, XH, CH, Flags);
288806c3fb27SDimitry Andric 
288906c3fb27SDimitry Andric     SDValue XLCL = DAG.getNode(ISD::FMUL, SL, VT, XL, CL, Flags);
289006c3fb27SDimitry Andric     SDValue Mad0 = getMad(DAG, SL, VT, XL, CH, XLCL, Flags);
289106c3fb27SDimitry Andric     PL = getMad(DAG, SL, VT, XH, CL, Mad0, Flags);
289206c3fb27SDimitry Andric   }
289306c3fb27SDimitry Andric 
289406c3fb27SDimitry Andric   SDValue E = DAG.getNode(ISD::FRINT, SL, VT, PH, Flags);
289506c3fb27SDimitry Andric 
289606c3fb27SDimitry Andric   // It is unsafe to contract this fsub into the PH multiply.
289706c3fb27SDimitry Andric   SDValue PHSubE = DAG.getNode(ISD::FSUB, SL, VT, PH, E, FlagsNoContract);
289806c3fb27SDimitry Andric 
289906c3fb27SDimitry Andric   SDValue A = DAG.getNode(ISD::FADD, SL, VT, PHSubE, PL, Flags);
290006c3fb27SDimitry Andric   SDValue IntE = DAG.getNode(ISD::FP_TO_SINT, SL, MVT::i32, E);
290106c3fb27SDimitry Andric   SDValue Exp2 = DAG.getNode(AMDGPUISD::EXP, SL, VT, A, Flags);
290206c3fb27SDimitry Andric 
290306c3fb27SDimitry Andric   SDValue R = DAG.getNode(ISD::FLDEXP, SL, VT, Exp2, IntE, Flags);
290406c3fb27SDimitry Andric 
290506c3fb27SDimitry Andric   SDValue UnderflowCheckConst =
290606c3fb27SDimitry Andric       DAG.getConstantFP(IsExp10 ? -0x1.66d3e8p+5f : -0x1.9d1da0p+6f, SL, VT);
290706c3fb27SDimitry Andric 
290806c3fb27SDimitry Andric   EVT SetCCVT = getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), VT);
290906c3fb27SDimitry Andric   SDValue Zero = DAG.getConstantFP(0.0, SL, VT);
291006c3fb27SDimitry Andric   SDValue Underflow =
291106c3fb27SDimitry Andric       DAG.getSetCC(SL, SetCCVT, X, UnderflowCheckConst, ISD::SETOLT);
291206c3fb27SDimitry Andric 
291306c3fb27SDimitry Andric   R = DAG.getNode(ISD::SELECT, SL, VT, Underflow, Zero, R);
291406c3fb27SDimitry Andric   const auto &Options = getTargetMachine().Options;
291506c3fb27SDimitry Andric 
291606c3fb27SDimitry Andric   if (!Flags.hasNoInfs() && !Options.NoInfsFPMath) {
291706c3fb27SDimitry Andric     SDValue OverflowCheckConst =
291806c3fb27SDimitry Andric         DAG.getConstantFP(IsExp10 ? 0x1.344136p+5f : 0x1.62e430p+6f, SL, VT);
291906c3fb27SDimitry Andric     SDValue Overflow =
292006c3fb27SDimitry Andric         DAG.getSetCC(SL, SetCCVT, X, OverflowCheckConst, ISD::SETOGT);
292106c3fb27SDimitry Andric     SDValue Inf =
292206c3fb27SDimitry Andric         DAG.getConstantFP(APFloat::getInf(APFloat::IEEEsingle()), SL, VT);
292306c3fb27SDimitry Andric     R = DAG.getNode(ISD::SELECT, SL, VT, Overflow, Inf, R);
292406c3fb27SDimitry Andric   }
292506c3fb27SDimitry Andric 
292606c3fb27SDimitry Andric   return R;
29270b57cec5SDimitry Andric }
29280b57cec5SDimitry Andric 
29290b57cec5SDimitry Andric static bool isCtlzOpc(unsigned Opc) {
29300b57cec5SDimitry Andric   return Opc == ISD::CTLZ || Opc == ISD::CTLZ_ZERO_UNDEF;
29310b57cec5SDimitry Andric }
29320b57cec5SDimitry Andric 
29330b57cec5SDimitry Andric static bool isCttzOpc(unsigned Opc) {
29340b57cec5SDimitry Andric   return Opc == ISD::CTTZ || Opc == ISD::CTTZ_ZERO_UNDEF;
29350b57cec5SDimitry Andric }
29360b57cec5SDimitry Andric 
29370b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerCTLZ_CTTZ(SDValue Op, SelectionDAG &DAG) const {
29380b57cec5SDimitry Andric   SDLoc SL(Op);
29390b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
29400b57cec5SDimitry Andric 
2941349cc55cSDimitry Andric   assert(isCtlzOpc(Op.getOpcode()) || isCttzOpc(Op.getOpcode()));
2942349cc55cSDimitry Andric   bool Ctlz = isCtlzOpc(Op.getOpcode());
2943349cc55cSDimitry Andric   unsigned NewOpc = Ctlz ? AMDGPUISD::FFBH_U32 : AMDGPUISD::FFBL_B32;
29440b57cec5SDimitry Andric 
2945349cc55cSDimitry Andric   bool ZeroUndef = Op.getOpcode() == ISD::CTLZ_ZERO_UNDEF ||
2946349cc55cSDimitry Andric                    Op.getOpcode() == ISD::CTTZ_ZERO_UNDEF;
29470b57cec5SDimitry Andric 
2948349cc55cSDimitry Andric   if (Src.getValueType() == MVT::i32) {
2949349cc55cSDimitry Andric     // (ctlz hi:lo) -> (umin (ffbh src), 32)
2950349cc55cSDimitry Andric     // (cttz hi:lo) -> (umin (ffbl src), 32)
2951349cc55cSDimitry Andric     // (ctlz_zero_undef src) -> (ffbh src)
2952349cc55cSDimitry Andric     // (cttz_zero_undef src) -> (ffbl src)
2953349cc55cSDimitry Andric     SDValue NewOpr = DAG.getNode(NewOpc, SL, MVT::i32, Src);
2954349cc55cSDimitry Andric     if (!ZeroUndef) {
2955349cc55cSDimitry Andric       const SDValue Const32 = DAG.getConstant(32, SL, MVT::i32);
2956349cc55cSDimitry Andric       NewOpr = DAG.getNode(ISD::UMIN, SL, MVT::i32, NewOpr, Const32);
2957349cc55cSDimitry Andric     }
2958349cc55cSDimitry Andric     return NewOpr;
29590b57cec5SDimitry Andric   }
29600b57cec5SDimitry Andric 
2961349cc55cSDimitry Andric   SDValue Lo, Hi;
2962349cc55cSDimitry Andric   std::tie(Lo, Hi) = split64BitValue(Src, DAG);
2963349cc55cSDimitry Andric 
2964349cc55cSDimitry Andric   SDValue OprLo = DAG.getNode(NewOpc, SL, MVT::i32, Lo);
2965349cc55cSDimitry Andric   SDValue OprHi = DAG.getNode(NewOpc, SL, MVT::i32, Hi);
2966349cc55cSDimitry Andric 
2967349cc55cSDimitry Andric   // (ctlz hi:lo) -> (umin3 (ffbh hi), (uaddsat (ffbh lo), 32), 64)
2968349cc55cSDimitry Andric   // (cttz hi:lo) -> (umin3 (uaddsat (ffbl hi), 32), (ffbl lo), 64)
2969349cc55cSDimitry Andric   // (ctlz_zero_undef hi:lo) -> (umin (ffbh hi), (add (ffbh lo), 32))
2970349cc55cSDimitry Andric   // (cttz_zero_undef hi:lo) -> (umin (add (ffbl hi), 32), (ffbl lo))
2971349cc55cSDimitry Andric 
2972349cc55cSDimitry Andric   unsigned AddOpc = ZeroUndef ? ISD::ADD : ISD::UADDSAT;
2973349cc55cSDimitry Andric   const SDValue Const32 = DAG.getConstant(32, SL, MVT::i32);
2974349cc55cSDimitry Andric   if (Ctlz)
2975349cc55cSDimitry Andric     OprLo = DAG.getNode(AddOpc, SL, MVT::i32, OprLo, Const32);
2976349cc55cSDimitry Andric   else
2977349cc55cSDimitry Andric     OprHi = DAG.getNode(AddOpc, SL, MVT::i32, OprHi, Const32);
2978349cc55cSDimitry Andric 
2979349cc55cSDimitry Andric   SDValue NewOpr;
2980349cc55cSDimitry Andric   NewOpr = DAG.getNode(ISD::UMIN, SL, MVT::i32, OprLo, OprHi);
29810b57cec5SDimitry Andric   if (!ZeroUndef) {
2982349cc55cSDimitry Andric     const SDValue Const64 = DAG.getConstant(64, SL, MVT::i32);
2983349cc55cSDimitry Andric     NewOpr = DAG.getNode(ISD::UMIN, SL, MVT::i32, NewOpr, Const64);
29840b57cec5SDimitry Andric   }
29850b57cec5SDimitry Andric 
29860b57cec5SDimitry Andric   return DAG.getNode(ISD::ZERO_EXTEND, SL, MVT::i64, NewOpr);
29870b57cec5SDimitry Andric }
29880b57cec5SDimitry Andric 
29890b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerINT_TO_FP32(SDValue Op, SelectionDAG &DAG,
29900b57cec5SDimitry Andric                                                bool Signed) const {
2991349cc55cSDimitry Andric   // The regular method converting a 64-bit integer to float roughly consists of
2992349cc55cSDimitry Andric   // 2 steps: normalization and rounding. In fact, after normalization, the
2993349cc55cSDimitry Andric   // conversion from a 64-bit integer to a float is essentially the same as the
2994349cc55cSDimitry Andric   // one from a 32-bit integer. The only difference is that it has more
2995349cc55cSDimitry Andric   // trailing bits to be rounded. To leverage the native 32-bit conversion, a
2996349cc55cSDimitry Andric   // 64-bit integer could be preprocessed and fit into a 32-bit integer then
2997349cc55cSDimitry Andric   // converted into the correct float number. The basic steps for the unsigned
2998349cc55cSDimitry Andric   // conversion are illustrated in the following pseudo code:
2999349cc55cSDimitry Andric   //
3000349cc55cSDimitry Andric   // f32 uitofp(i64 u) {
3001349cc55cSDimitry Andric   //   i32 hi, lo = split(u);
3002349cc55cSDimitry Andric   //   // Only count the leading zeros in hi as we have native support of the
3003349cc55cSDimitry Andric   //   // conversion from i32 to f32. If hi is all 0s, the conversion is
3004349cc55cSDimitry Andric   //   // reduced to a 32-bit one automatically.
3005349cc55cSDimitry Andric   //   i32 shamt = clz(hi); // Return 32 if hi is all 0s.
3006349cc55cSDimitry Andric   //   u <<= shamt;
3007349cc55cSDimitry Andric   //   hi, lo = split(u);
3008349cc55cSDimitry Andric   //   hi |= (lo != 0) ? 1 : 0; // Adjust rounding bit in hi based on lo.
3009349cc55cSDimitry Andric   //   // convert it as a 32-bit integer and scale the result back.
3010349cc55cSDimitry Andric   //   return uitofp(hi) * 2^(32 - shamt);
30110b57cec5SDimitry Andric   // }
3012349cc55cSDimitry Andric   //
3013349cc55cSDimitry Andric   // The signed one follows the same principle but uses 'ffbh_i32' to count its
3014349cc55cSDimitry Andric   // sign bits instead. If 'ffbh_i32' is not available, its absolute value is
3015349cc55cSDimitry Andric   // converted instead followed by negation based its sign bit.
30160b57cec5SDimitry Andric 
30170b57cec5SDimitry Andric   SDLoc SL(Op);
30180b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
30190b57cec5SDimitry Andric 
3020349cc55cSDimitry Andric   SDValue Lo, Hi;
3021349cc55cSDimitry Andric   std::tie(Lo, Hi) = split64BitValue(Src, DAG);
3022349cc55cSDimitry Andric   SDValue Sign;
3023349cc55cSDimitry Andric   SDValue ShAmt;
3024349cc55cSDimitry Andric   if (Signed && Subtarget->isGCN()) {
3025349cc55cSDimitry Andric     // We also need to consider the sign bit in Lo if Hi has just sign bits,
3026349cc55cSDimitry Andric     // i.e. Hi is 0 or -1. However, that only needs to take the MSB into
3027349cc55cSDimitry Andric     // account. That is, the maximal shift is
3028349cc55cSDimitry Andric     // - 32 if Lo and Hi have opposite signs;
3029349cc55cSDimitry Andric     // - 33 if Lo and Hi have the same sign.
3030349cc55cSDimitry Andric     //
3031349cc55cSDimitry Andric     // Or, MaxShAmt = 33 + OppositeSign, where
3032349cc55cSDimitry Andric     //
3033349cc55cSDimitry Andric     // OppositeSign is defined as ((Lo ^ Hi) >> 31), which is
3034349cc55cSDimitry Andric     // - -1 if Lo and Hi have opposite signs; and
3035349cc55cSDimitry Andric     // -  0 otherwise.
3036349cc55cSDimitry Andric     //
3037349cc55cSDimitry Andric     // All in all, ShAmt is calculated as
3038349cc55cSDimitry Andric     //
3039349cc55cSDimitry Andric     //  umin(sffbh(Hi), 33 + (Lo^Hi)>>31) - 1.
3040349cc55cSDimitry Andric     //
3041349cc55cSDimitry Andric     // or
3042349cc55cSDimitry Andric     //
3043349cc55cSDimitry Andric     //  umin(sffbh(Hi) - 1, 32 + (Lo^Hi)>>31).
3044349cc55cSDimitry Andric     //
3045349cc55cSDimitry Andric     // to reduce the critical path.
3046349cc55cSDimitry Andric     SDValue OppositeSign = DAG.getNode(
3047349cc55cSDimitry Andric         ISD::SRA, SL, MVT::i32, DAG.getNode(ISD::XOR, SL, MVT::i32, Lo, Hi),
3048349cc55cSDimitry Andric         DAG.getConstant(31, SL, MVT::i32));
3049349cc55cSDimitry Andric     SDValue MaxShAmt =
3050349cc55cSDimitry Andric         DAG.getNode(ISD::ADD, SL, MVT::i32, DAG.getConstant(32, SL, MVT::i32),
3051349cc55cSDimitry Andric                     OppositeSign);
3052349cc55cSDimitry Andric     // Count the leading sign bits.
3053349cc55cSDimitry Andric     ShAmt = DAG.getNode(AMDGPUISD::FFBH_I32, SL, MVT::i32, Hi);
3054349cc55cSDimitry Andric     // Different from unsigned conversion, the shift should be one bit less to
3055349cc55cSDimitry Andric     // preserve the sign bit.
3056349cc55cSDimitry Andric     ShAmt = DAG.getNode(ISD::SUB, SL, MVT::i32, ShAmt,
3057349cc55cSDimitry Andric                         DAG.getConstant(1, SL, MVT::i32));
3058349cc55cSDimitry Andric     ShAmt = DAG.getNode(ISD::UMIN, SL, MVT::i32, ShAmt, MaxShAmt);
3059349cc55cSDimitry Andric   } else {
30600b57cec5SDimitry Andric     if (Signed) {
3061349cc55cSDimitry Andric       // Without 'ffbh_i32', only leading zeros could be counted. Take the
3062349cc55cSDimitry Andric       // absolute value first.
3063349cc55cSDimitry Andric       Sign = DAG.getNode(ISD::SRA, SL, MVT::i64, Src,
3064349cc55cSDimitry Andric                          DAG.getConstant(63, SL, MVT::i64));
3065349cc55cSDimitry Andric       SDValue Abs =
3066349cc55cSDimitry Andric           DAG.getNode(ISD::XOR, SL, MVT::i64,
3067349cc55cSDimitry Andric                       DAG.getNode(ISD::ADD, SL, MVT::i64, Src, Sign), Sign);
3068349cc55cSDimitry Andric       std::tie(Lo, Hi) = split64BitValue(Abs, DAG);
30690b57cec5SDimitry Andric     }
3070349cc55cSDimitry Andric     // Count the leading zeros.
3071349cc55cSDimitry Andric     ShAmt = DAG.getNode(ISD::CTLZ, SL, MVT::i32, Hi);
3072349cc55cSDimitry Andric     // The shift amount for signed integers is [0, 32].
3073349cc55cSDimitry Andric   }
3074349cc55cSDimitry Andric   // Normalize the given 64-bit integer.
3075349cc55cSDimitry Andric   SDValue Norm = DAG.getNode(ISD::SHL, SL, MVT::i64, Src, ShAmt);
3076349cc55cSDimitry Andric   // Split it again.
3077349cc55cSDimitry Andric   std::tie(Lo, Hi) = split64BitValue(Norm, DAG);
3078349cc55cSDimitry Andric   // Calculate the adjust bit for rounding.
3079349cc55cSDimitry Andric   // (lo != 0) ? 1 : 0 => (lo >= 1) ? 1 : 0 => umin(1, lo)
3080349cc55cSDimitry Andric   SDValue Adjust = DAG.getNode(ISD::UMIN, SL, MVT::i32,
3081349cc55cSDimitry Andric                                DAG.getConstant(1, SL, MVT::i32), Lo);
3082349cc55cSDimitry Andric   // Get the 32-bit normalized integer.
3083349cc55cSDimitry Andric   Norm = DAG.getNode(ISD::OR, SL, MVT::i32, Hi, Adjust);
3084349cc55cSDimitry Andric   // Convert the normalized 32-bit integer into f32.
3085349cc55cSDimitry Andric   unsigned Opc =
3086349cc55cSDimitry Andric       (Signed && Subtarget->isGCN()) ? ISD::SINT_TO_FP : ISD::UINT_TO_FP;
3087349cc55cSDimitry Andric   SDValue FVal = DAG.getNode(Opc, SL, MVT::f32, Norm);
30880b57cec5SDimitry Andric 
3089349cc55cSDimitry Andric   // Finally, need to scale back the converted floating number as the original
3090349cc55cSDimitry Andric   // 64-bit integer is converted as a 32-bit one.
3091349cc55cSDimitry Andric   ShAmt = DAG.getNode(ISD::SUB, SL, MVT::i32, DAG.getConstant(32, SL, MVT::i32),
3092349cc55cSDimitry Andric                       ShAmt);
3093349cc55cSDimitry Andric   // On GCN, use LDEXP directly.
3094349cc55cSDimitry Andric   if (Subtarget->isGCN())
309506c3fb27SDimitry Andric     return DAG.getNode(ISD::FLDEXP, SL, MVT::f32, FVal, ShAmt);
30960b57cec5SDimitry Andric 
3097349cc55cSDimitry Andric   // Otherwise, align 'ShAmt' to the exponent part and add it into the exponent
3098349cc55cSDimitry Andric   // part directly to emulate the multiplication of 2^ShAmt. That 8-bit
3099349cc55cSDimitry Andric   // exponent is enough to avoid overflowing into the sign bit.
3100349cc55cSDimitry Andric   SDValue Exp = DAG.getNode(ISD::SHL, SL, MVT::i32, ShAmt,
3101349cc55cSDimitry Andric                             DAG.getConstant(23, SL, MVT::i32));
3102349cc55cSDimitry Andric   SDValue IVal =
3103349cc55cSDimitry Andric       DAG.getNode(ISD::ADD, SL, MVT::i32,
3104349cc55cSDimitry Andric                   DAG.getNode(ISD::BITCAST, SL, MVT::i32, FVal), Exp);
3105349cc55cSDimitry Andric   if (Signed) {
3106349cc55cSDimitry Andric     // Set the sign bit.
3107349cc55cSDimitry Andric     Sign = DAG.getNode(ISD::SHL, SL, MVT::i32,
3108349cc55cSDimitry Andric                        DAG.getNode(ISD::TRUNCATE, SL, MVT::i32, Sign),
3109349cc55cSDimitry Andric                        DAG.getConstant(31, SL, MVT::i32));
3110349cc55cSDimitry Andric     IVal = DAG.getNode(ISD::OR, SL, MVT::i32, IVal, Sign);
3111349cc55cSDimitry Andric   }
3112349cc55cSDimitry Andric   return DAG.getNode(ISD::BITCAST, SL, MVT::f32, IVal);
31130b57cec5SDimitry Andric }
31140b57cec5SDimitry Andric 
31150b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerINT_TO_FP64(SDValue Op, SelectionDAG &DAG,
31160b57cec5SDimitry Andric                                                bool Signed) const {
31170b57cec5SDimitry Andric   SDLoc SL(Op);
31180b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
31190b57cec5SDimitry Andric 
3120349cc55cSDimitry Andric   SDValue Lo, Hi;
3121349cc55cSDimitry Andric   std::tie(Lo, Hi) = split64BitValue(Src, DAG);
31220b57cec5SDimitry Andric 
31230b57cec5SDimitry Andric   SDValue CvtHi = DAG.getNode(Signed ? ISD::SINT_TO_FP : ISD::UINT_TO_FP,
31240b57cec5SDimitry Andric                               SL, MVT::f64, Hi);
31250b57cec5SDimitry Andric 
31260b57cec5SDimitry Andric   SDValue CvtLo = DAG.getNode(ISD::UINT_TO_FP, SL, MVT::f64, Lo);
31270b57cec5SDimitry Andric 
312806c3fb27SDimitry Andric   SDValue LdExp = DAG.getNode(ISD::FLDEXP, SL, MVT::f64, CvtHi,
31290b57cec5SDimitry Andric                               DAG.getConstant(32, SL, MVT::i32));
31300b57cec5SDimitry Andric   // TODO: Should this propagate fast-math-flags?
31310b57cec5SDimitry Andric   return DAG.getNode(ISD::FADD, SL, MVT::f64, LdExp, CvtLo);
31320b57cec5SDimitry Andric }
31330b57cec5SDimitry Andric 
31340b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerUINT_TO_FP(SDValue Op,
31350b57cec5SDimitry Andric                                                SelectionDAG &DAG) const {
31360b57cec5SDimitry Andric   // TODO: Factor out code common with LowerSINT_TO_FP.
31370b57cec5SDimitry Andric   EVT DestVT = Op.getValueType();
3138480093f4SDimitry Andric   SDValue Src = Op.getOperand(0);
3139480093f4SDimitry Andric   EVT SrcVT = Src.getValueType();
3140480093f4SDimitry Andric 
3141480093f4SDimitry Andric   if (SrcVT == MVT::i16) {
3142480093f4SDimitry Andric     if (DestVT == MVT::f16)
3143480093f4SDimitry Andric       return Op;
3144480093f4SDimitry Andric     SDLoc DL(Op);
3145480093f4SDimitry Andric 
3146480093f4SDimitry Andric     // Promote src to i32
3147480093f4SDimitry Andric     SDValue Ext = DAG.getNode(ISD::ZERO_EXTEND, DL, MVT::i32, Src);
3148480093f4SDimitry Andric     return DAG.getNode(ISD::UINT_TO_FP, DL, DestVT, Ext);
3149480093f4SDimitry Andric   }
3150480093f4SDimitry Andric 
3151480093f4SDimitry Andric   assert(SrcVT == MVT::i64 && "operation should be legal");
3152480093f4SDimitry Andric 
31530b57cec5SDimitry Andric   if (Subtarget->has16BitInsts() && DestVT == MVT::f16) {
31540b57cec5SDimitry Andric     SDLoc DL(Op);
31550b57cec5SDimitry Andric 
31560b57cec5SDimitry Andric     SDValue IntToFp32 = DAG.getNode(Op.getOpcode(), DL, MVT::f32, Src);
3157bdd1243dSDimitry Andric     SDValue FPRoundFlag =
3158bdd1243dSDimitry Andric         DAG.getIntPtrConstant(0, SDLoc(Op), /*isTarget=*/true);
31590b57cec5SDimitry Andric     SDValue FPRound =
31600b57cec5SDimitry Andric         DAG.getNode(ISD::FP_ROUND, DL, MVT::f16, IntToFp32, FPRoundFlag);
31610b57cec5SDimitry Andric 
31620b57cec5SDimitry Andric     return FPRound;
31630b57cec5SDimitry Andric   }
31640b57cec5SDimitry Andric 
31650b57cec5SDimitry Andric   if (DestVT == MVT::f32)
31660b57cec5SDimitry Andric     return LowerINT_TO_FP32(Op, DAG, false);
31670b57cec5SDimitry Andric 
31680b57cec5SDimitry Andric   assert(DestVT == MVT::f64);
31690b57cec5SDimitry Andric   return LowerINT_TO_FP64(Op, DAG, false);
31700b57cec5SDimitry Andric }
31710b57cec5SDimitry Andric 
31720b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerSINT_TO_FP(SDValue Op,
31730b57cec5SDimitry Andric                                               SelectionDAG &DAG) const {
3174480093f4SDimitry Andric   EVT DestVT = Op.getValueType();
3175480093f4SDimitry Andric 
3176480093f4SDimitry Andric   SDValue Src = Op.getOperand(0);
3177480093f4SDimitry Andric   EVT SrcVT = Src.getValueType();
3178480093f4SDimitry Andric 
3179480093f4SDimitry Andric   if (SrcVT == MVT::i16) {
3180480093f4SDimitry Andric     if (DestVT == MVT::f16)
3181480093f4SDimitry Andric       return Op;
3182480093f4SDimitry Andric 
3183480093f4SDimitry Andric     SDLoc DL(Op);
3184480093f4SDimitry Andric     // Promote src to i32
3185480093f4SDimitry Andric     SDValue Ext = DAG.getNode(ISD::SIGN_EXTEND, DL, MVT::i32, Src);
3186480093f4SDimitry Andric     return DAG.getNode(ISD::SINT_TO_FP, DL, DestVT, Ext);
3187480093f4SDimitry Andric   }
3188480093f4SDimitry Andric 
3189480093f4SDimitry Andric   assert(SrcVT == MVT::i64 && "operation should be legal");
31900b57cec5SDimitry Andric 
31910b57cec5SDimitry Andric   // TODO: Factor out code common with LowerUINT_TO_FP.
31920b57cec5SDimitry Andric 
31930b57cec5SDimitry Andric   if (Subtarget->has16BitInsts() && DestVT == MVT::f16) {
31940b57cec5SDimitry Andric     SDLoc DL(Op);
31950b57cec5SDimitry Andric     SDValue Src = Op.getOperand(0);
31960b57cec5SDimitry Andric 
31970b57cec5SDimitry Andric     SDValue IntToFp32 = DAG.getNode(Op.getOpcode(), DL, MVT::f32, Src);
3198bdd1243dSDimitry Andric     SDValue FPRoundFlag =
3199bdd1243dSDimitry Andric         DAG.getIntPtrConstant(0, SDLoc(Op), /*isTarget=*/true);
32000b57cec5SDimitry Andric     SDValue FPRound =
32010b57cec5SDimitry Andric         DAG.getNode(ISD::FP_ROUND, DL, MVT::f16, IntToFp32, FPRoundFlag);
32020b57cec5SDimitry Andric 
32030b57cec5SDimitry Andric     return FPRound;
32040b57cec5SDimitry Andric   }
32050b57cec5SDimitry Andric 
32060b57cec5SDimitry Andric   if (DestVT == MVT::f32)
32070b57cec5SDimitry Andric     return LowerINT_TO_FP32(Op, DAG, true);
32080b57cec5SDimitry Andric 
32090b57cec5SDimitry Andric   assert(DestVT == MVT::f64);
32100b57cec5SDimitry Andric   return LowerINT_TO_FP64(Op, DAG, true);
32110b57cec5SDimitry Andric }
32120b57cec5SDimitry Andric 
3213fe6060f1SDimitry Andric SDValue AMDGPUTargetLowering::LowerFP_TO_INT64(SDValue Op, SelectionDAG &DAG,
32140b57cec5SDimitry Andric                                                bool Signed) const {
32150b57cec5SDimitry Andric   SDLoc SL(Op);
32160b57cec5SDimitry Andric 
32170b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
3218fe6060f1SDimitry Andric   EVT SrcVT = Src.getValueType();
32190b57cec5SDimitry Andric 
3220fe6060f1SDimitry Andric   assert(SrcVT == MVT::f32 || SrcVT == MVT::f64);
32210b57cec5SDimitry Andric 
3222fe6060f1SDimitry Andric   // The basic idea of converting a floating point number into a pair of 32-bit
3223fe6060f1SDimitry Andric   // integers is illustrated as follows:
3224fe6060f1SDimitry Andric   //
3225fe6060f1SDimitry Andric   //     tf := trunc(val);
3226fe6060f1SDimitry Andric   //    hif := floor(tf * 2^-32);
3227fe6060f1SDimitry Andric   //    lof := tf - hif * 2^32; // lof is always positive due to floor.
3228fe6060f1SDimitry Andric   //     hi := fptoi(hif);
3229fe6060f1SDimitry Andric   //     lo := fptoi(lof);
3230fe6060f1SDimitry Andric   //
3231fe6060f1SDimitry Andric   SDValue Trunc = DAG.getNode(ISD::FTRUNC, SL, SrcVT, Src);
3232fe6060f1SDimitry Andric   SDValue Sign;
3233fe6060f1SDimitry Andric   if (Signed && SrcVT == MVT::f32) {
3234fe6060f1SDimitry Andric     // However, a 32-bit floating point number has only 23 bits mantissa and
3235fe6060f1SDimitry Andric     // it's not enough to hold all the significant bits of `lof` if val is
3236fe6060f1SDimitry Andric     // negative. To avoid the loss of precision, We need to take the absolute
3237fe6060f1SDimitry Andric     // value after truncating and flip the result back based on the original
3238fe6060f1SDimitry Andric     // signedness.
3239fe6060f1SDimitry Andric     Sign = DAG.getNode(ISD::SRA, SL, MVT::i32,
3240fe6060f1SDimitry Andric                        DAG.getNode(ISD::BITCAST, SL, MVT::i32, Trunc),
3241fe6060f1SDimitry Andric                        DAG.getConstant(31, SL, MVT::i32));
3242fe6060f1SDimitry Andric     Trunc = DAG.getNode(ISD::FABS, SL, SrcVT, Trunc);
3243fe6060f1SDimitry Andric   }
3244fe6060f1SDimitry Andric 
3245fe6060f1SDimitry Andric   SDValue K0, K1;
3246fe6060f1SDimitry Andric   if (SrcVT == MVT::f64) {
324706c3fb27SDimitry Andric     K0 = DAG.getConstantFP(
324806c3fb27SDimitry Andric         llvm::bit_cast<double>(UINT64_C(/*2^-32*/ 0x3df0000000000000)), SL,
324906c3fb27SDimitry Andric         SrcVT);
325006c3fb27SDimitry Andric     K1 = DAG.getConstantFP(
325106c3fb27SDimitry Andric         llvm::bit_cast<double>(UINT64_C(/*-2^32*/ 0xc1f0000000000000)), SL,
325206c3fb27SDimitry Andric         SrcVT);
3253fe6060f1SDimitry Andric   } else {
325406c3fb27SDimitry Andric     K0 = DAG.getConstantFP(
325506c3fb27SDimitry Andric         llvm::bit_cast<float>(UINT32_C(/*2^-32*/ 0x2f800000)), SL, SrcVT);
325606c3fb27SDimitry Andric     K1 = DAG.getConstantFP(
325706c3fb27SDimitry Andric         llvm::bit_cast<float>(UINT32_C(/*-2^32*/ 0xcf800000)), SL, SrcVT);
3258fe6060f1SDimitry Andric   }
32590b57cec5SDimitry Andric   // TODO: Should this propagate fast-math-flags?
3260fe6060f1SDimitry Andric   SDValue Mul = DAG.getNode(ISD::FMUL, SL, SrcVT, Trunc, K0);
32610b57cec5SDimitry Andric 
3262fe6060f1SDimitry Andric   SDValue FloorMul = DAG.getNode(ISD::FFLOOR, SL, SrcVT, Mul);
32630b57cec5SDimitry Andric 
3264fe6060f1SDimitry Andric   SDValue Fma = DAG.getNode(ISD::FMA, SL, SrcVT, FloorMul, K1, Trunc);
32650b57cec5SDimitry Andric 
3266fe6060f1SDimitry Andric   SDValue Hi = DAG.getNode((Signed && SrcVT == MVT::f64) ? ISD::FP_TO_SINT
3267fe6060f1SDimitry Andric                                                          : ISD::FP_TO_UINT,
3268fe6060f1SDimitry Andric                            SL, MVT::i32, FloorMul);
32690b57cec5SDimitry Andric   SDValue Lo = DAG.getNode(ISD::FP_TO_UINT, SL, MVT::i32, Fma);
32700b57cec5SDimitry Andric 
3271fe6060f1SDimitry Andric   SDValue Result = DAG.getNode(ISD::BITCAST, SL, MVT::i64,
3272fe6060f1SDimitry Andric                                DAG.getBuildVector(MVT::v2i32, SL, {Lo, Hi}));
32730b57cec5SDimitry Andric 
3274fe6060f1SDimitry Andric   if (Signed && SrcVT == MVT::f32) {
3275fe6060f1SDimitry Andric     assert(Sign);
3276fe6060f1SDimitry Andric     // Flip the result based on the signedness, which is either all 0s or 1s.
3277fe6060f1SDimitry Andric     Sign = DAG.getNode(ISD::BITCAST, SL, MVT::i64,
3278fe6060f1SDimitry Andric                        DAG.getBuildVector(MVT::v2i32, SL, {Sign, Sign}));
3279fe6060f1SDimitry Andric     // r := xor(r, sign) - sign;
3280fe6060f1SDimitry Andric     Result =
3281fe6060f1SDimitry Andric         DAG.getNode(ISD::SUB, SL, MVT::i64,
3282fe6060f1SDimitry Andric                     DAG.getNode(ISD::XOR, SL, MVT::i64, Result, Sign), Sign);
3283fe6060f1SDimitry Andric   }
3284fe6060f1SDimitry Andric 
3285fe6060f1SDimitry Andric   return Result;
32860b57cec5SDimitry Andric }
32870b57cec5SDimitry Andric 
32880b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFP_TO_FP16(SDValue Op, SelectionDAG &DAG) const {
32890b57cec5SDimitry Andric   SDLoc DL(Op);
32900b57cec5SDimitry Andric   SDValue N0 = Op.getOperand(0);
32910b57cec5SDimitry Andric 
32920b57cec5SDimitry Andric   // Convert to target node to get known bits
32930b57cec5SDimitry Andric   if (N0.getValueType() == MVT::f32)
32940b57cec5SDimitry Andric     return DAG.getNode(AMDGPUISD::FP_TO_FP16, DL, Op.getValueType(), N0);
32950b57cec5SDimitry Andric 
32960b57cec5SDimitry Andric   if (getTargetMachine().Options.UnsafeFPMath) {
32970b57cec5SDimitry Andric     // There is a generic expand for FP_TO_FP16 with unsafe fast math.
32980b57cec5SDimitry Andric     return SDValue();
32990b57cec5SDimitry Andric   }
33000b57cec5SDimitry Andric 
33010b57cec5SDimitry Andric   assert(N0.getSimpleValueType() == MVT::f64);
33020b57cec5SDimitry Andric 
33030b57cec5SDimitry Andric   // f64 -> f16 conversion using round-to-nearest-even rounding mode.
33040b57cec5SDimitry Andric   const unsigned ExpMask = 0x7ff;
33050b57cec5SDimitry Andric   const unsigned ExpBiasf64 = 1023;
33060b57cec5SDimitry Andric   const unsigned ExpBiasf16 = 15;
33070b57cec5SDimitry Andric   SDValue Zero = DAG.getConstant(0, DL, MVT::i32);
33080b57cec5SDimitry Andric   SDValue One = DAG.getConstant(1, DL, MVT::i32);
33090b57cec5SDimitry Andric   SDValue U = DAG.getNode(ISD::BITCAST, DL, MVT::i64, N0);
33100b57cec5SDimitry Andric   SDValue UH = DAG.getNode(ISD::SRL, DL, MVT::i64, U,
33110b57cec5SDimitry Andric                            DAG.getConstant(32, DL, MVT::i64));
33120b57cec5SDimitry Andric   UH = DAG.getZExtOrTrunc(UH, DL, MVT::i32);
33130b57cec5SDimitry Andric   U = DAG.getZExtOrTrunc(U, DL, MVT::i32);
33140b57cec5SDimitry Andric   SDValue E = DAG.getNode(ISD::SRL, DL, MVT::i32, UH,
33150b57cec5SDimitry Andric                           DAG.getConstant(20, DL, MVT::i64));
33160b57cec5SDimitry Andric   E = DAG.getNode(ISD::AND, DL, MVT::i32, E,
33170b57cec5SDimitry Andric                   DAG.getConstant(ExpMask, DL, MVT::i32));
33180b57cec5SDimitry Andric   // Subtract the fp64 exponent bias (1023) to get the real exponent and
33190b57cec5SDimitry Andric   // add the f16 bias (15) to get the biased exponent for the f16 format.
33200b57cec5SDimitry Andric   E = DAG.getNode(ISD::ADD, DL, MVT::i32, E,
33210b57cec5SDimitry Andric                   DAG.getConstant(-ExpBiasf64 + ExpBiasf16, DL, MVT::i32));
33220b57cec5SDimitry Andric 
33230b57cec5SDimitry Andric   SDValue M = DAG.getNode(ISD::SRL, DL, MVT::i32, UH,
33240b57cec5SDimitry Andric                           DAG.getConstant(8, DL, MVT::i32));
33250b57cec5SDimitry Andric   M = DAG.getNode(ISD::AND, DL, MVT::i32, M,
33260b57cec5SDimitry Andric                   DAG.getConstant(0xffe, DL, MVT::i32));
33270b57cec5SDimitry Andric 
33280b57cec5SDimitry Andric   SDValue MaskedSig = DAG.getNode(ISD::AND, DL, MVT::i32, UH,
33290b57cec5SDimitry Andric                                   DAG.getConstant(0x1ff, DL, MVT::i32));
33300b57cec5SDimitry Andric   MaskedSig = DAG.getNode(ISD::OR, DL, MVT::i32, MaskedSig, U);
33310b57cec5SDimitry Andric 
33320b57cec5SDimitry Andric   SDValue Lo40Set = DAG.getSelectCC(DL, MaskedSig, Zero, Zero, One, ISD::SETEQ);
33330b57cec5SDimitry Andric   M = DAG.getNode(ISD::OR, DL, MVT::i32, M, Lo40Set);
33340b57cec5SDimitry Andric 
33350b57cec5SDimitry Andric   // (M != 0 ? 0x0200 : 0) | 0x7c00;
33360b57cec5SDimitry Andric   SDValue I = DAG.getNode(ISD::OR, DL, MVT::i32,
33370b57cec5SDimitry Andric       DAG.getSelectCC(DL, M, Zero, DAG.getConstant(0x0200, DL, MVT::i32),
33380b57cec5SDimitry Andric                       Zero, ISD::SETNE), DAG.getConstant(0x7c00, DL, MVT::i32));
33390b57cec5SDimitry Andric 
33400b57cec5SDimitry Andric   // N = M | (E << 12);
33410b57cec5SDimitry Andric   SDValue N = DAG.getNode(ISD::OR, DL, MVT::i32, M,
33420b57cec5SDimitry Andric       DAG.getNode(ISD::SHL, DL, MVT::i32, E,
33430b57cec5SDimitry Andric                   DAG.getConstant(12, DL, MVT::i32)));
33440b57cec5SDimitry Andric 
33450b57cec5SDimitry Andric   // B = clamp(1-E, 0, 13);
33460b57cec5SDimitry Andric   SDValue OneSubExp = DAG.getNode(ISD::SUB, DL, MVT::i32,
33470b57cec5SDimitry Andric                                   One, E);
33480b57cec5SDimitry Andric   SDValue B = DAG.getNode(ISD::SMAX, DL, MVT::i32, OneSubExp, Zero);
33490b57cec5SDimitry Andric   B = DAG.getNode(ISD::SMIN, DL, MVT::i32, B,
33500b57cec5SDimitry Andric                   DAG.getConstant(13, DL, MVT::i32));
33510b57cec5SDimitry Andric 
33520b57cec5SDimitry Andric   SDValue SigSetHigh = DAG.getNode(ISD::OR, DL, MVT::i32, M,
33530b57cec5SDimitry Andric                                    DAG.getConstant(0x1000, DL, MVT::i32));
33540b57cec5SDimitry Andric 
33550b57cec5SDimitry Andric   SDValue D = DAG.getNode(ISD::SRL, DL, MVT::i32, SigSetHigh, B);
33560b57cec5SDimitry Andric   SDValue D0 = DAG.getNode(ISD::SHL, DL, MVT::i32, D, B);
33570b57cec5SDimitry Andric   SDValue D1 = DAG.getSelectCC(DL, D0, SigSetHigh, One, Zero, ISD::SETNE);
33580b57cec5SDimitry Andric   D = DAG.getNode(ISD::OR, DL, MVT::i32, D, D1);
33590b57cec5SDimitry Andric 
33600b57cec5SDimitry Andric   SDValue V = DAG.getSelectCC(DL, E, One, D, N, ISD::SETLT);
33610b57cec5SDimitry Andric   SDValue VLow3 = DAG.getNode(ISD::AND, DL, MVT::i32, V,
33620b57cec5SDimitry Andric                               DAG.getConstant(0x7, DL, MVT::i32));
33630b57cec5SDimitry Andric   V = DAG.getNode(ISD::SRL, DL, MVT::i32, V,
33640b57cec5SDimitry Andric                   DAG.getConstant(2, DL, MVT::i32));
33650b57cec5SDimitry Andric   SDValue V0 = DAG.getSelectCC(DL, VLow3, DAG.getConstant(3, DL, MVT::i32),
33660b57cec5SDimitry Andric                                One, Zero, ISD::SETEQ);
33670b57cec5SDimitry Andric   SDValue V1 = DAG.getSelectCC(DL, VLow3, DAG.getConstant(5, DL, MVT::i32),
33680b57cec5SDimitry Andric                                One, Zero, ISD::SETGT);
33690b57cec5SDimitry Andric   V1 = DAG.getNode(ISD::OR, DL, MVT::i32, V0, V1);
33700b57cec5SDimitry Andric   V = DAG.getNode(ISD::ADD, DL, MVT::i32, V, V1);
33710b57cec5SDimitry Andric 
33720b57cec5SDimitry Andric   V = DAG.getSelectCC(DL, E, DAG.getConstant(30, DL, MVT::i32),
33730b57cec5SDimitry Andric                       DAG.getConstant(0x7c00, DL, MVT::i32), V, ISD::SETGT);
33740b57cec5SDimitry Andric   V = DAG.getSelectCC(DL, E, DAG.getConstant(1039, DL, MVT::i32),
33750b57cec5SDimitry Andric                       I, V, ISD::SETEQ);
33760b57cec5SDimitry Andric 
33770b57cec5SDimitry Andric   // Extract the sign bit.
33780b57cec5SDimitry Andric   SDValue Sign = DAG.getNode(ISD::SRL, DL, MVT::i32, UH,
33790b57cec5SDimitry Andric                             DAG.getConstant(16, DL, MVT::i32));
33800b57cec5SDimitry Andric   Sign = DAG.getNode(ISD::AND, DL, MVT::i32, Sign,
33810b57cec5SDimitry Andric                      DAG.getConstant(0x8000, DL, MVT::i32));
33820b57cec5SDimitry Andric 
33830b57cec5SDimitry Andric   V = DAG.getNode(ISD::OR, DL, MVT::i32, Sign, V);
33840b57cec5SDimitry Andric   return DAG.getZExtOrTrunc(V, DL, Op.getValueType());
33850b57cec5SDimitry Andric }
33860b57cec5SDimitry Andric 
3387fe6060f1SDimitry Andric SDValue AMDGPUTargetLowering::LowerFP_TO_INT(SDValue Op,
33880b57cec5SDimitry Andric                                              SelectionDAG &DAG) const {
33890b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
3390fe6060f1SDimitry Andric   unsigned OpOpcode = Op.getOpcode();
33910b57cec5SDimitry Andric   EVT SrcVT = Src.getValueType();
3392fe6060f1SDimitry Andric   EVT DestVT = Op.getValueType();
3393fe6060f1SDimitry Andric 
3394fe6060f1SDimitry Andric   // Will be selected natively
3395fe6060f1SDimitry Andric   if (SrcVT == MVT::f16 && DestVT == MVT::i16)
3396fe6060f1SDimitry Andric     return Op;
3397fe6060f1SDimitry Andric 
3398fe6060f1SDimitry Andric   // Promote i16 to i32
3399fe6060f1SDimitry Andric   if (DestVT == MVT::i16 && (SrcVT == MVT::f32 || SrcVT == MVT::f64)) {
3400fe6060f1SDimitry Andric     SDLoc DL(Op);
3401fe6060f1SDimitry Andric 
3402fe6060f1SDimitry Andric     SDValue FpToInt32 = DAG.getNode(OpOpcode, DL, MVT::i32, Src);
3403fe6060f1SDimitry Andric     return DAG.getNode(ISD::TRUNCATE, DL, MVT::i16, FpToInt32);
3404fe6060f1SDimitry Andric   }
3405fe6060f1SDimitry Andric 
3406e8d8bef9SDimitry Andric   if (SrcVT == MVT::f16 ||
3407e8d8bef9SDimitry Andric       (SrcVT == MVT::f32 && Src.getOpcode() == ISD::FP16_TO_FP)) {
34080b57cec5SDimitry Andric     SDLoc DL(Op);
34090b57cec5SDimitry Andric 
3410fe6060f1SDimitry Andric     SDValue FpToInt32 = DAG.getNode(OpOpcode, DL, MVT::i32, Src);
3411fe6060f1SDimitry Andric     unsigned Ext =
3412fe6060f1SDimitry Andric         OpOpcode == ISD::FP_TO_SINT ? ISD::SIGN_EXTEND : ISD::ZERO_EXTEND;
3413fe6060f1SDimitry Andric     return DAG.getNode(Ext, DL, MVT::i64, FpToInt32);
34140b57cec5SDimitry Andric   }
34150b57cec5SDimitry Andric 
3416fe6060f1SDimitry Andric   if (DestVT == MVT::i64 && (SrcVT == MVT::f32 || SrcVT == MVT::f64))
3417fe6060f1SDimitry Andric     return LowerFP_TO_INT64(Op, DAG, OpOpcode == ISD::FP_TO_SINT);
34180b57cec5SDimitry Andric 
34190b57cec5SDimitry Andric   return SDValue();
34200b57cec5SDimitry Andric }
34210b57cec5SDimitry Andric 
34220b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerSIGN_EXTEND_INREG(SDValue Op,
34230b57cec5SDimitry Andric                                                      SelectionDAG &DAG) const {
34240b57cec5SDimitry Andric   EVT ExtraVT = cast<VTSDNode>(Op.getOperand(1))->getVT();
34250b57cec5SDimitry Andric   MVT VT = Op.getSimpleValueType();
34260b57cec5SDimitry Andric   MVT ScalarVT = VT.getScalarType();
34270b57cec5SDimitry Andric 
34280b57cec5SDimitry Andric   assert(VT.isVector());
34290b57cec5SDimitry Andric 
34300b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
34310b57cec5SDimitry Andric   SDLoc DL(Op);
34320b57cec5SDimitry Andric 
34330b57cec5SDimitry Andric   // TODO: Don't scalarize on Evergreen?
34340b57cec5SDimitry Andric   unsigned NElts = VT.getVectorNumElements();
34350b57cec5SDimitry Andric   SmallVector<SDValue, 8> Args;
34360b57cec5SDimitry Andric   DAG.ExtractVectorElements(Src, Args, 0, NElts);
34370b57cec5SDimitry Andric 
34380b57cec5SDimitry Andric   SDValue VTOp = DAG.getValueType(ExtraVT.getScalarType());
34390b57cec5SDimitry Andric   for (unsigned I = 0; I < NElts; ++I)
34400b57cec5SDimitry Andric     Args[I] = DAG.getNode(ISD::SIGN_EXTEND_INREG, DL, ScalarVT, Args[I], VTOp);
34410b57cec5SDimitry Andric 
34420b57cec5SDimitry Andric   return DAG.getBuildVector(VT, DL, Args);
34430b57cec5SDimitry Andric }
34440b57cec5SDimitry Andric 
34450b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
34460b57cec5SDimitry Andric // Custom DAG optimizations
34470b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
34480b57cec5SDimitry Andric 
34490b57cec5SDimitry Andric static bool isU24(SDValue Op, SelectionDAG &DAG) {
34500b57cec5SDimitry Andric   return AMDGPUTargetLowering::numBitsUnsigned(Op, DAG) <= 24;
34510b57cec5SDimitry Andric }
34520b57cec5SDimitry Andric 
34530b57cec5SDimitry Andric static bool isI24(SDValue Op, SelectionDAG &DAG) {
34540b57cec5SDimitry Andric   EVT VT = Op.getValueType();
34550b57cec5SDimitry Andric   return VT.getSizeInBits() >= 24 && // Types less than 24-bit should be treated
34560b57cec5SDimitry Andric                                      // as unsigned 24-bit values.
3457349cc55cSDimitry Andric          AMDGPUTargetLowering::numBitsSigned(Op, DAG) <= 24;
34580b57cec5SDimitry Andric }
34590b57cec5SDimitry Andric 
3460fe6060f1SDimitry Andric static SDValue simplifyMul24(SDNode *Node24,
34610b57cec5SDimitry Andric                              TargetLowering::DAGCombinerInfo &DCI) {
34620b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
34635ffd83dbSDimitry Andric   const TargetLowering &TLI = DAG.getTargetLoweringInfo();
34648bcb0991SDimitry Andric   bool IsIntrin = Node24->getOpcode() == ISD::INTRINSIC_WO_CHAIN;
34658bcb0991SDimitry Andric 
34668bcb0991SDimitry Andric   SDValue LHS = IsIntrin ? Node24->getOperand(1) : Node24->getOperand(0);
34678bcb0991SDimitry Andric   SDValue RHS = IsIntrin ? Node24->getOperand(2) : Node24->getOperand(1);
34688bcb0991SDimitry Andric   unsigned NewOpcode = Node24->getOpcode();
34698bcb0991SDimitry Andric   if (IsIntrin) {
34708bcb0991SDimitry Andric     unsigned IID = cast<ConstantSDNode>(Node24->getOperand(0))->getZExtValue();
3471349cc55cSDimitry Andric     switch (IID) {
3472349cc55cSDimitry Andric     case Intrinsic::amdgcn_mul_i24:
3473349cc55cSDimitry Andric       NewOpcode = AMDGPUISD::MUL_I24;
3474349cc55cSDimitry Andric       break;
3475349cc55cSDimitry Andric     case Intrinsic::amdgcn_mul_u24:
3476349cc55cSDimitry Andric       NewOpcode = AMDGPUISD::MUL_U24;
3477349cc55cSDimitry Andric       break;
3478349cc55cSDimitry Andric     case Intrinsic::amdgcn_mulhi_i24:
3479349cc55cSDimitry Andric       NewOpcode = AMDGPUISD::MULHI_I24;
3480349cc55cSDimitry Andric       break;
3481349cc55cSDimitry Andric     case Intrinsic::amdgcn_mulhi_u24:
3482349cc55cSDimitry Andric       NewOpcode = AMDGPUISD::MULHI_U24;
3483349cc55cSDimitry Andric       break;
3484349cc55cSDimitry Andric     default:
3485349cc55cSDimitry Andric       llvm_unreachable("Expected 24-bit mul intrinsic");
3486349cc55cSDimitry Andric     }
34878bcb0991SDimitry Andric   }
34880b57cec5SDimitry Andric 
34890b57cec5SDimitry Andric   APInt Demanded = APInt::getLowBitsSet(LHS.getValueSizeInBits(), 24);
34900b57cec5SDimitry Andric 
34915ffd83dbSDimitry Andric   // First try to simplify using SimplifyMultipleUseDemandedBits which allows
34925ffd83dbSDimitry Andric   // the operands to have other uses, but will only perform simplifications that
34935ffd83dbSDimitry Andric   // involve bypassing some nodes for this user.
34945ffd83dbSDimitry Andric   SDValue DemandedLHS = TLI.SimplifyMultipleUseDemandedBits(LHS, Demanded, DAG);
34955ffd83dbSDimitry Andric   SDValue DemandedRHS = TLI.SimplifyMultipleUseDemandedBits(RHS, Demanded, DAG);
34960b57cec5SDimitry Andric   if (DemandedLHS || DemandedRHS)
34978bcb0991SDimitry Andric     return DAG.getNode(NewOpcode, SDLoc(Node24), Node24->getVTList(),
34980b57cec5SDimitry Andric                        DemandedLHS ? DemandedLHS : LHS,
34990b57cec5SDimitry Andric                        DemandedRHS ? DemandedRHS : RHS);
35000b57cec5SDimitry Andric 
35010b57cec5SDimitry Andric   // Now try SimplifyDemandedBits which can simplify the nodes used by our
35020b57cec5SDimitry Andric   // operands if this node is the only user.
35030b57cec5SDimitry Andric   if (TLI.SimplifyDemandedBits(LHS, Demanded, DCI))
35040b57cec5SDimitry Andric     return SDValue(Node24, 0);
35050b57cec5SDimitry Andric   if (TLI.SimplifyDemandedBits(RHS, Demanded, DCI))
35060b57cec5SDimitry Andric     return SDValue(Node24, 0);
35070b57cec5SDimitry Andric 
35080b57cec5SDimitry Andric   return SDValue();
35090b57cec5SDimitry Andric }
35100b57cec5SDimitry Andric 
35110b57cec5SDimitry Andric template <typename IntTy>
35120b57cec5SDimitry Andric static SDValue constantFoldBFE(SelectionDAG &DAG, IntTy Src0, uint32_t Offset,
35130b57cec5SDimitry Andric                                uint32_t Width, const SDLoc &DL) {
35140b57cec5SDimitry Andric   if (Width + Offset < 32) {
35150b57cec5SDimitry Andric     uint32_t Shl = static_cast<uint32_t>(Src0) << (32 - Offset - Width);
35160b57cec5SDimitry Andric     IntTy Result = static_cast<IntTy>(Shl) >> (32 - Width);
35170b57cec5SDimitry Andric     return DAG.getConstant(Result, DL, MVT::i32);
35180b57cec5SDimitry Andric   }
35190b57cec5SDimitry Andric 
35200b57cec5SDimitry Andric   return DAG.getConstant(Src0 >> Offset, DL, MVT::i32);
35210b57cec5SDimitry Andric }
35220b57cec5SDimitry Andric 
35230b57cec5SDimitry Andric static bool hasVolatileUser(SDNode *Val) {
35240b57cec5SDimitry Andric   for (SDNode *U : Val->uses()) {
35250b57cec5SDimitry Andric     if (MemSDNode *M = dyn_cast<MemSDNode>(U)) {
35260b57cec5SDimitry Andric       if (M->isVolatile())
35270b57cec5SDimitry Andric         return true;
35280b57cec5SDimitry Andric     }
35290b57cec5SDimitry Andric   }
35300b57cec5SDimitry Andric 
35310b57cec5SDimitry Andric   return false;
35320b57cec5SDimitry Andric }
35330b57cec5SDimitry Andric 
35340b57cec5SDimitry Andric bool AMDGPUTargetLowering::shouldCombineMemoryType(EVT VT) const {
35350b57cec5SDimitry Andric   // i32 vectors are the canonical memory type.
35360b57cec5SDimitry Andric   if (VT.getScalarType() == MVT::i32 || isTypeLegal(VT))
35370b57cec5SDimitry Andric     return false;
35380b57cec5SDimitry Andric 
35390b57cec5SDimitry Andric   if (!VT.isByteSized())
35400b57cec5SDimitry Andric     return false;
35410b57cec5SDimitry Andric 
35420b57cec5SDimitry Andric   unsigned Size = VT.getStoreSize();
35430b57cec5SDimitry Andric 
35440b57cec5SDimitry Andric   if ((Size == 1 || Size == 2 || Size == 4) && !VT.isVector())
35450b57cec5SDimitry Andric     return false;
35460b57cec5SDimitry Andric 
35470b57cec5SDimitry Andric   if (Size == 3 || (Size > 4 && (Size % 4 != 0)))
35480b57cec5SDimitry Andric     return false;
35490b57cec5SDimitry Andric 
35500b57cec5SDimitry Andric   return true;
35510b57cec5SDimitry Andric }
35520b57cec5SDimitry Andric 
35530b57cec5SDimitry Andric // Replace load of an illegal type with a store of a bitcast to a friendlier
35540b57cec5SDimitry Andric // type.
35550b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performLoadCombine(SDNode *N,
35560b57cec5SDimitry Andric                                                  DAGCombinerInfo &DCI) const {
35570b57cec5SDimitry Andric   if (!DCI.isBeforeLegalize())
35580b57cec5SDimitry Andric     return SDValue();
35590b57cec5SDimitry Andric 
35600b57cec5SDimitry Andric   LoadSDNode *LN = cast<LoadSDNode>(N);
35615ffd83dbSDimitry Andric   if (!LN->isSimple() || !ISD::isNormalLoad(LN) || hasVolatileUser(LN))
35620b57cec5SDimitry Andric     return SDValue();
35630b57cec5SDimitry Andric 
35640b57cec5SDimitry Andric   SDLoc SL(N);
35650b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
35660b57cec5SDimitry Andric   EVT VT = LN->getMemoryVT();
35670b57cec5SDimitry Andric 
35680b57cec5SDimitry Andric   unsigned Size = VT.getStoreSize();
35695ffd83dbSDimitry Andric   Align Alignment = LN->getAlign();
35705ffd83dbSDimitry Andric   if (Alignment < Size && isTypeLegal(VT)) {
3571bdd1243dSDimitry Andric     unsigned IsFast;
35720b57cec5SDimitry Andric     unsigned AS = LN->getAddressSpace();
35730b57cec5SDimitry Andric 
35740b57cec5SDimitry Andric     // Expand unaligned loads earlier than legalization. Due to visitation order
35750b57cec5SDimitry Andric     // problems during legalization, the emitted instructions to pack and unpack
35760b57cec5SDimitry Andric     // the bytes again are not eliminated in the case of an unaligned copy.
3577fe6060f1SDimitry Andric     if (!allowsMisalignedMemoryAccesses(
3578fe6060f1SDimitry Andric             VT, AS, Alignment, LN->getMemOperand()->getFlags(), &IsFast)) {
3579480093f4SDimitry Andric       if (VT.isVector())
358081ad6265SDimitry Andric         return SplitVectorLoad(SDValue(LN, 0), DAG);
358181ad6265SDimitry Andric 
358281ad6265SDimitry Andric       SDValue Ops[2];
35830b57cec5SDimitry Andric       std::tie(Ops[0], Ops[1]) = expandUnalignedLoad(LN, DAG);
3584480093f4SDimitry Andric 
35850b57cec5SDimitry Andric       return DAG.getMergeValues(Ops, SDLoc(N));
35860b57cec5SDimitry Andric     }
35870b57cec5SDimitry Andric 
35880b57cec5SDimitry Andric     if (!IsFast)
35890b57cec5SDimitry Andric       return SDValue();
35900b57cec5SDimitry Andric   }
35910b57cec5SDimitry Andric 
35920b57cec5SDimitry Andric   if (!shouldCombineMemoryType(VT))
35930b57cec5SDimitry Andric     return SDValue();
35940b57cec5SDimitry Andric 
35950b57cec5SDimitry Andric   EVT NewVT = getEquivalentMemType(*DAG.getContext(), VT);
35960b57cec5SDimitry Andric 
35970b57cec5SDimitry Andric   SDValue NewLoad
35980b57cec5SDimitry Andric     = DAG.getLoad(NewVT, SL, LN->getChain(),
35990b57cec5SDimitry Andric                   LN->getBasePtr(), LN->getMemOperand());
36000b57cec5SDimitry Andric 
36010b57cec5SDimitry Andric   SDValue BC = DAG.getNode(ISD::BITCAST, SL, VT, NewLoad);
36020b57cec5SDimitry Andric   DCI.CombineTo(N, BC, NewLoad.getValue(1));
36030b57cec5SDimitry Andric   return SDValue(N, 0);
36040b57cec5SDimitry Andric }
36050b57cec5SDimitry Andric 
36060b57cec5SDimitry Andric // Replace store of an illegal type with a store of a bitcast to a friendlier
36070b57cec5SDimitry Andric // type.
36080b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performStoreCombine(SDNode *N,
36090b57cec5SDimitry Andric                                                   DAGCombinerInfo &DCI) const {
36100b57cec5SDimitry Andric   if (!DCI.isBeforeLegalize())
36110b57cec5SDimitry Andric     return SDValue();
36120b57cec5SDimitry Andric 
36130b57cec5SDimitry Andric   StoreSDNode *SN = cast<StoreSDNode>(N);
36145ffd83dbSDimitry Andric   if (!SN->isSimple() || !ISD::isNormalStore(SN))
36150b57cec5SDimitry Andric     return SDValue();
36160b57cec5SDimitry Andric 
36170b57cec5SDimitry Andric   EVT VT = SN->getMemoryVT();
36180b57cec5SDimitry Andric   unsigned Size = VT.getStoreSize();
36190b57cec5SDimitry Andric 
36200b57cec5SDimitry Andric   SDLoc SL(N);
36210b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
36225ffd83dbSDimitry Andric   Align Alignment = SN->getAlign();
36235ffd83dbSDimitry Andric   if (Alignment < Size && isTypeLegal(VT)) {
3624bdd1243dSDimitry Andric     unsigned IsFast;
36250b57cec5SDimitry Andric     unsigned AS = SN->getAddressSpace();
36260b57cec5SDimitry Andric 
36270b57cec5SDimitry Andric     // Expand unaligned stores earlier than legalization. Due to visitation
36280b57cec5SDimitry Andric     // order problems during legalization, the emitted instructions to pack and
36290b57cec5SDimitry Andric     // unpack the bytes again are not eliminated in the case of an unaligned
36300b57cec5SDimitry Andric     // copy.
3631fe6060f1SDimitry Andric     if (!allowsMisalignedMemoryAccesses(
3632fe6060f1SDimitry Andric             VT, AS, Alignment, SN->getMemOperand()->getFlags(), &IsFast)) {
36330b57cec5SDimitry Andric       if (VT.isVector())
363481ad6265SDimitry Andric         return SplitVectorStore(SDValue(SN, 0), DAG);
36350b57cec5SDimitry Andric 
36360b57cec5SDimitry Andric       return expandUnalignedStore(SN, DAG);
36370b57cec5SDimitry Andric     }
36380b57cec5SDimitry Andric 
36390b57cec5SDimitry Andric     if (!IsFast)
36400b57cec5SDimitry Andric       return SDValue();
36410b57cec5SDimitry Andric   }
36420b57cec5SDimitry Andric 
36430b57cec5SDimitry Andric   if (!shouldCombineMemoryType(VT))
36440b57cec5SDimitry Andric     return SDValue();
36450b57cec5SDimitry Andric 
36460b57cec5SDimitry Andric   EVT NewVT = getEquivalentMemType(*DAG.getContext(), VT);
36470b57cec5SDimitry Andric   SDValue Val = SN->getValue();
36480b57cec5SDimitry Andric 
36490b57cec5SDimitry Andric   //DCI.AddToWorklist(Val.getNode());
36500b57cec5SDimitry Andric 
36510b57cec5SDimitry Andric   bool OtherUses = !Val.hasOneUse();
36520b57cec5SDimitry Andric   SDValue CastVal = DAG.getNode(ISD::BITCAST, SL, NewVT, Val);
36530b57cec5SDimitry Andric   if (OtherUses) {
36540b57cec5SDimitry Andric     SDValue CastBack = DAG.getNode(ISD::BITCAST, SL, VT, CastVal);
36550b57cec5SDimitry Andric     DAG.ReplaceAllUsesOfValueWith(Val, CastBack);
36560b57cec5SDimitry Andric   }
36570b57cec5SDimitry Andric 
36580b57cec5SDimitry Andric   return DAG.getStore(SN->getChain(), SL, CastVal,
36590b57cec5SDimitry Andric                       SN->getBasePtr(), SN->getMemOperand());
36600b57cec5SDimitry Andric }
36610b57cec5SDimitry Andric 
36620b57cec5SDimitry Andric // FIXME: This should go in generic DAG combiner with an isTruncateFree check,
36630b57cec5SDimitry Andric // but isTruncateFree is inaccurate for i16 now because of SALU vs. VALU
36640b57cec5SDimitry Andric // issues.
36650b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performAssertSZExtCombine(SDNode *N,
36660b57cec5SDimitry Andric                                                         DAGCombinerInfo &DCI) const {
36670b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
36680b57cec5SDimitry Andric   SDValue N0 = N->getOperand(0);
36690b57cec5SDimitry Andric 
36700b57cec5SDimitry Andric   // (vt2 (assertzext (truncate vt0:x), vt1)) ->
36710b57cec5SDimitry Andric   //     (vt2 (truncate (assertzext vt0:x, vt1)))
36720b57cec5SDimitry Andric   if (N0.getOpcode() == ISD::TRUNCATE) {
36730b57cec5SDimitry Andric     SDValue N1 = N->getOperand(1);
36740b57cec5SDimitry Andric     EVT ExtVT = cast<VTSDNode>(N1)->getVT();
36750b57cec5SDimitry Andric     SDLoc SL(N);
36760b57cec5SDimitry Andric 
36770b57cec5SDimitry Andric     SDValue Src = N0.getOperand(0);
36780b57cec5SDimitry Andric     EVT SrcVT = Src.getValueType();
36790b57cec5SDimitry Andric     if (SrcVT.bitsGE(ExtVT)) {
36800b57cec5SDimitry Andric       SDValue NewInReg = DAG.getNode(N->getOpcode(), SL, SrcVT, Src, N1);
36810b57cec5SDimitry Andric       return DAG.getNode(ISD::TRUNCATE, SL, N->getValueType(0), NewInReg);
36820b57cec5SDimitry Andric     }
36830b57cec5SDimitry Andric   }
36840b57cec5SDimitry Andric 
36850b57cec5SDimitry Andric   return SDValue();
36860b57cec5SDimitry Andric }
36878bcb0991SDimitry Andric 
36888bcb0991SDimitry Andric SDValue AMDGPUTargetLowering::performIntrinsicWOChainCombine(
36898bcb0991SDimitry Andric   SDNode *N, DAGCombinerInfo &DCI) const {
36908bcb0991SDimitry Andric   unsigned IID = cast<ConstantSDNode>(N->getOperand(0))->getZExtValue();
36918bcb0991SDimitry Andric   switch (IID) {
36928bcb0991SDimitry Andric   case Intrinsic::amdgcn_mul_i24:
36938bcb0991SDimitry Andric   case Intrinsic::amdgcn_mul_u24:
3694349cc55cSDimitry Andric   case Intrinsic::amdgcn_mulhi_i24:
3695349cc55cSDimitry Andric   case Intrinsic::amdgcn_mulhi_u24:
3696fe6060f1SDimitry Andric     return simplifyMul24(N, DCI);
36975ffd83dbSDimitry Andric   case Intrinsic::amdgcn_fract:
36985ffd83dbSDimitry Andric   case Intrinsic::amdgcn_rsq:
36995ffd83dbSDimitry Andric   case Intrinsic::amdgcn_rcp_legacy:
37005ffd83dbSDimitry Andric   case Intrinsic::amdgcn_rsq_legacy:
37015ffd83dbSDimitry Andric   case Intrinsic::amdgcn_rsq_clamp:
37025ffd83dbSDimitry Andric   case Intrinsic::amdgcn_ldexp: {
37035ffd83dbSDimitry Andric     // FIXME: This is probably wrong. If src is an sNaN, it won't be quieted
37045ffd83dbSDimitry Andric     SDValue Src = N->getOperand(1);
37055ffd83dbSDimitry Andric     return Src.isUndef() ? Src : SDValue();
37065ffd83dbSDimitry Andric   }
370706c3fb27SDimitry Andric   case Intrinsic::amdgcn_frexp_exp: {
370806c3fb27SDimitry Andric     // frexp_exp (fneg x) -> frexp_exp x
370906c3fb27SDimitry Andric     // frexp_exp (fabs x) -> frexp_exp x
371006c3fb27SDimitry Andric     // frexp_exp (fneg (fabs x)) -> frexp_exp x
371106c3fb27SDimitry Andric     SDValue Src = N->getOperand(1);
371206c3fb27SDimitry Andric     SDValue PeekSign = peekFPSignOps(Src);
371306c3fb27SDimitry Andric     if (PeekSign == Src)
371406c3fb27SDimitry Andric       return SDValue();
371506c3fb27SDimitry Andric     return SDValue(DCI.DAG.UpdateNodeOperands(N, N->getOperand(0), PeekSign),
371606c3fb27SDimitry Andric                    0);
371706c3fb27SDimitry Andric   }
37188bcb0991SDimitry Andric   default:
37198bcb0991SDimitry Andric     return SDValue();
37208bcb0991SDimitry Andric   }
37218bcb0991SDimitry Andric }
37228bcb0991SDimitry Andric 
37230b57cec5SDimitry Andric /// Split the 64-bit value \p LHS into two 32-bit components, and perform the
37240b57cec5SDimitry Andric /// binary operation \p Opc to it with the corresponding constant operands.
37250b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::splitBinaryBitConstantOpImpl(
37260b57cec5SDimitry Andric   DAGCombinerInfo &DCI, const SDLoc &SL,
37270b57cec5SDimitry Andric   unsigned Opc, SDValue LHS,
37280b57cec5SDimitry Andric   uint32_t ValLo, uint32_t ValHi) const {
37290b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
37300b57cec5SDimitry Andric   SDValue Lo, Hi;
37310b57cec5SDimitry Andric   std::tie(Lo, Hi) = split64BitValue(LHS, DAG);
37320b57cec5SDimitry Andric 
37330b57cec5SDimitry Andric   SDValue LoRHS = DAG.getConstant(ValLo, SL, MVT::i32);
37340b57cec5SDimitry Andric   SDValue HiRHS = DAG.getConstant(ValHi, SL, MVT::i32);
37350b57cec5SDimitry Andric 
37360b57cec5SDimitry Andric   SDValue LoAnd = DAG.getNode(Opc, SL, MVT::i32, Lo, LoRHS);
37370b57cec5SDimitry Andric   SDValue HiAnd = DAG.getNode(Opc, SL, MVT::i32, Hi, HiRHS);
37380b57cec5SDimitry Andric 
37390b57cec5SDimitry Andric   // Re-visit the ands. It's possible we eliminated one of them and it could
37400b57cec5SDimitry Andric   // simplify the vector.
37410b57cec5SDimitry Andric   DCI.AddToWorklist(Lo.getNode());
37420b57cec5SDimitry Andric   DCI.AddToWorklist(Hi.getNode());
37430b57cec5SDimitry Andric 
37440b57cec5SDimitry Andric   SDValue Vec = DAG.getBuildVector(MVT::v2i32, SL, {LoAnd, HiAnd});
37450b57cec5SDimitry Andric   return DAG.getNode(ISD::BITCAST, SL, MVT::i64, Vec);
37460b57cec5SDimitry Andric }
37470b57cec5SDimitry Andric 
37480b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performShlCombine(SDNode *N,
37490b57cec5SDimitry Andric                                                 DAGCombinerInfo &DCI) const {
37500b57cec5SDimitry Andric   EVT VT = N->getValueType(0);
37510b57cec5SDimitry Andric 
37520b57cec5SDimitry Andric   ConstantSDNode *RHS = dyn_cast<ConstantSDNode>(N->getOperand(1));
37530b57cec5SDimitry Andric   if (!RHS)
37540b57cec5SDimitry Andric     return SDValue();
37550b57cec5SDimitry Andric 
37560b57cec5SDimitry Andric   SDValue LHS = N->getOperand(0);
37570b57cec5SDimitry Andric   unsigned RHSVal = RHS->getZExtValue();
37580b57cec5SDimitry Andric   if (!RHSVal)
37590b57cec5SDimitry Andric     return LHS;
37600b57cec5SDimitry Andric 
37610b57cec5SDimitry Andric   SDLoc SL(N);
37620b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
37630b57cec5SDimitry Andric 
37640b57cec5SDimitry Andric   switch (LHS->getOpcode()) {
37650b57cec5SDimitry Andric   default:
37660b57cec5SDimitry Andric     break;
37670b57cec5SDimitry Andric   case ISD::ZERO_EXTEND:
37680b57cec5SDimitry Andric   case ISD::SIGN_EXTEND:
37690b57cec5SDimitry Andric   case ISD::ANY_EXTEND: {
37700b57cec5SDimitry Andric     SDValue X = LHS->getOperand(0);
37710b57cec5SDimitry Andric 
37720b57cec5SDimitry Andric     if (VT == MVT::i32 && RHSVal == 16 && X.getValueType() == MVT::i16 &&
37730b57cec5SDimitry Andric         isOperationLegal(ISD::BUILD_VECTOR, MVT::v2i16)) {
37740b57cec5SDimitry Andric       // Prefer build_vector as the canonical form if packed types are legal.
37750b57cec5SDimitry Andric       // (shl ([asz]ext i16:x), 16 -> build_vector 0, x
37760b57cec5SDimitry Andric       SDValue Vec = DAG.getBuildVector(MVT::v2i16, SL,
37770b57cec5SDimitry Andric        { DAG.getConstant(0, SL, MVT::i16), LHS->getOperand(0) });
37780b57cec5SDimitry Andric       return DAG.getNode(ISD::BITCAST, SL, MVT::i32, Vec);
37790b57cec5SDimitry Andric     }
37800b57cec5SDimitry Andric 
37810b57cec5SDimitry Andric     // shl (ext x) => zext (shl x), if shift does not overflow int
37820b57cec5SDimitry Andric     if (VT != MVT::i64)
37830b57cec5SDimitry Andric       break;
37840b57cec5SDimitry Andric     KnownBits Known = DAG.computeKnownBits(X);
37850b57cec5SDimitry Andric     unsigned LZ = Known.countMinLeadingZeros();
37860b57cec5SDimitry Andric     if (LZ < RHSVal)
37870b57cec5SDimitry Andric       break;
37880b57cec5SDimitry Andric     EVT XVT = X.getValueType();
37890b57cec5SDimitry Andric     SDValue Shl = DAG.getNode(ISD::SHL, SL, XVT, X, SDValue(RHS, 0));
37900b57cec5SDimitry Andric     return DAG.getZExtOrTrunc(Shl, SL, VT);
37910b57cec5SDimitry Andric   }
37920b57cec5SDimitry Andric   }
37930b57cec5SDimitry Andric 
37940b57cec5SDimitry Andric   if (VT != MVT::i64)
37950b57cec5SDimitry Andric     return SDValue();
37960b57cec5SDimitry Andric 
37970b57cec5SDimitry Andric   // i64 (shl x, C) -> (build_pair 0, (shl x, C -32))
37980b57cec5SDimitry Andric 
37990b57cec5SDimitry Andric   // On some subtargets, 64-bit shift is a quarter rate instruction. In the
38000b57cec5SDimitry Andric   // common case, splitting this into a move and a 32-bit shift is faster and
38010b57cec5SDimitry Andric   // the same code size.
38020b57cec5SDimitry Andric   if (RHSVal < 32)
38030b57cec5SDimitry Andric     return SDValue();
38040b57cec5SDimitry Andric 
38050b57cec5SDimitry Andric   SDValue ShiftAmt = DAG.getConstant(RHSVal - 32, SL, MVT::i32);
38060b57cec5SDimitry Andric 
38070b57cec5SDimitry Andric   SDValue Lo = DAG.getNode(ISD::TRUNCATE, SL, MVT::i32, LHS);
38080b57cec5SDimitry Andric   SDValue NewShift = DAG.getNode(ISD::SHL, SL, MVT::i32, Lo, ShiftAmt);
38090b57cec5SDimitry Andric 
38100b57cec5SDimitry Andric   const SDValue Zero = DAG.getConstant(0, SL, MVT::i32);
38110b57cec5SDimitry Andric 
38120b57cec5SDimitry Andric   SDValue Vec = DAG.getBuildVector(MVT::v2i32, SL, {Zero, NewShift});
38130b57cec5SDimitry Andric   return DAG.getNode(ISD::BITCAST, SL, MVT::i64, Vec);
38140b57cec5SDimitry Andric }
38150b57cec5SDimitry Andric 
38160b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performSraCombine(SDNode *N,
38170b57cec5SDimitry Andric                                                 DAGCombinerInfo &DCI) const {
38180b57cec5SDimitry Andric   if (N->getValueType(0) != MVT::i64)
38190b57cec5SDimitry Andric     return SDValue();
38200b57cec5SDimitry Andric 
38210b57cec5SDimitry Andric   const ConstantSDNode *RHS = dyn_cast<ConstantSDNode>(N->getOperand(1));
38220b57cec5SDimitry Andric   if (!RHS)
38230b57cec5SDimitry Andric     return SDValue();
38240b57cec5SDimitry Andric 
38250b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
38260b57cec5SDimitry Andric   SDLoc SL(N);
38270b57cec5SDimitry Andric   unsigned RHSVal = RHS->getZExtValue();
38280b57cec5SDimitry Andric 
38290b57cec5SDimitry Andric   // (sra i64:x, 32) -> build_pair x, (sra hi_32(x), 31)
38300b57cec5SDimitry Andric   if (RHSVal == 32) {
38310b57cec5SDimitry Andric     SDValue Hi = getHiHalf64(N->getOperand(0), DAG);
38320b57cec5SDimitry Andric     SDValue NewShift = DAG.getNode(ISD::SRA, SL, MVT::i32, Hi,
38330b57cec5SDimitry Andric                                    DAG.getConstant(31, SL, MVT::i32));
38340b57cec5SDimitry Andric 
38350b57cec5SDimitry Andric     SDValue BuildVec = DAG.getBuildVector(MVT::v2i32, SL, {Hi, NewShift});
38360b57cec5SDimitry Andric     return DAG.getNode(ISD::BITCAST, SL, MVT::i64, BuildVec);
38370b57cec5SDimitry Andric   }
38380b57cec5SDimitry Andric 
38390b57cec5SDimitry Andric   // (sra i64:x, 63) -> build_pair (sra hi_32(x), 31), (sra hi_32(x), 31)
38400b57cec5SDimitry Andric   if (RHSVal == 63) {
38410b57cec5SDimitry Andric     SDValue Hi = getHiHalf64(N->getOperand(0), DAG);
38420b57cec5SDimitry Andric     SDValue NewShift = DAG.getNode(ISD::SRA, SL, MVT::i32, Hi,
38430b57cec5SDimitry Andric                                    DAG.getConstant(31, SL, MVT::i32));
38440b57cec5SDimitry Andric     SDValue BuildVec = DAG.getBuildVector(MVT::v2i32, SL, {NewShift, NewShift});
38450b57cec5SDimitry Andric     return DAG.getNode(ISD::BITCAST, SL, MVT::i64, BuildVec);
38460b57cec5SDimitry Andric   }
38470b57cec5SDimitry Andric 
38480b57cec5SDimitry Andric   return SDValue();
38490b57cec5SDimitry Andric }
38500b57cec5SDimitry Andric 
38510b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performSrlCombine(SDNode *N,
38520b57cec5SDimitry Andric                                                 DAGCombinerInfo &DCI) const {
38530b57cec5SDimitry Andric   auto *RHS = dyn_cast<ConstantSDNode>(N->getOperand(1));
38540b57cec5SDimitry Andric   if (!RHS)
38550b57cec5SDimitry Andric     return SDValue();
38560b57cec5SDimitry Andric 
38570b57cec5SDimitry Andric   EVT VT = N->getValueType(0);
38580b57cec5SDimitry Andric   SDValue LHS = N->getOperand(0);
38590b57cec5SDimitry Andric   unsigned ShiftAmt = RHS->getZExtValue();
38600b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
38610b57cec5SDimitry Andric   SDLoc SL(N);
38620b57cec5SDimitry Andric 
38630b57cec5SDimitry Andric   // fold (srl (and x, c1 << c2), c2) -> (and (srl(x, c2), c1)
38640b57cec5SDimitry Andric   // this improves the ability to match BFE patterns in isel.
38650b57cec5SDimitry Andric   if (LHS.getOpcode() == ISD::AND) {
38660b57cec5SDimitry Andric     if (auto *Mask = dyn_cast<ConstantSDNode>(LHS.getOperand(1))) {
386781ad6265SDimitry Andric       unsigned MaskIdx, MaskLen;
386881ad6265SDimitry Andric       if (Mask->getAPIntValue().isShiftedMask(MaskIdx, MaskLen) &&
386981ad6265SDimitry Andric           MaskIdx == ShiftAmt) {
38700b57cec5SDimitry Andric         return DAG.getNode(
38710b57cec5SDimitry Andric             ISD::AND, SL, VT,
38720b57cec5SDimitry Andric             DAG.getNode(ISD::SRL, SL, VT, LHS.getOperand(0), N->getOperand(1)),
38730b57cec5SDimitry Andric             DAG.getNode(ISD::SRL, SL, VT, LHS.getOperand(1), N->getOperand(1)));
38740b57cec5SDimitry Andric       }
38750b57cec5SDimitry Andric     }
38760b57cec5SDimitry Andric   }
38770b57cec5SDimitry Andric 
38780b57cec5SDimitry Andric   if (VT != MVT::i64)
38790b57cec5SDimitry Andric     return SDValue();
38800b57cec5SDimitry Andric 
38810b57cec5SDimitry Andric   if (ShiftAmt < 32)
38820b57cec5SDimitry Andric     return SDValue();
38830b57cec5SDimitry Andric 
38840b57cec5SDimitry Andric   // srl i64:x, C for C >= 32
38850b57cec5SDimitry Andric   // =>
38860b57cec5SDimitry Andric   //   build_pair (srl hi_32(x), C - 32), 0
38870b57cec5SDimitry Andric   SDValue Zero = DAG.getConstant(0, SL, MVT::i32);
38880b57cec5SDimitry Andric 
3889349cc55cSDimitry Andric   SDValue Hi = getHiHalf64(LHS, DAG);
38900b57cec5SDimitry Andric 
38910b57cec5SDimitry Andric   SDValue NewConst = DAG.getConstant(ShiftAmt - 32, SL, MVT::i32);
38920b57cec5SDimitry Andric   SDValue NewShift = DAG.getNode(ISD::SRL, SL, MVT::i32, Hi, NewConst);
38930b57cec5SDimitry Andric 
38940b57cec5SDimitry Andric   SDValue BuildPair = DAG.getBuildVector(MVT::v2i32, SL, {NewShift, Zero});
38950b57cec5SDimitry Andric 
38960b57cec5SDimitry Andric   return DAG.getNode(ISD::BITCAST, SL, MVT::i64, BuildPair);
38970b57cec5SDimitry Andric }
38980b57cec5SDimitry Andric 
38990b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performTruncateCombine(
39000b57cec5SDimitry Andric   SDNode *N, DAGCombinerInfo &DCI) const {
39010b57cec5SDimitry Andric   SDLoc SL(N);
39020b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
39030b57cec5SDimitry Andric   EVT VT = N->getValueType(0);
39040b57cec5SDimitry Andric   SDValue Src = N->getOperand(0);
39050b57cec5SDimitry Andric 
39060b57cec5SDimitry Andric   // vt1 (truncate (bitcast (build_vector vt0:x, ...))) -> vt1 (bitcast vt0:x)
39070b57cec5SDimitry Andric   if (Src.getOpcode() == ISD::BITCAST && !VT.isVector()) {
39080b57cec5SDimitry Andric     SDValue Vec = Src.getOperand(0);
39090b57cec5SDimitry Andric     if (Vec.getOpcode() == ISD::BUILD_VECTOR) {
39100b57cec5SDimitry Andric       SDValue Elt0 = Vec.getOperand(0);
39110b57cec5SDimitry Andric       EVT EltVT = Elt0.getValueType();
3912e8d8bef9SDimitry Andric       if (VT.getFixedSizeInBits() <= EltVT.getFixedSizeInBits()) {
39130b57cec5SDimitry Andric         if (EltVT.isFloatingPoint()) {
39140b57cec5SDimitry Andric           Elt0 = DAG.getNode(ISD::BITCAST, SL,
39150b57cec5SDimitry Andric                              EltVT.changeTypeToInteger(), Elt0);
39160b57cec5SDimitry Andric         }
39170b57cec5SDimitry Andric 
39180b57cec5SDimitry Andric         return DAG.getNode(ISD::TRUNCATE, SL, VT, Elt0);
39190b57cec5SDimitry Andric       }
39200b57cec5SDimitry Andric     }
39210b57cec5SDimitry Andric   }
39220b57cec5SDimitry Andric 
39230b57cec5SDimitry Andric   // Equivalent of above for accessing the high element of a vector as an
39240b57cec5SDimitry Andric   // integer operation.
39250b57cec5SDimitry Andric   // trunc (srl (bitcast (build_vector x, y))), 16 -> trunc (bitcast y)
39260b57cec5SDimitry Andric   if (Src.getOpcode() == ISD::SRL && !VT.isVector()) {
39270b57cec5SDimitry Andric     if (auto K = isConstOrConstSplat(Src.getOperand(1))) {
39280b57cec5SDimitry Andric       if (2 * K->getZExtValue() == Src.getValueType().getScalarSizeInBits()) {
39290b57cec5SDimitry Andric         SDValue BV = stripBitcast(Src.getOperand(0));
39300b57cec5SDimitry Andric         if (BV.getOpcode() == ISD::BUILD_VECTOR &&
39310b57cec5SDimitry Andric             BV.getValueType().getVectorNumElements() == 2) {
39320b57cec5SDimitry Andric           SDValue SrcElt = BV.getOperand(1);
39330b57cec5SDimitry Andric           EVT SrcEltVT = SrcElt.getValueType();
39340b57cec5SDimitry Andric           if (SrcEltVT.isFloatingPoint()) {
39350b57cec5SDimitry Andric             SrcElt = DAG.getNode(ISD::BITCAST, SL,
39360b57cec5SDimitry Andric                                  SrcEltVT.changeTypeToInteger(), SrcElt);
39370b57cec5SDimitry Andric           }
39380b57cec5SDimitry Andric 
39390b57cec5SDimitry Andric           return DAG.getNode(ISD::TRUNCATE, SL, VT, SrcElt);
39400b57cec5SDimitry Andric         }
39410b57cec5SDimitry Andric       }
39420b57cec5SDimitry Andric     }
39430b57cec5SDimitry Andric   }
39440b57cec5SDimitry Andric 
39450b57cec5SDimitry Andric   // Partially shrink 64-bit shifts to 32-bit if reduced to 16-bit.
39460b57cec5SDimitry Andric   //
39470b57cec5SDimitry Andric   // i16 (trunc (srl i64:x, K)), K <= 16 ->
39480b57cec5SDimitry Andric   //     i16 (trunc (srl (i32 (trunc x), K)))
39490b57cec5SDimitry Andric   if (VT.getScalarSizeInBits() < 32) {
39500b57cec5SDimitry Andric     EVT SrcVT = Src.getValueType();
39510b57cec5SDimitry Andric     if (SrcVT.getScalarSizeInBits() > 32 &&
39520b57cec5SDimitry Andric         (Src.getOpcode() == ISD::SRL ||
39530b57cec5SDimitry Andric          Src.getOpcode() == ISD::SRA ||
39540b57cec5SDimitry Andric          Src.getOpcode() == ISD::SHL)) {
39550b57cec5SDimitry Andric       SDValue Amt = Src.getOperand(1);
39560b57cec5SDimitry Andric       KnownBits Known = DAG.computeKnownBits(Amt);
3957bdd1243dSDimitry Andric 
3958bdd1243dSDimitry Andric       // - For left shifts, do the transform as long as the shift
3959bdd1243dSDimitry Andric       //   amount is still legal for i32, so when ShiftAmt < 32 (<= 31)
3960bdd1243dSDimitry Andric       // - For right shift, do it if ShiftAmt <= (32 - Size) to avoid
3961bdd1243dSDimitry Andric       //   losing information stored in the high bits when truncating.
3962bdd1243dSDimitry Andric       const unsigned MaxCstSize =
3963bdd1243dSDimitry Andric           (Src.getOpcode() == ISD::SHL) ? 31 : (32 - VT.getScalarSizeInBits());
3964bdd1243dSDimitry Andric       if (Known.getMaxValue().ule(MaxCstSize)) {
39650b57cec5SDimitry Andric         EVT MidVT = VT.isVector() ?
39660b57cec5SDimitry Andric           EVT::getVectorVT(*DAG.getContext(), MVT::i32,
39670b57cec5SDimitry Andric                            VT.getVectorNumElements()) : MVT::i32;
39680b57cec5SDimitry Andric 
39690b57cec5SDimitry Andric         EVT NewShiftVT = getShiftAmountTy(MidVT, DAG.getDataLayout());
39700b57cec5SDimitry Andric         SDValue Trunc = DAG.getNode(ISD::TRUNCATE, SL, MidVT,
39710b57cec5SDimitry Andric                                     Src.getOperand(0));
39720b57cec5SDimitry Andric         DCI.AddToWorklist(Trunc.getNode());
39730b57cec5SDimitry Andric 
39740b57cec5SDimitry Andric         if (Amt.getValueType() != NewShiftVT) {
39750b57cec5SDimitry Andric           Amt = DAG.getZExtOrTrunc(Amt, SL, NewShiftVT);
39760b57cec5SDimitry Andric           DCI.AddToWorklist(Amt.getNode());
39770b57cec5SDimitry Andric         }
39780b57cec5SDimitry Andric 
39790b57cec5SDimitry Andric         SDValue ShrunkShift = DAG.getNode(Src.getOpcode(), SL, MidVT,
39800b57cec5SDimitry Andric                                           Trunc, Amt);
39810b57cec5SDimitry Andric         return DAG.getNode(ISD::TRUNCATE, SL, VT, ShrunkShift);
39820b57cec5SDimitry Andric       }
39830b57cec5SDimitry Andric     }
39840b57cec5SDimitry Andric   }
39850b57cec5SDimitry Andric 
39860b57cec5SDimitry Andric   return SDValue();
39870b57cec5SDimitry Andric }
39880b57cec5SDimitry Andric 
39890b57cec5SDimitry Andric // We need to specifically handle i64 mul here to avoid unnecessary conversion
39900b57cec5SDimitry Andric // instructions. If we only match on the legalized i64 mul expansion,
39910b57cec5SDimitry Andric // SimplifyDemandedBits will be unable to remove them because there will be
39920b57cec5SDimitry Andric // multiple uses due to the separate mul + mulh[su].
39930b57cec5SDimitry Andric static SDValue getMul24(SelectionDAG &DAG, const SDLoc &SL,
39940b57cec5SDimitry Andric                         SDValue N0, SDValue N1, unsigned Size, bool Signed) {
39950b57cec5SDimitry Andric   if (Size <= 32) {
39960b57cec5SDimitry Andric     unsigned MulOpc = Signed ? AMDGPUISD::MUL_I24 : AMDGPUISD::MUL_U24;
39970b57cec5SDimitry Andric     return DAG.getNode(MulOpc, SL, MVT::i32, N0, N1);
39980b57cec5SDimitry Andric   }
39990b57cec5SDimitry Andric 
4000e8d8bef9SDimitry Andric   unsigned MulLoOpc = Signed ? AMDGPUISD::MUL_I24 : AMDGPUISD::MUL_U24;
4001e8d8bef9SDimitry Andric   unsigned MulHiOpc = Signed ? AMDGPUISD::MULHI_I24 : AMDGPUISD::MULHI_U24;
40020b57cec5SDimitry Andric 
4003e8d8bef9SDimitry Andric   SDValue MulLo = DAG.getNode(MulLoOpc, SL, MVT::i32, N0, N1);
4004e8d8bef9SDimitry Andric   SDValue MulHi = DAG.getNode(MulHiOpc, SL, MVT::i32, N0, N1);
40050b57cec5SDimitry Andric 
4006e8d8bef9SDimitry Andric   return DAG.getNode(ISD::BUILD_PAIR, SL, MVT::i64, MulLo, MulHi);
40070b57cec5SDimitry Andric }
40080b57cec5SDimitry Andric 
400906c3fb27SDimitry Andric /// If \p V is an add of a constant 1, returns the other operand. Otherwise
401006c3fb27SDimitry Andric /// return SDValue().
401106c3fb27SDimitry Andric static SDValue getAddOneOp(const SDNode *V) {
401206c3fb27SDimitry Andric   if (V->getOpcode() != ISD::ADD)
401306c3fb27SDimitry Andric     return SDValue();
401406c3fb27SDimitry Andric 
401506c3fb27SDimitry Andric   auto *C = dyn_cast<ConstantSDNode>(V->getOperand(1));
401606c3fb27SDimitry Andric   return C && C->isOne() ? V->getOperand(0) : SDValue();
401706c3fb27SDimitry Andric }
401806c3fb27SDimitry Andric 
40190b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performMulCombine(SDNode *N,
40200b57cec5SDimitry Andric                                                 DAGCombinerInfo &DCI) const {
40210b57cec5SDimitry Andric   EVT VT = N->getValueType(0);
40220b57cec5SDimitry Andric 
4023fe6060f1SDimitry Andric   // Don't generate 24-bit multiplies on values that are in SGPRs, since
4024fe6060f1SDimitry Andric   // we only have a 32-bit scalar multiply (avoid values being moved to VGPRs
4025fe6060f1SDimitry Andric   // unnecessarily). isDivergent() is used as an approximation of whether the
4026fe6060f1SDimitry Andric   // value is in an SGPR.
4027fe6060f1SDimitry Andric   if (!N->isDivergent())
4028fe6060f1SDimitry Andric     return SDValue();
4029fe6060f1SDimitry Andric 
40300b57cec5SDimitry Andric   unsigned Size = VT.getSizeInBits();
40310b57cec5SDimitry Andric   if (VT.isVector() || Size > 64)
40320b57cec5SDimitry Andric     return SDValue();
40330b57cec5SDimitry Andric 
40340b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
40350b57cec5SDimitry Andric   SDLoc DL(N);
40360b57cec5SDimitry Andric 
40370b57cec5SDimitry Andric   SDValue N0 = N->getOperand(0);
40380b57cec5SDimitry Andric   SDValue N1 = N->getOperand(1);
40390b57cec5SDimitry Andric 
404006c3fb27SDimitry Andric   // Undo InstCombine canonicalize X * (Y + 1) -> X * Y + X to enable mad
404106c3fb27SDimitry Andric   // matching.
404206c3fb27SDimitry Andric 
404306c3fb27SDimitry Andric   // mul x, (add y, 1) -> add (mul x, y), x
404406c3fb27SDimitry Andric   auto IsFoldableAdd = [](SDValue V) -> SDValue {
404506c3fb27SDimitry Andric     SDValue AddOp = getAddOneOp(V.getNode());
404606c3fb27SDimitry Andric     if (!AddOp)
404706c3fb27SDimitry Andric       return SDValue();
404806c3fb27SDimitry Andric 
404906c3fb27SDimitry Andric     if (V.hasOneUse() || all_of(V->uses(), [](const SDNode *U) -> bool {
405006c3fb27SDimitry Andric           return U->getOpcode() == ISD::MUL;
405106c3fb27SDimitry Andric         }))
405206c3fb27SDimitry Andric       return AddOp;
405306c3fb27SDimitry Andric 
405406c3fb27SDimitry Andric     return SDValue();
405506c3fb27SDimitry Andric   };
405606c3fb27SDimitry Andric 
405706c3fb27SDimitry Andric   // FIXME: The selection pattern is not properly checking for commuted
405806c3fb27SDimitry Andric   // operands, so we have to place the mul in the LHS
405906c3fb27SDimitry Andric   if (SDValue MulOper = IsFoldableAdd(N0)) {
406006c3fb27SDimitry Andric     SDValue MulVal = DAG.getNode(N->getOpcode(), DL, VT, N1, MulOper);
406106c3fb27SDimitry Andric     return DAG.getNode(ISD::ADD, DL, VT, MulVal, N1);
406206c3fb27SDimitry Andric   }
406306c3fb27SDimitry Andric 
406406c3fb27SDimitry Andric   if (SDValue MulOper = IsFoldableAdd(N1)) {
406506c3fb27SDimitry Andric     SDValue MulVal = DAG.getNode(N->getOpcode(), DL, VT, N0, MulOper);
406606c3fb27SDimitry Andric     return DAG.getNode(ISD::ADD, DL, VT, MulVal, N0);
406706c3fb27SDimitry Andric   }
406806c3fb27SDimitry Andric 
406906c3fb27SDimitry Andric   // Skip if already mul24.
407006c3fb27SDimitry Andric   if (N->getOpcode() != ISD::MUL)
407106c3fb27SDimitry Andric     return SDValue();
407206c3fb27SDimitry Andric 
407306c3fb27SDimitry Andric   // There are i16 integer mul/mad.
407406c3fb27SDimitry Andric   if (Subtarget->has16BitInsts() && VT.getScalarType().bitsLE(MVT::i16))
407506c3fb27SDimitry Andric     return SDValue();
407606c3fb27SDimitry Andric 
40770b57cec5SDimitry Andric   // SimplifyDemandedBits has the annoying habit of turning useful zero_extends
40780b57cec5SDimitry Andric   // in the source into any_extends if the result of the mul is truncated. Since
40790b57cec5SDimitry Andric   // we can assume the high bits are whatever we want, use the underlying value
40800b57cec5SDimitry Andric   // to avoid the unknown high bits from interfering.
40810b57cec5SDimitry Andric   if (N0.getOpcode() == ISD::ANY_EXTEND)
40820b57cec5SDimitry Andric     N0 = N0.getOperand(0);
40830b57cec5SDimitry Andric 
40840b57cec5SDimitry Andric   if (N1.getOpcode() == ISD::ANY_EXTEND)
40850b57cec5SDimitry Andric     N1 = N1.getOperand(0);
40860b57cec5SDimitry Andric 
40870b57cec5SDimitry Andric   SDValue Mul;
40880b57cec5SDimitry Andric 
40890b57cec5SDimitry Andric   if (Subtarget->hasMulU24() && isU24(N0, DAG) && isU24(N1, DAG)) {
40900b57cec5SDimitry Andric     N0 = DAG.getZExtOrTrunc(N0, DL, MVT::i32);
40910b57cec5SDimitry Andric     N1 = DAG.getZExtOrTrunc(N1, DL, MVT::i32);
40920b57cec5SDimitry Andric     Mul = getMul24(DAG, DL, N0, N1, Size, false);
40930b57cec5SDimitry Andric   } else if (Subtarget->hasMulI24() && isI24(N0, DAG) && isI24(N1, DAG)) {
40940b57cec5SDimitry Andric     N0 = DAG.getSExtOrTrunc(N0, DL, MVT::i32);
40950b57cec5SDimitry Andric     N1 = DAG.getSExtOrTrunc(N1, DL, MVT::i32);
40960b57cec5SDimitry Andric     Mul = getMul24(DAG, DL, N0, N1, Size, true);
40970b57cec5SDimitry Andric   } else {
40980b57cec5SDimitry Andric     return SDValue();
40990b57cec5SDimitry Andric   }
41000b57cec5SDimitry Andric 
41010b57cec5SDimitry Andric   // We need to use sext even for MUL_U24, because MUL_U24 is used
41020b57cec5SDimitry Andric   // for signed multiply of 8 and 16-bit types.
41030b57cec5SDimitry Andric   return DAG.getSExtOrTrunc(Mul, DL, VT);
41040b57cec5SDimitry Andric }
41050b57cec5SDimitry Andric 
41064824e7fdSDimitry Andric SDValue
41074824e7fdSDimitry Andric AMDGPUTargetLowering::performMulLoHiCombine(SDNode *N,
41084824e7fdSDimitry Andric                                             DAGCombinerInfo &DCI) const {
41094824e7fdSDimitry Andric   if (N->getValueType(0) != MVT::i32)
41104824e7fdSDimitry Andric     return SDValue();
41114824e7fdSDimitry Andric 
41124824e7fdSDimitry Andric   SelectionDAG &DAG = DCI.DAG;
41134824e7fdSDimitry Andric   SDLoc DL(N);
41144824e7fdSDimitry Andric 
41154824e7fdSDimitry Andric   SDValue N0 = N->getOperand(0);
41164824e7fdSDimitry Andric   SDValue N1 = N->getOperand(1);
41174824e7fdSDimitry Andric 
41184824e7fdSDimitry Andric   // SimplifyDemandedBits has the annoying habit of turning useful zero_extends
41194824e7fdSDimitry Andric   // in the source into any_extends if the result of the mul is truncated. Since
41204824e7fdSDimitry Andric   // we can assume the high bits are whatever we want, use the underlying value
41214824e7fdSDimitry Andric   // to avoid the unknown high bits from interfering.
41224824e7fdSDimitry Andric   if (N0.getOpcode() == ISD::ANY_EXTEND)
41234824e7fdSDimitry Andric     N0 = N0.getOperand(0);
41244824e7fdSDimitry Andric   if (N1.getOpcode() == ISD::ANY_EXTEND)
41254824e7fdSDimitry Andric     N1 = N1.getOperand(0);
41264824e7fdSDimitry Andric 
41274824e7fdSDimitry Andric   // Try to use two fast 24-bit multiplies (one for each half of the result)
41284824e7fdSDimitry Andric   // instead of one slow extending multiply.
41294824e7fdSDimitry Andric   unsigned LoOpcode, HiOpcode;
41304824e7fdSDimitry Andric   if (Subtarget->hasMulU24() && isU24(N0, DAG) && isU24(N1, DAG)) {
41314824e7fdSDimitry Andric     N0 = DAG.getZExtOrTrunc(N0, DL, MVT::i32);
41324824e7fdSDimitry Andric     N1 = DAG.getZExtOrTrunc(N1, DL, MVT::i32);
41334824e7fdSDimitry Andric     LoOpcode = AMDGPUISD::MUL_U24;
41344824e7fdSDimitry Andric     HiOpcode = AMDGPUISD::MULHI_U24;
41354824e7fdSDimitry Andric   } else if (Subtarget->hasMulI24() && isI24(N0, DAG) && isI24(N1, DAG)) {
41364824e7fdSDimitry Andric     N0 = DAG.getSExtOrTrunc(N0, DL, MVT::i32);
41374824e7fdSDimitry Andric     N1 = DAG.getSExtOrTrunc(N1, DL, MVT::i32);
41384824e7fdSDimitry Andric     LoOpcode = AMDGPUISD::MUL_I24;
41394824e7fdSDimitry Andric     HiOpcode = AMDGPUISD::MULHI_I24;
41404824e7fdSDimitry Andric   } else {
41414824e7fdSDimitry Andric     return SDValue();
41424824e7fdSDimitry Andric   }
41434824e7fdSDimitry Andric 
41444824e7fdSDimitry Andric   SDValue Lo = DAG.getNode(LoOpcode, DL, MVT::i32, N0, N1);
41454824e7fdSDimitry Andric   SDValue Hi = DAG.getNode(HiOpcode, DL, MVT::i32, N0, N1);
41464824e7fdSDimitry Andric   DCI.CombineTo(N, Lo, Hi);
41474824e7fdSDimitry Andric   return SDValue(N, 0);
41484824e7fdSDimitry Andric }
41494824e7fdSDimitry Andric 
41500b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performMulhsCombine(SDNode *N,
41510b57cec5SDimitry Andric                                                   DAGCombinerInfo &DCI) const {
41520b57cec5SDimitry Andric   EVT VT = N->getValueType(0);
41530b57cec5SDimitry Andric 
41540b57cec5SDimitry Andric   if (!Subtarget->hasMulI24() || VT.isVector())
41550b57cec5SDimitry Andric     return SDValue();
41560b57cec5SDimitry Andric 
4157fe6060f1SDimitry Andric   // Don't generate 24-bit multiplies on values that are in SGPRs, since
4158fe6060f1SDimitry Andric   // we only have a 32-bit scalar multiply (avoid values being moved to VGPRs
4159fe6060f1SDimitry Andric   // unnecessarily). isDivergent() is used as an approximation of whether the
4160fe6060f1SDimitry Andric   // value is in an SGPR.
4161fe6060f1SDimitry Andric   // This doesn't apply if no s_mul_hi is available (since we'll end up with a
4162fe6060f1SDimitry Andric   // valu op anyway)
4163fe6060f1SDimitry Andric   if (Subtarget->hasSMulHi() && !N->isDivergent())
4164fe6060f1SDimitry Andric     return SDValue();
4165fe6060f1SDimitry Andric 
41660b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
41670b57cec5SDimitry Andric   SDLoc DL(N);
41680b57cec5SDimitry Andric 
41690b57cec5SDimitry Andric   SDValue N0 = N->getOperand(0);
41700b57cec5SDimitry Andric   SDValue N1 = N->getOperand(1);
41710b57cec5SDimitry Andric 
41720b57cec5SDimitry Andric   if (!isI24(N0, DAG) || !isI24(N1, DAG))
41730b57cec5SDimitry Andric     return SDValue();
41740b57cec5SDimitry Andric 
41750b57cec5SDimitry Andric   N0 = DAG.getSExtOrTrunc(N0, DL, MVT::i32);
41760b57cec5SDimitry Andric   N1 = DAG.getSExtOrTrunc(N1, DL, MVT::i32);
41770b57cec5SDimitry Andric 
41780b57cec5SDimitry Andric   SDValue Mulhi = DAG.getNode(AMDGPUISD::MULHI_I24, DL, MVT::i32, N0, N1);
41790b57cec5SDimitry Andric   DCI.AddToWorklist(Mulhi.getNode());
41800b57cec5SDimitry Andric   return DAG.getSExtOrTrunc(Mulhi, DL, VT);
41810b57cec5SDimitry Andric }
41820b57cec5SDimitry Andric 
41830b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performMulhuCombine(SDNode *N,
41840b57cec5SDimitry Andric                                                   DAGCombinerInfo &DCI) const {
41850b57cec5SDimitry Andric   EVT VT = N->getValueType(0);
41860b57cec5SDimitry Andric 
41870b57cec5SDimitry Andric   if (!Subtarget->hasMulU24() || VT.isVector() || VT.getSizeInBits() > 32)
41880b57cec5SDimitry Andric     return SDValue();
41890b57cec5SDimitry Andric 
4190fe6060f1SDimitry Andric   // Don't generate 24-bit multiplies on values that are in SGPRs, since
4191fe6060f1SDimitry Andric   // we only have a 32-bit scalar multiply (avoid values being moved to VGPRs
4192fe6060f1SDimitry Andric   // unnecessarily). isDivergent() is used as an approximation of whether the
4193fe6060f1SDimitry Andric   // value is in an SGPR.
4194fe6060f1SDimitry Andric   // This doesn't apply if no s_mul_hi is available (since we'll end up with a
4195fe6060f1SDimitry Andric   // valu op anyway)
4196fe6060f1SDimitry Andric   if (Subtarget->hasSMulHi() && !N->isDivergent())
4197fe6060f1SDimitry Andric     return SDValue();
4198fe6060f1SDimitry Andric 
41990b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
42000b57cec5SDimitry Andric   SDLoc DL(N);
42010b57cec5SDimitry Andric 
42020b57cec5SDimitry Andric   SDValue N0 = N->getOperand(0);
42030b57cec5SDimitry Andric   SDValue N1 = N->getOperand(1);
42040b57cec5SDimitry Andric 
42050b57cec5SDimitry Andric   if (!isU24(N0, DAG) || !isU24(N1, DAG))
42060b57cec5SDimitry Andric     return SDValue();
42070b57cec5SDimitry Andric 
42080b57cec5SDimitry Andric   N0 = DAG.getZExtOrTrunc(N0, DL, MVT::i32);
42090b57cec5SDimitry Andric   N1 = DAG.getZExtOrTrunc(N1, DL, MVT::i32);
42100b57cec5SDimitry Andric 
42110b57cec5SDimitry Andric   SDValue Mulhi = DAG.getNode(AMDGPUISD::MULHI_U24, DL, MVT::i32, N0, N1);
42120b57cec5SDimitry Andric   DCI.AddToWorklist(Mulhi.getNode());
42130b57cec5SDimitry Andric   return DAG.getZExtOrTrunc(Mulhi, DL, VT);
42140b57cec5SDimitry Andric }
42150b57cec5SDimitry Andric 
42160b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::getFFBX_U32(SelectionDAG &DAG,
42170b57cec5SDimitry Andric                                           SDValue Op,
42180b57cec5SDimitry Andric                                           const SDLoc &DL,
42190b57cec5SDimitry Andric                                           unsigned Opc) const {
42200b57cec5SDimitry Andric   EVT VT = Op.getValueType();
42210b57cec5SDimitry Andric   EVT LegalVT = getTypeToTransformTo(*DAG.getContext(), VT);
42220b57cec5SDimitry Andric   if (LegalVT != MVT::i32 && (Subtarget->has16BitInsts() &&
42230b57cec5SDimitry Andric                               LegalVT != MVT::i16))
42240b57cec5SDimitry Andric     return SDValue();
42250b57cec5SDimitry Andric 
42260b57cec5SDimitry Andric   if (VT != MVT::i32)
42270b57cec5SDimitry Andric     Op = DAG.getNode(ISD::ZERO_EXTEND, DL, MVT::i32, Op);
42280b57cec5SDimitry Andric 
42290b57cec5SDimitry Andric   SDValue FFBX = DAG.getNode(Opc, DL, MVT::i32, Op);
42300b57cec5SDimitry Andric   if (VT != MVT::i32)
42310b57cec5SDimitry Andric     FFBX = DAG.getNode(ISD::TRUNCATE, DL, VT, FFBX);
42320b57cec5SDimitry Andric 
42330b57cec5SDimitry Andric   return FFBX;
42340b57cec5SDimitry Andric }
42350b57cec5SDimitry Andric 
42360b57cec5SDimitry Andric // The native instructions return -1 on 0 input. Optimize out a select that
42370b57cec5SDimitry Andric // produces -1 on 0.
42380b57cec5SDimitry Andric //
42390b57cec5SDimitry Andric // TODO: If zero is not undef, we could also do this if the output is compared
42400b57cec5SDimitry Andric // against the bitwidth.
42410b57cec5SDimitry Andric //
42420b57cec5SDimitry Andric // TODO: Should probably combine against FFBH_U32 instead of ctlz directly.
42430b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performCtlz_CttzCombine(const SDLoc &SL, SDValue Cond,
42440b57cec5SDimitry Andric                                                  SDValue LHS, SDValue RHS,
42450b57cec5SDimitry Andric                                                  DAGCombinerInfo &DCI) const {
42460b57cec5SDimitry Andric   ConstantSDNode *CmpRhs = dyn_cast<ConstantSDNode>(Cond.getOperand(1));
4247349cc55cSDimitry Andric   if (!CmpRhs || !CmpRhs->isZero())
42480b57cec5SDimitry Andric     return SDValue();
42490b57cec5SDimitry Andric 
42500b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
42510b57cec5SDimitry Andric   ISD::CondCode CCOpcode = cast<CondCodeSDNode>(Cond.getOperand(2))->get();
42520b57cec5SDimitry Andric   SDValue CmpLHS = Cond.getOperand(0);
42530b57cec5SDimitry Andric 
42540b57cec5SDimitry Andric   // select (setcc x, 0, eq), -1, (ctlz_zero_undef x) -> ffbh_u32 x
42550b57cec5SDimitry Andric   // select (setcc x, 0, eq), -1, (cttz_zero_undef x) -> ffbl_u32 x
42560b57cec5SDimitry Andric   if (CCOpcode == ISD::SETEQ &&
42570b57cec5SDimitry Andric       (isCtlzOpc(RHS.getOpcode()) || isCttzOpc(RHS.getOpcode())) &&
425806c3fb27SDimitry Andric       RHS.getOperand(0) == CmpLHS && isAllOnesConstant(LHS)) {
42595ffd83dbSDimitry Andric     unsigned Opc =
42605ffd83dbSDimitry Andric         isCttzOpc(RHS.getOpcode()) ? AMDGPUISD::FFBL_B32 : AMDGPUISD::FFBH_U32;
42610b57cec5SDimitry Andric     return getFFBX_U32(DAG, CmpLHS, SL, Opc);
42620b57cec5SDimitry Andric   }
42630b57cec5SDimitry Andric 
42640b57cec5SDimitry Andric   // select (setcc x, 0, ne), (ctlz_zero_undef x), -1 -> ffbh_u32 x
42650b57cec5SDimitry Andric   // select (setcc x, 0, ne), (cttz_zero_undef x), -1 -> ffbl_u32 x
42660b57cec5SDimitry Andric   if (CCOpcode == ISD::SETNE &&
42675ffd83dbSDimitry Andric       (isCtlzOpc(LHS.getOpcode()) || isCttzOpc(LHS.getOpcode())) &&
426806c3fb27SDimitry Andric       LHS.getOperand(0) == CmpLHS && isAllOnesConstant(RHS)) {
42695ffd83dbSDimitry Andric     unsigned Opc =
42705ffd83dbSDimitry Andric         isCttzOpc(LHS.getOpcode()) ? AMDGPUISD::FFBL_B32 : AMDGPUISD::FFBH_U32;
42715ffd83dbSDimitry Andric 
42720b57cec5SDimitry Andric     return getFFBX_U32(DAG, CmpLHS, SL, Opc);
42730b57cec5SDimitry Andric   }
42740b57cec5SDimitry Andric 
42750b57cec5SDimitry Andric   return SDValue();
42760b57cec5SDimitry Andric }
42770b57cec5SDimitry Andric 
42780b57cec5SDimitry Andric static SDValue distributeOpThroughSelect(TargetLowering::DAGCombinerInfo &DCI,
42790b57cec5SDimitry Andric                                          unsigned Op,
42800b57cec5SDimitry Andric                                          const SDLoc &SL,
42810b57cec5SDimitry Andric                                          SDValue Cond,
42820b57cec5SDimitry Andric                                          SDValue N1,
42830b57cec5SDimitry Andric                                          SDValue N2) {
42840b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
42850b57cec5SDimitry Andric   EVT VT = N1.getValueType();
42860b57cec5SDimitry Andric 
42870b57cec5SDimitry Andric   SDValue NewSelect = DAG.getNode(ISD::SELECT, SL, VT, Cond,
42880b57cec5SDimitry Andric                                   N1.getOperand(0), N2.getOperand(0));
42890b57cec5SDimitry Andric   DCI.AddToWorklist(NewSelect.getNode());
42900b57cec5SDimitry Andric   return DAG.getNode(Op, SL, VT, NewSelect);
42910b57cec5SDimitry Andric }
42920b57cec5SDimitry Andric 
42930b57cec5SDimitry Andric // Pull a free FP operation out of a select so it may fold into uses.
42940b57cec5SDimitry Andric //
42950b57cec5SDimitry Andric // select c, (fneg x), (fneg y) -> fneg (select c, x, y)
42960b57cec5SDimitry Andric // select c, (fneg x), k -> fneg (select c, x, (fneg k))
42970b57cec5SDimitry Andric //
42980b57cec5SDimitry Andric // select c, (fabs x), (fabs y) -> fabs (select c, x, y)
42990b57cec5SDimitry Andric // select c, (fabs x), +k -> fabs (select c, x, k)
430006c3fb27SDimitry Andric SDValue
430106c3fb27SDimitry Andric AMDGPUTargetLowering::foldFreeOpFromSelect(TargetLowering::DAGCombinerInfo &DCI,
430206c3fb27SDimitry Andric                                            SDValue N) const {
43030b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
43040b57cec5SDimitry Andric   SDValue Cond = N.getOperand(0);
43050b57cec5SDimitry Andric   SDValue LHS = N.getOperand(1);
43060b57cec5SDimitry Andric   SDValue RHS = N.getOperand(2);
43070b57cec5SDimitry Andric 
43080b57cec5SDimitry Andric   EVT VT = N.getValueType();
43090b57cec5SDimitry Andric   if ((LHS.getOpcode() == ISD::FABS && RHS.getOpcode() == ISD::FABS) ||
43100b57cec5SDimitry Andric       (LHS.getOpcode() == ISD::FNEG && RHS.getOpcode() == ISD::FNEG)) {
431106c3fb27SDimitry Andric     if (!AMDGPUTargetLowering::allUsesHaveSourceMods(N.getNode()))
431206c3fb27SDimitry Andric       return SDValue();
431306c3fb27SDimitry Andric 
43140b57cec5SDimitry Andric     return distributeOpThroughSelect(DCI, LHS.getOpcode(),
43150b57cec5SDimitry Andric                                      SDLoc(N), Cond, LHS, RHS);
43160b57cec5SDimitry Andric   }
43170b57cec5SDimitry Andric 
43180b57cec5SDimitry Andric   bool Inv = false;
43190b57cec5SDimitry Andric   if (RHS.getOpcode() == ISD::FABS || RHS.getOpcode() == ISD::FNEG) {
43200b57cec5SDimitry Andric     std::swap(LHS, RHS);
43210b57cec5SDimitry Andric     Inv = true;
43220b57cec5SDimitry Andric   }
43230b57cec5SDimitry Andric 
43240b57cec5SDimitry Andric   // TODO: Support vector constants.
43250b57cec5SDimitry Andric   ConstantFPSDNode *CRHS = dyn_cast<ConstantFPSDNode>(RHS);
432606c3fb27SDimitry Andric   if ((LHS.getOpcode() == ISD::FNEG || LHS.getOpcode() == ISD::FABS) && CRHS &&
432706c3fb27SDimitry Andric       !selectSupportsSourceMods(N.getNode())) {
43280b57cec5SDimitry Andric     SDLoc SL(N);
43290b57cec5SDimitry Andric     // If one side is an fneg/fabs and the other is a constant, we can push the
43300b57cec5SDimitry Andric     // fneg/fabs down. If it's an fabs, the constant needs to be non-negative.
43310b57cec5SDimitry Andric     SDValue NewLHS = LHS.getOperand(0);
43320b57cec5SDimitry Andric     SDValue NewRHS = RHS;
43330b57cec5SDimitry Andric 
43340b57cec5SDimitry Andric     // Careful: if the neg can be folded up, don't try to pull it back down.
43350b57cec5SDimitry Andric     bool ShouldFoldNeg = true;
43360b57cec5SDimitry Andric 
43370b57cec5SDimitry Andric     if (NewLHS.hasOneUse()) {
43380b57cec5SDimitry Andric       unsigned Opc = NewLHS.getOpcode();
433906c3fb27SDimitry Andric       if (LHS.getOpcode() == ISD::FNEG && fnegFoldsIntoOp(NewLHS.getNode()))
43400b57cec5SDimitry Andric         ShouldFoldNeg = false;
43410b57cec5SDimitry Andric       if (LHS.getOpcode() == ISD::FABS && Opc == ISD::FMUL)
43420b57cec5SDimitry Andric         ShouldFoldNeg = false;
43430b57cec5SDimitry Andric     }
43440b57cec5SDimitry Andric 
43450b57cec5SDimitry Andric     if (ShouldFoldNeg) {
434606c3fb27SDimitry Andric       if (LHS.getOpcode() == ISD::FABS && CRHS->isNegative())
434706c3fb27SDimitry Andric         return SDValue();
434806c3fb27SDimitry Andric 
434906c3fb27SDimitry Andric       // We're going to be forced to use a source modifier anyway, there's no
435006c3fb27SDimitry Andric       // point to pulling the negate out unless we can get a size reduction by
435106c3fb27SDimitry Andric       // negating the constant.
435206c3fb27SDimitry Andric       //
435306c3fb27SDimitry Andric       // TODO: Generalize to use getCheaperNegatedExpression which doesn't know
435406c3fb27SDimitry Andric       // about cheaper constants.
435506c3fb27SDimitry Andric       if (NewLHS.getOpcode() == ISD::FABS &&
435606c3fb27SDimitry Andric           getConstantNegateCost(CRHS) != NegatibleCost::Cheaper)
435706c3fb27SDimitry Andric         return SDValue();
435806c3fb27SDimitry Andric 
435906c3fb27SDimitry Andric       if (!AMDGPUTargetLowering::allUsesHaveSourceMods(N.getNode()))
436006c3fb27SDimitry Andric         return SDValue();
436106c3fb27SDimitry Andric 
43620b57cec5SDimitry Andric       if (LHS.getOpcode() == ISD::FNEG)
43630b57cec5SDimitry Andric         NewRHS = DAG.getNode(ISD::FNEG, SL, VT, RHS);
43640b57cec5SDimitry Andric 
43650b57cec5SDimitry Andric       if (Inv)
43660b57cec5SDimitry Andric         std::swap(NewLHS, NewRHS);
43670b57cec5SDimitry Andric 
43680b57cec5SDimitry Andric       SDValue NewSelect = DAG.getNode(ISD::SELECT, SL, VT,
43690b57cec5SDimitry Andric                                       Cond, NewLHS, NewRHS);
43700b57cec5SDimitry Andric       DCI.AddToWorklist(NewSelect.getNode());
43710b57cec5SDimitry Andric       return DAG.getNode(LHS.getOpcode(), SL, VT, NewSelect);
43720b57cec5SDimitry Andric     }
43730b57cec5SDimitry Andric   }
43740b57cec5SDimitry Andric 
43750b57cec5SDimitry Andric   return SDValue();
43760b57cec5SDimitry Andric }
43770b57cec5SDimitry Andric 
43780b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performSelectCombine(SDNode *N,
43790b57cec5SDimitry Andric                                                    DAGCombinerInfo &DCI) const {
43800b57cec5SDimitry Andric   if (SDValue Folded = foldFreeOpFromSelect(DCI, SDValue(N, 0)))
43810b57cec5SDimitry Andric     return Folded;
43820b57cec5SDimitry Andric 
43830b57cec5SDimitry Andric   SDValue Cond = N->getOperand(0);
43840b57cec5SDimitry Andric   if (Cond.getOpcode() != ISD::SETCC)
43850b57cec5SDimitry Andric     return SDValue();
43860b57cec5SDimitry Andric 
43870b57cec5SDimitry Andric   EVT VT = N->getValueType(0);
43880b57cec5SDimitry Andric   SDValue LHS = Cond.getOperand(0);
43890b57cec5SDimitry Andric   SDValue RHS = Cond.getOperand(1);
43900b57cec5SDimitry Andric   SDValue CC = Cond.getOperand(2);
43910b57cec5SDimitry Andric 
43920b57cec5SDimitry Andric   SDValue True = N->getOperand(1);
43930b57cec5SDimitry Andric   SDValue False = N->getOperand(2);
43940b57cec5SDimitry Andric 
43950b57cec5SDimitry Andric   if (Cond.hasOneUse()) { // TODO: Look for multiple select uses.
43960b57cec5SDimitry Andric     SelectionDAG &DAG = DCI.DAG;
43970b57cec5SDimitry Andric     if (DAG.isConstantValueOfAnyType(True) &&
43980b57cec5SDimitry Andric         !DAG.isConstantValueOfAnyType(False)) {
43990b57cec5SDimitry Andric       // Swap cmp + select pair to move constant to false input.
44000b57cec5SDimitry Andric       // This will allow using VOPC cndmasks more often.
44010b57cec5SDimitry Andric       // select (setcc x, y), k, x -> select (setccinv x, y), x, k
44020b57cec5SDimitry Andric 
44030b57cec5SDimitry Andric       SDLoc SL(N);
4404480093f4SDimitry Andric       ISD::CondCode NewCC =
4405480093f4SDimitry Andric           getSetCCInverse(cast<CondCodeSDNode>(CC)->get(), LHS.getValueType());
44060b57cec5SDimitry Andric 
44070b57cec5SDimitry Andric       SDValue NewCond = DAG.getSetCC(SL, Cond.getValueType(), LHS, RHS, NewCC);
44080b57cec5SDimitry Andric       return DAG.getNode(ISD::SELECT, SL, VT, NewCond, False, True);
44090b57cec5SDimitry Andric     }
44100b57cec5SDimitry Andric 
44110b57cec5SDimitry Andric     if (VT == MVT::f32 && Subtarget->hasFminFmaxLegacy()) {
44120b57cec5SDimitry Andric       SDValue MinMax
44130b57cec5SDimitry Andric         = combineFMinMaxLegacy(SDLoc(N), VT, LHS, RHS, True, False, CC, DCI);
44140b57cec5SDimitry Andric       // Revisit this node so we can catch min3/max3/med3 patterns.
44150b57cec5SDimitry Andric       //DCI.AddToWorklist(MinMax.getNode());
44160b57cec5SDimitry Andric       return MinMax;
44170b57cec5SDimitry Andric     }
44180b57cec5SDimitry Andric   }
44190b57cec5SDimitry Andric 
44200b57cec5SDimitry Andric   // There's no reason to not do this if the condition has other uses.
44210b57cec5SDimitry Andric   return performCtlz_CttzCombine(SDLoc(N), Cond, True, False, DCI);
44220b57cec5SDimitry Andric }
44230b57cec5SDimitry Andric 
44240b57cec5SDimitry Andric static bool isInv2Pi(const APFloat &APF) {
44250b57cec5SDimitry Andric   static const APFloat KF16(APFloat::IEEEhalf(), APInt(16, 0x3118));
44260b57cec5SDimitry Andric   static const APFloat KF32(APFloat::IEEEsingle(), APInt(32, 0x3e22f983));
44270b57cec5SDimitry Andric   static const APFloat KF64(APFloat::IEEEdouble(), APInt(64, 0x3fc45f306dc9c882));
44280b57cec5SDimitry Andric 
44290b57cec5SDimitry Andric   return APF.bitwiseIsEqual(KF16) ||
44300b57cec5SDimitry Andric          APF.bitwiseIsEqual(KF32) ||
44310b57cec5SDimitry Andric          APF.bitwiseIsEqual(KF64);
44320b57cec5SDimitry Andric }
44330b57cec5SDimitry Andric 
44340b57cec5SDimitry Andric // 0 and 1.0 / (0.5 * pi) do not have inline immmediates, so there is an
44350b57cec5SDimitry Andric // additional cost to negate them.
443606c3fb27SDimitry Andric TargetLowering::NegatibleCost
443706c3fb27SDimitry Andric AMDGPUTargetLowering::getConstantNegateCost(const ConstantFPSDNode *C) const {
443806c3fb27SDimitry Andric   if (C->isZero())
443906c3fb27SDimitry Andric     return C->isNegative() ? NegatibleCost::Cheaper : NegatibleCost::Expensive;
44400b57cec5SDimitry Andric 
44410b57cec5SDimitry Andric   if (Subtarget->hasInv2PiInlineImm() && isInv2Pi(C->getValueAPF()))
444206c3fb27SDimitry Andric     return C->isNegative() ? NegatibleCost::Cheaper : NegatibleCost::Expensive;
444306c3fb27SDimitry Andric 
444406c3fb27SDimitry Andric   return NegatibleCost::Neutral;
44450b57cec5SDimitry Andric }
44460b57cec5SDimitry Andric 
444706c3fb27SDimitry Andric bool AMDGPUTargetLowering::isConstantCostlierToNegate(SDValue N) const {
444806c3fb27SDimitry Andric   if (const ConstantFPSDNode *C = isConstOrConstSplatFP(N))
444906c3fb27SDimitry Andric     return getConstantNegateCost(C) == NegatibleCost::Expensive;
445006c3fb27SDimitry Andric   return false;
445106c3fb27SDimitry Andric }
445206c3fb27SDimitry Andric 
445306c3fb27SDimitry Andric bool AMDGPUTargetLowering::isConstantCheaperToNegate(SDValue N) const {
445406c3fb27SDimitry Andric   if (const ConstantFPSDNode *C = isConstOrConstSplatFP(N))
445506c3fb27SDimitry Andric     return getConstantNegateCost(C) == NegatibleCost::Cheaper;
44560b57cec5SDimitry Andric   return false;
44570b57cec5SDimitry Andric }
44580b57cec5SDimitry Andric 
44590b57cec5SDimitry Andric static unsigned inverseMinMax(unsigned Opc) {
44600b57cec5SDimitry Andric   switch (Opc) {
44610b57cec5SDimitry Andric   case ISD::FMAXNUM:
44620b57cec5SDimitry Andric     return ISD::FMINNUM;
44630b57cec5SDimitry Andric   case ISD::FMINNUM:
44640b57cec5SDimitry Andric     return ISD::FMAXNUM;
44650b57cec5SDimitry Andric   case ISD::FMAXNUM_IEEE:
44660b57cec5SDimitry Andric     return ISD::FMINNUM_IEEE;
44670b57cec5SDimitry Andric   case ISD::FMINNUM_IEEE:
44680b57cec5SDimitry Andric     return ISD::FMAXNUM_IEEE;
44690b57cec5SDimitry Andric   case AMDGPUISD::FMAX_LEGACY:
44700b57cec5SDimitry Andric     return AMDGPUISD::FMIN_LEGACY;
44710b57cec5SDimitry Andric   case AMDGPUISD::FMIN_LEGACY:
44720b57cec5SDimitry Andric     return  AMDGPUISD::FMAX_LEGACY;
44730b57cec5SDimitry Andric   default:
44740b57cec5SDimitry Andric     llvm_unreachable("invalid min/max opcode");
44750b57cec5SDimitry Andric   }
44760b57cec5SDimitry Andric }
44770b57cec5SDimitry Andric 
447806c3fb27SDimitry Andric /// \return true if it's profitable to try to push an fneg into its source
447906c3fb27SDimitry Andric /// instruction.
448006c3fb27SDimitry Andric bool AMDGPUTargetLowering::shouldFoldFNegIntoSrc(SDNode *N, SDValue N0) {
44810b57cec5SDimitry Andric   // If the input has multiple uses and we can either fold the negate down, or
44820b57cec5SDimitry Andric   // the other uses cannot, give up. This both prevents unprofitable
44830b57cec5SDimitry Andric   // transformations and infinite loops: we won't repeatedly try to fold around
44840b57cec5SDimitry Andric   // a negate that has no 'good' form.
44850b57cec5SDimitry Andric   if (N0.hasOneUse()) {
44860b57cec5SDimitry Andric     // This may be able to fold into the source, but at a code size cost. Don't
44870b57cec5SDimitry Andric     // fold if the fold into the user is free.
44880b57cec5SDimitry Andric     if (allUsesHaveSourceMods(N, 0))
448906c3fb27SDimitry Andric       return false;
44900b57cec5SDimitry Andric   } else {
449106c3fb27SDimitry Andric     if (fnegFoldsIntoOp(N0.getNode()) &&
44920b57cec5SDimitry Andric         (allUsesHaveSourceMods(N) || !allUsesHaveSourceMods(N0.getNode())))
449306c3fb27SDimitry Andric       return false;
44940b57cec5SDimitry Andric   }
44950b57cec5SDimitry Andric 
449606c3fb27SDimitry Andric   return true;
449706c3fb27SDimitry Andric }
449806c3fb27SDimitry Andric 
449906c3fb27SDimitry Andric SDValue AMDGPUTargetLowering::performFNegCombine(SDNode *N,
450006c3fb27SDimitry Andric                                                  DAGCombinerInfo &DCI) const {
450106c3fb27SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
450206c3fb27SDimitry Andric   SDValue N0 = N->getOperand(0);
450306c3fb27SDimitry Andric   EVT VT = N->getValueType(0);
450406c3fb27SDimitry Andric 
450506c3fb27SDimitry Andric   unsigned Opc = N0.getOpcode();
450606c3fb27SDimitry Andric 
450706c3fb27SDimitry Andric   if (!shouldFoldFNegIntoSrc(N, N0))
450806c3fb27SDimitry Andric     return SDValue();
450906c3fb27SDimitry Andric 
45100b57cec5SDimitry Andric   SDLoc SL(N);
45110b57cec5SDimitry Andric   switch (Opc) {
45120b57cec5SDimitry Andric   case ISD::FADD: {
45130b57cec5SDimitry Andric     if (!mayIgnoreSignedZero(N0))
45140b57cec5SDimitry Andric       return SDValue();
45150b57cec5SDimitry Andric 
45160b57cec5SDimitry Andric     // (fneg (fadd x, y)) -> (fadd (fneg x), (fneg y))
45170b57cec5SDimitry Andric     SDValue LHS = N0.getOperand(0);
45180b57cec5SDimitry Andric     SDValue RHS = N0.getOperand(1);
45190b57cec5SDimitry Andric 
45200b57cec5SDimitry Andric     if (LHS.getOpcode() != ISD::FNEG)
45210b57cec5SDimitry Andric       LHS = DAG.getNode(ISD::FNEG, SL, VT, LHS);
45220b57cec5SDimitry Andric     else
45230b57cec5SDimitry Andric       LHS = LHS.getOperand(0);
45240b57cec5SDimitry Andric 
45250b57cec5SDimitry Andric     if (RHS.getOpcode() != ISD::FNEG)
45260b57cec5SDimitry Andric       RHS = DAG.getNode(ISD::FNEG, SL, VT, RHS);
45270b57cec5SDimitry Andric     else
45280b57cec5SDimitry Andric       RHS = RHS.getOperand(0);
45290b57cec5SDimitry Andric 
45300b57cec5SDimitry Andric     SDValue Res = DAG.getNode(ISD::FADD, SL, VT, LHS, RHS, N0->getFlags());
45310b57cec5SDimitry Andric     if (Res.getOpcode() != ISD::FADD)
45320b57cec5SDimitry Andric       return SDValue(); // Op got folded away.
45330b57cec5SDimitry Andric     if (!N0.hasOneUse())
45340b57cec5SDimitry Andric       DAG.ReplaceAllUsesWith(N0, DAG.getNode(ISD::FNEG, SL, VT, Res));
45350b57cec5SDimitry Andric     return Res;
45360b57cec5SDimitry Andric   }
45370b57cec5SDimitry Andric   case ISD::FMUL:
45380b57cec5SDimitry Andric   case AMDGPUISD::FMUL_LEGACY: {
45390b57cec5SDimitry Andric     // (fneg (fmul x, y)) -> (fmul x, (fneg y))
45400b57cec5SDimitry Andric     // (fneg (fmul_legacy x, y)) -> (fmul_legacy x, (fneg y))
45410b57cec5SDimitry Andric     SDValue LHS = N0.getOperand(0);
45420b57cec5SDimitry Andric     SDValue RHS = N0.getOperand(1);
45430b57cec5SDimitry Andric 
45440b57cec5SDimitry Andric     if (LHS.getOpcode() == ISD::FNEG)
45450b57cec5SDimitry Andric       LHS = LHS.getOperand(0);
45460b57cec5SDimitry Andric     else if (RHS.getOpcode() == ISD::FNEG)
45470b57cec5SDimitry Andric       RHS = RHS.getOperand(0);
45480b57cec5SDimitry Andric     else
45490b57cec5SDimitry Andric       RHS = DAG.getNode(ISD::FNEG, SL, VT, RHS);
45500b57cec5SDimitry Andric 
45510b57cec5SDimitry Andric     SDValue Res = DAG.getNode(Opc, SL, VT, LHS, RHS, N0->getFlags());
45520b57cec5SDimitry Andric     if (Res.getOpcode() != Opc)
45530b57cec5SDimitry Andric       return SDValue(); // Op got folded away.
45540b57cec5SDimitry Andric     if (!N0.hasOneUse())
45550b57cec5SDimitry Andric       DAG.ReplaceAllUsesWith(N0, DAG.getNode(ISD::FNEG, SL, VT, Res));
45560b57cec5SDimitry Andric     return Res;
45570b57cec5SDimitry Andric   }
45580b57cec5SDimitry Andric   case ISD::FMA:
45590b57cec5SDimitry Andric   case ISD::FMAD: {
4560e8d8bef9SDimitry Andric     // TODO: handle llvm.amdgcn.fma.legacy
45610b57cec5SDimitry Andric     if (!mayIgnoreSignedZero(N0))
45620b57cec5SDimitry Andric       return SDValue();
45630b57cec5SDimitry Andric 
45640b57cec5SDimitry Andric     // (fneg (fma x, y, z)) -> (fma x, (fneg y), (fneg z))
45650b57cec5SDimitry Andric     SDValue LHS = N0.getOperand(0);
45660b57cec5SDimitry Andric     SDValue MHS = N0.getOperand(1);
45670b57cec5SDimitry Andric     SDValue RHS = N0.getOperand(2);
45680b57cec5SDimitry Andric 
45690b57cec5SDimitry Andric     if (LHS.getOpcode() == ISD::FNEG)
45700b57cec5SDimitry Andric       LHS = LHS.getOperand(0);
45710b57cec5SDimitry Andric     else if (MHS.getOpcode() == ISD::FNEG)
45720b57cec5SDimitry Andric       MHS = MHS.getOperand(0);
45730b57cec5SDimitry Andric     else
45740b57cec5SDimitry Andric       MHS = DAG.getNode(ISD::FNEG, SL, VT, MHS);
45750b57cec5SDimitry Andric 
45760b57cec5SDimitry Andric     if (RHS.getOpcode() != ISD::FNEG)
45770b57cec5SDimitry Andric       RHS = DAG.getNode(ISD::FNEG, SL, VT, RHS);
45780b57cec5SDimitry Andric     else
45790b57cec5SDimitry Andric       RHS = RHS.getOperand(0);
45800b57cec5SDimitry Andric 
45810b57cec5SDimitry Andric     SDValue Res = DAG.getNode(Opc, SL, VT, LHS, MHS, RHS);
45820b57cec5SDimitry Andric     if (Res.getOpcode() != Opc)
45830b57cec5SDimitry Andric       return SDValue(); // Op got folded away.
45840b57cec5SDimitry Andric     if (!N0.hasOneUse())
45850b57cec5SDimitry Andric       DAG.ReplaceAllUsesWith(N0, DAG.getNode(ISD::FNEG, SL, VT, Res));
45860b57cec5SDimitry Andric     return Res;
45870b57cec5SDimitry Andric   }
45880b57cec5SDimitry Andric   case ISD::FMAXNUM:
45890b57cec5SDimitry Andric   case ISD::FMINNUM:
45900b57cec5SDimitry Andric   case ISD::FMAXNUM_IEEE:
45910b57cec5SDimitry Andric   case ISD::FMINNUM_IEEE:
45920b57cec5SDimitry Andric   case AMDGPUISD::FMAX_LEGACY:
45930b57cec5SDimitry Andric   case AMDGPUISD::FMIN_LEGACY: {
45940b57cec5SDimitry Andric     // fneg (fmaxnum x, y) -> fminnum (fneg x), (fneg y)
45950b57cec5SDimitry Andric     // fneg (fminnum x, y) -> fmaxnum (fneg x), (fneg y)
45960b57cec5SDimitry Andric     // fneg (fmax_legacy x, y) -> fmin_legacy (fneg x), (fneg y)
45970b57cec5SDimitry Andric     // fneg (fmin_legacy x, y) -> fmax_legacy (fneg x), (fneg y)
45980b57cec5SDimitry Andric 
45990b57cec5SDimitry Andric     SDValue LHS = N0.getOperand(0);
46000b57cec5SDimitry Andric     SDValue RHS = N0.getOperand(1);
46010b57cec5SDimitry Andric 
46020b57cec5SDimitry Andric     // 0 doesn't have a negated inline immediate.
46030b57cec5SDimitry Andric     // TODO: This constant check should be generalized to other operations.
46040b57cec5SDimitry Andric     if (isConstantCostlierToNegate(RHS))
46050b57cec5SDimitry Andric       return SDValue();
46060b57cec5SDimitry Andric 
46070b57cec5SDimitry Andric     SDValue NegLHS = DAG.getNode(ISD::FNEG, SL, VT, LHS);
46080b57cec5SDimitry Andric     SDValue NegRHS = DAG.getNode(ISD::FNEG, SL, VT, RHS);
46090b57cec5SDimitry Andric     unsigned Opposite = inverseMinMax(Opc);
46100b57cec5SDimitry Andric 
46110b57cec5SDimitry Andric     SDValue Res = DAG.getNode(Opposite, SL, VT, NegLHS, NegRHS, N0->getFlags());
46120b57cec5SDimitry Andric     if (Res.getOpcode() != Opposite)
46130b57cec5SDimitry Andric       return SDValue(); // Op got folded away.
46140b57cec5SDimitry Andric     if (!N0.hasOneUse())
46150b57cec5SDimitry Andric       DAG.ReplaceAllUsesWith(N0, DAG.getNode(ISD::FNEG, SL, VT, Res));
46160b57cec5SDimitry Andric     return Res;
46170b57cec5SDimitry Andric   }
46180b57cec5SDimitry Andric   case AMDGPUISD::FMED3: {
46190b57cec5SDimitry Andric     SDValue Ops[3];
46200b57cec5SDimitry Andric     for (unsigned I = 0; I < 3; ++I)
46210b57cec5SDimitry Andric       Ops[I] = DAG.getNode(ISD::FNEG, SL, VT, N0->getOperand(I), N0->getFlags());
46220b57cec5SDimitry Andric 
46230b57cec5SDimitry Andric     SDValue Res = DAG.getNode(AMDGPUISD::FMED3, SL, VT, Ops, N0->getFlags());
46240b57cec5SDimitry Andric     if (Res.getOpcode() != AMDGPUISD::FMED3)
46250b57cec5SDimitry Andric       return SDValue(); // Op got folded away.
4626e8d8bef9SDimitry Andric 
4627e8d8bef9SDimitry Andric     if (!N0.hasOneUse()) {
4628e8d8bef9SDimitry Andric       SDValue Neg = DAG.getNode(ISD::FNEG, SL, VT, Res);
4629e8d8bef9SDimitry Andric       DAG.ReplaceAllUsesWith(N0, Neg);
4630e8d8bef9SDimitry Andric 
4631e8d8bef9SDimitry Andric       for (SDNode *U : Neg->uses())
4632e8d8bef9SDimitry Andric         DCI.AddToWorklist(U);
4633e8d8bef9SDimitry Andric     }
4634e8d8bef9SDimitry Andric 
46350b57cec5SDimitry Andric     return Res;
46360b57cec5SDimitry Andric   }
46370b57cec5SDimitry Andric   case ISD::FP_EXTEND:
46380b57cec5SDimitry Andric   case ISD::FTRUNC:
46390b57cec5SDimitry Andric   case ISD::FRINT:
46400b57cec5SDimitry Andric   case ISD::FNEARBYINT: // XXX - Should fround be handled?
46410b57cec5SDimitry Andric   case ISD::FSIN:
46420b57cec5SDimitry Andric   case ISD::FCANONICALIZE:
46430b57cec5SDimitry Andric   case AMDGPUISD::RCP:
46440b57cec5SDimitry Andric   case AMDGPUISD::RCP_LEGACY:
46450b57cec5SDimitry Andric   case AMDGPUISD::RCP_IFLAG:
46460b57cec5SDimitry Andric   case AMDGPUISD::SIN_HW: {
46470b57cec5SDimitry Andric     SDValue CvtSrc = N0.getOperand(0);
46480b57cec5SDimitry Andric     if (CvtSrc.getOpcode() == ISD::FNEG) {
46490b57cec5SDimitry Andric       // (fneg (fp_extend (fneg x))) -> (fp_extend x)
46500b57cec5SDimitry Andric       // (fneg (rcp (fneg x))) -> (rcp x)
46510b57cec5SDimitry Andric       return DAG.getNode(Opc, SL, VT, CvtSrc.getOperand(0));
46520b57cec5SDimitry Andric     }
46530b57cec5SDimitry Andric 
46540b57cec5SDimitry Andric     if (!N0.hasOneUse())
46550b57cec5SDimitry Andric       return SDValue();
46560b57cec5SDimitry Andric 
46570b57cec5SDimitry Andric     // (fneg (fp_extend x)) -> (fp_extend (fneg x))
46580b57cec5SDimitry Andric     // (fneg (rcp x)) -> (rcp (fneg x))
46590b57cec5SDimitry Andric     SDValue Neg = DAG.getNode(ISD::FNEG, SL, CvtSrc.getValueType(), CvtSrc);
46600b57cec5SDimitry Andric     return DAG.getNode(Opc, SL, VT, Neg, N0->getFlags());
46610b57cec5SDimitry Andric   }
46620b57cec5SDimitry Andric   case ISD::FP_ROUND: {
46630b57cec5SDimitry Andric     SDValue CvtSrc = N0.getOperand(0);
46640b57cec5SDimitry Andric 
46650b57cec5SDimitry Andric     if (CvtSrc.getOpcode() == ISD::FNEG) {
46660b57cec5SDimitry Andric       // (fneg (fp_round (fneg x))) -> (fp_round x)
46670b57cec5SDimitry Andric       return DAG.getNode(ISD::FP_ROUND, SL, VT,
46680b57cec5SDimitry Andric                          CvtSrc.getOperand(0), N0.getOperand(1));
46690b57cec5SDimitry Andric     }
46700b57cec5SDimitry Andric 
46710b57cec5SDimitry Andric     if (!N0.hasOneUse())
46720b57cec5SDimitry Andric       return SDValue();
46730b57cec5SDimitry Andric 
46740b57cec5SDimitry Andric     // (fneg (fp_round x)) -> (fp_round (fneg x))
46750b57cec5SDimitry Andric     SDValue Neg = DAG.getNode(ISD::FNEG, SL, CvtSrc.getValueType(), CvtSrc);
46760b57cec5SDimitry Andric     return DAG.getNode(ISD::FP_ROUND, SL, VT, Neg, N0.getOperand(1));
46770b57cec5SDimitry Andric   }
46780b57cec5SDimitry Andric   case ISD::FP16_TO_FP: {
46790b57cec5SDimitry Andric     // v_cvt_f32_f16 supports source modifiers on pre-VI targets without legal
46800b57cec5SDimitry Andric     // f16, but legalization of f16 fneg ends up pulling it out of the source.
46810b57cec5SDimitry Andric     // Put the fneg back as a legal source operation that can be matched later.
46820b57cec5SDimitry Andric     SDLoc SL(N);
46830b57cec5SDimitry Andric 
46840b57cec5SDimitry Andric     SDValue Src = N0.getOperand(0);
46850b57cec5SDimitry Andric     EVT SrcVT = Src.getValueType();
46860b57cec5SDimitry Andric 
46870b57cec5SDimitry Andric     // fneg (fp16_to_fp x) -> fp16_to_fp (xor x, 0x8000)
46880b57cec5SDimitry Andric     SDValue IntFNeg = DAG.getNode(ISD::XOR, SL, SrcVT, Src,
46890b57cec5SDimitry Andric                                   DAG.getConstant(0x8000, SL, SrcVT));
46900b57cec5SDimitry Andric     return DAG.getNode(ISD::FP16_TO_FP, SL, N->getValueType(0), IntFNeg);
46910b57cec5SDimitry Andric   }
469206c3fb27SDimitry Andric   case ISD::SELECT: {
469306c3fb27SDimitry Andric     // fneg (select c, a, b) -> select c, (fneg a), (fneg b)
469406c3fb27SDimitry Andric     // TODO: Invert conditions of foldFreeOpFromSelect
469506c3fb27SDimitry Andric     return SDValue();
469606c3fb27SDimitry Andric   }
469706c3fb27SDimitry Andric   case ISD::BITCAST: {
469806c3fb27SDimitry Andric     SDLoc SL(N);
469906c3fb27SDimitry Andric     SDValue BCSrc = N0.getOperand(0);
470006c3fb27SDimitry Andric     if (BCSrc.getOpcode() == ISD::BUILD_VECTOR) {
470106c3fb27SDimitry Andric       SDValue HighBits = BCSrc.getOperand(BCSrc.getNumOperands() - 1);
470206c3fb27SDimitry Andric       if (HighBits.getValueType().getSizeInBits() != 32 ||
470306c3fb27SDimitry Andric           !fnegFoldsIntoOp(HighBits.getNode()))
470406c3fb27SDimitry Andric         return SDValue();
470506c3fb27SDimitry Andric 
470606c3fb27SDimitry Andric       // f64 fneg only really needs to operate on the high half of of the
470706c3fb27SDimitry Andric       // register, so try to force it to an f32 operation to help make use of
470806c3fb27SDimitry Andric       // source modifiers.
470906c3fb27SDimitry Andric       //
471006c3fb27SDimitry Andric       //
471106c3fb27SDimitry Andric       // fneg (f64 (bitcast (build_vector x, y))) ->
471206c3fb27SDimitry Andric       // f64 (bitcast (build_vector (bitcast i32:x to f32),
471306c3fb27SDimitry Andric       //                            (fneg (bitcast i32:y to f32)))
471406c3fb27SDimitry Andric 
471506c3fb27SDimitry Andric       SDValue CastHi = DAG.getNode(ISD::BITCAST, SL, MVT::f32, HighBits);
471606c3fb27SDimitry Andric       SDValue NegHi = DAG.getNode(ISD::FNEG, SL, MVT::f32, CastHi);
471706c3fb27SDimitry Andric       SDValue CastBack =
471806c3fb27SDimitry Andric           DAG.getNode(ISD::BITCAST, SL, HighBits.getValueType(), NegHi);
471906c3fb27SDimitry Andric 
472006c3fb27SDimitry Andric       SmallVector<SDValue, 8> Ops(BCSrc->op_begin(), BCSrc->op_end());
472106c3fb27SDimitry Andric       Ops.back() = CastBack;
472206c3fb27SDimitry Andric       DCI.AddToWorklist(NegHi.getNode());
472306c3fb27SDimitry Andric       SDValue Build =
472406c3fb27SDimitry Andric           DAG.getNode(ISD::BUILD_VECTOR, SL, BCSrc.getValueType(), Ops);
472506c3fb27SDimitry Andric       SDValue Result = DAG.getNode(ISD::BITCAST, SL, VT, Build);
472606c3fb27SDimitry Andric 
472706c3fb27SDimitry Andric       if (!N0.hasOneUse())
472806c3fb27SDimitry Andric         DAG.ReplaceAllUsesWith(N0, DAG.getNode(ISD::FNEG, SL, VT, Result));
472906c3fb27SDimitry Andric       return Result;
473006c3fb27SDimitry Andric     }
473106c3fb27SDimitry Andric 
473206c3fb27SDimitry Andric     if (BCSrc.getOpcode() == ISD::SELECT && VT == MVT::f32 &&
473306c3fb27SDimitry Andric         BCSrc.hasOneUse()) {
473406c3fb27SDimitry Andric       // fneg (bitcast (f32 (select cond, i32:lhs, i32:rhs))) ->
473506c3fb27SDimitry Andric       //   select cond, (bitcast i32:lhs to f32), (bitcast i32:rhs to f32)
473606c3fb27SDimitry Andric 
473706c3fb27SDimitry Andric       // TODO: Cast back result for multiple uses is beneficial in some cases.
473806c3fb27SDimitry Andric 
473906c3fb27SDimitry Andric       SDValue LHS =
474006c3fb27SDimitry Andric           DAG.getNode(ISD::BITCAST, SL, MVT::f32, BCSrc.getOperand(1));
474106c3fb27SDimitry Andric       SDValue RHS =
474206c3fb27SDimitry Andric           DAG.getNode(ISD::BITCAST, SL, MVT::f32, BCSrc.getOperand(2));
474306c3fb27SDimitry Andric 
474406c3fb27SDimitry Andric       SDValue NegLHS = DAG.getNode(ISD::FNEG, SL, MVT::f32, LHS);
474506c3fb27SDimitry Andric       SDValue NegRHS = DAG.getNode(ISD::FNEG, SL, MVT::f32, RHS);
474606c3fb27SDimitry Andric 
474706c3fb27SDimitry Andric       return DAG.getNode(ISD::SELECT, SL, MVT::f32, BCSrc.getOperand(0), NegLHS,
474806c3fb27SDimitry Andric                          NegRHS);
474906c3fb27SDimitry Andric     }
475006c3fb27SDimitry Andric 
475106c3fb27SDimitry Andric     return SDValue();
475206c3fb27SDimitry Andric   }
47530b57cec5SDimitry Andric   default:
47540b57cec5SDimitry Andric     return SDValue();
47550b57cec5SDimitry Andric   }
47560b57cec5SDimitry Andric }
47570b57cec5SDimitry Andric 
47580b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performFAbsCombine(SDNode *N,
47590b57cec5SDimitry Andric                                                  DAGCombinerInfo &DCI) const {
47600b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
47610b57cec5SDimitry Andric   SDValue N0 = N->getOperand(0);
47620b57cec5SDimitry Andric 
47630b57cec5SDimitry Andric   if (!N0.hasOneUse())
47640b57cec5SDimitry Andric     return SDValue();
47650b57cec5SDimitry Andric 
47660b57cec5SDimitry Andric   switch (N0.getOpcode()) {
47670b57cec5SDimitry Andric   case ISD::FP16_TO_FP: {
47680b57cec5SDimitry Andric     assert(!Subtarget->has16BitInsts() && "should only see if f16 is illegal");
47690b57cec5SDimitry Andric     SDLoc SL(N);
47700b57cec5SDimitry Andric     SDValue Src = N0.getOperand(0);
47710b57cec5SDimitry Andric     EVT SrcVT = Src.getValueType();
47720b57cec5SDimitry Andric 
47730b57cec5SDimitry Andric     // fabs (fp16_to_fp x) -> fp16_to_fp (and x, 0x7fff)
47740b57cec5SDimitry Andric     SDValue IntFAbs = DAG.getNode(ISD::AND, SL, SrcVT, Src,
47750b57cec5SDimitry Andric                                   DAG.getConstant(0x7fff, SL, SrcVT));
47760b57cec5SDimitry Andric     return DAG.getNode(ISD::FP16_TO_FP, SL, N->getValueType(0), IntFAbs);
47770b57cec5SDimitry Andric   }
47780b57cec5SDimitry Andric   default:
47790b57cec5SDimitry Andric     return SDValue();
47800b57cec5SDimitry Andric   }
47810b57cec5SDimitry Andric }
47820b57cec5SDimitry Andric 
47830b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performRcpCombine(SDNode *N,
47840b57cec5SDimitry Andric                                                 DAGCombinerInfo &DCI) const {
47850b57cec5SDimitry Andric   const auto *CFP = dyn_cast<ConstantFPSDNode>(N->getOperand(0));
47860b57cec5SDimitry Andric   if (!CFP)
47870b57cec5SDimitry Andric     return SDValue();
47880b57cec5SDimitry Andric 
47890b57cec5SDimitry Andric   // XXX - Should this flush denormals?
47900b57cec5SDimitry Andric   const APFloat &Val = CFP->getValueAPF();
47910b57cec5SDimitry Andric   APFloat One(Val.getSemantics(), "1.0");
47920b57cec5SDimitry Andric   return DCI.DAG.getConstantFP(One / Val, SDLoc(N), N->getValueType(0));
47930b57cec5SDimitry Andric }
47940b57cec5SDimitry Andric 
47950b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::PerformDAGCombine(SDNode *N,
47960b57cec5SDimitry Andric                                                 DAGCombinerInfo &DCI) const {
47970b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
47980b57cec5SDimitry Andric   SDLoc DL(N);
47990b57cec5SDimitry Andric 
48000b57cec5SDimitry Andric   switch(N->getOpcode()) {
48010b57cec5SDimitry Andric   default:
48020b57cec5SDimitry Andric     break;
48030b57cec5SDimitry Andric   case ISD::BITCAST: {
48040b57cec5SDimitry Andric     EVT DestVT = N->getValueType(0);
48050b57cec5SDimitry Andric 
48060b57cec5SDimitry Andric     // Push casts through vector builds. This helps avoid emitting a large
48070b57cec5SDimitry Andric     // number of copies when materializing floating point vector constants.
48080b57cec5SDimitry Andric     //
48090b57cec5SDimitry Andric     // vNt1 bitcast (vNt0 (build_vector t0:x, t0:y)) =>
48100b57cec5SDimitry Andric     //   vnt1 = build_vector (t1 (bitcast t0:x)), (t1 (bitcast t0:y))
48110b57cec5SDimitry Andric     if (DestVT.isVector()) {
48120b57cec5SDimitry Andric       SDValue Src = N->getOperand(0);
48130b57cec5SDimitry Andric       if (Src.getOpcode() == ISD::BUILD_VECTOR) {
48140b57cec5SDimitry Andric         EVT SrcVT = Src.getValueType();
48150b57cec5SDimitry Andric         unsigned NElts = DestVT.getVectorNumElements();
48160b57cec5SDimitry Andric 
48170b57cec5SDimitry Andric         if (SrcVT.getVectorNumElements() == NElts) {
48180b57cec5SDimitry Andric           EVT DestEltVT = DestVT.getVectorElementType();
48190b57cec5SDimitry Andric 
48200b57cec5SDimitry Andric           SmallVector<SDValue, 8> CastedElts;
48210b57cec5SDimitry Andric           SDLoc SL(N);
48220b57cec5SDimitry Andric           for (unsigned I = 0, E = SrcVT.getVectorNumElements(); I != E; ++I) {
48230b57cec5SDimitry Andric             SDValue Elt = Src.getOperand(I);
48240b57cec5SDimitry Andric             CastedElts.push_back(DAG.getNode(ISD::BITCAST, DL, DestEltVT, Elt));
48250b57cec5SDimitry Andric           }
48260b57cec5SDimitry Andric 
48270b57cec5SDimitry Andric           return DAG.getBuildVector(DestVT, SL, CastedElts);
48280b57cec5SDimitry Andric         }
48290b57cec5SDimitry Andric       }
48300b57cec5SDimitry Andric     }
48310b57cec5SDimitry Andric 
4832e8d8bef9SDimitry Andric     if (DestVT.getSizeInBits() != 64 || !DestVT.isVector())
48330b57cec5SDimitry Andric       break;
48340b57cec5SDimitry Andric 
48350b57cec5SDimitry Andric     // Fold bitcasts of constants.
48360b57cec5SDimitry Andric     //
48370b57cec5SDimitry Andric     // v2i32 (bitcast i64:k) -> build_vector lo_32(k), hi_32(k)
48380b57cec5SDimitry Andric     // TODO: Generalize and move to DAGCombiner
48390b57cec5SDimitry Andric     SDValue Src = N->getOperand(0);
48400b57cec5SDimitry Andric     if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Src)) {
48410b57cec5SDimitry Andric       SDLoc SL(N);
48420b57cec5SDimitry Andric       uint64_t CVal = C->getZExtValue();
48430b57cec5SDimitry Andric       SDValue BV = DAG.getNode(ISD::BUILD_VECTOR, SL, MVT::v2i32,
48440b57cec5SDimitry Andric                                DAG.getConstant(Lo_32(CVal), SL, MVT::i32),
48450b57cec5SDimitry Andric                                DAG.getConstant(Hi_32(CVal), SL, MVT::i32));
48460b57cec5SDimitry Andric       return DAG.getNode(ISD::BITCAST, SL, DestVT, BV);
48470b57cec5SDimitry Andric     }
48480b57cec5SDimitry Andric 
48490b57cec5SDimitry Andric     if (ConstantFPSDNode *C = dyn_cast<ConstantFPSDNode>(Src)) {
48500b57cec5SDimitry Andric       const APInt &Val = C->getValueAPF().bitcastToAPInt();
48510b57cec5SDimitry Andric       SDLoc SL(N);
48520b57cec5SDimitry Andric       uint64_t CVal = Val.getZExtValue();
48530b57cec5SDimitry Andric       SDValue Vec = DAG.getNode(ISD::BUILD_VECTOR, SL, MVT::v2i32,
48540b57cec5SDimitry Andric                                 DAG.getConstant(Lo_32(CVal), SL, MVT::i32),
48550b57cec5SDimitry Andric                                 DAG.getConstant(Hi_32(CVal), SL, MVT::i32));
48560b57cec5SDimitry Andric 
48570b57cec5SDimitry Andric       return DAG.getNode(ISD::BITCAST, SL, DestVT, Vec);
48580b57cec5SDimitry Andric     }
48590b57cec5SDimitry Andric 
48600b57cec5SDimitry Andric     break;
48610b57cec5SDimitry Andric   }
48620b57cec5SDimitry Andric   case ISD::SHL: {
48630b57cec5SDimitry Andric     if (DCI.getDAGCombineLevel() < AfterLegalizeDAG)
48640b57cec5SDimitry Andric       break;
48650b57cec5SDimitry Andric 
48660b57cec5SDimitry Andric     return performShlCombine(N, DCI);
48670b57cec5SDimitry Andric   }
48680b57cec5SDimitry Andric   case ISD::SRL: {
48690b57cec5SDimitry Andric     if (DCI.getDAGCombineLevel() < AfterLegalizeDAG)
48700b57cec5SDimitry Andric       break;
48710b57cec5SDimitry Andric 
48720b57cec5SDimitry Andric     return performSrlCombine(N, DCI);
48730b57cec5SDimitry Andric   }
48740b57cec5SDimitry Andric   case ISD::SRA: {
48750b57cec5SDimitry Andric     if (DCI.getDAGCombineLevel() < AfterLegalizeDAG)
48760b57cec5SDimitry Andric       break;
48770b57cec5SDimitry Andric 
48780b57cec5SDimitry Andric     return performSraCombine(N, DCI);
48790b57cec5SDimitry Andric   }
48800b57cec5SDimitry Andric   case ISD::TRUNCATE:
48810b57cec5SDimitry Andric     return performTruncateCombine(N, DCI);
48820b57cec5SDimitry Andric   case ISD::MUL:
48830b57cec5SDimitry Andric     return performMulCombine(N, DCI);
488406c3fb27SDimitry Andric   case AMDGPUISD::MUL_U24:
488506c3fb27SDimitry Andric   case AMDGPUISD::MUL_I24: {
488606c3fb27SDimitry Andric     if (SDValue Simplified = simplifyMul24(N, DCI))
488706c3fb27SDimitry Andric       return Simplified;
488806c3fb27SDimitry Andric     return performMulCombine(N, DCI);
488906c3fb27SDimitry Andric   }
489006c3fb27SDimitry Andric   case AMDGPUISD::MULHI_I24:
489106c3fb27SDimitry Andric   case AMDGPUISD::MULHI_U24:
489206c3fb27SDimitry Andric     return simplifyMul24(N, DCI);
48934824e7fdSDimitry Andric   case ISD::SMUL_LOHI:
48944824e7fdSDimitry Andric   case ISD::UMUL_LOHI:
48954824e7fdSDimitry Andric     return performMulLoHiCombine(N, DCI);
48960b57cec5SDimitry Andric   case ISD::MULHS:
48970b57cec5SDimitry Andric     return performMulhsCombine(N, DCI);
48980b57cec5SDimitry Andric   case ISD::MULHU:
48990b57cec5SDimitry Andric     return performMulhuCombine(N, DCI);
49000b57cec5SDimitry Andric   case ISD::SELECT:
49010b57cec5SDimitry Andric     return performSelectCombine(N, DCI);
49020b57cec5SDimitry Andric   case ISD::FNEG:
49030b57cec5SDimitry Andric     return performFNegCombine(N, DCI);
49040b57cec5SDimitry Andric   case ISD::FABS:
49050b57cec5SDimitry Andric     return performFAbsCombine(N, DCI);
49060b57cec5SDimitry Andric   case AMDGPUISD::BFE_I32:
49070b57cec5SDimitry Andric   case AMDGPUISD::BFE_U32: {
49080b57cec5SDimitry Andric     assert(!N->getValueType(0).isVector() &&
49090b57cec5SDimitry Andric            "Vector handling of BFE not implemented");
49100b57cec5SDimitry Andric     ConstantSDNode *Width = dyn_cast<ConstantSDNode>(N->getOperand(2));
49110b57cec5SDimitry Andric     if (!Width)
49120b57cec5SDimitry Andric       break;
49130b57cec5SDimitry Andric 
49140b57cec5SDimitry Andric     uint32_t WidthVal = Width->getZExtValue() & 0x1f;
49150b57cec5SDimitry Andric     if (WidthVal == 0)
49160b57cec5SDimitry Andric       return DAG.getConstant(0, DL, MVT::i32);
49170b57cec5SDimitry Andric 
49180b57cec5SDimitry Andric     ConstantSDNode *Offset = dyn_cast<ConstantSDNode>(N->getOperand(1));
49190b57cec5SDimitry Andric     if (!Offset)
49200b57cec5SDimitry Andric       break;
49210b57cec5SDimitry Andric 
49220b57cec5SDimitry Andric     SDValue BitsFrom = N->getOperand(0);
49230b57cec5SDimitry Andric     uint32_t OffsetVal = Offset->getZExtValue() & 0x1f;
49240b57cec5SDimitry Andric 
49250b57cec5SDimitry Andric     bool Signed = N->getOpcode() == AMDGPUISD::BFE_I32;
49260b57cec5SDimitry Andric 
49270b57cec5SDimitry Andric     if (OffsetVal == 0) {
49280b57cec5SDimitry Andric       // This is already sign / zero extended, so try to fold away extra BFEs.
49290b57cec5SDimitry Andric       unsigned SignBits =  Signed ? (32 - WidthVal + 1) : (32 - WidthVal);
49300b57cec5SDimitry Andric 
49310b57cec5SDimitry Andric       unsigned OpSignBits = DAG.ComputeNumSignBits(BitsFrom);
49320b57cec5SDimitry Andric       if (OpSignBits >= SignBits)
49330b57cec5SDimitry Andric         return BitsFrom;
49340b57cec5SDimitry Andric 
49350b57cec5SDimitry Andric       EVT SmallVT = EVT::getIntegerVT(*DAG.getContext(), WidthVal);
49360b57cec5SDimitry Andric       if (Signed) {
49370b57cec5SDimitry Andric         // This is a sign_extend_inreg. Replace it to take advantage of existing
49380b57cec5SDimitry Andric         // DAG Combines. If not eliminated, we will match back to BFE during
49390b57cec5SDimitry Andric         // selection.
49400b57cec5SDimitry Andric 
49410b57cec5SDimitry Andric         // TODO: The sext_inreg of extended types ends, although we can could
49420b57cec5SDimitry Andric         // handle them in a single BFE.
49430b57cec5SDimitry Andric         return DAG.getNode(ISD::SIGN_EXTEND_INREG, DL, MVT::i32, BitsFrom,
49440b57cec5SDimitry Andric                            DAG.getValueType(SmallVT));
49450b57cec5SDimitry Andric       }
49460b57cec5SDimitry Andric 
49470b57cec5SDimitry Andric       return DAG.getZeroExtendInReg(BitsFrom, DL, SmallVT);
49480b57cec5SDimitry Andric     }
49490b57cec5SDimitry Andric 
49500b57cec5SDimitry Andric     if (ConstantSDNode *CVal = dyn_cast<ConstantSDNode>(BitsFrom)) {
49510b57cec5SDimitry Andric       if (Signed) {
49520b57cec5SDimitry Andric         return constantFoldBFE<int32_t>(DAG,
49530b57cec5SDimitry Andric                                         CVal->getSExtValue(),
49540b57cec5SDimitry Andric                                         OffsetVal,
49550b57cec5SDimitry Andric                                         WidthVal,
49560b57cec5SDimitry Andric                                         DL);
49570b57cec5SDimitry Andric       }
49580b57cec5SDimitry Andric 
49590b57cec5SDimitry Andric       return constantFoldBFE<uint32_t>(DAG,
49600b57cec5SDimitry Andric                                        CVal->getZExtValue(),
49610b57cec5SDimitry Andric                                        OffsetVal,
49620b57cec5SDimitry Andric                                        WidthVal,
49630b57cec5SDimitry Andric                                        DL);
49640b57cec5SDimitry Andric     }
49650b57cec5SDimitry Andric 
49660b57cec5SDimitry Andric     if ((OffsetVal + WidthVal) >= 32 &&
49670b57cec5SDimitry Andric         !(Subtarget->hasSDWA() && OffsetVal == 16 && WidthVal == 16)) {
49680b57cec5SDimitry Andric       SDValue ShiftVal = DAG.getConstant(OffsetVal, DL, MVT::i32);
49690b57cec5SDimitry Andric       return DAG.getNode(Signed ? ISD::SRA : ISD::SRL, DL, MVT::i32,
49700b57cec5SDimitry Andric                          BitsFrom, ShiftVal);
49710b57cec5SDimitry Andric     }
49720b57cec5SDimitry Andric 
49730b57cec5SDimitry Andric     if (BitsFrom.hasOneUse()) {
49740b57cec5SDimitry Andric       APInt Demanded = APInt::getBitsSet(32,
49750b57cec5SDimitry Andric                                          OffsetVal,
49760b57cec5SDimitry Andric                                          OffsetVal + WidthVal);
49770b57cec5SDimitry Andric 
49780b57cec5SDimitry Andric       KnownBits Known;
49790b57cec5SDimitry Andric       TargetLowering::TargetLoweringOpt TLO(DAG, !DCI.isBeforeLegalize(),
49800b57cec5SDimitry Andric                                             !DCI.isBeforeLegalizeOps());
49810b57cec5SDimitry Andric       const TargetLowering &TLI = DAG.getTargetLoweringInfo();
49820b57cec5SDimitry Andric       if (TLI.ShrinkDemandedConstant(BitsFrom, Demanded, TLO) ||
49830b57cec5SDimitry Andric           TLI.SimplifyDemandedBits(BitsFrom, Demanded, Known, TLO)) {
49840b57cec5SDimitry Andric         DCI.CommitTargetLoweringOpt(TLO);
49850b57cec5SDimitry Andric       }
49860b57cec5SDimitry Andric     }
49870b57cec5SDimitry Andric 
49880b57cec5SDimitry Andric     break;
49890b57cec5SDimitry Andric   }
49900b57cec5SDimitry Andric   case ISD::LOAD:
49910b57cec5SDimitry Andric     return performLoadCombine(N, DCI);
49920b57cec5SDimitry Andric   case ISD::STORE:
49930b57cec5SDimitry Andric     return performStoreCombine(N, DCI);
49940b57cec5SDimitry Andric   case AMDGPUISD::RCP:
49950b57cec5SDimitry Andric   case AMDGPUISD::RCP_IFLAG:
49960b57cec5SDimitry Andric     return performRcpCombine(N, DCI);
49970b57cec5SDimitry Andric   case ISD::AssertZext:
49980b57cec5SDimitry Andric   case ISD::AssertSext:
49990b57cec5SDimitry Andric     return performAssertSZExtCombine(N, DCI);
50008bcb0991SDimitry Andric   case ISD::INTRINSIC_WO_CHAIN:
50018bcb0991SDimitry Andric     return performIntrinsicWOChainCombine(N, DCI);
50020b57cec5SDimitry Andric   }
50030b57cec5SDimitry Andric   return SDValue();
50040b57cec5SDimitry Andric }
50050b57cec5SDimitry Andric 
50060b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
50070b57cec5SDimitry Andric // Helper functions
50080b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
50090b57cec5SDimitry Andric 
50100b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::CreateLiveInRegister(SelectionDAG &DAG,
50110b57cec5SDimitry Andric                                                    const TargetRegisterClass *RC,
50125ffd83dbSDimitry Andric                                                    Register Reg, EVT VT,
50130b57cec5SDimitry Andric                                                    const SDLoc &SL,
50140b57cec5SDimitry Andric                                                    bool RawReg) const {
50150b57cec5SDimitry Andric   MachineFunction &MF = DAG.getMachineFunction();
50160b57cec5SDimitry Andric   MachineRegisterInfo &MRI = MF.getRegInfo();
50175ffd83dbSDimitry Andric   Register VReg;
50180b57cec5SDimitry Andric 
50190b57cec5SDimitry Andric   if (!MRI.isLiveIn(Reg)) {
50200b57cec5SDimitry Andric     VReg = MRI.createVirtualRegister(RC);
50210b57cec5SDimitry Andric     MRI.addLiveIn(Reg, VReg);
50220b57cec5SDimitry Andric   } else {
50230b57cec5SDimitry Andric     VReg = MRI.getLiveInVirtReg(Reg);
50240b57cec5SDimitry Andric   }
50250b57cec5SDimitry Andric 
50260b57cec5SDimitry Andric   if (RawReg)
50270b57cec5SDimitry Andric     return DAG.getRegister(VReg, VT);
50280b57cec5SDimitry Andric 
50290b57cec5SDimitry Andric   return DAG.getCopyFromReg(DAG.getEntryNode(), SL, VReg, VT);
50300b57cec5SDimitry Andric }
50310b57cec5SDimitry Andric 
50328bcb0991SDimitry Andric // This may be called multiple times, and nothing prevents creating multiple
50338bcb0991SDimitry Andric // objects at the same offset. See if we already defined this object.
50348bcb0991SDimitry Andric static int getOrCreateFixedStackObject(MachineFrameInfo &MFI, unsigned Size,
50358bcb0991SDimitry Andric                                        int64_t Offset) {
50368bcb0991SDimitry Andric   for (int I = MFI.getObjectIndexBegin(); I < 0; ++I) {
50378bcb0991SDimitry Andric     if (MFI.getObjectOffset(I) == Offset) {
50388bcb0991SDimitry Andric       assert(MFI.getObjectSize(I) == Size);
50398bcb0991SDimitry Andric       return I;
50408bcb0991SDimitry Andric     }
50418bcb0991SDimitry Andric   }
50428bcb0991SDimitry Andric 
50438bcb0991SDimitry Andric   return MFI.CreateFixedObject(Size, Offset, true);
50448bcb0991SDimitry Andric }
50458bcb0991SDimitry Andric 
50460b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::loadStackInputValue(SelectionDAG &DAG,
50470b57cec5SDimitry Andric                                                   EVT VT,
50480b57cec5SDimitry Andric                                                   const SDLoc &SL,
50490b57cec5SDimitry Andric                                                   int64_t Offset) const {
50500b57cec5SDimitry Andric   MachineFunction &MF = DAG.getMachineFunction();
50510b57cec5SDimitry Andric   MachineFrameInfo &MFI = MF.getFrameInfo();
50528bcb0991SDimitry Andric   int FI = getOrCreateFixedStackObject(MFI, VT.getStoreSize(), Offset);
50530b57cec5SDimitry Andric 
50540b57cec5SDimitry Andric   auto SrcPtrInfo = MachinePointerInfo::getStack(MF, Offset);
50550b57cec5SDimitry Andric   SDValue Ptr = DAG.getFrameIndex(FI, MVT::i32);
50560b57cec5SDimitry Andric 
5057e8d8bef9SDimitry Andric   return DAG.getLoad(VT, SL, DAG.getEntryNode(), Ptr, SrcPtrInfo, Align(4),
50580b57cec5SDimitry Andric                      MachineMemOperand::MODereferenceable |
50590b57cec5SDimitry Andric                          MachineMemOperand::MOInvariant);
50600b57cec5SDimitry Andric }
50610b57cec5SDimitry Andric 
50620b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::storeStackInputValue(SelectionDAG &DAG,
50630b57cec5SDimitry Andric                                                    const SDLoc &SL,
50640b57cec5SDimitry Andric                                                    SDValue Chain,
50650b57cec5SDimitry Andric                                                    SDValue ArgVal,
50660b57cec5SDimitry Andric                                                    int64_t Offset) const {
50670b57cec5SDimitry Andric   MachineFunction &MF = DAG.getMachineFunction();
50680b57cec5SDimitry Andric   MachinePointerInfo DstInfo = MachinePointerInfo::getStack(MF, Offset);
5069fe6060f1SDimitry Andric   const SIMachineFunctionInfo *Info = MF.getInfo<SIMachineFunctionInfo>();
50700b57cec5SDimitry Andric 
50710b57cec5SDimitry Andric   SDValue Ptr = DAG.getConstant(Offset, SL, MVT::i32);
5072fe6060f1SDimitry Andric   // Stores to the argument stack area are relative to the stack pointer.
5073fe6060f1SDimitry Andric   SDValue SP =
5074fe6060f1SDimitry Andric       DAG.getCopyFromReg(Chain, SL, Info->getStackPtrOffsetReg(), MVT::i32);
5075fe6060f1SDimitry Andric   Ptr = DAG.getNode(ISD::ADD, SL, MVT::i32, SP, Ptr);
5076e8d8bef9SDimitry Andric   SDValue Store = DAG.getStore(Chain, SL, ArgVal, Ptr, DstInfo, Align(4),
50770b57cec5SDimitry Andric                                MachineMemOperand::MODereferenceable);
50780b57cec5SDimitry Andric   return Store;
50790b57cec5SDimitry Andric }
50800b57cec5SDimitry Andric 
50810b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::loadInputValue(SelectionDAG &DAG,
50820b57cec5SDimitry Andric                                              const TargetRegisterClass *RC,
50830b57cec5SDimitry Andric                                              EVT VT, const SDLoc &SL,
50840b57cec5SDimitry Andric                                              const ArgDescriptor &Arg) const {
50850b57cec5SDimitry Andric   assert(Arg && "Attempting to load missing argument");
50860b57cec5SDimitry Andric 
50870b57cec5SDimitry Andric   SDValue V = Arg.isRegister() ?
50880b57cec5SDimitry Andric     CreateLiveInRegister(DAG, RC, Arg.getRegister(), VT, SL) :
50890b57cec5SDimitry Andric     loadStackInputValue(DAG, VT, SL, Arg.getStackOffset());
50900b57cec5SDimitry Andric 
50910b57cec5SDimitry Andric   if (!Arg.isMasked())
50920b57cec5SDimitry Andric     return V;
50930b57cec5SDimitry Andric 
50940b57cec5SDimitry Andric   unsigned Mask = Arg.getMask();
509506c3fb27SDimitry Andric   unsigned Shift = llvm::countr_zero<unsigned>(Mask);
50960b57cec5SDimitry Andric   V = DAG.getNode(ISD::SRL, SL, VT, V,
50970b57cec5SDimitry Andric                   DAG.getShiftAmountConstant(Shift, VT, SL));
50980b57cec5SDimitry Andric   return DAG.getNode(ISD::AND, SL, VT, V,
50990b57cec5SDimitry Andric                      DAG.getConstant(Mask >> Shift, SL, VT));
51000b57cec5SDimitry Andric }
51010b57cec5SDimitry Andric 
51020b57cec5SDimitry Andric uint32_t AMDGPUTargetLowering::getImplicitParameterOffset(
510306c3fb27SDimitry Andric     uint64_t ExplicitKernArgSize, const ImplicitParameter Param) const {
510406c3fb27SDimitry Andric   unsigned ExplicitArgOffset = Subtarget->getExplicitKernelArgOffset();
510506c3fb27SDimitry Andric   const Align Alignment = Subtarget->getAlignmentForImplicitArgPtr();
510606c3fb27SDimitry Andric   uint64_t ArgOffset =
510706c3fb27SDimitry Andric       alignTo(ExplicitKernArgSize, Alignment) + ExplicitArgOffset;
51080b57cec5SDimitry Andric   switch (Param) {
510981ad6265SDimitry Andric   case FIRST_IMPLICIT:
51100b57cec5SDimitry Andric     return ArgOffset;
511181ad6265SDimitry Andric   case PRIVATE_BASE:
511281ad6265SDimitry Andric     return ArgOffset + AMDGPU::ImplicitArg::PRIVATE_BASE_OFFSET;
511381ad6265SDimitry Andric   case SHARED_BASE:
511481ad6265SDimitry Andric     return ArgOffset + AMDGPU::ImplicitArg::SHARED_BASE_OFFSET;
511581ad6265SDimitry Andric   case QUEUE_PTR:
511681ad6265SDimitry Andric     return ArgOffset + AMDGPU::ImplicitArg::QUEUE_PTR_OFFSET;
51170b57cec5SDimitry Andric   }
51180b57cec5SDimitry Andric   llvm_unreachable("unexpected implicit parameter type");
51190b57cec5SDimitry Andric }
51200b57cec5SDimitry Andric 
512106c3fb27SDimitry Andric uint32_t AMDGPUTargetLowering::getImplicitParameterOffset(
512206c3fb27SDimitry Andric     const MachineFunction &MF, const ImplicitParameter Param) const {
512306c3fb27SDimitry Andric   const AMDGPUMachineFunction *MFI = MF.getInfo<AMDGPUMachineFunction>();
512406c3fb27SDimitry Andric   return getImplicitParameterOffset(MFI->getExplicitKernArgSize(), Param);
512506c3fb27SDimitry Andric }
512606c3fb27SDimitry Andric 
51270b57cec5SDimitry Andric #define NODE_NAME_CASE(node) case AMDGPUISD::node: return #node;
51280b57cec5SDimitry Andric 
51290b57cec5SDimitry Andric const char* AMDGPUTargetLowering::getTargetNodeName(unsigned Opcode) const {
51300b57cec5SDimitry Andric   switch ((AMDGPUISD::NodeType)Opcode) {
51310b57cec5SDimitry Andric   case AMDGPUISD::FIRST_NUMBER: break;
51320b57cec5SDimitry Andric   // AMDIL DAG nodes
51330b57cec5SDimitry Andric   NODE_NAME_CASE(UMUL);
51340b57cec5SDimitry Andric   NODE_NAME_CASE(BRANCH_COND);
51350b57cec5SDimitry Andric 
51360b57cec5SDimitry Andric   // AMDGPU DAG nodes
51370b57cec5SDimitry Andric   NODE_NAME_CASE(IF)
51380b57cec5SDimitry Andric   NODE_NAME_CASE(ELSE)
51390b57cec5SDimitry Andric   NODE_NAME_CASE(LOOP)
51400b57cec5SDimitry Andric   NODE_NAME_CASE(CALL)
51410b57cec5SDimitry Andric   NODE_NAME_CASE(TC_RETURN)
514206c3fb27SDimitry Andric   NODE_NAME_CASE(TC_RETURN_GFX)
51430b57cec5SDimitry Andric   NODE_NAME_CASE(TRAP)
514406c3fb27SDimitry Andric   NODE_NAME_CASE(RET_GLUE)
51450b57cec5SDimitry Andric   NODE_NAME_CASE(RETURN_TO_EPILOG)
51460b57cec5SDimitry Andric   NODE_NAME_CASE(ENDPGM)
514706c3fb27SDimitry Andric   NODE_NAME_CASE(ENDPGM_TRAP)
51480b57cec5SDimitry Andric   NODE_NAME_CASE(DWORDADDR)
51490b57cec5SDimitry Andric   NODE_NAME_CASE(FRACT)
51500b57cec5SDimitry Andric   NODE_NAME_CASE(SETCC)
51510b57cec5SDimitry Andric   NODE_NAME_CASE(SETREG)
51528bcb0991SDimitry Andric   NODE_NAME_CASE(DENORM_MODE)
51530b57cec5SDimitry Andric   NODE_NAME_CASE(FMA_W_CHAIN)
51540b57cec5SDimitry Andric   NODE_NAME_CASE(FMUL_W_CHAIN)
51550b57cec5SDimitry Andric   NODE_NAME_CASE(CLAMP)
51560b57cec5SDimitry Andric   NODE_NAME_CASE(COS_HW)
51570b57cec5SDimitry Andric   NODE_NAME_CASE(SIN_HW)
51580b57cec5SDimitry Andric   NODE_NAME_CASE(FMAX_LEGACY)
51590b57cec5SDimitry Andric   NODE_NAME_CASE(FMIN_LEGACY)
51600b57cec5SDimitry Andric   NODE_NAME_CASE(FMAX3)
51610b57cec5SDimitry Andric   NODE_NAME_CASE(SMAX3)
51620b57cec5SDimitry Andric   NODE_NAME_CASE(UMAX3)
51630b57cec5SDimitry Andric   NODE_NAME_CASE(FMIN3)
51640b57cec5SDimitry Andric   NODE_NAME_CASE(SMIN3)
51650b57cec5SDimitry Andric   NODE_NAME_CASE(UMIN3)
51660b57cec5SDimitry Andric   NODE_NAME_CASE(FMED3)
51670b57cec5SDimitry Andric   NODE_NAME_CASE(SMED3)
51680b57cec5SDimitry Andric   NODE_NAME_CASE(UMED3)
51690b57cec5SDimitry Andric   NODE_NAME_CASE(FDOT2)
51700b57cec5SDimitry Andric   NODE_NAME_CASE(URECIP)
51710b57cec5SDimitry Andric   NODE_NAME_CASE(DIV_SCALE)
51720b57cec5SDimitry Andric   NODE_NAME_CASE(DIV_FMAS)
51730b57cec5SDimitry Andric   NODE_NAME_CASE(DIV_FIXUP)
51740b57cec5SDimitry Andric   NODE_NAME_CASE(FMAD_FTZ)
51750b57cec5SDimitry Andric   NODE_NAME_CASE(RCP)
51760b57cec5SDimitry Andric   NODE_NAME_CASE(RSQ)
51770b57cec5SDimitry Andric   NODE_NAME_CASE(RCP_LEGACY)
51780b57cec5SDimitry Andric   NODE_NAME_CASE(RCP_IFLAG)
517906c3fb27SDimitry Andric   NODE_NAME_CASE(LOG)
518006c3fb27SDimitry Andric   NODE_NAME_CASE(EXP)
51810b57cec5SDimitry Andric   NODE_NAME_CASE(FMUL_LEGACY)
51820b57cec5SDimitry Andric   NODE_NAME_CASE(RSQ_CLAMP)
51830b57cec5SDimitry Andric   NODE_NAME_CASE(FP_CLASS)
51840b57cec5SDimitry Andric   NODE_NAME_CASE(DOT4)
51850b57cec5SDimitry Andric   NODE_NAME_CASE(CARRY)
51860b57cec5SDimitry Andric   NODE_NAME_CASE(BORROW)
51870b57cec5SDimitry Andric   NODE_NAME_CASE(BFE_U32)
51880b57cec5SDimitry Andric   NODE_NAME_CASE(BFE_I32)
51890b57cec5SDimitry Andric   NODE_NAME_CASE(BFI)
51900b57cec5SDimitry Andric   NODE_NAME_CASE(BFM)
51910b57cec5SDimitry Andric   NODE_NAME_CASE(FFBH_U32)
51920b57cec5SDimitry Andric   NODE_NAME_CASE(FFBH_I32)
51930b57cec5SDimitry Andric   NODE_NAME_CASE(FFBL_B32)
51940b57cec5SDimitry Andric   NODE_NAME_CASE(MUL_U24)
51950b57cec5SDimitry Andric   NODE_NAME_CASE(MUL_I24)
51960b57cec5SDimitry Andric   NODE_NAME_CASE(MULHI_U24)
51970b57cec5SDimitry Andric   NODE_NAME_CASE(MULHI_I24)
51980b57cec5SDimitry Andric   NODE_NAME_CASE(MAD_U24)
51990b57cec5SDimitry Andric   NODE_NAME_CASE(MAD_I24)
52000b57cec5SDimitry Andric   NODE_NAME_CASE(MAD_I64_I32)
52010b57cec5SDimitry Andric   NODE_NAME_CASE(MAD_U64_U32)
52020b57cec5SDimitry Andric   NODE_NAME_CASE(PERM)
52030b57cec5SDimitry Andric   NODE_NAME_CASE(TEXTURE_FETCH)
52040b57cec5SDimitry Andric   NODE_NAME_CASE(R600_EXPORT)
52050b57cec5SDimitry Andric   NODE_NAME_CASE(CONST_ADDRESS)
52060b57cec5SDimitry Andric   NODE_NAME_CASE(REGISTER_LOAD)
52070b57cec5SDimitry Andric   NODE_NAME_CASE(REGISTER_STORE)
52080b57cec5SDimitry Andric   NODE_NAME_CASE(SAMPLE)
52090b57cec5SDimitry Andric   NODE_NAME_CASE(SAMPLEB)
52100b57cec5SDimitry Andric   NODE_NAME_CASE(SAMPLED)
52110b57cec5SDimitry Andric   NODE_NAME_CASE(SAMPLEL)
52120b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_F32_UBYTE0)
52130b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_F32_UBYTE1)
52140b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_F32_UBYTE2)
52150b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_F32_UBYTE3)
52160b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_PKRTZ_F16_F32)
52170b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_PKNORM_I16_F32)
52180b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_PKNORM_U16_F32)
52190b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_PK_I16_I32)
52200b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_PK_U16_U32)
52210b57cec5SDimitry Andric   NODE_NAME_CASE(FP_TO_FP16)
52220b57cec5SDimitry Andric   NODE_NAME_CASE(BUILD_VERTICAL_VECTOR)
52230b57cec5SDimitry Andric   NODE_NAME_CASE(CONST_DATA_PTR)
52240b57cec5SDimitry Andric   NODE_NAME_CASE(PC_ADD_REL_OFFSET)
52250b57cec5SDimitry Andric   NODE_NAME_CASE(LDS)
522681ad6265SDimitry Andric   NODE_NAME_CASE(FPTRUNC_ROUND_UPWARD)
522781ad6265SDimitry Andric   NODE_NAME_CASE(FPTRUNC_ROUND_DOWNWARD)
52280b57cec5SDimitry Andric   NODE_NAME_CASE(DUMMY_CHAIN)
52290b57cec5SDimitry Andric   case AMDGPUISD::FIRST_MEM_OPCODE_NUMBER: break;
52300b57cec5SDimitry Andric   NODE_NAME_CASE(LOAD_D16_HI)
52310b57cec5SDimitry Andric   NODE_NAME_CASE(LOAD_D16_LO)
52320b57cec5SDimitry Andric   NODE_NAME_CASE(LOAD_D16_HI_I8)
52330b57cec5SDimitry Andric   NODE_NAME_CASE(LOAD_D16_HI_U8)
52340b57cec5SDimitry Andric   NODE_NAME_CASE(LOAD_D16_LO_I8)
52350b57cec5SDimitry Andric   NODE_NAME_CASE(LOAD_D16_LO_U8)
52360b57cec5SDimitry Andric   NODE_NAME_CASE(STORE_MSKOR)
52370b57cec5SDimitry Andric   NODE_NAME_CASE(LOAD_CONSTANT)
52380b57cec5SDimitry Andric   NODE_NAME_CASE(TBUFFER_STORE_FORMAT)
52390b57cec5SDimitry Andric   NODE_NAME_CASE(TBUFFER_STORE_FORMAT_D16)
52400b57cec5SDimitry Andric   NODE_NAME_CASE(TBUFFER_LOAD_FORMAT)
52410b57cec5SDimitry Andric   NODE_NAME_CASE(TBUFFER_LOAD_FORMAT_D16)
52420b57cec5SDimitry Andric   NODE_NAME_CASE(DS_ORDERED_COUNT)
52430b57cec5SDimitry Andric   NODE_NAME_CASE(ATOMIC_CMP_SWAP)
52440b57cec5SDimitry Andric   NODE_NAME_CASE(ATOMIC_LOAD_FMIN)
52450b57cec5SDimitry Andric   NODE_NAME_CASE(ATOMIC_LOAD_FMAX)
52460b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_LOAD)
52470b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_LOAD_UBYTE)
52480b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_LOAD_USHORT)
52490b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_LOAD_BYTE)
52500b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_LOAD_SHORT)
52510b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_LOAD_FORMAT)
5252bdd1243dSDimitry Andric   NODE_NAME_CASE(BUFFER_LOAD_FORMAT_TFE)
52530b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_LOAD_FORMAT_D16)
52540b57cec5SDimitry Andric   NODE_NAME_CASE(SBUFFER_LOAD)
52550b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_STORE)
52560b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_STORE_BYTE)
52570b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_STORE_SHORT)
52580b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_STORE_FORMAT)
52590b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_STORE_FORMAT_D16)
52600b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_SWAP)
52610b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_ADD)
52620b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_SUB)
52630b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_SMIN)
52640b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_UMIN)
52650b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_SMAX)
52660b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_UMAX)
52670b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_AND)
52680b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_OR)
52690b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_XOR)
52708bcb0991SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_INC)
52718bcb0991SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_DEC)
52720b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_CMPSWAP)
52735ffd83dbSDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_CSUB)
52740b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_FADD)
5275fe6060f1SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_FMIN)
5276fe6060f1SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_FMAX)
52770b57cec5SDimitry Andric 
52780b57cec5SDimitry Andric   case AMDGPUISD::LAST_AMDGPU_ISD_NUMBER: break;
52790b57cec5SDimitry Andric   }
52800b57cec5SDimitry Andric   return nullptr;
52810b57cec5SDimitry Andric }
52820b57cec5SDimitry Andric 
52830b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::getSqrtEstimate(SDValue Operand,
52840b57cec5SDimitry Andric                                               SelectionDAG &DAG, int Enabled,
52850b57cec5SDimitry Andric                                               int &RefinementSteps,
52860b57cec5SDimitry Andric                                               bool &UseOneConstNR,
52870b57cec5SDimitry Andric                                               bool Reciprocal) const {
52880b57cec5SDimitry Andric   EVT VT = Operand.getValueType();
52890b57cec5SDimitry Andric 
52900b57cec5SDimitry Andric   if (VT == MVT::f32) {
52910b57cec5SDimitry Andric     RefinementSteps = 0;
52920b57cec5SDimitry Andric     return DAG.getNode(AMDGPUISD::RSQ, SDLoc(Operand), VT, Operand);
52930b57cec5SDimitry Andric   }
52940b57cec5SDimitry Andric 
52950b57cec5SDimitry Andric   // TODO: There is also f64 rsq instruction, but the documentation is less
52960b57cec5SDimitry Andric   // clear on its precision.
52970b57cec5SDimitry Andric 
52980b57cec5SDimitry Andric   return SDValue();
52990b57cec5SDimitry Andric }
53000b57cec5SDimitry Andric 
53010b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::getRecipEstimate(SDValue Operand,
53020b57cec5SDimitry Andric                                                SelectionDAG &DAG, int Enabled,
53030b57cec5SDimitry Andric                                                int &RefinementSteps) const {
53040b57cec5SDimitry Andric   EVT VT = Operand.getValueType();
53050b57cec5SDimitry Andric 
53060b57cec5SDimitry Andric   if (VT == MVT::f32) {
53070b57cec5SDimitry Andric     // Reciprocal, < 1 ulp error.
53080b57cec5SDimitry Andric     //
53090b57cec5SDimitry Andric     // This reciprocal approximation converges to < 0.5 ulp error with one
53100b57cec5SDimitry Andric     // newton rhapson performed with two fused multiple adds (FMAs).
53110b57cec5SDimitry Andric 
53120b57cec5SDimitry Andric     RefinementSteps = 0;
53130b57cec5SDimitry Andric     return DAG.getNode(AMDGPUISD::RCP, SDLoc(Operand), VT, Operand);
53140b57cec5SDimitry Andric   }
53150b57cec5SDimitry Andric 
53160b57cec5SDimitry Andric   // TODO: There is also f64 rcp instruction, but the documentation is less
53170b57cec5SDimitry Andric   // clear on its precision.
53180b57cec5SDimitry Andric 
53190b57cec5SDimitry Andric   return SDValue();
53200b57cec5SDimitry Andric }
53210b57cec5SDimitry Andric 
532281ad6265SDimitry Andric static unsigned workitemIntrinsicDim(unsigned ID) {
532381ad6265SDimitry Andric   switch (ID) {
532481ad6265SDimitry Andric   case Intrinsic::amdgcn_workitem_id_x:
532581ad6265SDimitry Andric     return 0;
532681ad6265SDimitry Andric   case Intrinsic::amdgcn_workitem_id_y:
532781ad6265SDimitry Andric     return 1;
532881ad6265SDimitry Andric   case Intrinsic::amdgcn_workitem_id_z:
532981ad6265SDimitry Andric     return 2;
533081ad6265SDimitry Andric   default:
533181ad6265SDimitry Andric     llvm_unreachable("not a workitem intrinsic");
533281ad6265SDimitry Andric   }
533381ad6265SDimitry Andric }
533481ad6265SDimitry Andric 
53350b57cec5SDimitry Andric void AMDGPUTargetLowering::computeKnownBitsForTargetNode(
53360b57cec5SDimitry Andric     const SDValue Op, KnownBits &Known,
53370b57cec5SDimitry Andric     const APInt &DemandedElts, const SelectionDAG &DAG, unsigned Depth) const {
53380b57cec5SDimitry Andric 
53390b57cec5SDimitry Andric   Known.resetAll(); // Don't know anything.
53400b57cec5SDimitry Andric 
53410b57cec5SDimitry Andric   unsigned Opc = Op.getOpcode();
53420b57cec5SDimitry Andric 
53430b57cec5SDimitry Andric   switch (Opc) {
53440b57cec5SDimitry Andric   default:
53450b57cec5SDimitry Andric     break;
53460b57cec5SDimitry Andric   case AMDGPUISD::CARRY:
53470b57cec5SDimitry Andric   case AMDGPUISD::BORROW: {
53480b57cec5SDimitry Andric     Known.Zero = APInt::getHighBitsSet(32, 31);
53490b57cec5SDimitry Andric     break;
53500b57cec5SDimitry Andric   }
53510b57cec5SDimitry Andric 
53520b57cec5SDimitry Andric   case AMDGPUISD::BFE_I32:
53530b57cec5SDimitry Andric   case AMDGPUISD::BFE_U32: {
53540b57cec5SDimitry Andric     ConstantSDNode *CWidth = dyn_cast<ConstantSDNode>(Op.getOperand(2));
53550b57cec5SDimitry Andric     if (!CWidth)
53560b57cec5SDimitry Andric       return;
53570b57cec5SDimitry Andric 
53580b57cec5SDimitry Andric     uint32_t Width = CWidth->getZExtValue() & 0x1f;
53590b57cec5SDimitry Andric 
53600b57cec5SDimitry Andric     if (Opc == AMDGPUISD::BFE_U32)
53610b57cec5SDimitry Andric       Known.Zero = APInt::getHighBitsSet(32, 32 - Width);
53620b57cec5SDimitry Andric 
53630b57cec5SDimitry Andric     break;
53640b57cec5SDimitry Andric   }
5365fe6060f1SDimitry Andric   case AMDGPUISD::FP_TO_FP16: {
53660b57cec5SDimitry Andric     unsigned BitWidth = Known.getBitWidth();
53670b57cec5SDimitry Andric 
53680b57cec5SDimitry Andric     // High bits are zero.
53690b57cec5SDimitry Andric     Known.Zero = APInt::getHighBitsSet(BitWidth, BitWidth - 16);
53700b57cec5SDimitry Andric     break;
53710b57cec5SDimitry Andric   }
53720b57cec5SDimitry Andric   case AMDGPUISD::MUL_U24:
53730b57cec5SDimitry Andric   case AMDGPUISD::MUL_I24: {
53740b57cec5SDimitry Andric     KnownBits LHSKnown = DAG.computeKnownBits(Op.getOperand(0), Depth + 1);
53750b57cec5SDimitry Andric     KnownBits RHSKnown = DAG.computeKnownBits(Op.getOperand(1), Depth + 1);
53760b57cec5SDimitry Andric     unsigned TrailZ = LHSKnown.countMinTrailingZeros() +
53770b57cec5SDimitry Andric                       RHSKnown.countMinTrailingZeros();
53780b57cec5SDimitry Andric     Known.Zero.setLowBits(std::min(TrailZ, 32u));
5379480093f4SDimitry Andric     // Skip extra check if all bits are known zeros.
5380480093f4SDimitry Andric     if (TrailZ >= 32)
5381480093f4SDimitry Andric       break;
53820b57cec5SDimitry Andric 
53830b57cec5SDimitry Andric     // Truncate to 24 bits.
53840b57cec5SDimitry Andric     LHSKnown = LHSKnown.trunc(24);
53850b57cec5SDimitry Andric     RHSKnown = RHSKnown.trunc(24);
53860b57cec5SDimitry Andric 
53870b57cec5SDimitry Andric     if (Opc == AMDGPUISD::MUL_I24) {
538804eeddc0SDimitry Andric       unsigned LHSValBits = LHSKnown.countMaxSignificantBits();
538904eeddc0SDimitry Andric       unsigned RHSValBits = RHSKnown.countMaxSignificantBits();
539004eeddc0SDimitry Andric       unsigned MaxValBits = LHSValBits + RHSValBits;
539104eeddc0SDimitry Andric       if (MaxValBits > 32)
53920b57cec5SDimitry Andric         break;
539304eeddc0SDimitry Andric       unsigned SignBits = 32 - MaxValBits + 1;
53940b57cec5SDimitry Andric       bool LHSNegative = LHSKnown.isNegative();
5395480093f4SDimitry Andric       bool LHSNonNegative = LHSKnown.isNonNegative();
5396480093f4SDimitry Andric       bool LHSPositive = LHSKnown.isStrictlyPositive();
53970b57cec5SDimitry Andric       bool RHSNegative = RHSKnown.isNegative();
5398480093f4SDimitry Andric       bool RHSNonNegative = RHSKnown.isNonNegative();
5399480093f4SDimitry Andric       bool RHSPositive = RHSKnown.isStrictlyPositive();
5400480093f4SDimitry Andric 
5401480093f4SDimitry Andric       if ((LHSNonNegative && RHSNonNegative) || (LHSNegative && RHSNegative))
540204eeddc0SDimitry Andric         Known.Zero.setHighBits(SignBits);
5403480093f4SDimitry Andric       else if ((LHSNegative && RHSPositive) || (LHSPositive && RHSNegative))
540404eeddc0SDimitry Andric         Known.One.setHighBits(SignBits);
54050b57cec5SDimitry Andric     } else {
540604eeddc0SDimitry Andric       unsigned LHSValBits = LHSKnown.countMaxActiveBits();
540704eeddc0SDimitry Andric       unsigned RHSValBits = RHSKnown.countMaxActiveBits();
540804eeddc0SDimitry Andric       unsigned MaxValBits = LHSValBits + RHSValBits;
54090b57cec5SDimitry Andric       if (MaxValBits >= 32)
54100b57cec5SDimitry Andric         break;
541104eeddc0SDimitry Andric       Known.Zero.setBitsFrom(MaxValBits);
54120b57cec5SDimitry Andric     }
54130b57cec5SDimitry Andric     break;
54140b57cec5SDimitry Andric   }
54150b57cec5SDimitry Andric   case AMDGPUISD::PERM: {
54160b57cec5SDimitry Andric     ConstantSDNode *CMask = dyn_cast<ConstantSDNode>(Op.getOperand(2));
54170b57cec5SDimitry Andric     if (!CMask)
54180b57cec5SDimitry Andric       return;
54190b57cec5SDimitry Andric 
54200b57cec5SDimitry Andric     KnownBits LHSKnown = DAG.computeKnownBits(Op.getOperand(0), Depth + 1);
54210b57cec5SDimitry Andric     KnownBits RHSKnown = DAG.computeKnownBits(Op.getOperand(1), Depth + 1);
54220b57cec5SDimitry Andric     unsigned Sel = CMask->getZExtValue();
54230b57cec5SDimitry Andric 
54240b57cec5SDimitry Andric     for (unsigned I = 0; I < 32; I += 8) {
54250b57cec5SDimitry Andric       unsigned SelBits = Sel & 0xff;
54260b57cec5SDimitry Andric       if (SelBits < 4) {
54270b57cec5SDimitry Andric         SelBits *= 8;
54280b57cec5SDimitry Andric         Known.One |= ((RHSKnown.One.getZExtValue() >> SelBits) & 0xff) << I;
54290b57cec5SDimitry Andric         Known.Zero |= ((RHSKnown.Zero.getZExtValue() >> SelBits) & 0xff) << I;
54300b57cec5SDimitry Andric       } else if (SelBits < 7) {
54310b57cec5SDimitry Andric         SelBits = (SelBits & 3) * 8;
54320b57cec5SDimitry Andric         Known.One |= ((LHSKnown.One.getZExtValue() >> SelBits) & 0xff) << I;
54330b57cec5SDimitry Andric         Known.Zero |= ((LHSKnown.Zero.getZExtValue() >> SelBits) & 0xff) << I;
54340b57cec5SDimitry Andric       } else if (SelBits == 0x0c) {
54358bcb0991SDimitry Andric         Known.Zero |= 0xFFull << I;
54360b57cec5SDimitry Andric       } else if (SelBits > 0x0c) {
54378bcb0991SDimitry Andric         Known.One |= 0xFFull << I;
54380b57cec5SDimitry Andric       }
54390b57cec5SDimitry Andric       Sel >>= 8;
54400b57cec5SDimitry Andric     }
54410b57cec5SDimitry Andric     break;
54420b57cec5SDimitry Andric   }
54430b57cec5SDimitry Andric   case AMDGPUISD::BUFFER_LOAD_UBYTE:  {
54440b57cec5SDimitry Andric     Known.Zero.setHighBits(24);
54450b57cec5SDimitry Andric     break;
54460b57cec5SDimitry Andric   }
54470b57cec5SDimitry Andric   case AMDGPUISD::BUFFER_LOAD_USHORT: {
54480b57cec5SDimitry Andric     Known.Zero.setHighBits(16);
54490b57cec5SDimitry Andric     break;
54500b57cec5SDimitry Andric   }
54510b57cec5SDimitry Andric   case AMDGPUISD::LDS: {
54520b57cec5SDimitry Andric     auto GA = cast<GlobalAddressSDNode>(Op.getOperand(0).getNode());
54535ffd83dbSDimitry Andric     Align Alignment = GA->getGlobal()->getPointerAlignment(DAG.getDataLayout());
54540b57cec5SDimitry Andric 
54550b57cec5SDimitry Andric     Known.Zero.setHighBits(16);
54565ffd83dbSDimitry Andric     Known.Zero.setLowBits(Log2(Alignment));
54570b57cec5SDimitry Andric     break;
54580b57cec5SDimitry Andric   }
545906c3fb27SDimitry Andric   case AMDGPUISD::SMIN3:
546006c3fb27SDimitry Andric   case AMDGPUISD::SMAX3:
546106c3fb27SDimitry Andric   case AMDGPUISD::SMED3:
546206c3fb27SDimitry Andric   case AMDGPUISD::UMIN3:
546306c3fb27SDimitry Andric   case AMDGPUISD::UMAX3:
546406c3fb27SDimitry Andric   case AMDGPUISD::UMED3: {
546506c3fb27SDimitry Andric     KnownBits Known2 = DAG.computeKnownBits(Op.getOperand(2), Depth + 1);
546606c3fb27SDimitry Andric     if (Known2.isUnknown())
546706c3fb27SDimitry Andric       break;
546806c3fb27SDimitry Andric 
546906c3fb27SDimitry Andric     KnownBits Known1 = DAG.computeKnownBits(Op.getOperand(1), Depth + 1);
547006c3fb27SDimitry Andric     if (Known1.isUnknown())
547106c3fb27SDimitry Andric       break;
547206c3fb27SDimitry Andric 
547306c3fb27SDimitry Andric     KnownBits Known0 = DAG.computeKnownBits(Op.getOperand(0), Depth + 1);
547406c3fb27SDimitry Andric     if (Known0.isUnknown())
547506c3fb27SDimitry Andric       break;
547606c3fb27SDimitry Andric 
547706c3fb27SDimitry Andric     // TODO: Handle LeadZero/LeadOne from UMIN/UMAX handling.
547806c3fb27SDimitry Andric     Known.Zero = Known0.Zero & Known1.Zero & Known2.Zero;
547906c3fb27SDimitry Andric     Known.One = Known0.One & Known1.One & Known2.One;
548006c3fb27SDimitry Andric     break;
548106c3fb27SDimitry Andric   }
54820b57cec5SDimitry Andric   case ISD::INTRINSIC_WO_CHAIN: {
54830b57cec5SDimitry Andric     unsigned IID = cast<ConstantSDNode>(Op.getOperand(0))->getZExtValue();
54840b57cec5SDimitry Andric     switch (IID) {
548581ad6265SDimitry Andric     case Intrinsic::amdgcn_workitem_id_x:
548681ad6265SDimitry Andric     case Intrinsic::amdgcn_workitem_id_y:
548781ad6265SDimitry Andric     case Intrinsic::amdgcn_workitem_id_z: {
548881ad6265SDimitry Andric       unsigned MaxValue = Subtarget->getMaxWorkitemID(
548981ad6265SDimitry Andric           DAG.getMachineFunction().getFunction(), workitemIntrinsicDim(IID));
549006c3fb27SDimitry Andric       Known.Zero.setHighBits(llvm::countl_zero(MaxValue));
549181ad6265SDimitry Andric       break;
549281ad6265SDimitry Andric     }
54930b57cec5SDimitry Andric     default:
54940b57cec5SDimitry Andric       break;
54950b57cec5SDimitry Andric     }
54960b57cec5SDimitry Andric   }
54970b57cec5SDimitry Andric   }
54980b57cec5SDimitry Andric }
54990b57cec5SDimitry Andric 
55000b57cec5SDimitry Andric unsigned AMDGPUTargetLowering::ComputeNumSignBitsForTargetNode(
55010b57cec5SDimitry Andric     SDValue Op, const APInt &DemandedElts, const SelectionDAG &DAG,
55020b57cec5SDimitry Andric     unsigned Depth) const {
55030b57cec5SDimitry Andric   switch (Op.getOpcode()) {
55040b57cec5SDimitry Andric   case AMDGPUISD::BFE_I32: {
55050b57cec5SDimitry Andric     ConstantSDNode *Width = dyn_cast<ConstantSDNode>(Op.getOperand(2));
55060b57cec5SDimitry Andric     if (!Width)
55070b57cec5SDimitry Andric       return 1;
55080b57cec5SDimitry Andric 
55090b57cec5SDimitry Andric     unsigned SignBits = 32 - Width->getZExtValue() + 1;
55100b57cec5SDimitry Andric     if (!isNullConstant(Op.getOperand(1)))
55110b57cec5SDimitry Andric       return SignBits;
55120b57cec5SDimitry Andric 
55130b57cec5SDimitry Andric     // TODO: Could probably figure something out with non-0 offsets.
55140b57cec5SDimitry Andric     unsigned Op0SignBits = DAG.ComputeNumSignBits(Op.getOperand(0), Depth + 1);
55150b57cec5SDimitry Andric     return std::max(SignBits, Op0SignBits);
55160b57cec5SDimitry Andric   }
55170b57cec5SDimitry Andric 
55180b57cec5SDimitry Andric   case AMDGPUISD::BFE_U32: {
55190b57cec5SDimitry Andric     ConstantSDNode *Width = dyn_cast<ConstantSDNode>(Op.getOperand(2));
55200b57cec5SDimitry Andric     return Width ? 32 - (Width->getZExtValue() & 0x1f) : 1;
55210b57cec5SDimitry Andric   }
55220b57cec5SDimitry Andric 
55230b57cec5SDimitry Andric   case AMDGPUISD::CARRY:
55240b57cec5SDimitry Andric   case AMDGPUISD::BORROW:
55250b57cec5SDimitry Andric     return 31;
55260b57cec5SDimitry Andric   case AMDGPUISD::BUFFER_LOAD_BYTE:
55270b57cec5SDimitry Andric     return 25;
55280b57cec5SDimitry Andric   case AMDGPUISD::BUFFER_LOAD_SHORT:
55290b57cec5SDimitry Andric     return 17;
55300b57cec5SDimitry Andric   case AMDGPUISD::BUFFER_LOAD_UBYTE:
55310b57cec5SDimitry Andric     return 24;
55320b57cec5SDimitry Andric   case AMDGPUISD::BUFFER_LOAD_USHORT:
55330b57cec5SDimitry Andric     return 16;
55340b57cec5SDimitry Andric   case AMDGPUISD::FP_TO_FP16:
55350b57cec5SDimitry Andric     return 16;
553606c3fb27SDimitry Andric   case AMDGPUISD::SMIN3:
553706c3fb27SDimitry Andric   case AMDGPUISD::SMAX3:
553806c3fb27SDimitry Andric   case AMDGPUISD::SMED3:
553906c3fb27SDimitry Andric   case AMDGPUISD::UMIN3:
554006c3fb27SDimitry Andric   case AMDGPUISD::UMAX3:
554106c3fb27SDimitry Andric   case AMDGPUISD::UMED3: {
554206c3fb27SDimitry Andric     unsigned Tmp2 = DAG.ComputeNumSignBits(Op.getOperand(2), Depth + 1);
554306c3fb27SDimitry Andric     if (Tmp2 == 1)
554406c3fb27SDimitry Andric       return 1; // Early out.
554506c3fb27SDimitry Andric 
554606c3fb27SDimitry Andric     unsigned Tmp1 = DAG.ComputeNumSignBits(Op.getOperand(1), Depth + 1);
554706c3fb27SDimitry Andric     if (Tmp1 == 1)
554806c3fb27SDimitry Andric       return 1; // Early out.
554906c3fb27SDimitry Andric 
555006c3fb27SDimitry Andric     unsigned Tmp0 = DAG.ComputeNumSignBits(Op.getOperand(0), Depth + 1);
555106c3fb27SDimitry Andric     if (Tmp0 == 1)
555206c3fb27SDimitry Andric       return 1; // Early out.
555306c3fb27SDimitry Andric 
555406c3fb27SDimitry Andric     return std::min(Tmp0, std::min(Tmp1, Tmp2));
555506c3fb27SDimitry Andric   }
55560b57cec5SDimitry Andric   default:
55570b57cec5SDimitry Andric     return 1;
55580b57cec5SDimitry Andric   }
55590b57cec5SDimitry Andric }
55600b57cec5SDimitry Andric 
55615ffd83dbSDimitry Andric unsigned AMDGPUTargetLowering::computeNumSignBitsForTargetInstr(
55625ffd83dbSDimitry Andric   GISelKnownBits &Analysis, Register R,
55635ffd83dbSDimitry Andric   const APInt &DemandedElts, const MachineRegisterInfo &MRI,
55645ffd83dbSDimitry Andric   unsigned Depth) const {
55655ffd83dbSDimitry Andric   const MachineInstr *MI = MRI.getVRegDef(R);
55665ffd83dbSDimitry Andric   if (!MI)
55675ffd83dbSDimitry Andric     return 1;
55685ffd83dbSDimitry Andric 
55695ffd83dbSDimitry Andric   // TODO: Check range metadata on MMO.
55705ffd83dbSDimitry Andric   switch (MI->getOpcode()) {
55715ffd83dbSDimitry Andric   case AMDGPU::G_AMDGPU_BUFFER_LOAD_SBYTE:
55725ffd83dbSDimitry Andric     return 25;
55735ffd83dbSDimitry Andric   case AMDGPU::G_AMDGPU_BUFFER_LOAD_SSHORT:
55745ffd83dbSDimitry Andric     return 17;
55755ffd83dbSDimitry Andric   case AMDGPU::G_AMDGPU_BUFFER_LOAD_UBYTE:
55765ffd83dbSDimitry Andric     return 24;
55775ffd83dbSDimitry Andric   case AMDGPU::G_AMDGPU_BUFFER_LOAD_USHORT:
55785ffd83dbSDimitry Andric     return 16;
557906c3fb27SDimitry Andric   case AMDGPU::G_AMDGPU_SMED3:
558006c3fb27SDimitry Andric   case AMDGPU::G_AMDGPU_UMED3: {
558106c3fb27SDimitry Andric     auto [Dst, Src0, Src1, Src2] = MI->getFirst4Regs();
558206c3fb27SDimitry Andric     unsigned Tmp2 = Analysis.computeNumSignBits(Src2, DemandedElts, Depth + 1);
558306c3fb27SDimitry Andric     if (Tmp2 == 1)
558406c3fb27SDimitry Andric       return 1;
558506c3fb27SDimitry Andric     unsigned Tmp1 = Analysis.computeNumSignBits(Src1, DemandedElts, Depth + 1);
558606c3fb27SDimitry Andric     if (Tmp1 == 1)
558706c3fb27SDimitry Andric       return 1;
558806c3fb27SDimitry Andric     unsigned Tmp0 = Analysis.computeNumSignBits(Src0, DemandedElts, Depth + 1);
558906c3fb27SDimitry Andric     if (Tmp0 == 1)
559006c3fb27SDimitry Andric       return 1;
559106c3fb27SDimitry Andric     return std::min(Tmp0, std::min(Tmp1, Tmp2));
559206c3fb27SDimitry Andric   }
55935ffd83dbSDimitry Andric   default:
55945ffd83dbSDimitry Andric     return 1;
55955ffd83dbSDimitry Andric   }
55965ffd83dbSDimitry Andric }
55975ffd83dbSDimitry Andric 
55980b57cec5SDimitry Andric bool AMDGPUTargetLowering::isKnownNeverNaNForTargetNode(SDValue Op,
55990b57cec5SDimitry Andric                                                         const SelectionDAG &DAG,
56000b57cec5SDimitry Andric                                                         bool SNaN,
56010b57cec5SDimitry Andric                                                         unsigned Depth) const {
56020b57cec5SDimitry Andric   unsigned Opcode = Op.getOpcode();
56030b57cec5SDimitry Andric   switch (Opcode) {
56040b57cec5SDimitry Andric   case AMDGPUISD::FMIN_LEGACY:
56050b57cec5SDimitry Andric   case AMDGPUISD::FMAX_LEGACY: {
56060b57cec5SDimitry Andric     if (SNaN)
56070b57cec5SDimitry Andric       return true;
56080b57cec5SDimitry Andric 
56090b57cec5SDimitry Andric     // TODO: Can check no nans on one of the operands for each one, but which
56100b57cec5SDimitry Andric     // one?
56110b57cec5SDimitry Andric     return false;
56120b57cec5SDimitry Andric   }
56130b57cec5SDimitry Andric   case AMDGPUISD::FMUL_LEGACY:
56140b57cec5SDimitry Andric   case AMDGPUISD::CVT_PKRTZ_F16_F32: {
56150b57cec5SDimitry Andric     if (SNaN)
56160b57cec5SDimitry Andric       return true;
56170b57cec5SDimitry Andric     return DAG.isKnownNeverNaN(Op.getOperand(0), SNaN, Depth + 1) &&
56180b57cec5SDimitry Andric            DAG.isKnownNeverNaN(Op.getOperand(1), SNaN, Depth + 1);
56190b57cec5SDimitry Andric   }
56200b57cec5SDimitry Andric   case AMDGPUISD::FMED3:
56210b57cec5SDimitry Andric   case AMDGPUISD::FMIN3:
56220b57cec5SDimitry Andric   case AMDGPUISD::FMAX3:
56230b57cec5SDimitry Andric   case AMDGPUISD::FMAD_FTZ: {
56240b57cec5SDimitry Andric     if (SNaN)
56250b57cec5SDimitry Andric       return true;
56260b57cec5SDimitry Andric     return DAG.isKnownNeverNaN(Op.getOperand(0), SNaN, Depth + 1) &&
56270b57cec5SDimitry Andric            DAG.isKnownNeverNaN(Op.getOperand(1), SNaN, Depth + 1) &&
56280b57cec5SDimitry Andric            DAG.isKnownNeverNaN(Op.getOperand(2), SNaN, Depth + 1);
56290b57cec5SDimitry Andric   }
56300b57cec5SDimitry Andric   case AMDGPUISD::CVT_F32_UBYTE0:
56310b57cec5SDimitry Andric   case AMDGPUISD::CVT_F32_UBYTE1:
56320b57cec5SDimitry Andric   case AMDGPUISD::CVT_F32_UBYTE2:
56330b57cec5SDimitry Andric   case AMDGPUISD::CVT_F32_UBYTE3:
56340b57cec5SDimitry Andric     return true;
56350b57cec5SDimitry Andric 
56360b57cec5SDimitry Andric   case AMDGPUISD::RCP:
56370b57cec5SDimitry Andric   case AMDGPUISD::RSQ:
56380b57cec5SDimitry Andric   case AMDGPUISD::RCP_LEGACY:
56390b57cec5SDimitry Andric   case AMDGPUISD::RSQ_CLAMP: {
56400b57cec5SDimitry Andric     if (SNaN)
56410b57cec5SDimitry Andric       return true;
56420b57cec5SDimitry Andric 
56430b57cec5SDimitry Andric     // TODO: Need is known positive check.
56440b57cec5SDimitry Andric     return false;
56450b57cec5SDimitry Andric   }
564606c3fb27SDimitry Andric   case ISD::FLDEXP:
56470b57cec5SDimitry Andric   case AMDGPUISD::FRACT: {
56480b57cec5SDimitry Andric     if (SNaN)
56490b57cec5SDimitry Andric       return true;
56500b57cec5SDimitry Andric     return DAG.isKnownNeverNaN(Op.getOperand(0), SNaN, Depth + 1);
56510b57cec5SDimitry Andric   }
56520b57cec5SDimitry Andric   case AMDGPUISD::DIV_SCALE:
56530b57cec5SDimitry Andric   case AMDGPUISD::DIV_FMAS:
56540b57cec5SDimitry Andric   case AMDGPUISD::DIV_FIXUP:
56550b57cec5SDimitry Andric     // TODO: Refine on operands.
56560b57cec5SDimitry Andric     return SNaN;
56570b57cec5SDimitry Andric   case AMDGPUISD::SIN_HW:
56580b57cec5SDimitry Andric   case AMDGPUISD::COS_HW: {
56590b57cec5SDimitry Andric     // TODO: Need check for infinity
56600b57cec5SDimitry Andric     return SNaN;
56610b57cec5SDimitry Andric   }
56620b57cec5SDimitry Andric   case ISD::INTRINSIC_WO_CHAIN: {
56630b57cec5SDimitry Andric     unsigned IntrinsicID
56640b57cec5SDimitry Andric       = cast<ConstantSDNode>(Op.getOperand(0))->getZExtValue();
56650b57cec5SDimitry Andric     // TODO: Handle more intrinsics
56660b57cec5SDimitry Andric     switch (IntrinsicID) {
56670b57cec5SDimitry Andric     case Intrinsic::amdgcn_cubeid:
56680b57cec5SDimitry Andric       return true;
56690b57cec5SDimitry Andric 
56700b57cec5SDimitry Andric     case Intrinsic::amdgcn_frexp_mant: {
56710b57cec5SDimitry Andric       if (SNaN)
56720b57cec5SDimitry Andric         return true;
56730b57cec5SDimitry Andric       return DAG.isKnownNeverNaN(Op.getOperand(1), SNaN, Depth + 1);
56740b57cec5SDimitry Andric     }
56750b57cec5SDimitry Andric     case Intrinsic::amdgcn_cvt_pkrtz: {
56760b57cec5SDimitry Andric       if (SNaN)
56770b57cec5SDimitry Andric         return true;
56780b57cec5SDimitry Andric       return DAG.isKnownNeverNaN(Op.getOperand(1), SNaN, Depth + 1) &&
56790b57cec5SDimitry Andric              DAG.isKnownNeverNaN(Op.getOperand(2), SNaN, Depth + 1);
56800b57cec5SDimitry Andric     }
56815ffd83dbSDimitry Andric     case Intrinsic::amdgcn_rcp:
56825ffd83dbSDimitry Andric     case Intrinsic::amdgcn_rsq:
56835ffd83dbSDimitry Andric     case Intrinsic::amdgcn_rcp_legacy:
56845ffd83dbSDimitry Andric     case Intrinsic::amdgcn_rsq_legacy:
56855ffd83dbSDimitry Andric     case Intrinsic::amdgcn_rsq_clamp: {
56865ffd83dbSDimitry Andric       if (SNaN)
56875ffd83dbSDimitry Andric         return true;
56885ffd83dbSDimitry Andric 
56895ffd83dbSDimitry Andric       // TODO: Need is known positive check.
56905ffd83dbSDimitry Andric       return false;
56915ffd83dbSDimitry Andric     }
56925ffd83dbSDimitry Andric     case Intrinsic::amdgcn_trig_preop:
56930b57cec5SDimitry Andric     case Intrinsic::amdgcn_fdot2:
56940b57cec5SDimitry Andric       // TODO: Refine on operand
56950b57cec5SDimitry Andric       return SNaN;
5696e8d8bef9SDimitry Andric     case Intrinsic::amdgcn_fma_legacy:
5697e8d8bef9SDimitry Andric       if (SNaN)
5698e8d8bef9SDimitry Andric         return true;
5699e8d8bef9SDimitry Andric       return DAG.isKnownNeverNaN(Op.getOperand(1), SNaN, Depth + 1) &&
5700e8d8bef9SDimitry Andric              DAG.isKnownNeverNaN(Op.getOperand(2), SNaN, Depth + 1) &&
5701e8d8bef9SDimitry Andric              DAG.isKnownNeverNaN(Op.getOperand(3), SNaN, Depth + 1);
57020b57cec5SDimitry Andric     default:
57030b57cec5SDimitry Andric       return false;
57040b57cec5SDimitry Andric     }
57050b57cec5SDimitry Andric   }
57060b57cec5SDimitry Andric   default:
57070b57cec5SDimitry Andric     return false;
57080b57cec5SDimitry Andric   }
57090b57cec5SDimitry Andric }
57100b57cec5SDimitry Andric 
571106c3fb27SDimitry Andric bool AMDGPUTargetLowering::isReassocProfitable(MachineRegisterInfo &MRI,
571206c3fb27SDimitry Andric                                                Register N0, Register N1) const {
571306c3fb27SDimitry Andric   return MRI.hasOneNonDBGUse(N0); // FIXME: handle regbanks
571406c3fb27SDimitry Andric }
571506c3fb27SDimitry Andric 
57160b57cec5SDimitry Andric TargetLowering::AtomicExpansionKind
57170b57cec5SDimitry Andric AMDGPUTargetLowering::shouldExpandAtomicRMWInIR(AtomicRMWInst *RMW) const {
57180b57cec5SDimitry Andric   switch (RMW->getOperation()) {
57190b57cec5SDimitry Andric   case AtomicRMWInst::Nand:
57200b57cec5SDimitry Andric   case AtomicRMWInst::FAdd:
57210b57cec5SDimitry Andric   case AtomicRMWInst::FSub:
5722753f127fSDimitry Andric   case AtomicRMWInst::FMax:
5723753f127fSDimitry Andric   case AtomicRMWInst::FMin:
57240b57cec5SDimitry Andric     return AtomicExpansionKind::CmpXChg;
5725bdd1243dSDimitry Andric   default: {
5726bdd1243dSDimitry Andric     if (auto *IntTy = dyn_cast<IntegerType>(RMW->getType())) {
5727bdd1243dSDimitry Andric       unsigned Size = IntTy->getBitWidth();
5728bdd1243dSDimitry Andric       if (Size == 32 || Size == 64)
57290b57cec5SDimitry Andric         return AtomicExpansionKind::None;
57300b57cec5SDimitry Andric     }
5731bdd1243dSDimitry Andric 
5732bdd1243dSDimitry Andric     return AtomicExpansionKind::CmpXChg;
5733bdd1243dSDimitry Andric   }
5734bdd1243dSDimitry Andric   }
57350b57cec5SDimitry Andric }
5736fe6060f1SDimitry Andric 
573704eeddc0SDimitry Andric bool AMDGPUTargetLowering::isConstantUnsignedBitfieldExtractLegal(
5738fe6060f1SDimitry Andric     unsigned Opc, LLT Ty1, LLT Ty2) const {
573904eeddc0SDimitry Andric   return (Ty1 == LLT::scalar(32) || Ty1 == LLT::scalar(64)) &&
574004eeddc0SDimitry Andric          Ty2 == LLT::scalar(32);
5741fe6060f1SDimitry Andric }
574206c3fb27SDimitry Andric 
574306c3fb27SDimitry Andric /// Whether it is profitable to sink the operands of an
574406c3fb27SDimitry Andric /// Instruction I to the basic block of I.
574506c3fb27SDimitry Andric /// This helps using several modifiers (like abs and neg) more often.
574606c3fb27SDimitry Andric bool AMDGPUTargetLowering::shouldSinkOperands(
574706c3fb27SDimitry Andric     Instruction *I, SmallVectorImpl<Use *> &Ops) const {
574806c3fb27SDimitry Andric   using namespace PatternMatch;
574906c3fb27SDimitry Andric 
575006c3fb27SDimitry Andric   for (auto &Op : I->operands()) {
575106c3fb27SDimitry Andric     // Ensure we are not already sinking this operand.
575206c3fb27SDimitry Andric     if (any_of(Ops, [&](Use *U) { return U->get() == Op.get(); }))
575306c3fb27SDimitry Andric       continue;
575406c3fb27SDimitry Andric 
575506c3fb27SDimitry Andric     if (match(&Op, m_FAbs(m_Value())) || match(&Op, m_FNeg(m_Value())))
575606c3fb27SDimitry Andric       Ops.push_back(&Op);
575706c3fb27SDimitry Andric   }
575806c3fb27SDimitry Andric 
575906c3fb27SDimitry Andric   return !Ops.empty();
576006c3fb27SDimitry Andric }
5761