diff --git a/llvm/test/Transforms/InstCombine/fsh.ll b/llvm/test/Transforms/InstCombine/fsh.ll index f0bdb6f79f7e..c88beb72ce05 100644 --- a/llvm/test/Transforms/InstCombine/fsh.ll +++ b/llvm/test/Transforms/InstCombine/fsh.ll @@ -559,3 +559,84 @@ define i16 @fshr_bswap(i16 %x) { ret i16 %r } +define i32 @fshl_mask_args_same1(i32 %a) { +; CHECK-LABEL: @fshl_mask_args_same1( +; CHECK-NEXT: [[TMP1:%.*]] = and i32 [[A:%.*]], -65536 +; CHECK-NEXT: [[TMP2:%.*]] = call i32 @llvm.fshl.i32(i32 [[TMP1]], i32 [[TMP1]], i32 16) +; CHECK-NEXT: ret i32 [[TMP2]] +; + %tmp1 = and i32 %a, 4294901760 ; 0xffff0000 + %tmp2 = call i32 @llvm.fshl.i32(i32 %tmp1, i32 %tmp1, i32 16) + ret i32 %tmp2 +} + +define i32 @fshl_mask_args_same2(i32 %a) { +; CHECK-LABEL: @fshl_mask_args_same2( +; CHECK-NEXT: [[TMP1:%.*]] = and i32 [[A:%.*]], 255 +; CHECK-NEXT: [[TMP2:%.*]] = call i32 @llvm.fshl.i32(i32 [[TMP1]], i32 [[TMP1]], i32 8) +; CHECK-NEXT: ret i32 [[TMP2]] +; + %tmp1 = and i32 %a, 255 + %tmp2 = call i32 @llvm.fshl.i32(i32 %tmp1, i32 %tmp1, i32 8) + ret i32 %tmp2 +} + +define i32 @fshl_mask_args_same3(i32 %a) { +; CHECK-LABEL: @fshl_mask_args_same3( +; CHECK-NEXT: [[TMP1:%.*]] = and i32 [[A:%.*]], 255 +; CHECK-NEXT: [[TMP2:%.*]] = call i32 @llvm.fshl.i32(i32 [[TMP1]], i32 [[TMP1]], i32 24) +; CHECK-NEXT: ret i32 [[TMP2]] +; + %tmp1 = and i32 %a, 255 + %tmp2 = call i32 @llvm.fshl.i32(i32 %tmp1, i32 %tmp1, i32 24) + ret i32 %tmp2 +} + +define i32 @fshl_mask_args_different(i32 %a) { +; CHECK-LABEL: @fshl_mask_args_different( +; CHECK-NEXT: [[TMP2:%.*]] = and i32 [[A:%.*]], -65536 +; CHECK-NEXT: [[TMP1:%.*]] = and i32 [[A]], -16777216 +; CHECK-NEXT: [[TMP3:%.*]] = call i32 @llvm.fshl.i32(i32 [[TMP2]], i32 [[TMP1]], i32 17) +; CHECK-NEXT: ret i32 [[TMP3]] +; + %tmp2 = and i32 %a, 4294901760 ; 0xfffff00f + %tmp1 = and i32 %a, 4278190080 ; 0xff00f00f + %tmp3 = call i32 @llvm.fshl.i32(i32 %tmp2, i32 %tmp1, i32 17) + ret i32 %tmp3 +} + +define <2 x i31> @fshr_mask_args_same_vector(<2 x i31> %a) { +; CHECK-LABEL: @fshr_mask_args_same_vector( +; CHECK-NEXT: [[TMP1:%.*]] = and <2 x i31> [[A:%.*]], +; CHECK-NEXT: [[TMP3:%.*]] = call <2 x i31> @llvm.fshl.v2i31(<2 x i31> [[A]], <2 x i31> [[TMP1]], <2 x i31> ) +; CHECK-NEXT: ret <2 x i31> [[TMP3]] +; + %tmp1 = and <2 x i31> %a, + %tmp2 = and <2 x i31> %a, + %tmp3 = call <2 x i31> @llvm.fshl.v2i31(<2 x i31> %tmp2, <2 x i31> %tmp1, <2 x i31> ) + ret <2 x i31> %tmp3 +} + +define <2 x i32> @fshr_mask_args_same_vector2(<2 x i32> %a, <2 x i32> %b) { +; CHECK-LABEL: @fshr_mask_args_same_vector2( +; CHECK-NEXT: [[TMP1:%.*]] = and <2 x i32> [[A:%.*]], +; CHECK-NEXT: [[TMP3:%.*]] = call <2 x i32> @llvm.fshl.v2i32(<2 x i32> [[TMP1]], <2 x i32> [[TMP1]], <2 x i32> ) +; CHECK-NEXT: ret <2 x i32> [[TMP3]] +; + %tmp1 = and <2 x i32> %a, + %tmp2 = and <2 x i32> %a, + %tmp3 = call <2 x i32> @llvm.fshr.v2i32(<2 x i32> %tmp1, <2 x i32> %tmp1, <2 x i32> ) + ret <2 x i32> %tmp3 +} + +define <2 x i31> @fshr_mask_args_same_vector3_different_but_still_prunable(<2 x i31> %a) { +; CHECK-LABEL: @fshr_mask_args_same_vector3_different_but_still_prunable( +; CHECK-NEXT: [[TMP1:%.*]] = and <2 x i31> [[A:%.*]], +; CHECK-NEXT: [[TMP3:%.*]] = call <2 x i31> @llvm.fshl.v2i31(<2 x i31> [[A]], <2 x i31> [[TMP1]], <2 x i31> ) +; CHECK-NEXT: ret <2 x i31> [[TMP3]] +; + %tmp1 = and <2 x i31> %a, + %tmp2 = and <2 x i31> %a, + %tmp3 = call <2 x i31> @llvm.fshl.v2i31(<2 x i31> %tmp2, <2 x i31> %tmp1, <2 x i31> ) + ret <2 x i31> %tmp3 +}