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