Optimize the sequence blend(sign_extend(x)) to blend(shl(x)) since SSE blend instructions only look at the highest bit.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@147426 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Nadav Rotem 2012-01-02 08:05:46 +00:00
parent 47f79bb58e
commit a46f35d3d6
5 changed files with 49 additions and 10 deletions

View File

@ -13133,6 +13133,24 @@ static SDValue PerformSELECTCombine(SDNode *N, SelectionDAG &DAG,
}
}
// The VSELECT instruction is lowered to SSE blend instructions. In many cases
// the mask is sign-extended to fill the entire lane. However, we only care
// for the highest bit. Convert sign_extend to srl because it is cheaper.
// (vselect(sign_extend(x))) -> vselect(srl(x))
if (N->getOpcode() == ISD::VSELECT &&
Cond.getOpcode() == ISD::SIGN_EXTEND_INREG && Cond.hasOneUse()) {
EVT CondVT = Cond.getValueType();
EVT SExtTy = cast<VTSDNode>(Cond.getOperand(1))->getVT();
unsigned BitsDiff = CondVT.getScalarType().getSizeInBits() -
SExtTy.getScalarType().getSizeInBits();
EVT ShiftType = EVT::getVectorVT(*DAG.getContext(),
MVT::i32, CondVT.getVectorNumElements());
SDValue SHL = DAG.getNode(ISD::SHL, DL, CondVT, Cond.getOperand(0),
DAG.getConstant(BitsDiff, ShiftType));
return DAG.getNode(ISD::VSELECT, DL, VT, SHL, LHS, RHS);
}
return SDValue();
}

View File

@ -5,8 +5,10 @@ target triple = "x86_64-apple-darwin11.2.0"
; CHECK: @foo8
; CHECK: psll
; CHECK: psraw
; CHECK: pblendvb
; CHECK-NOT: sra
; CHECK: pandn
; CHECK: pand
; CHECK: or
; CHECK: ret
define void @foo8(float* nocapture %RET) nounwind {
allocas:

View File

@ -0,0 +1,15 @@
; RUN: llc < %s -mtriple=x86_64-apple-darwin -mcpu=corei7 -promote-elements -mattr=+sse41 | FileCheck %s
; CHECK: foo
define <4 x double> @foo(<4 x double> %x, <4 x double> %y) {
; CHECK: cmpnlepd
; CHECK: psllq
; CHECK-NEXT: blendvpd
; CHECK: psllq
; CHECK-NEXT: blendvpd
; CHECK: ret
%min_is_x = fcmp ult <4 x double> %x, %y
%min = select <4 x i1> %min_is_x, <4 x double> %x, <4 x double> %y
ret <4 x double> %min
}

View File

@ -28,10 +28,10 @@ define void@vsel_i32(<4 x i32>* %v1, <4 x i32>* %v2) {
; Without forcing instructions, fall back to the preferred PS domain.
; CHECK: vsel_i64
; CHECK: xorps
; CHECK: andps
; CHECK: andnps
; CHECK: orps
; CHECK: pxor
; CHECK: and
; CHECK: andn
; CHECK: or
; CHECK: ret
define void@vsel_i64(<4 x i64>* %v1, <4 x i64>* %v2) {
@ -44,10 +44,10 @@ define void@vsel_i64(<4 x i64>* %v1, <4 x i64>* %v2) {
; Without forcing instructions, fall back to the preferred PS domain.
; CHECK: vsel_double
; CHECK: xorps
; CHECK: andps
; CHECK: andnps
; CHECK: orps
; CHECK: xor
; CHECK: and
; CHECK: andn
; CHECK: or
; CHECK: ret
define void@vsel_double(<4 x double>* %v1, <4 x double>* %v2) {

View File

@ -36,6 +36,7 @@ define <4 x i32> @vsel_i32(<4 x i32> %v1, <4 x i32> %v2) {
;CHECK: vsel_double
;CHECK-NOT: sra
;CHECK: blendvpd
;CHECK: ret
define <4 x double> @vsel_double(<4 x double> %v1, <4 x double> %v2) {
@ -54,6 +55,7 @@ define <4 x i64> @vsel_i64(<4 x i64> %v1, <4 x i64> %v2) {
;CHECK: vsel_i8
;CHECK-NOT: sra
;CHECK: pblendvb
;CHECK: ret
define <16 x i8> @vsel_i8(<16 x i8> %v1, <16 x i8> %v2) {
@ -65,6 +67,7 @@ define <16 x i8> @vsel_i8(<16 x i8> %v1, <16 x i8> %v2) {
; CHECK: A
define <2 x double> @A(<2 x double> %x, <2 x double> %y) {
; CHECK: cmplepd
; CHECK-NOT: sra
; CHECK: blendvpd
%max_is_x = fcmp oge <2 x double> %x, %y
%max = select <2 x i1> %max_is_x, <2 x double> %x, <2 x double> %y
@ -74,6 +77,7 @@ define <2 x double> @A(<2 x double> %x, <2 x double> %y) {
; CHECK: B
define <2 x double> @B(<2 x double> %x, <2 x double> %y) {
; CHECK: cmpnlepd
; CHECK-NOT: sra
; CHECK: blendvpd
%min_is_x = fcmp ult <2 x double> %x, %y
%min = select <2 x i1> %min_is_x, <2 x double> %x, <2 x double> %y