Skip to content

Commit

Permalink
[Test][AggressiveInstCombine] Add one more test for shift truncation
Browse files Browse the repository at this point in the history
Add test for which `OrigBitWidth != SrcBitWidth`
(https://reviews.llvm.org/D108091#2950131)
  • Loading branch information
anton-afanasyev committed Aug 18, 2021
1 parent 5aa0f9c commit 0988488
Showing 1 changed file with 30 additions and 0 deletions.
30 changes: 30 additions & 0 deletions llvm/test/Transforms/AggressiveInstCombine/trunc_shifts.ll
Expand Up @@ -92,6 +92,36 @@ define i32 @shl_check_no_overflow(i32 %call62, i16 %call63) {
ret i32 %conv68
}

define i16 @shl_smaller_bitwidth(i8 %x) {
; CHECK-LABEL: @shl_smaller_bitwidth(
; CHECK-NEXT: [[ZEXT:%.*]] = zext i8 [[X:%.*]] to i16
; CHECK-NEXT: [[SHL:%.*]] = shl i16 [[ZEXT]], 1
; CHECK-NEXT: [[AND:%.*]] = and i16 [[SHL]], 42
; CHECK-NEXT: ret i16 [[AND]]
;
%zext = zext i8 %x to i16
%shl = shl i16 %zext, 1
%zext2 = zext i16 %shl to i32
%and = and i32 %zext2, 42
%trunc = trunc i32 %and to i16
ret i16 %trunc
}

define i16 @shl_larger_bitwidth(i8 %x) {
; CHECK-LABEL: @shl_larger_bitwidth(
; CHECK-NEXT: [[ZEXT:%.*]] = zext i8 [[X:%.*]] to i16
; CHECK-NEXT: [[SHL:%.*]] = shl i16 [[ZEXT]], 1
; CHECK-NEXT: [[AND:%.*]] = and i16 [[SHL]], 42
; CHECK-NEXT: ret i16 [[AND]]
;
%zext = zext i8 %x to i64
%shl = shl i64 %zext, 1
%zext2 = trunc i64 %shl to i32
%and = and i32 %zext2, 42
%trunc = trunc i32 %and to i16
ret i16 %trunc
}

define <2 x i16> @shl_vector(<2 x i8> %x) {
; CHECK-LABEL: @shl_vector(
; CHECK-NEXT: [[Z:%.*]] = zext <2 x i8> [[X:%.*]] to <2 x i16>
Expand Down

0 comments on commit 0988488

Please sign in to comment.