xref: /freebsd/contrib/llvm-project/llvm/lib/Target/AMDGPU/AMDGPUISelLowering.cpp (revision 8bcb0991864975618c09697b1aca10683346d9f0)
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"
170b57cec5SDimitry Andric #include "AMDGPUCallLowering.h"
180b57cec5SDimitry Andric #include "AMDGPUFrameLowering.h"
190b57cec5SDimitry Andric #include "AMDGPURegisterInfo.h"
200b57cec5SDimitry Andric #include "AMDGPUSubtarget.h"
210b57cec5SDimitry Andric #include "AMDGPUTargetMachine.h"
220b57cec5SDimitry Andric #include "Utils/AMDGPUBaseInfo.h"
230b57cec5SDimitry Andric #include "R600MachineFunctionInfo.h"
240b57cec5SDimitry Andric #include "SIInstrInfo.h"
250b57cec5SDimitry Andric #include "SIMachineFunctionInfo.h"
260b57cec5SDimitry Andric #include "MCTargetDesc/AMDGPUMCTargetDesc.h"
270b57cec5SDimitry Andric #include "llvm/CodeGen/Analysis.h"
280b57cec5SDimitry Andric #include "llvm/CodeGen/CallingConvLower.h"
290b57cec5SDimitry Andric #include "llvm/CodeGen/MachineFunction.h"
300b57cec5SDimitry Andric #include "llvm/CodeGen/MachineRegisterInfo.h"
310b57cec5SDimitry Andric #include "llvm/CodeGen/SelectionDAG.h"
320b57cec5SDimitry Andric #include "llvm/CodeGen/TargetLoweringObjectFileImpl.h"
330b57cec5SDimitry Andric #include "llvm/IR/DataLayout.h"
340b57cec5SDimitry Andric #include "llvm/IR/DiagnosticInfo.h"
350b57cec5SDimitry Andric #include "llvm/Support/KnownBits.h"
36*8bcb0991SDimitry Andric #include "llvm/Support/MathExtras.h"
370b57cec5SDimitry Andric using namespace llvm;
380b57cec5SDimitry Andric 
390b57cec5SDimitry Andric #include "AMDGPUGenCallingConv.inc"
400b57cec5SDimitry Andric 
410b57cec5SDimitry Andric // Find a larger type to do a load / store of a vector with.
420b57cec5SDimitry Andric EVT AMDGPUTargetLowering::getEquivalentMemType(LLVMContext &Ctx, EVT VT) {
430b57cec5SDimitry Andric   unsigned StoreSize = VT.getStoreSizeInBits();
440b57cec5SDimitry Andric   if (StoreSize <= 32)
450b57cec5SDimitry Andric     return EVT::getIntegerVT(Ctx, StoreSize);
460b57cec5SDimitry Andric 
470b57cec5SDimitry Andric   assert(StoreSize % 32 == 0 && "Store size not a multiple of 32");
480b57cec5SDimitry Andric   return EVT::getVectorVT(Ctx, MVT::i32, StoreSize / 32);
490b57cec5SDimitry Andric }
500b57cec5SDimitry Andric 
510b57cec5SDimitry Andric unsigned AMDGPUTargetLowering::numBitsUnsigned(SDValue Op, SelectionDAG &DAG) {
520b57cec5SDimitry Andric   EVT VT = Op.getValueType();
530b57cec5SDimitry Andric   KnownBits Known = DAG.computeKnownBits(Op);
540b57cec5SDimitry Andric   return VT.getSizeInBits() - Known.countMinLeadingZeros();
550b57cec5SDimitry Andric }
560b57cec5SDimitry Andric 
570b57cec5SDimitry Andric unsigned AMDGPUTargetLowering::numBitsSigned(SDValue Op, SelectionDAG &DAG) {
580b57cec5SDimitry Andric   EVT VT = Op.getValueType();
590b57cec5SDimitry Andric 
600b57cec5SDimitry Andric   // In order for this to be a signed 24-bit value, bit 23, must
610b57cec5SDimitry Andric   // be a sign bit.
620b57cec5SDimitry Andric   return VT.getSizeInBits() - DAG.ComputeNumSignBits(Op);
630b57cec5SDimitry Andric }
640b57cec5SDimitry Andric 
650b57cec5SDimitry Andric AMDGPUTargetLowering::AMDGPUTargetLowering(const TargetMachine &TM,
660b57cec5SDimitry Andric                                            const AMDGPUSubtarget &STI)
670b57cec5SDimitry Andric     : TargetLowering(TM), Subtarget(&STI) {
680b57cec5SDimitry Andric   // Lower floating point store/load to integer store/load to reduce the number
690b57cec5SDimitry Andric   // of patterns in tablegen.
700b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::f32, Promote);
710b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::f32, MVT::i32);
720b57cec5SDimitry Andric 
730b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v2f32, Promote);
740b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v2f32, MVT::v2i32);
750b57cec5SDimitry Andric 
760b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v3f32, Promote);
770b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v3f32, MVT::v3i32);
780b57cec5SDimitry Andric 
790b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v4f32, Promote);
800b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v4f32, MVT::v4i32);
810b57cec5SDimitry Andric 
820b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v5f32, Promote);
830b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v5f32, MVT::v5i32);
840b57cec5SDimitry Andric 
850b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v8f32, Promote);
860b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v8f32, MVT::v8i32);
870b57cec5SDimitry Andric 
880b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v16f32, Promote);
890b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v16f32, MVT::v16i32);
900b57cec5SDimitry Andric 
910b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v32f32, Promote);
920b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v32f32, MVT::v32i32);
930b57cec5SDimitry Andric 
940b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::i64, Promote);
950b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::i64, MVT::v2i32);
960b57cec5SDimitry Andric 
970b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v2i64, Promote);
980b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v2i64, MVT::v4i32);
990b57cec5SDimitry Andric 
1000b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::f64, Promote);
1010b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::f64, MVT::v2i32);
1020b57cec5SDimitry Andric 
1030b57cec5SDimitry Andric   setOperationAction(ISD::LOAD, MVT::v2f64, Promote);
1040b57cec5SDimitry Andric   AddPromotedToType(ISD::LOAD, MVT::v2f64, MVT::v4i32);
1050b57cec5SDimitry Andric 
1060b57cec5SDimitry Andric   // There are no 64-bit extloads. These should be done as a 32-bit extload and
1070b57cec5SDimitry Andric   // an extension to 64-bit.
1080b57cec5SDimitry Andric   for (MVT VT : MVT::integer_valuetypes()) {
1090b57cec5SDimitry Andric     setLoadExtAction(ISD::EXTLOAD, MVT::i64, VT, Expand);
1100b57cec5SDimitry Andric     setLoadExtAction(ISD::SEXTLOAD, MVT::i64, VT, Expand);
1110b57cec5SDimitry Andric     setLoadExtAction(ISD::ZEXTLOAD, MVT::i64, VT, Expand);
1120b57cec5SDimitry Andric   }
1130b57cec5SDimitry Andric 
1140b57cec5SDimitry Andric   for (MVT VT : MVT::integer_valuetypes()) {
1150b57cec5SDimitry Andric     if (VT == MVT::i64)
1160b57cec5SDimitry Andric       continue;
1170b57cec5SDimitry Andric 
1180b57cec5SDimitry Andric     setLoadExtAction(ISD::SEXTLOAD, VT, MVT::i1, Promote);
1190b57cec5SDimitry Andric     setLoadExtAction(ISD::SEXTLOAD, VT, MVT::i8, Legal);
1200b57cec5SDimitry Andric     setLoadExtAction(ISD::SEXTLOAD, VT, MVT::i16, Legal);
1210b57cec5SDimitry Andric     setLoadExtAction(ISD::SEXTLOAD, VT, MVT::i32, Expand);
1220b57cec5SDimitry Andric 
1230b57cec5SDimitry Andric     setLoadExtAction(ISD::ZEXTLOAD, VT, MVT::i1, Promote);
1240b57cec5SDimitry Andric     setLoadExtAction(ISD::ZEXTLOAD, VT, MVT::i8, Legal);
1250b57cec5SDimitry Andric     setLoadExtAction(ISD::ZEXTLOAD, VT, MVT::i16, Legal);
1260b57cec5SDimitry Andric     setLoadExtAction(ISD::ZEXTLOAD, VT, MVT::i32, Expand);
1270b57cec5SDimitry Andric 
1280b57cec5SDimitry Andric     setLoadExtAction(ISD::EXTLOAD, VT, MVT::i1, Promote);
1290b57cec5SDimitry Andric     setLoadExtAction(ISD::EXTLOAD, VT, MVT::i8, Legal);
1300b57cec5SDimitry Andric     setLoadExtAction(ISD::EXTLOAD, VT, MVT::i16, Legal);
1310b57cec5SDimitry Andric     setLoadExtAction(ISD::EXTLOAD, VT, MVT::i32, Expand);
1320b57cec5SDimitry Andric   }
1330b57cec5SDimitry Andric 
134*8bcb0991SDimitry Andric   for (MVT VT : MVT::integer_fixedlen_vector_valuetypes()) {
1350b57cec5SDimitry Andric     setLoadExtAction(ISD::EXTLOAD, VT, MVT::v2i8, Expand);
1360b57cec5SDimitry Andric     setLoadExtAction(ISD::SEXTLOAD, VT, MVT::v2i8, Expand);
1370b57cec5SDimitry Andric     setLoadExtAction(ISD::ZEXTLOAD, VT, MVT::v2i8, Expand);
1380b57cec5SDimitry Andric     setLoadExtAction(ISD::EXTLOAD, VT, MVT::v4i8, Expand);
1390b57cec5SDimitry Andric     setLoadExtAction(ISD::SEXTLOAD, VT, MVT::v4i8, Expand);
1400b57cec5SDimitry Andric     setLoadExtAction(ISD::ZEXTLOAD, VT, MVT::v4i8, Expand);
1410b57cec5SDimitry Andric     setLoadExtAction(ISD::EXTLOAD, VT, MVT::v2i16, Expand);
1420b57cec5SDimitry Andric     setLoadExtAction(ISD::SEXTLOAD, VT, MVT::v2i16, Expand);
1430b57cec5SDimitry Andric     setLoadExtAction(ISD::ZEXTLOAD, VT, MVT::v2i16, Expand);
144*8bcb0991SDimitry Andric     setLoadExtAction(ISD::EXTLOAD, VT, MVT::v3i16, Expand);
145*8bcb0991SDimitry Andric     setLoadExtAction(ISD::SEXTLOAD, VT, MVT::v3i16, Expand);
146*8bcb0991SDimitry Andric     setLoadExtAction(ISD::ZEXTLOAD, VT, MVT::v3i16, Expand);
1470b57cec5SDimitry Andric     setLoadExtAction(ISD::EXTLOAD, VT, MVT::v4i16, Expand);
1480b57cec5SDimitry Andric     setLoadExtAction(ISD::SEXTLOAD, VT, MVT::v4i16, Expand);
1490b57cec5SDimitry Andric     setLoadExtAction(ISD::ZEXTLOAD, VT, MVT::v4i16, Expand);
1500b57cec5SDimitry Andric   }
1510b57cec5SDimitry Andric 
1520b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::f32, MVT::f16, Expand);
1530b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v2f32, MVT::v2f16, Expand);
154*8bcb0991SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v3f32, MVT::v3f16, Expand);
1550b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v4f32, MVT::v4f16, Expand);
1560b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v8f32, MVT::v8f16, Expand);
157*8bcb0991SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v16f32, MVT::v16f16, Expand);
158*8bcb0991SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v32f32, MVT::v32f16, Expand);
1590b57cec5SDimitry Andric 
1600b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::f64, MVT::f32, Expand);
1610b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v2f64, MVT::v2f32, Expand);
1620b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v4f64, MVT::v4f32, Expand);
1630b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v8f64, MVT::v8f32, Expand);
1640b57cec5SDimitry Andric 
1650b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::f64, MVT::f16, Expand);
1660b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v2f64, MVT::v2f16, Expand);
1670b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v4f64, MVT::v4f16, Expand);
1680b57cec5SDimitry Andric   setLoadExtAction(ISD::EXTLOAD, MVT::v8f64, MVT::v8f16, Expand);
1690b57cec5SDimitry Andric 
1700b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::f32, Promote);
1710b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::f32, MVT::i32);
1720b57cec5SDimitry Andric 
1730b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v2f32, Promote);
1740b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v2f32, MVT::v2i32);
1750b57cec5SDimitry Andric 
1760b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v3f32, Promote);
1770b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v3f32, MVT::v3i32);
1780b57cec5SDimitry Andric 
1790b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v4f32, Promote);
1800b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v4f32, MVT::v4i32);
1810b57cec5SDimitry Andric 
1820b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v5f32, Promote);
1830b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v5f32, MVT::v5i32);
1840b57cec5SDimitry Andric 
1850b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v8f32, Promote);
1860b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v8f32, MVT::v8i32);
1870b57cec5SDimitry Andric 
1880b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v16f32, Promote);
1890b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v16f32, MVT::v16i32);
1900b57cec5SDimitry Andric 
1910b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v32f32, Promote);
1920b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v32f32, MVT::v32i32);
1930b57cec5SDimitry Andric 
1940b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::i64, Promote);
1950b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::i64, MVT::v2i32);
1960b57cec5SDimitry Andric 
1970b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v2i64, Promote);
1980b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v2i64, MVT::v4i32);
1990b57cec5SDimitry Andric 
2000b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::f64, Promote);
2010b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::f64, MVT::v2i32);
2020b57cec5SDimitry Andric 
2030b57cec5SDimitry Andric   setOperationAction(ISD::STORE, MVT::v2f64, Promote);
2040b57cec5SDimitry Andric   AddPromotedToType(ISD::STORE, MVT::v2f64, MVT::v4i32);
2050b57cec5SDimitry Andric 
2060b57cec5SDimitry Andric   setTruncStoreAction(MVT::i64, MVT::i1, Expand);
2070b57cec5SDimitry Andric   setTruncStoreAction(MVT::i64, MVT::i8, Expand);
2080b57cec5SDimitry Andric   setTruncStoreAction(MVT::i64, MVT::i16, Expand);
2090b57cec5SDimitry Andric   setTruncStoreAction(MVT::i64, MVT::i32, Expand);
2100b57cec5SDimitry Andric 
2110b57cec5SDimitry Andric   setTruncStoreAction(MVT::v2i64, MVT::v2i1, Expand);
2120b57cec5SDimitry Andric   setTruncStoreAction(MVT::v2i64, MVT::v2i8, Expand);
2130b57cec5SDimitry Andric   setTruncStoreAction(MVT::v2i64, MVT::v2i16, Expand);
2140b57cec5SDimitry Andric   setTruncStoreAction(MVT::v2i64, MVT::v2i32, Expand);
2150b57cec5SDimitry Andric 
2160b57cec5SDimitry Andric   setTruncStoreAction(MVT::f32, MVT::f16, Expand);
2170b57cec5SDimitry Andric   setTruncStoreAction(MVT::v2f32, MVT::v2f16, Expand);
218*8bcb0991SDimitry Andric   setTruncStoreAction(MVT::v3f32, MVT::v3f16, Expand);
2190b57cec5SDimitry Andric   setTruncStoreAction(MVT::v4f32, MVT::v4f16, Expand);
2200b57cec5SDimitry Andric   setTruncStoreAction(MVT::v8f32, MVT::v8f16, Expand);
221*8bcb0991SDimitry Andric   setTruncStoreAction(MVT::v16f32, MVT::v16f16, Expand);
222*8bcb0991SDimitry Andric   setTruncStoreAction(MVT::v32f32, MVT::v32f16, Expand);
2230b57cec5SDimitry Andric 
2240b57cec5SDimitry Andric   setTruncStoreAction(MVT::f64, MVT::f16, Expand);
2250b57cec5SDimitry Andric   setTruncStoreAction(MVT::f64, MVT::f32, Expand);
2260b57cec5SDimitry Andric 
2270b57cec5SDimitry Andric   setTruncStoreAction(MVT::v2f64, MVT::v2f32, Expand);
2280b57cec5SDimitry Andric   setTruncStoreAction(MVT::v2f64, MVT::v2f16, Expand);
2290b57cec5SDimitry Andric 
2300b57cec5SDimitry Andric   setTruncStoreAction(MVT::v4f64, MVT::v4f32, Expand);
2310b57cec5SDimitry Andric   setTruncStoreAction(MVT::v4f64, MVT::v4f16, Expand);
2320b57cec5SDimitry Andric 
2330b57cec5SDimitry Andric   setTruncStoreAction(MVT::v8f64, MVT::v8f32, Expand);
2340b57cec5SDimitry Andric   setTruncStoreAction(MVT::v8f64, MVT::v8f16, Expand);
2350b57cec5SDimitry Andric 
2360b57cec5SDimitry Andric 
2370b57cec5SDimitry Andric   setOperationAction(ISD::Constant, MVT::i32, Legal);
2380b57cec5SDimitry Andric   setOperationAction(ISD::Constant, MVT::i64, Legal);
2390b57cec5SDimitry Andric   setOperationAction(ISD::ConstantFP, MVT::f32, Legal);
2400b57cec5SDimitry Andric   setOperationAction(ISD::ConstantFP, MVT::f64, Legal);
2410b57cec5SDimitry Andric 
2420b57cec5SDimitry Andric   setOperationAction(ISD::BR_JT, MVT::Other, Expand);
2430b57cec5SDimitry Andric   setOperationAction(ISD::BRIND, MVT::Other, Expand);
2440b57cec5SDimitry Andric 
2450b57cec5SDimitry Andric   // This is totally unsupported, just custom lower to produce an error.
2460b57cec5SDimitry Andric   setOperationAction(ISD::DYNAMIC_STACKALLOC, MVT::i32, Custom);
2470b57cec5SDimitry Andric 
2480b57cec5SDimitry Andric   // Library functions.  These default to Expand, but we have instructions
2490b57cec5SDimitry Andric   // for them.
2500b57cec5SDimitry Andric   setOperationAction(ISD::FCEIL,  MVT::f32, Legal);
2510b57cec5SDimitry Andric   setOperationAction(ISD::FEXP2,  MVT::f32, Legal);
2520b57cec5SDimitry Andric   setOperationAction(ISD::FPOW,   MVT::f32, Legal);
2530b57cec5SDimitry Andric   setOperationAction(ISD::FLOG2,  MVT::f32, Legal);
2540b57cec5SDimitry Andric   setOperationAction(ISD::FABS,   MVT::f32, Legal);
2550b57cec5SDimitry Andric   setOperationAction(ISD::FFLOOR, MVT::f32, Legal);
2560b57cec5SDimitry Andric   setOperationAction(ISD::FRINT,  MVT::f32, Legal);
2570b57cec5SDimitry Andric   setOperationAction(ISD::FTRUNC, MVT::f32, Legal);
2580b57cec5SDimitry Andric   setOperationAction(ISD::FMINNUM, MVT::f32, Legal);
2590b57cec5SDimitry Andric   setOperationAction(ISD::FMAXNUM, MVT::f32, Legal);
2600b57cec5SDimitry Andric 
2610b57cec5SDimitry Andric   setOperationAction(ISD::FROUND, MVT::f32, Custom);
2620b57cec5SDimitry Andric   setOperationAction(ISD::FROUND, MVT::f64, Custom);
2630b57cec5SDimitry Andric 
2640b57cec5SDimitry Andric   setOperationAction(ISD::FLOG, MVT::f32, Custom);
2650b57cec5SDimitry Andric   setOperationAction(ISD::FLOG10, MVT::f32, Custom);
2660b57cec5SDimitry Andric   setOperationAction(ISD::FEXP, MVT::f32, Custom);
2670b57cec5SDimitry Andric 
2680b57cec5SDimitry Andric 
2690b57cec5SDimitry Andric   setOperationAction(ISD::FNEARBYINT, MVT::f32, Custom);
2700b57cec5SDimitry Andric   setOperationAction(ISD::FNEARBYINT, MVT::f64, Custom);
2710b57cec5SDimitry Andric 
2720b57cec5SDimitry Andric   setOperationAction(ISD::FREM, MVT::f32, Custom);
2730b57cec5SDimitry Andric   setOperationAction(ISD::FREM, MVT::f64, Custom);
2740b57cec5SDimitry Andric 
2750b57cec5SDimitry Andric   // Expand to fneg + fadd.
2760b57cec5SDimitry Andric   setOperationAction(ISD::FSUB, MVT::f64, Expand);
2770b57cec5SDimitry Andric 
2780b57cec5SDimitry Andric   setOperationAction(ISD::CONCAT_VECTORS, MVT::v3i32, Custom);
2790b57cec5SDimitry Andric   setOperationAction(ISD::CONCAT_VECTORS, MVT::v3f32, Custom);
2800b57cec5SDimitry Andric   setOperationAction(ISD::CONCAT_VECTORS, MVT::v4i32, Custom);
2810b57cec5SDimitry Andric   setOperationAction(ISD::CONCAT_VECTORS, MVT::v4f32, Custom);
2820b57cec5SDimitry Andric   setOperationAction(ISD::CONCAT_VECTORS, MVT::v5i32, Custom);
2830b57cec5SDimitry Andric   setOperationAction(ISD::CONCAT_VECTORS, MVT::v5f32, Custom);
2840b57cec5SDimitry Andric   setOperationAction(ISD::CONCAT_VECTORS, MVT::v8i32, Custom);
2850b57cec5SDimitry Andric   setOperationAction(ISD::CONCAT_VECTORS, MVT::v8f32, Custom);
2860b57cec5SDimitry Andric   setOperationAction(ISD::EXTRACT_SUBVECTOR, MVT::v2f32, Custom);
2870b57cec5SDimitry Andric   setOperationAction(ISD::EXTRACT_SUBVECTOR, MVT::v2i32, Custom);
2880b57cec5SDimitry Andric   setOperationAction(ISD::EXTRACT_SUBVECTOR, MVT::v3f32, Custom);
2890b57cec5SDimitry Andric   setOperationAction(ISD::EXTRACT_SUBVECTOR, MVT::v3i32, Custom);
2900b57cec5SDimitry Andric   setOperationAction(ISD::EXTRACT_SUBVECTOR, MVT::v4f32, Custom);
2910b57cec5SDimitry Andric   setOperationAction(ISD::EXTRACT_SUBVECTOR, MVT::v4i32, Custom);
2920b57cec5SDimitry Andric   setOperationAction(ISD::EXTRACT_SUBVECTOR, MVT::v5f32, Custom);
2930b57cec5SDimitry Andric   setOperationAction(ISD::EXTRACT_SUBVECTOR, MVT::v5i32, Custom);
2940b57cec5SDimitry Andric   setOperationAction(ISD::EXTRACT_SUBVECTOR, MVT::v8f32, Custom);
2950b57cec5SDimitry Andric   setOperationAction(ISD::EXTRACT_SUBVECTOR, MVT::v8i32, Custom);
2960b57cec5SDimitry Andric   setOperationAction(ISD::EXTRACT_SUBVECTOR, MVT::v16f32, Custom);
2970b57cec5SDimitry Andric   setOperationAction(ISD::EXTRACT_SUBVECTOR, MVT::v16i32, Custom);
2980b57cec5SDimitry Andric   setOperationAction(ISD::EXTRACT_SUBVECTOR, MVT::v32f32, Custom);
2990b57cec5SDimitry Andric   setOperationAction(ISD::EXTRACT_SUBVECTOR, MVT::v32i32, Custom);
3000b57cec5SDimitry Andric 
3010b57cec5SDimitry Andric   setOperationAction(ISD::FP16_TO_FP, MVT::f64, Expand);
3020b57cec5SDimitry Andric   setOperationAction(ISD::FP_TO_FP16, MVT::f64, Custom);
3030b57cec5SDimitry Andric   setOperationAction(ISD::FP_TO_FP16, MVT::f32, Custom);
3040b57cec5SDimitry Andric 
3050b57cec5SDimitry Andric   const MVT ScalarIntVTs[] = { MVT::i32, MVT::i64 };
3060b57cec5SDimitry Andric   for (MVT VT : ScalarIntVTs) {
3070b57cec5SDimitry Andric     // These should use [SU]DIVREM, so set them to expand
3080b57cec5SDimitry Andric     setOperationAction(ISD::SDIV, VT, Expand);
3090b57cec5SDimitry Andric     setOperationAction(ISD::UDIV, VT, Expand);
3100b57cec5SDimitry Andric     setOperationAction(ISD::SREM, VT, Expand);
3110b57cec5SDimitry Andric     setOperationAction(ISD::UREM, VT, Expand);
3120b57cec5SDimitry Andric 
3130b57cec5SDimitry Andric     // GPU does not have divrem function for signed or unsigned.
3140b57cec5SDimitry Andric     setOperationAction(ISD::SDIVREM, VT, Custom);
3150b57cec5SDimitry Andric     setOperationAction(ISD::UDIVREM, VT, Custom);
3160b57cec5SDimitry Andric 
3170b57cec5SDimitry Andric     // GPU does not have [S|U]MUL_LOHI functions as a single instruction.
3180b57cec5SDimitry Andric     setOperationAction(ISD::SMUL_LOHI, VT, Expand);
3190b57cec5SDimitry Andric     setOperationAction(ISD::UMUL_LOHI, VT, Expand);
3200b57cec5SDimitry Andric 
3210b57cec5SDimitry Andric     setOperationAction(ISD::BSWAP, VT, Expand);
3220b57cec5SDimitry Andric     setOperationAction(ISD::CTTZ, VT, Expand);
3230b57cec5SDimitry Andric     setOperationAction(ISD::CTLZ, VT, Expand);
3240b57cec5SDimitry Andric 
3250b57cec5SDimitry Andric     // AMDGPU uses ADDC/SUBC/ADDE/SUBE
3260b57cec5SDimitry Andric     setOperationAction(ISD::ADDC, VT, Legal);
3270b57cec5SDimitry Andric     setOperationAction(ISD::SUBC, VT, Legal);
3280b57cec5SDimitry Andric     setOperationAction(ISD::ADDE, VT, Legal);
3290b57cec5SDimitry Andric     setOperationAction(ISD::SUBE, VT, Legal);
3300b57cec5SDimitry Andric   }
3310b57cec5SDimitry Andric 
3320b57cec5SDimitry Andric   // The hardware supports 32-bit ROTR, but not ROTL.
3330b57cec5SDimitry Andric   setOperationAction(ISD::ROTL, MVT::i32, Expand);
3340b57cec5SDimitry Andric   setOperationAction(ISD::ROTL, MVT::i64, Expand);
3350b57cec5SDimitry Andric   setOperationAction(ISD::ROTR, MVT::i64, Expand);
3360b57cec5SDimitry Andric 
3370b57cec5SDimitry Andric   setOperationAction(ISD::MUL, MVT::i64, Expand);
3380b57cec5SDimitry Andric   setOperationAction(ISD::MULHU, MVT::i64, Expand);
3390b57cec5SDimitry Andric   setOperationAction(ISD::MULHS, MVT::i64, Expand);
3400b57cec5SDimitry Andric   setOperationAction(ISD::UINT_TO_FP, MVT::i64, Custom);
3410b57cec5SDimitry Andric   setOperationAction(ISD::SINT_TO_FP, MVT::i64, Custom);
3420b57cec5SDimitry Andric   setOperationAction(ISD::FP_TO_SINT, MVT::i64, Custom);
3430b57cec5SDimitry Andric   setOperationAction(ISD::FP_TO_UINT, MVT::i64, Custom);
3440b57cec5SDimitry Andric   setOperationAction(ISD::SELECT_CC, MVT::i64, Expand);
3450b57cec5SDimitry Andric 
3460b57cec5SDimitry Andric   setOperationAction(ISD::SMIN, MVT::i32, Legal);
3470b57cec5SDimitry Andric   setOperationAction(ISD::UMIN, MVT::i32, Legal);
3480b57cec5SDimitry Andric   setOperationAction(ISD::SMAX, MVT::i32, Legal);
3490b57cec5SDimitry Andric   setOperationAction(ISD::UMAX, MVT::i32, Legal);
3500b57cec5SDimitry Andric 
3510b57cec5SDimitry Andric   setOperationAction(ISD::CTTZ, MVT::i64, Custom);
3520b57cec5SDimitry Andric   setOperationAction(ISD::CTTZ_ZERO_UNDEF, MVT::i64, Custom);
3530b57cec5SDimitry Andric   setOperationAction(ISD::CTLZ, MVT::i64, Custom);
3540b57cec5SDimitry Andric   setOperationAction(ISD::CTLZ_ZERO_UNDEF, MVT::i64, Custom);
3550b57cec5SDimitry Andric 
3560b57cec5SDimitry Andric   static const MVT::SimpleValueType VectorIntTypes[] = {
3570b57cec5SDimitry Andric     MVT::v2i32, MVT::v3i32, MVT::v4i32, MVT::v5i32
3580b57cec5SDimitry Andric   };
3590b57cec5SDimitry Andric 
3600b57cec5SDimitry Andric   for (MVT VT : VectorIntTypes) {
3610b57cec5SDimitry Andric     // Expand the following operations for the current type by default.
3620b57cec5SDimitry Andric     setOperationAction(ISD::ADD,  VT, Expand);
3630b57cec5SDimitry Andric     setOperationAction(ISD::AND,  VT, Expand);
3640b57cec5SDimitry Andric     setOperationAction(ISD::FP_TO_SINT, VT, Expand);
3650b57cec5SDimitry Andric     setOperationAction(ISD::FP_TO_UINT, VT, Expand);
3660b57cec5SDimitry Andric     setOperationAction(ISD::MUL,  VT, Expand);
3670b57cec5SDimitry Andric     setOperationAction(ISD::MULHU, VT, Expand);
3680b57cec5SDimitry Andric     setOperationAction(ISD::MULHS, VT, Expand);
3690b57cec5SDimitry Andric     setOperationAction(ISD::OR,   VT, Expand);
3700b57cec5SDimitry Andric     setOperationAction(ISD::SHL,  VT, Expand);
3710b57cec5SDimitry Andric     setOperationAction(ISD::SRA,  VT, Expand);
3720b57cec5SDimitry Andric     setOperationAction(ISD::SRL,  VT, Expand);
3730b57cec5SDimitry Andric     setOperationAction(ISD::ROTL, VT, Expand);
3740b57cec5SDimitry Andric     setOperationAction(ISD::ROTR, VT, Expand);
3750b57cec5SDimitry Andric     setOperationAction(ISD::SUB,  VT, Expand);
3760b57cec5SDimitry Andric     setOperationAction(ISD::SINT_TO_FP, VT, Expand);
3770b57cec5SDimitry Andric     setOperationAction(ISD::UINT_TO_FP, VT, Expand);
3780b57cec5SDimitry Andric     setOperationAction(ISD::SDIV, VT, Expand);
3790b57cec5SDimitry Andric     setOperationAction(ISD::UDIV, VT, Expand);
3800b57cec5SDimitry Andric     setOperationAction(ISD::SREM, VT, Expand);
3810b57cec5SDimitry Andric     setOperationAction(ISD::UREM, VT, Expand);
3820b57cec5SDimitry Andric     setOperationAction(ISD::SMUL_LOHI, VT, Expand);
3830b57cec5SDimitry Andric     setOperationAction(ISD::UMUL_LOHI, VT, Expand);
3840b57cec5SDimitry Andric     setOperationAction(ISD::SDIVREM, VT, Custom);
3850b57cec5SDimitry Andric     setOperationAction(ISD::UDIVREM, VT, Expand);
3860b57cec5SDimitry Andric     setOperationAction(ISD::SELECT, VT, Expand);
3870b57cec5SDimitry Andric     setOperationAction(ISD::VSELECT, VT, Expand);
3880b57cec5SDimitry Andric     setOperationAction(ISD::SELECT_CC, VT, Expand);
3890b57cec5SDimitry Andric     setOperationAction(ISD::XOR,  VT, Expand);
3900b57cec5SDimitry Andric     setOperationAction(ISD::BSWAP, VT, Expand);
3910b57cec5SDimitry Andric     setOperationAction(ISD::CTPOP, VT, Expand);
3920b57cec5SDimitry Andric     setOperationAction(ISD::CTTZ, VT, Expand);
3930b57cec5SDimitry Andric     setOperationAction(ISD::CTLZ, VT, Expand);
3940b57cec5SDimitry Andric     setOperationAction(ISD::VECTOR_SHUFFLE, VT, Expand);
3950b57cec5SDimitry Andric     setOperationAction(ISD::SETCC, VT, Expand);
3960b57cec5SDimitry Andric   }
3970b57cec5SDimitry Andric 
3980b57cec5SDimitry Andric   static const MVT::SimpleValueType FloatVectorTypes[] = {
3990b57cec5SDimitry Andric      MVT::v2f32, MVT::v3f32, MVT::v4f32, MVT::v5f32
4000b57cec5SDimitry Andric   };
4010b57cec5SDimitry Andric 
4020b57cec5SDimitry Andric   for (MVT VT : FloatVectorTypes) {
4030b57cec5SDimitry Andric     setOperationAction(ISD::FABS, VT, Expand);
4040b57cec5SDimitry Andric     setOperationAction(ISD::FMINNUM, VT, Expand);
4050b57cec5SDimitry Andric     setOperationAction(ISD::FMAXNUM, VT, Expand);
4060b57cec5SDimitry Andric     setOperationAction(ISD::FADD, VT, Expand);
4070b57cec5SDimitry Andric     setOperationAction(ISD::FCEIL, VT, Expand);
4080b57cec5SDimitry Andric     setOperationAction(ISD::FCOS, VT, Expand);
4090b57cec5SDimitry Andric     setOperationAction(ISD::FDIV, VT, Expand);
4100b57cec5SDimitry Andric     setOperationAction(ISD::FEXP2, VT, Expand);
4110b57cec5SDimitry Andric     setOperationAction(ISD::FEXP, VT, Expand);
4120b57cec5SDimitry Andric     setOperationAction(ISD::FLOG2, VT, Expand);
4130b57cec5SDimitry Andric     setOperationAction(ISD::FREM, VT, Expand);
4140b57cec5SDimitry Andric     setOperationAction(ISD::FLOG, VT, Expand);
4150b57cec5SDimitry Andric     setOperationAction(ISD::FLOG10, VT, Expand);
4160b57cec5SDimitry Andric     setOperationAction(ISD::FPOW, VT, Expand);
4170b57cec5SDimitry Andric     setOperationAction(ISD::FFLOOR, VT, Expand);
4180b57cec5SDimitry Andric     setOperationAction(ISD::FTRUNC, VT, Expand);
4190b57cec5SDimitry Andric     setOperationAction(ISD::FMUL, VT, Expand);
4200b57cec5SDimitry Andric     setOperationAction(ISD::FMA, VT, Expand);
4210b57cec5SDimitry Andric     setOperationAction(ISD::FRINT, VT, Expand);
4220b57cec5SDimitry Andric     setOperationAction(ISD::FNEARBYINT, VT, Expand);
4230b57cec5SDimitry Andric     setOperationAction(ISD::FSQRT, VT, Expand);
4240b57cec5SDimitry Andric     setOperationAction(ISD::FSIN, VT, Expand);
4250b57cec5SDimitry Andric     setOperationAction(ISD::FSUB, VT, Expand);
4260b57cec5SDimitry Andric     setOperationAction(ISD::FNEG, VT, Expand);
4270b57cec5SDimitry Andric     setOperationAction(ISD::VSELECT, VT, Expand);
4280b57cec5SDimitry Andric     setOperationAction(ISD::SELECT_CC, VT, Expand);
4290b57cec5SDimitry Andric     setOperationAction(ISD::FCOPYSIGN, VT, Expand);
4300b57cec5SDimitry Andric     setOperationAction(ISD::VECTOR_SHUFFLE, VT, Expand);
4310b57cec5SDimitry Andric     setOperationAction(ISD::SETCC, VT, Expand);
4320b57cec5SDimitry Andric     setOperationAction(ISD::FCANONICALIZE, VT, Expand);
4330b57cec5SDimitry Andric   }
4340b57cec5SDimitry Andric 
4350b57cec5SDimitry Andric   // This causes using an unrolled select operation rather than expansion with
4360b57cec5SDimitry Andric   // bit operations. This is in general better, but the alternative using BFI
4370b57cec5SDimitry Andric   // instructions may be better if the select sources are SGPRs.
4380b57cec5SDimitry Andric   setOperationAction(ISD::SELECT, MVT::v2f32, Promote);
4390b57cec5SDimitry Andric   AddPromotedToType(ISD::SELECT, MVT::v2f32, MVT::v2i32);
4400b57cec5SDimitry Andric 
4410b57cec5SDimitry Andric   setOperationAction(ISD::SELECT, MVT::v3f32, Promote);
4420b57cec5SDimitry Andric   AddPromotedToType(ISD::SELECT, MVT::v3f32, MVT::v3i32);
4430b57cec5SDimitry Andric 
4440b57cec5SDimitry Andric   setOperationAction(ISD::SELECT, MVT::v4f32, Promote);
4450b57cec5SDimitry Andric   AddPromotedToType(ISD::SELECT, MVT::v4f32, MVT::v4i32);
4460b57cec5SDimitry Andric 
4470b57cec5SDimitry Andric   setOperationAction(ISD::SELECT, MVT::v5f32, Promote);
4480b57cec5SDimitry Andric   AddPromotedToType(ISD::SELECT, MVT::v5f32, MVT::v5i32);
4490b57cec5SDimitry Andric 
4500b57cec5SDimitry Andric   // There are no libcalls of any kind.
4510b57cec5SDimitry Andric   for (int I = 0; I < RTLIB::UNKNOWN_LIBCALL; ++I)
4520b57cec5SDimitry Andric     setLibcallName(static_cast<RTLIB::Libcall>(I), nullptr);
4530b57cec5SDimitry Andric 
4540b57cec5SDimitry Andric   setBooleanContents(ZeroOrNegativeOneBooleanContent);
4550b57cec5SDimitry Andric   setBooleanVectorContents(ZeroOrNegativeOneBooleanContent);
4560b57cec5SDimitry Andric 
4570b57cec5SDimitry Andric   setSchedulingPreference(Sched::RegPressure);
4580b57cec5SDimitry Andric   setJumpIsExpensive(true);
4590b57cec5SDimitry Andric 
4600b57cec5SDimitry Andric   // FIXME: This is only partially true. If we have to do vector compares, any
4610b57cec5SDimitry Andric   // SGPR pair can be a condition register. If we have a uniform condition, we
4620b57cec5SDimitry Andric   // are better off doing SALU operations, where there is only one SCC. For now,
4630b57cec5SDimitry Andric   // we don't have a way of knowing during instruction selection if a condition
4640b57cec5SDimitry Andric   // will be uniform and we always use vector compares. Assume we are using
4650b57cec5SDimitry Andric   // vector compares until that is fixed.
4660b57cec5SDimitry Andric   setHasMultipleConditionRegisters(true);
4670b57cec5SDimitry Andric 
4680b57cec5SDimitry Andric   setMinCmpXchgSizeInBits(32);
4690b57cec5SDimitry Andric   setSupportsUnalignedAtomics(false);
4700b57cec5SDimitry Andric 
4710b57cec5SDimitry Andric   PredictableSelectIsExpensive = false;
4720b57cec5SDimitry Andric 
4730b57cec5SDimitry Andric   // We want to find all load dependencies for long chains of stores to enable
4740b57cec5SDimitry Andric   // merging into very wide vectors. The problem is with vectors with > 4
4750b57cec5SDimitry Andric   // elements. MergeConsecutiveStores will attempt to merge these because x8/x16
4760b57cec5SDimitry Andric   // vectors are a legal type, even though we have to split the loads
4770b57cec5SDimitry Andric   // usually. When we can more precisely specify load legality per address
4780b57cec5SDimitry Andric   // space, we should be able to make FindBetterChain/MergeConsecutiveStores
4790b57cec5SDimitry Andric   // smarter so that they can figure out what to do in 2 iterations without all
4800b57cec5SDimitry Andric   // N > 4 stores on the same chain.
4810b57cec5SDimitry Andric   GatherAllAliasesMaxDepth = 16;
4820b57cec5SDimitry Andric 
4830b57cec5SDimitry Andric   // memcpy/memmove/memset are expanded in the IR, so we shouldn't need to worry
4840b57cec5SDimitry Andric   // about these during lowering.
4850b57cec5SDimitry Andric   MaxStoresPerMemcpy  = 0xffffffff;
4860b57cec5SDimitry Andric   MaxStoresPerMemmove = 0xffffffff;
4870b57cec5SDimitry Andric   MaxStoresPerMemset  = 0xffffffff;
4880b57cec5SDimitry Andric 
4890b57cec5SDimitry Andric   setTargetDAGCombine(ISD::BITCAST);
4900b57cec5SDimitry Andric   setTargetDAGCombine(ISD::SHL);
4910b57cec5SDimitry Andric   setTargetDAGCombine(ISD::SRA);
4920b57cec5SDimitry Andric   setTargetDAGCombine(ISD::SRL);
4930b57cec5SDimitry Andric   setTargetDAGCombine(ISD::TRUNCATE);
4940b57cec5SDimitry Andric   setTargetDAGCombine(ISD::MUL);
4950b57cec5SDimitry Andric   setTargetDAGCombine(ISD::MULHU);
4960b57cec5SDimitry Andric   setTargetDAGCombine(ISD::MULHS);
4970b57cec5SDimitry Andric   setTargetDAGCombine(ISD::SELECT);
4980b57cec5SDimitry Andric   setTargetDAGCombine(ISD::SELECT_CC);
4990b57cec5SDimitry Andric   setTargetDAGCombine(ISD::STORE);
5000b57cec5SDimitry Andric   setTargetDAGCombine(ISD::FADD);
5010b57cec5SDimitry Andric   setTargetDAGCombine(ISD::FSUB);
5020b57cec5SDimitry Andric   setTargetDAGCombine(ISD::FNEG);
5030b57cec5SDimitry Andric   setTargetDAGCombine(ISD::FABS);
5040b57cec5SDimitry Andric   setTargetDAGCombine(ISD::AssertZext);
5050b57cec5SDimitry Andric   setTargetDAGCombine(ISD::AssertSext);
506*8bcb0991SDimitry Andric   setTargetDAGCombine(ISD::INTRINSIC_WO_CHAIN);
5070b57cec5SDimitry Andric }
5080b57cec5SDimitry Andric 
5090b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
5100b57cec5SDimitry Andric // Target Information
5110b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
5120b57cec5SDimitry Andric 
5130b57cec5SDimitry Andric LLVM_READNONE
5140b57cec5SDimitry Andric static bool fnegFoldsIntoOp(unsigned Opc) {
5150b57cec5SDimitry Andric   switch (Opc) {
5160b57cec5SDimitry Andric   case ISD::FADD:
5170b57cec5SDimitry Andric   case ISD::FSUB:
5180b57cec5SDimitry Andric   case ISD::FMUL:
5190b57cec5SDimitry Andric   case ISD::FMA:
5200b57cec5SDimitry Andric   case ISD::FMAD:
5210b57cec5SDimitry Andric   case ISD::FMINNUM:
5220b57cec5SDimitry Andric   case ISD::FMAXNUM:
5230b57cec5SDimitry Andric   case ISD::FMINNUM_IEEE:
5240b57cec5SDimitry Andric   case ISD::FMAXNUM_IEEE:
5250b57cec5SDimitry Andric   case ISD::FSIN:
5260b57cec5SDimitry Andric   case ISD::FTRUNC:
5270b57cec5SDimitry Andric   case ISD::FRINT:
5280b57cec5SDimitry Andric   case ISD::FNEARBYINT:
5290b57cec5SDimitry Andric   case ISD::FCANONICALIZE:
5300b57cec5SDimitry Andric   case AMDGPUISD::RCP:
5310b57cec5SDimitry Andric   case AMDGPUISD::RCP_LEGACY:
5320b57cec5SDimitry Andric   case AMDGPUISD::RCP_IFLAG:
5330b57cec5SDimitry Andric   case AMDGPUISD::SIN_HW:
5340b57cec5SDimitry Andric   case AMDGPUISD::FMUL_LEGACY:
5350b57cec5SDimitry Andric   case AMDGPUISD::FMIN_LEGACY:
5360b57cec5SDimitry Andric   case AMDGPUISD::FMAX_LEGACY:
5370b57cec5SDimitry Andric   case AMDGPUISD::FMED3:
5380b57cec5SDimitry Andric     return true;
5390b57cec5SDimitry Andric   default:
5400b57cec5SDimitry Andric     return false;
5410b57cec5SDimitry Andric   }
5420b57cec5SDimitry Andric }
5430b57cec5SDimitry Andric 
5440b57cec5SDimitry Andric /// \p returns true if the operation will definitely need to use a 64-bit
5450b57cec5SDimitry Andric /// encoding, and thus will use a VOP3 encoding regardless of the source
5460b57cec5SDimitry Andric /// modifiers.
5470b57cec5SDimitry Andric LLVM_READONLY
5480b57cec5SDimitry Andric static bool opMustUseVOP3Encoding(const SDNode *N, MVT VT) {
5490b57cec5SDimitry Andric   return N->getNumOperands() > 2 || VT == MVT::f64;
5500b57cec5SDimitry Andric }
5510b57cec5SDimitry Andric 
5520b57cec5SDimitry Andric // Most FP instructions support source modifiers, but this could be refined
5530b57cec5SDimitry Andric // slightly.
5540b57cec5SDimitry Andric LLVM_READONLY
5550b57cec5SDimitry Andric static bool hasSourceMods(const SDNode *N) {
5560b57cec5SDimitry Andric   if (isa<MemSDNode>(N))
5570b57cec5SDimitry Andric     return false;
5580b57cec5SDimitry Andric 
5590b57cec5SDimitry Andric   switch (N->getOpcode()) {
5600b57cec5SDimitry Andric   case ISD::CopyToReg:
5610b57cec5SDimitry Andric   case ISD::SELECT:
5620b57cec5SDimitry Andric   case ISD::FDIV:
5630b57cec5SDimitry Andric   case ISD::FREM:
5640b57cec5SDimitry Andric   case ISD::INLINEASM:
5650b57cec5SDimitry Andric   case ISD::INLINEASM_BR:
5660b57cec5SDimitry Andric   case AMDGPUISD::DIV_SCALE:
567*8bcb0991SDimitry Andric   case ISD::INTRINSIC_W_CHAIN:
5680b57cec5SDimitry Andric 
5690b57cec5SDimitry Andric   // TODO: Should really be looking at the users of the bitcast. These are
5700b57cec5SDimitry Andric   // problematic because bitcasts are used to legalize all stores to integer
5710b57cec5SDimitry Andric   // types.
5720b57cec5SDimitry Andric   case ISD::BITCAST:
5730b57cec5SDimitry Andric     return false;
574*8bcb0991SDimitry Andric   case ISD::INTRINSIC_WO_CHAIN: {
575*8bcb0991SDimitry Andric     switch (cast<ConstantSDNode>(N->getOperand(0))->getZExtValue()) {
576*8bcb0991SDimitry Andric     case Intrinsic::amdgcn_interp_p1:
577*8bcb0991SDimitry Andric     case Intrinsic::amdgcn_interp_p2:
578*8bcb0991SDimitry Andric     case Intrinsic::amdgcn_interp_mov:
579*8bcb0991SDimitry Andric     case Intrinsic::amdgcn_interp_p1_f16:
580*8bcb0991SDimitry Andric     case Intrinsic::amdgcn_interp_p2_f16:
581*8bcb0991SDimitry Andric       return false;
582*8bcb0991SDimitry Andric     default:
583*8bcb0991SDimitry Andric       return true;
584*8bcb0991SDimitry Andric     }
585*8bcb0991SDimitry Andric   }
5860b57cec5SDimitry Andric   default:
5870b57cec5SDimitry Andric     return true;
5880b57cec5SDimitry Andric   }
5890b57cec5SDimitry Andric }
5900b57cec5SDimitry Andric 
5910b57cec5SDimitry Andric bool AMDGPUTargetLowering::allUsesHaveSourceMods(const SDNode *N,
5920b57cec5SDimitry Andric                                                  unsigned CostThreshold) {
5930b57cec5SDimitry Andric   // Some users (such as 3-operand FMA/MAD) must use a VOP3 encoding, and thus
5940b57cec5SDimitry Andric   // it is truly free to use a source modifier in all cases. If there are
5950b57cec5SDimitry Andric   // multiple users but for each one will necessitate using VOP3, there will be
5960b57cec5SDimitry Andric   // a code size increase. Try to avoid increasing code size unless we know it
5970b57cec5SDimitry Andric   // will save on the instruction count.
5980b57cec5SDimitry Andric   unsigned NumMayIncreaseSize = 0;
5990b57cec5SDimitry Andric   MVT VT = N->getValueType(0).getScalarType().getSimpleVT();
6000b57cec5SDimitry Andric 
6010b57cec5SDimitry Andric   // XXX - Should this limit number of uses to check?
6020b57cec5SDimitry Andric   for (const SDNode *U : N->uses()) {
6030b57cec5SDimitry Andric     if (!hasSourceMods(U))
6040b57cec5SDimitry Andric       return false;
6050b57cec5SDimitry Andric 
6060b57cec5SDimitry Andric     if (!opMustUseVOP3Encoding(U, VT)) {
6070b57cec5SDimitry Andric       if (++NumMayIncreaseSize > CostThreshold)
6080b57cec5SDimitry Andric         return false;
6090b57cec5SDimitry Andric     }
6100b57cec5SDimitry Andric   }
6110b57cec5SDimitry Andric 
6120b57cec5SDimitry Andric   return true;
6130b57cec5SDimitry Andric }
6140b57cec5SDimitry Andric 
6150b57cec5SDimitry Andric MVT AMDGPUTargetLowering::getVectorIdxTy(const DataLayout &) const {
6160b57cec5SDimitry Andric   return MVT::i32;
6170b57cec5SDimitry Andric }
6180b57cec5SDimitry Andric 
6190b57cec5SDimitry Andric bool AMDGPUTargetLowering::isSelectSupported(SelectSupportKind SelType) const {
6200b57cec5SDimitry Andric   return true;
6210b57cec5SDimitry Andric }
6220b57cec5SDimitry Andric 
6230b57cec5SDimitry Andric // The backend supports 32 and 64 bit floating point immediates.
6240b57cec5SDimitry Andric // FIXME: Why are we reporting vectors of FP immediates as legal?
6250b57cec5SDimitry Andric bool AMDGPUTargetLowering::isFPImmLegal(const APFloat &Imm, EVT VT,
6260b57cec5SDimitry Andric                                         bool ForCodeSize) const {
6270b57cec5SDimitry Andric   EVT ScalarVT = VT.getScalarType();
6280b57cec5SDimitry Andric   return (ScalarVT == MVT::f32 || ScalarVT == MVT::f64 ||
6290b57cec5SDimitry Andric          (ScalarVT == MVT::f16 && Subtarget->has16BitInsts()));
6300b57cec5SDimitry Andric }
6310b57cec5SDimitry Andric 
6320b57cec5SDimitry Andric // We don't want to shrink f64 / f32 constants.
6330b57cec5SDimitry Andric bool AMDGPUTargetLowering::ShouldShrinkFPConstant(EVT VT) const {
6340b57cec5SDimitry Andric   EVT ScalarVT = VT.getScalarType();
6350b57cec5SDimitry Andric   return (ScalarVT != MVT::f32 && ScalarVT != MVT::f64);
6360b57cec5SDimitry Andric }
6370b57cec5SDimitry Andric 
6380b57cec5SDimitry Andric bool AMDGPUTargetLowering::shouldReduceLoadWidth(SDNode *N,
6390b57cec5SDimitry Andric                                                  ISD::LoadExtType ExtTy,
6400b57cec5SDimitry Andric                                                  EVT NewVT) const {
6410b57cec5SDimitry Andric   // TODO: This may be worth removing. Check regression tests for diffs.
6420b57cec5SDimitry Andric   if (!TargetLoweringBase::shouldReduceLoadWidth(N, ExtTy, NewVT))
6430b57cec5SDimitry Andric     return false;
6440b57cec5SDimitry Andric 
6450b57cec5SDimitry Andric   unsigned NewSize = NewVT.getStoreSizeInBits();
6460b57cec5SDimitry Andric 
6470b57cec5SDimitry Andric   // If we are reducing to a 32-bit load, this is always better.
6480b57cec5SDimitry Andric   if (NewSize == 32)
6490b57cec5SDimitry Andric     return true;
6500b57cec5SDimitry Andric 
6510b57cec5SDimitry Andric   EVT OldVT = N->getValueType(0);
6520b57cec5SDimitry Andric   unsigned OldSize = OldVT.getStoreSizeInBits();
6530b57cec5SDimitry Andric 
6540b57cec5SDimitry Andric   MemSDNode *MN = cast<MemSDNode>(N);
6550b57cec5SDimitry Andric   unsigned AS = MN->getAddressSpace();
6560b57cec5SDimitry Andric   // Do not shrink an aligned scalar load to sub-dword.
6570b57cec5SDimitry Andric   // Scalar engine cannot do sub-dword loads.
6580b57cec5SDimitry Andric   if (OldSize >= 32 && NewSize < 32 && MN->getAlignment() >= 4 &&
6590b57cec5SDimitry Andric       (AS == AMDGPUAS::CONSTANT_ADDRESS ||
6600b57cec5SDimitry Andric        AS == AMDGPUAS::CONSTANT_ADDRESS_32BIT ||
6610b57cec5SDimitry Andric        (isa<LoadSDNode>(N) &&
6620b57cec5SDimitry Andric         AS == AMDGPUAS::GLOBAL_ADDRESS && MN->isInvariant())) &&
6630b57cec5SDimitry Andric       AMDGPUInstrInfo::isUniformMMO(MN->getMemOperand()))
6640b57cec5SDimitry Andric     return false;
6650b57cec5SDimitry Andric 
6660b57cec5SDimitry Andric   // Don't produce extloads from sub 32-bit types. SI doesn't have scalar
6670b57cec5SDimitry Andric   // extloads, so doing one requires using a buffer_load. In cases where we
6680b57cec5SDimitry Andric   // still couldn't use a scalar load, using the wider load shouldn't really
6690b57cec5SDimitry Andric   // hurt anything.
6700b57cec5SDimitry Andric 
6710b57cec5SDimitry Andric   // If the old size already had to be an extload, there's no harm in continuing
6720b57cec5SDimitry Andric   // to reduce the width.
6730b57cec5SDimitry Andric   return (OldSize < 32);
6740b57cec5SDimitry Andric }
6750b57cec5SDimitry Andric 
6760b57cec5SDimitry Andric bool AMDGPUTargetLowering::isLoadBitCastBeneficial(EVT LoadTy, EVT CastTy,
6770b57cec5SDimitry Andric                                                    const SelectionDAG &DAG,
6780b57cec5SDimitry Andric                                                    const MachineMemOperand &MMO) const {
6790b57cec5SDimitry Andric 
6800b57cec5SDimitry Andric   assert(LoadTy.getSizeInBits() == CastTy.getSizeInBits());
6810b57cec5SDimitry Andric 
6820b57cec5SDimitry Andric   if (LoadTy.getScalarType() == MVT::i32)
6830b57cec5SDimitry Andric     return false;
6840b57cec5SDimitry Andric 
6850b57cec5SDimitry Andric   unsigned LScalarSize = LoadTy.getScalarSizeInBits();
6860b57cec5SDimitry Andric   unsigned CastScalarSize = CastTy.getScalarSizeInBits();
6870b57cec5SDimitry Andric 
6880b57cec5SDimitry Andric   if ((LScalarSize >= CastScalarSize) && (CastScalarSize < 32))
6890b57cec5SDimitry Andric     return false;
6900b57cec5SDimitry Andric 
6910b57cec5SDimitry Andric   bool Fast = false;
692*8bcb0991SDimitry Andric   return allowsMemoryAccessForAlignment(*DAG.getContext(), DAG.getDataLayout(),
693*8bcb0991SDimitry Andric                                         CastTy, MMO, &Fast) &&
694*8bcb0991SDimitry Andric          Fast;
6950b57cec5SDimitry Andric }
6960b57cec5SDimitry Andric 
6970b57cec5SDimitry Andric // SI+ has instructions for cttz / ctlz for 32-bit values. This is probably also
6980b57cec5SDimitry Andric // profitable with the expansion for 64-bit since it's generally good to
6990b57cec5SDimitry Andric // speculate things.
7000b57cec5SDimitry Andric // FIXME: These should really have the size as a parameter.
7010b57cec5SDimitry Andric bool AMDGPUTargetLowering::isCheapToSpeculateCttz() const {
7020b57cec5SDimitry Andric   return true;
7030b57cec5SDimitry Andric }
7040b57cec5SDimitry Andric 
7050b57cec5SDimitry Andric bool AMDGPUTargetLowering::isCheapToSpeculateCtlz() const {
7060b57cec5SDimitry Andric   return true;
7070b57cec5SDimitry Andric }
7080b57cec5SDimitry Andric 
7090b57cec5SDimitry Andric bool AMDGPUTargetLowering::isSDNodeAlwaysUniform(const SDNode * N) const {
7100b57cec5SDimitry Andric   switch (N->getOpcode()) {
7110b57cec5SDimitry Andric     default:
7120b57cec5SDimitry Andric     return false;
7130b57cec5SDimitry Andric     case ISD::EntryToken:
7140b57cec5SDimitry Andric     case ISD::TokenFactor:
7150b57cec5SDimitry Andric       return true;
7160b57cec5SDimitry Andric     case ISD::INTRINSIC_WO_CHAIN:
7170b57cec5SDimitry Andric     {
7180b57cec5SDimitry Andric       unsigned IntrID = cast<ConstantSDNode>(N->getOperand(0))->getZExtValue();
7190b57cec5SDimitry Andric       switch (IntrID) {
7200b57cec5SDimitry Andric         default:
7210b57cec5SDimitry Andric         return false;
7220b57cec5SDimitry Andric         case Intrinsic::amdgcn_readfirstlane:
7230b57cec5SDimitry Andric         case Intrinsic::amdgcn_readlane:
7240b57cec5SDimitry Andric           return true;
7250b57cec5SDimitry Andric       }
7260b57cec5SDimitry Andric     }
7270b57cec5SDimitry Andric     break;
7280b57cec5SDimitry Andric     case ISD::LOAD:
7290b57cec5SDimitry Andric     {
730*8bcb0991SDimitry Andric       if (cast<LoadSDNode>(N)->getMemOperand()->getAddrSpace() ==
731*8bcb0991SDimitry Andric           AMDGPUAS::CONSTANT_ADDRESS_32BIT)
7320b57cec5SDimitry Andric         return true;
7330b57cec5SDimitry Andric       return false;
7340b57cec5SDimitry Andric     }
7350b57cec5SDimitry Andric     break;
7360b57cec5SDimitry Andric   }
7370b57cec5SDimitry Andric }
7380b57cec5SDimitry Andric 
7390b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
7400b57cec5SDimitry Andric // Target Properties
7410b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
7420b57cec5SDimitry Andric 
7430b57cec5SDimitry Andric bool AMDGPUTargetLowering::isFAbsFree(EVT VT) const {
7440b57cec5SDimitry Andric   assert(VT.isFloatingPoint());
7450b57cec5SDimitry Andric 
7460b57cec5SDimitry Andric   // Packed operations do not have a fabs modifier.
7470b57cec5SDimitry Andric   return VT == MVT::f32 || VT == MVT::f64 ||
7480b57cec5SDimitry Andric          (Subtarget->has16BitInsts() && VT == MVT::f16);
7490b57cec5SDimitry Andric }
7500b57cec5SDimitry Andric 
7510b57cec5SDimitry Andric bool AMDGPUTargetLowering::isFNegFree(EVT VT) const {
7520b57cec5SDimitry Andric   assert(VT.isFloatingPoint());
7530b57cec5SDimitry Andric   return VT == MVT::f32 || VT == MVT::f64 ||
7540b57cec5SDimitry Andric          (Subtarget->has16BitInsts() && VT == MVT::f16) ||
7550b57cec5SDimitry Andric          (Subtarget->hasVOP3PInsts() && VT == MVT::v2f16);
7560b57cec5SDimitry Andric }
7570b57cec5SDimitry Andric 
7580b57cec5SDimitry Andric bool AMDGPUTargetLowering:: storeOfVectorConstantIsCheap(EVT MemVT,
7590b57cec5SDimitry Andric                                                          unsigned NumElem,
7600b57cec5SDimitry Andric                                                          unsigned AS) const {
7610b57cec5SDimitry Andric   return true;
7620b57cec5SDimitry Andric }
7630b57cec5SDimitry Andric 
7640b57cec5SDimitry Andric bool AMDGPUTargetLowering::aggressivelyPreferBuildVectorSources(EVT VecVT) const {
7650b57cec5SDimitry Andric   // There are few operations which truly have vector input operands. Any vector
7660b57cec5SDimitry Andric   // operation is going to involve operations on each component, and a
7670b57cec5SDimitry Andric   // build_vector will be a copy per element, so it always makes sense to use a
7680b57cec5SDimitry Andric   // build_vector input in place of the extracted element to avoid a copy into a
7690b57cec5SDimitry Andric   // super register.
7700b57cec5SDimitry Andric   //
7710b57cec5SDimitry Andric   // We should probably only do this if all users are extracts only, but this
7720b57cec5SDimitry Andric   // should be the common case.
7730b57cec5SDimitry Andric   return true;
7740b57cec5SDimitry Andric }
7750b57cec5SDimitry Andric 
7760b57cec5SDimitry Andric bool AMDGPUTargetLowering::isTruncateFree(EVT Source, EVT Dest) const {
7770b57cec5SDimitry Andric   // Truncate is just accessing a subregister.
7780b57cec5SDimitry Andric 
7790b57cec5SDimitry Andric   unsigned SrcSize = Source.getSizeInBits();
7800b57cec5SDimitry Andric   unsigned DestSize = Dest.getSizeInBits();
7810b57cec5SDimitry Andric 
7820b57cec5SDimitry Andric   return DestSize < SrcSize && DestSize % 32 == 0 ;
7830b57cec5SDimitry Andric }
7840b57cec5SDimitry Andric 
7850b57cec5SDimitry Andric bool AMDGPUTargetLowering::isTruncateFree(Type *Source, Type *Dest) const {
7860b57cec5SDimitry Andric   // Truncate is just accessing a subregister.
7870b57cec5SDimitry Andric 
7880b57cec5SDimitry Andric   unsigned SrcSize = Source->getScalarSizeInBits();
7890b57cec5SDimitry Andric   unsigned DestSize = Dest->getScalarSizeInBits();
7900b57cec5SDimitry Andric 
7910b57cec5SDimitry Andric   if (DestSize== 16 && Subtarget->has16BitInsts())
7920b57cec5SDimitry Andric     return SrcSize >= 32;
7930b57cec5SDimitry Andric 
7940b57cec5SDimitry Andric   return DestSize < SrcSize && DestSize % 32 == 0;
7950b57cec5SDimitry Andric }
7960b57cec5SDimitry Andric 
7970b57cec5SDimitry Andric bool AMDGPUTargetLowering::isZExtFree(Type *Src, Type *Dest) const {
7980b57cec5SDimitry Andric   unsigned SrcSize = Src->getScalarSizeInBits();
7990b57cec5SDimitry Andric   unsigned DestSize = Dest->getScalarSizeInBits();
8000b57cec5SDimitry Andric 
8010b57cec5SDimitry Andric   if (SrcSize == 16 && Subtarget->has16BitInsts())
8020b57cec5SDimitry Andric     return DestSize >= 32;
8030b57cec5SDimitry Andric 
8040b57cec5SDimitry Andric   return SrcSize == 32 && DestSize == 64;
8050b57cec5SDimitry Andric }
8060b57cec5SDimitry Andric 
8070b57cec5SDimitry Andric bool AMDGPUTargetLowering::isZExtFree(EVT Src, EVT Dest) const {
8080b57cec5SDimitry Andric   // Any register load of a 64-bit value really requires 2 32-bit moves. For all
8090b57cec5SDimitry Andric   // practical purposes, the extra mov 0 to load a 64-bit is free.  As used,
8100b57cec5SDimitry Andric   // this will enable reducing 64-bit operations the 32-bit, which is always
8110b57cec5SDimitry Andric   // good.
8120b57cec5SDimitry Andric 
8130b57cec5SDimitry Andric   if (Src == MVT::i16)
8140b57cec5SDimitry Andric     return Dest == MVT::i32 ||Dest == MVT::i64 ;
8150b57cec5SDimitry Andric 
8160b57cec5SDimitry Andric   return Src == MVT::i32 && Dest == MVT::i64;
8170b57cec5SDimitry Andric }
8180b57cec5SDimitry Andric 
8190b57cec5SDimitry Andric bool AMDGPUTargetLowering::isZExtFree(SDValue Val, EVT VT2) const {
8200b57cec5SDimitry Andric   return isZExtFree(Val.getValueType(), VT2);
8210b57cec5SDimitry Andric }
8220b57cec5SDimitry Andric 
8230b57cec5SDimitry Andric bool AMDGPUTargetLowering::isNarrowingProfitable(EVT SrcVT, EVT DestVT) const {
8240b57cec5SDimitry Andric   // There aren't really 64-bit registers, but pairs of 32-bit ones and only a
8250b57cec5SDimitry Andric   // limited number of native 64-bit operations. Shrinking an operation to fit
8260b57cec5SDimitry Andric   // in a single 32-bit register should always be helpful. As currently used,
8270b57cec5SDimitry Andric   // this is much less general than the name suggests, and is only used in
8280b57cec5SDimitry Andric   // places trying to reduce the sizes of loads. Shrinking loads to < 32-bits is
8290b57cec5SDimitry Andric   // not profitable, and may actually be harmful.
8300b57cec5SDimitry Andric   return SrcVT.getSizeInBits() > 32 && DestVT.getSizeInBits() == 32;
8310b57cec5SDimitry Andric }
8320b57cec5SDimitry Andric 
8330b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
8340b57cec5SDimitry Andric // TargetLowering Callbacks
8350b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
8360b57cec5SDimitry Andric 
8370b57cec5SDimitry Andric CCAssignFn *AMDGPUCallLowering::CCAssignFnForCall(CallingConv::ID CC,
8380b57cec5SDimitry Andric                                                   bool IsVarArg) {
8390b57cec5SDimitry Andric   switch (CC) {
8400b57cec5SDimitry Andric   case CallingConv::AMDGPU_VS:
8410b57cec5SDimitry Andric   case CallingConv::AMDGPU_GS:
8420b57cec5SDimitry Andric   case CallingConv::AMDGPU_PS:
8430b57cec5SDimitry Andric   case CallingConv::AMDGPU_CS:
8440b57cec5SDimitry Andric   case CallingConv::AMDGPU_HS:
8450b57cec5SDimitry Andric   case CallingConv::AMDGPU_ES:
8460b57cec5SDimitry Andric   case CallingConv::AMDGPU_LS:
8470b57cec5SDimitry Andric     return CC_AMDGPU;
8480b57cec5SDimitry Andric   case CallingConv::C:
8490b57cec5SDimitry Andric   case CallingConv::Fast:
8500b57cec5SDimitry Andric   case CallingConv::Cold:
8510b57cec5SDimitry Andric     return CC_AMDGPU_Func;
8520b57cec5SDimitry Andric   case CallingConv::AMDGPU_KERNEL:
8530b57cec5SDimitry Andric   case CallingConv::SPIR_KERNEL:
8540b57cec5SDimitry Andric   default:
8550b57cec5SDimitry Andric     report_fatal_error("Unsupported calling convention for call");
8560b57cec5SDimitry Andric   }
8570b57cec5SDimitry Andric }
8580b57cec5SDimitry Andric 
8590b57cec5SDimitry Andric CCAssignFn *AMDGPUCallLowering::CCAssignFnForReturn(CallingConv::ID CC,
8600b57cec5SDimitry Andric                                                     bool IsVarArg) {
8610b57cec5SDimitry Andric   switch (CC) {
8620b57cec5SDimitry Andric   case CallingConv::AMDGPU_KERNEL:
8630b57cec5SDimitry Andric   case CallingConv::SPIR_KERNEL:
8640b57cec5SDimitry Andric     llvm_unreachable("kernels should not be handled here");
8650b57cec5SDimitry Andric   case CallingConv::AMDGPU_VS:
8660b57cec5SDimitry Andric   case CallingConv::AMDGPU_GS:
8670b57cec5SDimitry Andric   case CallingConv::AMDGPU_PS:
8680b57cec5SDimitry Andric   case CallingConv::AMDGPU_CS:
8690b57cec5SDimitry Andric   case CallingConv::AMDGPU_HS:
8700b57cec5SDimitry Andric   case CallingConv::AMDGPU_ES:
8710b57cec5SDimitry Andric   case CallingConv::AMDGPU_LS:
8720b57cec5SDimitry Andric     return RetCC_SI_Shader;
8730b57cec5SDimitry Andric   case CallingConv::C:
8740b57cec5SDimitry Andric   case CallingConv::Fast:
8750b57cec5SDimitry Andric   case CallingConv::Cold:
8760b57cec5SDimitry Andric     return RetCC_AMDGPU_Func;
8770b57cec5SDimitry Andric   default:
8780b57cec5SDimitry Andric     report_fatal_error("Unsupported calling convention.");
8790b57cec5SDimitry Andric   }
8800b57cec5SDimitry Andric }
8810b57cec5SDimitry Andric 
8820b57cec5SDimitry Andric /// The SelectionDAGBuilder will automatically promote function arguments
8830b57cec5SDimitry Andric /// with illegal types.  However, this does not work for the AMDGPU targets
8840b57cec5SDimitry Andric /// since the function arguments are stored in memory as these illegal types.
8850b57cec5SDimitry Andric /// In order to handle this properly we need to get the original types sizes
8860b57cec5SDimitry Andric /// from the LLVM IR Function and fixup the ISD:InputArg values before
8870b57cec5SDimitry Andric /// passing them to AnalyzeFormalArguments()
8880b57cec5SDimitry Andric 
8890b57cec5SDimitry Andric /// When the SelectionDAGBuilder computes the Ins, it takes care of splitting
8900b57cec5SDimitry Andric /// input values across multiple registers.  Each item in the Ins array
8910b57cec5SDimitry Andric /// represents a single value that will be stored in registers.  Ins[x].VT is
8920b57cec5SDimitry Andric /// the value type of the value that will be stored in the register, so
8930b57cec5SDimitry Andric /// whatever SDNode we lower the argument to needs to be this type.
8940b57cec5SDimitry Andric ///
8950b57cec5SDimitry Andric /// In order to correctly lower the arguments we need to know the size of each
8960b57cec5SDimitry Andric /// argument.  Since Ins[x].VT gives us the size of the register that will
8970b57cec5SDimitry Andric /// hold the value, we need to look at Ins[x].ArgVT to see the 'real' type
8980b57cec5SDimitry Andric /// for the orignal function argument so that we can deduce the correct memory
8990b57cec5SDimitry Andric /// type to use for Ins[x].  In most cases the correct memory type will be
9000b57cec5SDimitry Andric /// Ins[x].ArgVT.  However, this will not always be the case.  If, for example,
9010b57cec5SDimitry Andric /// we have a kernel argument of type v8i8, this argument will be split into
9020b57cec5SDimitry Andric /// 8 parts and each part will be represented by its own item in the Ins array.
9030b57cec5SDimitry Andric /// For each part the Ins[x].ArgVT will be the v8i8, which is the full type of
9040b57cec5SDimitry Andric /// the argument before it was split.  From this, we deduce that the memory type
9050b57cec5SDimitry Andric /// for each individual part is i8.  We pass the memory type as LocVT to the
9060b57cec5SDimitry Andric /// calling convention analysis function and the register type (Ins[x].VT) as
9070b57cec5SDimitry Andric /// the ValVT.
9080b57cec5SDimitry Andric void AMDGPUTargetLowering::analyzeFormalArgumentsCompute(
9090b57cec5SDimitry Andric   CCState &State,
9100b57cec5SDimitry Andric   const SmallVectorImpl<ISD::InputArg> &Ins) const {
9110b57cec5SDimitry Andric   const MachineFunction &MF = State.getMachineFunction();
9120b57cec5SDimitry Andric   const Function &Fn = MF.getFunction();
9130b57cec5SDimitry Andric   LLVMContext &Ctx = Fn.getParent()->getContext();
9140b57cec5SDimitry Andric   const AMDGPUSubtarget &ST = AMDGPUSubtarget::get(MF);
9150b57cec5SDimitry Andric   const unsigned ExplicitOffset = ST.getExplicitKernelArgOffset(Fn);
9160b57cec5SDimitry Andric   CallingConv::ID CC = Fn.getCallingConv();
9170b57cec5SDimitry Andric 
9180b57cec5SDimitry Andric   unsigned MaxAlign = 1;
9190b57cec5SDimitry Andric   uint64_t ExplicitArgOffset = 0;
9200b57cec5SDimitry Andric   const DataLayout &DL = Fn.getParent()->getDataLayout();
9210b57cec5SDimitry Andric 
9220b57cec5SDimitry Andric   unsigned InIndex = 0;
9230b57cec5SDimitry Andric 
9240b57cec5SDimitry Andric   for (const Argument &Arg : Fn.args()) {
9250b57cec5SDimitry Andric     Type *BaseArgTy = Arg.getType();
9260b57cec5SDimitry Andric     unsigned Align = DL.getABITypeAlignment(BaseArgTy);
9270b57cec5SDimitry Andric     MaxAlign = std::max(Align, MaxAlign);
9280b57cec5SDimitry Andric     unsigned AllocSize = DL.getTypeAllocSize(BaseArgTy);
9290b57cec5SDimitry Andric 
9300b57cec5SDimitry Andric     uint64_t ArgOffset = alignTo(ExplicitArgOffset, Align) + ExplicitOffset;
9310b57cec5SDimitry Andric     ExplicitArgOffset = alignTo(ExplicitArgOffset, Align) + AllocSize;
9320b57cec5SDimitry Andric 
9330b57cec5SDimitry Andric     // We're basically throwing away everything passed into us and starting over
9340b57cec5SDimitry Andric     // to get accurate in-memory offsets. The "PartOffset" is completely useless
9350b57cec5SDimitry Andric     // to us as computed in Ins.
9360b57cec5SDimitry Andric     //
9370b57cec5SDimitry Andric     // We also need to figure out what type legalization is trying to do to get
9380b57cec5SDimitry Andric     // the correct memory offsets.
9390b57cec5SDimitry Andric 
9400b57cec5SDimitry Andric     SmallVector<EVT, 16> ValueVTs;
9410b57cec5SDimitry Andric     SmallVector<uint64_t, 16> Offsets;
9420b57cec5SDimitry Andric     ComputeValueVTs(*this, DL, BaseArgTy, ValueVTs, &Offsets, ArgOffset);
9430b57cec5SDimitry Andric 
9440b57cec5SDimitry Andric     for (unsigned Value = 0, NumValues = ValueVTs.size();
9450b57cec5SDimitry Andric          Value != NumValues; ++Value) {
9460b57cec5SDimitry Andric       uint64_t BasePartOffset = Offsets[Value];
9470b57cec5SDimitry Andric 
9480b57cec5SDimitry Andric       EVT ArgVT = ValueVTs[Value];
9490b57cec5SDimitry Andric       EVT MemVT = ArgVT;
9500b57cec5SDimitry Andric       MVT RegisterVT = getRegisterTypeForCallingConv(Ctx, CC, ArgVT);
9510b57cec5SDimitry Andric       unsigned NumRegs = getNumRegistersForCallingConv(Ctx, CC, ArgVT);
9520b57cec5SDimitry Andric 
9530b57cec5SDimitry Andric       if (NumRegs == 1) {
9540b57cec5SDimitry Andric         // This argument is not split, so the IR type is the memory type.
9550b57cec5SDimitry Andric         if (ArgVT.isExtended()) {
9560b57cec5SDimitry Andric           // We have an extended type, like i24, so we should just use the
9570b57cec5SDimitry Andric           // register type.
9580b57cec5SDimitry Andric           MemVT = RegisterVT;
9590b57cec5SDimitry Andric         } else {
9600b57cec5SDimitry Andric           MemVT = ArgVT;
9610b57cec5SDimitry Andric         }
9620b57cec5SDimitry Andric       } else if (ArgVT.isVector() && RegisterVT.isVector() &&
9630b57cec5SDimitry Andric                  ArgVT.getScalarType() == RegisterVT.getScalarType()) {
9640b57cec5SDimitry Andric         assert(ArgVT.getVectorNumElements() > RegisterVT.getVectorNumElements());
9650b57cec5SDimitry Andric         // We have a vector value which has been split into a vector with
9660b57cec5SDimitry Andric         // the same scalar type, but fewer elements.  This should handle
9670b57cec5SDimitry Andric         // all the floating-point vector types.
9680b57cec5SDimitry Andric         MemVT = RegisterVT;
9690b57cec5SDimitry Andric       } else if (ArgVT.isVector() &&
9700b57cec5SDimitry Andric                  ArgVT.getVectorNumElements() == NumRegs) {
9710b57cec5SDimitry Andric         // This arg has been split so that each element is stored in a separate
9720b57cec5SDimitry Andric         // register.
9730b57cec5SDimitry Andric         MemVT = ArgVT.getScalarType();
9740b57cec5SDimitry Andric       } else if (ArgVT.isExtended()) {
9750b57cec5SDimitry Andric         // We have an extended type, like i65.
9760b57cec5SDimitry Andric         MemVT = RegisterVT;
9770b57cec5SDimitry Andric       } else {
9780b57cec5SDimitry Andric         unsigned MemoryBits = ArgVT.getStoreSizeInBits() / NumRegs;
9790b57cec5SDimitry Andric         assert(ArgVT.getStoreSizeInBits() % NumRegs == 0);
9800b57cec5SDimitry Andric         if (RegisterVT.isInteger()) {
9810b57cec5SDimitry Andric           MemVT = EVT::getIntegerVT(State.getContext(), MemoryBits);
9820b57cec5SDimitry Andric         } else if (RegisterVT.isVector()) {
9830b57cec5SDimitry Andric           assert(!RegisterVT.getScalarType().isFloatingPoint());
9840b57cec5SDimitry Andric           unsigned NumElements = RegisterVT.getVectorNumElements();
9850b57cec5SDimitry Andric           assert(MemoryBits % NumElements == 0);
9860b57cec5SDimitry Andric           // This vector type has been split into another vector type with
9870b57cec5SDimitry Andric           // a different elements size.
9880b57cec5SDimitry Andric           EVT ScalarVT = EVT::getIntegerVT(State.getContext(),
9890b57cec5SDimitry Andric                                            MemoryBits / NumElements);
9900b57cec5SDimitry Andric           MemVT = EVT::getVectorVT(State.getContext(), ScalarVT, NumElements);
9910b57cec5SDimitry Andric         } else {
9920b57cec5SDimitry Andric           llvm_unreachable("cannot deduce memory type.");
9930b57cec5SDimitry Andric         }
9940b57cec5SDimitry Andric       }
9950b57cec5SDimitry Andric 
9960b57cec5SDimitry Andric       // Convert one element vectors to scalar.
9970b57cec5SDimitry Andric       if (MemVT.isVector() && MemVT.getVectorNumElements() == 1)
9980b57cec5SDimitry Andric         MemVT = MemVT.getScalarType();
9990b57cec5SDimitry Andric 
10000b57cec5SDimitry Andric       // Round up vec3/vec5 argument.
10010b57cec5SDimitry Andric       if (MemVT.isVector() && !MemVT.isPow2VectorType()) {
10020b57cec5SDimitry Andric         assert(MemVT.getVectorNumElements() == 3 ||
10030b57cec5SDimitry Andric                MemVT.getVectorNumElements() == 5);
10040b57cec5SDimitry Andric         MemVT = MemVT.getPow2VectorType(State.getContext());
10050b57cec5SDimitry Andric       }
10060b57cec5SDimitry Andric 
10070b57cec5SDimitry Andric       unsigned PartOffset = 0;
10080b57cec5SDimitry Andric       for (unsigned i = 0; i != NumRegs; ++i) {
10090b57cec5SDimitry Andric         State.addLoc(CCValAssign::getCustomMem(InIndex++, RegisterVT,
10100b57cec5SDimitry Andric                                                BasePartOffset + PartOffset,
10110b57cec5SDimitry Andric                                                MemVT.getSimpleVT(),
10120b57cec5SDimitry Andric                                                CCValAssign::Full));
10130b57cec5SDimitry Andric         PartOffset += MemVT.getStoreSize();
10140b57cec5SDimitry Andric       }
10150b57cec5SDimitry Andric     }
10160b57cec5SDimitry Andric   }
10170b57cec5SDimitry Andric }
10180b57cec5SDimitry Andric 
10190b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerReturn(
10200b57cec5SDimitry Andric   SDValue Chain, CallingConv::ID CallConv,
10210b57cec5SDimitry Andric   bool isVarArg,
10220b57cec5SDimitry Andric   const SmallVectorImpl<ISD::OutputArg> &Outs,
10230b57cec5SDimitry Andric   const SmallVectorImpl<SDValue> &OutVals,
10240b57cec5SDimitry Andric   const SDLoc &DL, SelectionDAG &DAG) const {
10250b57cec5SDimitry Andric   // FIXME: Fails for r600 tests
10260b57cec5SDimitry Andric   //assert(!isVarArg && Outs.empty() && OutVals.empty() &&
10270b57cec5SDimitry Andric   // "wave terminate should not have return values");
10280b57cec5SDimitry Andric   return DAG.getNode(AMDGPUISD::ENDPGM, DL, MVT::Other, Chain);
10290b57cec5SDimitry Andric }
10300b57cec5SDimitry Andric 
10310b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
10320b57cec5SDimitry Andric // Target specific lowering
10330b57cec5SDimitry Andric //===---------------------------------------------------------------------===//
10340b57cec5SDimitry Andric 
10350b57cec5SDimitry Andric /// Selects the correct CCAssignFn for a given CallingConvention value.
10360b57cec5SDimitry Andric CCAssignFn *AMDGPUTargetLowering::CCAssignFnForCall(CallingConv::ID CC,
10370b57cec5SDimitry Andric                                                     bool IsVarArg) {
10380b57cec5SDimitry Andric   return AMDGPUCallLowering::CCAssignFnForCall(CC, IsVarArg);
10390b57cec5SDimitry Andric }
10400b57cec5SDimitry Andric 
10410b57cec5SDimitry Andric CCAssignFn *AMDGPUTargetLowering::CCAssignFnForReturn(CallingConv::ID CC,
10420b57cec5SDimitry Andric                                                       bool IsVarArg) {
10430b57cec5SDimitry Andric   return AMDGPUCallLowering::CCAssignFnForReturn(CC, IsVarArg);
10440b57cec5SDimitry Andric }
10450b57cec5SDimitry Andric 
10460b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::addTokenForArgument(SDValue Chain,
10470b57cec5SDimitry Andric                                                   SelectionDAG &DAG,
10480b57cec5SDimitry Andric                                                   MachineFrameInfo &MFI,
10490b57cec5SDimitry Andric                                                   int ClobberedFI) const {
10500b57cec5SDimitry Andric   SmallVector<SDValue, 8> ArgChains;
10510b57cec5SDimitry Andric   int64_t FirstByte = MFI.getObjectOffset(ClobberedFI);
10520b57cec5SDimitry Andric   int64_t LastByte = FirstByte + MFI.getObjectSize(ClobberedFI) - 1;
10530b57cec5SDimitry Andric 
10540b57cec5SDimitry Andric   // Include the original chain at the beginning of the list. When this is
10550b57cec5SDimitry Andric   // used by target LowerCall hooks, this helps legalize find the
10560b57cec5SDimitry Andric   // CALLSEQ_BEGIN node.
10570b57cec5SDimitry Andric   ArgChains.push_back(Chain);
10580b57cec5SDimitry Andric 
10590b57cec5SDimitry Andric   // Add a chain value for each stack argument corresponding
10600b57cec5SDimitry Andric   for (SDNode::use_iterator U = DAG.getEntryNode().getNode()->use_begin(),
10610b57cec5SDimitry Andric                             UE = DAG.getEntryNode().getNode()->use_end();
10620b57cec5SDimitry Andric        U != UE; ++U) {
10630b57cec5SDimitry Andric     if (LoadSDNode *L = dyn_cast<LoadSDNode>(*U)) {
10640b57cec5SDimitry Andric       if (FrameIndexSDNode *FI = dyn_cast<FrameIndexSDNode>(L->getBasePtr())) {
10650b57cec5SDimitry Andric         if (FI->getIndex() < 0) {
10660b57cec5SDimitry Andric           int64_t InFirstByte = MFI.getObjectOffset(FI->getIndex());
10670b57cec5SDimitry Andric           int64_t InLastByte = InFirstByte;
10680b57cec5SDimitry Andric           InLastByte += MFI.getObjectSize(FI->getIndex()) - 1;
10690b57cec5SDimitry Andric 
10700b57cec5SDimitry Andric           if ((InFirstByte <= FirstByte && FirstByte <= InLastByte) ||
10710b57cec5SDimitry Andric               (FirstByte <= InFirstByte && InFirstByte <= LastByte))
10720b57cec5SDimitry Andric             ArgChains.push_back(SDValue(L, 1));
10730b57cec5SDimitry Andric         }
10740b57cec5SDimitry Andric       }
10750b57cec5SDimitry Andric     }
10760b57cec5SDimitry Andric   }
10770b57cec5SDimitry Andric 
10780b57cec5SDimitry Andric   // Build a tokenfactor for all the chains.
10790b57cec5SDimitry Andric   return DAG.getNode(ISD::TokenFactor, SDLoc(Chain), MVT::Other, ArgChains);
10800b57cec5SDimitry Andric }
10810b57cec5SDimitry Andric 
10820b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::lowerUnhandledCall(CallLoweringInfo &CLI,
10830b57cec5SDimitry Andric                                                  SmallVectorImpl<SDValue> &InVals,
10840b57cec5SDimitry Andric                                                  StringRef Reason) const {
10850b57cec5SDimitry Andric   SDValue Callee = CLI.Callee;
10860b57cec5SDimitry Andric   SelectionDAG &DAG = CLI.DAG;
10870b57cec5SDimitry Andric 
10880b57cec5SDimitry Andric   const Function &Fn = DAG.getMachineFunction().getFunction();
10890b57cec5SDimitry Andric 
10900b57cec5SDimitry Andric   StringRef FuncName("<unknown>");
10910b57cec5SDimitry Andric 
10920b57cec5SDimitry Andric   if (const ExternalSymbolSDNode *G = dyn_cast<ExternalSymbolSDNode>(Callee))
10930b57cec5SDimitry Andric     FuncName = G->getSymbol();
10940b57cec5SDimitry Andric   else if (const GlobalAddressSDNode *G = dyn_cast<GlobalAddressSDNode>(Callee))
10950b57cec5SDimitry Andric     FuncName = G->getGlobal()->getName();
10960b57cec5SDimitry Andric 
10970b57cec5SDimitry Andric   DiagnosticInfoUnsupported NoCalls(
10980b57cec5SDimitry Andric     Fn, Reason + FuncName, CLI.DL.getDebugLoc());
10990b57cec5SDimitry Andric   DAG.getContext()->diagnose(NoCalls);
11000b57cec5SDimitry Andric 
11010b57cec5SDimitry Andric   if (!CLI.IsTailCall) {
11020b57cec5SDimitry Andric     for (unsigned I = 0, E = CLI.Ins.size(); I != E; ++I)
11030b57cec5SDimitry Andric       InVals.push_back(DAG.getUNDEF(CLI.Ins[I].VT));
11040b57cec5SDimitry Andric   }
11050b57cec5SDimitry Andric 
11060b57cec5SDimitry Andric   return DAG.getEntryNode();
11070b57cec5SDimitry Andric }
11080b57cec5SDimitry Andric 
11090b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerCall(CallLoweringInfo &CLI,
11100b57cec5SDimitry Andric                                         SmallVectorImpl<SDValue> &InVals) const {
11110b57cec5SDimitry Andric   return lowerUnhandledCall(CLI, InVals, "unsupported call to function ");
11120b57cec5SDimitry Andric }
11130b57cec5SDimitry Andric 
11140b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerDYNAMIC_STACKALLOC(SDValue Op,
11150b57cec5SDimitry Andric                                                       SelectionDAG &DAG) const {
11160b57cec5SDimitry Andric   const Function &Fn = DAG.getMachineFunction().getFunction();
11170b57cec5SDimitry Andric 
11180b57cec5SDimitry Andric   DiagnosticInfoUnsupported NoDynamicAlloca(Fn, "unsupported dynamic alloca",
11190b57cec5SDimitry Andric                                             SDLoc(Op).getDebugLoc());
11200b57cec5SDimitry Andric   DAG.getContext()->diagnose(NoDynamicAlloca);
11210b57cec5SDimitry Andric   auto Ops = {DAG.getConstant(0, SDLoc(), Op.getValueType()), Op.getOperand(0)};
11220b57cec5SDimitry Andric   return DAG.getMergeValues(Ops, SDLoc());
11230b57cec5SDimitry Andric }
11240b57cec5SDimitry Andric 
11250b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerOperation(SDValue Op,
11260b57cec5SDimitry Andric                                              SelectionDAG &DAG) const {
11270b57cec5SDimitry Andric   switch (Op.getOpcode()) {
11280b57cec5SDimitry Andric   default:
11290b57cec5SDimitry Andric     Op->print(errs(), &DAG);
11300b57cec5SDimitry Andric     llvm_unreachable("Custom lowering code for this"
11310b57cec5SDimitry Andric                      "instruction is not implemented yet!");
11320b57cec5SDimitry Andric     break;
11330b57cec5SDimitry Andric   case ISD::SIGN_EXTEND_INREG: return LowerSIGN_EXTEND_INREG(Op, DAG);
11340b57cec5SDimitry Andric   case ISD::CONCAT_VECTORS: return LowerCONCAT_VECTORS(Op, DAG);
11350b57cec5SDimitry Andric   case ISD::EXTRACT_SUBVECTOR: return LowerEXTRACT_SUBVECTOR(Op, DAG);
11360b57cec5SDimitry Andric   case ISD::UDIVREM: return LowerUDIVREM(Op, DAG);
11370b57cec5SDimitry Andric   case ISD::SDIVREM: return LowerSDIVREM(Op, DAG);
11380b57cec5SDimitry Andric   case ISD::FREM: return LowerFREM(Op, DAG);
11390b57cec5SDimitry Andric   case ISD::FCEIL: return LowerFCEIL(Op, DAG);
11400b57cec5SDimitry Andric   case ISD::FTRUNC: return LowerFTRUNC(Op, DAG);
11410b57cec5SDimitry Andric   case ISD::FRINT: return LowerFRINT(Op, DAG);
11420b57cec5SDimitry Andric   case ISD::FNEARBYINT: return LowerFNEARBYINT(Op, DAG);
11430b57cec5SDimitry Andric   case ISD::FROUND: return LowerFROUND(Op, DAG);
11440b57cec5SDimitry Andric   case ISD::FFLOOR: return LowerFFLOOR(Op, DAG);
11450b57cec5SDimitry Andric   case ISD::FLOG:
1146*8bcb0991SDimitry Andric     return LowerFLOG(Op, DAG, 1.0F / numbers::log2ef);
11470b57cec5SDimitry Andric   case ISD::FLOG10:
1148*8bcb0991SDimitry Andric     return LowerFLOG(Op, DAG, numbers::ln2f / numbers::ln10f);
11490b57cec5SDimitry Andric   case ISD::FEXP:
11500b57cec5SDimitry Andric     return lowerFEXP(Op, DAG);
11510b57cec5SDimitry Andric   case ISD::SINT_TO_FP: return LowerSINT_TO_FP(Op, DAG);
11520b57cec5SDimitry Andric   case ISD::UINT_TO_FP: return LowerUINT_TO_FP(Op, DAG);
11530b57cec5SDimitry Andric   case ISD::FP_TO_FP16: return LowerFP_TO_FP16(Op, DAG);
11540b57cec5SDimitry Andric   case ISD::FP_TO_SINT: return LowerFP_TO_SINT(Op, DAG);
11550b57cec5SDimitry Andric   case ISD::FP_TO_UINT: return LowerFP_TO_UINT(Op, DAG);
11560b57cec5SDimitry Andric   case ISD::CTTZ:
11570b57cec5SDimitry Andric   case ISD::CTTZ_ZERO_UNDEF:
11580b57cec5SDimitry Andric   case ISD::CTLZ:
11590b57cec5SDimitry Andric   case ISD::CTLZ_ZERO_UNDEF:
11600b57cec5SDimitry Andric     return LowerCTLZ_CTTZ(Op, DAG);
11610b57cec5SDimitry Andric   case ISD::DYNAMIC_STACKALLOC: return LowerDYNAMIC_STACKALLOC(Op, DAG);
11620b57cec5SDimitry Andric   }
11630b57cec5SDimitry Andric   return Op;
11640b57cec5SDimitry Andric }
11650b57cec5SDimitry Andric 
11660b57cec5SDimitry Andric void AMDGPUTargetLowering::ReplaceNodeResults(SDNode *N,
11670b57cec5SDimitry Andric                                               SmallVectorImpl<SDValue> &Results,
11680b57cec5SDimitry Andric                                               SelectionDAG &DAG) const {
11690b57cec5SDimitry Andric   switch (N->getOpcode()) {
11700b57cec5SDimitry Andric   case ISD::SIGN_EXTEND_INREG:
11710b57cec5SDimitry Andric     // Different parts of legalization seem to interpret which type of
11720b57cec5SDimitry Andric     // sign_extend_inreg is the one to check for custom lowering. The extended
11730b57cec5SDimitry Andric     // from type is what really matters, but some places check for custom
11740b57cec5SDimitry Andric     // lowering of the result type. This results in trying to use
11750b57cec5SDimitry Andric     // ReplaceNodeResults to sext_in_reg to an illegal type, so we'll just do
11760b57cec5SDimitry Andric     // nothing here and let the illegal result integer be handled normally.
11770b57cec5SDimitry Andric     return;
11780b57cec5SDimitry Andric   default:
11790b57cec5SDimitry Andric     return;
11800b57cec5SDimitry Andric   }
11810b57cec5SDimitry Andric }
11820b57cec5SDimitry Andric 
1183*8bcb0991SDimitry Andric bool AMDGPUTargetLowering::hasDefinedInitializer(const GlobalValue *GV) {
11840b57cec5SDimitry Andric   const GlobalVariable *GVar = dyn_cast<GlobalVariable>(GV);
11850b57cec5SDimitry Andric   if (!GVar || !GVar->hasInitializer())
11860b57cec5SDimitry Andric     return false;
11870b57cec5SDimitry Andric 
11880b57cec5SDimitry Andric   return !isa<UndefValue>(GVar->getInitializer());
11890b57cec5SDimitry Andric }
11900b57cec5SDimitry Andric 
11910b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerGlobalAddress(AMDGPUMachineFunction* MFI,
11920b57cec5SDimitry Andric                                                  SDValue Op,
11930b57cec5SDimitry Andric                                                  SelectionDAG &DAG) const {
11940b57cec5SDimitry Andric 
11950b57cec5SDimitry Andric   const DataLayout &DL = DAG.getDataLayout();
11960b57cec5SDimitry Andric   GlobalAddressSDNode *G = cast<GlobalAddressSDNode>(Op);
11970b57cec5SDimitry Andric   const GlobalValue *GV = G->getGlobal();
11980b57cec5SDimitry Andric 
11990b57cec5SDimitry Andric   if (G->getAddressSpace() == AMDGPUAS::LOCAL_ADDRESS ||
12000b57cec5SDimitry Andric       G->getAddressSpace() == AMDGPUAS::REGION_ADDRESS) {
12010b57cec5SDimitry Andric     if (!MFI->isEntryFunction()) {
12020b57cec5SDimitry Andric       const Function &Fn = DAG.getMachineFunction().getFunction();
12030b57cec5SDimitry Andric       DiagnosticInfoUnsupported BadLDSDecl(
12040b57cec5SDimitry Andric         Fn, "local memory global used by non-kernel function", SDLoc(Op).getDebugLoc());
12050b57cec5SDimitry Andric       DAG.getContext()->diagnose(BadLDSDecl);
12060b57cec5SDimitry Andric     }
12070b57cec5SDimitry Andric 
12080b57cec5SDimitry Andric     // XXX: What does the value of G->getOffset() mean?
12090b57cec5SDimitry Andric     assert(G->getOffset() == 0 &&
12100b57cec5SDimitry Andric          "Do not know what to do with an non-zero offset");
12110b57cec5SDimitry Andric 
12120b57cec5SDimitry Andric     // TODO: We could emit code to handle the initialization somewhere.
12130b57cec5SDimitry Andric     if (!hasDefinedInitializer(GV)) {
12140b57cec5SDimitry Andric       unsigned Offset = MFI->allocateLDSGlobal(DL, *GV);
12150b57cec5SDimitry Andric       return DAG.getConstant(Offset, SDLoc(Op), Op.getValueType());
12160b57cec5SDimitry Andric     }
12170b57cec5SDimitry Andric   }
12180b57cec5SDimitry Andric 
12190b57cec5SDimitry Andric   const Function &Fn = DAG.getMachineFunction().getFunction();
12200b57cec5SDimitry Andric   DiagnosticInfoUnsupported BadInit(
12210b57cec5SDimitry Andric       Fn, "unsupported initializer for address space", SDLoc(Op).getDebugLoc());
12220b57cec5SDimitry Andric   DAG.getContext()->diagnose(BadInit);
12230b57cec5SDimitry Andric   return SDValue();
12240b57cec5SDimitry Andric }
12250b57cec5SDimitry Andric 
12260b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerCONCAT_VECTORS(SDValue Op,
12270b57cec5SDimitry Andric                                                   SelectionDAG &DAG) const {
12280b57cec5SDimitry Andric   SmallVector<SDValue, 8> Args;
12290b57cec5SDimitry Andric 
12300b57cec5SDimitry Andric   EVT VT = Op.getValueType();
12310b57cec5SDimitry Andric   if (VT == MVT::v4i16 || VT == MVT::v4f16) {
12320b57cec5SDimitry Andric     SDLoc SL(Op);
12330b57cec5SDimitry Andric     SDValue Lo = DAG.getNode(ISD::BITCAST, SL, MVT::i32, Op.getOperand(0));
12340b57cec5SDimitry Andric     SDValue Hi = DAG.getNode(ISD::BITCAST, SL, MVT::i32, Op.getOperand(1));
12350b57cec5SDimitry Andric 
12360b57cec5SDimitry Andric     SDValue BV = DAG.getBuildVector(MVT::v2i32, SL, { Lo, Hi });
12370b57cec5SDimitry Andric     return DAG.getNode(ISD::BITCAST, SL, VT, BV);
12380b57cec5SDimitry Andric   }
12390b57cec5SDimitry Andric 
12400b57cec5SDimitry Andric   for (const SDUse &U : Op->ops())
12410b57cec5SDimitry Andric     DAG.ExtractVectorElements(U.get(), Args);
12420b57cec5SDimitry Andric 
12430b57cec5SDimitry Andric   return DAG.getBuildVector(Op.getValueType(), SDLoc(Op), Args);
12440b57cec5SDimitry Andric }
12450b57cec5SDimitry Andric 
12460b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerEXTRACT_SUBVECTOR(SDValue Op,
12470b57cec5SDimitry Andric                                                      SelectionDAG &DAG) const {
12480b57cec5SDimitry Andric 
12490b57cec5SDimitry Andric   SmallVector<SDValue, 8> Args;
12500b57cec5SDimitry Andric   unsigned Start = cast<ConstantSDNode>(Op.getOperand(1))->getZExtValue();
12510b57cec5SDimitry Andric   EVT VT = Op.getValueType();
12520b57cec5SDimitry Andric   DAG.ExtractVectorElements(Op.getOperand(0), Args, Start,
12530b57cec5SDimitry Andric                             VT.getVectorNumElements());
12540b57cec5SDimitry Andric 
12550b57cec5SDimitry Andric   return DAG.getBuildVector(Op.getValueType(), SDLoc(Op), Args);
12560b57cec5SDimitry Andric }
12570b57cec5SDimitry Andric 
12580b57cec5SDimitry Andric /// Generate Min/Max node
12590b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::combineFMinMaxLegacy(const SDLoc &DL, EVT VT,
12600b57cec5SDimitry Andric                                                    SDValue LHS, SDValue RHS,
12610b57cec5SDimitry Andric                                                    SDValue True, SDValue False,
12620b57cec5SDimitry Andric                                                    SDValue CC,
12630b57cec5SDimitry Andric                                                    DAGCombinerInfo &DCI) const {
12640b57cec5SDimitry Andric   if (!(LHS == True && RHS == False) && !(LHS == False && RHS == True))
12650b57cec5SDimitry Andric     return SDValue();
12660b57cec5SDimitry Andric 
12670b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
12680b57cec5SDimitry Andric   ISD::CondCode CCOpcode = cast<CondCodeSDNode>(CC)->get();
12690b57cec5SDimitry Andric   switch (CCOpcode) {
12700b57cec5SDimitry Andric   case ISD::SETOEQ:
12710b57cec5SDimitry Andric   case ISD::SETONE:
12720b57cec5SDimitry Andric   case ISD::SETUNE:
12730b57cec5SDimitry Andric   case ISD::SETNE:
12740b57cec5SDimitry Andric   case ISD::SETUEQ:
12750b57cec5SDimitry Andric   case ISD::SETEQ:
12760b57cec5SDimitry Andric   case ISD::SETFALSE:
12770b57cec5SDimitry Andric   case ISD::SETFALSE2:
12780b57cec5SDimitry Andric   case ISD::SETTRUE:
12790b57cec5SDimitry Andric   case ISD::SETTRUE2:
12800b57cec5SDimitry Andric   case ISD::SETUO:
12810b57cec5SDimitry Andric   case ISD::SETO:
12820b57cec5SDimitry Andric     break;
12830b57cec5SDimitry Andric   case ISD::SETULE:
12840b57cec5SDimitry Andric   case ISD::SETULT: {
12850b57cec5SDimitry Andric     if (LHS == True)
12860b57cec5SDimitry Andric       return DAG.getNode(AMDGPUISD::FMIN_LEGACY, DL, VT, RHS, LHS);
12870b57cec5SDimitry Andric     return DAG.getNode(AMDGPUISD::FMAX_LEGACY, DL, VT, LHS, RHS);
12880b57cec5SDimitry Andric   }
12890b57cec5SDimitry Andric   case ISD::SETOLE:
12900b57cec5SDimitry Andric   case ISD::SETOLT:
12910b57cec5SDimitry Andric   case ISD::SETLE:
12920b57cec5SDimitry Andric   case ISD::SETLT: {
12930b57cec5SDimitry Andric     // Ordered. Assume ordered for undefined.
12940b57cec5SDimitry Andric 
12950b57cec5SDimitry Andric     // Only do this after legalization to avoid interfering with other combines
12960b57cec5SDimitry Andric     // which might occur.
12970b57cec5SDimitry Andric     if (DCI.getDAGCombineLevel() < AfterLegalizeDAG &&
12980b57cec5SDimitry Andric         !DCI.isCalledByLegalizer())
12990b57cec5SDimitry Andric       return SDValue();
13000b57cec5SDimitry Andric 
13010b57cec5SDimitry Andric     // We need to permute the operands to get the correct NaN behavior. The
13020b57cec5SDimitry Andric     // selected operand is the second one based on the failing compare with NaN,
13030b57cec5SDimitry Andric     // so permute it based on the compare type the hardware uses.
13040b57cec5SDimitry Andric     if (LHS == True)
13050b57cec5SDimitry Andric       return DAG.getNode(AMDGPUISD::FMIN_LEGACY, DL, VT, LHS, RHS);
13060b57cec5SDimitry Andric     return DAG.getNode(AMDGPUISD::FMAX_LEGACY, DL, VT, RHS, LHS);
13070b57cec5SDimitry Andric   }
13080b57cec5SDimitry Andric   case ISD::SETUGE:
13090b57cec5SDimitry Andric   case ISD::SETUGT: {
13100b57cec5SDimitry Andric     if (LHS == True)
13110b57cec5SDimitry Andric       return DAG.getNode(AMDGPUISD::FMAX_LEGACY, DL, VT, RHS, LHS);
13120b57cec5SDimitry Andric     return DAG.getNode(AMDGPUISD::FMIN_LEGACY, DL, VT, LHS, RHS);
13130b57cec5SDimitry Andric   }
13140b57cec5SDimitry Andric   case ISD::SETGT:
13150b57cec5SDimitry Andric   case ISD::SETGE:
13160b57cec5SDimitry Andric   case ISD::SETOGE:
13170b57cec5SDimitry Andric   case ISD::SETOGT: {
13180b57cec5SDimitry Andric     if (DCI.getDAGCombineLevel() < AfterLegalizeDAG &&
13190b57cec5SDimitry Andric         !DCI.isCalledByLegalizer())
13200b57cec5SDimitry Andric       return SDValue();
13210b57cec5SDimitry Andric 
13220b57cec5SDimitry Andric     if (LHS == True)
13230b57cec5SDimitry Andric       return DAG.getNode(AMDGPUISD::FMAX_LEGACY, DL, VT, LHS, RHS);
13240b57cec5SDimitry Andric     return DAG.getNode(AMDGPUISD::FMIN_LEGACY, DL, VT, RHS, LHS);
13250b57cec5SDimitry Andric   }
13260b57cec5SDimitry Andric   case ISD::SETCC_INVALID:
13270b57cec5SDimitry Andric     llvm_unreachable("Invalid setcc condcode!");
13280b57cec5SDimitry Andric   }
13290b57cec5SDimitry Andric   return SDValue();
13300b57cec5SDimitry Andric }
13310b57cec5SDimitry Andric 
13320b57cec5SDimitry Andric std::pair<SDValue, SDValue>
13330b57cec5SDimitry Andric AMDGPUTargetLowering::split64BitValue(SDValue Op, SelectionDAG &DAG) const {
13340b57cec5SDimitry Andric   SDLoc SL(Op);
13350b57cec5SDimitry Andric 
13360b57cec5SDimitry Andric   SDValue Vec = DAG.getNode(ISD::BITCAST, SL, MVT::v2i32, Op);
13370b57cec5SDimitry Andric 
13380b57cec5SDimitry Andric   const SDValue Zero = DAG.getConstant(0, SL, MVT::i32);
13390b57cec5SDimitry Andric   const SDValue One = DAG.getConstant(1, SL, MVT::i32);
13400b57cec5SDimitry Andric 
13410b57cec5SDimitry Andric   SDValue Lo = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, MVT::i32, Vec, Zero);
13420b57cec5SDimitry Andric   SDValue Hi = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, MVT::i32, Vec, One);
13430b57cec5SDimitry Andric 
13440b57cec5SDimitry Andric   return std::make_pair(Lo, Hi);
13450b57cec5SDimitry Andric }
13460b57cec5SDimitry Andric 
13470b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::getLoHalf64(SDValue Op, SelectionDAG &DAG) const {
13480b57cec5SDimitry Andric   SDLoc SL(Op);
13490b57cec5SDimitry Andric 
13500b57cec5SDimitry Andric   SDValue Vec = DAG.getNode(ISD::BITCAST, SL, MVT::v2i32, Op);
13510b57cec5SDimitry Andric   const SDValue Zero = DAG.getConstant(0, SL, MVT::i32);
13520b57cec5SDimitry Andric   return DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, MVT::i32, Vec, Zero);
13530b57cec5SDimitry Andric }
13540b57cec5SDimitry Andric 
13550b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::getHiHalf64(SDValue Op, SelectionDAG &DAG) const {
13560b57cec5SDimitry Andric   SDLoc SL(Op);
13570b57cec5SDimitry Andric 
13580b57cec5SDimitry Andric   SDValue Vec = DAG.getNode(ISD::BITCAST, SL, MVT::v2i32, Op);
13590b57cec5SDimitry Andric   const SDValue One = DAG.getConstant(1, SL, MVT::i32);
13600b57cec5SDimitry Andric   return DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, MVT::i32, Vec, One);
13610b57cec5SDimitry Andric }
13620b57cec5SDimitry Andric 
13630b57cec5SDimitry Andric // Split a vector type into two parts. The first part is a power of two vector.
13640b57cec5SDimitry Andric // The second part is whatever is left over, and is a scalar if it would
13650b57cec5SDimitry Andric // otherwise be a 1-vector.
13660b57cec5SDimitry Andric std::pair<EVT, EVT>
13670b57cec5SDimitry Andric AMDGPUTargetLowering::getSplitDestVTs(const EVT &VT, SelectionDAG &DAG) const {
13680b57cec5SDimitry Andric   EVT LoVT, HiVT;
13690b57cec5SDimitry Andric   EVT EltVT = VT.getVectorElementType();
13700b57cec5SDimitry Andric   unsigned NumElts = VT.getVectorNumElements();
13710b57cec5SDimitry Andric   unsigned LoNumElts = PowerOf2Ceil((NumElts + 1) / 2);
13720b57cec5SDimitry Andric   LoVT = EVT::getVectorVT(*DAG.getContext(), EltVT, LoNumElts);
13730b57cec5SDimitry Andric   HiVT = NumElts - LoNumElts == 1
13740b57cec5SDimitry Andric              ? EltVT
13750b57cec5SDimitry Andric              : EVT::getVectorVT(*DAG.getContext(), EltVT, NumElts - LoNumElts);
13760b57cec5SDimitry Andric   return std::make_pair(LoVT, HiVT);
13770b57cec5SDimitry Andric }
13780b57cec5SDimitry Andric 
13790b57cec5SDimitry Andric // Split a vector value into two parts of types LoVT and HiVT. HiVT could be
13800b57cec5SDimitry Andric // scalar.
13810b57cec5SDimitry Andric std::pair<SDValue, SDValue>
13820b57cec5SDimitry Andric AMDGPUTargetLowering::splitVector(const SDValue &N, const SDLoc &DL,
13830b57cec5SDimitry Andric                                   const EVT &LoVT, const EVT &HiVT,
13840b57cec5SDimitry Andric                                   SelectionDAG &DAG) const {
13850b57cec5SDimitry Andric   assert(LoVT.getVectorNumElements() +
13860b57cec5SDimitry Andric                  (HiVT.isVector() ? HiVT.getVectorNumElements() : 1) <=
13870b57cec5SDimitry Andric              N.getValueType().getVectorNumElements() &&
13880b57cec5SDimitry Andric          "More vector elements requested than available!");
13890b57cec5SDimitry Andric   auto IdxTy = getVectorIdxTy(DAG.getDataLayout());
13900b57cec5SDimitry Andric   SDValue Lo = DAG.getNode(ISD::EXTRACT_SUBVECTOR, DL, LoVT, N,
13910b57cec5SDimitry Andric                            DAG.getConstant(0, DL, IdxTy));
13920b57cec5SDimitry Andric   SDValue Hi = DAG.getNode(
13930b57cec5SDimitry Andric       HiVT.isVector() ? ISD::EXTRACT_SUBVECTOR : ISD::EXTRACT_VECTOR_ELT, DL,
13940b57cec5SDimitry Andric       HiVT, N, DAG.getConstant(LoVT.getVectorNumElements(), DL, IdxTy));
13950b57cec5SDimitry Andric   return std::make_pair(Lo, Hi);
13960b57cec5SDimitry Andric }
13970b57cec5SDimitry Andric 
13980b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::SplitVectorLoad(const SDValue Op,
13990b57cec5SDimitry Andric                                               SelectionDAG &DAG) const {
14000b57cec5SDimitry Andric   LoadSDNode *Load = cast<LoadSDNode>(Op);
14010b57cec5SDimitry Andric   EVT VT = Op.getValueType();
14020b57cec5SDimitry Andric 
14030b57cec5SDimitry Andric 
14040b57cec5SDimitry Andric   // If this is a 2 element vector, we really want to scalarize and not create
14050b57cec5SDimitry Andric   // weird 1 element vectors.
14060b57cec5SDimitry Andric   if (VT.getVectorNumElements() == 2)
14070b57cec5SDimitry Andric     return scalarizeVectorLoad(Load, DAG);
14080b57cec5SDimitry Andric 
14090b57cec5SDimitry Andric   SDValue BasePtr = Load->getBasePtr();
14100b57cec5SDimitry Andric   EVT MemVT = Load->getMemoryVT();
14110b57cec5SDimitry Andric   SDLoc SL(Op);
14120b57cec5SDimitry Andric 
14130b57cec5SDimitry Andric   const MachinePointerInfo &SrcValue = Load->getMemOperand()->getPointerInfo();
14140b57cec5SDimitry Andric 
14150b57cec5SDimitry Andric   EVT LoVT, HiVT;
14160b57cec5SDimitry Andric   EVT LoMemVT, HiMemVT;
14170b57cec5SDimitry Andric   SDValue Lo, Hi;
14180b57cec5SDimitry Andric 
14190b57cec5SDimitry Andric   std::tie(LoVT, HiVT) = getSplitDestVTs(VT, DAG);
14200b57cec5SDimitry Andric   std::tie(LoMemVT, HiMemVT) = getSplitDestVTs(MemVT, DAG);
14210b57cec5SDimitry Andric   std::tie(Lo, Hi) = splitVector(Op, SL, LoVT, HiVT, DAG);
14220b57cec5SDimitry Andric 
14230b57cec5SDimitry Andric   unsigned Size = LoMemVT.getStoreSize();
14240b57cec5SDimitry Andric   unsigned BaseAlign = Load->getAlignment();
14250b57cec5SDimitry Andric   unsigned HiAlign = MinAlign(BaseAlign, Size);
14260b57cec5SDimitry Andric 
14270b57cec5SDimitry Andric   SDValue LoLoad = DAG.getExtLoad(Load->getExtensionType(), SL, LoVT,
14280b57cec5SDimitry Andric                                   Load->getChain(), BasePtr, SrcValue, LoMemVT,
14290b57cec5SDimitry Andric                                   BaseAlign, Load->getMemOperand()->getFlags());
14300b57cec5SDimitry Andric   SDValue HiPtr = DAG.getObjectPtrOffset(SL, BasePtr, Size);
14310b57cec5SDimitry Andric   SDValue HiLoad =
14320b57cec5SDimitry Andric       DAG.getExtLoad(Load->getExtensionType(), SL, HiVT, Load->getChain(),
14330b57cec5SDimitry Andric                      HiPtr, SrcValue.getWithOffset(LoMemVT.getStoreSize()),
14340b57cec5SDimitry Andric                      HiMemVT, HiAlign, Load->getMemOperand()->getFlags());
14350b57cec5SDimitry Andric 
14360b57cec5SDimitry Andric   auto IdxTy = getVectorIdxTy(DAG.getDataLayout());
14370b57cec5SDimitry Andric   SDValue Join;
14380b57cec5SDimitry Andric   if (LoVT == HiVT) {
14390b57cec5SDimitry Andric     // This is the case that the vector is power of two so was evenly split.
14400b57cec5SDimitry Andric     Join = DAG.getNode(ISD::CONCAT_VECTORS, SL, VT, LoLoad, HiLoad);
14410b57cec5SDimitry Andric   } else {
14420b57cec5SDimitry Andric     Join = DAG.getNode(ISD::INSERT_SUBVECTOR, SL, VT, DAG.getUNDEF(VT), LoLoad,
14430b57cec5SDimitry Andric                        DAG.getConstant(0, SL, IdxTy));
14440b57cec5SDimitry Andric     Join = DAG.getNode(HiVT.isVector() ? ISD::INSERT_SUBVECTOR
14450b57cec5SDimitry Andric                                        : ISD::INSERT_VECTOR_ELT,
14460b57cec5SDimitry Andric                        SL, VT, Join, HiLoad,
14470b57cec5SDimitry Andric                        DAG.getConstant(LoVT.getVectorNumElements(), SL, IdxTy));
14480b57cec5SDimitry Andric   }
14490b57cec5SDimitry Andric 
14500b57cec5SDimitry Andric   SDValue Ops[] = {Join, DAG.getNode(ISD::TokenFactor, SL, MVT::Other,
14510b57cec5SDimitry Andric                                      LoLoad.getValue(1), HiLoad.getValue(1))};
14520b57cec5SDimitry Andric 
14530b57cec5SDimitry Andric   return DAG.getMergeValues(Ops, SL);
14540b57cec5SDimitry Andric }
14550b57cec5SDimitry Andric 
14560b57cec5SDimitry Andric // Widen a vector load from vec3 to vec4.
14570b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::WidenVectorLoad(SDValue Op,
14580b57cec5SDimitry Andric                                               SelectionDAG &DAG) const {
14590b57cec5SDimitry Andric   LoadSDNode *Load = cast<LoadSDNode>(Op);
14600b57cec5SDimitry Andric   EVT VT = Op.getValueType();
14610b57cec5SDimitry Andric   assert(VT.getVectorNumElements() == 3);
14620b57cec5SDimitry Andric   SDValue BasePtr = Load->getBasePtr();
14630b57cec5SDimitry Andric   EVT MemVT = Load->getMemoryVT();
14640b57cec5SDimitry Andric   SDLoc SL(Op);
14650b57cec5SDimitry Andric   const MachinePointerInfo &SrcValue = Load->getMemOperand()->getPointerInfo();
14660b57cec5SDimitry Andric   unsigned BaseAlign = Load->getAlignment();
14670b57cec5SDimitry Andric 
14680b57cec5SDimitry Andric   EVT WideVT =
14690b57cec5SDimitry Andric       EVT::getVectorVT(*DAG.getContext(), VT.getVectorElementType(), 4);
14700b57cec5SDimitry Andric   EVT WideMemVT =
14710b57cec5SDimitry Andric       EVT::getVectorVT(*DAG.getContext(), MemVT.getVectorElementType(), 4);
14720b57cec5SDimitry Andric   SDValue WideLoad = DAG.getExtLoad(
14730b57cec5SDimitry Andric       Load->getExtensionType(), SL, WideVT, Load->getChain(), BasePtr, SrcValue,
14740b57cec5SDimitry Andric       WideMemVT, BaseAlign, Load->getMemOperand()->getFlags());
14750b57cec5SDimitry Andric   return DAG.getMergeValues(
14760b57cec5SDimitry Andric       {DAG.getNode(ISD::EXTRACT_SUBVECTOR, SL, VT, WideLoad,
14770b57cec5SDimitry Andric                    DAG.getConstant(0, SL, getVectorIdxTy(DAG.getDataLayout()))),
14780b57cec5SDimitry Andric        WideLoad.getValue(1)},
14790b57cec5SDimitry Andric       SL);
14800b57cec5SDimitry Andric }
14810b57cec5SDimitry Andric 
14820b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::SplitVectorStore(SDValue Op,
14830b57cec5SDimitry Andric                                                SelectionDAG &DAG) const {
14840b57cec5SDimitry Andric   StoreSDNode *Store = cast<StoreSDNode>(Op);
14850b57cec5SDimitry Andric   SDValue Val = Store->getValue();
14860b57cec5SDimitry Andric   EVT VT = Val.getValueType();
14870b57cec5SDimitry Andric 
14880b57cec5SDimitry Andric   // If this is a 2 element vector, we really want to scalarize and not create
14890b57cec5SDimitry Andric   // weird 1 element vectors.
14900b57cec5SDimitry Andric   if (VT.getVectorNumElements() == 2)
14910b57cec5SDimitry Andric     return scalarizeVectorStore(Store, DAG);
14920b57cec5SDimitry Andric 
14930b57cec5SDimitry Andric   EVT MemVT = Store->getMemoryVT();
14940b57cec5SDimitry Andric   SDValue Chain = Store->getChain();
14950b57cec5SDimitry Andric   SDValue BasePtr = Store->getBasePtr();
14960b57cec5SDimitry Andric   SDLoc SL(Op);
14970b57cec5SDimitry Andric 
14980b57cec5SDimitry Andric   EVT LoVT, HiVT;
14990b57cec5SDimitry Andric   EVT LoMemVT, HiMemVT;
15000b57cec5SDimitry Andric   SDValue Lo, Hi;
15010b57cec5SDimitry Andric 
15020b57cec5SDimitry Andric   std::tie(LoVT, HiVT) = getSplitDestVTs(VT, DAG);
15030b57cec5SDimitry Andric   std::tie(LoMemVT, HiMemVT) = getSplitDestVTs(MemVT, DAG);
15040b57cec5SDimitry Andric   std::tie(Lo, Hi) = splitVector(Val, SL, LoVT, HiVT, DAG);
15050b57cec5SDimitry Andric 
15060b57cec5SDimitry Andric   SDValue HiPtr = DAG.getObjectPtrOffset(SL, BasePtr, LoMemVT.getStoreSize());
15070b57cec5SDimitry Andric 
15080b57cec5SDimitry Andric   const MachinePointerInfo &SrcValue = Store->getMemOperand()->getPointerInfo();
15090b57cec5SDimitry Andric   unsigned BaseAlign = Store->getAlignment();
15100b57cec5SDimitry Andric   unsigned Size = LoMemVT.getStoreSize();
15110b57cec5SDimitry Andric   unsigned HiAlign = MinAlign(BaseAlign, Size);
15120b57cec5SDimitry Andric 
15130b57cec5SDimitry Andric   SDValue LoStore =
15140b57cec5SDimitry Andric       DAG.getTruncStore(Chain, SL, Lo, BasePtr, SrcValue, LoMemVT, BaseAlign,
15150b57cec5SDimitry Andric                         Store->getMemOperand()->getFlags());
15160b57cec5SDimitry Andric   SDValue HiStore =
15170b57cec5SDimitry Andric       DAG.getTruncStore(Chain, SL, Hi, HiPtr, SrcValue.getWithOffset(Size),
15180b57cec5SDimitry Andric                         HiMemVT, HiAlign, Store->getMemOperand()->getFlags());
15190b57cec5SDimitry Andric 
15200b57cec5SDimitry Andric   return DAG.getNode(ISD::TokenFactor, SL, MVT::Other, LoStore, HiStore);
15210b57cec5SDimitry Andric }
15220b57cec5SDimitry Andric 
15230b57cec5SDimitry Andric // This is a shortcut for integer division because we have fast i32<->f32
15240b57cec5SDimitry Andric // conversions, and fast f32 reciprocal instructions. The fractional part of a
15250b57cec5SDimitry Andric // float is enough to accurately represent up to a 24-bit signed integer.
15260b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerDIVREM24(SDValue Op, SelectionDAG &DAG,
15270b57cec5SDimitry Andric                                             bool Sign) const {
15280b57cec5SDimitry Andric   SDLoc DL(Op);
15290b57cec5SDimitry Andric   EVT VT = Op.getValueType();
15300b57cec5SDimitry Andric   SDValue LHS = Op.getOperand(0);
15310b57cec5SDimitry Andric   SDValue RHS = Op.getOperand(1);
15320b57cec5SDimitry Andric   MVT IntVT = MVT::i32;
15330b57cec5SDimitry Andric   MVT FltVT = MVT::f32;
15340b57cec5SDimitry Andric 
15350b57cec5SDimitry Andric   unsigned LHSSignBits = DAG.ComputeNumSignBits(LHS);
15360b57cec5SDimitry Andric   if (LHSSignBits < 9)
15370b57cec5SDimitry Andric     return SDValue();
15380b57cec5SDimitry Andric 
15390b57cec5SDimitry Andric   unsigned RHSSignBits = DAG.ComputeNumSignBits(RHS);
15400b57cec5SDimitry Andric   if (RHSSignBits < 9)
15410b57cec5SDimitry Andric     return SDValue();
15420b57cec5SDimitry Andric 
15430b57cec5SDimitry Andric   unsigned BitSize = VT.getSizeInBits();
15440b57cec5SDimitry Andric   unsigned SignBits = std::min(LHSSignBits, RHSSignBits);
15450b57cec5SDimitry Andric   unsigned DivBits = BitSize - SignBits;
15460b57cec5SDimitry Andric   if (Sign)
15470b57cec5SDimitry Andric     ++DivBits;
15480b57cec5SDimitry Andric 
15490b57cec5SDimitry Andric   ISD::NodeType ToFp = Sign ? ISD::SINT_TO_FP : ISD::UINT_TO_FP;
15500b57cec5SDimitry Andric   ISD::NodeType ToInt = Sign ? ISD::FP_TO_SINT : ISD::FP_TO_UINT;
15510b57cec5SDimitry Andric 
15520b57cec5SDimitry Andric   SDValue jq = DAG.getConstant(1, DL, IntVT);
15530b57cec5SDimitry Andric 
15540b57cec5SDimitry Andric   if (Sign) {
15550b57cec5SDimitry Andric     // char|short jq = ia ^ ib;
15560b57cec5SDimitry Andric     jq = DAG.getNode(ISD::XOR, DL, VT, LHS, RHS);
15570b57cec5SDimitry Andric 
15580b57cec5SDimitry Andric     // jq = jq >> (bitsize - 2)
15590b57cec5SDimitry Andric     jq = DAG.getNode(ISD::SRA, DL, VT, jq,
15600b57cec5SDimitry Andric                      DAG.getConstant(BitSize - 2, DL, VT));
15610b57cec5SDimitry Andric 
15620b57cec5SDimitry Andric     // jq = jq | 0x1
15630b57cec5SDimitry Andric     jq = DAG.getNode(ISD::OR, DL, VT, jq, DAG.getConstant(1, DL, VT));
15640b57cec5SDimitry Andric   }
15650b57cec5SDimitry Andric 
15660b57cec5SDimitry Andric   // int ia = (int)LHS;
15670b57cec5SDimitry Andric   SDValue ia = LHS;
15680b57cec5SDimitry Andric 
15690b57cec5SDimitry Andric   // int ib, (int)RHS;
15700b57cec5SDimitry Andric   SDValue ib = RHS;
15710b57cec5SDimitry Andric 
15720b57cec5SDimitry Andric   // float fa = (float)ia;
15730b57cec5SDimitry Andric   SDValue fa = DAG.getNode(ToFp, DL, FltVT, ia);
15740b57cec5SDimitry Andric 
15750b57cec5SDimitry Andric   // float fb = (float)ib;
15760b57cec5SDimitry Andric   SDValue fb = DAG.getNode(ToFp, DL, FltVT, ib);
15770b57cec5SDimitry Andric 
15780b57cec5SDimitry Andric   SDValue fq = DAG.getNode(ISD::FMUL, DL, FltVT,
15790b57cec5SDimitry Andric                            fa, DAG.getNode(AMDGPUISD::RCP, DL, FltVT, fb));
15800b57cec5SDimitry Andric 
15810b57cec5SDimitry Andric   // fq = trunc(fq);
15820b57cec5SDimitry Andric   fq = DAG.getNode(ISD::FTRUNC, DL, FltVT, fq);
15830b57cec5SDimitry Andric 
15840b57cec5SDimitry Andric   // float fqneg = -fq;
15850b57cec5SDimitry Andric   SDValue fqneg = DAG.getNode(ISD::FNEG, DL, FltVT, fq);
15860b57cec5SDimitry Andric 
15870b57cec5SDimitry Andric   // float fr = mad(fqneg, fb, fa);
15880b57cec5SDimitry Andric   unsigned OpCode = Subtarget->hasFP32Denormals() ?
15890b57cec5SDimitry Andric                     (unsigned)AMDGPUISD::FMAD_FTZ :
15900b57cec5SDimitry Andric                     (unsigned)ISD::FMAD;
15910b57cec5SDimitry Andric   SDValue fr = DAG.getNode(OpCode, DL, FltVT, fqneg, fb, fa);
15920b57cec5SDimitry Andric 
15930b57cec5SDimitry Andric   // int iq = (int)fq;
15940b57cec5SDimitry Andric   SDValue iq = DAG.getNode(ToInt, DL, IntVT, fq);
15950b57cec5SDimitry Andric 
15960b57cec5SDimitry Andric   // fr = fabs(fr);
15970b57cec5SDimitry Andric   fr = DAG.getNode(ISD::FABS, DL, FltVT, fr);
15980b57cec5SDimitry Andric 
15990b57cec5SDimitry Andric   // fb = fabs(fb);
16000b57cec5SDimitry Andric   fb = DAG.getNode(ISD::FABS, DL, FltVT, fb);
16010b57cec5SDimitry Andric 
16020b57cec5SDimitry Andric   EVT SetCCVT = getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), VT);
16030b57cec5SDimitry Andric 
16040b57cec5SDimitry Andric   // int cv = fr >= fb;
16050b57cec5SDimitry Andric   SDValue cv = DAG.getSetCC(DL, SetCCVT, fr, fb, ISD::SETOGE);
16060b57cec5SDimitry Andric 
16070b57cec5SDimitry Andric   // jq = (cv ? jq : 0);
16080b57cec5SDimitry Andric   jq = DAG.getNode(ISD::SELECT, DL, VT, cv, jq, DAG.getConstant(0, DL, VT));
16090b57cec5SDimitry Andric 
16100b57cec5SDimitry Andric   // dst = iq + jq;
16110b57cec5SDimitry Andric   SDValue Div = DAG.getNode(ISD::ADD, DL, VT, iq, jq);
16120b57cec5SDimitry Andric 
16130b57cec5SDimitry Andric   // Rem needs compensation, it's easier to recompute it
16140b57cec5SDimitry Andric   SDValue Rem = DAG.getNode(ISD::MUL, DL, VT, Div, RHS);
16150b57cec5SDimitry Andric   Rem = DAG.getNode(ISD::SUB, DL, VT, LHS, Rem);
16160b57cec5SDimitry Andric 
16170b57cec5SDimitry Andric   // Truncate to number of bits this divide really is.
16180b57cec5SDimitry Andric   if (Sign) {
16190b57cec5SDimitry Andric     SDValue InRegSize
16200b57cec5SDimitry Andric       = DAG.getValueType(EVT::getIntegerVT(*DAG.getContext(), DivBits));
16210b57cec5SDimitry Andric     Div = DAG.getNode(ISD::SIGN_EXTEND_INREG, DL, VT, Div, InRegSize);
16220b57cec5SDimitry Andric     Rem = DAG.getNode(ISD::SIGN_EXTEND_INREG, DL, VT, Rem, InRegSize);
16230b57cec5SDimitry Andric   } else {
16240b57cec5SDimitry Andric     SDValue TruncMask = DAG.getConstant((UINT64_C(1) << DivBits) - 1, DL, VT);
16250b57cec5SDimitry Andric     Div = DAG.getNode(ISD::AND, DL, VT, Div, TruncMask);
16260b57cec5SDimitry Andric     Rem = DAG.getNode(ISD::AND, DL, VT, Rem, TruncMask);
16270b57cec5SDimitry Andric   }
16280b57cec5SDimitry Andric 
16290b57cec5SDimitry Andric   return DAG.getMergeValues({ Div, Rem }, DL);
16300b57cec5SDimitry Andric }
16310b57cec5SDimitry Andric 
16320b57cec5SDimitry Andric void AMDGPUTargetLowering::LowerUDIVREM64(SDValue Op,
16330b57cec5SDimitry Andric                                       SelectionDAG &DAG,
16340b57cec5SDimitry Andric                                       SmallVectorImpl<SDValue> &Results) const {
16350b57cec5SDimitry Andric   SDLoc DL(Op);
16360b57cec5SDimitry Andric   EVT VT = Op.getValueType();
16370b57cec5SDimitry Andric 
16380b57cec5SDimitry Andric   assert(VT == MVT::i64 && "LowerUDIVREM64 expects an i64");
16390b57cec5SDimitry Andric 
16400b57cec5SDimitry Andric   EVT HalfVT = VT.getHalfSizedIntegerVT(*DAG.getContext());
16410b57cec5SDimitry Andric 
16420b57cec5SDimitry Andric   SDValue One = DAG.getConstant(1, DL, HalfVT);
16430b57cec5SDimitry Andric   SDValue Zero = DAG.getConstant(0, DL, HalfVT);
16440b57cec5SDimitry Andric 
16450b57cec5SDimitry Andric   //HiLo split
16460b57cec5SDimitry Andric   SDValue LHS = Op.getOperand(0);
16470b57cec5SDimitry Andric   SDValue LHS_Lo = DAG.getNode(ISD::EXTRACT_ELEMENT, DL, HalfVT, LHS, Zero);
16480b57cec5SDimitry Andric   SDValue LHS_Hi = DAG.getNode(ISD::EXTRACT_ELEMENT, DL, HalfVT, LHS, One);
16490b57cec5SDimitry Andric 
16500b57cec5SDimitry Andric   SDValue RHS = Op.getOperand(1);
16510b57cec5SDimitry Andric   SDValue RHS_Lo = DAG.getNode(ISD::EXTRACT_ELEMENT, DL, HalfVT, RHS, Zero);
16520b57cec5SDimitry Andric   SDValue RHS_Hi = DAG.getNode(ISD::EXTRACT_ELEMENT, DL, HalfVT, RHS, One);
16530b57cec5SDimitry Andric 
16540b57cec5SDimitry Andric   if (DAG.MaskedValueIsZero(RHS, APInt::getHighBitsSet(64, 32)) &&
16550b57cec5SDimitry Andric       DAG.MaskedValueIsZero(LHS, APInt::getHighBitsSet(64, 32))) {
16560b57cec5SDimitry Andric 
16570b57cec5SDimitry Andric     SDValue Res = DAG.getNode(ISD::UDIVREM, DL, DAG.getVTList(HalfVT, HalfVT),
16580b57cec5SDimitry Andric                               LHS_Lo, RHS_Lo);
16590b57cec5SDimitry Andric 
16600b57cec5SDimitry Andric     SDValue DIV = DAG.getBuildVector(MVT::v2i32, DL, {Res.getValue(0), Zero});
16610b57cec5SDimitry Andric     SDValue REM = DAG.getBuildVector(MVT::v2i32, DL, {Res.getValue(1), Zero});
16620b57cec5SDimitry Andric 
16630b57cec5SDimitry Andric     Results.push_back(DAG.getNode(ISD::BITCAST, DL, MVT::i64, DIV));
16640b57cec5SDimitry Andric     Results.push_back(DAG.getNode(ISD::BITCAST, DL, MVT::i64, REM));
16650b57cec5SDimitry Andric     return;
16660b57cec5SDimitry Andric   }
16670b57cec5SDimitry Andric 
16680b57cec5SDimitry Andric   if (isTypeLegal(MVT::i64)) {
16690b57cec5SDimitry Andric     // Compute denominator reciprocal.
16700b57cec5SDimitry Andric     unsigned FMAD = Subtarget->hasFP32Denormals() ?
16710b57cec5SDimitry Andric                     (unsigned)AMDGPUISD::FMAD_FTZ :
16720b57cec5SDimitry Andric                     (unsigned)ISD::FMAD;
16730b57cec5SDimitry Andric 
16740b57cec5SDimitry Andric     SDValue Cvt_Lo = DAG.getNode(ISD::UINT_TO_FP, DL, MVT::f32, RHS_Lo);
16750b57cec5SDimitry Andric     SDValue Cvt_Hi = DAG.getNode(ISD::UINT_TO_FP, DL, MVT::f32, RHS_Hi);
16760b57cec5SDimitry Andric     SDValue Mad1 = DAG.getNode(FMAD, DL, MVT::f32, Cvt_Hi,
16770b57cec5SDimitry Andric       DAG.getConstantFP(APInt(32, 0x4f800000).bitsToFloat(), DL, MVT::f32),
16780b57cec5SDimitry Andric       Cvt_Lo);
16790b57cec5SDimitry Andric     SDValue Rcp = DAG.getNode(AMDGPUISD::RCP, DL, MVT::f32, Mad1);
16800b57cec5SDimitry Andric     SDValue Mul1 = DAG.getNode(ISD::FMUL, DL, MVT::f32, Rcp,
16810b57cec5SDimitry Andric       DAG.getConstantFP(APInt(32, 0x5f7ffffc).bitsToFloat(), DL, MVT::f32));
16820b57cec5SDimitry Andric     SDValue Mul2 = DAG.getNode(ISD::FMUL, DL, MVT::f32, Mul1,
16830b57cec5SDimitry Andric       DAG.getConstantFP(APInt(32, 0x2f800000).bitsToFloat(), DL, MVT::f32));
16840b57cec5SDimitry Andric     SDValue Trunc = DAG.getNode(ISD::FTRUNC, DL, MVT::f32, Mul2);
16850b57cec5SDimitry Andric     SDValue Mad2 = DAG.getNode(FMAD, DL, MVT::f32, Trunc,
16860b57cec5SDimitry Andric       DAG.getConstantFP(APInt(32, 0xcf800000).bitsToFloat(), DL, MVT::f32),
16870b57cec5SDimitry Andric       Mul1);
16880b57cec5SDimitry Andric     SDValue Rcp_Lo = DAG.getNode(ISD::FP_TO_UINT, DL, HalfVT, Mad2);
16890b57cec5SDimitry Andric     SDValue Rcp_Hi = DAG.getNode(ISD::FP_TO_UINT, DL, HalfVT, Trunc);
16900b57cec5SDimitry Andric     SDValue Rcp64 = DAG.getBitcast(VT,
16910b57cec5SDimitry Andric                         DAG.getBuildVector(MVT::v2i32, DL, {Rcp_Lo, Rcp_Hi}));
16920b57cec5SDimitry Andric 
16930b57cec5SDimitry Andric     SDValue Zero64 = DAG.getConstant(0, DL, VT);
16940b57cec5SDimitry Andric     SDValue One64  = DAG.getConstant(1, DL, VT);
16950b57cec5SDimitry Andric     SDValue Zero1 = DAG.getConstant(0, DL, MVT::i1);
16960b57cec5SDimitry Andric     SDVTList HalfCarryVT = DAG.getVTList(HalfVT, MVT::i1);
16970b57cec5SDimitry Andric 
16980b57cec5SDimitry Andric     SDValue Neg_RHS = DAG.getNode(ISD::SUB, DL, VT, Zero64, RHS);
16990b57cec5SDimitry Andric     SDValue Mullo1 = DAG.getNode(ISD::MUL, DL, VT, Neg_RHS, Rcp64);
17000b57cec5SDimitry Andric     SDValue Mulhi1 = DAG.getNode(ISD::MULHU, DL, VT, Rcp64, Mullo1);
17010b57cec5SDimitry Andric     SDValue Mulhi1_Lo = DAG.getNode(ISD::EXTRACT_ELEMENT, DL, HalfVT, Mulhi1,
17020b57cec5SDimitry Andric                                     Zero);
17030b57cec5SDimitry Andric     SDValue Mulhi1_Hi = DAG.getNode(ISD::EXTRACT_ELEMENT, DL, HalfVT, Mulhi1,
17040b57cec5SDimitry Andric                                     One);
17050b57cec5SDimitry Andric 
17060b57cec5SDimitry Andric     SDValue Add1_Lo = DAG.getNode(ISD::ADDCARRY, DL, HalfCarryVT, Rcp_Lo,
17070b57cec5SDimitry Andric                                   Mulhi1_Lo, Zero1);
17080b57cec5SDimitry Andric     SDValue Add1_Hi = DAG.getNode(ISD::ADDCARRY, DL, HalfCarryVT, Rcp_Hi,
17090b57cec5SDimitry Andric                                   Mulhi1_Hi, Add1_Lo.getValue(1));
17100b57cec5SDimitry Andric     SDValue Add1_HiNc = DAG.getNode(ISD::ADD, DL, HalfVT, Rcp_Hi, Mulhi1_Hi);
17110b57cec5SDimitry Andric     SDValue Add1 = DAG.getBitcast(VT,
17120b57cec5SDimitry Andric                         DAG.getBuildVector(MVT::v2i32, DL, {Add1_Lo, Add1_Hi}));
17130b57cec5SDimitry Andric 
17140b57cec5SDimitry Andric     SDValue Mullo2 = DAG.getNode(ISD::MUL, DL, VT, Neg_RHS, Add1);
17150b57cec5SDimitry Andric     SDValue Mulhi2 = DAG.getNode(ISD::MULHU, DL, VT, Add1, Mullo2);
17160b57cec5SDimitry Andric     SDValue Mulhi2_Lo = DAG.getNode(ISD::EXTRACT_ELEMENT, DL, HalfVT, Mulhi2,
17170b57cec5SDimitry Andric                                     Zero);
17180b57cec5SDimitry Andric     SDValue Mulhi2_Hi = DAG.getNode(ISD::EXTRACT_ELEMENT, DL, HalfVT, Mulhi2,
17190b57cec5SDimitry Andric                                     One);
17200b57cec5SDimitry Andric 
17210b57cec5SDimitry Andric     SDValue Add2_Lo = DAG.getNode(ISD::ADDCARRY, DL, HalfCarryVT, Add1_Lo,
17220b57cec5SDimitry Andric                                   Mulhi2_Lo, Zero1);
17230b57cec5SDimitry Andric     SDValue Add2_HiC = DAG.getNode(ISD::ADDCARRY, DL, HalfCarryVT, Add1_HiNc,
17240b57cec5SDimitry Andric                                    Mulhi2_Hi, Add1_Lo.getValue(1));
17250b57cec5SDimitry Andric     SDValue Add2_Hi = DAG.getNode(ISD::ADDCARRY, DL, HalfCarryVT, Add2_HiC,
17260b57cec5SDimitry Andric                                   Zero, Add2_Lo.getValue(1));
17270b57cec5SDimitry Andric     SDValue Add2 = DAG.getBitcast(VT,
17280b57cec5SDimitry Andric                         DAG.getBuildVector(MVT::v2i32, DL, {Add2_Lo, Add2_Hi}));
17290b57cec5SDimitry Andric     SDValue Mulhi3 = DAG.getNode(ISD::MULHU, DL, VT, LHS, Add2);
17300b57cec5SDimitry Andric 
17310b57cec5SDimitry Andric     SDValue Mul3 = DAG.getNode(ISD::MUL, DL, VT, RHS, Mulhi3);
17320b57cec5SDimitry Andric 
17330b57cec5SDimitry Andric     SDValue Mul3_Lo = DAG.getNode(ISD::EXTRACT_ELEMENT, DL, HalfVT, Mul3, Zero);
17340b57cec5SDimitry Andric     SDValue Mul3_Hi = DAG.getNode(ISD::EXTRACT_ELEMENT, DL, HalfVT, Mul3, One);
17350b57cec5SDimitry Andric     SDValue Sub1_Lo = DAG.getNode(ISD::SUBCARRY, DL, HalfCarryVT, LHS_Lo,
17360b57cec5SDimitry Andric                                   Mul3_Lo, Zero1);
17370b57cec5SDimitry Andric     SDValue Sub1_Hi = DAG.getNode(ISD::SUBCARRY, DL, HalfCarryVT, LHS_Hi,
17380b57cec5SDimitry Andric                                   Mul3_Hi, Sub1_Lo.getValue(1));
17390b57cec5SDimitry Andric     SDValue Sub1_Mi = DAG.getNode(ISD::SUB, DL, HalfVT, LHS_Hi, Mul3_Hi);
17400b57cec5SDimitry Andric     SDValue Sub1 = DAG.getBitcast(VT,
17410b57cec5SDimitry Andric                         DAG.getBuildVector(MVT::v2i32, DL, {Sub1_Lo, Sub1_Hi}));
17420b57cec5SDimitry Andric 
17430b57cec5SDimitry Andric     SDValue MinusOne = DAG.getConstant(0xffffffffu, DL, HalfVT);
17440b57cec5SDimitry Andric     SDValue C1 = DAG.getSelectCC(DL, Sub1_Hi, RHS_Hi, MinusOne, Zero,
17450b57cec5SDimitry Andric                                  ISD::SETUGE);
17460b57cec5SDimitry Andric     SDValue C2 = DAG.getSelectCC(DL, Sub1_Lo, RHS_Lo, MinusOne, Zero,
17470b57cec5SDimitry Andric                                  ISD::SETUGE);
17480b57cec5SDimitry Andric     SDValue C3 = DAG.getSelectCC(DL, Sub1_Hi, RHS_Hi, C2, C1, ISD::SETEQ);
17490b57cec5SDimitry Andric 
17500b57cec5SDimitry Andric     // TODO: Here and below portions of the code can be enclosed into if/endif.
17510b57cec5SDimitry Andric     // Currently control flow is unconditional and we have 4 selects after
17520b57cec5SDimitry Andric     // potential endif to substitute PHIs.
17530b57cec5SDimitry Andric 
17540b57cec5SDimitry Andric     // if C3 != 0 ...
17550b57cec5SDimitry Andric     SDValue Sub2_Lo = DAG.getNode(ISD::SUBCARRY, DL, HalfCarryVT, Sub1_Lo,
17560b57cec5SDimitry Andric                                   RHS_Lo, Zero1);
17570b57cec5SDimitry Andric     SDValue Sub2_Mi = DAG.getNode(ISD::SUBCARRY, DL, HalfCarryVT, Sub1_Mi,
17580b57cec5SDimitry Andric                                   RHS_Hi, Sub1_Lo.getValue(1));
17590b57cec5SDimitry Andric     SDValue Sub2_Hi = DAG.getNode(ISD::SUBCARRY, DL, HalfCarryVT, Sub2_Mi,
17600b57cec5SDimitry Andric                                   Zero, Sub2_Lo.getValue(1));
17610b57cec5SDimitry Andric     SDValue Sub2 = DAG.getBitcast(VT,
17620b57cec5SDimitry Andric                         DAG.getBuildVector(MVT::v2i32, DL, {Sub2_Lo, Sub2_Hi}));
17630b57cec5SDimitry Andric 
17640b57cec5SDimitry Andric     SDValue Add3 = DAG.getNode(ISD::ADD, DL, VT, Mulhi3, One64);
17650b57cec5SDimitry Andric 
17660b57cec5SDimitry Andric     SDValue C4 = DAG.getSelectCC(DL, Sub2_Hi, RHS_Hi, MinusOne, Zero,
17670b57cec5SDimitry Andric                                  ISD::SETUGE);
17680b57cec5SDimitry Andric     SDValue C5 = DAG.getSelectCC(DL, Sub2_Lo, RHS_Lo, MinusOne, Zero,
17690b57cec5SDimitry Andric                                  ISD::SETUGE);
17700b57cec5SDimitry Andric     SDValue C6 = DAG.getSelectCC(DL, Sub2_Hi, RHS_Hi, C5, C4, ISD::SETEQ);
17710b57cec5SDimitry Andric 
17720b57cec5SDimitry Andric     // if (C6 != 0)
17730b57cec5SDimitry Andric     SDValue Add4 = DAG.getNode(ISD::ADD, DL, VT, Add3, One64);
17740b57cec5SDimitry Andric 
17750b57cec5SDimitry Andric     SDValue Sub3_Lo = DAG.getNode(ISD::SUBCARRY, DL, HalfCarryVT, Sub2_Lo,
17760b57cec5SDimitry Andric                                   RHS_Lo, Zero1);
17770b57cec5SDimitry Andric     SDValue Sub3_Mi = DAG.getNode(ISD::SUBCARRY, DL, HalfCarryVT, Sub2_Mi,
17780b57cec5SDimitry Andric                                   RHS_Hi, Sub2_Lo.getValue(1));
17790b57cec5SDimitry Andric     SDValue Sub3_Hi = DAG.getNode(ISD::SUBCARRY, DL, HalfCarryVT, Sub3_Mi,
17800b57cec5SDimitry Andric                                   Zero, Sub3_Lo.getValue(1));
17810b57cec5SDimitry Andric     SDValue Sub3 = DAG.getBitcast(VT,
17820b57cec5SDimitry Andric                         DAG.getBuildVector(MVT::v2i32, DL, {Sub3_Lo, Sub3_Hi}));
17830b57cec5SDimitry Andric 
17840b57cec5SDimitry Andric     // endif C6
17850b57cec5SDimitry Andric     // endif C3
17860b57cec5SDimitry Andric 
17870b57cec5SDimitry Andric     SDValue Sel1 = DAG.getSelectCC(DL, C6, Zero, Add4, Add3, ISD::SETNE);
17880b57cec5SDimitry Andric     SDValue Div  = DAG.getSelectCC(DL, C3, Zero, Sel1, Mulhi3, ISD::SETNE);
17890b57cec5SDimitry Andric 
17900b57cec5SDimitry Andric     SDValue Sel2 = DAG.getSelectCC(DL, C6, Zero, Sub3, Sub2, ISD::SETNE);
17910b57cec5SDimitry Andric     SDValue Rem  = DAG.getSelectCC(DL, C3, Zero, Sel2, Sub1, ISD::SETNE);
17920b57cec5SDimitry Andric 
17930b57cec5SDimitry Andric     Results.push_back(Div);
17940b57cec5SDimitry Andric     Results.push_back(Rem);
17950b57cec5SDimitry Andric 
17960b57cec5SDimitry Andric     return;
17970b57cec5SDimitry Andric   }
17980b57cec5SDimitry Andric 
17990b57cec5SDimitry Andric   // r600 expandion.
18000b57cec5SDimitry Andric   // Get Speculative values
18010b57cec5SDimitry Andric   SDValue DIV_Part = DAG.getNode(ISD::UDIV, DL, HalfVT, LHS_Hi, RHS_Lo);
18020b57cec5SDimitry Andric   SDValue REM_Part = DAG.getNode(ISD::UREM, DL, HalfVT, LHS_Hi, RHS_Lo);
18030b57cec5SDimitry Andric 
18040b57cec5SDimitry Andric   SDValue REM_Lo = DAG.getSelectCC(DL, RHS_Hi, Zero, REM_Part, LHS_Hi, ISD::SETEQ);
18050b57cec5SDimitry Andric   SDValue REM = DAG.getBuildVector(MVT::v2i32, DL, {REM_Lo, Zero});
18060b57cec5SDimitry Andric   REM = DAG.getNode(ISD::BITCAST, DL, MVT::i64, REM);
18070b57cec5SDimitry Andric 
18080b57cec5SDimitry Andric   SDValue DIV_Hi = DAG.getSelectCC(DL, RHS_Hi, Zero, DIV_Part, Zero, ISD::SETEQ);
18090b57cec5SDimitry Andric   SDValue DIV_Lo = Zero;
18100b57cec5SDimitry Andric 
18110b57cec5SDimitry Andric   const unsigned halfBitWidth = HalfVT.getSizeInBits();
18120b57cec5SDimitry Andric 
18130b57cec5SDimitry Andric   for (unsigned i = 0; i < halfBitWidth; ++i) {
18140b57cec5SDimitry Andric     const unsigned bitPos = halfBitWidth - i - 1;
18150b57cec5SDimitry Andric     SDValue POS = DAG.getConstant(bitPos, DL, HalfVT);
18160b57cec5SDimitry Andric     // Get value of high bit
18170b57cec5SDimitry Andric     SDValue HBit = DAG.getNode(ISD::SRL, DL, HalfVT, LHS_Lo, POS);
18180b57cec5SDimitry Andric     HBit = DAG.getNode(ISD::AND, DL, HalfVT, HBit, One);
18190b57cec5SDimitry Andric     HBit = DAG.getNode(ISD::ZERO_EXTEND, DL, VT, HBit);
18200b57cec5SDimitry Andric 
18210b57cec5SDimitry Andric     // Shift
18220b57cec5SDimitry Andric     REM = DAG.getNode(ISD::SHL, DL, VT, REM, DAG.getConstant(1, DL, VT));
18230b57cec5SDimitry Andric     // Add LHS high bit
18240b57cec5SDimitry Andric     REM = DAG.getNode(ISD::OR, DL, VT, REM, HBit);
18250b57cec5SDimitry Andric 
18260b57cec5SDimitry Andric     SDValue BIT = DAG.getConstant(1ULL << bitPos, DL, HalfVT);
18270b57cec5SDimitry Andric     SDValue realBIT = DAG.getSelectCC(DL, REM, RHS, BIT, Zero, ISD::SETUGE);
18280b57cec5SDimitry Andric 
18290b57cec5SDimitry Andric     DIV_Lo = DAG.getNode(ISD::OR, DL, HalfVT, DIV_Lo, realBIT);
18300b57cec5SDimitry Andric 
18310b57cec5SDimitry Andric     // Update REM
18320b57cec5SDimitry Andric     SDValue REM_sub = DAG.getNode(ISD::SUB, DL, VT, REM, RHS);
18330b57cec5SDimitry Andric     REM = DAG.getSelectCC(DL, REM, RHS, REM_sub, REM, ISD::SETUGE);
18340b57cec5SDimitry Andric   }
18350b57cec5SDimitry Andric 
18360b57cec5SDimitry Andric   SDValue DIV = DAG.getBuildVector(MVT::v2i32, DL, {DIV_Lo, DIV_Hi});
18370b57cec5SDimitry Andric   DIV = DAG.getNode(ISD::BITCAST, DL, MVT::i64, DIV);
18380b57cec5SDimitry Andric   Results.push_back(DIV);
18390b57cec5SDimitry Andric   Results.push_back(REM);
18400b57cec5SDimitry Andric }
18410b57cec5SDimitry Andric 
18420b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerUDIVREM(SDValue Op,
18430b57cec5SDimitry Andric                                            SelectionDAG &DAG) const {
18440b57cec5SDimitry Andric   SDLoc DL(Op);
18450b57cec5SDimitry Andric   EVT VT = Op.getValueType();
18460b57cec5SDimitry Andric 
18470b57cec5SDimitry Andric   if (VT == MVT::i64) {
18480b57cec5SDimitry Andric     SmallVector<SDValue, 2> Results;
18490b57cec5SDimitry Andric     LowerUDIVREM64(Op, DAG, Results);
18500b57cec5SDimitry Andric     return DAG.getMergeValues(Results, DL);
18510b57cec5SDimitry Andric   }
18520b57cec5SDimitry Andric 
18530b57cec5SDimitry Andric   if (VT == MVT::i32) {
18540b57cec5SDimitry Andric     if (SDValue Res = LowerDIVREM24(Op, DAG, false))
18550b57cec5SDimitry Andric       return Res;
18560b57cec5SDimitry Andric   }
18570b57cec5SDimitry Andric 
18580b57cec5SDimitry Andric   SDValue Num = Op.getOperand(0);
18590b57cec5SDimitry Andric   SDValue Den = Op.getOperand(1);
18600b57cec5SDimitry Andric 
18610b57cec5SDimitry Andric   // RCP =  URECIP(Den) = 2^32 / Den + e
18620b57cec5SDimitry Andric   // e is rounding error.
18630b57cec5SDimitry Andric   SDValue RCP = DAG.getNode(AMDGPUISD::URECIP, DL, VT, Den);
18640b57cec5SDimitry Andric 
18650b57cec5SDimitry Andric   // RCP_LO = mul(RCP, Den) */
18660b57cec5SDimitry Andric   SDValue RCP_LO = DAG.getNode(ISD::MUL, DL, VT, RCP, Den);
18670b57cec5SDimitry Andric 
18680b57cec5SDimitry Andric   // RCP_HI = mulhu (RCP, Den) */
18690b57cec5SDimitry Andric   SDValue RCP_HI = DAG.getNode(ISD::MULHU, DL, VT, RCP, Den);
18700b57cec5SDimitry Andric 
18710b57cec5SDimitry Andric   // NEG_RCP_LO = -RCP_LO
18720b57cec5SDimitry Andric   SDValue NEG_RCP_LO = DAG.getNode(ISD::SUB, DL, VT, DAG.getConstant(0, DL, VT),
18730b57cec5SDimitry Andric                                                      RCP_LO);
18740b57cec5SDimitry Andric 
18750b57cec5SDimitry Andric   // ABS_RCP_LO = (RCP_HI == 0 ? NEG_RCP_LO : RCP_LO)
18760b57cec5SDimitry Andric   SDValue ABS_RCP_LO = DAG.getSelectCC(DL, RCP_HI, DAG.getConstant(0, DL, VT),
18770b57cec5SDimitry Andric                                            NEG_RCP_LO, RCP_LO,
18780b57cec5SDimitry Andric                                            ISD::SETEQ);
18790b57cec5SDimitry Andric   // Calculate the rounding error from the URECIP instruction
18800b57cec5SDimitry Andric   // E = mulhu(ABS_RCP_LO, RCP)
18810b57cec5SDimitry Andric   SDValue E = DAG.getNode(ISD::MULHU, DL, VT, ABS_RCP_LO, RCP);
18820b57cec5SDimitry Andric 
18830b57cec5SDimitry Andric   // RCP_A_E = RCP + E
18840b57cec5SDimitry Andric   SDValue RCP_A_E = DAG.getNode(ISD::ADD, DL, VT, RCP, E);
18850b57cec5SDimitry Andric 
18860b57cec5SDimitry Andric   // RCP_S_E = RCP - E
18870b57cec5SDimitry Andric   SDValue RCP_S_E = DAG.getNode(ISD::SUB, DL, VT, RCP, E);
18880b57cec5SDimitry Andric 
18890b57cec5SDimitry Andric   // Tmp0 = (RCP_HI == 0 ? RCP_A_E : RCP_SUB_E)
18900b57cec5SDimitry Andric   SDValue Tmp0 = DAG.getSelectCC(DL, RCP_HI, DAG.getConstant(0, DL, VT),
18910b57cec5SDimitry Andric                                      RCP_A_E, RCP_S_E,
18920b57cec5SDimitry Andric                                      ISD::SETEQ);
18930b57cec5SDimitry Andric   // Quotient = mulhu(Tmp0, Num)
18940b57cec5SDimitry Andric   SDValue Quotient = DAG.getNode(ISD::MULHU, DL, VT, Tmp0, Num);
18950b57cec5SDimitry Andric 
18960b57cec5SDimitry Andric   // Num_S_Remainder = Quotient * Den
18970b57cec5SDimitry Andric   SDValue Num_S_Remainder = DAG.getNode(ISD::MUL, DL, VT, Quotient, Den);
18980b57cec5SDimitry Andric 
18990b57cec5SDimitry Andric   // Remainder = Num - Num_S_Remainder
19000b57cec5SDimitry Andric   SDValue Remainder = DAG.getNode(ISD::SUB, DL, VT, Num, Num_S_Remainder);
19010b57cec5SDimitry Andric 
19020b57cec5SDimitry Andric   // Remainder_GE_Den = (Remainder >= Den ? -1 : 0)
19030b57cec5SDimitry Andric   SDValue Remainder_GE_Den = DAG.getSelectCC(DL, Remainder, Den,
19040b57cec5SDimitry Andric                                                  DAG.getConstant(-1, DL, VT),
19050b57cec5SDimitry Andric                                                  DAG.getConstant(0, DL, VT),
19060b57cec5SDimitry Andric                                                  ISD::SETUGE);
19070b57cec5SDimitry Andric   // Remainder_GE_Zero = (Num >= Num_S_Remainder ? -1 : 0)
19080b57cec5SDimitry Andric   SDValue Remainder_GE_Zero = DAG.getSelectCC(DL, Num,
19090b57cec5SDimitry Andric                                                   Num_S_Remainder,
19100b57cec5SDimitry Andric                                                   DAG.getConstant(-1, DL, VT),
19110b57cec5SDimitry Andric                                                   DAG.getConstant(0, DL, VT),
19120b57cec5SDimitry Andric                                                   ISD::SETUGE);
19130b57cec5SDimitry Andric   // Tmp1 = Remainder_GE_Den & Remainder_GE_Zero
19140b57cec5SDimitry Andric   SDValue Tmp1 = DAG.getNode(ISD::AND, DL, VT, Remainder_GE_Den,
19150b57cec5SDimitry Andric                                                Remainder_GE_Zero);
19160b57cec5SDimitry Andric 
19170b57cec5SDimitry Andric   // Calculate Division result:
19180b57cec5SDimitry Andric 
19190b57cec5SDimitry Andric   // Quotient_A_One = Quotient + 1
19200b57cec5SDimitry Andric   SDValue Quotient_A_One = DAG.getNode(ISD::ADD, DL, VT, Quotient,
19210b57cec5SDimitry Andric                                        DAG.getConstant(1, DL, VT));
19220b57cec5SDimitry Andric 
19230b57cec5SDimitry Andric   // Quotient_S_One = Quotient - 1
19240b57cec5SDimitry Andric   SDValue Quotient_S_One = DAG.getNode(ISD::SUB, DL, VT, Quotient,
19250b57cec5SDimitry Andric                                        DAG.getConstant(1, DL, VT));
19260b57cec5SDimitry Andric 
19270b57cec5SDimitry Andric   // Div = (Tmp1 == 0 ? Quotient : Quotient_A_One)
19280b57cec5SDimitry Andric   SDValue Div = DAG.getSelectCC(DL, Tmp1, DAG.getConstant(0, DL, VT),
19290b57cec5SDimitry Andric                                      Quotient, Quotient_A_One, ISD::SETEQ);
19300b57cec5SDimitry Andric 
19310b57cec5SDimitry Andric   // Div = (Remainder_GE_Zero == 0 ? Quotient_S_One : Div)
19320b57cec5SDimitry Andric   Div = DAG.getSelectCC(DL, Remainder_GE_Zero, DAG.getConstant(0, DL, VT),
19330b57cec5SDimitry Andric                             Quotient_S_One, Div, ISD::SETEQ);
19340b57cec5SDimitry Andric 
19350b57cec5SDimitry Andric   // Calculate Rem result:
19360b57cec5SDimitry Andric 
19370b57cec5SDimitry Andric   // Remainder_S_Den = Remainder - Den
19380b57cec5SDimitry Andric   SDValue Remainder_S_Den = DAG.getNode(ISD::SUB, DL, VT, Remainder, Den);
19390b57cec5SDimitry Andric 
19400b57cec5SDimitry Andric   // Remainder_A_Den = Remainder + Den
19410b57cec5SDimitry Andric   SDValue Remainder_A_Den = DAG.getNode(ISD::ADD, DL, VT, Remainder, Den);
19420b57cec5SDimitry Andric 
19430b57cec5SDimitry Andric   // Rem = (Tmp1 == 0 ? Remainder : Remainder_S_Den)
19440b57cec5SDimitry Andric   SDValue Rem = DAG.getSelectCC(DL, Tmp1, DAG.getConstant(0, DL, VT),
19450b57cec5SDimitry Andric                                     Remainder, Remainder_S_Den, ISD::SETEQ);
19460b57cec5SDimitry Andric 
19470b57cec5SDimitry Andric   // Rem = (Remainder_GE_Zero == 0 ? Remainder_A_Den : Rem)
19480b57cec5SDimitry Andric   Rem = DAG.getSelectCC(DL, Remainder_GE_Zero, DAG.getConstant(0, DL, VT),
19490b57cec5SDimitry Andric                             Remainder_A_Den, Rem, ISD::SETEQ);
19500b57cec5SDimitry Andric   SDValue Ops[2] = {
19510b57cec5SDimitry Andric     Div,
19520b57cec5SDimitry Andric     Rem
19530b57cec5SDimitry Andric   };
19540b57cec5SDimitry Andric   return DAG.getMergeValues(Ops, DL);
19550b57cec5SDimitry Andric }
19560b57cec5SDimitry Andric 
19570b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerSDIVREM(SDValue Op,
19580b57cec5SDimitry Andric                                            SelectionDAG &DAG) const {
19590b57cec5SDimitry Andric   SDLoc DL(Op);
19600b57cec5SDimitry Andric   EVT VT = Op.getValueType();
19610b57cec5SDimitry Andric 
19620b57cec5SDimitry Andric   SDValue LHS = Op.getOperand(0);
19630b57cec5SDimitry Andric   SDValue RHS = Op.getOperand(1);
19640b57cec5SDimitry Andric 
19650b57cec5SDimitry Andric   SDValue Zero = DAG.getConstant(0, DL, VT);
19660b57cec5SDimitry Andric   SDValue NegOne = DAG.getConstant(-1, DL, VT);
19670b57cec5SDimitry Andric 
19680b57cec5SDimitry Andric   if (VT == MVT::i32) {
19690b57cec5SDimitry Andric     if (SDValue Res = LowerDIVREM24(Op, DAG, true))
19700b57cec5SDimitry Andric       return Res;
19710b57cec5SDimitry Andric   }
19720b57cec5SDimitry Andric 
19730b57cec5SDimitry Andric   if (VT == MVT::i64 &&
19740b57cec5SDimitry Andric       DAG.ComputeNumSignBits(LHS) > 32 &&
19750b57cec5SDimitry Andric       DAG.ComputeNumSignBits(RHS) > 32) {
19760b57cec5SDimitry Andric     EVT HalfVT = VT.getHalfSizedIntegerVT(*DAG.getContext());
19770b57cec5SDimitry Andric 
19780b57cec5SDimitry Andric     //HiLo split
19790b57cec5SDimitry Andric     SDValue LHS_Lo = DAG.getNode(ISD::EXTRACT_ELEMENT, DL, HalfVT, LHS, Zero);
19800b57cec5SDimitry Andric     SDValue RHS_Lo = DAG.getNode(ISD::EXTRACT_ELEMENT, DL, HalfVT, RHS, Zero);
19810b57cec5SDimitry Andric     SDValue DIVREM = DAG.getNode(ISD::SDIVREM, DL, DAG.getVTList(HalfVT, HalfVT),
19820b57cec5SDimitry Andric                                  LHS_Lo, RHS_Lo);
19830b57cec5SDimitry Andric     SDValue Res[2] = {
19840b57cec5SDimitry Andric       DAG.getNode(ISD::SIGN_EXTEND, DL, VT, DIVREM.getValue(0)),
19850b57cec5SDimitry Andric       DAG.getNode(ISD::SIGN_EXTEND, DL, VT, DIVREM.getValue(1))
19860b57cec5SDimitry Andric     };
19870b57cec5SDimitry Andric     return DAG.getMergeValues(Res, DL);
19880b57cec5SDimitry Andric   }
19890b57cec5SDimitry Andric 
19900b57cec5SDimitry Andric   SDValue LHSign = DAG.getSelectCC(DL, LHS, Zero, NegOne, Zero, ISD::SETLT);
19910b57cec5SDimitry Andric   SDValue RHSign = DAG.getSelectCC(DL, RHS, Zero, NegOne, Zero, ISD::SETLT);
19920b57cec5SDimitry Andric   SDValue DSign = DAG.getNode(ISD::XOR, DL, VT, LHSign, RHSign);
19930b57cec5SDimitry Andric   SDValue RSign = LHSign; // Remainder sign is the same as LHS
19940b57cec5SDimitry Andric 
19950b57cec5SDimitry Andric   LHS = DAG.getNode(ISD::ADD, DL, VT, LHS, LHSign);
19960b57cec5SDimitry Andric   RHS = DAG.getNode(ISD::ADD, DL, VT, RHS, RHSign);
19970b57cec5SDimitry Andric 
19980b57cec5SDimitry Andric   LHS = DAG.getNode(ISD::XOR, DL, VT, LHS, LHSign);
19990b57cec5SDimitry Andric   RHS = DAG.getNode(ISD::XOR, DL, VT, RHS, RHSign);
20000b57cec5SDimitry Andric 
20010b57cec5SDimitry Andric   SDValue Div = DAG.getNode(ISD::UDIVREM, DL, DAG.getVTList(VT, VT), LHS, RHS);
20020b57cec5SDimitry Andric   SDValue Rem = Div.getValue(1);
20030b57cec5SDimitry Andric 
20040b57cec5SDimitry Andric   Div = DAG.getNode(ISD::XOR, DL, VT, Div, DSign);
20050b57cec5SDimitry Andric   Rem = DAG.getNode(ISD::XOR, DL, VT, Rem, RSign);
20060b57cec5SDimitry Andric 
20070b57cec5SDimitry Andric   Div = DAG.getNode(ISD::SUB, DL, VT, Div, DSign);
20080b57cec5SDimitry Andric   Rem = DAG.getNode(ISD::SUB, DL, VT, Rem, RSign);
20090b57cec5SDimitry Andric 
20100b57cec5SDimitry Andric   SDValue Res[2] = {
20110b57cec5SDimitry Andric     Div,
20120b57cec5SDimitry Andric     Rem
20130b57cec5SDimitry Andric   };
20140b57cec5SDimitry Andric   return DAG.getMergeValues(Res, DL);
20150b57cec5SDimitry Andric }
20160b57cec5SDimitry Andric 
20170b57cec5SDimitry Andric // (frem x, y) -> (fsub x, (fmul (ftrunc (fdiv x, y)), y))
20180b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFREM(SDValue Op, SelectionDAG &DAG) const {
20190b57cec5SDimitry Andric   SDLoc SL(Op);
20200b57cec5SDimitry Andric   EVT VT = Op.getValueType();
20210b57cec5SDimitry Andric   SDValue X = Op.getOperand(0);
20220b57cec5SDimitry Andric   SDValue Y = Op.getOperand(1);
20230b57cec5SDimitry Andric 
20240b57cec5SDimitry Andric   // TODO: Should this propagate fast-math-flags?
20250b57cec5SDimitry Andric 
20260b57cec5SDimitry Andric   SDValue Div = DAG.getNode(ISD::FDIV, SL, VT, X, Y);
20270b57cec5SDimitry Andric   SDValue Floor = DAG.getNode(ISD::FTRUNC, SL, VT, Div);
20280b57cec5SDimitry Andric   SDValue Mul = DAG.getNode(ISD::FMUL, SL, VT, Floor, Y);
20290b57cec5SDimitry Andric 
20300b57cec5SDimitry Andric   return DAG.getNode(ISD::FSUB, SL, VT, X, Mul);
20310b57cec5SDimitry Andric }
20320b57cec5SDimitry Andric 
20330b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFCEIL(SDValue Op, SelectionDAG &DAG) const {
20340b57cec5SDimitry Andric   SDLoc SL(Op);
20350b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
20360b57cec5SDimitry Andric 
20370b57cec5SDimitry Andric   // result = trunc(src)
20380b57cec5SDimitry Andric   // if (src > 0.0 && src != result)
20390b57cec5SDimitry Andric   //   result += 1.0
20400b57cec5SDimitry Andric 
20410b57cec5SDimitry Andric   SDValue Trunc = DAG.getNode(ISD::FTRUNC, SL, MVT::f64, Src);
20420b57cec5SDimitry Andric 
20430b57cec5SDimitry Andric   const SDValue Zero = DAG.getConstantFP(0.0, SL, MVT::f64);
20440b57cec5SDimitry Andric   const SDValue One = DAG.getConstantFP(1.0, SL, MVT::f64);
20450b57cec5SDimitry Andric 
20460b57cec5SDimitry Andric   EVT SetCCVT =
20470b57cec5SDimitry Andric       getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), MVT::f64);
20480b57cec5SDimitry Andric 
20490b57cec5SDimitry Andric   SDValue Lt0 = DAG.getSetCC(SL, SetCCVT, Src, Zero, ISD::SETOGT);
20500b57cec5SDimitry Andric   SDValue NeTrunc = DAG.getSetCC(SL, SetCCVT, Src, Trunc, ISD::SETONE);
20510b57cec5SDimitry Andric   SDValue And = DAG.getNode(ISD::AND, SL, SetCCVT, Lt0, NeTrunc);
20520b57cec5SDimitry Andric 
20530b57cec5SDimitry Andric   SDValue Add = DAG.getNode(ISD::SELECT, SL, MVT::f64, And, One, Zero);
20540b57cec5SDimitry Andric   // TODO: Should this propagate fast-math-flags?
20550b57cec5SDimitry Andric   return DAG.getNode(ISD::FADD, SL, MVT::f64, Trunc, Add);
20560b57cec5SDimitry Andric }
20570b57cec5SDimitry Andric 
20580b57cec5SDimitry Andric static SDValue extractF64Exponent(SDValue Hi, const SDLoc &SL,
20590b57cec5SDimitry Andric                                   SelectionDAG &DAG) {
20600b57cec5SDimitry Andric   const unsigned FractBits = 52;
20610b57cec5SDimitry Andric   const unsigned ExpBits = 11;
20620b57cec5SDimitry Andric 
20630b57cec5SDimitry Andric   SDValue ExpPart = DAG.getNode(AMDGPUISD::BFE_U32, SL, MVT::i32,
20640b57cec5SDimitry Andric                                 Hi,
20650b57cec5SDimitry Andric                                 DAG.getConstant(FractBits - 32, SL, MVT::i32),
20660b57cec5SDimitry Andric                                 DAG.getConstant(ExpBits, SL, MVT::i32));
20670b57cec5SDimitry Andric   SDValue Exp = DAG.getNode(ISD::SUB, SL, MVT::i32, ExpPart,
20680b57cec5SDimitry Andric                             DAG.getConstant(1023, SL, MVT::i32));
20690b57cec5SDimitry Andric 
20700b57cec5SDimitry Andric   return Exp;
20710b57cec5SDimitry Andric }
20720b57cec5SDimitry Andric 
20730b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFTRUNC(SDValue Op, SelectionDAG &DAG) const {
20740b57cec5SDimitry Andric   SDLoc SL(Op);
20750b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
20760b57cec5SDimitry Andric 
20770b57cec5SDimitry Andric   assert(Op.getValueType() == MVT::f64);
20780b57cec5SDimitry Andric 
20790b57cec5SDimitry Andric   const SDValue Zero = DAG.getConstant(0, SL, MVT::i32);
20800b57cec5SDimitry Andric   const SDValue One = DAG.getConstant(1, SL, MVT::i32);
20810b57cec5SDimitry Andric 
20820b57cec5SDimitry Andric   SDValue VecSrc = DAG.getNode(ISD::BITCAST, SL, MVT::v2i32, Src);
20830b57cec5SDimitry Andric 
20840b57cec5SDimitry Andric   // Extract the upper half, since this is where we will find the sign and
20850b57cec5SDimitry Andric   // exponent.
20860b57cec5SDimitry Andric   SDValue Hi = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, MVT::i32, VecSrc, One);
20870b57cec5SDimitry Andric 
20880b57cec5SDimitry Andric   SDValue Exp = extractF64Exponent(Hi, SL, DAG);
20890b57cec5SDimitry Andric 
20900b57cec5SDimitry Andric   const unsigned FractBits = 52;
20910b57cec5SDimitry Andric 
20920b57cec5SDimitry Andric   // Extract the sign bit.
20930b57cec5SDimitry Andric   const SDValue SignBitMask = DAG.getConstant(UINT32_C(1) << 31, SL, MVT::i32);
20940b57cec5SDimitry Andric   SDValue SignBit = DAG.getNode(ISD::AND, SL, MVT::i32, Hi, SignBitMask);
20950b57cec5SDimitry Andric 
20960b57cec5SDimitry Andric   // Extend back to 64-bits.
20970b57cec5SDimitry Andric   SDValue SignBit64 = DAG.getBuildVector(MVT::v2i32, SL, {Zero, SignBit});
20980b57cec5SDimitry Andric   SignBit64 = DAG.getNode(ISD::BITCAST, SL, MVT::i64, SignBit64);
20990b57cec5SDimitry Andric 
21000b57cec5SDimitry Andric   SDValue BcInt = DAG.getNode(ISD::BITCAST, SL, MVT::i64, Src);
21010b57cec5SDimitry Andric   const SDValue FractMask
21020b57cec5SDimitry Andric     = DAG.getConstant((UINT64_C(1) << FractBits) - 1, SL, MVT::i64);
21030b57cec5SDimitry Andric 
21040b57cec5SDimitry Andric   SDValue Shr = DAG.getNode(ISD::SRA, SL, MVT::i64, FractMask, Exp);
21050b57cec5SDimitry Andric   SDValue Not = DAG.getNOT(SL, Shr, MVT::i64);
21060b57cec5SDimitry Andric   SDValue Tmp0 = DAG.getNode(ISD::AND, SL, MVT::i64, BcInt, Not);
21070b57cec5SDimitry Andric 
21080b57cec5SDimitry Andric   EVT SetCCVT =
21090b57cec5SDimitry Andric       getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), MVT::i32);
21100b57cec5SDimitry Andric 
21110b57cec5SDimitry Andric   const SDValue FiftyOne = DAG.getConstant(FractBits - 1, SL, MVT::i32);
21120b57cec5SDimitry Andric 
21130b57cec5SDimitry Andric   SDValue ExpLt0 = DAG.getSetCC(SL, SetCCVT, Exp, Zero, ISD::SETLT);
21140b57cec5SDimitry Andric   SDValue ExpGt51 = DAG.getSetCC(SL, SetCCVT, Exp, FiftyOne, ISD::SETGT);
21150b57cec5SDimitry Andric 
21160b57cec5SDimitry Andric   SDValue Tmp1 = DAG.getNode(ISD::SELECT, SL, MVT::i64, ExpLt0, SignBit64, Tmp0);
21170b57cec5SDimitry Andric   SDValue Tmp2 = DAG.getNode(ISD::SELECT, SL, MVT::i64, ExpGt51, BcInt, Tmp1);
21180b57cec5SDimitry Andric 
21190b57cec5SDimitry Andric   return DAG.getNode(ISD::BITCAST, SL, MVT::f64, Tmp2);
21200b57cec5SDimitry Andric }
21210b57cec5SDimitry Andric 
21220b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFRINT(SDValue Op, SelectionDAG &DAG) const {
21230b57cec5SDimitry Andric   SDLoc SL(Op);
21240b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
21250b57cec5SDimitry Andric 
21260b57cec5SDimitry Andric   assert(Op.getValueType() == MVT::f64);
21270b57cec5SDimitry Andric 
21280b57cec5SDimitry Andric   APFloat C1Val(APFloat::IEEEdouble(), "0x1.0p+52");
21290b57cec5SDimitry Andric   SDValue C1 = DAG.getConstantFP(C1Val, SL, MVT::f64);
21300b57cec5SDimitry Andric   SDValue CopySign = DAG.getNode(ISD::FCOPYSIGN, SL, MVT::f64, C1, Src);
21310b57cec5SDimitry Andric 
21320b57cec5SDimitry Andric   // TODO: Should this propagate fast-math-flags?
21330b57cec5SDimitry Andric 
21340b57cec5SDimitry Andric   SDValue Tmp1 = DAG.getNode(ISD::FADD, SL, MVT::f64, Src, CopySign);
21350b57cec5SDimitry Andric   SDValue Tmp2 = DAG.getNode(ISD::FSUB, SL, MVT::f64, Tmp1, CopySign);
21360b57cec5SDimitry Andric 
21370b57cec5SDimitry Andric   SDValue Fabs = DAG.getNode(ISD::FABS, SL, MVT::f64, Src);
21380b57cec5SDimitry Andric 
21390b57cec5SDimitry Andric   APFloat C2Val(APFloat::IEEEdouble(), "0x1.fffffffffffffp+51");
21400b57cec5SDimitry Andric   SDValue C2 = DAG.getConstantFP(C2Val, SL, MVT::f64);
21410b57cec5SDimitry Andric 
21420b57cec5SDimitry Andric   EVT SetCCVT =
21430b57cec5SDimitry Andric       getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), MVT::f64);
21440b57cec5SDimitry Andric   SDValue Cond = DAG.getSetCC(SL, SetCCVT, Fabs, C2, ISD::SETOGT);
21450b57cec5SDimitry Andric 
21460b57cec5SDimitry Andric   return DAG.getSelect(SL, MVT::f64, Cond, Src, Tmp2);
21470b57cec5SDimitry Andric }
21480b57cec5SDimitry Andric 
21490b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFNEARBYINT(SDValue Op, SelectionDAG &DAG) const {
21500b57cec5SDimitry Andric   // FNEARBYINT and FRINT are the same, except in their handling of FP
21510b57cec5SDimitry Andric   // exceptions. Those aren't really meaningful for us, and OpenCL only has
21520b57cec5SDimitry Andric   // rint, so just treat them as equivalent.
21530b57cec5SDimitry Andric   return DAG.getNode(ISD::FRINT, SDLoc(Op), Op.getValueType(), Op.getOperand(0));
21540b57cec5SDimitry Andric }
21550b57cec5SDimitry Andric 
21560b57cec5SDimitry Andric // XXX - May require not supporting f32 denormals?
21570b57cec5SDimitry Andric 
21580b57cec5SDimitry Andric // Don't handle v2f16. The extra instructions to scalarize and repack around the
21590b57cec5SDimitry Andric // compare and vselect end up producing worse code than scalarizing the whole
21600b57cec5SDimitry Andric // operation.
21610b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFROUND32_16(SDValue Op, SelectionDAG &DAG) const {
21620b57cec5SDimitry Andric   SDLoc SL(Op);
21630b57cec5SDimitry Andric   SDValue X = Op.getOperand(0);
21640b57cec5SDimitry Andric   EVT VT = Op.getValueType();
21650b57cec5SDimitry Andric 
21660b57cec5SDimitry Andric   SDValue T = DAG.getNode(ISD::FTRUNC, SL, VT, X);
21670b57cec5SDimitry Andric 
21680b57cec5SDimitry Andric   // TODO: Should this propagate fast-math-flags?
21690b57cec5SDimitry Andric 
21700b57cec5SDimitry Andric   SDValue Diff = DAG.getNode(ISD::FSUB, SL, VT, X, T);
21710b57cec5SDimitry Andric 
21720b57cec5SDimitry Andric   SDValue AbsDiff = DAG.getNode(ISD::FABS, SL, VT, Diff);
21730b57cec5SDimitry Andric 
21740b57cec5SDimitry Andric   const SDValue Zero = DAG.getConstantFP(0.0, SL, VT);
21750b57cec5SDimitry Andric   const SDValue One = DAG.getConstantFP(1.0, SL, VT);
21760b57cec5SDimitry Andric   const SDValue Half = DAG.getConstantFP(0.5, SL, VT);
21770b57cec5SDimitry Andric 
21780b57cec5SDimitry Andric   SDValue SignOne = DAG.getNode(ISD::FCOPYSIGN, SL, VT, One, X);
21790b57cec5SDimitry Andric 
21800b57cec5SDimitry Andric   EVT SetCCVT =
21810b57cec5SDimitry Andric       getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), VT);
21820b57cec5SDimitry Andric 
21830b57cec5SDimitry Andric   SDValue Cmp = DAG.getSetCC(SL, SetCCVT, AbsDiff, Half, ISD::SETOGE);
21840b57cec5SDimitry Andric 
21850b57cec5SDimitry Andric   SDValue Sel = DAG.getNode(ISD::SELECT, SL, VT, Cmp, SignOne, Zero);
21860b57cec5SDimitry Andric 
21870b57cec5SDimitry Andric   return DAG.getNode(ISD::FADD, SL, VT, T, Sel);
21880b57cec5SDimitry Andric }
21890b57cec5SDimitry Andric 
21900b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFROUND64(SDValue Op, SelectionDAG &DAG) const {
21910b57cec5SDimitry Andric   SDLoc SL(Op);
21920b57cec5SDimitry Andric   SDValue X = Op.getOperand(0);
21930b57cec5SDimitry Andric 
21940b57cec5SDimitry Andric   SDValue L = DAG.getNode(ISD::BITCAST, SL, MVT::i64, X);
21950b57cec5SDimitry Andric 
21960b57cec5SDimitry Andric   const SDValue Zero = DAG.getConstant(0, SL, MVT::i32);
21970b57cec5SDimitry Andric   const SDValue One = DAG.getConstant(1, SL, MVT::i32);
21980b57cec5SDimitry Andric   const SDValue NegOne = DAG.getConstant(-1, SL, MVT::i32);
21990b57cec5SDimitry Andric   const SDValue FiftyOne = DAG.getConstant(51, SL, MVT::i32);
22000b57cec5SDimitry Andric   EVT SetCCVT =
22010b57cec5SDimitry Andric       getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), MVT::i32);
22020b57cec5SDimitry Andric 
22030b57cec5SDimitry Andric   SDValue BC = DAG.getNode(ISD::BITCAST, SL, MVT::v2i32, X);
22040b57cec5SDimitry Andric 
22050b57cec5SDimitry Andric   SDValue Hi = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, MVT::i32, BC, One);
22060b57cec5SDimitry Andric 
22070b57cec5SDimitry Andric   SDValue Exp = extractF64Exponent(Hi, SL, DAG);
22080b57cec5SDimitry Andric 
22090b57cec5SDimitry Andric   const SDValue Mask = DAG.getConstant(INT64_C(0x000fffffffffffff), SL,
22100b57cec5SDimitry Andric                                        MVT::i64);
22110b57cec5SDimitry Andric 
22120b57cec5SDimitry Andric   SDValue M = DAG.getNode(ISD::SRA, SL, MVT::i64, Mask, Exp);
22130b57cec5SDimitry Andric   SDValue D = DAG.getNode(ISD::SRA, SL, MVT::i64,
22140b57cec5SDimitry Andric                           DAG.getConstant(INT64_C(0x0008000000000000), SL,
22150b57cec5SDimitry Andric                                           MVT::i64),
22160b57cec5SDimitry Andric                           Exp);
22170b57cec5SDimitry Andric 
22180b57cec5SDimitry Andric   SDValue Tmp0 = DAG.getNode(ISD::AND, SL, MVT::i64, L, M);
22190b57cec5SDimitry Andric   SDValue Tmp1 = DAG.getSetCC(SL, SetCCVT,
22200b57cec5SDimitry Andric                               DAG.getConstant(0, SL, MVT::i64), Tmp0,
22210b57cec5SDimitry Andric                               ISD::SETNE);
22220b57cec5SDimitry Andric 
22230b57cec5SDimitry Andric   SDValue Tmp2 = DAG.getNode(ISD::SELECT, SL, MVT::i64, Tmp1,
22240b57cec5SDimitry Andric                              D, DAG.getConstant(0, SL, MVT::i64));
22250b57cec5SDimitry Andric   SDValue K = DAG.getNode(ISD::ADD, SL, MVT::i64, L, Tmp2);
22260b57cec5SDimitry Andric 
22270b57cec5SDimitry Andric   K = DAG.getNode(ISD::AND, SL, MVT::i64, K, DAG.getNOT(SL, M, MVT::i64));
22280b57cec5SDimitry Andric   K = DAG.getNode(ISD::BITCAST, SL, MVT::f64, K);
22290b57cec5SDimitry Andric 
22300b57cec5SDimitry Andric   SDValue ExpLt0 = DAG.getSetCC(SL, SetCCVT, Exp, Zero, ISD::SETLT);
22310b57cec5SDimitry Andric   SDValue ExpGt51 = DAG.getSetCC(SL, SetCCVT, Exp, FiftyOne, ISD::SETGT);
22320b57cec5SDimitry Andric   SDValue ExpEqNegOne = DAG.getSetCC(SL, SetCCVT, NegOne, Exp, ISD::SETEQ);
22330b57cec5SDimitry Andric 
22340b57cec5SDimitry Andric   SDValue Mag = DAG.getNode(ISD::SELECT, SL, MVT::f64,
22350b57cec5SDimitry Andric                             ExpEqNegOne,
22360b57cec5SDimitry Andric                             DAG.getConstantFP(1.0, SL, MVT::f64),
22370b57cec5SDimitry Andric                             DAG.getConstantFP(0.0, SL, MVT::f64));
22380b57cec5SDimitry Andric 
22390b57cec5SDimitry Andric   SDValue S = DAG.getNode(ISD::FCOPYSIGN, SL, MVT::f64, Mag, X);
22400b57cec5SDimitry Andric 
22410b57cec5SDimitry Andric   K = DAG.getNode(ISD::SELECT, SL, MVT::f64, ExpLt0, S, K);
22420b57cec5SDimitry Andric   K = DAG.getNode(ISD::SELECT, SL, MVT::f64, ExpGt51, X, K);
22430b57cec5SDimitry Andric 
22440b57cec5SDimitry Andric   return K;
22450b57cec5SDimitry Andric }
22460b57cec5SDimitry Andric 
22470b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFROUND(SDValue Op, SelectionDAG &DAG) const {
22480b57cec5SDimitry Andric   EVT VT = Op.getValueType();
22490b57cec5SDimitry Andric 
22500b57cec5SDimitry Andric   if (VT == MVT::f32 || VT == MVT::f16)
22510b57cec5SDimitry Andric     return LowerFROUND32_16(Op, DAG);
22520b57cec5SDimitry Andric 
22530b57cec5SDimitry Andric   if (VT == MVT::f64)
22540b57cec5SDimitry Andric     return LowerFROUND64(Op, DAG);
22550b57cec5SDimitry Andric 
22560b57cec5SDimitry Andric   llvm_unreachable("unhandled type");
22570b57cec5SDimitry Andric }
22580b57cec5SDimitry Andric 
22590b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFFLOOR(SDValue Op, SelectionDAG &DAG) const {
22600b57cec5SDimitry Andric   SDLoc SL(Op);
22610b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
22620b57cec5SDimitry Andric 
22630b57cec5SDimitry Andric   // result = trunc(src);
22640b57cec5SDimitry Andric   // if (src < 0.0 && src != result)
22650b57cec5SDimitry Andric   //   result += -1.0.
22660b57cec5SDimitry Andric 
22670b57cec5SDimitry Andric   SDValue Trunc = DAG.getNode(ISD::FTRUNC, SL, MVT::f64, Src);
22680b57cec5SDimitry Andric 
22690b57cec5SDimitry Andric   const SDValue Zero = DAG.getConstantFP(0.0, SL, MVT::f64);
22700b57cec5SDimitry Andric   const SDValue NegOne = DAG.getConstantFP(-1.0, SL, MVT::f64);
22710b57cec5SDimitry Andric 
22720b57cec5SDimitry Andric   EVT SetCCVT =
22730b57cec5SDimitry Andric       getSetCCResultType(DAG.getDataLayout(), *DAG.getContext(), MVT::f64);
22740b57cec5SDimitry Andric 
22750b57cec5SDimitry Andric   SDValue Lt0 = DAG.getSetCC(SL, SetCCVT, Src, Zero, ISD::SETOLT);
22760b57cec5SDimitry Andric   SDValue NeTrunc = DAG.getSetCC(SL, SetCCVT, Src, Trunc, ISD::SETONE);
22770b57cec5SDimitry Andric   SDValue And = DAG.getNode(ISD::AND, SL, SetCCVT, Lt0, NeTrunc);
22780b57cec5SDimitry Andric 
22790b57cec5SDimitry Andric   SDValue Add = DAG.getNode(ISD::SELECT, SL, MVT::f64, And, NegOne, Zero);
22800b57cec5SDimitry Andric   // TODO: Should this propagate fast-math-flags?
22810b57cec5SDimitry Andric   return DAG.getNode(ISD::FADD, SL, MVT::f64, Trunc, Add);
22820b57cec5SDimitry Andric }
22830b57cec5SDimitry Andric 
22840b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFLOG(SDValue Op, SelectionDAG &DAG,
22850b57cec5SDimitry Andric                                         double Log2BaseInverted) const {
22860b57cec5SDimitry Andric   EVT VT = Op.getValueType();
22870b57cec5SDimitry Andric 
22880b57cec5SDimitry Andric   SDLoc SL(Op);
22890b57cec5SDimitry Andric   SDValue Operand = Op.getOperand(0);
22900b57cec5SDimitry Andric   SDValue Log2Operand = DAG.getNode(ISD::FLOG2, SL, VT, Operand);
22910b57cec5SDimitry Andric   SDValue Log2BaseInvertedOperand = DAG.getConstantFP(Log2BaseInverted, SL, VT);
22920b57cec5SDimitry Andric 
22930b57cec5SDimitry Andric   return DAG.getNode(ISD::FMUL, SL, VT, Log2Operand, Log2BaseInvertedOperand);
22940b57cec5SDimitry Andric }
22950b57cec5SDimitry Andric 
22960b57cec5SDimitry Andric // exp2(M_LOG2E_F * f);
22970b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::lowerFEXP(SDValue Op, SelectionDAG &DAG) const {
22980b57cec5SDimitry Andric   EVT VT = Op.getValueType();
22990b57cec5SDimitry Andric   SDLoc SL(Op);
23000b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
23010b57cec5SDimitry Andric 
2302*8bcb0991SDimitry Andric   const SDValue K = DAG.getConstantFP(numbers::log2e, SL, VT);
23030b57cec5SDimitry Andric   SDValue Mul = DAG.getNode(ISD::FMUL, SL, VT, Src, K, Op->getFlags());
23040b57cec5SDimitry Andric   return DAG.getNode(ISD::FEXP2, SL, VT, Mul, Op->getFlags());
23050b57cec5SDimitry Andric }
23060b57cec5SDimitry Andric 
23070b57cec5SDimitry Andric static bool isCtlzOpc(unsigned Opc) {
23080b57cec5SDimitry Andric   return Opc == ISD::CTLZ || Opc == ISD::CTLZ_ZERO_UNDEF;
23090b57cec5SDimitry Andric }
23100b57cec5SDimitry Andric 
23110b57cec5SDimitry Andric static bool isCttzOpc(unsigned Opc) {
23120b57cec5SDimitry Andric   return Opc == ISD::CTTZ || Opc == ISD::CTTZ_ZERO_UNDEF;
23130b57cec5SDimitry Andric }
23140b57cec5SDimitry Andric 
23150b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerCTLZ_CTTZ(SDValue Op, SelectionDAG &DAG) const {
23160b57cec5SDimitry Andric   SDLoc SL(Op);
23170b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
23180b57cec5SDimitry Andric   bool ZeroUndef = Op.getOpcode() == ISD::CTTZ_ZERO_UNDEF ||
23190b57cec5SDimitry Andric                    Op.getOpcode() == ISD::CTLZ_ZERO_UNDEF;
23200b57cec5SDimitry Andric 
23210b57cec5SDimitry Andric   unsigned ISDOpc, NewOpc;
23220b57cec5SDimitry Andric   if (isCtlzOpc(Op.getOpcode())) {
23230b57cec5SDimitry Andric     ISDOpc = ISD::CTLZ_ZERO_UNDEF;
23240b57cec5SDimitry Andric     NewOpc = AMDGPUISD::FFBH_U32;
23250b57cec5SDimitry Andric   } else if (isCttzOpc(Op.getOpcode())) {
23260b57cec5SDimitry Andric     ISDOpc = ISD::CTTZ_ZERO_UNDEF;
23270b57cec5SDimitry Andric     NewOpc = AMDGPUISD::FFBL_B32;
23280b57cec5SDimitry Andric   } else
23290b57cec5SDimitry Andric     llvm_unreachable("Unexpected OPCode!!!");
23300b57cec5SDimitry Andric 
23310b57cec5SDimitry Andric 
23320b57cec5SDimitry Andric   if (ZeroUndef && Src.getValueType() == MVT::i32)
23330b57cec5SDimitry Andric     return DAG.getNode(NewOpc, SL, MVT::i32, Src);
23340b57cec5SDimitry Andric 
23350b57cec5SDimitry Andric   SDValue Vec = DAG.getNode(ISD::BITCAST, SL, MVT::v2i32, Src);
23360b57cec5SDimitry Andric 
23370b57cec5SDimitry Andric   const SDValue Zero = DAG.getConstant(0, SL, MVT::i32);
23380b57cec5SDimitry Andric   const SDValue One = DAG.getConstant(1, SL, MVT::i32);
23390b57cec5SDimitry Andric 
23400b57cec5SDimitry Andric   SDValue Lo = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, MVT::i32, Vec, Zero);
23410b57cec5SDimitry Andric   SDValue Hi = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, MVT::i32, Vec, One);
23420b57cec5SDimitry Andric 
23430b57cec5SDimitry Andric   EVT SetCCVT = getSetCCResultType(DAG.getDataLayout(),
23440b57cec5SDimitry Andric                                    *DAG.getContext(), MVT::i32);
23450b57cec5SDimitry Andric 
23460b57cec5SDimitry Andric   SDValue HiOrLo = isCtlzOpc(Op.getOpcode()) ? Hi : Lo;
23470b57cec5SDimitry Andric   SDValue Hi0orLo0 = DAG.getSetCC(SL, SetCCVT, HiOrLo, Zero, ISD::SETEQ);
23480b57cec5SDimitry Andric 
23490b57cec5SDimitry Andric   SDValue OprLo = DAG.getNode(ISDOpc, SL, MVT::i32, Lo);
23500b57cec5SDimitry Andric   SDValue OprHi = DAG.getNode(ISDOpc, SL, MVT::i32, Hi);
23510b57cec5SDimitry Andric 
23520b57cec5SDimitry Andric   const SDValue Bits32 = DAG.getConstant(32, SL, MVT::i32);
23530b57cec5SDimitry Andric   SDValue Add, NewOpr;
23540b57cec5SDimitry Andric   if (isCtlzOpc(Op.getOpcode())) {
23550b57cec5SDimitry Andric     Add = DAG.getNode(ISD::ADD, SL, MVT::i32, OprLo, Bits32);
23560b57cec5SDimitry Andric     // ctlz(x) = hi_32(x) == 0 ? ctlz(lo_32(x)) + 32 : ctlz(hi_32(x))
23570b57cec5SDimitry Andric     NewOpr = DAG.getNode(ISD::SELECT, SL, MVT::i32, Hi0orLo0, Add, OprHi);
23580b57cec5SDimitry Andric   } else {
23590b57cec5SDimitry Andric     Add = DAG.getNode(ISD::ADD, SL, MVT::i32, OprHi, Bits32);
23600b57cec5SDimitry Andric     // cttz(x) = lo_32(x) == 0 ? cttz(hi_32(x)) + 32 : cttz(lo_32(x))
23610b57cec5SDimitry Andric     NewOpr = DAG.getNode(ISD::SELECT, SL, MVT::i32, Hi0orLo0, Add, OprLo);
23620b57cec5SDimitry Andric   }
23630b57cec5SDimitry Andric 
23640b57cec5SDimitry Andric   if (!ZeroUndef) {
23650b57cec5SDimitry Andric     // Test if the full 64-bit input is zero.
23660b57cec5SDimitry Andric 
23670b57cec5SDimitry Andric     // FIXME: DAG combines turn what should be an s_and_b64 into a v_or_b32,
23680b57cec5SDimitry Andric     // which we probably don't want.
23690b57cec5SDimitry Andric     SDValue LoOrHi = isCtlzOpc(Op.getOpcode()) ? Lo : Hi;
23700b57cec5SDimitry Andric     SDValue Lo0OrHi0 = DAG.getSetCC(SL, SetCCVT, LoOrHi, Zero, ISD::SETEQ);
23710b57cec5SDimitry Andric     SDValue SrcIsZero = DAG.getNode(ISD::AND, SL, SetCCVT, Lo0OrHi0, Hi0orLo0);
23720b57cec5SDimitry Andric 
23730b57cec5SDimitry Andric     // TODO: If i64 setcc is half rate, it can result in 1 fewer instruction
23740b57cec5SDimitry Andric     // with the same cycles, otherwise it is slower.
23750b57cec5SDimitry Andric     // SDValue SrcIsZero = DAG.getSetCC(SL, SetCCVT, Src,
23760b57cec5SDimitry Andric     // DAG.getConstant(0, SL, MVT::i64), ISD::SETEQ);
23770b57cec5SDimitry Andric 
23780b57cec5SDimitry Andric     const SDValue Bits32 = DAG.getConstant(64, SL, MVT::i32);
23790b57cec5SDimitry Andric 
23800b57cec5SDimitry Andric     // The instruction returns -1 for 0 input, but the defined intrinsic
23810b57cec5SDimitry Andric     // behavior is to return the number of bits.
23820b57cec5SDimitry Andric     NewOpr = DAG.getNode(ISD::SELECT, SL, MVT::i32,
23830b57cec5SDimitry Andric                          SrcIsZero, Bits32, NewOpr);
23840b57cec5SDimitry Andric   }
23850b57cec5SDimitry Andric 
23860b57cec5SDimitry Andric   return DAG.getNode(ISD::ZERO_EXTEND, SL, MVT::i64, NewOpr);
23870b57cec5SDimitry Andric }
23880b57cec5SDimitry Andric 
23890b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerINT_TO_FP32(SDValue Op, SelectionDAG &DAG,
23900b57cec5SDimitry Andric                                                bool Signed) const {
23910b57cec5SDimitry Andric   // Unsigned
23920b57cec5SDimitry Andric   // cul2f(ulong u)
23930b57cec5SDimitry Andric   //{
23940b57cec5SDimitry Andric   //  uint lz = clz(u);
23950b57cec5SDimitry Andric   //  uint e = (u != 0) ? 127U + 63U - lz : 0;
23960b57cec5SDimitry Andric   //  u = (u << lz) & 0x7fffffffffffffffUL;
23970b57cec5SDimitry Andric   //  ulong t = u & 0xffffffffffUL;
23980b57cec5SDimitry Andric   //  uint v = (e << 23) | (uint)(u >> 40);
23990b57cec5SDimitry Andric   //  uint r = t > 0x8000000000UL ? 1U : (t == 0x8000000000UL ? v & 1U : 0U);
24000b57cec5SDimitry Andric   //  return as_float(v + r);
24010b57cec5SDimitry Andric   //}
24020b57cec5SDimitry Andric   // Signed
24030b57cec5SDimitry Andric   // cl2f(long l)
24040b57cec5SDimitry Andric   //{
24050b57cec5SDimitry Andric   //  long s = l >> 63;
24060b57cec5SDimitry Andric   //  float r = cul2f((l + s) ^ s);
24070b57cec5SDimitry Andric   //  return s ? -r : r;
24080b57cec5SDimitry Andric   //}
24090b57cec5SDimitry Andric 
24100b57cec5SDimitry Andric   SDLoc SL(Op);
24110b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
24120b57cec5SDimitry Andric   SDValue L = Src;
24130b57cec5SDimitry Andric 
24140b57cec5SDimitry Andric   SDValue S;
24150b57cec5SDimitry Andric   if (Signed) {
24160b57cec5SDimitry Andric     const SDValue SignBit = DAG.getConstant(63, SL, MVT::i64);
24170b57cec5SDimitry Andric     S = DAG.getNode(ISD::SRA, SL, MVT::i64, L, SignBit);
24180b57cec5SDimitry Andric 
24190b57cec5SDimitry Andric     SDValue LPlusS = DAG.getNode(ISD::ADD, SL, MVT::i64, L, S);
24200b57cec5SDimitry Andric     L = DAG.getNode(ISD::XOR, SL, MVT::i64, LPlusS, S);
24210b57cec5SDimitry Andric   }
24220b57cec5SDimitry Andric 
24230b57cec5SDimitry Andric   EVT SetCCVT = getSetCCResultType(DAG.getDataLayout(),
24240b57cec5SDimitry Andric                                    *DAG.getContext(), MVT::f32);
24250b57cec5SDimitry Andric 
24260b57cec5SDimitry Andric 
24270b57cec5SDimitry Andric   SDValue ZeroI32 = DAG.getConstant(0, SL, MVT::i32);
24280b57cec5SDimitry Andric   SDValue ZeroI64 = DAG.getConstant(0, SL, MVT::i64);
24290b57cec5SDimitry Andric   SDValue LZ = DAG.getNode(ISD::CTLZ_ZERO_UNDEF, SL, MVT::i64, L);
24300b57cec5SDimitry Andric   LZ = DAG.getNode(ISD::TRUNCATE, SL, MVT::i32, LZ);
24310b57cec5SDimitry Andric 
24320b57cec5SDimitry Andric   SDValue K = DAG.getConstant(127U + 63U, SL, MVT::i32);
24330b57cec5SDimitry Andric   SDValue E = DAG.getSelect(SL, MVT::i32,
24340b57cec5SDimitry Andric     DAG.getSetCC(SL, SetCCVT, L, ZeroI64, ISD::SETNE),
24350b57cec5SDimitry Andric     DAG.getNode(ISD::SUB, SL, MVT::i32, K, LZ),
24360b57cec5SDimitry Andric     ZeroI32);
24370b57cec5SDimitry Andric 
24380b57cec5SDimitry Andric   SDValue U = DAG.getNode(ISD::AND, SL, MVT::i64,
24390b57cec5SDimitry Andric     DAG.getNode(ISD::SHL, SL, MVT::i64, L, LZ),
24400b57cec5SDimitry Andric     DAG.getConstant((-1ULL) >> 1, SL, MVT::i64));
24410b57cec5SDimitry Andric 
24420b57cec5SDimitry Andric   SDValue T = DAG.getNode(ISD::AND, SL, MVT::i64, U,
24430b57cec5SDimitry Andric                           DAG.getConstant(0xffffffffffULL, SL, MVT::i64));
24440b57cec5SDimitry Andric 
24450b57cec5SDimitry Andric   SDValue UShl = DAG.getNode(ISD::SRL, SL, MVT::i64,
24460b57cec5SDimitry Andric                              U, DAG.getConstant(40, SL, MVT::i64));
24470b57cec5SDimitry Andric 
24480b57cec5SDimitry Andric   SDValue V = DAG.getNode(ISD::OR, SL, MVT::i32,
24490b57cec5SDimitry Andric     DAG.getNode(ISD::SHL, SL, MVT::i32, E, DAG.getConstant(23, SL, MVT::i32)),
24500b57cec5SDimitry Andric     DAG.getNode(ISD::TRUNCATE, SL, MVT::i32,  UShl));
24510b57cec5SDimitry Andric 
24520b57cec5SDimitry Andric   SDValue C = DAG.getConstant(0x8000000000ULL, SL, MVT::i64);
24530b57cec5SDimitry Andric   SDValue RCmp = DAG.getSetCC(SL, SetCCVT, T, C, ISD::SETUGT);
24540b57cec5SDimitry Andric   SDValue TCmp = DAG.getSetCC(SL, SetCCVT, T, C, ISD::SETEQ);
24550b57cec5SDimitry Andric 
24560b57cec5SDimitry Andric   SDValue One = DAG.getConstant(1, SL, MVT::i32);
24570b57cec5SDimitry Andric 
24580b57cec5SDimitry Andric   SDValue VTrunc1 = DAG.getNode(ISD::AND, SL, MVT::i32, V, One);
24590b57cec5SDimitry Andric 
24600b57cec5SDimitry Andric   SDValue R = DAG.getSelect(SL, MVT::i32,
24610b57cec5SDimitry Andric     RCmp,
24620b57cec5SDimitry Andric     One,
24630b57cec5SDimitry Andric     DAG.getSelect(SL, MVT::i32, TCmp, VTrunc1, ZeroI32));
24640b57cec5SDimitry Andric   R = DAG.getNode(ISD::ADD, SL, MVT::i32, V, R);
24650b57cec5SDimitry Andric   R = DAG.getNode(ISD::BITCAST, SL, MVT::f32, R);
24660b57cec5SDimitry Andric 
24670b57cec5SDimitry Andric   if (!Signed)
24680b57cec5SDimitry Andric     return R;
24690b57cec5SDimitry Andric 
24700b57cec5SDimitry Andric   SDValue RNeg = DAG.getNode(ISD::FNEG, SL, MVT::f32, R);
24710b57cec5SDimitry Andric   return DAG.getSelect(SL, MVT::f32, DAG.getSExtOrTrunc(S, SL, SetCCVT), RNeg, R);
24720b57cec5SDimitry Andric }
24730b57cec5SDimitry Andric 
24740b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerINT_TO_FP64(SDValue Op, SelectionDAG &DAG,
24750b57cec5SDimitry Andric                                                bool Signed) const {
24760b57cec5SDimitry Andric   SDLoc SL(Op);
24770b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
24780b57cec5SDimitry Andric 
24790b57cec5SDimitry Andric   SDValue BC = DAG.getNode(ISD::BITCAST, SL, MVT::v2i32, Src);
24800b57cec5SDimitry Andric 
24810b57cec5SDimitry Andric   SDValue Lo = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, MVT::i32, BC,
24820b57cec5SDimitry Andric                            DAG.getConstant(0, SL, MVT::i32));
24830b57cec5SDimitry Andric   SDValue Hi = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, MVT::i32, BC,
24840b57cec5SDimitry Andric                            DAG.getConstant(1, SL, MVT::i32));
24850b57cec5SDimitry Andric 
24860b57cec5SDimitry Andric   SDValue CvtHi = DAG.getNode(Signed ? ISD::SINT_TO_FP : ISD::UINT_TO_FP,
24870b57cec5SDimitry Andric                               SL, MVT::f64, Hi);
24880b57cec5SDimitry Andric 
24890b57cec5SDimitry Andric   SDValue CvtLo = DAG.getNode(ISD::UINT_TO_FP, SL, MVT::f64, Lo);
24900b57cec5SDimitry Andric 
24910b57cec5SDimitry Andric   SDValue LdExp = DAG.getNode(AMDGPUISD::LDEXP, SL, MVT::f64, CvtHi,
24920b57cec5SDimitry Andric                               DAG.getConstant(32, SL, MVT::i32));
24930b57cec5SDimitry Andric   // TODO: Should this propagate fast-math-flags?
24940b57cec5SDimitry Andric   return DAG.getNode(ISD::FADD, SL, MVT::f64, LdExp, CvtLo);
24950b57cec5SDimitry Andric }
24960b57cec5SDimitry Andric 
24970b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerUINT_TO_FP(SDValue Op,
24980b57cec5SDimitry Andric                                                SelectionDAG &DAG) const {
24990b57cec5SDimitry Andric   assert(Op.getOperand(0).getValueType() == MVT::i64 &&
25000b57cec5SDimitry Andric          "operation should be legal");
25010b57cec5SDimitry Andric 
25020b57cec5SDimitry Andric   // TODO: Factor out code common with LowerSINT_TO_FP.
25030b57cec5SDimitry Andric 
25040b57cec5SDimitry Andric   EVT DestVT = Op.getValueType();
25050b57cec5SDimitry Andric   if (Subtarget->has16BitInsts() && DestVT == MVT::f16) {
25060b57cec5SDimitry Andric     SDLoc DL(Op);
25070b57cec5SDimitry Andric     SDValue Src = Op.getOperand(0);
25080b57cec5SDimitry Andric 
25090b57cec5SDimitry Andric     SDValue IntToFp32 = DAG.getNode(Op.getOpcode(), DL, MVT::f32, Src);
25100b57cec5SDimitry Andric     SDValue FPRoundFlag = DAG.getIntPtrConstant(0, SDLoc(Op));
25110b57cec5SDimitry Andric     SDValue FPRound =
25120b57cec5SDimitry Andric         DAG.getNode(ISD::FP_ROUND, DL, MVT::f16, IntToFp32, FPRoundFlag);
25130b57cec5SDimitry Andric 
25140b57cec5SDimitry Andric     return FPRound;
25150b57cec5SDimitry Andric   }
25160b57cec5SDimitry Andric 
25170b57cec5SDimitry Andric   if (DestVT == MVT::f32)
25180b57cec5SDimitry Andric     return LowerINT_TO_FP32(Op, DAG, false);
25190b57cec5SDimitry Andric 
25200b57cec5SDimitry Andric   assert(DestVT == MVT::f64);
25210b57cec5SDimitry Andric   return LowerINT_TO_FP64(Op, DAG, false);
25220b57cec5SDimitry Andric }
25230b57cec5SDimitry Andric 
25240b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerSINT_TO_FP(SDValue Op,
25250b57cec5SDimitry Andric                                               SelectionDAG &DAG) const {
25260b57cec5SDimitry Andric   assert(Op.getOperand(0).getValueType() == MVT::i64 &&
25270b57cec5SDimitry Andric          "operation should be legal");
25280b57cec5SDimitry Andric 
25290b57cec5SDimitry Andric   // TODO: Factor out code common with LowerUINT_TO_FP.
25300b57cec5SDimitry Andric 
25310b57cec5SDimitry Andric   EVT DestVT = Op.getValueType();
25320b57cec5SDimitry Andric   if (Subtarget->has16BitInsts() && DestVT == MVT::f16) {
25330b57cec5SDimitry Andric     SDLoc DL(Op);
25340b57cec5SDimitry Andric     SDValue Src = Op.getOperand(0);
25350b57cec5SDimitry Andric 
25360b57cec5SDimitry Andric     SDValue IntToFp32 = DAG.getNode(Op.getOpcode(), DL, MVT::f32, Src);
25370b57cec5SDimitry Andric     SDValue FPRoundFlag = DAG.getIntPtrConstant(0, SDLoc(Op));
25380b57cec5SDimitry Andric     SDValue FPRound =
25390b57cec5SDimitry Andric         DAG.getNode(ISD::FP_ROUND, DL, MVT::f16, IntToFp32, FPRoundFlag);
25400b57cec5SDimitry Andric 
25410b57cec5SDimitry Andric     return FPRound;
25420b57cec5SDimitry Andric   }
25430b57cec5SDimitry Andric 
25440b57cec5SDimitry Andric   if (DestVT == MVT::f32)
25450b57cec5SDimitry Andric     return LowerINT_TO_FP32(Op, DAG, true);
25460b57cec5SDimitry Andric 
25470b57cec5SDimitry Andric   assert(DestVT == MVT::f64);
25480b57cec5SDimitry Andric   return LowerINT_TO_FP64(Op, DAG, true);
25490b57cec5SDimitry Andric }
25500b57cec5SDimitry Andric 
25510b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFP64_TO_INT(SDValue Op, SelectionDAG &DAG,
25520b57cec5SDimitry Andric                                                bool Signed) const {
25530b57cec5SDimitry Andric   SDLoc SL(Op);
25540b57cec5SDimitry Andric 
25550b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
25560b57cec5SDimitry Andric 
25570b57cec5SDimitry Andric   SDValue Trunc = DAG.getNode(ISD::FTRUNC, SL, MVT::f64, Src);
25580b57cec5SDimitry Andric 
25590b57cec5SDimitry Andric   SDValue K0 = DAG.getConstantFP(BitsToDouble(UINT64_C(0x3df0000000000000)), SL,
25600b57cec5SDimitry Andric                                  MVT::f64);
25610b57cec5SDimitry Andric   SDValue K1 = DAG.getConstantFP(BitsToDouble(UINT64_C(0xc1f0000000000000)), SL,
25620b57cec5SDimitry Andric                                  MVT::f64);
25630b57cec5SDimitry Andric   // TODO: Should this propagate fast-math-flags?
25640b57cec5SDimitry Andric   SDValue Mul = DAG.getNode(ISD::FMUL, SL, MVT::f64, Trunc, K0);
25650b57cec5SDimitry Andric 
25660b57cec5SDimitry Andric   SDValue FloorMul = DAG.getNode(ISD::FFLOOR, SL, MVT::f64, Mul);
25670b57cec5SDimitry Andric 
25680b57cec5SDimitry Andric 
25690b57cec5SDimitry Andric   SDValue Fma = DAG.getNode(ISD::FMA, SL, MVT::f64, FloorMul, K1, Trunc);
25700b57cec5SDimitry Andric 
25710b57cec5SDimitry Andric   SDValue Hi = DAG.getNode(Signed ? ISD::FP_TO_SINT : ISD::FP_TO_UINT, SL,
25720b57cec5SDimitry Andric                            MVT::i32, FloorMul);
25730b57cec5SDimitry Andric   SDValue Lo = DAG.getNode(ISD::FP_TO_UINT, SL, MVT::i32, Fma);
25740b57cec5SDimitry Andric 
25750b57cec5SDimitry Andric   SDValue Result = DAG.getBuildVector(MVT::v2i32, SL, {Lo, Hi});
25760b57cec5SDimitry Andric 
25770b57cec5SDimitry Andric   return DAG.getNode(ISD::BITCAST, SL, MVT::i64, Result);
25780b57cec5SDimitry Andric }
25790b57cec5SDimitry Andric 
25800b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFP_TO_FP16(SDValue Op, SelectionDAG &DAG) const {
25810b57cec5SDimitry Andric   SDLoc DL(Op);
25820b57cec5SDimitry Andric   SDValue N0 = Op.getOperand(0);
25830b57cec5SDimitry Andric 
25840b57cec5SDimitry Andric   // Convert to target node to get known bits
25850b57cec5SDimitry Andric   if (N0.getValueType() == MVT::f32)
25860b57cec5SDimitry Andric     return DAG.getNode(AMDGPUISD::FP_TO_FP16, DL, Op.getValueType(), N0);
25870b57cec5SDimitry Andric 
25880b57cec5SDimitry Andric   if (getTargetMachine().Options.UnsafeFPMath) {
25890b57cec5SDimitry Andric     // There is a generic expand for FP_TO_FP16 with unsafe fast math.
25900b57cec5SDimitry Andric     return SDValue();
25910b57cec5SDimitry Andric   }
25920b57cec5SDimitry Andric 
25930b57cec5SDimitry Andric   assert(N0.getSimpleValueType() == MVT::f64);
25940b57cec5SDimitry Andric 
25950b57cec5SDimitry Andric   // f64 -> f16 conversion using round-to-nearest-even rounding mode.
25960b57cec5SDimitry Andric   const unsigned ExpMask = 0x7ff;
25970b57cec5SDimitry Andric   const unsigned ExpBiasf64 = 1023;
25980b57cec5SDimitry Andric   const unsigned ExpBiasf16 = 15;
25990b57cec5SDimitry Andric   SDValue Zero = DAG.getConstant(0, DL, MVT::i32);
26000b57cec5SDimitry Andric   SDValue One = DAG.getConstant(1, DL, MVT::i32);
26010b57cec5SDimitry Andric   SDValue U = DAG.getNode(ISD::BITCAST, DL, MVT::i64, N0);
26020b57cec5SDimitry Andric   SDValue UH = DAG.getNode(ISD::SRL, DL, MVT::i64, U,
26030b57cec5SDimitry Andric                            DAG.getConstant(32, DL, MVT::i64));
26040b57cec5SDimitry Andric   UH = DAG.getZExtOrTrunc(UH, DL, MVT::i32);
26050b57cec5SDimitry Andric   U = DAG.getZExtOrTrunc(U, DL, MVT::i32);
26060b57cec5SDimitry Andric   SDValue E = DAG.getNode(ISD::SRL, DL, MVT::i32, UH,
26070b57cec5SDimitry Andric                           DAG.getConstant(20, DL, MVT::i64));
26080b57cec5SDimitry Andric   E = DAG.getNode(ISD::AND, DL, MVT::i32, E,
26090b57cec5SDimitry Andric                   DAG.getConstant(ExpMask, DL, MVT::i32));
26100b57cec5SDimitry Andric   // Subtract the fp64 exponent bias (1023) to get the real exponent and
26110b57cec5SDimitry Andric   // add the f16 bias (15) to get the biased exponent for the f16 format.
26120b57cec5SDimitry Andric   E = DAG.getNode(ISD::ADD, DL, MVT::i32, E,
26130b57cec5SDimitry Andric                   DAG.getConstant(-ExpBiasf64 + ExpBiasf16, DL, MVT::i32));
26140b57cec5SDimitry Andric 
26150b57cec5SDimitry Andric   SDValue M = DAG.getNode(ISD::SRL, DL, MVT::i32, UH,
26160b57cec5SDimitry Andric                           DAG.getConstant(8, DL, MVT::i32));
26170b57cec5SDimitry Andric   M = DAG.getNode(ISD::AND, DL, MVT::i32, M,
26180b57cec5SDimitry Andric                   DAG.getConstant(0xffe, DL, MVT::i32));
26190b57cec5SDimitry Andric 
26200b57cec5SDimitry Andric   SDValue MaskedSig = DAG.getNode(ISD::AND, DL, MVT::i32, UH,
26210b57cec5SDimitry Andric                                   DAG.getConstant(0x1ff, DL, MVT::i32));
26220b57cec5SDimitry Andric   MaskedSig = DAG.getNode(ISD::OR, DL, MVT::i32, MaskedSig, U);
26230b57cec5SDimitry Andric 
26240b57cec5SDimitry Andric   SDValue Lo40Set = DAG.getSelectCC(DL, MaskedSig, Zero, Zero, One, ISD::SETEQ);
26250b57cec5SDimitry Andric   M = DAG.getNode(ISD::OR, DL, MVT::i32, M, Lo40Set);
26260b57cec5SDimitry Andric 
26270b57cec5SDimitry Andric   // (M != 0 ? 0x0200 : 0) | 0x7c00;
26280b57cec5SDimitry Andric   SDValue I = DAG.getNode(ISD::OR, DL, MVT::i32,
26290b57cec5SDimitry Andric       DAG.getSelectCC(DL, M, Zero, DAG.getConstant(0x0200, DL, MVT::i32),
26300b57cec5SDimitry Andric                       Zero, ISD::SETNE), DAG.getConstant(0x7c00, DL, MVT::i32));
26310b57cec5SDimitry Andric 
26320b57cec5SDimitry Andric   // N = M | (E << 12);
26330b57cec5SDimitry Andric   SDValue N = DAG.getNode(ISD::OR, DL, MVT::i32, M,
26340b57cec5SDimitry Andric       DAG.getNode(ISD::SHL, DL, MVT::i32, E,
26350b57cec5SDimitry Andric                   DAG.getConstant(12, DL, MVT::i32)));
26360b57cec5SDimitry Andric 
26370b57cec5SDimitry Andric   // B = clamp(1-E, 0, 13);
26380b57cec5SDimitry Andric   SDValue OneSubExp = DAG.getNode(ISD::SUB, DL, MVT::i32,
26390b57cec5SDimitry Andric                                   One, E);
26400b57cec5SDimitry Andric   SDValue B = DAG.getNode(ISD::SMAX, DL, MVT::i32, OneSubExp, Zero);
26410b57cec5SDimitry Andric   B = DAG.getNode(ISD::SMIN, DL, MVT::i32, B,
26420b57cec5SDimitry Andric                   DAG.getConstant(13, DL, MVT::i32));
26430b57cec5SDimitry Andric 
26440b57cec5SDimitry Andric   SDValue SigSetHigh = DAG.getNode(ISD::OR, DL, MVT::i32, M,
26450b57cec5SDimitry Andric                                    DAG.getConstant(0x1000, DL, MVT::i32));
26460b57cec5SDimitry Andric 
26470b57cec5SDimitry Andric   SDValue D = DAG.getNode(ISD::SRL, DL, MVT::i32, SigSetHigh, B);
26480b57cec5SDimitry Andric   SDValue D0 = DAG.getNode(ISD::SHL, DL, MVT::i32, D, B);
26490b57cec5SDimitry Andric   SDValue D1 = DAG.getSelectCC(DL, D0, SigSetHigh, One, Zero, ISD::SETNE);
26500b57cec5SDimitry Andric   D = DAG.getNode(ISD::OR, DL, MVT::i32, D, D1);
26510b57cec5SDimitry Andric 
26520b57cec5SDimitry Andric   SDValue V = DAG.getSelectCC(DL, E, One, D, N, ISD::SETLT);
26530b57cec5SDimitry Andric   SDValue VLow3 = DAG.getNode(ISD::AND, DL, MVT::i32, V,
26540b57cec5SDimitry Andric                               DAG.getConstant(0x7, DL, MVT::i32));
26550b57cec5SDimitry Andric   V = DAG.getNode(ISD::SRL, DL, MVT::i32, V,
26560b57cec5SDimitry Andric                   DAG.getConstant(2, DL, MVT::i32));
26570b57cec5SDimitry Andric   SDValue V0 = DAG.getSelectCC(DL, VLow3, DAG.getConstant(3, DL, MVT::i32),
26580b57cec5SDimitry Andric                                One, Zero, ISD::SETEQ);
26590b57cec5SDimitry Andric   SDValue V1 = DAG.getSelectCC(DL, VLow3, DAG.getConstant(5, DL, MVT::i32),
26600b57cec5SDimitry Andric                                One, Zero, ISD::SETGT);
26610b57cec5SDimitry Andric   V1 = DAG.getNode(ISD::OR, DL, MVT::i32, V0, V1);
26620b57cec5SDimitry Andric   V = DAG.getNode(ISD::ADD, DL, MVT::i32, V, V1);
26630b57cec5SDimitry Andric 
26640b57cec5SDimitry Andric   V = DAG.getSelectCC(DL, E, DAG.getConstant(30, DL, MVT::i32),
26650b57cec5SDimitry Andric                       DAG.getConstant(0x7c00, DL, MVT::i32), V, ISD::SETGT);
26660b57cec5SDimitry Andric   V = DAG.getSelectCC(DL, E, DAG.getConstant(1039, DL, MVT::i32),
26670b57cec5SDimitry Andric                       I, V, ISD::SETEQ);
26680b57cec5SDimitry Andric 
26690b57cec5SDimitry Andric   // Extract the sign bit.
26700b57cec5SDimitry Andric   SDValue Sign = DAG.getNode(ISD::SRL, DL, MVT::i32, UH,
26710b57cec5SDimitry Andric                             DAG.getConstant(16, DL, MVT::i32));
26720b57cec5SDimitry Andric   Sign = DAG.getNode(ISD::AND, DL, MVT::i32, Sign,
26730b57cec5SDimitry Andric                      DAG.getConstant(0x8000, DL, MVT::i32));
26740b57cec5SDimitry Andric 
26750b57cec5SDimitry Andric   V = DAG.getNode(ISD::OR, DL, MVT::i32, Sign, V);
26760b57cec5SDimitry Andric   return DAG.getZExtOrTrunc(V, DL, Op.getValueType());
26770b57cec5SDimitry Andric }
26780b57cec5SDimitry Andric 
26790b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFP_TO_SINT(SDValue Op,
26800b57cec5SDimitry Andric                                               SelectionDAG &DAG) const {
26810b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
26820b57cec5SDimitry Andric 
26830b57cec5SDimitry Andric   // TODO: Factor out code common with LowerFP_TO_UINT.
26840b57cec5SDimitry Andric 
26850b57cec5SDimitry Andric   EVT SrcVT = Src.getValueType();
26860b57cec5SDimitry Andric   if (Subtarget->has16BitInsts() && SrcVT == MVT::f16) {
26870b57cec5SDimitry Andric     SDLoc DL(Op);
26880b57cec5SDimitry Andric 
26890b57cec5SDimitry Andric     SDValue FPExtend = DAG.getNode(ISD::FP_EXTEND, DL, MVT::f32, Src);
26900b57cec5SDimitry Andric     SDValue FpToInt32 =
26910b57cec5SDimitry Andric         DAG.getNode(Op.getOpcode(), DL, MVT::i64, FPExtend);
26920b57cec5SDimitry Andric 
26930b57cec5SDimitry Andric     return FpToInt32;
26940b57cec5SDimitry Andric   }
26950b57cec5SDimitry Andric 
26960b57cec5SDimitry Andric   if (Op.getValueType() == MVT::i64 && Src.getValueType() == MVT::f64)
26970b57cec5SDimitry Andric     return LowerFP64_TO_INT(Op, DAG, true);
26980b57cec5SDimitry Andric 
26990b57cec5SDimitry Andric   return SDValue();
27000b57cec5SDimitry Andric }
27010b57cec5SDimitry Andric 
27020b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerFP_TO_UINT(SDValue Op,
27030b57cec5SDimitry Andric                                               SelectionDAG &DAG) const {
27040b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
27050b57cec5SDimitry Andric 
27060b57cec5SDimitry Andric   // TODO: Factor out code common with LowerFP_TO_SINT.
27070b57cec5SDimitry Andric 
27080b57cec5SDimitry Andric   EVT SrcVT = Src.getValueType();
27090b57cec5SDimitry Andric   if (Subtarget->has16BitInsts() && SrcVT == MVT::f16) {
27100b57cec5SDimitry Andric     SDLoc DL(Op);
27110b57cec5SDimitry Andric 
27120b57cec5SDimitry Andric     SDValue FPExtend = DAG.getNode(ISD::FP_EXTEND, DL, MVT::f32, Src);
27130b57cec5SDimitry Andric     SDValue FpToInt32 =
27140b57cec5SDimitry Andric         DAG.getNode(Op.getOpcode(), DL, MVT::i64, FPExtend);
27150b57cec5SDimitry Andric 
27160b57cec5SDimitry Andric     return FpToInt32;
27170b57cec5SDimitry Andric   }
27180b57cec5SDimitry Andric 
27190b57cec5SDimitry Andric   if (Op.getValueType() == MVT::i64 && Src.getValueType() == MVT::f64)
27200b57cec5SDimitry Andric     return LowerFP64_TO_INT(Op, DAG, false);
27210b57cec5SDimitry Andric 
27220b57cec5SDimitry Andric   return SDValue();
27230b57cec5SDimitry Andric }
27240b57cec5SDimitry Andric 
27250b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::LowerSIGN_EXTEND_INREG(SDValue Op,
27260b57cec5SDimitry Andric                                                      SelectionDAG &DAG) const {
27270b57cec5SDimitry Andric   EVT ExtraVT = cast<VTSDNode>(Op.getOperand(1))->getVT();
27280b57cec5SDimitry Andric   MVT VT = Op.getSimpleValueType();
27290b57cec5SDimitry Andric   MVT ScalarVT = VT.getScalarType();
27300b57cec5SDimitry Andric 
27310b57cec5SDimitry Andric   assert(VT.isVector());
27320b57cec5SDimitry Andric 
27330b57cec5SDimitry Andric   SDValue Src = Op.getOperand(0);
27340b57cec5SDimitry Andric   SDLoc DL(Op);
27350b57cec5SDimitry Andric 
27360b57cec5SDimitry Andric   // TODO: Don't scalarize on Evergreen?
27370b57cec5SDimitry Andric   unsigned NElts = VT.getVectorNumElements();
27380b57cec5SDimitry Andric   SmallVector<SDValue, 8> Args;
27390b57cec5SDimitry Andric   DAG.ExtractVectorElements(Src, Args, 0, NElts);
27400b57cec5SDimitry Andric 
27410b57cec5SDimitry Andric   SDValue VTOp = DAG.getValueType(ExtraVT.getScalarType());
27420b57cec5SDimitry Andric   for (unsigned I = 0; I < NElts; ++I)
27430b57cec5SDimitry Andric     Args[I] = DAG.getNode(ISD::SIGN_EXTEND_INREG, DL, ScalarVT, Args[I], VTOp);
27440b57cec5SDimitry Andric 
27450b57cec5SDimitry Andric   return DAG.getBuildVector(VT, DL, Args);
27460b57cec5SDimitry Andric }
27470b57cec5SDimitry Andric 
27480b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
27490b57cec5SDimitry Andric // Custom DAG optimizations
27500b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
27510b57cec5SDimitry Andric 
27520b57cec5SDimitry Andric static bool isU24(SDValue Op, SelectionDAG &DAG) {
27530b57cec5SDimitry Andric   return AMDGPUTargetLowering::numBitsUnsigned(Op, DAG) <= 24;
27540b57cec5SDimitry Andric }
27550b57cec5SDimitry Andric 
27560b57cec5SDimitry Andric static bool isI24(SDValue Op, SelectionDAG &DAG) {
27570b57cec5SDimitry Andric   EVT VT = Op.getValueType();
27580b57cec5SDimitry Andric   return VT.getSizeInBits() >= 24 && // Types less than 24-bit should be treated
27590b57cec5SDimitry Andric                                      // as unsigned 24-bit values.
27600b57cec5SDimitry Andric     AMDGPUTargetLowering::numBitsSigned(Op, DAG) < 24;
27610b57cec5SDimitry Andric }
27620b57cec5SDimitry Andric 
27630b57cec5SDimitry Andric static SDValue simplifyI24(SDNode *Node24,
27640b57cec5SDimitry Andric                            TargetLowering::DAGCombinerInfo &DCI) {
27650b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
2766*8bcb0991SDimitry Andric   bool IsIntrin = Node24->getOpcode() == ISD::INTRINSIC_WO_CHAIN;
2767*8bcb0991SDimitry Andric 
2768*8bcb0991SDimitry Andric   SDValue LHS = IsIntrin ? Node24->getOperand(1) : Node24->getOperand(0);
2769*8bcb0991SDimitry Andric   SDValue RHS = IsIntrin ? Node24->getOperand(2) : Node24->getOperand(1);
2770*8bcb0991SDimitry Andric   unsigned NewOpcode = Node24->getOpcode();
2771*8bcb0991SDimitry Andric   if (IsIntrin) {
2772*8bcb0991SDimitry Andric     unsigned IID = cast<ConstantSDNode>(Node24->getOperand(0))->getZExtValue();
2773*8bcb0991SDimitry Andric     NewOpcode = IID == Intrinsic::amdgcn_mul_i24 ?
2774*8bcb0991SDimitry Andric       AMDGPUISD::MUL_I24 : AMDGPUISD::MUL_U24;
2775*8bcb0991SDimitry Andric   }
27760b57cec5SDimitry Andric 
27770b57cec5SDimitry Andric   APInt Demanded = APInt::getLowBitsSet(LHS.getValueSizeInBits(), 24);
27780b57cec5SDimitry Andric 
27790b57cec5SDimitry Andric   // First try to simplify using GetDemandedBits which allows the operands to
27800b57cec5SDimitry Andric   // have other uses, but will only perform simplifications that involve
27810b57cec5SDimitry Andric   // bypassing some nodes for this user.
27820b57cec5SDimitry Andric   SDValue DemandedLHS = DAG.GetDemandedBits(LHS, Demanded);
27830b57cec5SDimitry Andric   SDValue DemandedRHS = DAG.GetDemandedBits(RHS, Demanded);
27840b57cec5SDimitry Andric   if (DemandedLHS || DemandedRHS)
2785*8bcb0991SDimitry Andric     return DAG.getNode(NewOpcode, SDLoc(Node24), Node24->getVTList(),
27860b57cec5SDimitry Andric                        DemandedLHS ? DemandedLHS : LHS,
27870b57cec5SDimitry Andric                        DemandedRHS ? DemandedRHS : RHS);
27880b57cec5SDimitry Andric 
27890b57cec5SDimitry Andric   // Now try SimplifyDemandedBits which can simplify the nodes used by our
27900b57cec5SDimitry Andric   // operands if this node is the only user.
27910b57cec5SDimitry Andric   const TargetLowering &TLI = DAG.getTargetLoweringInfo();
27920b57cec5SDimitry Andric   if (TLI.SimplifyDemandedBits(LHS, Demanded, DCI))
27930b57cec5SDimitry Andric     return SDValue(Node24, 0);
27940b57cec5SDimitry Andric   if (TLI.SimplifyDemandedBits(RHS, Demanded, DCI))
27950b57cec5SDimitry Andric     return SDValue(Node24, 0);
27960b57cec5SDimitry Andric 
27970b57cec5SDimitry Andric   return SDValue();
27980b57cec5SDimitry Andric }
27990b57cec5SDimitry Andric 
28000b57cec5SDimitry Andric template <typename IntTy>
28010b57cec5SDimitry Andric static SDValue constantFoldBFE(SelectionDAG &DAG, IntTy Src0, uint32_t Offset,
28020b57cec5SDimitry Andric                                uint32_t Width, const SDLoc &DL) {
28030b57cec5SDimitry Andric   if (Width + Offset < 32) {
28040b57cec5SDimitry Andric     uint32_t Shl = static_cast<uint32_t>(Src0) << (32 - Offset - Width);
28050b57cec5SDimitry Andric     IntTy Result = static_cast<IntTy>(Shl) >> (32 - Width);
28060b57cec5SDimitry Andric     return DAG.getConstant(Result, DL, MVT::i32);
28070b57cec5SDimitry Andric   }
28080b57cec5SDimitry Andric 
28090b57cec5SDimitry Andric   return DAG.getConstant(Src0 >> Offset, DL, MVT::i32);
28100b57cec5SDimitry Andric }
28110b57cec5SDimitry Andric 
28120b57cec5SDimitry Andric static bool hasVolatileUser(SDNode *Val) {
28130b57cec5SDimitry Andric   for (SDNode *U : Val->uses()) {
28140b57cec5SDimitry Andric     if (MemSDNode *M = dyn_cast<MemSDNode>(U)) {
28150b57cec5SDimitry Andric       if (M->isVolatile())
28160b57cec5SDimitry Andric         return true;
28170b57cec5SDimitry Andric     }
28180b57cec5SDimitry Andric   }
28190b57cec5SDimitry Andric 
28200b57cec5SDimitry Andric   return false;
28210b57cec5SDimitry Andric }
28220b57cec5SDimitry Andric 
28230b57cec5SDimitry Andric bool AMDGPUTargetLowering::shouldCombineMemoryType(EVT VT) const {
28240b57cec5SDimitry Andric   // i32 vectors are the canonical memory type.
28250b57cec5SDimitry Andric   if (VT.getScalarType() == MVT::i32 || isTypeLegal(VT))
28260b57cec5SDimitry Andric     return false;
28270b57cec5SDimitry Andric 
28280b57cec5SDimitry Andric   if (!VT.isByteSized())
28290b57cec5SDimitry Andric     return false;
28300b57cec5SDimitry Andric 
28310b57cec5SDimitry Andric   unsigned Size = VT.getStoreSize();
28320b57cec5SDimitry Andric 
28330b57cec5SDimitry Andric   if ((Size == 1 || Size == 2 || Size == 4) && !VT.isVector())
28340b57cec5SDimitry Andric     return false;
28350b57cec5SDimitry Andric 
28360b57cec5SDimitry Andric   if (Size == 3 || (Size > 4 && (Size % 4 != 0)))
28370b57cec5SDimitry Andric     return false;
28380b57cec5SDimitry Andric 
28390b57cec5SDimitry Andric   return true;
28400b57cec5SDimitry Andric }
28410b57cec5SDimitry Andric 
28420b57cec5SDimitry Andric // Replace load of an illegal type with a store of a bitcast to a friendlier
28430b57cec5SDimitry Andric // type.
28440b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performLoadCombine(SDNode *N,
28450b57cec5SDimitry Andric                                                  DAGCombinerInfo &DCI) const {
28460b57cec5SDimitry Andric   if (!DCI.isBeforeLegalize())
28470b57cec5SDimitry Andric     return SDValue();
28480b57cec5SDimitry Andric 
28490b57cec5SDimitry Andric   LoadSDNode *LN = cast<LoadSDNode>(N);
28500b57cec5SDimitry Andric   if (LN->isVolatile() || !ISD::isNormalLoad(LN) || hasVolatileUser(LN))
28510b57cec5SDimitry Andric     return SDValue();
28520b57cec5SDimitry Andric 
28530b57cec5SDimitry Andric   SDLoc SL(N);
28540b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
28550b57cec5SDimitry Andric   EVT VT = LN->getMemoryVT();
28560b57cec5SDimitry Andric 
28570b57cec5SDimitry Andric   unsigned Size = VT.getStoreSize();
28580b57cec5SDimitry Andric   unsigned Align = LN->getAlignment();
28590b57cec5SDimitry Andric   if (Align < Size && isTypeLegal(VT)) {
28600b57cec5SDimitry Andric     bool IsFast;
28610b57cec5SDimitry Andric     unsigned AS = LN->getAddressSpace();
28620b57cec5SDimitry Andric 
28630b57cec5SDimitry Andric     // Expand unaligned loads earlier than legalization. Due to visitation order
28640b57cec5SDimitry Andric     // problems during legalization, the emitted instructions to pack and unpack
28650b57cec5SDimitry Andric     // the bytes again are not eliminated in the case of an unaligned copy.
28660b57cec5SDimitry Andric     if (!allowsMisalignedMemoryAccesses(
28670b57cec5SDimitry Andric             VT, AS, Align, LN->getMemOperand()->getFlags(), &IsFast)) {
28680b57cec5SDimitry Andric       if (VT.isVector())
28690b57cec5SDimitry Andric         return scalarizeVectorLoad(LN, DAG);
28700b57cec5SDimitry Andric 
28710b57cec5SDimitry Andric       SDValue Ops[2];
28720b57cec5SDimitry Andric       std::tie(Ops[0], Ops[1]) = expandUnalignedLoad(LN, DAG);
28730b57cec5SDimitry Andric       return DAG.getMergeValues(Ops, SDLoc(N));
28740b57cec5SDimitry Andric     }
28750b57cec5SDimitry Andric 
28760b57cec5SDimitry Andric     if (!IsFast)
28770b57cec5SDimitry Andric       return SDValue();
28780b57cec5SDimitry Andric   }
28790b57cec5SDimitry Andric 
28800b57cec5SDimitry Andric   if (!shouldCombineMemoryType(VT))
28810b57cec5SDimitry Andric     return SDValue();
28820b57cec5SDimitry Andric 
28830b57cec5SDimitry Andric   EVT NewVT = getEquivalentMemType(*DAG.getContext(), VT);
28840b57cec5SDimitry Andric 
28850b57cec5SDimitry Andric   SDValue NewLoad
28860b57cec5SDimitry Andric     = DAG.getLoad(NewVT, SL, LN->getChain(),
28870b57cec5SDimitry Andric                   LN->getBasePtr(), LN->getMemOperand());
28880b57cec5SDimitry Andric 
28890b57cec5SDimitry Andric   SDValue BC = DAG.getNode(ISD::BITCAST, SL, VT, NewLoad);
28900b57cec5SDimitry Andric   DCI.CombineTo(N, BC, NewLoad.getValue(1));
28910b57cec5SDimitry Andric   return SDValue(N, 0);
28920b57cec5SDimitry Andric }
28930b57cec5SDimitry Andric 
28940b57cec5SDimitry Andric // Replace store of an illegal type with a store of a bitcast to a friendlier
28950b57cec5SDimitry Andric // type.
28960b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performStoreCombine(SDNode *N,
28970b57cec5SDimitry Andric                                                   DAGCombinerInfo &DCI) const {
28980b57cec5SDimitry Andric   if (!DCI.isBeforeLegalize())
28990b57cec5SDimitry Andric     return SDValue();
29000b57cec5SDimitry Andric 
29010b57cec5SDimitry Andric   StoreSDNode *SN = cast<StoreSDNode>(N);
29020b57cec5SDimitry Andric   if (SN->isVolatile() || !ISD::isNormalStore(SN))
29030b57cec5SDimitry Andric     return SDValue();
29040b57cec5SDimitry Andric 
29050b57cec5SDimitry Andric   EVT VT = SN->getMemoryVT();
29060b57cec5SDimitry Andric   unsigned Size = VT.getStoreSize();
29070b57cec5SDimitry Andric 
29080b57cec5SDimitry Andric   SDLoc SL(N);
29090b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
29100b57cec5SDimitry Andric   unsigned Align = SN->getAlignment();
29110b57cec5SDimitry Andric   if (Align < Size && isTypeLegal(VT)) {
29120b57cec5SDimitry Andric     bool IsFast;
29130b57cec5SDimitry Andric     unsigned AS = SN->getAddressSpace();
29140b57cec5SDimitry Andric 
29150b57cec5SDimitry Andric     // Expand unaligned stores earlier than legalization. Due to visitation
29160b57cec5SDimitry Andric     // order problems during legalization, the emitted instructions to pack and
29170b57cec5SDimitry Andric     // unpack the bytes again are not eliminated in the case of an unaligned
29180b57cec5SDimitry Andric     // copy.
29190b57cec5SDimitry Andric     if (!allowsMisalignedMemoryAccesses(
29200b57cec5SDimitry Andric             VT, AS, Align, SN->getMemOperand()->getFlags(), &IsFast)) {
29210b57cec5SDimitry Andric       if (VT.isVector())
29220b57cec5SDimitry Andric         return scalarizeVectorStore(SN, DAG);
29230b57cec5SDimitry Andric 
29240b57cec5SDimitry Andric       return expandUnalignedStore(SN, DAG);
29250b57cec5SDimitry Andric     }
29260b57cec5SDimitry Andric 
29270b57cec5SDimitry Andric     if (!IsFast)
29280b57cec5SDimitry Andric       return SDValue();
29290b57cec5SDimitry Andric   }
29300b57cec5SDimitry Andric 
29310b57cec5SDimitry Andric   if (!shouldCombineMemoryType(VT))
29320b57cec5SDimitry Andric     return SDValue();
29330b57cec5SDimitry Andric 
29340b57cec5SDimitry Andric   EVT NewVT = getEquivalentMemType(*DAG.getContext(), VT);
29350b57cec5SDimitry Andric   SDValue Val = SN->getValue();
29360b57cec5SDimitry Andric 
29370b57cec5SDimitry Andric   //DCI.AddToWorklist(Val.getNode());
29380b57cec5SDimitry Andric 
29390b57cec5SDimitry Andric   bool OtherUses = !Val.hasOneUse();
29400b57cec5SDimitry Andric   SDValue CastVal = DAG.getNode(ISD::BITCAST, SL, NewVT, Val);
29410b57cec5SDimitry Andric   if (OtherUses) {
29420b57cec5SDimitry Andric     SDValue CastBack = DAG.getNode(ISD::BITCAST, SL, VT, CastVal);
29430b57cec5SDimitry Andric     DAG.ReplaceAllUsesOfValueWith(Val, CastBack);
29440b57cec5SDimitry Andric   }
29450b57cec5SDimitry Andric 
29460b57cec5SDimitry Andric   return DAG.getStore(SN->getChain(), SL, CastVal,
29470b57cec5SDimitry Andric                       SN->getBasePtr(), SN->getMemOperand());
29480b57cec5SDimitry Andric }
29490b57cec5SDimitry Andric 
29500b57cec5SDimitry Andric // FIXME: This should go in generic DAG combiner with an isTruncateFree check,
29510b57cec5SDimitry Andric // but isTruncateFree is inaccurate for i16 now because of SALU vs. VALU
29520b57cec5SDimitry Andric // issues.
29530b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performAssertSZExtCombine(SDNode *N,
29540b57cec5SDimitry Andric                                                         DAGCombinerInfo &DCI) const {
29550b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
29560b57cec5SDimitry Andric   SDValue N0 = N->getOperand(0);
29570b57cec5SDimitry Andric 
29580b57cec5SDimitry Andric   // (vt2 (assertzext (truncate vt0:x), vt1)) ->
29590b57cec5SDimitry Andric   //     (vt2 (truncate (assertzext vt0:x, vt1)))
29600b57cec5SDimitry Andric   if (N0.getOpcode() == ISD::TRUNCATE) {
29610b57cec5SDimitry Andric     SDValue N1 = N->getOperand(1);
29620b57cec5SDimitry Andric     EVT ExtVT = cast<VTSDNode>(N1)->getVT();
29630b57cec5SDimitry Andric     SDLoc SL(N);
29640b57cec5SDimitry Andric 
29650b57cec5SDimitry Andric     SDValue Src = N0.getOperand(0);
29660b57cec5SDimitry Andric     EVT SrcVT = Src.getValueType();
29670b57cec5SDimitry Andric     if (SrcVT.bitsGE(ExtVT)) {
29680b57cec5SDimitry Andric       SDValue NewInReg = DAG.getNode(N->getOpcode(), SL, SrcVT, Src, N1);
29690b57cec5SDimitry Andric       return DAG.getNode(ISD::TRUNCATE, SL, N->getValueType(0), NewInReg);
29700b57cec5SDimitry Andric     }
29710b57cec5SDimitry Andric   }
29720b57cec5SDimitry Andric 
29730b57cec5SDimitry Andric   return SDValue();
29740b57cec5SDimitry Andric }
2975*8bcb0991SDimitry Andric 
2976*8bcb0991SDimitry Andric SDValue AMDGPUTargetLowering::performIntrinsicWOChainCombine(
2977*8bcb0991SDimitry Andric   SDNode *N, DAGCombinerInfo &DCI) const {
2978*8bcb0991SDimitry Andric   unsigned IID = cast<ConstantSDNode>(N->getOperand(0))->getZExtValue();
2979*8bcb0991SDimitry Andric   switch (IID) {
2980*8bcb0991SDimitry Andric   case Intrinsic::amdgcn_mul_i24:
2981*8bcb0991SDimitry Andric   case Intrinsic::amdgcn_mul_u24:
2982*8bcb0991SDimitry Andric     return simplifyI24(N, DCI);
2983*8bcb0991SDimitry Andric   default:
2984*8bcb0991SDimitry Andric     return SDValue();
2985*8bcb0991SDimitry Andric   }
2986*8bcb0991SDimitry Andric }
2987*8bcb0991SDimitry Andric 
29880b57cec5SDimitry Andric /// Split the 64-bit value \p LHS into two 32-bit components, and perform the
29890b57cec5SDimitry Andric /// binary operation \p Opc to it with the corresponding constant operands.
29900b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::splitBinaryBitConstantOpImpl(
29910b57cec5SDimitry Andric   DAGCombinerInfo &DCI, const SDLoc &SL,
29920b57cec5SDimitry Andric   unsigned Opc, SDValue LHS,
29930b57cec5SDimitry Andric   uint32_t ValLo, uint32_t ValHi) const {
29940b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
29950b57cec5SDimitry Andric   SDValue Lo, Hi;
29960b57cec5SDimitry Andric   std::tie(Lo, Hi) = split64BitValue(LHS, DAG);
29970b57cec5SDimitry Andric 
29980b57cec5SDimitry Andric   SDValue LoRHS = DAG.getConstant(ValLo, SL, MVT::i32);
29990b57cec5SDimitry Andric   SDValue HiRHS = DAG.getConstant(ValHi, SL, MVT::i32);
30000b57cec5SDimitry Andric 
30010b57cec5SDimitry Andric   SDValue LoAnd = DAG.getNode(Opc, SL, MVT::i32, Lo, LoRHS);
30020b57cec5SDimitry Andric   SDValue HiAnd = DAG.getNode(Opc, SL, MVT::i32, Hi, HiRHS);
30030b57cec5SDimitry Andric 
30040b57cec5SDimitry Andric   // Re-visit the ands. It's possible we eliminated one of them and it could
30050b57cec5SDimitry Andric   // simplify the vector.
30060b57cec5SDimitry Andric   DCI.AddToWorklist(Lo.getNode());
30070b57cec5SDimitry Andric   DCI.AddToWorklist(Hi.getNode());
30080b57cec5SDimitry Andric 
30090b57cec5SDimitry Andric   SDValue Vec = DAG.getBuildVector(MVT::v2i32, SL, {LoAnd, HiAnd});
30100b57cec5SDimitry Andric   return DAG.getNode(ISD::BITCAST, SL, MVT::i64, Vec);
30110b57cec5SDimitry Andric }
30120b57cec5SDimitry Andric 
30130b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performShlCombine(SDNode *N,
30140b57cec5SDimitry Andric                                                 DAGCombinerInfo &DCI) const {
30150b57cec5SDimitry Andric   EVT VT = N->getValueType(0);
30160b57cec5SDimitry Andric 
30170b57cec5SDimitry Andric   ConstantSDNode *RHS = dyn_cast<ConstantSDNode>(N->getOperand(1));
30180b57cec5SDimitry Andric   if (!RHS)
30190b57cec5SDimitry Andric     return SDValue();
30200b57cec5SDimitry Andric 
30210b57cec5SDimitry Andric   SDValue LHS = N->getOperand(0);
30220b57cec5SDimitry Andric   unsigned RHSVal = RHS->getZExtValue();
30230b57cec5SDimitry Andric   if (!RHSVal)
30240b57cec5SDimitry Andric     return LHS;
30250b57cec5SDimitry Andric 
30260b57cec5SDimitry Andric   SDLoc SL(N);
30270b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
30280b57cec5SDimitry Andric 
30290b57cec5SDimitry Andric   switch (LHS->getOpcode()) {
30300b57cec5SDimitry Andric   default:
30310b57cec5SDimitry Andric     break;
30320b57cec5SDimitry Andric   case ISD::ZERO_EXTEND:
30330b57cec5SDimitry Andric   case ISD::SIGN_EXTEND:
30340b57cec5SDimitry Andric   case ISD::ANY_EXTEND: {
30350b57cec5SDimitry Andric     SDValue X = LHS->getOperand(0);
30360b57cec5SDimitry Andric 
30370b57cec5SDimitry Andric     if (VT == MVT::i32 && RHSVal == 16 && X.getValueType() == MVT::i16 &&
30380b57cec5SDimitry Andric         isOperationLegal(ISD::BUILD_VECTOR, MVT::v2i16)) {
30390b57cec5SDimitry Andric       // Prefer build_vector as the canonical form if packed types are legal.
30400b57cec5SDimitry Andric       // (shl ([asz]ext i16:x), 16 -> build_vector 0, x
30410b57cec5SDimitry Andric       SDValue Vec = DAG.getBuildVector(MVT::v2i16, SL,
30420b57cec5SDimitry Andric        { DAG.getConstant(0, SL, MVT::i16), LHS->getOperand(0) });
30430b57cec5SDimitry Andric       return DAG.getNode(ISD::BITCAST, SL, MVT::i32, Vec);
30440b57cec5SDimitry Andric     }
30450b57cec5SDimitry Andric 
30460b57cec5SDimitry Andric     // shl (ext x) => zext (shl x), if shift does not overflow int
30470b57cec5SDimitry Andric     if (VT != MVT::i64)
30480b57cec5SDimitry Andric       break;
30490b57cec5SDimitry Andric     KnownBits Known = DAG.computeKnownBits(X);
30500b57cec5SDimitry Andric     unsigned LZ = Known.countMinLeadingZeros();
30510b57cec5SDimitry Andric     if (LZ < RHSVal)
30520b57cec5SDimitry Andric       break;
30530b57cec5SDimitry Andric     EVT XVT = X.getValueType();
30540b57cec5SDimitry Andric     SDValue Shl = DAG.getNode(ISD::SHL, SL, XVT, X, SDValue(RHS, 0));
30550b57cec5SDimitry Andric     return DAG.getZExtOrTrunc(Shl, SL, VT);
30560b57cec5SDimitry Andric   }
30570b57cec5SDimitry Andric   }
30580b57cec5SDimitry Andric 
30590b57cec5SDimitry Andric   if (VT != MVT::i64)
30600b57cec5SDimitry Andric     return SDValue();
30610b57cec5SDimitry Andric 
30620b57cec5SDimitry Andric   // i64 (shl x, C) -> (build_pair 0, (shl x, C -32))
30630b57cec5SDimitry Andric 
30640b57cec5SDimitry Andric   // On some subtargets, 64-bit shift is a quarter rate instruction. In the
30650b57cec5SDimitry Andric   // common case, splitting this into a move and a 32-bit shift is faster and
30660b57cec5SDimitry Andric   // the same code size.
30670b57cec5SDimitry Andric   if (RHSVal < 32)
30680b57cec5SDimitry Andric     return SDValue();
30690b57cec5SDimitry Andric 
30700b57cec5SDimitry Andric   SDValue ShiftAmt = DAG.getConstant(RHSVal - 32, SL, MVT::i32);
30710b57cec5SDimitry Andric 
30720b57cec5SDimitry Andric   SDValue Lo = DAG.getNode(ISD::TRUNCATE, SL, MVT::i32, LHS);
30730b57cec5SDimitry Andric   SDValue NewShift = DAG.getNode(ISD::SHL, SL, MVT::i32, Lo, ShiftAmt);
30740b57cec5SDimitry Andric 
30750b57cec5SDimitry Andric   const SDValue Zero = DAG.getConstant(0, SL, MVT::i32);
30760b57cec5SDimitry Andric 
30770b57cec5SDimitry Andric   SDValue Vec = DAG.getBuildVector(MVT::v2i32, SL, {Zero, NewShift});
30780b57cec5SDimitry Andric   return DAG.getNode(ISD::BITCAST, SL, MVT::i64, Vec);
30790b57cec5SDimitry Andric }
30800b57cec5SDimitry Andric 
30810b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performSraCombine(SDNode *N,
30820b57cec5SDimitry Andric                                                 DAGCombinerInfo &DCI) const {
30830b57cec5SDimitry Andric   if (N->getValueType(0) != MVT::i64)
30840b57cec5SDimitry Andric     return SDValue();
30850b57cec5SDimitry Andric 
30860b57cec5SDimitry Andric   const ConstantSDNode *RHS = dyn_cast<ConstantSDNode>(N->getOperand(1));
30870b57cec5SDimitry Andric   if (!RHS)
30880b57cec5SDimitry Andric     return SDValue();
30890b57cec5SDimitry Andric 
30900b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
30910b57cec5SDimitry Andric   SDLoc SL(N);
30920b57cec5SDimitry Andric   unsigned RHSVal = RHS->getZExtValue();
30930b57cec5SDimitry Andric 
30940b57cec5SDimitry Andric   // (sra i64:x, 32) -> build_pair x, (sra hi_32(x), 31)
30950b57cec5SDimitry Andric   if (RHSVal == 32) {
30960b57cec5SDimitry Andric     SDValue Hi = getHiHalf64(N->getOperand(0), DAG);
30970b57cec5SDimitry Andric     SDValue NewShift = DAG.getNode(ISD::SRA, SL, MVT::i32, Hi,
30980b57cec5SDimitry Andric                                    DAG.getConstant(31, SL, MVT::i32));
30990b57cec5SDimitry Andric 
31000b57cec5SDimitry Andric     SDValue BuildVec = DAG.getBuildVector(MVT::v2i32, SL, {Hi, NewShift});
31010b57cec5SDimitry Andric     return DAG.getNode(ISD::BITCAST, SL, MVT::i64, BuildVec);
31020b57cec5SDimitry Andric   }
31030b57cec5SDimitry Andric 
31040b57cec5SDimitry Andric   // (sra i64:x, 63) -> build_pair (sra hi_32(x), 31), (sra hi_32(x), 31)
31050b57cec5SDimitry Andric   if (RHSVal == 63) {
31060b57cec5SDimitry Andric     SDValue Hi = getHiHalf64(N->getOperand(0), DAG);
31070b57cec5SDimitry Andric     SDValue NewShift = DAG.getNode(ISD::SRA, SL, MVT::i32, Hi,
31080b57cec5SDimitry Andric                                    DAG.getConstant(31, SL, MVT::i32));
31090b57cec5SDimitry Andric     SDValue BuildVec = DAG.getBuildVector(MVT::v2i32, SL, {NewShift, NewShift});
31100b57cec5SDimitry Andric     return DAG.getNode(ISD::BITCAST, SL, MVT::i64, BuildVec);
31110b57cec5SDimitry Andric   }
31120b57cec5SDimitry Andric 
31130b57cec5SDimitry Andric   return SDValue();
31140b57cec5SDimitry Andric }
31150b57cec5SDimitry Andric 
31160b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performSrlCombine(SDNode *N,
31170b57cec5SDimitry Andric                                                 DAGCombinerInfo &DCI) const {
31180b57cec5SDimitry Andric   auto *RHS = dyn_cast<ConstantSDNode>(N->getOperand(1));
31190b57cec5SDimitry Andric   if (!RHS)
31200b57cec5SDimitry Andric     return SDValue();
31210b57cec5SDimitry Andric 
31220b57cec5SDimitry Andric   EVT VT = N->getValueType(0);
31230b57cec5SDimitry Andric   SDValue LHS = N->getOperand(0);
31240b57cec5SDimitry Andric   unsigned ShiftAmt = RHS->getZExtValue();
31250b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
31260b57cec5SDimitry Andric   SDLoc SL(N);
31270b57cec5SDimitry Andric 
31280b57cec5SDimitry Andric   // fold (srl (and x, c1 << c2), c2) -> (and (srl(x, c2), c1)
31290b57cec5SDimitry Andric   // this improves the ability to match BFE patterns in isel.
31300b57cec5SDimitry Andric   if (LHS.getOpcode() == ISD::AND) {
31310b57cec5SDimitry Andric     if (auto *Mask = dyn_cast<ConstantSDNode>(LHS.getOperand(1))) {
31320b57cec5SDimitry Andric       if (Mask->getAPIntValue().isShiftedMask() &&
31330b57cec5SDimitry Andric           Mask->getAPIntValue().countTrailingZeros() == ShiftAmt) {
31340b57cec5SDimitry Andric         return DAG.getNode(
31350b57cec5SDimitry Andric             ISD::AND, SL, VT,
31360b57cec5SDimitry Andric             DAG.getNode(ISD::SRL, SL, VT, LHS.getOperand(0), N->getOperand(1)),
31370b57cec5SDimitry Andric             DAG.getNode(ISD::SRL, SL, VT, LHS.getOperand(1), N->getOperand(1)));
31380b57cec5SDimitry Andric       }
31390b57cec5SDimitry Andric     }
31400b57cec5SDimitry Andric   }
31410b57cec5SDimitry Andric 
31420b57cec5SDimitry Andric   if (VT != MVT::i64)
31430b57cec5SDimitry Andric     return SDValue();
31440b57cec5SDimitry Andric 
31450b57cec5SDimitry Andric   if (ShiftAmt < 32)
31460b57cec5SDimitry Andric     return SDValue();
31470b57cec5SDimitry Andric 
31480b57cec5SDimitry Andric   // srl i64:x, C for C >= 32
31490b57cec5SDimitry Andric   // =>
31500b57cec5SDimitry Andric   //   build_pair (srl hi_32(x), C - 32), 0
31510b57cec5SDimitry Andric   SDValue One = DAG.getConstant(1, SL, MVT::i32);
31520b57cec5SDimitry Andric   SDValue Zero = DAG.getConstant(0, SL, MVT::i32);
31530b57cec5SDimitry Andric 
31540b57cec5SDimitry Andric   SDValue VecOp = DAG.getNode(ISD::BITCAST, SL, MVT::v2i32, LHS);
31550b57cec5SDimitry Andric   SDValue Hi = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, MVT::i32, VecOp, One);
31560b57cec5SDimitry Andric 
31570b57cec5SDimitry Andric   SDValue NewConst = DAG.getConstant(ShiftAmt - 32, SL, MVT::i32);
31580b57cec5SDimitry Andric   SDValue NewShift = DAG.getNode(ISD::SRL, SL, MVT::i32, Hi, NewConst);
31590b57cec5SDimitry Andric 
31600b57cec5SDimitry Andric   SDValue BuildPair = DAG.getBuildVector(MVT::v2i32, SL, {NewShift, Zero});
31610b57cec5SDimitry Andric 
31620b57cec5SDimitry Andric   return DAG.getNode(ISD::BITCAST, SL, MVT::i64, BuildPair);
31630b57cec5SDimitry Andric }
31640b57cec5SDimitry Andric 
31650b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performTruncateCombine(
31660b57cec5SDimitry Andric   SDNode *N, DAGCombinerInfo &DCI) const {
31670b57cec5SDimitry Andric   SDLoc SL(N);
31680b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
31690b57cec5SDimitry Andric   EVT VT = N->getValueType(0);
31700b57cec5SDimitry Andric   SDValue Src = N->getOperand(0);
31710b57cec5SDimitry Andric 
31720b57cec5SDimitry Andric   // vt1 (truncate (bitcast (build_vector vt0:x, ...))) -> vt1 (bitcast vt0:x)
31730b57cec5SDimitry Andric   if (Src.getOpcode() == ISD::BITCAST && !VT.isVector()) {
31740b57cec5SDimitry Andric     SDValue Vec = Src.getOperand(0);
31750b57cec5SDimitry Andric     if (Vec.getOpcode() == ISD::BUILD_VECTOR) {
31760b57cec5SDimitry Andric       SDValue Elt0 = Vec.getOperand(0);
31770b57cec5SDimitry Andric       EVT EltVT = Elt0.getValueType();
31780b57cec5SDimitry Andric       if (VT.getSizeInBits() <= EltVT.getSizeInBits()) {
31790b57cec5SDimitry Andric         if (EltVT.isFloatingPoint()) {
31800b57cec5SDimitry Andric           Elt0 = DAG.getNode(ISD::BITCAST, SL,
31810b57cec5SDimitry Andric                              EltVT.changeTypeToInteger(), Elt0);
31820b57cec5SDimitry Andric         }
31830b57cec5SDimitry Andric 
31840b57cec5SDimitry Andric         return DAG.getNode(ISD::TRUNCATE, SL, VT, Elt0);
31850b57cec5SDimitry Andric       }
31860b57cec5SDimitry Andric     }
31870b57cec5SDimitry Andric   }
31880b57cec5SDimitry Andric 
31890b57cec5SDimitry Andric   // Equivalent of above for accessing the high element of a vector as an
31900b57cec5SDimitry Andric   // integer operation.
31910b57cec5SDimitry Andric   // trunc (srl (bitcast (build_vector x, y))), 16 -> trunc (bitcast y)
31920b57cec5SDimitry Andric   if (Src.getOpcode() == ISD::SRL && !VT.isVector()) {
31930b57cec5SDimitry Andric     if (auto K = isConstOrConstSplat(Src.getOperand(1))) {
31940b57cec5SDimitry Andric       if (2 * K->getZExtValue() == Src.getValueType().getScalarSizeInBits()) {
31950b57cec5SDimitry Andric         SDValue BV = stripBitcast(Src.getOperand(0));
31960b57cec5SDimitry Andric         if (BV.getOpcode() == ISD::BUILD_VECTOR &&
31970b57cec5SDimitry Andric             BV.getValueType().getVectorNumElements() == 2) {
31980b57cec5SDimitry Andric           SDValue SrcElt = BV.getOperand(1);
31990b57cec5SDimitry Andric           EVT SrcEltVT = SrcElt.getValueType();
32000b57cec5SDimitry Andric           if (SrcEltVT.isFloatingPoint()) {
32010b57cec5SDimitry Andric             SrcElt = DAG.getNode(ISD::BITCAST, SL,
32020b57cec5SDimitry Andric                                  SrcEltVT.changeTypeToInteger(), SrcElt);
32030b57cec5SDimitry Andric           }
32040b57cec5SDimitry Andric 
32050b57cec5SDimitry Andric           return DAG.getNode(ISD::TRUNCATE, SL, VT, SrcElt);
32060b57cec5SDimitry Andric         }
32070b57cec5SDimitry Andric       }
32080b57cec5SDimitry Andric     }
32090b57cec5SDimitry Andric   }
32100b57cec5SDimitry Andric 
32110b57cec5SDimitry Andric   // Partially shrink 64-bit shifts to 32-bit if reduced to 16-bit.
32120b57cec5SDimitry Andric   //
32130b57cec5SDimitry Andric   // i16 (trunc (srl i64:x, K)), K <= 16 ->
32140b57cec5SDimitry Andric   //     i16 (trunc (srl (i32 (trunc x), K)))
32150b57cec5SDimitry Andric   if (VT.getScalarSizeInBits() < 32) {
32160b57cec5SDimitry Andric     EVT SrcVT = Src.getValueType();
32170b57cec5SDimitry Andric     if (SrcVT.getScalarSizeInBits() > 32 &&
32180b57cec5SDimitry Andric         (Src.getOpcode() == ISD::SRL ||
32190b57cec5SDimitry Andric          Src.getOpcode() == ISD::SRA ||
32200b57cec5SDimitry Andric          Src.getOpcode() == ISD::SHL)) {
32210b57cec5SDimitry Andric       SDValue Amt = Src.getOperand(1);
32220b57cec5SDimitry Andric       KnownBits Known = DAG.computeKnownBits(Amt);
32230b57cec5SDimitry Andric       unsigned Size = VT.getScalarSizeInBits();
32240b57cec5SDimitry Andric       if ((Known.isConstant() && Known.getConstant().ule(Size)) ||
32250b57cec5SDimitry Andric           (Known.getBitWidth() - Known.countMinLeadingZeros() <= Log2_32(Size))) {
32260b57cec5SDimitry Andric         EVT MidVT = VT.isVector() ?
32270b57cec5SDimitry Andric           EVT::getVectorVT(*DAG.getContext(), MVT::i32,
32280b57cec5SDimitry Andric                            VT.getVectorNumElements()) : MVT::i32;
32290b57cec5SDimitry Andric 
32300b57cec5SDimitry Andric         EVT NewShiftVT = getShiftAmountTy(MidVT, DAG.getDataLayout());
32310b57cec5SDimitry Andric         SDValue Trunc = DAG.getNode(ISD::TRUNCATE, SL, MidVT,
32320b57cec5SDimitry Andric                                     Src.getOperand(0));
32330b57cec5SDimitry Andric         DCI.AddToWorklist(Trunc.getNode());
32340b57cec5SDimitry Andric 
32350b57cec5SDimitry Andric         if (Amt.getValueType() != NewShiftVT) {
32360b57cec5SDimitry Andric           Amt = DAG.getZExtOrTrunc(Amt, SL, NewShiftVT);
32370b57cec5SDimitry Andric           DCI.AddToWorklist(Amt.getNode());
32380b57cec5SDimitry Andric         }
32390b57cec5SDimitry Andric 
32400b57cec5SDimitry Andric         SDValue ShrunkShift = DAG.getNode(Src.getOpcode(), SL, MidVT,
32410b57cec5SDimitry Andric                                           Trunc, Amt);
32420b57cec5SDimitry Andric         return DAG.getNode(ISD::TRUNCATE, SL, VT, ShrunkShift);
32430b57cec5SDimitry Andric       }
32440b57cec5SDimitry Andric     }
32450b57cec5SDimitry Andric   }
32460b57cec5SDimitry Andric 
32470b57cec5SDimitry Andric   return SDValue();
32480b57cec5SDimitry Andric }
32490b57cec5SDimitry Andric 
32500b57cec5SDimitry Andric // We need to specifically handle i64 mul here to avoid unnecessary conversion
32510b57cec5SDimitry Andric // instructions. If we only match on the legalized i64 mul expansion,
32520b57cec5SDimitry Andric // SimplifyDemandedBits will be unable to remove them because there will be
32530b57cec5SDimitry Andric // multiple uses due to the separate mul + mulh[su].
32540b57cec5SDimitry Andric static SDValue getMul24(SelectionDAG &DAG, const SDLoc &SL,
32550b57cec5SDimitry Andric                         SDValue N0, SDValue N1, unsigned Size, bool Signed) {
32560b57cec5SDimitry Andric   if (Size <= 32) {
32570b57cec5SDimitry Andric     unsigned MulOpc = Signed ? AMDGPUISD::MUL_I24 : AMDGPUISD::MUL_U24;
32580b57cec5SDimitry Andric     return DAG.getNode(MulOpc, SL, MVT::i32, N0, N1);
32590b57cec5SDimitry Andric   }
32600b57cec5SDimitry Andric 
32610b57cec5SDimitry Andric   // Because we want to eliminate extension instructions before the
32620b57cec5SDimitry Andric   // operation, we need to create a single user here (i.e. not the separate
32630b57cec5SDimitry Andric   // mul_lo + mul_hi) so that SimplifyDemandedBits will deal with it.
32640b57cec5SDimitry Andric 
32650b57cec5SDimitry Andric   unsigned MulOpc = Signed ? AMDGPUISD::MUL_LOHI_I24 : AMDGPUISD::MUL_LOHI_U24;
32660b57cec5SDimitry Andric 
32670b57cec5SDimitry Andric   SDValue Mul = DAG.getNode(MulOpc, SL,
32680b57cec5SDimitry Andric                             DAG.getVTList(MVT::i32, MVT::i32), N0, N1);
32690b57cec5SDimitry Andric 
32700b57cec5SDimitry Andric   return DAG.getNode(ISD::BUILD_PAIR, SL, MVT::i64,
32710b57cec5SDimitry Andric                      Mul.getValue(0), Mul.getValue(1));
32720b57cec5SDimitry Andric }
32730b57cec5SDimitry Andric 
32740b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performMulCombine(SDNode *N,
32750b57cec5SDimitry Andric                                                 DAGCombinerInfo &DCI) const {
32760b57cec5SDimitry Andric   EVT VT = N->getValueType(0);
32770b57cec5SDimitry Andric 
32780b57cec5SDimitry Andric   unsigned Size = VT.getSizeInBits();
32790b57cec5SDimitry Andric   if (VT.isVector() || Size > 64)
32800b57cec5SDimitry Andric     return SDValue();
32810b57cec5SDimitry Andric 
32820b57cec5SDimitry Andric   // There are i16 integer mul/mad.
32830b57cec5SDimitry Andric   if (Subtarget->has16BitInsts() && VT.getScalarType().bitsLE(MVT::i16))
32840b57cec5SDimitry Andric     return SDValue();
32850b57cec5SDimitry Andric 
32860b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
32870b57cec5SDimitry Andric   SDLoc DL(N);
32880b57cec5SDimitry Andric 
32890b57cec5SDimitry Andric   SDValue N0 = N->getOperand(0);
32900b57cec5SDimitry Andric   SDValue N1 = N->getOperand(1);
32910b57cec5SDimitry Andric 
32920b57cec5SDimitry Andric   // SimplifyDemandedBits has the annoying habit of turning useful zero_extends
32930b57cec5SDimitry Andric   // in the source into any_extends if the result of the mul is truncated. Since
32940b57cec5SDimitry Andric   // we can assume the high bits are whatever we want, use the underlying value
32950b57cec5SDimitry Andric   // to avoid the unknown high bits from interfering.
32960b57cec5SDimitry Andric   if (N0.getOpcode() == ISD::ANY_EXTEND)
32970b57cec5SDimitry Andric     N0 = N0.getOperand(0);
32980b57cec5SDimitry Andric 
32990b57cec5SDimitry Andric   if (N1.getOpcode() == ISD::ANY_EXTEND)
33000b57cec5SDimitry Andric     N1 = N1.getOperand(0);
33010b57cec5SDimitry Andric 
33020b57cec5SDimitry Andric   SDValue Mul;
33030b57cec5SDimitry Andric 
33040b57cec5SDimitry Andric   if (Subtarget->hasMulU24() && isU24(N0, DAG) && isU24(N1, DAG)) {
33050b57cec5SDimitry Andric     N0 = DAG.getZExtOrTrunc(N0, DL, MVT::i32);
33060b57cec5SDimitry Andric     N1 = DAG.getZExtOrTrunc(N1, DL, MVT::i32);
33070b57cec5SDimitry Andric     Mul = getMul24(DAG, DL, N0, N1, Size, false);
33080b57cec5SDimitry Andric   } else if (Subtarget->hasMulI24() && isI24(N0, DAG) && isI24(N1, DAG)) {
33090b57cec5SDimitry Andric     N0 = DAG.getSExtOrTrunc(N0, DL, MVT::i32);
33100b57cec5SDimitry Andric     N1 = DAG.getSExtOrTrunc(N1, DL, MVT::i32);
33110b57cec5SDimitry Andric     Mul = getMul24(DAG, DL, N0, N1, Size, true);
33120b57cec5SDimitry Andric   } else {
33130b57cec5SDimitry Andric     return SDValue();
33140b57cec5SDimitry Andric   }
33150b57cec5SDimitry Andric 
33160b57cec5SDimitry Andric   // We need to use sext even for MUL_U24, because MUL_U24 is used
33170b57cec5SDimitry Andric   // for signed multiply of 8 and 16-bit types.
33180b57cec5SDimitry Andric   return DAG.getSExtOrTrunc(Mul, DL, VT);
33190b57cec5SDimitry Andric }
33200b57cec5SDimitry Andric 
33210b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performMulhsCombine(SDNode *N,
33220b57cec5SDimitry Andric                                                   DAGCombinerInfo &DCI) const {
33230b57cec5SDimitry Andric   EVT VT = N->getValueType(0);
33240b57cec5SDimitry Andric 
33250b57cec5SDimitry Andric   if (!Subtarget->hasMulI24() || VT.isVector())
33260b57cec5SDimitry Andric     return SDValue();
33270b57cec5SDimitry Andric 
33280b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
33290b57cec5SDimitry Andric   SDLoc DL(N);
33300b57cec5SDimitry Andric 
33310b57cec5SDimitry Andric   SDValue N0 = N->getOperand(0);
33320b57cec5SDimitry Andric   SDValue N1 = N->getOperand(1);
33330b57cec5SDimitry Andric 
33340b57cec5SDimitry Andric   if (!isI24(N0, DAG) || !isI24(N1, DAG))
33350b57cec5SDimitry Andric     return SDValue();
33360b57cec5SDimitry Andric 
33370b57cec5SDimitry Andric   N0 = DAG.getSExtOrTrunc(N0, DL, MVT::i32);
33380b57cec5SDimitry Andric   N1 = DAG.getSExtOrTrunc(N1, DL, MVT::i32);
33390b57cec5SDimitry Andric 
33400b57cec5SDimitry Andric   SDValue Mulhi = DAG.getNode(AMDGPUISD::MULHI_I24, DL, MVT::i32, N0, N1);
33410b57cec5SDimitry Andric   DCI.AddToWorklist(Mulhi.getNode());
33420b57cec5SDimitry Andric   return DAG.getSExtOrTrunc(Mulhi, DL, VT);
33430b57cec5SDimitry Andric }
33440b57cec5SDimitry Andric 
33450b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performMulhuCombine(SDNode *N,
33460b57cec5SDimitry Andric                                                   DAGCombinerInfo &DCI) const {
33470b57cec5SDimitry Andric   EVT VT = N->getValueType(0);
33480b57cec5SDimitry Andric 
33490b57cec5SDimitry Andric   if (!Subtarget->hasMulU24() || VT.isVector() || VT.getSizeInBits() > 32)
33500b57cec5SDimitry Andric     return SDValue();
33510b57cec5SDimitry Andric 
33520b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
33530b57cec5SDimitry Andric   SDLoc DL(N);
33540b57cec5SDimitry Andric 
33550b57cec5SDimitry Andric   SDValue N0 = N->getOperand(0);
33560b57cec5SDimitry Andric   SDValue N1 = N->getOperand(1);
33570b57cec5SDimitry Andric 
33580b57cec5SDimitry Andric   if (!isU24(N0, DAG) || !isU24(N1, DAG))
33590b57cec5SDimitry Andric     return SDValue();
33600b57cec5SDimitry Andric 
33610b57cec5SDimitry Andric   N0 = DAG.getZExtOrTrunc(N0, DL, MVT::i32);
33620b57cec5SDimitry Andric   N1 = DAG.getZExtOrTrunc(N1, DL, MVT::i32);
33630b57cec5SDimitry Andric 
33640b57cec5SDimitry Andric   SDValue Mulhi = DAG.getNode(AMDGPUISD::MULHI_U24, DL, MVT::i32, N0, N1);
33650b57cec5SDimitry Andric   DCI.AddToWorklist(Mulhi.getNode());
33660b57cec5SDimitry Andric   return DAG.getZExtOrTrunc(Mulhi, DL, VT);
33670b57cec5SDimitry Andric }
33680b57cec5SDimitry Andric 
33690b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performMulLoHi24Combine(
33700b57cec5SDimitry Andric   SDNode *N, DAGCombinerInfo &DCI) const {
33710b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
33720b57cec5SDimitry Andric 
33730b57cec5SDimitry Andric   // Simplify demanded bits before splitting into multiple users.
33740b57cec5SDimitry Andric   if (SDValue V = simplifyI24(N, DCI))
33750b57cec5SDimitry Andric     return V;
33760b57cec5SDimitry Andric 
33770b57cec5SDimitry Andric   SDValue N0 = N->getOperand(0);
33780b57cec5SDimitry Andric   SDValue N1 = N->getOperand(1);
33790b57cec5SDimitry Andric 
33800b57cec5SDimitry Andric   bool Signed = (N->getOpcode() == AMDGPUISD::MUL_LOHI_I24);
33810b57cec5SDimitry Andric 
33820b57cec5SDimitry Andric   unsigned MulLoOpc = Signed ? AMDGPUISD::MUL_I24 : AMDGPUISD::MUL_U24;
33830b57cec5SDimitry Andric   unsigned MulHiOpc = Signed ? AMDGPUISD::MULHI_I24 : AMDGPUISD::MULHI_U24;
33840b57cec5SDimitry Andric 
33850b57cec5SDimitry Andric   SDLoc SL(N);
33860b57cec5SDimitry Andric 
33870b57cec5SDimitry Andric   SDValue MulLo = DAG.getNode(MulLoOpc, SL, MVT::i32, N0, N1);
33880b57cec5SDimitry Andric   SDValue MulHi = DAG.getNode(MulHiOpc, SL, MVT::i32, N0, N1);
33890b57cec5SDimitry Andric   return DAG.getMergeValues({ MulLo, MulHi }, SL);
33900b57cec5SDimitry Andric }
33910b57cec5SDimitry Andric 
33920b57cec5SDimitry Andric static bool isNegativeOne(SDValue Val) {
33930b57cec5SDimitry Andric   if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Val))
33940b57cec5SDimitry Andric     return C->isAllOnesValue();
33950b57cec5SDimitry Andric   return false;
33960b57cec5SDimitry Andric }
33970b57cec5SDimitry Andric 
33980b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::getFFBX_U32(SelectionDAG &DAG,
33990b57cec5SDimitry Andric                                           SDValue Op,
34000b57cec5SDimitry Andric                                           const SDLoc &DL,
34010b57cec5SDimitry Andric                                           unsigned Opc) const {
34020b57cec5SDimitry Andric   EVT VT = Op.getValueType();
34030b57cec5SDimitry Andric   EVT LegalVT = getTypeToTransformTo(*DAG.getContext(), VT);
34040b57cec5SDimitry Andric   if (LegalVT != MVT::i32 && (Subtarget->has16BitInsts() &&
34050b57cec5SDimitry Andric                               LegalVT != MVT::i16))
34060b57cec5SDimitry Andric     return SDValue();
34070b57cec5SDimitry Andric 
34080b57cec5SDimitry Andric   if (VT != MVT::i32)
34090b57cec5SDimitry Andric     Op = DAG.getNode(ISD::ZERO_EXTEND, DL, MVT::i32, Op);
34100b57cec5SDimitry Andric 
34110b57cec5SDimitry Andric   SDValue FFBX = DAG.getNode(Opc, DL, MVT::i32, Op);
34120b57cec5SDimitry Andric   if (VT != MVT::i32)
34130b57cec5SDimitry Andric     FFBX = DAG.getNode(ISD::TRUNCATE, DL, VT, FFBX);
34140b57cec5SDimitry Andric 
34150b57cec5SDimitry Andric   return FFBX;
34160b57cec5SDimitry Andric }
34170b57cec5SDimitry Andric 
34180b57cec5SDimitry Andric // The native instructions return -1 on 0 input. Optimize out a select that
34190b57cec5SDimitry Andric // produces -1 on 0.
34200b57cec5SDimitry Andric //
34210b57cec5SDimitry Andric // TODO: If zero is not undef, we could also do this if the output is compared
34220b57cec5SDimitry Andric // against the bitwidth.
34230b57cec5SDimitry Andric //
34240b57cec5SDimitry Andric // TODO: Should probably combine against FFBH_U32 instead of ctlz directly.
34250b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performCtlz_CttzCombine(const SDLoc &SL, SDValue Cond,
34260b57cec5SDimitry Andric                                                  SDValue LHS, SDValue RHS,
34270b57cec5SDimitry Andric                                                  DAGCombinerInfo &DCI) const {
34280b57cec5SDimitry Andric   ConstantSDNode *CmpRhs = dyn_cast<ConstantSDNode>(Cond.getOperand(1));
34290b57cec5SDimitry Andric   if (!CmpRhs || !CmpRhs->isNullValue())
34300b57cec5SDimitry Andric     return SDValue();
34310b57cec5SDimitry Andric 
34320b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
34330b57cec5SDimitry Andric   ISD::CondCode CCOpcode = cast<CondCodeSDNode>(Cond.getOperand(2))->get();
34340b57cec5SDimitry Andric   SDValue CmpLHS = Cond.getOperand(0);
34350b57cec5SDimitry Andric 
34360b57cec5SDimitry Andric   unsigned Opc = isCttzOpc(RHS.getOpcode()) ? AMDGPUISD::FFBL_B32 :
34370b57cec5SDimitry Andric                                            AMDGPUISD::FFBH_U32;
34380b57cec5SDimitry Andric 
34390b57cec5SDimitry Andric   // select (setcc x, 0, eq), -1, (ctlz_zero_undef x) -> ffbh_u32 x
34400b57cec5SDimitry Andric   // select (setcc x, 0, eq), -1, (cttz_zero_undef x) -> ffbl_u32 x
34410b57cec5SDimitry Andric   if (CCOpcode == ISD::SETEQ &&
34420b57cec5SDimitry Andric       (isCtlzOpc(RHS.getOpcode()) || isCttzOpc(RHS.getOpcode())) &&
34430b57cec5SDimitry Andric       RHS.getOperand(0) == CmpLHS &&
34440b57cec5SDimitry Andric       isNegativeOne(LHS)) {
34450b57cec5SDimitry Andric     return getFFBX_U32(DAG, CmpLHS, SL, Opc);
34460b57cec5SDimitry Andric   }
34470b57cec5SDimitry Andric 
34480b57cec5SDimitry Andric   // select (setcc x, 0, ne), (ctlz_zero_undef x), -1 -> ffbh_u32 x
34490b57cec5SDimitry Andric   // select (setcc x, 0, ne), (cttz_zero_undef x), -1 -> ffbl_u32 x
34500b57cec5SDimitry Andric   if (CCOpcode == ISD::SETNE &&
34510b57cec5SDimitry Andric       (isCtlzOpc(LHS.getOpcode()) || isCttzOpc(RHS.getOpcode())) &&
34520b57cec5SDimitry Andric       LHS.getOperand(0) == CmpLHS &&
34530b57cec5SDimitry Andric       isNegativeOne(RHS)) {
34540b57cec5SDimitry Andric     return getFFBX_U32(DAG, CmpLHS, SL, Opc);
34550b57cec5SDimitry Andric   }
34560b57cec5SDimitry Andric 
34570b57cec5SDimitry Andric   return SDValue();
34580b57cec5SDimitry Andric }
34590b57cec5SDimitry Andric 
34600b57cec5SDimitry Andric static SDValue distributeOpThroughSelect(TargetLowering::DAGCombinerInfo &DCI,
34610b57cec5SDimitry Andric                                          unsigned Op,
34620b57cec5SDimitry Andric                                          const SDLoc &SL,
34630b57cec5SDimitry Andric                                          SDValue Cond,
34640b57cec5SDimitry Andric                                          SDValue N1,
34650b57cec5SDimitry Andric                                          SDValue N2) {
34660b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
34670b57cec5SDimitry Andric   EVT VT = N1.getValueType();
34680b57cec5SDimitry Andric 
34690b57cec5SDimitry Andric   SDValue NewSelect = DAG.getNode(ISD::SELECT, SL, VT, Cond,
34700b57cec5SDimitry Andric                                   N1.getOperand(0), N2.getOperand(0));
34710b57cec5SDimitry Andric   DCI.AddToWorklist(NewSelect.getNode());
34720b57cec5SDimitry Andric   return DAG.getNode(Op, SL, VT, NewSelect);
34730b57cec5SDimitry Andric }
34740b57cec5SDimitry Andric 
34750b57cec5SDimitry Andric // Pull a free FP operation out of a select so it may fold into uses.
34760b57cec5SDimitry Andric //
34770b57cec5SDimitry Andric // select c, (fneg x), (fneg y) -> fneg (select c, x, y)
34780b57cec5SDimitry Andric // select c, (fneg x), k -> fneg (select c, x, (fneg k))
34790b57cec5SDimitry Andric //
34800b57cec5SDimitry Andric // select c, (fabs x), (fabs y) -> fabs (select c, x, y)
34810b57cec5SDimitry Andric // select c, (fabs x), +k -> fabs (select c, x, k)
34820b57cec5SDimitry Andric static SDValue foldFreeOpFromSelect(TargetLowering::DAGCombinerInfo &DCI,
34830b57cec5SDimitry Andric                                     SDValue N) {
34840b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
34850b57cec5SDimitry Andric   SDValue Cond = N.getOperand(0);
34860b57cec5SDimitry Andric   SDValue LHS = N.getOperand(1);
34870b57cec5SDimitry Andric   SDValue RHS = N.getOperand(2);
34880b57cec5SDimitry Andric 
34890b57cec5SDimitry Andric   EVT VT = N.getValueType();
34900b57cec5SDimitry Andric   if ((LHS.getOpcode() == ISD::FABS && RHS.getOpcode() == ISD::FABS) ||
34910b57cec5SDimitry Andric       (LHS.getOpcode() == ISD::FNEG && RHS.getOpcode() == ISD::FNEG)) {
34920b57cec5SDimitry Andric     return distributeOpThroughSelect(DCI, LHS.getOpcode(),
34930b57cec5SDimitry Andric                                      SDLoc(N), Cond, LHS, RHS);
34940b57cec5SDimitry Andric   }
34950b57cec5SDimitry Andric 
34960b57cec5SDimitry Andric   bool Inv = false;
34970b57cec5SDimitry Andric   if (RHS.getOpcode() == ISD::FABS || RHS.getOpcode() == ISD::FNEG) {
34980b57cec5SDimitry Andric     std::swap(LHS, RHS);
34990b57cec5SDimitry Andric     Inv = true;
35000b57cec5SDimitry Andric   }
35010b57cec5SDimitry Andric 
35020b57cec5SDimitry Andric   // TODO: Support vector constants.
35030b57cec5SDimitry Andric   ConstantFPSDNode *CRHS = dyn_cast<ConstantFPSDNode>(RHS);
35040b57cec5SDimitry Andric   if ((LHS.getOpcode() == ISD::FNEG || LHS.getOpcode() == ISD::FABS) && CRHS) {
35050b57cec5SDimitry Andric     SDLoc SL(N);
35060b57cec5SDimitry Andric     // If one side is an fneg/fabs and the other is a constant, we can push the
35070b57cec5SDimitry Andric     // fneg/fabs down. If it's an fabs, the constant needs to be non-negative.
35080b57cec5SDimitry Andric     SDValue NewLHS = LHS.getOperand(0);
35090b57cec5SDimitry Andric     SDValue NewRHS = RHS;
35100b57cec5SDimitry Andric 
35110b57cec5SDimitry Andric     // Careful: if the neg can be folded up, don't try to pull it back down.
35120b57cec5SDimitry Andric     bool ShouldFoldNeg = true;
35130b57cec5SDimitry Andric 
35140b57cec5SDimitry Andric     if (NewLHS.hasOneUse()) {
35150b57cec5SDimitry Andric       unsigned Opc = NewLHS.getOpcode();
35160b57cec5SDimitry Andric       if (LHS.getOpcode() == ISD::FNEG && fnegFoldsIntoOp(Opc))
35170b57cec5SDimitry Andric         ShouldFoldNeg = false;
35180b57cec5SDimitry Andric       if (LHS.getOpcode() == ISD::FABS && Opc == ISD::FMUL)
35190b57cec5SDimitry Andric         ShouldFoldNeg = false;
35200b57cec5SDimitry Andric     }
35210b57cec5SDimitry Andric 
35220b57cec5SDimitry Andric     if (ShouldFoldNeg) {
35230b57cec5SDimitry Andric       if (LHS.getOpcode() == ISD::FNEG)
35240b57cec5SDimitry Andric         NewRHS = DAG.getNode(ISD::FNEG, SL, VT, RHS);
35250b57cec5SDimitry Andric       else if (CRHS->isNegative())
35260b57cec5SDimitry Andric         return SDValue();
35270b57cec5SDimitry Andric 
35280b57cec5SDimitry Andric       if (Inv)
35290b57cec5SDimitry Andric         std::swap(NewLHS, NewRHS);
35300b57cec5SDimitry Andric 
35310b57cec5SDimitry Andric       SDValue NewSelect = DAG.getNode(ISD::SELECT, SL, VT,
35320b57cec5SDimitry Andric                                       Cond, NewLHS, NewRHS);
35330b57cec5SDimitry Andric       DCI.AddToWorklist(NewSelect.getNode());
35340b57cec5SDimitry Andric       return DAG.getNode(LHS.getOpcode(), SL, VT, NewSelect);
35350b57cec5SDimitry Andric     }
35360b57cec5SDimitry Andric   }
35370b57cec5SDimitry Andric 
35380b57cec5SDimitry Andric   return SDValue();
35390b57cec5SDimitry Andric }
35400b57cec5SDimitry Andric 
35410b57cec5SDimitry Andric 
35420b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performSelectCombine(SDNode *N,
35430b57cec5SDimitry Andric                                                    DAGCombinerInfo &DCI) const {
35440b57cec5SDimitry Andric   if (SDValue Folded = foldFreeOpFromSelect(DCI, SDValue(N, 0)))
35450b57cec5SDimitry Andric     return Folded;
35460b57cec5SDimitry Andric 
35470b57cec5SDimitry Andric   SDValue Cond = N->getOperand(0);
35480b57cec5SDimitry Andric   if (Cond.getOpcode() != ISD::SETCC)
35490b57cec5SDimitry Andric     return SDValue();
35500b57cec5SDimitry Andric 
35510b57cec5SDimitry Andric   EVT VT = N->getValueType(0);
35520b57cec5SDimitry Andric   SDValue LHS = Cond.getOperand(0);
35530b57cec5SDimitry Andric   SDValue RHS = Cond.getOperand(1);
35540b57cec5SDimitry Andric   SDValue CC = Cond.getOperand(2);
35550b57cec5SDimitry Andric 
35560b57cec5SDimitry Andric   SDValue True = N->getOperand(1);
35570b57cec5SDimitry Andric   SDValue False = N->getOperand(2);
35580b57cec5SDimitry Andric 
35590b57cec5SDimitry Andric   if (Cond.hasOneUse()) { // TODO: Look for multiple select uses.
35600b57cec5SDimitry Andric     SelectionDAG &DAG = DCI.DAG;
35610b57cec5SDimitry Andric     if (DAG.isConstantValueOfAnyType(True) &&
35620b57cec5SDimitry Andric         !DAG.isConstantValueOfAnyType(False)) {
35630b57cec5SDimitry Andric       // Swap cmp + select pair to move constant to false input.
35640b57cec5SDimitry Andric       // This will allow using VOPC cndmasks more often.
35650b57cec5SDimitry Andric       // select (setcc x, y), k, x -> select (setccinv x, y), x, k
35660b57cec5SDimitry Andric 
35670b57cec5SDimitry Andric       SDLoc SL(N);
35680b57cec5SDimitry Andric       ISD::CondCode NewCC = getSetCCInverse(cast<CondCodeSDNode>(CC)->get(),
35690b57cec5SDimitry Andric                                             LHS.getValueType().isInteger());
35700b57cec5SDimitry Andric 
35710b57cec5SDimitry Andric       SDValue NewCond = DAG.getSetCC(SL, Cond.getValueType(), LHS, RHS, NewCC);
35720b57cec5SDimitry Andric       return DAG.getNode(ISD::SELECT, SL, VT, NewCond, False, True);
35730b57cec5SDimitry Andric     }
35740b57cec5SDimitry Andric 
35750b57cec5SDimitry Andric     if (VT == MVT::f32 && Subtarget->hasFminFmaxLegacy()) {
35760b57cec5SDimitry Andric       SDValue MinMax
35770b57cec5SDimitry Andric         = combineFMinMaxLegacy(SDLoc(N), VT, LHS, RHS, True, False, CC, DCI);
35780b57cec5SDimitry Andric       // Revisit this node so we can catch min3/max3/med3 patterns.
35790b57cec5SDimitry Andric       //DCI.AddToWorklist(MinMax.getNode());
35800b57cec5SDimitry Andric       return MinMax;
35810b57cec5SDimitry Andric     }
35820b57cec5SDimitry Andric   }
35830b57cec5SDimitry Andric 
35840b57cec5SDimitry Andric   // There's no reason to not do this if the condition has other uses.
35850b57cec5SDimitry Andric   return performCtlz_CttzCombine(SDLoc(N), Cond, True, False, DCI);
35860b57cec5SDimitry Andric }
35870b57cec5SDimitry Andric 
35880b57cec5SDimitry Andric static bool isInv2Pi(const APFloat &APF) {
35890b57cec5SDimitry Andric   static const APFloat KF16(APFloat::IEEEhalf(), APInt(16, 0x3118));
35900b57cec5SDimitry Andric   static const APFloat KF32(APFloat::IEEEsingle(), APInt(32, 0x3e22f983));
35910b57cec5SDimitry Andric   static const APFloat KF64(APFloat::IEEEdouble(), APInt(64, 0x3fc45f306dc9c882));
35920b57cec5SDimitry Andric 
35930b57cec5SDimitry Andric   return APF.bitwiseIsEqual(KF16) ||
35940b57cec5SDimitry Andric          APF.bitwiseIsEqual(KF32) ||
35950b57cec5SDimitry Andric          APF.bitwiseIsEqual(KF64);
35960b57cec5SDimitry Andric }
35970b57cec5SDimitry Andric 
35980b57cec5SDimitry Andric // 0 and 1.0 / (0.5 * pi) do not have inline immmediates, so there is an
35990b57cec5SDimitry Andric // additional cost to negate them.
36000b57cec5SDimitry Andric bool AMDGPUTargetLowering::isConstantCostlierToNegate(SDValue N) const {
36010b57cec5SDimitry Andric   if (const ConstantFPSDNode *C = isConstOrConstSplatFP(N)) {
36020b57cec5SDimitry Andric     if (C->isZero() && !C->isNegative())
36030b57cec5SDimitry Andric       return true;
36040b57cec5SDimitry Andric 
36050b57cec5SDimitry Andric     if (Subtarget->hasInv2PiInlineImm() && isInv2Pi(C->getValueAPF()))
36060b57cec5SDimitry Andric       return true;
36070b57cec5SDimitry Andric   }
36080b57cec5SDimitry Andric 
36090b57cec5SDimitry Andric   return false;
36100b57cec5SDimitry Andric }
36110b57cec5SDimitry Andric 
36120b57cec5SDimitry Andric static unsigned inverseMinMax(unsigned Opc) {
36130b57cec5SDimitry Andric   switch (Opc) {
36140b57cec5SDimitry Andric   case ISD::FMAXNUM:
36150b57cec5SDimitry Andric     return ISD::FMINNUM;
36160b57cec5SDimitry Andric   case ISD::FMINNUM:
36170b57cec5SDimitry Andric     return ISD::FMAXNUM;
36180b57cec5SDimitry Andric   case ISD::FMAXNUM_IEEE:
36190b57cec5SDimitry Andric     return ISD::FMINNUM_IEEE;
36200b57cec5SDimitry Andric   case ISD::FMINNUM_IEEE:
36210b57cec5SDimitry Andric     return ISD::FMAXNUM_IEEE;
36220b57cec5SDimitry Andric   case AMDGPUISD::FMAX_LEGACY:
36230b57cec5SDimitry Andric     return AMDGPUISD::FMIN_LEGACY;
36240b57cec5SDimitry Andric   case AMDGPUISD::FMIN_LEGACY:
36250b57cec5SDimitry Andric     return  AMDGPUISD::FMAX_LEGACY;
36260b57cec5SDimitry Andric   default:
36270b57cec5SDimitry Andric     llvm_unreachable("invalid min/max opcode");
36280b57cec5SDimitry Andric   }
36290b57cec5SDimitry Andric }
36300b57cec5SDimitry Andric 
36310b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performFNegCombine(SDNode *N,
36320b57cec5SDimitry Andric                                                  DAGCombinerInfo &DCI) const {
36330b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
36340b57cec5SDimitry Andric   SDValue N0 = N->getOperand(0);
36350b57cec5SDimitry Andric   EVT VT = N->getValueType(0);
36360b57cec5SDimitry Andric 
36370b57cec5SDimitry Andric   unsigned Opc = N0.getOpcode();
36380b57cec5SDimitry Andric 
36390b57cec5SDimitry Andric   // If the input has multiple uses and we can either fold the negate down, or
36400b57cec5SDimitry Andric   // the other uses cannot, give up. This both prevents unprofitable
36410b57cec5SDimitry Andric   // transformations and infinite loops: we won't repeatedly try to fold around
36420b57cec5SDimitry Andric   // a negate that has no 'good' form.
36430b57cec5SDimitry Andric   if (N0.hasOneUse()) {
36440b57cec5SDimitry Andric     // This may be able to fold into the source, but at a code size cost. Don't
36450b57cec5SDimitry Andric     // fold if the fold into the user is free.
36460b57cec5SDimitry Andric     if (allUsesHaveSourceMods(N, 0))
36470b57cec5SDimitry Andric       return SDValue();
36480b57cec5SDimitry Andric   } else {
36490b57cec5SDimitry Andric     if (fnegFoldsIntoOp(Opc) &&
36500b57cec5SDimitry Andric         (allUsesHaveSourceMods(N) || !allUsesHaveSourceMods(N0.getNode())))
36510b57cec5SDimitry Andric       return SDValue();
36520b57cec5SDimitry Andric   }
36530b57cec5SDimitry Andric 
36540b57cec5SDimitry Andric   SDLoc SL(N);
36550b57cec5SDimitry Andric   switch (Opc) {
36560b57cec5SDimitry Andric   case ISD::FADD: {
36570b57cec5SDimitry Andric     if (!mayIgnoreSignedZero(N0))
36580b57cec5SDimitry Andric       return SDValue();
36590b57cec5SDimitry Andric 
36600b57cec5SDimitry Andric     // (fneg (fadd x, y)) -> (fadd (fneg x), (fneg y))
36610b57cec5SDimitry Andric     SDValue LHS = N0.getOperand(0);
36620b57cec5SDimitry Andric     SDValue RHS = N0.getOperand(1);
36630b57cec5SDimitry Andric 
36640b57cec5SDimitry Andric     if (LHS.getOpcode() != ISD::FNEG)
36650b57cec5SDimitry Andric       LHS = DAG.getNode(ISD::FNEG, SL, VT, LHS);
36660b57cec5SDimitry Andric     else
36670b57cec5SDimitry Andric       LHS = LHS.getOperand(0);
36680b57cec5SDimitry Andric 
36690b57cec5SDimitry Andric     if (RHS.getOpcode() != ISD::FNEG)
36700b57cec5SDimitry Andric       RHS = DAG.getNode(ISD::FNEG, SL, VT, RHS);
36710b57cec5SDimitry Andric     else
36720b57cec5SDimitry Andric       RHS = RHS.getOperand(0);
36730b57cec5SDimitry Andric 
36740b57cec5SDimitry Andric     SDValue Res = DAG.getNode(ISD::FADD, SL, VT, LHS, RHS, N0->getFlags());
36750b57cec5SDimitry Andric     if (Res.getOpcode() != ISD::FADD)
36760b57cec5SDimitry Andric       return SDValue(); // Op got folded away.
36770b57cec5SDimitry Andric     if (!N0.hasOneUse())
36780b57cec5SDimitry Andric       DAG.ReplaceAllUsesWith(N0, DAG.getNode(ISD::FNEG, SL, VT, Res));
36790b57cec5SDimitry Andric     return Res;
36800b57cec5SDimitry Andric   }
36810b57cec5SDimitry Andric   case ISD::FMUL:
36820b57cec5SDimitry Andric   case AMDGPUISD::FMUL_LEGACY: {
36830b57cec5SDimitry Andric     // (fneg (fmul x, y)) -> (fmul x, (fneg y))
36840b57cec5SDimitry Andric     // (fneg (fmul_legacy x, y)) -> (fmul_legacy x, (fneg y))
36850b57cec5SDimitry Andric     SDValue LHS = N0.getOperand(0);
36860b57cec5SDimitry Andric     SDValue RHS = N0.getOperand(1);
36870b57cec5SDimitry Andric 
36880b57cec5SDimitry Andric     if (LHS.getOpcode() == ISD::FNEG)
36890b57cec5SDimitry Andric       LHS = LHS.getOperand(0);
36900b57cec5SDimitry Andric     else if (RHS.getOpcode() == ISD::FNEG)
36910b57cec5SDimitry Andric       RHS = RHS.getOperand(0);
36920b57cec5SDimitry Andric     else
36930b57cec5SDimitry Andric       RHS = DAG.getNode(ISD::FNEG, SL, VT, RHS);
36940b57cec5SDimitry Andric 
36950b57cec5SDimitry Andric     SDValue Res = DAG.getNode(Opc, SL, VT, LHS, RHS, N0->getFlags());
36960b57cec5SDimitry Andric     if (Res.getOpcode() != Opc)
36970b57cec5SDimitry Andric       return SDValue(); // Op got folded away.
36980b57cec5SDimitry Andric     if (!N0.hasOneUse())
36990b57cec5SDimitry Andric       DAG.ReplaceAllUsesWith(N0, DAG.getNode(ISD::FNEG, SL, VT, Res));
37000b57cec5SDimitry Andric     return Res;
37010b57cec5SDimitry Andric   }
37020b57cec5SDimitry Andric   case ISD::FMA:
37030b57cec5SDimitry Andric   case ISD::FMAD: {
37040b57cec5SDimitry Andric     if (!mayIgnoreSignedZero(N0))
37050b57cec5SDimitry Andric       return SDValue();
37060b57cec5SDimitry Andric 
37070b57cec5SDimitry Andric     // (fneg (fma x, y, z)) -> (fma x, (fneg y), (fneg z))
37080b57cec5SDimitry Andric     SDValue LHS = N0.getOperand(0);
37090b57cec5SDimitry Andric     SDValue MHS = N0.getOperand(1);
37100b57cec5SDimitry Andric     SDValue RHS = N0.getOperand(2);
37110b57cec5SDimitry Andric 
37120b57cec5SDimitry Andric     if (LHS.getOpcode() == ISD::FNEG)
37130b57cec5SDimitry Andric       LHS = LHS.getOperand(0);
37140b57cec5SDimitry Andric     else if (MHS.getOpcode() == ISD::FNEG)
37150b57cec5SDimitry Andric       MHS = MHS.getOperand(0);
37160b57cec5SDimitry Andric     else
37170b57cec5SDimitry Andric       MHS = DAG.getNode(ISD::FNEG, SL, VT, MHS);
37180b57cec5SDimitry Andric 
37190b57cec5SDimitry Andric     if (RHS.getOpcode() != ISD::FNEG)
37200b57cec5SDimitry Andric       RHS = DAG.getNode(ISD::FNEG, SL, VT, RHS);
37210b57cec5SDimitry Andric     else
37220b57cec5SDimitry Andric       RHS = RHS.getOperand(0);
37230b57cec5SDimitry Andric 
37240b57cec5SDimitry Andric     SDValue Res = DAG.getNode(Opc, SL, VT, LHS, MHS, RHS);
37250b57cec5SDimitry Andric     if (Res.getOpcode() != Opc)
37260b57cec5SDimitry Andric       return SDValue(); // Op got folded away.
37270b57cec5SDimitry Andric     if (!N0.hasOneUse())
37280b57cec5SDimitry Andric       DAG.ReplaceAllUsesWith(N0, DAG.getNode(ISD::FNEG, SL, VT, Res));
37290b57cec5SDimitry Andric     return Res;
37300b57cec5SDimitry Andric   }
37310b57cec5SDimitry Andric   case ISD::FMAXNUM:
37320b57cec5SDimitry Andric   case ISD::FMINNUM:
37330b57cec5SDimitry Andric   case ISD::FMAXNUM_IEEE:
37340b57cec5SDimitry Andric   case ISD::FMINNUM_IEEE:
37350b57cec5SDimitry Andric   case AMDGPUISD::FMAX_LEGACY:
37360b57cec5SDimitry Andric   case AMDGPUISD::FMIN_LEGACY: {
37370b57cec5SDimitry Andric     // fneg (fmaxnum x, y) -> fminnum (fneg x), (fneg y)
37380b57cec5SDimitry Andric     // fneg (fminnum x, y) -> fmaxnum (fneg x), (fneg y)
37390b57cec5SDimitry Andric     // fneg (fmax_legacy x, y) -> fmin_legacy (fneg x), (fneg y)
37400b57cec5SDimitry Andric     // fneg (fmin_legacy x, y) -> fmax_legacy (fneg x), (fneg y)
37410b57cec5SDimitry Andric 
37420b57cec5SDimitry Andric     SDValue LHS = N0.getOperand(0);
37430b57cec5SDimitry Andric     SDValue RHS = N0.getOperand(1);
37440b57cec5SDimitry Andric 
37450b57cec5SDimitry Andric     // 0 doesn't have a negated inline immediate.
37460b57cec5SDimitry Andric     // TODO: This constant check should be generalized to other operations.
37470b57cec5SDimitry Andric     if (isConstantCostlierToNegate(RHS))
37480b57cec5SDimitry Andric       return SDValue();
37490b57cec5SDimitry Andric 
37500b57cec5SDimitry Andric     SDValue NegLHS = DAG.getNode(ISD::FNEG, SL, VT, LHS);
37510b57cec5SDimitry Andric     SDValue NegRHS = DAG.getNode(ISD::FNEG, SL, VT, RHS);
37520b57cec5SDimitry Andric     unsigned Opposite = inverseMinMax(Opc);
37530b57cec5SDimitry Andric 
37540b57cec5SDimitry Andric     SDValue Res = DAG.getNode(Opposite, SL, VT, NegLHS, NegRHS, N0->getFlags());
37550b57cec5SDimitry Andric     if (Res.getOpcode() != Opposite)
37560b57cec5SDimitry Andric       return SDValue(); // Op got folded away.
37570b57cec5SDimitry Andric     if (!N0.hasOneUse())
37580b57cec5SDimitry Andric       DAG.ReplaceAllUsesWith(N0, DAG.getNode(ISD::FNEG, SL, VT, Res));
37590b57cec5SDimitry Andric     return Res;
37600b57cec5SDimitry Andric   }
37610b57cec5SDimitry Andric   case AMDGPUISD::FMED3: {
37620b57cec5SDimitry Andric     SDValue Ops[3];
37630b57cec5SDimitry Andric     for (unsigned I = 0; I < 3; ++I)
37640b57cec5SDimitry Andric       Ops[I] = DAG.getNode(ISD::FNEG, SL, VT, N0->getOperand(I), N0->getFlags());
37650b57cec5SDimitry Andric 
37660b57cec5SDimitry Andric     SDValue Res = DAG.getNode(AMDGPUISD::FMED3, SL, VT, Ops, N0->getFlags());
37670b57cec5SDimitry Andric     if (Res.getOpcode() != AMDGPUISD::FMED3)
37680b57cec5SDimitry Andric       return SDValue(); // Op got folded away.
37690b57cec5SDimitry Andric     if (!N0.hasOneUse())
37700b57cec5SDimitry Andric       DAG.ReplaceAllUsesWith(N0, DAG.getNode(ISD::FNEG, SL, VT, Res));
37710b57cec5SDimitry Andric     return Res;
37720b57cec5SDimitry Andric   }
37730b57cec5SDimitry Andric   case ISD::FP_EXTEND:
37740b57cec5SDimitry Andric   case ISD::FTRUNC:
37750b57cec5SDimitry Andric   case ISD::FRINT:
37760b57cec5SDimitry Andric   case ISD::FNEARBYINT: // XXX - Should fround be handled?
37770b57cec5SDimitry Andric   case ISD::FSIN:
37780b57cec5SDimitry Andric   case ISD::FCANONICALIZE:
37790b57cec5SDimitry Andric   case AMDGPUISD::RCP:
37800b57cec5SDimitry Andric   case AMDGPUISD::RCP_LEGACY:
37810b57cec5SDimitry Andric   case AMDGPUISD::RCP_IFLAG:
37820b57cec5SDimitry Andric   case AMDGPUISD::SIN_HW: {
37830b57cec5SDimitry Andric     SDValue CvtSrc = N0.getOperand(0);
37840b57cec5SDimitry Andric     if (CvtSrc.getOpcode() == ISD::FNEG) {
37850b57cec5SDimitry Andric       // (fneg (fp_extend (fneg x))) -> (fp_extend x)
37860b57cec5SDimitry Andric       // (fneg (rcp (fneg x))) -> (rcp x)
37870b57cec5SDimitry Andric       return DAG.getNode(Opc, SL, VT, CvtSrc.getOperand(0));
37880b57cec5SDimitry Andric     }
37890b57cec5SDimitry Andric 
37900b57cec5SDimitry Andric     if (!N0.hasOneUse())
37910b57cec5SDimitry Andric       return SDValue();
37920b57cec5SDimitry Andric 
37930b57cec5SDimitry Andric     // (fneg (fp_extend x)) -> (fp_extend (fneg x))
37940b57cec5SDimitry Andric     // (fneg (rcp x)) -> (rcp (fneg x))
37950b57cec5SDimitry Andric     SDValue Neg = DAG.getNode(ISD::FNEG, SL, CvtSrc.getValueType(), CvtSrc);
37960b57cec5SDimitry Andric     return DAG.getNode(Opc, SL, VT, Neg, N0->getFlags());
37970b57cec5SDimitry Andric   }
37980b57cec5SDimitry Andric   case ISD::FP_ROUND: {
37990b57cec5SDimitry Andric     SDValue CvtSrc = N0.getOperand(0);
38000b57cec5SDimitry Andric 
38010b57cec5SDimitry Andric     if (CvtSrc.getOpcode() == ISD::FNEG) {
38020b57cec5SDimitry Andric       // (fneg (fp_round (fneg x))) -> (fp_round x)
38030b57cec5SDimitry Andric       return DAG.getNode(ISD::FP_ROUND, SL, VT,
38040b57cec5SDimitry Andric                          CvtSrc.getOperand(0), N0.getOperand(1));
38050b57cec5SDimitry Andric     }
38060b57cec5SDimitry Andric 
38070b57cec5SDimitry Andric     if (!N0.hasOneUse())
38080b57cec5SDimitry Andric       return SDValue();
38090b57cec5SDimitry Andric 
38100b57cec5SDimitry Andric     // (fneg (fp_round x)) -> (fp_round (fneg x))
38110b57cec5SDimitry Andric     SDValue Neg = DAG.getNode(ISD::FNEG, SL, CvtSrc.getValueType(), CvtSrc);
38120b57cec5SDimitry Andric     return DAG.getNode(ISD::FP_ROUND, SL, VT, Neg, N0.getOperand(1));
38130b57cec5SDimitry Andric   }
38140b57cec5SDimitry Andric   case ISD::FP16_TO_FP: {
38150b57cec5SDimitry Andric     // v_cvt_f32_f16 supports source modifiers on pre-VI targets without legal
38160b57cec5SDimitry Andric     // f16, but legalization of f16 fneg ends up pulling it out of the source.
38170b57cec5SDimitry Andric     // Put the fneg back as a legal source operation that can be matched later.
38180b57cec5SDimitry Andric     SDLoc SL(N);
38190b57cec5SDimitry Andric 
38200b57cec5SDimitry Andric     SDValue Src = N0.getOperand(0);
38210b57cec5SDimitry Andric     EVT SrcVT = Src.getValueType();
38220b57cec5SDimitry Andric 
38230b57cec5SDimitry Andric     // fneg (fp16_to_fp x) -> fp16_to_fp (xor x, 0x8000)
38240b57cec5SDimitry Andric     SDValue IntFNeg = DAG.getNode(ISD::XOR, SL, SrcVT, Src,
38250b57cec5SDimitry Andric                                   DAG.getConstant(0x8000, SL, SrcVT));
38260b57cec5SDimitry Andric     return DAG.getNode(ISD::FP16_TO_FP, SL, N->getValueType(0), IntFNeg);
38270b57cec5SDimitry Andric   }
38280b57cec5SDimitry Andric   default:
38290b57cec5SDimitry Andric     return SDValue();
38300b57cec5SDimitry Andric   }
38310b57cec5SDimitry Andric }
38320b57cec5SDimitry Andric 
38330b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performFAbsCombine(SDNode *N,
38340b57cec5SDimitry Andric                                                  DAGCombinerInfo &DCI) const {
38350b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
38360b57cec5SDimitry Andric   SDValue N0 = N->getOperand(0);
38370b57cec5SDimitry Andric 
38380b57cec5SDimitry Andric   if (!N0.hasOneUse())
38390b57cec5SDimitry Andric     return SDValue();
38400b57cec5SDimitry Andric 
38410b57cec5SDimitry Andric   switch (N0.getOpcode()) {
38420b57cec5SDimitry Andric   case ISD::FP16_TO_FP: {
38430b57cec5SDimitry Andric     assert(!Subtarget->has16BitInsts() && "should only see if f16 is illegal");
38440b57cec5SDimitry Andric     SDLoc SL(N);
38450b57cec5SDimitry Andric     SDValue Src = N0.getOperand(0);
38460b57cec5SDimitry Andric     EVT SrcVT = Src.getValueType();
38470b57cec5SDimitry Andric 
38480b57cec5SDimitry Andric     // fabs (fp16_to_fp x) -> fp16_to_fp (and x, 0x7fff)
38490b57cec5SDimitry Andric     SDValue IntFAbs = DAG.getNode(ISD::AND, SL, SrcVT, Src,
38500b57cec5SDimitry Andric                                   DAG.getConstant(0x7fff, SL, SrcVT));
38510b57cec5SDimitry Andric     return DAG.getNode(ISD::FP16_TO_FP, SL, N->getValueType(0), IntFAbs);
38520b57cec5SDimitry Andric   }
38530b57cec5SDimitry Andric   default:
38540b57cec5SDimitry Andric     return SDValue();
38550b57cec5SDimitry Andric   }
38560b57cec5SDimitry Andric }
38570b57cec5SDimitry Andric 
38580b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::performRcpCombine(SDNode *N,
38590b57cec5SDimitry Andric                                                 DAGCombinerInfo &DCI) const {
38600b57cec5SDimitry Andric   const auto *CFP = dyn_cast<ConstantFPSDNode>(N->getOperand(0));
38610b57cec5SDimitry Andric   if (!CFP)
38620b57cec5SDimitry Andric     return SDValue();
38630b57cec5SDimitry Andric 
38640b57cec5SDimitry Andric   // XXX - Should this flush denormals?
38650b57cec5SDimitry Andric   const APFloat &Val = CFP->getValueAPF();
38660b57cec5SDimitry Andric   APFloat One(Val.getSemantics(), "1.0");
38670b57cec5SDimitry Andric   return DCI.DAG.getConstantFP(One / Val, SDLoc(N), N->getValueType(0));
38680b57cec5SDimitry Andric }
38690b57cec5SDimitry Andric 
38700b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::PerformDAGCombine(SDNode *N,
38710b57cec5SDimitry Andric                                                 DAGCombinerInfo &DCI) const {
38720b57cec5SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
38730b57cec5SDimitry Andric   SDLoc DL(N);
38740b57cec5SDimitry Andric 
38750b57cec5SDimitry Andric   switch(N->getOpcode()) {
38760b57cec5SDimitry Andric   default:
38770b57cec5SDimitry Andric     break;
38780b57cec5SDimitry Andric   case ISD::BITCAST: {
38790b57cec5SDimitry Andric     EVT DestVT = N->getValueType(0);
38800b57cec5SDimitry Andric 
38810b57cec5SDimitry Andric     // Push casts through vector builds. This helps avoid emitting a large
38820b57cec5SDimitry Andric     // number of copies when materializing floating point vector constants.
38830b57cec5SDimitry Andric     //
38840b57cec5SDimitry Andric     // vNt1 bitcast (vNt0 (build_vector t0:x, t0:y)) =>
38850b57cec5SDimitry Andric     //   vnt1 = build_vector (t1 (bitcast t0:x)), (t1 (bitcast t0:y))
38860b57cec5SDimitry Andric     if (DestVT.isVector()) {
38870b57cec5SDimitry Andric       SDValue Src = N->getOperand(0);
38880b57cec5SDimitry Andric       if (Src.getOpcode() == ISD::BUILD_VECTOR) {
38890b57cec5SDimitry Andric         EVT SrcVT = Src.getValueType();
38900b57cec5SDimitry Andric         unsigned NElts = DestVT.getVectorNumElements();
38910b57cec5SDimitry Andric 
38920b57cec5SDimitry Andric         if (SrcVT.getVectorNumElements() == NElts) {
38930b57cec5SDimitry Andric           EVT DestEltVT = DestVT.getVectorElementType();
38940b57cec5SDimitry Andric 
38950b57cec5SDimitry Andric           SmallVector<SDValue, 8> CastedElts;
38960b57cec5SDimitry Andric           SDLoc SL(N);
38970b57cec5SDimitry Andric           for (unsigned I = 0, E = SrcVT.getVectorNumElements(); I != E; ++I) {
38980b57cec5SDimitry Andric             SDValue Elt = Src.getOperand(I);
38990b57cec5SDimitry Andric             CastedElts.push_back(DAG.getNode(ISD::BITCAST, DL, DestEltVT, Elt));
39000b57cec5SDimitry Andric           }
39010b57cec5SDimitry Andric 
39020b57cec5SDimitry Andric           return DAG.getBuildVector(DestVT, SL, CastedElts);
39030b57cec5SDimitry Andric         }
39040b57cec5SDimitry Andric       }
39050b57cec5SDimitry Andric     }
39060b57cec5SDimitry Andric 
39070b57cec5SDimitry Andric     if (DestVT.getSizeInBits() != 64 && !DestVT.isVector())
39080b57cec5SDimitry Andric       break;
39090b57cec5SDimitry Andric 
39100b57cec5SDimitry Andric     // Fold bitcasts of constants.
39110b57cec5SDimitry Andric     //
39120b57cec5SDimitry Andric     // v2i32 (bitcast i64:k) -> build_vector lo_32(k), hi_32(k)
39130b57cec5SDimitry Andric     // TODO: Generalize and move to DAGCombiner
39140b57cec5SDimitry Andric     SDValue Src = N->getOperand(0);
39150b57cec5SDimitry Andric     if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Src)) {
39160b57cec5SDimitry Andric       if (Src.getValueType() == MVT::i64) {
39170b57cec5SDimitry Andric         SDLoc SL(N);
39180b57cec5SDimitry Andric         uint64_t CVal = C->getZExtValue();
39190b57cec5SDimitry Andric         SDValue BV = DAG.getNode(ISD::BUILD_VECTOR, SL, MVT::v2i32,
39200b57cec5SDimitry Andric                                  DAG.getConstant(Lo_32(CVal), SL, MVT::i32),
39210b57cec5SDimitry Andric                                  DAG.getConstant(Hi_32(CVal), SL, MVT::i32));
39220b57cec5SDimitry Andric         return DAG.getNode(ISD::BITCAST, SL, DestVT, BV);
39230b57cec5SDimitry Andric       }
39240b57cec5SDimitry Andric     }
39250b57cec5SDimitry Andric 
39260b57cec5SDimitry Andric     if (ConstantFPSDNode *C = dyn_cast<ConstantFPSDNode>(Src)) {
39270b57cec5SDimitry Andric       const APInt &Val = C->getValueAPF().bitcastToAPInt();
39280b57cec5SDimitry Andric       SDLoc SL(N);
39290b57cec5SDimitry Andric       uint64_t CVal = Val.getZExtValue();
39300b57cec5SDimitry Andric       SDValue Vec = DAG.getNode(ISD::BUILD_VECTOR, SL, MVT::v2i32,
39310b57cec5SDimitry Andric                                 DAG.getConstant(Lo_32(CVal), SL, MVT::i32),
39320b57cec5SDimitry Andric                                 DAG.getConstant(Hi_32(CVal), SL, MVT::i32));
39330b57cec5SDimitry Andric 
39340b57cec5SDimitry Andric       return DAG.getNode(ISD::BITCAST, SL, DestVT, Vec);
39350b57cec5SDimitry Andric     }
39360b57cec5SDimitry Andric 
39370b57cec5SDimitry Andric     break;
39380b57cec5SDimitry Andric   }
39390b57cec5SDimitry Andric   case ISD::SHL: {
39400b57cec5SDimitry Andric     if (DCI.getDAGCombineLevel() < AfterLegalizeDAG)
39410b57cec5SDimitry Andric       break;
39420b57cec5SDimitry Andric 
39430b57cec5SDimitry Andric     return performShlCombine(N, DCI);
39440b57cec5SDimitry Andric   }
39450b57cec5SDimitry Andric   case ISD::SRL: {
39460b57cec5SDimitry Andric     if (DCI.getDAGCombineLevel() < AfterLegalizeDAG)
39470b57cec5SDimitry Andric       break;
39480b57cec5SDimitry Andric 
39490b57cec5SDimitry Andric     return performSrlCombine(N, DCI);
39500b57cec5SDimitry Andric   }
39510b57cec5SDimitry Andric   case ISD::SRA: {
39520b57cec5SDimitry Andric     if (DCI.getDAGCombineLevel() < AfterLegalizeDAG)
39530b57cec5SDimitry Andric       break;
39540b57cec5SDimitry Andric 
39550b57cec5SDimitry Andric     return performSraCombine(N, DCI);
39560b57cec5SDimitry Andric   }
39570b57cec5SDimitry Andric   case ISD::TRUNCATE:
39580b57cec5SDimitry Andric     return performTruncateCombine(N, DCI);
39590b57cec5SDimitry Andric   case ISD::MUL:
39600b57cec5SDimitry Andric     return performMulCombine(N, DCI);
39610b57cec5SDimitry Andric   case ISD::MULHS:
39620b57cec5SDimitry Andric     return performMulhsCombine(N, DCI);
39630b57cec5SDimitry Andric   case ISD::MULHU:
39640b57cec5SDimitry Andric     return performMulhuCombine(N, DCI);
39650b57cec5SDimitry Andric   case AMDGPUISD::MUL_I24:
39660b57cec5SDimitry Andric   case AMDGPUISD::MUL_U24:
39670b57cec5SDimitry Andric   case AMDGPUISD::MULHI_I24:
39680b57cec5SDimitry Andric   case AMDGPUISD::MULHI_U24: {
39690b57cec5SDimitry Andric     if (SDValue V = simplifyI24(N, DCI))
39700b57cec5SDimitry Andric       return V;
39710b57cec5SDimitry Andric     return SDValue();
39720b57cec5SDimitry Andric   }
39730b57cec5SDimitry Andric   case AMDGPUISD::MUL_LOHI_I24:
39740b57cec5SDimitry Andric   case AMDGPUISD::MUL_LOHI_U24:
39750b57cec5SDimitry Andric     return performMulLoHi24Combine(N, DCI);
39760b57cec5SDimitry Andric   case ISD::SELECT:
39770b57cec5SDimitry Andric     return performSelectCombine(N, DCI);
39780b57cec5SDimitry Andric   case ISD::FNEG:
39790b57cec5SDimitry Andric     return performFNegCombine(N, DCI);
39800b57cec5SDimitry Andric   case ISD::FABS:
39810b57cec5SDimitry Andric     return performFAbsCombine(N, DCI);
39820b57cec5SDimitry Andric   case AMDGPUISD::BFE_I32:
39830b57cec5SDimitry Andric   case AMDGPUISD::BFE_U32: {
39840b57cec5SDimitry Andric     assert(!N->getValueType(0).isVector() &&
39850b57cec5SDimitry Andric            "Vector handling of BFE not implemented");
39860b57cec5SDimitry Andric     ConstantSDNode *Width = dyn_cast<ConstantSDNode>(N->getOperand(2));
39870b57cec5SDimitry Andric     if (!Width)
39880b57cec5SDimitry Andric       break;
39890b57cec5SDimitry Andric 
39900b57cec5SDimitry Andric     uint32_t WidthVal = Width->getZExtValue() & 0x1f;
39910b57cec5SDimitry Andric     if (WidthVal == 0)
39920b57cec5SDimitry Andric       return DAG.getConstant(0, DL, MVT::i32);
39930b57cec5SDimitry Andric 
39940b57cec5SDimitry Andric     ConstantSDNode *Offset = dyn_cast<ConstantSDNode>(N->getOperand(1));
39950b57cec5SDimitry Andric     if (!Offset)
39960b57cec5SDimitry Andric       break;
39970b57cec5SDimitry Andric 
39980b57cec5SDimitry Andric     SDValue BitsFrom = N->getOperand(0);
39990b57cec5SDimitry Andric     uint32_t OffsetVal = Offset->getZExtValue() & 0x1f;
40000b57cec5SDimitry Andric 
40010b57cec5SDimitry Andric     bool Signed = N->getOpcode() == AMDGPUISD::BFE_I32;
40020b57cec5SDimitry Andric 
40030b57cec5SDimitry Andric     if (OffsetVal == 0) {
40040b57cec5SDimitry Andric       // This is already sign / zero extended, so try to fold away extra BFEs.
40050b57cec5SDimitry Andric       unsigned SignBits =  Signed ? (32 - WidthVal + 1) : (32 - WidthVal);
40060b57cec5SDimitry Andric 
40070b57cec5SDimitry Andric       unsigned OpSignBits = DAG.ComputeNumSignBits(BitsFrom);
40080b57cec5SDimitry Andric       if (OpSignBits >= SignBits)
40090b57cec5SDimitry Andric         return BitsFrom;
40100b57cec5SDimitry Andric 
40110b57cec5SDimitry Andric       EVT SmallVT = EVT::getIntegerVT(*DAG.getContext(), WidthVal);
40120b57cec5SDimitry Andric       if (Signed) {
40130b57cec5SDimitry Andric         // This is a sign_extend_inreg. Replace it to take advantage of existing
40140b57cec5SDimitry Andric         // DAG Combines. If not eliminated, we will match back to BFE during
40150b57cec5SDimitry Andric         // selection.
40160b57cec5SDimitry Andric 
40170b57cec5SDimitry Andric         // TODO: The sext_inreg of extended types ends, although we can could
40180b57cec5SDimitry Andric         // handle them in a single BFE.
40190b57cec5SDimitry Andric         return DAG.getNode(ISD::SIGN_EXTEND_INREG, DL, MVT::i32, BitsFrom,
40200b57cec5SDimitry Andric                            DAG.getValueType(SmallVT));
40210b57cec5SDimitry Andric       }
40220b57cec5SDimitry Andric 
40230b57cec5SDimitry Andric       return DAG.getZeroExtendInReg(BitsFrom, DL, SmallVT);
40240b57cec5SDimitry Andric     }
40250b57cec5SDimitry Andric 
40260b57cec5SDimitry Andric     if (ConstantSDNode *CVal = dyn_cast<ConstantSDNode>(BitsFrom)) {
40270b57cec5SDimitry Andric       if (Signed) {
40280b57cec5SDimitry Andric         return constantFoldBFE<int32_t>(DAG,
40290b57cec5SDimitry Andric                                         CVal->getSExtValue(),
40300b57cec5SDimitry Andric                                         OffsetVal,
40310b57cec5SDimitry Andric                                         WidthVal,
40320b57cec5SDimitry Andric                                         DL);
40330b57cec5SDimitry Andric       }
40340b57cec5SDimitry Andric 
40350b57cec5SDimitry Andric       return constantFoldBFE<uint32_t>(DAG,
40360b57cec5SDimitry Andric                                        CVal->getZExtValue(),
40370b57cec5SDimitry Andric                                        OffsetVal,
40380b57cec5SDimitry Andric                                        WidthVal,
40390b57cec5SDimitry Andric                                        DL);
40400b57cec5SDimitry Andric     }
40410b57cec5SDimitry Andric 
40420b57cec5SDimitry Andric     if ((OffsetVal + WidthVal) >= 32 &&
40430b57cec5SDimitry Andric         !(Subtarget->hasSDWA() && OffsetVal == 16 && WidthVal == 16)) {
40440b57cec5SDimitry Andric       SDValue ShiftVal = DAG.getConstant(OffsetVal, DL, MVT::i32);
40450b57cec5SDimitry Andric       return DAG.getNode(Signed ? ISD::SRA : ISD::SRL, DL, MVT::i32,
40460b57cec5SDimitry Andric                          BitsFrom, ShiftVal);
40470b57cec5SDimitry Andric     }
40480b57cec5SDimitry Andric 
40490b57cec5SDimitry Andric     if (BitsFrom.hasOneUse()) {
40500b57cec5SDimitry Andric       APInt Demanded = APInt::getBitsSet(32,
40510b57cec5SDimitry Andric                                          OffsetVal,
40520b57cec5SDimitry Andric                                          OffsetVal + WidthVal);
40530b57cec5SDimitry Andric 
40540b57cec5SDimitry Andric       KnownBits Known;
40550b57cec5SDimitry Andric       TargetLowering::TargetLoweringOpt TLO(DAG, !DCI.isBeforeLegalize(),
40560b57cec5SDimitry Andric                                             !DCI.isBeforeLegalizeOps());
40570b57cec5SDimitry Andric       const TargetLowering &TLI = DAG.getTargetLoweringInfo();
40580b57cec5SDimitry Andric       if (TLI.ShrinkDemandedConstant(BitsFrom, Demanded, TLO) ||
40590b57cec5SDimitry Andric           TLI.SimplifyDemandedBits(BitsFrom, Demanded, Known, TLO)) {
40600b57cec5SDimitry Andric         DCI.CommitTargetLoweringOpt(TLO);
40610b57cec5SDimitry Andric       }
40620b57cec5SDimitry Andric     }
40630b57cec5SDimitry Andric 
40640b57cec5SDimitry Andric     break;
40650b57cec5SDimitry Andric   }
40660b57cec5SDimitry Andric   case ISD::LOAD:
40670b57cec5SDimitry Andric     return performLoadCombine(N, DCI);
40680b57cec5SDimitry Andric   case ISD::STORE:
40690b57cec5SDimitry Andric     return performStoreCombine(N, DCI);
40700b57cec5SDimitry Andric   case AMDGPUISD::RCP:
40710b57cec5SDimitry Andric   case AMDGPUISD::RCP_IFLAG:
40720b57cec5SDimitry Andric     return performRcpCombine(N, DCI);
40730b57cec5SDimitry Andric   case ISD::AssertZext:
40740b57cec5SDimitry Andric   case ISD::AssertSext:
40750b57cec5SDimitry Andric     return performAssertSZExtCombine(N, DCI);
4076*8bcb0991SDimitry Andric   case ISD::INTRINSIC_WO_CHAIN:
4077*8bcb0991SDimitry Andric     return performIntrinsicWOChainCombine(N, DCI);
40780b57cec5SDimitry Andric   }
40790b57cec5SDimitry Andric   return SDValue();
40800b57cec5SDimitry Andric }
40810b57cec5SDimitry Andric 
40820b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
40830b57cec5SDimitry Andric // Helper functions
40840b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
40850b57cec5SDimitry Andric 
40860b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::CreateLiveInRegister(SelectionDAG &DAG,
40870b57cec5SDimitry Andric                                                    const TargetRegisterClass *RC,
40880b57cec5SDimitry Andric                                                    unsigned Reg, EVT VT,
40890b57cec5SDimitry Andric                                                    const SDLoc &SL,
40900b57cec5SDimitry Andric                                                    bool RawReg) const {
40910b57cec5SDimitry Andric   MachineFunction &MF = DAG.getMachineFunction();
40920b57cec5SDimitry Andric   MachineRegisterInfo &MRI = MF.getRegInfo();
40930b57cec5SDimitry Andric   unsigned VReg;
40940b57cec5SDimitry Andric 
40950b57cec5SDimitry Andric   if (!MRI.isLiveIn(Reg)) {
40960b57cec5SDimitry Andric     VReg = MRI.createVirtualRegister(RC);
40970b57cec5SDimitry Andric     MRI.addLiveIn(Reg, VReg);
40980b57cec5SDimitry Andric   } else {
40990b57cec5SDimitry Andric     VReg = MRI.getLiveInVirtReg(Reg);
41000b57cec5SDimitry Andric   }
41010b57cec5SDimitry Andric 
41020b57cec5SDimitry Andric   if (RawReg)
41030b57cec5SDimitry Andric     return DAG.getRegister(VReg, VT);
41040b57cec5SDimitry Andric 
41050b57cec5SDimitry Andric   return DAG.getCopyFromReg(DAG.getEntryNode(), SL, VReg, VT);
41060b57cec5SDimitry Andric }
41070b57cec5SDimitry Andric 
4108*8bcb0991SDimitry Andric // This may be called multiple times, and nothing prevents creating multiple
4109*8bcb0991SDimitry Andric // objects at the same offset. See if we already defined this object.
4110*8bcb0991SDimitry Andric static int getOrCreateFixedStackObject(MachineFrameInfo &MFI, unsigned Size,
4111*8bcb0991SDimitry Andric                                        int64_t Offset) {
4112*8bcb0991SDimitry Andric   for (int I = MFI.getObjectIndexBegin(); I < 0; ++I) {
4113*8bcb0991SDimitry Andric     if (MFI.getObjectOffset(I) == Offset) {
4114*8bcb0991SDimitry Andric       assert(MFI.getObjectSize(I) == Size);
4115*8bcb0991SDimitry Andric       return I;
4116*8bcb0991SDimitry Andric     }
4117*8bcb0991SDimitry Andric   }
4118*8bcb0991SDimitry Andric 
4119*8bcb0991SDimitry Andric   return MFI.CreateFixedObject(Size, Offset, true);
4120*8bcb0991SDimitry Andric }
4121*8bcb0991SDimitry Andric 
41220b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::loadStackInputValue(SelectionDAG &DAG,
41230b57cec5SDimitry Andric                                                   EVT VT,
41240b57cec5SDimitry Andric                                                   const SDLoc &SL,
41250b57cec5SDimitry Andric                                                   int64_t Offset) const {
41260b57cec5SDimitry Andric   MachineFunction &MF = DAG.getMachineFunction();
41270b57cec5SDimitry Andric   MachineFrameInfo &MFI = MF.getFrameInfo();
4128*8bcb0991SDimitry Andric   int FI = getOrCreateFixedStackObject(MFI, VT.getStoreSize(), Offset);
41290b57cec5SDimitry Andric 
41300b57cec5SDimitry Andric   auto SrcPtrInfo = MachinePointerInfo::getStack(MF, Offset);
41310b57cec5SDimitry Andric   SDValue Ptr = DAG.getFrameIndex(FI, MVT::i32);
41320b57cec5SDimitry Andric 
41330b57cec5SDimitry Andric   return DAG.getLoad(VT, SL, DAG.getEntryNode(), Ptr, SrcPtrInfo, 4,
41340b57cec5SDimitry Andric                      MachineMemOperand::MODereferenceable |
41350b57cec5SDimitry Andric                      MachineMemOperand::MOInvariant);
41360b57cec5SDimitry Andric }
41370b57cec5SDimitry Andric 
41380b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::storeStackInputValue(SelectionDAG &DAG,
41390b57cec5SDimitry Andric                                                    const SDLoc &SL,
41400b57cec5SDimitry Andric                                                    SDValue Chain,
41410b57cec5SDimitry Andric                                                    SDValue ArgVal,
41420b57cec5SDimitry Andric                                                    int64_t Offset) const {
41430b57cec5SDimitry Andric   MachineFunction &MF = DAG.getMachineFunction();
41440b57cec5SDimitry Andric   MachinePointerInfo DstInfo = MachinePointerInfo::getStack(MF, Offset);
41450b57cec5SDimitry Andric 
41460b57cec5SDimitry Andric   SDValue Ptr = DAG.getConstant(Offset, SL, MVT::i32);
41470b57cec5SDimitry Andric   SDValue Store = DAG.getStore(Chain, SL, ArgVal, Ptr, DstInfo, 4,
41480b57cec5SDimitry Andric                                MachineMemOperand::MODereferenceable);
41490b57cec5SDimitry Andric   return Store;
41500b57cec5SDimitry Andric }
41510b57cec5SDimitry Andric 
41520b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::loadInputValue(SelectionDAG &DAG,
41530b57cec5SDimitry Andric                                              const TargetRegisterClass *RC,
41540b57cec5SDimitry Andric                                              EVT VT, const SDLoc &SL,
41550b57cec5SDimitry Andric                                              const ArgDescriptor &Arg) const {
41560b57cec5SDimitry Andric   assert(Arg && "Attempting to load missing argument");
41570b57cec5SDimitry Andric 
41580b57cec5SDimitry Andric   SDValue V = Arg.isRegister() ?
41590b57cec5SDimitry Andric     CreateLiveInRegister(DAG, RC, Arg.getRegister(), VT, SL) :
41600b57cec5SDimitry Andric     loadStackInputValue(DAG, VT, SL, Arg.getStackOffset());
41610b57cec5SDimitry Andric 
41620b57cec5SDimitry Andric   if (!Arg.isMasked())
41630b57cec5SDimitry Andric     return V;
41640b57cec5SDimitry Andric 
41650b57cec5SDimitry Andric   unsigned Mask = Arg.getMask();
41660b57cec5SDimitry Andric   unsigned Shift = countTrailingZeros<unsigned>(Mask);
41670b57cec5SDimitry Andric   V = DAG.getNode(ISD::SRL, SL, VT, V,
41680b57cec5SDimitry Andric                   DAG.getShiftAmountConstant(Shift, VT, SL));
41690b57cec5SDimitry Andric   return DAG.getNode(ISD::AND, SL, VT, V,
41700b57cec5SDimitry Andric                      DAG.getConstant(Mask >> Shift, SL, VT));
41710b57cec5SDimitry Andric }
41720b57cec5SDimitry Andric 
41730b57cec5SDimitry Andric uint32_t AMDGPUTargetLowering::getImplicitParameterOffset(
41740b57cec5SDimitry Andric     const MachineFunction &MF, const ImplicitParameter Param) const {
41750b57cec5SDimitry Andric   const AMDGPUMachineFunction *MFI = MF.getInfo<AMDGPUMachineFunction>();
41760b57cec5SDimitry Andric   const AMDGPUSubtarget &ST =
41770b57cec5SDimitry Andric       AMDGPUSubtarget::get(getTargetMachine(), MF.getFunction());
41780b57cec5SDimitry Andric   unsigned ExplicitArgOffset = ST.getExplicitKernelArgOffset(MF.getFunction());
4179*8bcb0991SDimitry Andric   const Align Alignment = ST.getAlignmentForImplicitArgPtr();
41800b57cec5SDimitry Andric   uint64_t ArgOffset = alignTo(MFI->getExplicitKernArgSize(), Alignment) +
41810b57cec5SDimitry Andric                        ExplicitArgOffset;
41820b57cec5SDimitry Andric   switch (Param) {
41830b57cec5SDimitry Andric   case GRID_DIM:
41840b57cec5SDimitry Andric     return ArgOffset;
41850b57cec5SDimitry Andric   case GRID_OFFSET:
41860b57cec5SDimitry Andric     return ArgOffset + 4;
41870b57cec5SDimitry Andric   }
41880b57cec5SDimitry Andric   llvm_unreachable("unexpected implicit parameter type");
41890b57cec5SDimitry Andric }
41900b57cec5SDimitry Andric 
41910b57cec5SDimitry Andric #define NODE_NAME_CASE(node) case AMDGPUISD::node: return #node;
41920b57cec5SDimitry Andric 
41930b57cec5SDimitry Andric const char* AMDGPUTargetLowering::getTargetNodeName(unsigned Opcode) const {
41940b57cec5SDimitry Andric   switch ((AMDGPUISD::NodeType)Opcode) {
41950b57cec5SDimitry Andric   case AMDGPUISD::FIRST_NUMBER: break;
41960b57cec5SDimitry Andric   // AMDIL DAG nodes
41970b57cec5SDimitry Andric   NODE_NAME_CASE(UMUL);
41980b57cec5SDimitry Andric   NODE_NAME_CASE(BRANCH_COND);
41990b57cec5SDimitry Andric 
42000b57cec5SDimitry Andric   // AMDGPU DAG nodes
42010b57cec5SDimitry Andric   NODE_NAME_CASE(IF)
42020b57cec5SDimitry Andric   NODE_NAME_CASE(ELSE)
42030b57cec5SDimitry Andric   NODE_NAME_CASE(LOOP)
42040b57cec5SDimitry Andric   NODE_NAME_CASE(CALL)
42050b57cec5SDimitry Andric   NODE_NAME_CASE(TC_RETURN)
42060b57cec5SDimitry Andric   NODE_NAME_CASE(TRAP)
42070b57cec5SDimitry Andric   NODE_NAME_CASE(RET_FLAG)
42080b57cec5SDimitry Andric   NODE_NAME_CASE(RETURN_TO_EPILOG)
42090b57cec5SDimitry Andric   NODE_NAME_CASE(ENDPGM)
42100b57cec5SDimitry Andric   NODE_NAME_CASE(DWORDADDR)
42110b57cec5SDimitry Andric   NODE_NAME_CASE(FRACT)
42120b57cec5SDimitry Andric   NODE_NAME_CASE(SETCC)
42130b57cec5SDimitry Andric   NODE_NAME_CASE(SETREG)
4214*8bcb0991SDimitry Andric   NODE_NAME_CASE(DENORM_MODE)
42150b57cec5SDimitry Andric   NODE_NAME_CASE(FMA_W_CHAIN)
42160b57cec5SDimitry Andric   NODE_NAME_CASE(FMUL_W_CHAIN)
42170b57cec5SDimitry Andric   NODE_NAME_CASE(CLAMP)
42180b57cec5SDimitry Andric   NODE_NAME_CASE(COS_HW)
42190b57cec5SDimitry Andric   NODE_NAME_CASE(SIN_HW)
42200b57cec5SDimitry Andric   NODE_NAME_CASE(FMAX_LEGACY)
42210b57cec5SDimitry Andric   NODE_NAME_CASE(FMIN_LEGACY)
42220b57cec5SDimitry Andric   NODE_NAME_CASE(FMAX3)
42230b57cec5SDimitry Andric   NODE_NAME_CASE(SMAX3)
42240b57cec5SDimitry Andric   NODE_NAME_CASE(UMAX3)
42250b57cec5SDimitry Andric   NODE_NAME_CASE(FMIN3)
42260b57cec5SDimitry Andric   NODE_NAME_CASE(SMIN3)
42270b57cec5SDimitry Andric   NODE_NAME_CASE(UMIN3)
42280b57cec5SDimitry Andric   NODE_NAME_CASE(FMED3)
42290b57cec5SDimitry Andric   NODE_NAME_CASE(SMED3)
42300b57cec5SDimitry Andric   NODE_NAME_CASE(UMED3)
42310b57cec5SDimitry Andric   NODE_NAME_CASE(FDOT2)
42320b57cec5SDimitry Andric   NODE_NAME_CASE(URECIP)
42330b57cec5SDimitry Andric   NODE_NAME_CASE(DIV_SCALE)
42340b57cec5SDimitry Andric   NODE_NAME_CASE(DIV_FMAS)
42350b57cec5SDimitry Andric   NODE_NAME_CASE(DIV_FIXUP)
42360b57cec5SDimitry Andric   NODE_NAME_CASE(FMAD_FTZ)
42370b57cec5SDimitry Andric   NODE_NAME_CASE(TRIG_PREOP)
42380b57cec5SDimitry Andric   NODE_NAME_CASE(RCP)
42390b57cec5SDimitry Andric   NODE_NAME_CASE(RSQ)
42400b57cec5SDimitry Andric   NODE_NAME_CASE(RCP_LEGACY)
42410b57cec5SDimitry Andric   NODE_NAME_CASE(RSQ_LEGACY)
42420b57cec5SDimitry Andric   NODE_NAME_CASE(RCP_IFLAG)
42430b57cec5SDimitry Andric   NODE_NAME_CASE(FMUL_LEGACY)
42440b57cec5SDimitry Andric   NODE_NAME_CASE(RSQ_CLAMP)
42450b57cec5SDimitry Andric   NODE_NAME_CASE(LDEXP)
42460b57cec5SDimitry Andric   NODE_NAME_CASE(FP_CLASS)
42470b57cec5SDimitry Andric   NODE_NAME_CASE(DOT4)
42480b57cec5SDimitry Andric   NODE_NAME_CASE(CARRY)
42490b57cec5SDimitry Andric   NODE_NAME_CASE(BORROW)
42500b57cec5SDimitry Andric   NODE_NAME_CASE(BFE_U32)
42510b57cec5SDimitry Andric   NODE_NAME_CASE(BFE_I32)
42520b57cec5SDimitry Andric   NODE_NAME_CASE(BFI)
42530b57cec5SDimitry Andric   NODE_NAME_CASE(BFM)
42540b57cec5SDimitry Andric   NODE_NAME_CASE(FFBH_U32)
42550b57cec5SDimitry Andric   NODE_NAME_CASE(FFBH_I32)
42560b57cec5SDimitry Andric   NODE_NAME_CASE(FFBL_B32)
42570b57cec5SDimitry Andric   NODE_NAME_CASE(MUL_U24)
42580b57cec5SDimitry Andric   NODE_NAME_CASE(MUL_I24)
42590b57cec5SDimitry Andric   NODE_NAME_CASE(MULHI_U24)
42600b57cec5SDimitry Andric   NODE_NAME_CASE(MULHI_I24)
42610b57cec5SDimitry Andric   NODE_NAME_CASE(MUL_LOHI_U24)
42620b57cec5SDimitry Andric   NODE_NAME_CASE(MUL_LOHI_I24)
42630b57cec5SDimitry Andric   NODE_NAME_CASE(MAD_U24)
42640b57cec5SDimitry Andric   NODE_NAME_CASE(MAD_I24)
42650b57cec5SDimitry Andric   NODE_NAME_CASE(MAD_I64_I32)
42660b57cec5SDimitry Andric   NODE_NAME_CASE(MAD_U64_U32)
42670b57cec5SDimitry Andric   NODE_NAME_CASE(PERM)
42680b57cec5SDimitry Andric   NODE_NAME_CASE(TEXTURE_FETCH)
42690b57cec5SDimitry Andric   NODE_NAME_CASE(EXPORT)
42700b57cec5SDimitry Andric   NODE_NAME_CASE(EXPORT_DONE)
42710b57cec5SDimitry Andric   NODE_NAME_CASE(R600_EXPORT)
42720b57cec5SDimitry Andric   NODE_NAME_CASE(CONST_ADDRESS)
42730b57cec5SDimitry Andric   NODE_NAME_CASE(REGISTER_LOAD)
42740b57cec5SDimitry Andric   NODE_NAME_CASE(REGISTER_STORE)
42750b57cec5SDimitry Andric   NODE_NAME_CASE(SAMPLE)
42760b57cec5SDimitry Andric   NODE_NAME_CASE(SAMPLEB)
42770b57cec5SDimitry Andric   NODE_NAME_CASE(SAMPLED)
42780b57cec5SDimitry Andric   NODE_NAME_CASE(SAMPLEL)
42790b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_F32_UBYTE0)
42800b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_F32_UBYTE1)
42810b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_F32_UBYTE2)
42820b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_F32_UBYTE3)
42830b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_PKRTZ_F16_F32)
42840b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_PKNORM_I16_F32)
42850b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_PKNORM_U16_F32)
42860b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_PK_I16_I32)
42870b57cec5SDimitry Andric   NODE_NAME_CASE(CVT_PK_U16_U32)
42880b57cec5SDimitry Andric   NODE_NAME_CASE(FP_TO_FP16)
42890b57cec5SDimitry Andric   NODE_NAME_CASE(FP16_ZEXT)
42900b57cec5SDimitry Andric   NODE_NAME_CASE(BUILD_VERTICAL_VECTOR)
42910b57cec5SDimitry Andric   NODE_NAME_CASE(CONST_DATA_PTR)
42920b57cec5SDimitry Andric   NODE_NAME_CASE(PC_ADD_REL_OFFSET)
42930b57cec5SDimitry Andric   NODE_NAME_CASE(LDS)
42940b57cec5SDimitry Andric   NODE_NAME_CASE(KILL)
42950b57cec5SDimitry Andric   NODE_NAME_CASE(DUMMY_CHAIN)
42960b57cec5SDimitry Andric   case AMDGPUISD::FIRST_MEM_OPCODE_NUMBER: break;
42970b57cec5SDimitry Andric   NODE_NAME_CASE(INTERP_P1LL_F16)
42980b57cec5SDimitry Andric   NODE_NAME_CASE(INTERP_P1LV_F16)
42990b57cec5SDimitry Andric   NODE_NAME_CASE(INTERP_P2_F16)
43000b57cec5SDimitry Andric   NODE_NAME_CASE(LOAD_D16_HI)
43010b57cec5SDimitry Andric   NODE_NAME_CASE(LOAD_D16_LO)
43020b57cec5SDimitry Andric   NODE_NAME_CASE(LOAD_D16_HI_I8)
43030b57cec5SDimitry Andric   NODE_NAME_CASE(LOAD_D16_HI_U8)
43040b57cec5SDimitry Andric   NODE_NAME_CASE(LOAD_D16_LO_I8)
43050b57cec5SDimitry Andric   NODE_NAME_CASE(LOAD_D16_LO_U8)
43060b57cec5SDimitry Andric   NODE_NAME_CASE(STORE_MSKOR)
43070b57cec5SDimitry Andric   NODE_NAME_CASE(LOAD_CONSTANT)
43080b57cec5SDimitry Andric   NODE_NAME_CASE(TBUFFER_STORE_FORMAT)
43090b57cec5SDimitry Andric   NODE_NAME_CASE(TBUFFER_STORE_FORMAT_D16)
43100b57cec5SDimitry Andric   NODE_NAME_CASE(TBUFFER_LOAD_FORMAT)
43110b57cec5SDimitry Andric   NODE_NAME_CASE(TBUFFER_LOAD_FORMAT_D16)
43120b57cec5SDimitry Andric   NODE_NAME_CASE(DS_ORDERED_COUNT)
43130b57cec5SDimitry Andric   NODE_NAME_CASE(ATOMIC_CMP_SWAP)
43140b57cec5SDimitry Andric   NODE_NAME_CASE(ATOMIC_INC)
43150b57cec5SDimitry Andric   NODE_NAME_CASE(ATOMIC_DEC)
43160b57cec5SDimitry Andric   NODE_NAME_CASE(ATOMIC_LOAD_FMIN)
43170b57cec5SDimitry Andric   NODE_NAME_CASE(ATOMIC_LOAD_FMAX)
43180b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_LOAD)
43190b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_LOAD_UBYTE)
43200b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_LOAD_USHORT)
43210b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_LOAD_BYTE)
43220b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_LOAD_SHORT)
43230b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_LOAD_FORMAT)
43240b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_LOAD_FORMAT_D16)
43250b57cec5SDimitry Andric   NODE_NAME_CASE(SBUFFER_LOAD)
43260b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_STORE)
43270b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_STORE_BYTE)
43280b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_STORE_SHORT)
43290b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_STORE_FORMAT)
43300b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_STORE_FORMAT_D16)
43310b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_SWAP)
43320b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_ADD)
43330b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_SUB)
43340b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_SMIN)
43350b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_UMIN)
43360b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_SMAX)
43370b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_UMAX)
43380b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_AND)
43390b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_OR)
43400b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_XOR)
4341*8bcb0991SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_INC)
4342*8bcb0991SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_DEC)
43430b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_CMPSWAP)
43440b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_FADD)
43450b57cec5SDimitry Andric   NODE_NAME_CASE(BUFFER_ATOMIC_PK_FADD)
43460b57cec5SDimitry Andric   NODE_NAME_CASE(ATOMIC_FADD)
43470b57cec5SDimitry Andric   NODE_NAME_CASE(ATOMIC_PK_FADD)
43480b57cec5SDimitry Andric 
43490b57cec5SDimitry Andric   case AMDGPUISD::LAST_AMDGPU_ISD_NUMBER: break;
43500b57cec5SDimitry Andric   }
43510b57cec5SDimitry Andric   return nullptr;
43520b57cec5SDimitry Andric }
43530b57cec5SDimitry Andric 
43540b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::getSqrtEstimate(SDValue Operand,
43550b57cec5SDimitry Andric                                               SelectionDAG &DAG, int Enabled,
43560b57cec5SDimitry Andric                                               int &RefinementSteps,
43570b57cec5SDimitry Andric                                               bool &UseOneConstNR,
43580b57cec5SDimitry Andric                                               bool Reciprocal) const {
43590b57cec5SDimitry Andric   EVT VT = Operand.getValueType();
43600b57cec5SDimitry Andric 
43610b57cec5SDimitry Andric   if (VT == MVT::f32) {
43620b57cec5SDimitry Andric     RefinementSteps = 0;
43630b57cec5SDimitry Andric     return DAG.getNode(AMDGPUISD::RSQ, SDLoc(Operand), VT, Operand);
43640b57cec5SDimitry Andric   }
43650b57cec5SDimitry Andric 
43660b57cec5SDimitry Andric   // TODO: There is also f64 rsq instruction, but the documentation is less
43670b57cec5SDimitry Andric   // clear on its precision.
43680b57cec5SDimitry Andric 
43690b57cec5SDimitry Andric   return SDValue();
43700b57cec5SDimitry Andric }
43710b57cec5SDimitry Andric 
43720b57cec5SDimitry Andric SDValue AMDGPUTargetLowering::getRecipEstimate(SDValue Operand,
43730b57cec5SDimitry Andric                                                SelectionDAG &DAG, int Enabled,
43740b57cec5SDimitry Andric                                                int &RefinementSteps) const {
43750b57cec5SDimitry Andric   EVT VT = Operand.getValueType();
43760b57cec5SDimitry Andric 
43770b57cec5SDimitry Andric   if (VT == MVT::f32) {
43780b57cec5SDimitry Andric     // Reciprocal, < 1 ulp error.
43790b57cec5SDimitry Andric     //
43800b57cec5SDimitry Andric     // This reciprocal approximation converges to < 0.5 ulp error with one
43810b57cec5SDimitry Andric     // newton rhapson performed with two fused multiple adds (FMAs).
43820b57cec5SDimitry Andric 
43830b57cec5SDimitry Andric     RefinementSteps = 0;
43840b57cec5SDimitry Andric     return DAG.getNode(AMDGPUISD::RCP, SDLoc(Operand), VT, Operand);
43850b57cec5SDimitry Andric   }
43860b57cec5SDimitry Andric 
43870b57cec5SDimitry Andric   // TODO: There is also f64 rcp instruction, but the documentation is less
43880b57cec5SDimitry Andric   // clear on its precision.
43890b57cec5SDimitry Andric 
43900b57cec5SDimitry Andric   return SDValue();
43910b57cec5SDimitry Andric }
43920b57cec5SDimitry Andric 
43930b57cec5SDimitry Andric void AMDGPUTargetLowering::computeKnownBitsForTargetNode(
43940b57cec5SDimitry Andric     const SDValue Op, KnownBits &Known,
43950b57cec5SDimitry Andric     const APInt &DemandedElts, const SelectionDAG &DAG, unsigned Depth) const {
43960b57cec5SDimitry Andric 
43970b57cec5SDimitry Andric   Known.resetAll(); // Don't know anything.
43980b57cec5SDimitry Andric 
43990b57cec5SDimitry Andric   unsigned Opc = Op.getOpcode();
44000b57cec5SDimitry Andric 
44010b57cec5SDimitry Andric   switch (Opc) {
44020b57cec5SDimitry Andric   default:
44030b57cec5SDimitry Andric     break;
44040b57cec5SDimitry Andric   case AMDGPUISD::CARRY:
44050b57cec5SDimitry Andric   case AMDGPUISD::BORROW: {
44060b57cec5SDimitry Andric     Known.Zero = APInt::getHighBitsSet(32, 31);
44070b57cec5SDimitry Andric     break;
44080b57cec5SDimitry Andric   }
44090b57cec5SDimitry Andric 
44100b57cec5SDimitry Andric   case AMDGPUISD::BFE_I32:
44110b57cec5SDimitry Andric   case AMDGPUISD::BFE_U32: {
44120b57cec5SDimitry Andric     ConstantSDNode *CWidth = dyn_cast<ConstantSDNode>(Op.getOperand(2));
44130b57cec5SDimitry Andric     if (!CWidth)
44140b57cec5SDimitry Andric       return;
44150b57cec5SDimitry Andric 
44160b57cec5SDimitry Andric     uint32_t Width = CWidth->getZExtValue() & 0x1f;
44170b57cec5SDimitry Andric 
44180b57cec5SDimitry Andric     if (Opc == AMDGPUISD::BFE_U32)
44190b57cec5SDimitry Andric       Known.Zero = APInt::getHighBitsSet(32, 32 - Width);
44200b57cec5SDimitry Andric 
44210b57cec5SDimitry Andric     break;
44220b57cec5SDimitry Andric   }
44230b57cec5SDimitry Andric   case AMDGPUISD::FP_TO_FP16:
44240b57cec5SDimitry Andric   case AMDGPUISD::FP16_ZEXT: {
44250b57cec5SDimitry Andric     unsigned BitWidth = Known.getBitWidth();
44260b57cec5SDimitry Andric 
44270b57cec5SDimitry Andric     // High bits are zero.
44280b57cec5SDimitry Andric     Known.Zero = APInt::getHighBitsSet(BitWidth, BitWidth - 16);
44290b57cec5SDimitry Andric     break;
44300b57cec5SDimitry Andric   }
44310b57cec5SDimitry Andric   case AMDGPUISD::MUL_U24:
44320b57cec5SDimitry Andric   case AMDGPUISD::MUL_I24: {
44330b57cec5SDimitry Andric     KnownBits LHSKnown = DAG.computeKnownBits(Op.getOperand(0), Depth + 1);
44340b57cec5SDimitry Andric     KnownBits RHSKnown = DAG.computeKnownBits(Op.getOperand(1), Depth + 1);
44350b57cec5SDimitry Andric     unsigned TrailZ = LHSKnown.countMinTrailingZeros() +
44360b57cec5SDimitry Andric                       RHSKnown.countMinTrailingZeros();
44370b57cec5SDimitry Andric     Known.Zero.setLowBits(std::min(TrailZ, 32u));
44380b57cec5SDimitry Andric 
44390b57cec5SDimitry Andric     // Truncate to 24 bits.
44400b57cec5SDimitry Andric     LHSKnown = LHSKnown.trunc(24);
44410b57cec5SDimitry Andric     RHSKnown = RHSKnown.trunc(24);
44420b57cec5SDimitry Andric 
44430b57cec5SDimitry Andric     bool Negative = false;
44440b57cec5SDimitry Andric     if (Opc == AMDGPUISD::MUL_I24) {
44450b57cec5SDimitry Andric       unsigned LHSValBits = 24 - LHSKnown.countMinSignBits();
44460b57cec5SDimitry Andric       unsigned RHSValBits = 24 - RHSKnown.countMinSignBits();
44470b57cec5SDimitry Andric       unsigned MaxValBits = std::min(LHSValBits + RHSValBits, 32u);
44480b57cec5SDimitry Andric       if (MaxValBits >= 32)
44490b57cec5SDimitry Andric         break;
44500b57cec5SDimitry Andric       bool LHSNegative = LHSKnown.isNegative();
44510b57cec5SDimitry Andric       bool LHSPositive = LHSKnown.isNonNegative();
44520b57cec5SDimitry Andric       bool RHSNegative = RHSKnown.isNegative();
44530b57cec5SDimitry Andric       bool RHSPositive = RHSKnown.isNonNegative();
44540b57cec5SDimitry Andric       if ((!LHSNegative && !LHSPositive) || (!RHSNegative && !RHSPositive))
44550b57cec5SDimitry Andric         break;
44560b57cec5SDimitry Andric       Negative = (LHSNegative && RHSPositive) || (LHSPositive && RHSNegative);
44570b57cec5SDimitry Andric       if (Negative)
44580b57cec5SDimitry Andric         Known.One.setHighBits(32 - MaxValBits);
44590b57cec5SDimitry Andric       else
44600b57cec5SDimitry Andric         Known.Zero.setHighBits(32 - MaxValBits);
44610b57cec5SDimitry Andric     } else {
44620b57cec5SDimitry Andric       unsigned LHSValBits = 24 - LHSKnown.countMinLeadingZeros();
44630b57cec5SDimitry Andric       unsigned RHSValBits = 24 - RHSKnown.countMinLeadingZeros();
44640b57cec5SDimitry Andric       unsigned MaxValBits = std::min(LHSValBits + RHSValBits, 32u);
44650b57cec5SDimitry Andric       if (MaxValBits >= 32)
44660b57cec5SDimitry Andric         break;
44670b57cec5SDimitry Andric       Known.Zero.setHighBits(32 - MaxValBits);
44680b57cec5SDimitry Andric     }
44690b57cec5SDimitry Andric     break;
44700b57cec5SDimitry Andric   }
44710b57cec5SDimitry Andric   case AMDGPUISD::PERM: {
44720b57cec5SDimitry Andric     ConstantSDNode *CMask = dyn_cast<ConstantSDNode>(Op.getOperand(2));
44730b57cec5SDimitry Andric     if (!CMask)
44740b57cec5SDimitry Andric       return;
44750b57cec5SDimitry Andric 
44760b57cec5SDimitry Andric     KnownBits LHSKnown = DAG.computeKnownBits(Op.getOperand(0), Depth + 1);
44770b57cec5SDimitry Andric     KnownBits RHSKnown = DAG.computeKnownBits(Op.getOperand(1), Depth + 1);
44780b57cec5SDimitry Andric     unsigned Sel = CMask->getZExtValue();
44790b57cec5SDimitry Andric 
44800b57cec5SDimitry Andric     for (unsigned I = 0; I < 32; I += 8) {
44810b57cec5SDimitry Andric       unsigned SelBits = Sel & 0xff;
44820b57cec5SDimitry Andric       if (SelBits < 4) {
44830b57cec5SDimitry Andric         SelBits *= 8;
44840b57cec5SDimitry Andric         Known.One |= ((RHSKnown.One.getZExtValue() >> SelBits) & 0xff) << I;
44850b57cec5SDimitry Andric         Known.Zero |= ((RHSKnown.Zero.getZExtValue() >> SelBits) & 0xff) << I;
44860b57cec5SDimitry Andric       } else if (SelBits < 7) {
44870b57cec5SDimitry Andric         SelBits = (SelBits & 3) * 8;
44880b57cec5SDimitry Andric         Known.One |= ((LHSKnown.One.getZExtValue() >> SelBits) & 0xff) << I;
44890b57cec5SDimitry Andric         Known.Zero |= ((LHSKnown.Zero.getZExtValue() >> SelBits) & 0xff) << I;
44900b57cec5SDimitry Andric       } else if (SelBits == 0x0c) {
4491*8bcb0991SDimitry Andric         Known.Zero |= 0xFFull << I;
44920b57cec5SDimitry Andric       } else if (SelBits > 0x0c) {
4493*8bcb0991SDimitry Andric         Known.One |= 0xFFull << I;
44940b57cec5SDimitry Andric       }
44950b57cec5SDimitry Andric       Sel >>= 8;
44960b57cec5SDimitry Andric     }
44970b57cec5SDimitry Andric     break;
44980b57cec5SDimitry Andric   }
44990b57cec5SDimitry Andric   case AMDGPUISD::BUFFER_LOAD_UBYTE:  {
45000b57cec5SDimitry Andric     Known.Zero.setHighBits(24);
45010b57cec5SDimitry Andric     break;
45020b57cec5SDimitry Andric   }
45030b57cec5SDimitry Andric   case AMDGPUISD::BUFFER_LOAD_USHORT: {
45040b57cec5SDimitry Andric     Known.Zero.setHighBits(16);
45050b57cec5SDimitry Andric     break;
45060b57cec5SDimitry Andric   }
45070b57cec5SDimitry Andric   case AMDGPUISD::LDS: {
45080b57cec5SDimitry Andric     auto GA = cast<GlobalAddressSDNode>(Op.getOperand(0).getNode());
45090b57cec5SDimitry Andric     unsigned Align = GA->getGlobal()->getAlignment();
45100b57cec5SDimitry Andric 
45110b57cec5SDimitry Andric     Known.Zero.setHighBits(16);
45120b57cec5SDimitry Andric     if (Align)
45130b57cec5SDimitry Andric       Known.Zero.setLowBits(Log2_32(Align));
45140b57cec5SDimitry Andric     break;
45150b57cec5SDimitry Andric   }
45160b57cec5SDimitry Andric   case ISD::INTRINSIC_WO_CHAIN: {
45170b57cec5SDimitry Andric     unsigned IID = cast<ConstantSDNode>(Op.getOperand(0))->getZExtValue();
45180b57cec5SDimitry Andric     switch (IID) {
45190b57cec5SDimitry Andric     case Intrinsic::amdgcn_mbcnt_lo:
45200b57cec5SDimitry Andric     case Intrinsic::amdgcn_mbcnt_hi: {
45210b57cec5SDimitry Andric       const GCNSubtarget &ST =
45220b57cec5SDimitry Andric           DAG.getMachineFunction().getSubtarget<GCNSubtarget>();
45230b57cec5SDimitry Andric       // These return at most the wavefront size - 1.
45240b57cec5SDimitry Andric       unsigned Size = Op.getValueType().getSizeInBits();
45250b57cec5SDimitry Andric       Known.Zero.setHighBits(Size - ST.getWavefrontSizeLog2());
45260b57cec5SDimitry Andric       break;
45270b57cec5SDimitry Andric     }
45280b57cec5SDimitry Andric     default:
45290b57cec5SDimitry Andric       break;
45300b57cec5SDimitry Andric     }
45310b57cec5SDimitry Andric   }
45320b57cec5SDimitry Andric   }
45330b57cec5SDimitry Andric }
45340b57cec5SDimitry Andric 
45350b57cec5SDimitry Andric unsigned AMDGPUTargetLowering::ComputeNumSignBitsForTargetNode(
45360b57cec5SDimitry Andric     SDValue Op, const APInt &DemandedElts, const SelectionDAG &DAG,
45370b57cec5SDimitry Andric     unsigned Depth) const {
45380b57cec5SDimitry Andric   switch (Op.getOpcode()) {
45390b57cec5SDimitry Andric   case AMDGPUISD::BFE_I32: {
45400b57cec5SDimitry Andric     ConstantSDNode *Width = dyn_cast<ConstantSDNode>(Op.getOperand(2));
45410b57cec5SDimitry Andric     if (!Width)
45420b57cec5SDimitry Andric       return 1;
45430b57cec5SDimitry Andric 
45440b57cec5SDimitry Andric     unsigned SignBits = 32 - Width->getZExtValue() + 1;
45450b57cec5SDimitry Andric     if (!isNullConstant(Op.getOperand(1)))
45460b57cec5SDimitry Andric       return SignBits;
45470b57cec5SDimitry Andric 
45480b57cec5SDimitry Andric     // TODO: Could probably figure something out with non-0 offsets.
45490b57cec5SDimitry Andric     unsigned Op0SignBits = DAG.ComputeNumSignBits(Op.getOperand(0), Depth + 1);
45500b57cec5SDimitry Andric     return std::max(SignBits, Op0SignBits);
45510b57cec5SDimitry Andric   }
45520b57cec5SDimitry Andric 
45530b57cec5SDimitry Andric   case AMDGPUISD::BFE_U32: {
45540b57cec5SDimitry Andric     ConstantSDNode *Width = dyn_cast<ConstantSDNode>(Op.getOperand(2));
45550b57cec5SDimitry Andric     return Width ? 32 - (Width->getZExtValue() & 0x1f) : 1;
45560b57cec5SDimitry Andric   }
45570b57cec5SDimitry Andric 
45580b57cec5SDimitry Andric   case AMDGPUISD::CARRY:
45590b57cec5SDimitry Andric   case AMDGPUISD::BORROW:
45600b57cec5SDimitry Andric     return 31;
45610b57cec5SDimitry Andric   case AMDGPUISD::BUFFER_LOAD_BYTE:
45620b57cec5SDimitry Andric     return 25;
45630b57cec5SDimitry Andric   case AMDGPUISD::BUFFER_LOAD_SHORT:
45640b57cec5SDimitry Andric     return 17;
45650b57cec5SDimitry Andric   case AMDGPUISD::BUFFER_LOAD_UBYTE:
45660b57cec5SDimitry Andric     return 24;
45670b57cec5SDimitry Andric   case AMDGPUISD::BUFFER_LOAD_USHORT:
45680b57cec5SDimitry Andric     return 16;
45690b57cec5SDimitry Andric   case AMDGPUISD::FP_TO_FP16:
45700b57cec5SDimitry Andric   case AMDGPUISD::FP16_ZEXT:
45710b57cec5SDimitry Andric     return 16;
45720b57cec5SDimitry Andric   default:
45730b57cec5SDimitry Andric     return 1;
45740b57cec5SDimitry Andric   }
45750b57cec5SDimitry Andric }
45760b57cec5SDimitry Andric 
45770b57cec5SDimitry Andric bool AMDGPUTargetLowering::isKnownNeverNaNForTargetNode(SDValue Op,
45780b57cec5SDimitry Andric                                                         const SelectionDAG &DAG,
45790b57cec5SDimitry Andric                                                         bool SNaN,
45800b57cec5SDimitry Andric                                                         unsigned Depth) const {
45810b57cec5SDimitry Andric   unsigned Opcode = Op.getOpcode();
45820b57cec5SDimitry Andric   switch (Opcode) {
45830b57cec5SDimitry Andric   case AMDGPUISD::FMIN_LEGACY:
45840b57cec5SDimitry Andric   case AMDGPUISD::FMAX_LEGACY: {
45850b57cec5SDimitry Andric     if (SNaN)
45860b57cec5SDimitry Andric       return true;
45870b57cec5SDimitry Andric 
45880b57cec5SDimitry Andric     // TODO: Can check no nans on one of the operands for each one, but which
45890b57cec5SDimitry Andric     // one?
45900b57cec5SDimitry Andric     return false;
45910b57cec5SDimitry Andric   }
45920b57cec5SDimitry Andric   case AMDGPUISD::FMUL_LEGACY:
45930b57cec5SDimitry Andric   case AMDGPUISD::CVT_PKRTZ_F16_F32: {
45940b57cec5SDimitry Andric     if (SNaN)
45950b57cec5SDimitry Andric       return true;
45960b57cec5SDimitry Andric     return DAG.isKnownNeverNaN(Op.getOperand(0), SNaN, Depth + 1) &&
45970b57cec5SDimitry Andric            DAG.isKnownNeverNaN(Op.getOperand(1), SNaN, Depth + 1);
45980b57cec5SDimitry Andric   }
45990b57cec5SDimitry Andric   case AMDGPUISD::FMED3:
46000b57cec5SDimitry Andric   case AMDGPUISD::FMIN3:
46010b57cec5SDimitry Andric   case AMDGPUISD::FMAX3:
46020b57cec5SDimitry Andric   case AMDGPUISD::FMAD_FTZ: {
46030b57cec5SDimitry Andric     if (SNaN)
46040b57cec5SDimitry Andric       return true;
46050b57cec5SDimitry Andric     return DAG.isKnownNeverNaN(Op.getOperand(0), SNaN, Depth + 1) &&
46060b57cec5SDimitry Andric            DAG.isKnownNeverNaN(Op.getOperand(1), SNaN, Depth + 1) &&
46070b57cec5SDimitry Andric            DAG.isKnownNeverNaN(Op.getOperand(2), SNaN, Depth + 1);
46080b57cec5SDimitry Andric   }
46090b57cec5SDimitry Andric   case AMDGPUISD::CVT_F32_UBYTE0:
46100b57cec5SDimitry Andric   case AMDGPUISD::CVT_F32_UBYTE1:
46110b57cec5SDimitry Andric   case AMDGPUISD::CVT_F32_UBYTE2:
46120b57cec5SDimitry Andric   case AMDGPUISD::CVT_F32_UBYTE3:
46130b57cec5SDimitry Andric     return true;
46140b57cec5SDimitry Andric 
46150b57cec5SDimitry Andric   case AMDGPUISD::RCP:
46160b57cec5SDimitry Andric   case AMDGPUISD::RSQ:
46170b57cec5SDimitry Andric   case AMDGPUISD::RCP_LEGACY:
46180b57cec5SDimitry Andric   case AMDGPUISD::RSQ_LEGACY:
46190b57cec5SDimitry Andric   case AMDGPUISD::RSQ_CLAMP: {
46200b57cec5SDimitry Andric     if (SNaN)
46210b57cec5SDimitry Andric       return true;
46220b57cec5SDimitry Andric 
46230b57cec5SDimitry Andric     // TODO: Need is known positive check.
46240b57cec5SDimitry Andric     return false;
46250b57cec5SDimitry Andric   }
46260b57cec5SDimitry Andric   case AMDGPUISD::LDEXP:
46270b57cec5SDimitry Andric   case AMDGPUISD::FRACT: {
46280b57cec5SDimitry Andric     if (SNaN)
46290b57cec5SDimitry Andric       return true;
46300b57cec5SDimitry Andric     return DAG.isKnownNeverNaN(Op.getOperand(0), SNaN, Depth + 1);
46310b57cec5SDimitry Andric   }
46320b57cec5SDimitry Andric   case AMDGPUISD::DIV_SCALE:
46330b57cec5SDimitry Andric   case AMDGPUISD::DIV_FMAS:
46340b57cec5SDimitry Andric   case AMDGPUISD::DIV_FIXUP:
46350b57cec5SDimitry Andric   case AMDGPUISD::TRIG_PREOP:
46360b57cec5SDimitry Andric     // TODO: Refine on operands.
46370b57cec5SDimitry Andric     return SNaN;
46380b57cec5SDimitry Andric   case AMDGPUISD::SIN_HW:
46390b57cec5SDimitry Andric   case AMDGPUISD::COS_HW: {
46400b57cec5SDimitry Andric     // TODO: Need check for infinity
46410b57cec5SDimitry Andric     return SNaN;
46420b57cec5SDimitry Andric   }
46430b57cec5SDimitry Andric   case ISD::INTRINSIC_WO_CHAIN: {
46440b57cec5SDimitry Andric     unsigned IntrinsicID
46450b57cec5SDimitry Andric       = cast<ConstantSDNode>(Op.getOperand(0))->getZExtValue();
46460b57cec5SDimitry Andric     // TODO: Handle more intrinsics
46470b57cec5SDimitry Andric     switch (IntrinsicID) {
46480b57cec5SDimitry Andric     case Intrinsic::amdgcn_cubeid:
46490b57cec5SDimitry Andric       return true;
46500b57cec5SDimitry Andric 
46510b57cec5SDimitry Andric     case Intrinsic::amdgcn_frexp_mant: {
46520b57cec5SDimitry Andric       if (SNaN)
46530b57cec5SDimitry Andric         return true;
46540b57cec5SDimitry Andric       return DAG.isKnownNeverNaN(Op.getOperand(1), SNaN, Depth + 1);
46550b57cec5SDimitry Andric     }
46560b57cec5SDimitry Andric     case Intrinsic::amdgcn_cvt_pkrtz: {
46570b57cec5SDimitry Andric       if (SNaN)
46580b57cec5SDimitry Andric         return true;
46590b57cec5SDimitry Andric       return DAG.isKnownNeverNaN(Op.getOperand(1), SNaN, Depth + 1) &&
46600b57cec5SDimitry Andric              DAG.isKnownNeverNaN(Op.getOperand(2), SNaN, Depth + 1);
46610b57cec5SDimitry Andric     }
46620b57cec5SDimitry Andric     case Intrinsic::amdgcn_fdot2:
46630b57cec5SDimitry Andric       // TODO: Refine on operand
46640b57cec5SDimitry Andric       return SNaN;
46650b57cec5SDimitry Andric     default:
46660b57cec5SDimitry Andric       return false;
46670b57cec5SDimitry Andric     }
46680b57cec5SDimitry Andric   }
46690b57cec5SDimitry Andric   default:
46700b57cec5SDimitry Andric     return false;
46710b57cec5SDimitry Andric   }
46720b57cec5SDimitry Andric }
46730b57cec5SDimitry Andric 
46740b57cec5SDimitry Andric TargetLowering::AtomicExpansionKind
46750b57cec5SDimitry Andric AMDGPUTargetLowering::shouldExpandAtomicRMWInIR(AtomicRMWInst *RMW) const {
46760b57cec5SDimitry Andric   switch (RMW->getOperation()) {
46770b57cec5SDimitry Andric   case AtomicRMWInst::Nand:
46780b57cec5SDimitry Andric   case AtomicRMWInst::FAdd:
46790b57cec5SDimitry Andric   case AtomicRMWInst::FSub:
46800b57cec5SDimitry Andric     return AtomicExpansionKind::CmpXChg;
46810b57cec5SDimitry Andric   default:
46820b57cec5SDimitry Andric     return AtomicExpansionKind::None;
46830b57cec5SDimitry Andric   }
46840b57cec5SDimitry Andric }
4685