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