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