Index: lib/Transforms/InstCombine/InstCombineShifts.cpp =================================================================== --- lib/Transforms/InstCombine/InstCombineShifts.cpp +++ lib/Transforms/InstCombine/InstCombineShifts.cpp @@ -802,6 +802,16 @@ return &I; } } + + // Transform (x << y) >> y to x & (-1 >> y) + Value *X, *Y; + if (match(&I, + m_LShr(m_OneUse(m_Shl(m_Value(X), m_Value(Y))), m_Deferred(Y)))) { + Constant *AllOnes = ConstantInt::getAllOnesValue(Ty); + Value *Mask = Builder.CreateLShr(AllOnes, Y); + return BinaryOperator::CreateAnd(Mask, X); + } + return nullptr; } Index: test/Transforms/InstCombine/canonicalize-shl-lshr-to-masking.ll =================================================================== --- test/Transforms/InstCombine/canonicalize-shl-lshr-to-masking.ll +++ test/Transforms/InstCombine/canonicalize-shl-lshr-to-masking.ll @@ -14,8 +14,8 @@ define i32 @positive_samevar(i32 %x, i32 %y) { ; CHECK-LABEL: @positive_samevar( -; CHECK-NEXT: [[TMP0:%.*]] = shl i32 [[X:%.*]], [[Y:%.*]] -; CHECK-NEXT: [[RET:%.*]] = lshr i32 [[TMP0]], [[Y]] +; CHECK-NEXT: [[TMP1:%.*]] = lshr i32 -1, [[Y:%.*]] +; CHECK-NEXT: [[RET:%.*]] = and i32 [[TMP1]], [[X:%.*]] ; CHECK-NEXT: ret i32 [[RET]] ; %tmp0 = shl i32 %x, %y @@ -113,8 +113,8 @@ define <2 x i32> @positive_samevar_vec(<2 x i32> %x, <2 x i32> %y) { ; CHECK-LABEL: @positive_samevar_vec( -; CHECK-NEXT: [[TMP0:%.*]] = shl <2 x i32> [[X:%.*]], [[Y:%.*]] -; CHECK-NEXT: [[RET:%.*]] = lshr <2 x i32> [[TMP0]], [[Y]] +; CHECK-NEXT: [[TMP1:%.*]] = lshr <2 x i32> , [[Y:%.*]] +; CHECK-NEXT: [[RET:%.*]] = and <2 x i32> [[TMP1]], [[X:%.*]] ; CHECK-NEXT: ret <2 x i32> [[RET]] ; %tmp0 = shl <2 x i32> %x, %y @@ -160,8 +160,7 @@ define <3 x i32> @positive_sameconst_vec_undef2(<3 x i32> %x) { ; CHECK-LABEL: @positive_sameconst_vec_undef2( -; CHECK-NEXT: [[TMP0:%.*]] = shl <3 x i32> [[X:%.*]], -; CHECK-NEXT: [[RET:%.*]] = lshr <3 x i32> [[TMP0]], +; CHECK-NEXT: [[RET:%.*]] = and <3 x i32> [[X:%.*]], ; CHECK-NEXT: ret <3 x i32> [[RET]] ; %tmp0 = shl <3 x i32> %x,