diff --git a/llvm/test/Transforms/InstCombine/div-shift.ll b/llvm/test/Transforms/InstCombine/div-shift.ll index 5982e73185c4b0..1251bd99619a2f 100644 --- a/llvm/test/Transforms/InstCombine/div-shift.ll +++ b/llvm/test/Transforms/InstCombine/div-shift.ll @@ -618,3 +618,115 @@ define i8 @udiv_shl_nuw_use(i8 %x, i8 %y, i8 %z) { %d = udiv i8 %x, %s ret i8 %d } + +define i8 @udiv_lshr_mul_nuw(i8 %x, i8 %y, i8 %z) { +; CHECK-LABEL: @udiv_lshr_mul_nuw( +; CHECK-NEXT: [[M:%.*]] = mul nuw i8 [[X:%.*]], [[Y:%.*]] +; CHECK-NEXT: [[S:%.*]] = lshr i8 [[M]], [[Z:%.*]] +; CHECK-NEXT: [[DIV:%.*]] = udiv i8 [[S]], [[X]] +; CHECK-NEXT: ret i8 [[DIV]] +; + %m = mul nuw i8 %x, %y + %s = lshr i8 %m, %z + %div = udiv i8 %s, %x + ret i8 %div +} + +define <2 x i4> @udiv_lshr_mul_nuw_commute1(<2 x i4> %x, <2 x i4> %y, <2 x i4> %z) { +; CHECK-LABEL: @udiv_lshr_mul_nuw_commute1( +; CHECK-NEXT: [[M:%.*]] = mul nuw <2 x i4> [[Y:%.*]], [[X:%.*]] +; CHECK-NEXT: [[S:%.*]] = lshr <2 x i4> [[M]], [[Z:%.*]] +; CHECK-NEXT: [[DIV:%.*]] = udiv <2 x i4> [[S]], [[X]] +; CHECK-NEXT: ret <2 x i4> [[DIV]] +; + %m = mul nuw <2 x i4> %y, %x + %s = lshr <2 x i4> %m, %z + %div = udiv <2 x i4> %s, %x + ret <2 x i4> %div +} + +define i8 @udiv_lshr_mul_nuw_commute2(i8 %x, i8 %y, i8 %z) { +; CHECK-LABEL: @udiv_lshr_mul_nuw_commute2( +; CHECK-NEXT: [[M:%.*]] = mul nuw i8 [[Y:%.*]], [[X:%.*]] +; CHECK-NEXT: [[S:%.*]] = lshr i8 [[Z:%.*]], [[M]] +; CHECK-NEXT: [[DIV:%.*]] = udiv i8 [[S]], [[X]] +; CHECK-NEXT: ret i8 [[DIV]] +; + %m = mul nuw i8 %y, %x + %s = lshr i8 %z, %m + %div = udiv i8 %s, %x + ret i8 %div +} + +define i8 @udiv_lshr_mul_nuw_use1(i8 %x, i8 %y, i8 %z) { +; CHECK-LABEL: @udiv_lshr_mul_nuw_use1( +; CHECK-NEXT: [[M:%.*]] = mul nuw i8 [[X:%.*]], [[Y:%.*]] +; CHECK-NEXT: call void @use(i8 [[M]]) +; CHECK-NEXT: [[S:%.*]] = lshr i8 [[M]], [[Z:%.*]] +; CHECK-NEXT: [[DIV:%.*]] = udiv i8 [[S]], [[X]] +; CHECK-NEXT: ret i8 [[DIV]] +; + %m = mul nuw i8 %x, %y + call void @use(i8 %m) + %s = lshr i8 %m, %z + %div = udiv i8 %s, %x + ret i8 %div +} + +define i8 @udiv_lshr_mul_nuw_use2(i8 %x, i8 %y, i8 %z) { +; CHECK-LABEL: @udiv_lshr_mul_nuw_use2( +; CHECK-NEXT: [[M:%.*]] = mul nuw i8 [[X:%.*]], [[Y:%.*]] +; CHECK-NEXT: [[S:%.*]] = lshr i8 [[M]], [[Z:%.*]] +; CHECK-NEXT: call void @use(i8 [[S]]) +; CHECK-NEXT: [[DIV:%.*]] = udiv i8 [[S]], [[X]] +; CHECK-NEXT: ret i8 [[DIV]] +; + %m = mul nuw i8 %x, %y + %s = lshr i8 %m, %z + call void @use(i8 %s) + %div = udiv i8 %s, %x + ret i8 %div +} + +define i8 @udiv_lshr_mul_nuw_use3(i8 %x, i8 %y, i8 %z) { +; CHECK-LABEL: @udiv_lshr_mul_nuw_use3( +; CHECK-NEXT: [[M:%.*]] = mul nuw i8 [[X:%.*]], [[Y:%.*]] +; CHECK-NEXT: call void @use(i8 [[M]]) +; CHECK-NEXT: [[S:%.*]] = lshr i8 [[M]], [[Z:%.*]] +; CHECK-NEXT: call void @use(i8 [[S]]) +; CHECK-NEXT: [[DIV:%.*]] = udiv i8 [[S]], [[X]] +; CHECK-NEXT: ret i8 [[DIV]] +; + %m = mul nuw i8 %x, %y + call void @use(i8 %m) + %s = lshr i8 %m, %z + call void @use(i8 %s) + %div = udiv i8 %s, %x + ret i8 %div +} + +define i8 @udiv_lshr_mul_nsw(i8 %x, i8 %y, i8 %z) { +; CHECK-LABEL: @udiv_lshr_mul_nsw( +; CHECK-NEXT: [[M:%.*]] = mul nsw i8 [[X:%.*]], [[Y:%.*]] +; CHECK-NEXT: [[S:%.*]] = lshr i8 [[M]], [[Z:%.*]] +; CHECK-NEXT: [[DIV:%.*]] = udiv i8 [[S]], [[X]] +; CHECK-NEXT: ret i8 [[DIV]] +; + %m = mul nsw i8 %x, %y + %s = lshr i8 %m, %z + %div = udiv i8 %s, %x + ret i8 %div +} + +define i8 @sdiv_lshr_mul_nsw(i8 %x, i8 %y, i8 %z) { +; CHECK-LABEL: @sdiv_lshr_mul_nsw( +; CHECK-NEXT: [[M:%.*]] = mul nsw i8 [[X:%.*]], [[Y:%.*]] +; CHECK-NEXT: [[S:%.*]] = lshr i8 [[M]], [[Z:%.*]] +; CHECK-NEXT: [[DIV:%.*]] = sdiv i8 [[S]], [[X]] +; CHECK-NEXT: ret i8 [[DIV]] +; + %m = mul nsw i8 %x, %y + %s = lshr i8 %m, %z + %div = sdiv i8 %s, %x + ret i8 %div +}