diff --git a/llvm/lib/Target/SystemZ/SystemZISelLowering.h b/llvm/lib/Target/SystemZ/SystemZISelLowering.h --- a/llvm/lib/Target/SystemZ/SystemZISelLowering.h +++ b/llvm/lib/Target/SystemZ/SystemZISelLowering.h @@ -650,6 +650,7 @@ SDValue lowerSIGN_EXTEND_VECTOR_INREG(SDValue Op, SelectionDAG &DAG) const; SDValue lowerZERO_EXTEND_VECTOR_INREG(SDValue Op, SelectionDAG &DAG) const; SDValue lowerShift(SDValue Op, SelectionDAG &DAG, unsigned ByScalar) const; + SDValue lowerIS_FPCLASS(SDValue Op, SelectionDAG &DAG) const; bool canTreatAsByteVector(EVT VT) const; SDValue combineExtract(const SDLoc &DL, EVT ElemVT, EVT VecVT, SDValue OrigOp, diff --git a/llvm/lib/Target/SystemZ/SystemZISelLowering.cpp b/llvm/lib/Target/SystemZ/SystemZISelLowering.cpp --- a/llvm/lib/Target/SystemZ/SystemZISelLowering.cpp +++ b/llvm/lib/Target/SystemZ/SystemZISelLowering.cpp @@ -473,6 +473,9 @@ setOperationAction(ISD::FREM, VT, Expand); setOperationAction(ISD::FPOW, VT, Expand); + // Special treatment. + setOperationAction(ISD::IS_FPCLASS, VT, Custom); + // Handle constrained floating-point operations. setOperationAction(ISD::STRICT_FADD, VT, Legal); setOperationAction(ISD::STRICT_FSUB, VT, Legal); @@ -5464,6 +5467,42 @@ return Op; } +SDValue SystemZTargetLowering::lowerIS_FPCLASS(SDValue Op, + SelectionDAG &DAG) const { + SDLoc DL(Op); + MVT ResultVT = Op.getSimpleValueType(); + SDValue Arg = Op.getOperand(0); + MVT ArgVT = Arg.getSimpleValueType(); + auto CNode = cast(Op.getOperand(1)); + unsigned Check = CNode->getZExtValue(); + + unsigned TDCMask = 0; + if (Check & ISD::fcSNan) + TDCMask |= SystemZ::TDCMASK_SNAN_PLUS | SystemZ::TDCMASK_SNAN_MINUS; + if (Check & ISD::fcQNan) + TDCMask |= SystemZ::TDCMASK_QNAN_PLUS | SystemZ::TDCMASK_QNAN_MINUS; + if (Check & ISD::fcPosInf) + TDCMask |= SystemZ::TDCMASK_INFINITY_PLUS; + if (Check & ISD::fcNegInf) + TDCMask |= SystemZ::TDCMASK_INFINITY_MINUS; + if (Check & ISD::fcPosNormal) + TDCMask |= SystemZ::TDCMASK_NORMAL_PLUS; + if (Check & ISD::fcNegNormal) + TDCMask |= SystemZ::TDCMASK_NORMAL_MINUS; + if (Check & ISD::fcPosSubnormal) + TDCMask |= SystemZ::TDCMASK_SUBNORMAL_PLUS; + if (Check & ISD::fcNegSubnormal) + TDCMask |= SystemZ::TDCMASK_SUBNORMAL_MINUS; + if (Check & ISD::fcPosZero) + TDCMask |= SystemZ::TDCMASK_ZERO_PLUS; + if (Check & ISD::fcNegZero) + TDCMask |= SystemZ::TDCMASK_ZERO_MINUS; + SDValue TDCMaskV = DAG.getConstant(TDCMask, DL, MVT::i32); + + SDValue Intr = DAG.getNode(SystemZISD::TDC, DL, ResultVT, Arg, TDCMaskV); + return getCCResult(DAG, Intr); +} + SDValue SystemZTargetLowering::LowerOperation(SDValue Op, SelectionDAG &DAG) const { switch (Op.getOpcode()) { @@ -5581,6 +5620,8 @@ return lowerShift(Op, DAG, SystemZISD::VSRL_BY_SCALAR); case ISD::SRA: return lowerShift(Op, DAG, SystemZISD::VSRA_BY_SCALAR); + case ISD::IS_FPCLASS: + return lowerIS_FPCLASS(Op, DAG); default: llvm_unreachable("Unexpected node to lower"); } diff --git a/llvm/test/CodeGen/SystemZ/fpclass.ll b/llvm/test/CodeGen/SystemZ/fpclass.ll new file mode 100644 --- /dev/null +++ b/llvm/test/CodeGen/SystemZ/fpclass.ll @@ -0,0 +1,144 @@ +; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py +; Test intrinsic 'is_fpclass'. +; +; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s + +declare i1 @llvm.is.fpclass.f32(float, metadata) +declare i1 @llvm.is.fpclass.f64(double, metadata) +declare i1 @llvm.is.fpclass.f128(fp128, metadata) + + +define i1 @isnan_f(float %x) { +; CHECK-LABEL: isnan_f: +; CHECK: # %bb.0: +; CHECK-NEXT: tceb %f0, 15 +; CHECK-NEXT: ipm %r2 +; CHECK-NEXT: srl %r2, 28 +; CHECK-NEXT: br %r14 + %1 = call i1 @llvm.is.fpclass.f32(float %x, metadata !"nan") + ret i1 %1 +} + +define i1 @isnan_d(double %x) { +; CHECK-LABEL: isnan_d: +; CHECK: # %bb.0: +; CHECK-NEXT: tcdb %f0, 15 +; CHECK-NEXT: ipm %r2 +; CHECK-NEXT: srl %r2, 28 +; CHECK-NEXT: br %r14 + %1 = call i1 @llvm.is.fpclass.f64(double %x, metadata !"nan") + ret i1 %1 +} + +define i1 @isnan_x(fp128 %x) { +; CHECK-LABEL: isnan_x: +; CHECK: # %bb.0: +; CHECK-NEXT: ld %f0, 0(%r2) +; CHECK-NEXT: ld %f2, 8(%r2) +; CHECK-NEXT: tcxb %f0, 15 +; CHECK-NEXT: ipm %r2 +; CHECK-NEXT: srl %r2, 28 +; CHECK-NEXT: br %r14 + %1 = call i1 @llvm.is.fpclass.f128(fp128 %x, metadata !"nan") + ret i1 %1 +} + +define i1 @isqnan_f(float %x) { +; CHECK-LABEL: isqnan_f: +; CHECK: # %bb.0: +; CHECK-NEXT: tceb %f0, 12 +; CHECK-NEXT: ipm %r2 +; CHECK-NEXT: srl %r2, 28 +; CHECK-NEXT: br %r14 + %1 = call i1 @llvm.is.fpclass.f32(float %x, metadata !"qnan") + ret i1 %1 +} + +define i1 @issnan_f(float %x) { +; CHECK-LABEL: issnan_f: +; CHECK: # %bb.0: +; CHECK-NEXT: tceb %f0, 3 +; CHECK-NEXT: ipm %r2 +; CHECK-NEXT: srl %r2, 28 +; CHECK-NEXT: br %r14 + %1 = call i1 @llvm.is.fpclass.f32(float %x, metadata !"snan") + ret i1 %1 +} + +define i1 @isinf_f(float %x) { +; CHECK-LABEL: isinf_f: +; CHECK: # %bb.0: +; CHECK-NEXT: tceb %f0, 48 +; CHECK-NEXT: ipm %r2 +; CHECK-NEXT: srl %r2, 28 +; CHECK-NEXT: br %r14 + %1 = call i1 @llvm.is.fpclass.f32(float %x, metadata !"inf") + ret i1 %1 +} + +define i1 @isposinf_f(float %x) { +; CHECK-LABEL: isposinf_f: +; CHECK: # %bb.0: +; CHECK-NEXT: tceb %f0, 32 +; CHECK-NEXT: ipm %r2 +; CHECK-NEXT: srl %r2, 28 +; CHECK-NEXT: br %r14 + %1 = call i1 @llvm.is.fpclass.f32(float %x, metadata !"+inf") + ret i1 %1 +} + +define i1 @isneginf_f(float %x) { +; CHECK-LABEL: isneginf_f: +; CHECK: # %bb.0: +; CHECK-NEXT: tceb %f0, 16 +; CHECK-NEXT: ipm %r2 +; CHECK-NEXT: srl %r2, 28 +; CHECK-NEXT: br %r14 + %1 = call i1 @llvm.is.fpclass.f32(float %x, metadata !"-inf") + ret i1 %1 +} + +define i1 @isfinite_f(float %x) { +; CHECK-LABEL: isfinite_f: +; CHECK: # %bb.0: +; CHECK-NEXT: tceb %f0, 4032 +; CHECK-NEXT: ipm %r2 +; CHECK-NEXT: srl %r2, 28 +; CHECK-NEXT: br %r14 + %1 = call i1 @llvm.is.fpclass.f32(float %x, metadata !"finite") + ret i1 %1 +} + +define i1 @isposfinite_f(float %x) { +; CHECK-LABEL: isposfinite_f: +; CHECK: # %bb.0: +; CHECK-NEXT: tceb %f0, 2688 +; CHECK-NEXT: ipm %r2 +; CHECK-NEXT: srl %r2, 28 +; CHECK-NEXT: br %r14 + %1 = call i1 @llvm.is.fpclass.f32(float %x, metadata !"+finite") + ret i1 %1 +} + +define i1 @isnegfinite_f(float %x) { +; CHECK-LABEL: isnegfinite_f: +; CHECK: # %bb.0: +; CHECK-NEXT: tceb %f0, 1344 +; CHECK-NEXT: ipm %r2 +; CHECK-NEXT: srl %r2, 28 +; CHECK-NEXT: br %r14 + %1 = call i1 @llvm.is.fpclass.f32(float %x, metadata !"-finite") + ret i1 %1 +} + +define i1 @isnotfinite_f(float %x) { +; CHECK-LABEL: isnotfinite_f: +; CHECK: # %bb.0: +; CHECK-NEXT: tceb %f0, 63 +; CHECK-NEXT: ipm %r2 +; CHECK-NEXT: srl %r2, 28 +; CHECK-NEXT: br %r14 + %1 = call i1 @llvm.is.fpclass.f32(float %x, metadata !"nan|inf") + ret i1 %1 +} +