diff --git a/llvm/docs/LangRef.rst b/llvm/docs/LangRef.rst --- a/llvm/docs/LangRef.rst +++ b/llvm/docs/LangRef.rst @@ -17727,6 +17727,64 @@ ``TargetTransformInfo::hasActiveVectorLength()`` returns true when the target has native support for %evl. +.. _int_vp_select: + +'``llvm.vp.select.*``' Intrinsics +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ + +Syntax: +""""""" +This is an overloaded intrinsic. + +:: + + declare <16 x i32> @llvm.vp.select.v16i32 (<16 x i1> , <16 x i32> , <16 x i32> , i32 ) + declare @llvm.vp.select.nxv4i64 ( , , , i32 ) + +Overview: +""""""""" + +The '``llvm.vp.select``' intrinsic is used to choose one value based on a +condition, without IR-level branching. + +Arguments: +"""""""""" + +The first operand is a vector of ``i1`` and indicates the mask. The second +operand is the value that is selected when the condition is true. The third +operand is the value that is selected when the condition is false. +The vectors must be of the same size. +The fourth operand is the explicit vector length. + +#. The optional ``fast-math flags`` marker indicates that the select has one or more + :ref:`fast-math flags `. These are optimization hints to enable + otherwise unsafe floating-point optimizations. Fast-math flags are only valid + for selects that return a floating-point scalar or vector type, or an array + (nested to any depth) of floating-point scalar or vector types. + +Semantics: +"""""""""" + +The intrinsic selects lanes from the second and third operand depending on a +condition. + +For all lanes at positions below ``%evl``, the mask parameter is the condition. +For all lanes below ``%evl`` where the mask is true the lane is taken from the second operand. +Otherwise, the lane is taken from the third operand. +All result lanes at positions greater or equal than ``%evl`` are undefined. + +Example: +"""""""" + +.. code-block:: llvm + + %r = call <4 x i32> @llvm.vp.select.v4i32(<4 x i1> %m, <4 x i32> %on_true, <4 x i32> %on_false, i32 %evl) + + ;;; Expansion. + ;; Any result is legal on lanes at and above %evl. + %also.r = select <4 x i1> %m, <4 x i32> %on_true, <4 x i32> %on_false + + .. _int_vp_add: diff --git a/llvm/include/llvm/IR/Intrinsics.td b/llvm/include/llvm/IR/Intrinsics.td --- a/llvm/include/llvm/IR/Intrinsics.td +++ b/llvm/include/llvm/IR/Intrinsics.td @@ -1507,6 +1507,12 @@ LLVMScalarOrSameVectorWidth<0, llvm_i1_ty>, llvm_i32_ty]>; } +// Shuffles. +def int_vp_select : DefaultAttrsIntrinsic<[ llvm_anyvector_ty ], + [ LLVMScalarOrSameVectorWidth<0, llvm_i1_ty>, + LLVMMatchType<0>, + LLVMMatchType<0>, + llvm_i32_ty]>; // Reductions let IntrProperties = [IntrSpeculatable, IntrNoMem, IntrNoSync, IntrWillReturn] in { diff --git a/llvm/include/llvm/IR/VPIntrinsics.def b/llvm/include/llvm/IR/VPIntrinsics.def --- a/llvm/include/llvm/IR/VPIntrinsics.def +++ b/llvm/include/llvm/IR/VPIntrinsics.def @@ -325,6 +325,16 @@ ///// } Reduction +///// Shuffles { + +// llvm.vp.select(mask,on_true,on_false,vlen) +BEGIN_REGISTER_VP_INTRINSIC(vp_select, 0, 3) +// BEGIN_REGISTER_VP_SDNODE(VP_SELECT, -1, vp_select, 0, 4) +// END_REGISTER_CASES(vp_select, VP_SELECT) +END_REGISTER_VP_INTRINSIC(vp_select) + +///// } Shuffles + #undef BEGIN_REGISTER_VP #undef BEGIN_REGISTER_VP_INTRINSIC #undef BEGIN_REGISTER_VP_SDNODE diff --git a/llvm/lib/IR/IntrinsicInst.cpp b/llvm/lib/IR/IntrinsicInst.cpp --- a/llvm/lib/IR/IntrinsicInst.cpp +++ b/llvm/lib/IR/IntrinsicInst.cpp @@ -482,6 +482,9 @@ VPFunc = Intrinsic::getDeclaration(M, VPID, OverloadTy); break; } + case Intrinsic::vp_select: + VPFunc = Intrinsic::getDeclaration(M, VPID, {Params[1]->getType()}); + break; case Intrinsic::vp_load: VPFunc = Intrinsic::getDeclaration( M, VPID, diff --git a/llvm/unittests/IR/VPIntrinsicTest.cpp b/llvm/unittests/IR/VPIntrinsicTest.cpp --- a/llvm/unittests/IR/VPIntrinsicTest.cpp +++ b/llvm/unittests/IR/VPIntrinsicTest.cpp @@ -68,6 +68,8 @@ Str << " declare float @llvm.vp.reduce." << ReductionOpcode << ".v8f32(float, <8 x float>, <8 x i1>, i32) "; + Str << " declare <8 x i32> @llvm.vp.select.v8i32(<8 x i1>, <8 x i32>, <8 x " + "i32>, i32)"; return parseAssemblyString(Str.str(), Err, C); } };