diff options
Diffstat (limited to 'llvm/lib')
| -rw-r--r-- | llvm/lib/Target/ARM/ARMISelDAGToDAG.cpp | 84 | ||||
| -rw-r--r-- | llvm/lib/Target/ARM/ARMInstrNEON.td | 24 | ||||
| -rw-r--r-- | llvm/lib/Target/ARM/NEONPreAllocPass.cpp | 16 | 
3 files changed, 105 insertions, 19 deletions
| diff --git a/llvm/lib/Target/ARM/ARMISelDAGToDAG.cpp b/llvm/lib/Target/ARM/ARMISelDAGToDAG.cpp index ba11f8cde05..a5d79c6c38b 100644 --- a/llvm/lib/Target/ARM/ARMISelDAGToDAG.cpp +++ b/llvm/lib/Target/ARM/ARMISelDAGToDAG.cpp @@ -1596,7 +1596,7 @@ SDNode *ARMDAGToDAGISel::Select(SDValue Op) {        EVT RegVT;        unsigned Opc2 = 0;        switch (VT.getSimpleVT().SimpleTy) { -      default: llvm_unreachable("unhandled vld2lane type"); +      default: llvm_unreachable("unhandled vld3lane type");        case MVT::v8i16:          Opc = ARM::VLD3LNq16a;          Opc2 = ARM::VLD3LNq16b; @@ -1650,21 +1650,83 @@ SDNode *ARMDAGToDAGISel::Select(SDValue Op) {        SDValue MemAddr, MemUpdate, MemOpc;        if (!SelectAddrMode6(Op, N->getOperand(2), MemAddr, MemUpdate, MemOpc))          return NULL; +      if (VT.is64BitVector()) { +        switch (VT.getSimpleVT().SimpleTy) { +        default: llvm_unreachable("unhandled vld4lane type"); +        case MVT::v8i8:  Opc = ARM::VLD4LNd8; break; +        case MVT::v4i16: Opc = ARM::VLD4LNd16; break; +        case MVT::v2f32: +        case MVT::v2i32: Opc = ARM::VLD4LNd32; break; +        } +        SDValue Chain = N->getOperand(0); +        const SDValue Ops[] = { MemAddr, MemUpdate, MemOpc, +                                N->getOperand(3), N->getOperand(4), +                                N->getOperand(5), N->getOperand(6), +                                N->getOperand(7), Chain }; +        std::vector<EVT> ResTys(4, VT); +        ResTys.push_back(MVT::Other); +        return CurDAG->getMachineNode(Opc, dl, ResTys, Ops, 9); +      } +      // Quad registers are handled by extracting subregs, doing the load, +      // and then inserting the results as subregs. +      EVT RegVT; +      unsigned Opc2 = 0;        switch (VT.getSimpleVT().SimpleTy) {        default: llvm_unreachable("unhandled vld4lane type"); -      case MVT::v8i8:  Opc = ARM::VLD4LNd8; break; -      case MVT::v4i16: Opc = ARM::VLD4LNd16; break; -      case MVT::v2f32: -      case MVT::v2i32: Opc = ARM::VLD4LNd32; break; +      case MVT::v8i16: +        Opc = ARM::VLD4LNq16a; +        Opc2 = ARM::VLD4LNq16b; +        RegVT = MVT::v4i16; +        break; +      case MVT::v4f32: +        Opc = ARM::VLD4LNq32a; +        Opc2 = ARM::VLD4LNq32b; +        RegVT = MVT::v2f32; +        break; +      case MVT::v4i32: +        Opc = ARM::VLD4LNq32a; +        Opc2 = ARM::VLD4LNq32b; +        RegVT = MVT::v2i32; +        break;        }        SDValue Chain = N->getOperand(0); -      const SDValue Ops[] = { MemAddr, MemUpdate, MemOpc, -                              N->getOperand(3), N->getOperand(4), -                              N->getOperand(5), N->getOperand(6), -                              N->getOperand(7), Chain }; -      std::vector<EVT> ResTys(4, VT); +      unsigned Lane = cast<ConstantSDNode>(N->getOperand(7))->getZExtValue(); +      unsigned NumElts = RegVT.getVectorNumElements(); +      int SubregIdx = (Lane < NumElts) ? ARM::DSUBREG_0 : ARM::DSUBREG_1; + +      SDValue D0 = CurDAG->getTargetExtractSubreg(SubregIdx, dl, RegVT, +                                                  N->getOperand(3)); +      SDValue D1 = CurDAG->getTargetExtractSubreg(SubregIdx, dl, RegVT, +                                                  N->getOperand(4)); +      SDValue D2 = CurDAG->getTargetExtractSubreg(SubregIdx, dl, RegVT, +                                                  N->getOperand(5)); +      SDValue D3 = CurDAG->getTargetExtractSubreg(SubregIdx, dl, RegVT, +                                                  N->getOperand(6)); +      const SDValue Ops[] = { MemAddr, MemUpdate, MemOpc, D0, D1, D2, D3, +                              getI32Imm(Lane % NumElts), Chain }; +      std::vector<EVT> ResTys(4, RegVT);        ResTys.push_back(MVT::Other); -      return CurDAG->getMachineNode(Opc, dl, ResTys, Ops, 9); +      SDNode *VLdLn = CurDAG->getMachineNode((Lane < NumElts) ? Opc : Opc2, +                                             dl, ResTys, Ops, 9); +      SDValue Q0 = CurDAG->getTargetInsertSubreg(SubregIdx, dl, VT, +                                                 N->getOperand(3), +                                                 SDValue(VLdLn, 0)); +      SDValue Q1 = CurDAG->getTargetInsertSubreg(SubregIdx, dl, VT, +                                                 N->getOperand(4), +                                                 SDValue(VLdLn, 1)); +      SDValue Q2 = CurDAG->getTargetInsertSubreg(SubregIdx, dl, VT, +                                                 N->getOperand(5), +                                                 SDValue(VLdLn, 2)); +      SDValue Q3 = CurDAG->getTargetInsertSubreg(SubregIdx, dl, VT, +                                                 N->getOperand(6), +                                                 SDValue(VLdLn, 3)); +      Chain = SDValue(VLdLn, 4); +      ReplaceUses(SDValue(N, 0), Q0); +      ReplaceUses(SDValue(N, 1), Q1); +      ReplaceUses(SDValue(N, 2), Q2); +      ReplaceUses(SDValue(N, 3), Q3); +      ReplaceUses(SDValue(N, 4), Chain); +      return NULL;      }      case Intrinsic::arm_neon_vst2: { diff --git a/llvm/lib/Target/ARM/ARMInstrNEON.td b/llvm/lib/Target/ARM/ARMInstrNEON.td index 5c4ccc95352..b8dd6632ad6 100644 --- a/llvm/lib/Target/ARM/ARMInstrNEON.td +++ b/llvm/lib/Target/ARM/ARMInstrNEON.td @@ -299,15 +299,15 @@ def VLD3LNd16 : VLD3LN<0b0110, "vld3.16">;  def VLD3LNd32 : VLD3LN<0b1010, "vld3.32">;  // vld3 to double-spaced even registers. -def VLD3LNq16a: VLD3LN<0b0101, "vld3.16">; -def VLD3LNq32a: VLD3LN<0b1001, "vld3.32">; +def VLD3LNq16a: VLD3LN<0b0110, "vld3.16">; +def VLD3LNq32a: VLD3LN<0b1010, "vld3.32">;  // vld3 to double-spaced odd registers. -def VLD3LNq16b: VLD3LN<0b0101, "vld3.16">; -def VLD3LNq32b: VLD3LN<0b1001, "vld3.32">; +def VLD3LNq16b: VLD3LN<0b0110, "vld3.16">; +def VLD3LNq32b: VLD3LN<0b1010, "vld3.32">;  //   VLD4LN   : Vector Load (single 4-element structure to one lane) -class VLD4LND<bits<4> op11_8, string OpcodeStr> +class VLD4LN<bits<4> op11_8, string OpcodeStr>    : NLdSt<1,0b10,op11_8,0b0000,            (outs DPR:$dst1, DPR:$dst2, DPR:$dst3, DPR:$dst4),            (ins addrmode6:$addr, DPR:$src1, DPR:$src2, DPR:$src3, DPR:$src4, @@ -316,9 +316,17 @@ class VLD4LND<bits<4> op11_8, string OpcodeStr>            "\t\\{$dst1[$lane],$dst2[$lane],$dst3[$lane],$dst4[$lane]\\}, $addr"),            "$src1 = $dst1, $src2 = $dst2, $src3 = $dst3, $src4 = $dst4", []>; -def VLD4LNd8  : VLD4LND<0b0011, "vld4.8">; -def VLD4LNd16 : VLD4LND<0b0111, "vld4.16">; -def VLD4LNd32 : VLD4LND<0b1011, "vld4.32">; +def VLD4LNd8  : VLD4LN<0b0011, "vld4.8">; +def VLD4LNd16 : VLD4LN<0b0111, "vld4.16">; +def VLD4LNd32 : VLD4LN<0b1011, "vld4.32">; + +// vld4 to double-spaced even registers. +def VLD4LNq16a: VLD4LN<0b0111, "vld4.16">; +def VLD4LNq32a: VLD4LN<0b1011, "vld4.32">; + +// vld4 to double-spaced odd registers. +def VLD4LNq16b: VLD4LN<0b0111, "vld4.16">; +def VLD4LNq32b: VLD4LN<0b1011, "vld4.32">;  //   VLD1DUP  : Vector Load (single element to all lanes)  //   VLD2DUP  : Vector Load (single 2-element structure to all lanes) diff --git a/llvm/lib/Target/ARM/NEONPreAllocPass.cpp b/llvm/lib/Target/ARM/NEONPreAllocPass.cpp index 53bc9c8d63e..f909d48afa4 100644 --- a/llvm/lib/Target/ARM/NEONPreAllocPass.cpp +++ b/llvm/lib/Target/ARM/NEONPreAllocPass.cpp @@ -154,6 +154,22 @@ static bool isNEONMultiRegOp(int Opcode, unsigned &FirstOpnd, unsigned &NumRegs,      Stride = 2;      return true; +  case ARM::VLD4LNq16a: +  case ARM::VLD4LNq32a: +    FirstOpnd = 0; +    NumRegs = 4; +    Offset = 0; +    Stride = 2; +    return true; + +  case ARM::VLD4LNq16b: +  case ARM::VLD4LNq32b: +    FirstOpnd = 0; +    NumRegs = 4; +    Offset = 1; +    Stride = 2; +    return true; +    case ARM::VST2d8:    case ARM::VST2d16:    case ARM::VST2d32: | 

