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