Skip to content

Commit

Permalink
[InstCombine] add tests for div with common factor; NFC
Browse files Browse the repository at this point in the history
  • Loading branch information
@rotateright
rotateright committed Oct 6, 2022
1 parent 08c4a67 commit 34cca2b
Showing 1 changed file with 254 additions and 8 deletions.
262 changes: 254 additions & 8 deletions llvm/test/Transforms/InstCombine/div-shift.ll
View file
Expand Up @@ -10,13 +10,11 @@ declare i8 @llvm.smax.i8(i8, i8)

define i32 @t1(i16 zeroext %x, i32 %y) {
; CHECK-LABEL: @t1(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[CONV:%.*]] = zext i16 [[X:%.*]] to i32
; CHECK-NEXT: [[TMP0:%.*]] = add i32 [[Y:%.*]], 1
; CHECK-NEXT: [[D1:%.*]] = lshr i32 [[CONV]], [[TMP0]]
; CHECK-NEXT: [[TMP1:%.*]] = add i32 [[Y:%.*]], 1
; CHECK-NEXT: [[D1:%.*]] = lshr i32 [[CONV]], [[TMP1]]
; CHECK-NEXT: ret i32 [[D1]]
;
entry:
%conv = zext i16 %x to i32
%s = shl i32 2, %y
%d = sdiv i32 %conv, %s
Expand All @@ -25,13 +23,11 @@ entry:

define <2 x i32> @t1vec(<2 x i16> %x, <2 x i32> %y) {
; CHECK-LABEL: @t1vec(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[CONV:%.*]] = zext <2 x i16> [[X:%.*]] to <2 x i32>
; CHECK-NEXT: [[TMP0:%.*]] = add <2 x i32> [[Y:%.*]], <i32 1, i32 1>
; CHECK-NEXT: [[D1:%.*]] = lshr <2 x i32> [[CONV]], [[TMP0]]
; CHECK-NEXT: [[TMP1:%.*]] = add <2 x i32> [[Y:%.*]], <i32 1, i32 1>
; CHECK-NEXT: [[D1:%.*]] = lshr <2 x i32> [[CONV]], [[TMP1]]
; CHECK-NEXT: ret <2 x i32> [[D1]]
;
entry:
%conv = zext <2 x i16> %x to <2 x i32>
%s = shl <2 x i32> <i32 2, i32 2>, %y
%d = sdiv <2 x i32> %conv, %s
Expand Down Expand Up @@ -298,3 +294,253 @@ define <2 x i32> @t16(<2 x i32> %x, <2 x i32> %y) {
%r = udiv <2 x i32> %shl, %x
ret <2 x i32> %r
}

define i5 @sdiv_mul_shl_nsw(i5 %x, i5 %y, i5 %z) {
; CHECK-LABEL: @sdiv_mul_shl_nsw(
; CHECK-NEXT: [[M1:%.*]] = mul nsw i5 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: [[M2:%.*]] = shl nsw i5 [[X]], [[Z:%.*]]
; CHECK-NEXT: [[D:%.*]] = sdiv i5 [[M1]], [[M2]]
; CHECK-NEXT: ret i5 [[D]]
;
%m1 = mul nsw i5 %x, %y
%m2 = shl nsw i5 %x, %z
%d = sdiv i5 %m1, %m2
ret i5 %d
}

define i5 @sdiv_mul_shl_nsw_commute1(i5 %x, i5 %y, i5 %z) {
; CHECK-LABEL: @sdiv_mul_shl_nsw_commute1(
; CHECK-NEXT: [[M1:%.*]] = mul nsw i5 [[Y:%.*]], [[X:%.*]]
; CHECK-NEXT: [[M2:%.*]] = shl nsw i5 [[X]], [[Z:%.*]]
; CHECK-NEXT: [[D:%.*]] = sdiv i5 [[M1]], [[M2]]
; CHECK-NEXT: ret i5 [[D]]
;
%m1 = mul nsw i5 %y, %x
%m2 = shl nsw i5 %x, %z
%d = sdiv i5 %m1, %m2
ret i5 %d
}

define i5 @sdiv_mul_shl_nsw_commute2(i5 %x, i5 %y, i5 %z) {
; CHECK-LABEL: @sdiv_mul_shl_nsw_commute2(
; CHECK-NEXT: [[M1:%.*]] = mul nsw i5 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: [[M2:%.*]] = shl nsw i5 [[Z:%.*]], [[X]]
; CHECK-NEXT: [[D:%.*]] = sdiv i5 [[M1]], [[M2]]
; CHECK-NEXT: ret i5 [[D]]
;
%m1 = mul nsw i5 %x, %y
%m2 = shl nsw i5 %z, %x
%d = sdiv i5 %m1, %m2
ret i5 %d
}

define i8 @sdiv_mul_shl_nsw_use1(i8 %x, i8 %y, i8 %z) {
; CHECK-LABEL: @sdiv_mul_shl_nsw_use1(
; CHECK-NEXT: [[M1:%.*]] = mul nsw i8 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: call void @use(i8 [[M1]])
; CHECK-NEXT: [[M2:%.*]] = shl nsw i8 [[X]], [[Z:%.*]]
; CHECK-NEXT: [[D:%.*]] = sdiv i8 [[M1]], [[M2]]
; CHECK-NEXT: ret i8 [[D]]
;
%m1 = mul nsw i8 %x, %y
call void @use(i8 %m1)
%m2 = shl nsw i8 %x, %z
%d = sdiv i8 %m1, %m2
ret i8 %d
}

define i8 @sdiv_mul_shl_nsw_use2(i8 %x, i8 %y, i8 %z) {
; CHECK-LABEL: @sdiv_mul_shl_nsw_use2(
; CHECK-NEXT: [[M1:%.*]] = mul nsw i8 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: [[M2:%.*]] = shl nsw i8 [[X]], [[Z:%.*]]
; CHECK-NEXT: call void @use(i8 [[M2]])
; CHECK-NEXT: [[D:%.*]] = sdiv i8 [[M1]], [[M2]]
; CHECK-NEXT: ret i8 [[D]]
;
%m1 = mul nsw i8 %x, %y
%m2 = shl nsw i8 %x, %z
call void @use(i8 %m2)
%d = sdiv i8 %m1, %m2
ret i8 %d
}

define i8 @sdiv_mul_shl_nsw_use3(i8 %x, i8 %y, i8 %z) {
; CHECK-LABEL: @sdiv_mul_shl_nsw_use3(
; CHECK-NEXT: [[M1:%.*]] = mul nsw i8 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: call void @use(i8 [[M1]])
; CHECK-NEXT: [[M2:%.*]] = shl nsw i8 [[X]], [[Z:%.*]]
; CHECK-NEXT: call void @use(i8 [[M2]])
; CHECK-NEXT: [[D:%.*]] = sdiv i8 [[M1]], [[M2]]
; CHECK-NEXT: ret i8 [[D]]
;
%m1 = mul nsw i8 %x, %y
call void @use(i8 %m1)
%m2 = shl nsw i8 %x, %z
call void @use(i8 %m2)
%d = sdiv i8 %m1, %m2
ret i8 %d
}

define i5 @sdiv_shl_mul_nsw(i5 %x, i5 %y, i5 %z) {
; CHECK-LABEL: @sdiv_shl_mul_nsw(
; CHECK-NEXT: [[M1:%.*]] = shl nsw i5 [[Z:%.*]], [[X:%.*]]
; CHECK-NEXT: [[M2:%.*]] = mul nsw i5 [[X]], [[Y:%.*]]
; CHECK-NEXT: [[D:%.*]] = sdiv i5 [[M1]], [[M2]]
; CHECK-NEXT: ret i5 [[D]]
;
%m1 = shl nsw i5 %z, %x
%m2 = mul nsw i5 %x, %y
%d = sdiv i5 %m1, %m2
ret i5 %d
}

define i5 @sdiv_mul_shl_missing_nsw1(i5 %x, i5 %y, i5 %z) {
; CHECK-LABEL: @sdiv_mul_shl_missing_nsw1(
; CHECK-NEXT: [[M1:%.*]] = mul nsw i5 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: [[M2:%.*]] = shl nuw i5 [[Y]], [[Z:%.*]]
; CHECK-NEXT: [[D:%.*]] = sdiv i5 [[M1]], [[M2]]
; CHECK-NEXT: ret i5 [[D]]
;
%m1 = mul nsw i5 %x, %y
%m2 = shl nuw i5 %y, %z
%d = sdiv i5 %m1, %m2
ret i5 %d
}

define i5 @sdiv_mul_shl_missing_nsw2(i5 %x, i5 %y, i5 %z) {
; CHECK-LABEL: @sdiv_mul_shl_missing_nsw2(
; CHECK-NEXT: [[M1:%.*]] = mul nuw i5 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: [[M2:%.*]] = shl nsw i5 [[Y]], [[Z:%.*]]
; CHECK-NEXT: [[D:%.*]] = sdiv i5 [[M1]], [[M2]]
; CHECK-NEXT: ret i5 [[D]]
;
%m1 = mul nuw i5 %x, %y
%m2 = shl nsw i5 %y, %z
%d = sdiv i5 %m1, %m2
ret i5 %d
}

define i5 @udiv_mul_shl_nuw(i5 %x, i5 %y, i5 %z) {
; CHECK-LABEL: @udiv_mul_shl_nuw(
; CHECK-NEXT: [[M1:%.*]] = mul nuw i5 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: [[M2:%.*]] = shl nuw i5 [[X]], [[Z:%.*]]
; CHECK-NEXT: [[D:%.*]] = udiv i5 [[M1]], [[M2]]
; CHECK-NEXT: ret i5 [[D]]
;
%m1 = mul nuw i5 %x, %y
%m2 = shl nuw i5 %x, %z
%d = udiv i5 %m1, %m2
ret i5 %d
}

define i5 @udiv_mul_shl_nuw_commute1(i5 %x, i5 %y, i5 %z) {
; CHECK-LABEL: @udiv_mul_shl_nuw_commute1(
; CHECK-NEXT: [[M1:%.*]] = mul nuw i5 [[Y:%.*]], [[X:%.*]]
; CHECK-NEXT: [[M2:%.*]] = shl nuw i5 [[X]], [[Z:%.*]]
; CHECK-NEXT: [[D:%.*]] = udiv i5 [[M1]], [[M2]]
; CHECK-NEXT: ret i5 [[D]]
;
%m1 = mul nuw i5 %y, %x
%m2 = shl nuw i5 %x, %z
%d = udiv i5 %m1, %m2
ret i5 %d
}

define i5 @udiv_mul_shl_nuw_commute2(i5 %x, i5 %y, i5 %z) {
; CHECK-LABEL: @udiv_mul_shl_nuw_commute2(
; CHECK-NEXT: [[M1:%.*]] = mul nuw i5 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: [[M2:%.*]] = shl nuw i5 [[Z:%.*]], [[X]]
; CHECK-NEXT: [[D:%.*]] = udiv i5 [[M1]], [[M2]]
; CHECK-NEXT: ret i5 [[D]]
;
%m1 = mul nuw i5 %x, %y
%m2 = shl nuw i5 %z, %x
%d = udiv i5 %m1, %m2
ret i5 %d
}

define i8 @udiv_mul_shl_nsw_use1(i8 %x, i8 %y, i8 %z) {
; CHECK-LABEL: @udiv_mul_shl_nsw_use1(
; CHECK-NEXT: [[M1:%.*]] = mul nuw i8 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: call void @use(i8 [[M1]])
; CHECK-NEXT: [[M2:%.*]] = shl nuw i8 [[X]], [[Z:%.*]]
; CHECK-NEXT: [[D:%.*]] = udiv i8 [[M1]], [[M2]]
; CHECK-NEXT: ret i8 [[D]]
;
%m1 = mul nuw i8 %x, %y
call void @use(i8 %m1)
%m2 = shl nuw i8 %x, %z
%d = udiv i8 %m1, %m2
ret i8 %d
}

define i8 @udiv_mul_shl_nsw_use2(i8 %x, i8 %y, i8 %z) {
; CHECK-LABEL: @udiv_mul_shl_nsw_use2(
; CHECK-NEXT: [[M1:%.*]] = mul nuw i8 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: [[M2:%.*]] = shl nuw i8 [[X]], [[Z:%.*]]
; CHECK-NEXT: call void @use(i8 [[M2]])
; CHECK-NEXT: [[D:%.*]] = udiv i8 [[M1]], [[M2]]
; CHECK-NEXT: ret i8 [[D]]
;
%m1 = mul nuw i8 %x, %y
%m2 = shl nuw i8 %x, %z
call void @use(i8 %m2)
%d = udiv i8 %m1, %m2
ret i8 %d
}

define i8 @udiv_mul_shl_nsw_use3(i8 %x, i8 %y, i8 %z) {
; CHECK-LABEL: @udiv_mul_shl_nsw_use3(
; CHECK-NEXT: [[M1:%.*]] = mul nuw i8 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: call void @use(i8 [[M1]])
; CHECK-NEXT: [[M2:%.*]] = shl nuw i8 [[X]], [[Z:%.*]]
; CHECK-NEXT: call void @use(i8 [[M2]])
; CHECK-NEXT: [[D:%.*]] = udiv i8 [[M1]], [[M2]]
; CHECK-NEXT: ret i8 [[D]]
;
%m1 = mul nuw i8 %x, %y
call void @use(i8 %m1)
%m2 = shl nuw i8 %x, %z
call void @use(i8 %m2)
%d = udiv i8 %m1, %m2
ret i8 %d
}

define i5 @udiv_shl_mul_nuw(i5 %x, i5 %y, i5 %z) {
; CHECK-LABEL: @udiv_shl_mul_nuw(
; CHECK-NEXT: [[M1:%.*]] = shl nuw i5 [[Z:%.*]], [[X:%.*]]
; CHECK-NEXT: [[M2:%.*]] = mul nuw i5 [[X]], [[Y:%.*]]
; CHECK-NEXT: [[D:%.*]] = udiv i5 [[M1]], [[M2]]
; CHECK-NEXT: ret i5 [[D]]
;
%m1 = shl nuw i5 %z, %x
%m2 = mul nuw i5 %x, %y
%d = udiv i5 %m1, %m2
ret i5 %d
}

define i5 @udiv_mul_shl_missing_nsw1(i5 %x, i5 %y, i5 %z) {
; CHECK-LABEL: @udiv_mul_shl_missing_nsw1(
; CHECK-NEXT: [[M1:%.*]] = mul nsw i5 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: [[M2:%.*]] = shl nuw i5 [[Y]], [[Z:%.*]]
; CHECK-NEXT: [[D:%.*]] = udiv i5 [[M1]], [[M2]]
; CHECK-NEXT: ret i5 [[D]]
;
%m1 = mul nsw i5 %x, %y
%m2 = shl nuw i5 %y, %z
%d = udiv i5 %m1, %m2
ret i5 %d
}

define i5 @udiv_mul_shl_missing_nsw2(i5 %x, i5 %y, i5 %z) {
; CHECK-LABEL: @udiv_mul_shl_missing_nsw2(
; CHECK-NEXT: [[M1:%.*]] = mul nuw i5 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: [[M2:%.*]] = shl nsw i5 [[Y]], [[Z:%.*]]
; CHECK-NEXT: [[D:%.*]] = udiv i5 [[M1]], [[M2]]
; CHECK-NEXT: ret i5 [[D]]
;
%m1 = mul nuw i5 %x, %y
%m2 = shl nsw i5 %y, %z
%d = udiv i5 %m1, %m2
ret i5 %d
}

0 comments on commit 34cca2b

Please sign in to comment.