diff --git a/llvm/lib/Analysis/InstructionSimplify.cpp b/llvm/lib/Analysis/InstructionSimplify.cpp index e397a228afee0..d1ac8d9fbdfd1 100644 --- a/llvm/lib/Analysis/InstructionSimplify.cpp +++ b/llvm/lib/Analysis/InstructionSimplify.cpp @@ -6969,6 +6969,23 @@ static Value *simplifyIntrinsic(CallBase *Call, Value *Callee, } return nullptr; } + case Intrinsic::experimental_vp_reverse: { + Value *Vec = Call->getArgOperand(0); + Value *Mask = Call->getArgOperand(1); + Value *EVL = Call->getArgOperand(2); + + Value *X; + // vp.reverse(vp.reverse(X)) == X (with all ones mask and matching EVL) + if (match(Mask, m_AllOnes()) && + match(Vec, m_Intrinsic( + m_Value(X), m_AllOnes(), m_Specific(EVL)))) + return X; + + // vp.reverse(splat(X)) -> splat(X) (regardless of mask and EVL) + if (isSplatValue(Vec)) + return Vec; + return nullptr; + } default: return nullptr; } diff --git a/llvm/test/Transforms/InstSimplify/vp-reverse.ll b/llvm/test/Transforms/InstSimplify/vp-reverse.ll index 3c3bb871dc610..f19a2ac8ca9e1 100644 --- a/llvm/test/Transforms/InstSimplify/vp-reverse.ll +++ b/llvm/test/Transforms/InstSimplify/vp-reverse.ll @@ -3,9 +3,7 @@ define @rev_of_rev( %a, i32 %evl) { ; CHECK-LABEL: @rev_of_rev( -; CHECK-NEXT: [[A_REV:%.*]] = tail call @llvm.experimental.vp.reverse.nxv4i32( [[A:%.*]], splat (i1 true), i32 [[EVL:%.*]]) -; CHECK-NEXT: [[RES:%.*]] = tail call @llvm.experimental.vp.reverse.nxv4i32( [[A_REV]], splat (i1 true), i32 [[EVL]]) -; CHECK-NEXT: ret [[RES]] +; CHECK-NEXT: ret [[A:%.*]] ; %a.rev = tail call @llvm.experimental.vp.reverse( %a, splat (i1 true), i32 %evl) %res = tail call @llvm.experimental.vp.reverse( %a.rev, splat (i1 true), i32 %evl) @@ -25,8 +23,7 @@ define @rev_of_rev_diffevl( %a, i32 %evl) { define @rev_of_poison(i32 %evl) { ; CHECK-LABEL: @rev_of_poison( -; CHECK-NEXT: [[REV:%.*]] = tail call @llvm.experimental.vp.reverse.nxv4i32( poison, splat (i1 true), i32 [[EVL:%.*]]) -; CHECK-NEXT: ret [[REV]] +; CHECK-NEXT: ret poison ; %rev = tail call @llvm.experimental.vp.reverse( poison, splat (i1 true), i32 %evl) ret %rev @@ -34,8 +31,7 @@ define @rev_of_poison(i32 %evl) { define @rev_of_undef(i32 %evl) { ; CHECK-LABEL: @rev_of_undef( -; CHECK-NEXT: [[REV:%.*]] = tail call @llvm.experimental.vp.reverse.nxv4i32( undef, splat (i1 true), i32 [[EVL:%.*]]) -; CHECK-NEXT: ret [[REV]] +; CHECK-NEXT: ret undef ; %rev = tail call @llvm.experimental.vp.reverse( undef, splat (i1 true), i32 %evl) ret %rev @@ -43,8 +39,7 @@ define @rev_of_undef(i32 %evl) { define @rev_of_zero(i32 %evl) { ; CHECK-LABEL: @rev_of_zero( -; CHECK-NEXT: [[REV:%.*]] = tail call @llvm.experimental.vp.reverse.nxv4i32( zeroinitializer, splat (i1 true), i32 [[EVL:%.*]]) -; CHECK-NEXT: ret [[REV]] +; CHECK-NEXT: ret zeroinitializer ; %rev = tail call @llvm.experimental.vp.reverse( zeroinitializer, splat (i1 true), i32 %evl) ret %rev @@ -54,8 +49,7 @@ define @rev_of_splat(i32 %a, i32 %evl) { ; CHECK-LABEL: @rev_of_splat( ; CHECK-NEXT: [[A_INS:%.*]] = insertelement poison, i32 [[A:%.*]], i32 0 ; CHECK-NEXT: [[A_VEC:%.*]] = shufflevector [[A_INS]], poison, zeroinitializer -; CHECK-NEXT: [[REV:%.*]] = tail call @llvm.experimental.vp.reverse.nxv4i32( [[A_VEC]], splat (i1 true), i32 [[EVL:%.*]]) -; CHECK-NEXT: ret [[REV]] +; CHECK-NEXT: ret [[A_VEC]] ; %a.ins = insertelement poison, i32 %a, i32 0 %a.vec = shufflevector %a.ins, poison, zeroinitializer @@ -67,8 +61,7 @@ define @rev_of_splat2(i32 %a, %m, i32 %evl) ; CHECK-LABEL: @rev_of_splat2( ; CHECK-NEXT: [[A_INS:%.*]] = insertelement poison, i32 [[A:%.*]], i32 0 ; CHECK-NEXT: [[A_VEC:%.*]] = shufflevector [[A_INS]], poison, zeroinitializer -; CHECK-NEXT: [[REV:%.*]] = tail call @llvm.experimental.vp.reverse.nxv4i32( [[A_VEC]], [[M:%.*]], i32 [[EVL:%.*]]) -; CHECK-NEXT: ret [[REV]] +; CHECK-NEXT: ret [[A_VEC]] ; %a.ins = insertelement poison, i32 %a, i32 0 %a.vec = shufflevector %a.ins, poison, zeroinitializer