160 lines
4.8 KiB
LLVM
160 lines
4.8 KiB
LLVM
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
|
|
; RUN: opt < %s -passes=aggressive-instcombine -S | FileCheck %s
|
|
|
|
define i16 @shl_1(i8 %x) {
|
|
; CHECK-LABEL: @shl_1(
|
|
; CHECK-NEXT: [[ZEXT:%.*]] = zext i8 [[X:%.*]] to i16
|
|
; CHECK-NEXT: [[SHL:%.*]] = shl i16 [[ZEXT]], 1
|
|
; CHECK-NEXT: ret i16 [[SHL]]
|
|
;
|
|
%zext = zext i8 %x to i32
|
|
%shl = shl i32 %zext, 1
|
|
%trunc = trunc i32 %shl to i16
|
|
ret i16 %trunc
|
|
}
|
|
|
|
define i16 @shl_15(i8 %x) {
|
|
; CHECK-LABEL: @shl_15(
|
|
; CHECK-NEXT: [[ZEXT:%.*]] = zext i8 [[X:%.*]] to i16
|
|
; CHECK-NEXT: [[SHL:%.*]] = shl i16 [[ZEXT]], 15
|
|
; CHECK-NEXT: ret i16 [[SHL]]
|
|
;
|
|
%zext = zext i8 %x to i32
|
|
%shl = shl i32 %zext, 15
|
|
%trunc = trunc i32 %shl to i16
|
|
ret i16 %trunc
|
|
}
|
|
|
|
; Negative test - shift amount isn't less than target bitwidth
|
|
|
|
define i16 @shl_16(i8 %x) {
|
|
; CHECK-LABEL: @shl_16(
|
|
; CHECK-NEXT: [[ZEXT:%.*]] = zext i8 [[X:%.*]] to i32
|
|
; CHECK-NEXT: [[SHL:%.*]] = shl i32 [[ZEXT]], 16
|
|
; CHECK-NEXT: [[TRUNC:%.*]] = trunc i32 [[SHL]] to i16
|
|
; CHECK-NEXT: ret i16 [[TRUNC]]
|
|
;
|
|
%zext = zext i8 %x to i32
|
|
%shl = shl i32 %zext, 16
|
|
%trunc = trunc i32 %shl to i16
|
|
ret i16 %trunc
|
|
}
|
|
|
|
; Negative test -- variable shift amount
|
|
|
|
define i16 @shl_var_shift_amount(i8 %x, i8 %y) {
|
|
; CHECK-LABEL: @shl_var_shift_amount(
|
|
; CHECK-NEXT: [[ZEXT_X:%.*]] = zext i8 [[X:%.*]] to i32
|
|
; CHECK-NEXT: [[ZEXT_Y:%.*]] = zext i8 [[Y:%.*]] to i32
|
|
; CHECK-NEXT: [[SHL:%.*]] = shl i32 [[ZEXT_X]], [[ZEXT_Y]]
|
|
; CHECK-NEXT: [[TRUNC:%.*]] = trunc i32 [[SHL]] to i16
|
|
; CHECK-NEXT: ret i16 [[TRUNC]]
|
|
;
|
|
%zext.x = zext i8 %x to i32
|
|
%zext.y = zext i8 %y to i32
|
|
%shl = shl i32 %zext.x, %zext.y
|
|
%trunc = trunc i32 %shl to i16
|
|
ret i16 %trunc
|
|
}
|
|
|
|
define i16 @shl_var_bounded_shift_amount(i8 %x, i8 %y) {
|
|
; CHECK-LABEL: @shl_var_bounded_shift_amount(
|
|
; CHECK-NEXT: [[ZEXT_X:%.*]] = zext i8 [[X:%.*]] to i16
|
|
; CHECK-NEXT: [[ZEXT_Y:%.*]] = zext i8 [[Y:%.*]] to i16
|
|
; CHECK-NEXT: [[AND:%.*]] = and i16 [[ZEXT_Y]], 15
|
|
; CHECK-NEXT: [[SHL:%.*]] = shl i16 [[ZEXT_X]], [[AND]]
|
|
; CHECK-NEXT: ret i16 [[SHL]]
|
|
;
|
|
%zext.x = zext i8 %x to i32
|
|
%zext.y = zext i8 %y to i32
|
|
%and = and i32 %zext.y, 15
|
|
%shl = shl i32 %zext.x, %and
|
|
%trunc = trunc i32 %shl to i16
|
|
ret i16 %trunc
|
|
}
|
|
|
|
; Negative test (https://reviews.llvm.org/D108091#2950930)
|
|
|
|
define i32 @shl_check_no_overflow(i32 %x, i16 %amt) {
|
|
; CHECK-LABEL: @shl_check_no_overflow(
|
|
; CHECK-NEXT: [[ZEXT:%.*]] = zext i32 [[X:%.*]] to i64
|
|
; CHECK-NEXT: [[SEXT:%.*]] = sext i16 [[AMT:%.*]] to i64
|
|
; CHECK-NEXT: [[AND:%.*]] = and i64 [[SEXT]], 4294967295
|
|
; CHECK-NEXT: [[SHL:%.*]] = shl i64 [[ZEXT]], [[AND]]
|
|
; CHECK-NEXT: [[TRUNC:%.*]] = trunc i64 [[SHL]] to i32
|
|
; CHECK-NEXT: ret i32 [[TRUNC]]
|
|
;
|
|
%zext = zext i32 %x to i64
|
|
%sext = sext i16 %amt to i64
|
|
%and = and i64 %sext, 4294967295
|
|
%shl = shl i64 %zext, %and
|
|
%trunc = trunc i64 %shl to i32
|
|
ret i32 %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>
|
|
; CHECK-NEXT: [[S:%.*]] = shl <2 x i16> [[Z]], <i16 4, i16 10>
|
|
; CHECK-NEXT: ret <2 x i16> [[S]]
|
|
;
|
|
%z = zext <2 x i8> %x to <2 x i32>
|
|
%s = shl <2 x i32> %z, <i32 4, i32 10>
|
|
%t = trunc <2 x i32> %s to <2 x i16>
|
|
ret <2 x i16> %t
|
|
}
|
|
|
|
; Negative test - can only fold to <2 x i16>, requiring new vector type
|
|
|
|
define <2 x i8> @shl_vector_no_new_vector_type(<2 x i8> %x) {
|
|
; CHECK-LABEL: @shl_vector_no_new_vector_type(
|
|
; CHECK-NEXT: [[Z:%.*]] = zext <2 x i8> [[X:%.*]] to <2 x i32>
|
|
; CHECK-NEXT: [[S:%.*]] = shl <2 x i32> [[Z]], <i32 4, i32 10>
|
|
; CHECK-NEXT: [[T:%.*]] = trunc <2 x i32> [[S]] to <2 x i8>
|
|
; CHECK-NEXT: ret <2 x i8> [[T]]
|
|
;
|
|
%z = zext <2 x i8> %x to <2 x i32>
|
|
%s = shl <2 x i32> %z, <i32 4, i32 10>
|
|
%t = trunc <2 x i32> %s to <2 x i8>
|
|
ret <2 x i8> %t
|
|
}
|
|
|
|
; Negative test
|
|
|
|
define <2 x i16> @shl_vector_large_shift_amount(<2 x i8> %x) {
|
|
; CHECK-LABEL: @shl_vector_large_shift_amount(
|
|
; CHECK-NEXT: [[Z:%.*]] = zext <2 x i8> [[X:%.*]] to <2 x i32>
|
|
; CHECK-NEXT: [[S:%.*]] = shl <2 x i32> [[Z]], <i32 16, i32 5>
|
|
; CHECK-NEXT: [[T:%.*]] = trunc <2 x i32> [[S]] to <2 x i16>
|
|
; CHECK-NEXT: ret <2 x i16> [[T]]
|
|
;
|
|
%z = zext <2 x i8> %x to <2 x i32>
|
|
%s = shl <2 x i32> %z, <i32 16, i32 5>
|
|
%t = trunc <2 x i32> %s to <2 x i16>
|
|
ret <2 x i16> %t
|
|
}
|
|
|
|
define i16 @shl_nuw(i8 %x) {
|
|
; CHECK-LABEL: @shl_nuw(
|
|
; CHECK-NEXT: [[Z:%.*]] = zext i8 [[X:%.*]] to i16
|
|
; CHECK-NEXT: [[S:%.*]] = shl i16 [[Z]], 15
|
|
; CHECK-NEXT: ret i16 [[S]]
|
|
;
|
|
%z = zext i8 %x to i32
|
|
%s = shl nuw i32 %z, 15
|
|
%t = trunc i32 %s to i16
|
|
ret i16 %t
|
|
}
|
|
|
|
define i16 @shl_nsw(i8 %x) {
|
|
; CHECK-LABEL: @shl_nsw(
|
|
; CHECK-NEXT: [[Z:%.*]] = zext i8 [[X:%.*]] to i16
|
|
; CHECK-NEXT: [[S:%.*]] = shl i16 [[Z]], 15
|
|
; CHECK-NEXT: ret i16 [[S]]
|
|
;
|
|
%z = zext i8 %x to i32
|
|
%s = shl nsw i32 %z, 15
|
|
%t = trunc i32 %s to i16
|
|
ret i16 %t
|
|
}
|