Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 1 | //===-- SIISelLowering.cpp - SI DAG Lowering Implementation ---------------===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | /// \file |
| 11 | /// \brief Custom DAG lowering for SI |
| 12 | // |
| 13 | //===----------------------------------------------------------------------===// |
| 14 | |
| 15 | #include "SIISelLowering.h" |
| 16 | #include "AMDIL.h" |
Christian Konig | 99ee0f4 | 2013-03-07 09:04:14 +0000 | [diff] [blame] | 17 | #include "AMDGPU.h" |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 18 | #include "AMDILIntrinsicInfo.h" |
| 19 | #include "SIInstrInfo.h" |
| 20 | #include "SIMachineFunctionInfo.h" |
| 21 | #include "SIRegisterInfo.h" |
Christian Konig | 2c8f6d5 | 2013-03-07 09:03:52 +0000 | [diff] [blame] | 22 | #include "llvm/IR/Function.h" |
| 23 | #include "llvm/CodeGen/CallingConvLower.h" |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 24 | #include "llvm/CodeGen/MachineInstrBuilder.h" |
| 25 | #include "llvm/CodeGen/MachineRegisterInfo.h" |
| 26 | #include "llvm/CodeGen/SelectionDAG.h" |
| 27 | |
| 28 | using namespace llvm; |
| 29 | |
| 30 | SITargetLowering::SITargetLowering(TargetMachine &TM) : |
| 31 | AMDGPUTargetLowering(TM), |
Christian Konig | f82901a | 2013-02-26 17:52:23 +0000 | [diff] [blame] | 32 | TII(static_cast<const SIInstrInfo*>(TM.getInstrInfo())), |
| 33 | TRI(TM.getRegisterInfo()) { |
Christian Konig | 2214f14 | 2013-03-07 09:03:38 +0000 | [diff] [blame] | 34 | |
Christian Konig | a881179 | 2013-02-16 11:28:30 +0000 | [diff] [blame] | 35 | addRegisterClass(MVT::i1, &AMDGPU::SReg_64RegClass); |
Christian Konig | 2214f14 | 2013-03-07 09:03:38 +0000 | [diff] [blame] | 36 | addRegisterClass(MVT::i64, &AMDGPU::SReg_64RegClass); |
| 37 | |
| 38 | addRegisterClass(MVT::v16i8, &AMDGPU::SReg_128RegClass); |
| 39 | addRegisterClass(MVT::v32i8, &AMDGPU::SReg_256RegClass); |
| 40 | addRegisterClass(MVT::v64i8, &AMDGPU::SReg_512RegClass); |
| 41 | |
| 42 | addRegisterClass(MVT::i32, &AMDGPU::VReg_32RegClass); |
| 43 | addRegisterClass(MVT::f32, &AMDGPU::VReg_32RegClass); |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 44 | |
Tom Stellard | 538ceeb | 2013-02-07 17:02:09 +0000 | [diff] [blame] | 45 | addRegisterClass(MVT::v1i32, &AMDGPU::VReg_32RegClass); |
Christian Konig | 2214f14 | 2013-03-07 09:03:38 +0000 | [diff] [blame] | 46 | |
Tom Stellard | 538ceeb | 2013-02-07 17:02:09 +0000 | [diff] [blame] | 47 | addRegisterClass(MVT::v2i32, &AMDGPU::VReg_64RegClass); |
Christian Konig | 2214f14 | 2013-03-07 09:03:38 +0000 | [diff] [blame] | 48 | addRegisterClass(MVT::v2f32, &AMDGPU::VReg_64RegClass); |
| 49 | |
Tom Stellard | 538ceeb | 2013-02-07 17:02:09 +0000 | [diff] [blame] | 50 | addRegisterClass(MVT::v4i32, &AMDGPU::VReg_128RegClass); |
Christian Konig | 2214f14 | 2013-03-07 09:03:38 +0000 | [diff] [blame] | 51 | addRegisterClass(MVT::v4f32, &AMDGPU::VReg_128RegClass); |
Tom Stellard | 754f80f | 2013-04-05 23:31:51 +0000 | [diff] [blame^] | 52 | addRegisterClass(MVT::i128, &AMDGPU::SReg_128RegClass); |
Christian Konig | 2214f14 | 2013-03-07 09:03:38 +0000 | [diff] [blame] | 53 | |
Tom Stellard | 538ceeb | 2013-02-07 17:02:09 +0000 | [diff] [blame] | 54 | addRegisterClass(MVT::v8i32, &AMDGPU::VReg_256RegClass); |
Christian Konig | 2214f14 | 2013-03-07 09:03:38 +0000 | [diff] [blame] | 55 | addRegisterClass(MVT::v8f32, &AMDGPU::VReg_256RegClass); |
| 56 | |
Tom Stellard | 538ceeb | 2013-02-07 17:02:09 +0000 | [diff] [blame] | 57 | addRegisterClass(MVT::v16i32, &AMDGPU::VReg_512RegClass); |
Christian Konig | 2214f14 | 2013-03-07 09:03:38 +0000 | [diff] [blame] | 58 | addRegisterClass(MVT::v16f32, &AMDGPU::VReg_512RegClass); |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 59 | |
| 60 | computeRegisterProperties(); |
| 61 | |
Christian Konig | 2989ffc | 2013-03-18 11:34:16 +0000 | [diff] [blame] | 62 | setOperationAction(ISD::VECTOR_SHUFFLE, MVT::v8i32, Expand); |
| 63 | setOperationAction(ISD::VECTOR_SHUFFLE, MVT::v8f32, Expand); |
| 64 | setOperationAction(ISD::VECTOR_SHUFFLE, MVT::v16i32, Expand); |
| 65 | setOperationAction(ISD::VECTOR_SHUFFLE, MVT::v16f32, Expand); |
| 66 | |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 67 | setOperationAction(ISD::ADD, MVT::i64, Legal); |
| 68 | setOperationAction(ISD::ADD, MVT::i32, Legal); |
| 69 | |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 70 | setOperationAction(ISD::SELECT_CC, MVT::f32, Custom); |
| 71 | setOperationAction(ISD::SELECT_CC, MVT::i32, Custom); |
| 72 | |
| 73 | setOperationAction(ISD::SELECT_CC, MVT::Other, Expand); |
Tom Stellard | 754f80f | 2013-04-05 23:31:51 +0000 | [diff] [blame^] | 74 | |
| 75 | setOperationAction(ISD::STORE, MVT::i32, Custom); |
| 76 | setOperationAction(ISD::STORE, MVT::i64, Custom); |
| 77 | |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 78 | setTargetDAGCombine(ISD::SELECT_CC); |
| 79 | |
| 80 | setTargetDAGCombine(ISD::SETCC); |
Michel Danzer | f52a672 | 2013-03-08 10:58:01 +0000 | [diff] [blame] | 81 | |
Christian Konig | eecebd0 | 2013-03-26 14:04:02 +0000 | [diff] [blame] | 82 | setSchedulingPreference(Sched::RegPressure); |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 83 | } |
| 84 | |
Christian Konig | 2c8f6d5 | 2013-03-07 09:03:52 +0000 | [diff] [blame] | 85 | SDValue SITargetLowering::LowerFormalArguments( |
| 86 | SDValue Chain, |
| 87 | CallingConv::ID CallConv, |
| 88 | bool isVarArg, |
| 89 | const SmallVectorImpl<ISD::InputArg> &Ins, |
| 90 | DebugLoc DL, SelectionDAG &DAG, |
| 91 | SmallVectorImpl<SDValue> &InVals) const { |
| 92 | |
| 93 | const TargetRegisterInfo *TRI = getTargetMachine().getRegisterInfo(); |
| 94 | |
| 95 | MachineFunction &MF = DAG.getMachineFunction(); |
| 96 | FunctionType *FType = MF.getFunction()->getFunctionType(); |
Christian Konig | 99ee0f4 | 2013-03-07 09:04:14 +0000 | [diff] [blame] | 97 | SIMachineFunctionInfo *Info = MF.getInfo<SIMachineFunctionInfo>(); |
Christian Konig | 2c8f6d5 | 2013-03-07 09:03:52 +0000 | [diff] [blame] | 98 | |
| 99 | assert(CallConv == CallingConv::C); |
| 100 | |
| 101 | SmallVector<ISD::InputArg, 16> Splits; |
Christian Konig | 99ee0f4 | 2013-03-07 09:04:14 +0000 | [diff] [blame] | 102 | uint32_t Skipped = 0; |
| 103 | |
| 104 | for (unsigned i = 0, e = Ins.size(), PSInputNum = 0; i != e; ++i) { |
Christian Konig | 2c8f6d5 | 2013-03-07 09:03:52 +0000 | [diff] [blame] | 105 | const ISD::InputArg &Arg = Ins[i]; |
| 106 | |
Christian Konig | 99ee0f4 | 2013-03-07 09:04:14 +0000 | [diff] [blame] | 107 | // First check if it's a PS input addr |
| 108 | if (Info->ShaderType == ShaderType::PIXEL && !Arg.Flags.isInReg()) { |
| 109 | |
| 110 | assert((PSInputNum <= 15) && "Too many PS inputs!"); |
| 111 | |
| 112 | if (!Arg.Used) { |
| 113 | // We can savely skip PS inputs |
| 114 | Skipped |= 1 << i; |
| 115 | ++PSInputNum; |
| 116 | continue; |
| 117 | } |
| 118 | |
| 119 | Info->PSInputAddr |= 1 << PSInputNum++; |
| 120 | } |
| 121 | |
| 122 | // Second split vertices into their elements |
Christian Konig | 2c8f6d5 | 2013-03-07 09:03:52 +0000 | [diff] [blame] | 123 | if (Arg.VT.isVector()) { |
| 124 | ISD::InputArg NewArg = Arg; |
| 125 | NewArg.Flags.setSplit(); |
| 126 | NewArg.VT = Arg.VT.getVectorElementType(); |
| 127 | |
| 128 | // We REALLY want the ORIGINAL number of vertex elements here, e.g. a |
| 129 | // three or five element vertex only needs three or five registers, |
| 130 | // NOT four or eigth. |
| 131 | Type *ParamType = FType->getParamType(Arg.OrigArgIndex); |
| 132 | unsigned NumElements = ParamType->getVectorNumElements(); |
| 133 | |
| 134 | for (unsigned j = 0; j != NumElements; ++j) { |
| 135 | Splits.push_back(NewArg); |
| 136 | NewArg.PartOffset += NewArg.VT.getStoreSize(); |
| 137 | } |
| 138 | |
| 139 | } else { |
| 140 | Splits.push_back(Arg); |
| 141 | } |
| 142 | } |
| 143 | |
| 144 | SmallVector<CCValAssign, 16> ArgLocs; |
| 145 | CCState CCInfo(CallConv, isVarArg, DAG.getMachineFunction(), |
| 146 | getTargetMachine(), ArgLocs, *DAG.getContext()); |
| 147 | |
Christian Konig | 99ee0f4 | 2013-03-07 09:04:14 +0000 | [diff] [blame] | 148 | // At least one interpolation mode must be enabled or else the GPU will hang. |
| 149 | if (Info->ShaderType == ShaderType::PIXEL && (Info->PSInputAddr & 0x7F) == 0) { |
| 150 | Info->PSInputAddr |= 1; |
| 151 | CCInfo.AllocateReg(AMDGPU::VGPR0); |
| 152 | CCInfo.AllocateReg(AMDGPU::VGPR1); |
| 153 | } |
| 154 | |
Christian Konig | 2c8f6d5 | 2013-03-07 09:03:52 +0000 | [diff] [blame] | 155 | AnalyzeFormalArguments(CCInfo, Splits); |
| 156 | |
| 157 | for (unsigned i = 0, e = Ins.size(), ArgIdx = 0; i != e; ++i) { |
| 158 | |
Christian Konig | 99ee0f4 | 2013-03-07 09:04:14 +0000 | [diff] [blame] | 159 | if (Skipped & (1 << i)) { |
| 160 | InVals.push_back(SDValue()); |
| 161 | continue; |
| 162 | } |
| 163 | |
Christian Konig | 2c8f6d5 | 2013-03-07 09:03:52 +0000 | [diff] [blame] | 164 | CCValAssign &VA = ArgLocs[ArgIdx++]; |
| 165 | assert(VA.isRegLoc() && "Parameter must be in a register!"); |
| 166 | |
| 167 | unsigned Reg = VA.getLocReg(); |
| 168 | MVT VT = VA.getLocVT(); |
| 169 | |
| 170 | if (VT == MVT::i64) { |
| 171 | // For now assume it is a pointer |
| 172 | Reg = TRI->getMatchingSuperReg(Reg, AMDGPU::sub0, |
| 173 | &AMDGPU::SReg_64RegClass); |
| 174 | Reg = MF.addLiveIn(Reg, &AMDGPU::SReg_64RegClass); |
| 175 | InVals.push_back(DAG.getCopyFromReg(Chain, DL, Reg, VT)); |
| 176 | continue; |
| 177 | } |
| 178 | |
| 179 | const TargetRegisterClass *RC = TRI->getMinimalPhysRegClass(Reg, VT); |
| 180 | |
| 181 | Reg = MF.addLiveIn(Reg, RC); |
| 182 | SDValue Val = DAG.getCopyFromReg(Chain, DL, Reg, VT); |
| 183 | |
| 184 | const ISD::InputArg &Arg = Ins[i]; |
| 185 | if (Arg.VT.isVector()) { |
| 186 | |
| 187 | // Build a vector from the registers |
| 188 | Type *ParamType = FType->getParamType(Arg.OrigArgIndex); |
| 189 | unsigned NumElements = ParamType->getVectorNumElements(); |
| 190 | |
| 191 | SmallVector<SDValue, 4> Regs; |
| 192 | Regs.push_back(Val); |
| 193 | for (unsigned j = 1; j != NumElements; ++j) { |
| 194 | Reg = ArgLocs[ArgIdx++].getLocReg(); |
| 195 | Reg = MF.addLiveIn(Reg, RC); |
| 196 | Regs.push_back(DAG.getCopyFromReg(Chain, DL, Reg, VT)); |
| 197 | } |
| 198 | |
| 199 | // Fill up the missing vector elements |
| 200 | NumElements = Arg.VT.getVectorNumElements() - NumElements; |
| 201 | for (unsigned j = 0; j != NumElements; ++j) |
| 202 | Regs.push_back(DAG.getUNDEF(VT)); |
| 203 | |
| 204 | InVals.push_back(DAG.getNode(ISD::BUILD_VECTOR, DL, Arg.VT, |
| 205 | Regs.data(), Regs.size())); |
| 206 | continue; |
| 207 | } |
| 208 | |
| 209 | InVals.push_back(Val); |
| 210 | } |
| 211 | return Chain; |
| 212 | } |
| 213 | |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 214 | MachineBasicBlock * SITargetLowering::EmitInstrWithCustomInserter( |
| 215 | MachineInstr * MI, MachineBasicBlock * BB) const { |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 216 | |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 217 | switch (MI->getOpcode()) { |
| 218 | default: |
| 219 | return AMDGPUTargetLowering::EmitInstrWithCustomInserter(MI, BB); |
| 220 | case AMDGPU::BRANCH: return BB; |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 221 | } |
| 222 | return BB; |
| 223 | } |
| 224 | |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 225 | EVT SITargetLowering::getSetCCResultType(EVT VT) const { |
| 226 | return MVT::i1; |
| 227 | } |
| 228 | |
Christian Konig | 082a14a | 2013-03-18 11:34:05 +0000 | [diff] [blame] | 229 | MVT SITargetLowering::getScalarShiftAmountTy(EVT VT) const { |
| 230 | return MVT::i32; |
| 231 | } |
| 232 | |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 233 | //===----------------------------------------------------------------------===// |
| 234 | // Custom DAG Lowering Operations |
| 235 | //===----------------------------------------------------------------------===// |
| 236 | |
| 237 | SDValue SITargetLowering::LowerOperation(SDValue Op, SelectionDAG &DAG) const { |
| 238 | switch (Op.getOpcode()) { |
| 239 | default: return AMDGPUTargetLowering::LowerOperation(Op, DAG); |
Tom Stellard | f879435 | 2012-12-19 22:10:31 +0000 | [diff] [blame] | 240 | case ISD::BRCOND: return LowerBRCOND(Op, DAG); |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 241 | case ISD::SELECT_CC: return LowerSELECT_CC(Op, DAG); |
Tom Stellard | 754f80f | 2013-04-05 23:31:51 +0000 | [diff] [blame^] | 242 | case ISD::STORE: return LowerSTORE(Op, DAG); |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 243 | } |
| 244 | return SDValue(); |
| 245 | } |
| 246 | |
Tom Stellard | f879435 | 2012-12-19 22:10:31 +0000 | [diff] [blame] | 247 | /// \brief Helper function for LowerBRCOND |
| 248 | static SDNode *findUser(SDValue Value, unsigned Opcode) { |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 249 | |
Tom Stellard | f879435 | 2012-12-19 22:10:31 +0000 | [diff] [blame] | 250 | SDNode *Parent = Value.getNode(); |
| 251 | for (SDNode::use_iterator I = Parent->use_begin(), E = Parent->use_end(); |
| 252 | I != E; ++I) { |
| 253 | |
| 254 | if (I.getUse().get() != Value) |
| 255 | continue; |
| 256 | |
| 257 | if (I->getOpcode() == Opcode) |
| 258 | return *I; |
| 259 | } |
| 260 | return 0; |
| 261 | } |
| 262 | |
| 263 | /// This transforms the control flow intrinsics to get the branch destination as |
| 264 | /// last parameter, also switches branch target with BR if the need arise |
| 265 | SDValue SITargetLowering::LowerBRCOND(SDValue BRCOND, |
| 266 | SelectionDAG &DAG) const { |
| 267 | |
| 268 | DebugLoc DL = BRCOND.getDebugLoc(); |
| 269 | |
| 270 | SDNode *Intr = BRCOND.getOperand(1).getNode(); |
| 271 | SDValue Target = BRCOND.getOperand(2); |
| 272 | SDNode *BR = 0; |
| 273 | |
| 274 | if (Intr->getOpcode() == ISD::SETCC) { |
| 275 | // As long as we negate the condition everything is fine |
| 276 | SDNode *SetCC = Intr; |
| 277 | assert(SetCC->getConstantOperandVal(1) == 1); |
NAKAMURA Takumi | 458a827 | 2013-01-07 11:14:44 +0000 | [diff] [blame] | 278 | assert(cast<CondCodeSDNode>(SetCC->getOperand(2).getNode())->get() == |
| 279 | ISD::SETNE); |
Tom Stellard | f879435 | 2012-12-19 22:10:31 +0000 | [diff] [blame] | 280 | Intr = SetCC->getOperand(0).getNode(); |
| 281 | |
| 282 | } else { |
| 283 | // Get the target from BR if we don't negate the condition |
| 284 | BR = findUser(BRCOND, ISD::BR); |
| 285 | Target = BR->getOperand(1); |
| 286 | } |
| 287 | |
| 288 | assert(Intr->getOpcode() == ISD::INTRINSIC_W_CHAIN); |
| 289 | |
| 290 | // Build the result and |
| 291 | SmallVector<EVT, 4> Res; |
| 292 | for (unsigned i = 1, e = Intr->getNumValues(); i != e; ++i) |
| 293 | Res.push_back(Intr->getValueType(i)); |
| 294 | |
| 295 | // operands of the new intrinsic call |
| 296 | SmallVector<SDValue, 4> Ops; |
| 297 | Ops.push_back(BRCOND.getOperand(0)); |
| 298 | for (unsigned i = 1, e = Intr->getNumOperands(); i != e; ++i) |
| 299 | Ops.push_back(Intr->getOperand(i)); |
| 300 | Ops.push_back(Target); |
| 301 | |
| 302 | // build the new intrinsic call |
| 303 | SDNode *Result = DAG.getNode( |
| 304 | Res.size() > 1 ? ISD::INTRINSIC_W_CHAIN : ISD::INTRINSIC_VOID, DL, |
| 305 | DAG.getVTList(Res.data(), Res.size()), Ops.data(), Ops.size()).getNode(); |
| 306 | |
| 307 | if (BR) { |
| 308 | // Give the branch instruction our target |
| 309 | SDValue Ops[] = { |
| 310 | BR->getOperand(0), |
| 311 | BRCOND.getOperand(2) |
| 312 | }; |
| 313 | DAG.MorphNodeTo(BR, ISD::BR, BR->getVTList(), Ops, 2); |
| 314 | } |
| 315 | |
| 316 | SDValue Chain = SDValue(Result, Result->getNumValues() - 1); |
| 317 | |
| 318 | // Copy the intrinsic results to registers |
| 319 | for (unsigned i = 1, e = Intr->getNumValues() - 1; i != e; ++i) { |
| 320 | SDNode *CopyToReg = findUser(SDValue(Intr, i), ISD::CopyToReg); |
| 321 | if (!CopyToReg) |
| 322 | continue; |
| 323 | |
| 324 | Chain = DAG.getCopyToReg( |
| 325 | Chain, DL, |
| 326 | CopyToReg->getOperand(1), |
| 327 | SDValue(Result, i - 1), |
| 328 | SDValue()); |
| 329 | |
| 330 | DAG.ReplaceAllUsesWith(SDValue(CopyToReg, 0), CopyToReg->getOperand(0)); |
| 331 | } |
| 332 | |
| 333 | // Remove the old intrinsic from the chain |
| 334 | DAG.ReplaceAllUsesOfValueWith( |
| 335 | SDValue(Intr, Intr->getNumValues() - 1), |
| 336 | Intr->getOperand(0)); |
| 337 | |
| 338 | return Chain; |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 339 | } |
| 340 | |
Tom Stellard | 754f80f | 2013-04-05 23:31:51 +0000 | [diff] [blame^] | 341 | #define RSRC_DATA_FORMAT 0xf00000000000 |
| 342 | |
| 343 | SDValue SITargetLowering::LowerSTORE(SDValue Op, SelectionDAG &DAG) const { |
| 344 | StoreSDNode *StoreNode = cast<StoreSDNode>(Op); |
| 345 | SDValue Chain = Op.getOperand(0); |
| 346 | SDValue Value = Op.getOperand(1); |
| 347 | SDValue VirtualAddress = Op.getOperand(2); |
| 348 | DebugLoc DL = Op.getDebugLoc(); |
| 349 | |
| 350 | if (StoreNode->getAddressSpace() != AMDGPUAS::GLOBAL_ADDRESS) { |
| 351 | return SDValue(); |
| 352 | } |
| 353 | |
| 354 | SDValue SrcSrc = DAG.getNode(ISD::BUILD_PAIR, DL, MVT::i128, |
| 355 | DAG.getConstant(0, MVT::i64), |
| 356 | DAG.getConstant(RSRC_DATA_FORMAT, MVT::i64)); |
| 357 | |
| 358 | SDValue Ops[2]; |
| 359 | Ops[0] = DAG.getNode(AMDGPUISD::BUFFER_STORE, DL, MVT::Other, Chain, |
| 360 | Value, SrcSrc, VirtualAddress); |
| 361 | Ops[1] = Chain; |
| 362 | |
| 363 | return DAG.getMergeValues(Ops, 2, DL); |
| 364 | |
| 365 | } |
| 366 | |
Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 367 | SDValue SITargetLowering::LowerSELECT_CC(SDValue Op, SelectionDAG &DAG) const { |
| 368 | SDValue LHS = Op.getOperand(0); |
| 369 | SDValue RHS = Op.getOperand(1); |
| 370 | SDValue True = Op.getOperand(2); |
| 371 | SDValue False = Op.getOperand(3); |
| 372 | SDValue CC = Op.getOperand(4); |
| 373 | EVT VT = Op.getValueType(); |
| 374 | DebugLoc DL = Op.getDebugLoc(); |
| 375 | |
| 376 | // Possible Min/Max pattern |
| 377 | SDValue MinMax = LowerMinMax(Op, DAG); |
| 378 | if (MinMax.getNode()) { |
| 379 | return MinMax; |
| 380 | } |
| 381 | |
| 382 | SDValue Cond = DAG.getNode(ISD::SETCC, DL, MVT::i1, LHS, RHS, CC); |
| 383 | return DAG.getNode(ISD::SELECT, DL, VT, Cond, True, False); |
| 384 | } |
| 385 | |
| 386 | //===----------------------------------------------------------------------===// |
| 387 | // Custom DAG optimizations |
| 388 | //===----------------------------------------------------------------------===// |
| 389 | |
| 390 | SDValue SITargetLowering::PerformDAGCombine(SDNode *N, |
| 391 | DAGCombinerInfo &DCI) const { |
| 392 | SelectionDAG &DAG = DCI.DAG; |
| 393 | DebugLoc DL = N->getDebugLoc(); |
| 394 | EVT VT = N->getValueType(0); |
| 395 | |
| 396 | switch (N->getOpcode()) { |
| 397 | default: break; |
| 398 | case ISD::SELECT_CC: { |
| 399 | N->dump(); |
| 400 | ConstantSDNode *True, *False; |
| 401 | // i1 selectcc(l, r, -1, 0, cc) -> i1 setcc(l, r, cc) |
| 402 | if ((True = dyn_cast<ConstantSDNode>(N->getOperand(2))) |
| 403 | && (False = dyn_cast<ConstantSDNode>(N->getOperand(3))) |
| 404 | && True->isAllOnesValue() |
| 405 | && False->isNullValue() |
| 406 | && VT == MVT::i1) { |
| 407 | return DAG.getNode(ISD::SETCC, DL, VT, N->getOperand(0), |
| 408 | N->getOperand(1), N->getOperand(4)); |
| 409 | |
| 410 | } |
| 411 | break; |
| 412 | } |
| 413 | case ISD::SETCC: { |
| 414 | SDValue Arg0 = N->getOperand(0); |
| 415 | SDValue Arg1 = N->getOperand(1); |
| 416 | SDValue CC = N->getOperand(2); |
| 417 | ConstantSDNode * C = NULL; |
| 418 | ISD::CondCode CCOp = dyn_cast<CondCodeSDNode>(CC)->get(); |
| 419 | |
| 420 | // i1 setcc (sext(i1), 0, setne) -> i1 setcc(i1, 0, setne) |
| 421 | if (VT == MVT::i1 |
| 422 | && Arg0.getOpcode() == ISD::SIGN_EXTEND |
| 423 | && Arg0.getOperand(0).getValueType() == MVT::i1 |
| 424 | && (C = dyn_cast<ConstantSDNode>(Arg1)) |
| 425 | && C->isNullValue() |
| 426 | && CCOp == ISD::SETNE) { |
| 427 | return SimplifySetCC(VT, Arg0.getOperand(0), |
| 428 | DAG.getConstant(0, MVT::i1), CCOp, true, DCI, DL); |
| 429 | } |
| 430 | break; |
| 431 | } |
| 432 | } |
| 433 | return SDValue(); |
| 434 | } |
Christian Konig | d910b7d | 2013-02-26 17:52:16 +0000 | [diff] [blame] | 435 | |
Christian Konig | f82901a | 2013-02-26 17:52:23 +0000 | [diff] [blame] | 436 | /// \brief Test if RegClass is one of the VSrc classes |
| 437 | static bool isVSrc(unsigned RegClass) { |
| 438 | return AMDGPU::VSrc_32RegClassID == RegClass || |
| 439 | AMDGPU::VSrc_64RegClassID == RegClass; |
| 440 | } |
| 441 | |
| 442 | /// \brief Test if RegClass is one of the SSrc classes |
| 443 | static bool isSSrc(unsigned RegClass) { |
| 444 | return AMDGPU::SSrc_32RegClassID == RegClass || |
| 445 | AMDGPU::SSrc_64RegClassID == RegClass; |
| 446 | } |
| 447 | |
| 448 | /// \brief Analyze the possible immediate value Op |
| 449 | /// |
| 450 | /// Returns -1 if it isn't an immediate, 0 if it's and inline immediate |
| 451 | /// and the immediate value if it's a literal immediate |
| 452 | int32_t SITargetLowering::analyzeImmediate(const SDNode *N) const { |
| 453 | |
| 454 | union { |
| 455 | int32_t I; |
| 456 | float F; |
| 457 | } Imm; |
| 458 | |
Tom Stellard | edbf1eb | 2013-04-05 23:31:20 +0000 | [diff] [blame] | 459 | if (const ConstantSDNode *Node = dyn_cast<ConstantSDNode>(N)) { |
| 460 | if (Node->getZExtValue() >> 32) { |
| 461 | return -1; |
| 462 | } |
Christian Konig | f82901a | 2013-02-26 17:52:23 +0000 | [diff] [blame] | 463 | Imm.I = Node->getSExtValue(); |
Tom Stellard | edbf1eb | 2013-04-05 23:31:20 +0000 | [diff] [blame] | 464 | } else if (const ConstantFPSDNode *Node = dyn_cast<ConstantFPSDNode>(N)) |
Christian Konig | f82901a | 2013-02-26 17:52:23 +0000 | [diff] [blame] | 465 | Imm.F = Node->getValueAPF().convertToFloat(); |
| 466 | else |
| 467 | return -1; // It isn't an immediate |
| 468 | |
| 469 | if ((Imm.I >= -16 && Imm.I <= 64) || |
| 470 | Imm.F == 0.5f || Imm.F == -0.5f || |
| 471 | Imm.F == 1.0f || Imm.F == -1.0f || |
| 472 | Imm.F == 2.0f || Imm.F == -2.0f || |
| 473 | Imm.F == 4.0f || Imm.F == -4.0f) |
| 474 | return 0; // It's an inline immediate |
| 475 | |
| 476 | return Imm.I; // It's a literal immediate |
| 477 | } |
| 478 | |
| 479 | /// \brief Try to fold an immediate directly into an instruction |
| 480 | bool SITargetLowering::foldImm(SDValue &Operand, int32_t &Immediate, |
| 481 | bool &ScalarSlotUsed) const { |
| 482 | |
| 483 | MachineSDNode *Mov = dyn_cast<MachineSDNode>(Operand); |
| 484 | if (Mov == 0 || !TII->isMov(Mov->getMachineOpcode())) |
| 485 | return false; |
| 486 | |
| 487 | const SDValue &Op = Mov->getOperand(0); |
| 488 | int32_t Value = analyzeImmediate(Op.getNode()); |
| 489 | if (Value == -1) { |
| 490 | // Not an immediate at all |
| 491 | return false; |
| 492 | |
| 493 | } else if (Value == 0) { |
| 494 | // Inline immediates can always be fold |
| 495 | Operand = Op; |
| 496 | return true; |
| 497 | |
| 498 | } else if (Value == Immediate) { |
| 499 | // Already fold literal immediate |
| 500 | Operand = Op; |
| 501 | return true; |
| 502 | |
| 503 | } else if (!ScalarSlotUsed && !Immediate) { |
| 504 | // Fold this literal immediate |
| 505 | ScalarSlotUsed = true; |
| 506 | Immediate = Value; |
| 507 | Operand = Op; |
| 508 | return true; |
| 509 | |
| 510 | } |
| 511 | |
| 512 | return false; |
| 513 | } |
| 514 | |
| 515 | /// \brief Does "Op" fit into register class "RegClass" ? |
| 516 | bool SITargetLowering::fitsRegClass(SelectionDAG &DAG, SDValue &Op, |
| 517 | unsigned RegClass) const { |
| 518 | |
| 519 | MachineRegisterInfo &MRI = DAG.getMachineFunction().getRegInfo(); |
| 520 | SDNode *Node = Op.getNode(); |
| 521 | |
Christian Konig | 8370dbb | 2013-03-26 14:04:17 +0000 | [diff] [blame] | 522 | const TargetRegisterClass *OpClass; |
Christian Konig | f82901a | 2013-02-26 17:52:23 +0000 | [diff] [blame] | 523 | if (MachineSDNode *MN = dyn_cast<MachineSDNode>(Node)) { |
| 524 | const MCInstrDesc &Desc = TII->get(MN->getMachineOpcode()); |
Christian Konig | 8370dbb | 2013-03-26 14:04:17 +0000 | [diff] [blame] | 525 | int OpClassID = Desc.OpInfo[Op.getResNo()].RegClass; |
| 526 | if (OpClassID == -1) |
| 527 | OpClass = getRegClassFor(Op.getSimpleValueType()); |
| 528 | else |
| 529 | OpClass = TRI->getRegClass(OpClassID); |
Christian Konig | f82901a | 2013-02-26 17:52:23 +0000 | [diff] [blame] | 530 | |
| 531 | } else if (Node->getOpcode() == ISD::CopyFromReg) { |
| 532 | RegisterSDNode *Reg = cast<RegisterSDNode>(Node->getOperand(1).getNode()); |
Christian Konig | 8370dbb | 2013-03-26 14:04:17 +0000 | [diff] [blame] | 533 | OpClass = MRI.getRegClass(Reg->getReg()); |
Christian Konig | f82901a | 2013-02-26 17:52:23 +0000 | [diff] [blame] | 534 | |
| 535 | } else |
| 536 | return false; |
| 537 | |
Christian Konig | 8370dbb | 2013-03-26 14:04:17 +0000 | [diff] [blame] | 538 | return TRI->getRegClass(RegClass)->hasSubClassEq(OpClass); |
Christian Konig | f82901a | 2013-02-26 17:52:23 +0000 | [diff] [blame] | 539 | } |
| 540 | |
| 541 | /// \brief Make sure that we don't exeed the number of allowed scalars |
| 542 | void SITargetLowering::ensureSRegLimit(SelectionDAG &DAG, SDValue &Operand, |
| 543 | unsigned RegClass, |
| 544 | bool &ScalarSlotUsed) const { |
| 545 | |
| 546 | // First map the operands register class to a destination class |
| 547 | if (RegClass == AMDGPU::VSrc_32RegClassID) |
| 548 | RegClass = AMDGPU::VReg_32RegClassID; |
| 549 | else if (RegClass == AMDGPU::VSrc_64RegClassID) |
| 550 | RegClass = AMDGPU::VReg_64RegClassID; |
| 551 | else |
| 552 | return; |
| 553 | |
| 554 | // Nothing todo if they fit naturaly |
| 555 | if (fitsRegClass(DAG, Operand, RegClass)) |
| 556 | return; |
| 557 | |
| 558 | // If the scalar slot isn't used yet use it now |
| 559 | if (!ScalarSlotUsed) { |
| 560 | ScalarSlotUsed = true; |
| 561 | return; |
| 562 | } |
| 563 | |
| 564 | // This is a conservative aproach, it is possible that we can't determine |
| 565 | // the correct register class and copy too often, but better save than sorry. |
| 566 | SDValue RC = DAG.getTargetConstant(RegClass, MVT::i32); |
| 567 | SDNode *Node = DAG.getMachineNode(TargetOpcode::COPY_TO_REGCLASS, DebugLoc(), |
| 568 | Operand.getValueType(), Operand, RC); |
| 569 | Operand = SDValue(Node, 0); |
| 570 | } |
| 571 | |
Christian Konig | d910b7d | 2013-02-26 17:52:16 +0000 | [diff] [blame] | 572 | SDNode *SITargetLowering::PostISelFolding(MachineSDNode *Node, |
| 573 | SelectionDAG &DAG) const { |
Christian Konig | f82901a | 2013-02-26 17:52:23 +0000 | [diff] [blame] | 574 | |
| 575 | // Original encoding (either e32 or e64) |
| 576 | int Opcode = Node->getMachineOpcode(); |
| 577 | const MCInstrDesc *Desc = &TII->get(Opcode); |
| 578 | |
| 579 | unsigned NumDefs = Desc->getNumDefs(); |
| 580 | unsigned NumOps = Desc->getNumOperands(); |
| 581 | |
Christian Konig | 3c14580 | 2013-03-27 09:12:59 +0000 | [diff] [blame] | 582 | // Commuted opcode if available |
| 583 | int OpcodeRev = Desc->isCommutable() ? TII->commuteOpcode(Opcode) : -1; |
| 584 | const MCInstrDesc *DescRev = OpcodeRev == -1 ? 0 : &TII->get(OpcodeRev); |
| 585 | |
| 586 | assert(!DescRev || DescRev->getNumDefs() == NumDefs); |
| 587 | assert(!DescRev || DescRev->getNumOperands() == NumOps); |
| 588 | |
Christian Konig | e500e44 | 2013-02-26 17:52:47 +0000 | [diff] [blame] | 589 | // e64 version if available, -1 otherwise |
| 590 | int OpcodeE64 = AMDGPU::getVOPe64(Opcode); |
| 591 | const MCInstrDesc *DescE64 = OpcodeE64 == -1 ? 0 : &TII->get(OpcodeE64); |
| 592 | |
| 593 | assert(!DescE64 || DescE64->getNumDefs() == NumDefs); |
| 594 | assert(!DescE64 || DescE64->getNumOperands() == (NumOps + 4)); |
| 595 | |
Christian Konig | f82901a | 2013-02-26 17:52:23 +0000 | [diff] [blame] | 596 | int32_t Immediate = Desc->getSize() == 4 ? 0 : -1; |
| 597 | bool HaveVSrc = false, HaveSSrc = false; |
| 598 | |
| 599 | // First figure out what we alread have in this instruction |
| 600 | for (unsigned i = 0, e = Node->getNumOperands(), Op = NumDefs; |
| 601 | i != e && Op < NumOps; ++i, ++Op) { |
| 602 | |
| 603 | unsigned RegClass = Desc->OpInfo[Op].RegClass; |
| 604 | if (isVSrc(RegClass)) |
| 605 | HaveVSrc = true; |
| 606 | else if (isSSrc(RegClass)) |
| 607 | HaveSSrc = true; |
| 608 | else |
| 609 | continue; |
| 610 | |
| 611 | int32_t Imm = analyzeImmediate(Node->getOperand(i).getNode()); |
| 612 | if (Imm != -1 && Imm != 0) { |
| 613 | // Literal immediate |
| 614 | Immediate = Imm; |
| 615 | } |
| 616 | } |
| 617 | |
| 618 | // If we neither have VSrc nor SSrc it makes no sense to continue |
| 619 | if (!HaveVSrc && !HaveSSrc) |
| 620 | return Node; |
| 621 | |
| 622 | // No scalar allowed when we have both VSrc and SSrc |
| 623 | bool ScalarSlotUsed = HaveVSrc && HaveSSrc; |
| 624 | |
| 625 | // Second go over the operands and try to fold them |
| 626 | std::vector<SDValue> Ops; |
Christian Konig | e500e44 | 2013-02-26 17:52:47 +0000 | [diff] [blame] | 627 | bool Promote2e64 = false; |
Christian Konig | f82901a | 2013-02-26 17:52:23 +0000 | [diff] [blame] | 628 | for (unsigned i = 0, e = Node->getNumOperands(), Op = NumDefs; |
| 629 | i != e && Op < NumOps; ++i, ++Op) { |
| 630 | |
| 631 | const SDValue &Operand = Node->getOperand(i); |
| 632 | Ops.push_back(Operand); |
| 633 | |
| 634 | // Already folded immediate ? |
| 635 | if (isa<ConstantSDNode>(Operand.getNode()) || |
| 636 | isa<ConstantFPSDNode>(Operand.getNode())) |
| 637 | continue; |
| 638 | |
| 639 | // Is this a VSrc or SSrc operand ? |
| 640 | unsigned RegClass = Desc->OpInfo[Op].RegClass; |
Christian Konig | 8370dbb | 2013-03-26 14:04:17 +0000 | [diff] [blame] | 641 | if (isVSrc(RegClass) || isSSrc(RegClass)) { |
| 642 | // Try to fold the immediates |
| 643 | if (!foldImm(Ops[i], Immediate, ScalarSlotUsed)) { |
| 644 | // Folding didn't worked, make sure we don't hit the SReg limit |
| 645 | ensureSRegLimit(DAG, Ops[i], RegClass, ScalarSlotUsed); |
| 646 | } |
| 647 | continue; |
| 648 | } |
Christian Konig | 6612ac3 | 2013-02-26 17:52:36 +0000 | [diff] [blame] | 649 | |
Christian Konig | 3c14580 | 2013-03-27 09:12:59 +0000 | [diff] [blame] | 650 | if (i == 1 && DescRev && fitsRegClass(DAG, Ops[0], RegClass)) { |
Christian Konig | 6612ac3 | 2013-02-26 17:52:36 +0000 | [diff] [blame] | 651 | |
Christian Konig | 8370dbb | 2013-03-26 14:04:17 +0000 | [diff] [blame] | 652 | unsigned OtherRegClass = Desc->OpInfo[NumDefs].RegClass; |
| 653 | assert(isVSrc(OtherRegClass) || isSSrc(OtherRegClass)); |
| 654 | |
| 655 | // Test if it makes sense to swap operands |
| 656 | if (foldImm(Ops[1], Immediate, ScalarSlotUsed) || |
| 657 | (!fitsRegClass(DAG, Ops[1], RegClass) && |
| 658 | fitsRegClass(DAG, Ops[1], OtherRegClass))) { |
Christian Konig | 6612ac3 | 2013-02-26 17:52:36 +0000 | [diff] [blame] | 659 | |
| 660 | // Swap commutable operands |
| 661 | SDValue Tmp = Ops[1]; |
| 662 | Ops[1] = Ops[0]; |
| 663 | Ops[0] = Tmp; |
Christian Konig | 3c14580 | 2013-03-27 09:12:59 +0000 | [diff] [blame] | 664 | |
| 665 | Desc = DescRev; |
| 666 | DescRev = 0; |
Christian Konig | 8370dbb | 2013-03-26 14:04:17 +0000 | [diff] [blame] | 667 | continue; |
Christian Konig | 6612ac3 | 2013-02-26 17:52:36 +0000 | [diff] [blame] | 668 | } |
Christian Konig | 6612ac3 | 2013-02-26 17:52:36 +0000 | [diff] [blame] | 669 | } |
Christian Konig | f82901a | 2013-02-26 17:52:23 +0000 | [diff] [blame] | 670 | |
Christian Konig | 8370dbb | 2013-03-26 14:04:17 +0000 | [diff] [blame] | 671 | if (DescE64 && !Immediate) { |
| 672 | |
| 673 | // Test if it makes sense to switch to e64 encoding |
| 674 | unsigned OtherRegClass = DescE64->OpInfo[Op].RegClass; |
| 675 | if (!isVSrc(OtherRegClass) && !isSSrc(OtherRegClass)) |
| 676 | continue; |
| 677 | |
| 678 | int32_t TmpImm = -1; |
| 679 | if (foldImm(Ops[i], TmpImm, ScalarSlotUsed) || |
| 680 | (!fitsRegClass(DAG, Ops[i], RegClass) && |
| 681 | fitsRegClass(DAG, Ops[1], OtherRegClass))) { |
| 682 | |
| 683 | // Switch to e64 encoding |
| 684 | Immediate = -1; |
| 685 | Promote2e64 = true; |
| 686 | Desc = DescE64; |
| 687 | DescE64 = 0; |
| 688 | } |
Christian Konig | f82901a | 2013-02-26 17:52:23 +0000 | [diff] [blame] | 689 | } |
| 690 | } |
| 691 | |
Christian Konig | e500e44 | 2013-02-26 17:52:47 +0000 | [diff] [blame] | 692 | if (Promote2e64) { |
| 693 | // Add the modifier flags while promoting |
| 694 | for (unsigned i = 0; i < 4; ++i) |
| 695 | Ops.push_back(DAG.getTargetConstant(0, MVT::i32)); |
| 696 | } |
| 697 | |
Christian Konig | f82901a | 2013-02-26 17:52:23 +0000 | [diff] [blame] | 698 | // Add optional chain and glue |
| 699 | for (unsigned i = NumOps - NumDefs, e = Node->getNumOperands(); i < e; ++i) |
| 700 | Ops.push_back(Node->getOperand(i)); |
| 701 | |
Christian Konig | 3c14580 | 2013-03-27 09:12:59 +0000 | [diff] [blame] | 702 | // Create a complete new instruction |
| 703 | return DAG.getMachineNode(Desc->Opcode, Node->getDebugLoc(), |
| 704 | Node->getVTList(), Ops.data(), Ops.size()); |
Christian Konig | d910b7d | 2013-02-26 17:52:16 +0000 | [diff] [blame] | 705 | } |