168 lines
5.5 KiB
LLVM
168 lines
5.5 KiB
LLVM
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
|
|
; RUN: opt < %s -passes=instcombine -S | FileCheck %s
|
|
|
|
; (x << k) ? 2^k * x : 0 --> 2^k * x
|
|
|
|
define i32 @test_eq(i32 %x) {
|
|
; CHECK-LABEL: @test_eq(
|
|
; CHECK-NEXT: [[MUL:%.*]] = shl i32 [[X:%.*]], 2
|
|
; CHECK-NEXT: ret i32 [[MUL]]
|
|
;
|
|
%shl.mask = and i32 %x, 1073741823
|
|
%tobool = icmp eq i32 %shl.mask, 0
|
|
%mul = shl i32 %x, 2
|
|
%cond = select i1 %tobool, i32 0, i32 %mul
|
|
ret i32 %cond
|
|
}
|
|
|
|
define <2 x i32> @test_eq_vect(<2 x i32> %x) {
|
|
; CHECK-LABEL: @test_eq_vect(
|
|
; CHECK-NEXT: [[MUL:%.*]] = shl <2 x i32> [[X:%.*]], <i32 2, i32 2>
|
|
; CHECK-NEXT: ret <2 x i32> [[MUL]]
|
|
;
|
|
%shl.mask = and <2 x i32> %x, <i32 1073741823, i32 1073741823>
|
|
%tobool = icmp eq <2 x i32> %shl.mask, zeroinitializer
|
|
%mul = shl <2 x i32> %x, <i32 2, i32 2>
|
|
%cond = select <2 x i1> %tobool, <2 x i32> zeroinitializer, <2 x i32> %mul
|
|
ret <2 x i32> %cond
|
|
}
|
|
|
|
define i32 @test_ne(i32 %x) {
|
|
; CHECK-LABEL: @test_ne(
|
|
; CHECK-NEXT: [[MUL:%.*]] = shl i32 [[X:%.*]], 2
|
|
; CHECK-NEXT: ret i32 [[MUL]]
|
|
;
|
|
%shl.mask = and i32 %x, 1073741823
|
|
%tobool.not = icmp ne i32 %shl.mask, 0
|
|
%mul = shl i32 %x, 2
|
|
%cond = select i1 %tobool.not, i32 %mul, i32 0
|
|
ret i32 %cond
|
|
}
|
|
|
|
define <2 x i32> @test_ne_vect(<2 x i32> %x) {
|
|
; CHECK-LABEL: @test_ne_vect(
|
|
; CHECK-NEXT: [[MUL:%.*]] = shl <2 x i32> [[X:%.*]], <i32 2, i32 2>
|
|
; CHECK-NEXT: ret <2 x i32> [[MUL]]
|
|
;
|
|
%shl.mask = and <2 x i32> %x, <i32 1073741823, i32 1073741823>
|
|
%tobool.not = icmp ne <2 x i32> %shl.mask, zeroinitializer
|
|
%mul = shl <2 x i32> %x, <i32 2, i32 2>
|
|
%cond = select <2 x i1> %tobool.not, <2 x i32> %mul, <2 x i32> zeroinitializer
|
|
ret <2 x i32> %cond
|
|
}
|
|
|
|
define i32 @test_nuw_dropped(i32 %x) {
|
|
; CHECK-LABEL: @test_nuw_dropped(
|
|
; CHECK-NEXT: [[MUL:%.*]] = shl i32 [[X:%.*]], 2
|
|
; CHECK-NEXT: ret i32 [[MUL]]
|
|
;
|
|
%shl.mask = and i32 %x, 1073741823
|
|
%tobool = icmp eq i32 %shl.mask, 0
|
|
%mul = shl nuw i32 %x, 2
|
|
%cond = select i1 %tobool, i32 0, i32 %mul
|
|
ret i32 %cond
|
|
}
|
|
|
|
define i32 @test_nsw_dropped(i32 %x) {
|
|
; CHECK-LABEL: @test_nsw_dropped(
|
|
; CHECK-NEXT: [[MUL:%.*]] = shl i32 [[X:%.*]], 2
|
|
; CHECK-NEXT: ret i32 [[MUL]]
|
|
;
|
|
%shl.mask = and i32 %x, 1073741823
|
|
%tobool = icmp eq i32 %shl.mask, 0
|
|
%mul = shl nsw i32 %x, 2
|
|
%cond = select i1 %tobool, i32 0, i32 %mul
|
|
ret i32 %cond
|
|
}
|
|
|
|
declare void @use_multi(i32, i1, i32)
|
|
|
|
; This will not be canonicalized.
|
|
define i32 @test_multi_use(i32 %x) {
|
|
; CHECK-LABEL: @test_multi_use(
|
|
; CHECK-NEXT: [[SHL_MASK:%.*]] = and i32 [[X:%.*]], 1073741823
|
|
; CHECK-NEXT: [[TOBOOL_NOT:%.*]] = icmp ne i32 [[SHL_MASK]], 0
|
|
; CHECK-NEXT: [[MUL:%.*]] = shl i32 [[X]], 2
|
|
; CHECK-NEXT: call void @use_multi(i32 [[SHL_MASK]], i1 [[TOBOOL_NOT]], i32 [[MUL]])
|
|
; CHECK-NEXT: ret i32 [[MUL]]
|
|
;
|
|
%shl.mask = and i32 %x, 1073741823
|
|
%tobool.not = icmp ne i32 %shl.mask, 0
|
|
%mul = shl i32 %x, 2
|
|
%cond = select i1 %tobool.not, i32 %mul, i32 0
|
|
call void @use_multi(i32 %shl.mask, i1 %tobool.not, i32 %mul)
|
|
ret i32 %cond
|
|
}
|
|
|
|
define i32 @test_multi_use_nuw_dropped(i32 %x) {
|
|
; CHECK-LABEL: @test_multi_use_nuw_dropped(
|
|
; CHECK-NEXT: [[SHL_MASK:%.*]] = and i32 [[X:%.*]], 1073741823
|
|
; CHECK-NEXT: [[TOBOOL:%.*]] = icmp eq i32 [[SHL_MASK]], 0
|
|
; CHECK-NEXT: [[MUL:%.*]] = shl i32 [[X]], 2
|
|
; CHECK-NEXT: call void @use_multi(i32 [[SHL_MASK]], i1 [[TOBOOL]], i32 [[MUL]])
|
|
; CHECK-NEXT: ret i32 [[MUL]]
|
|
;
|
|
%shl.mask = and i32 %x, 1073741823
|
|
%tobool = icmp eq i32 %shl.mask, 0
|
|
%mul = shl nuw i32 %x, 2
|
|
%cond = select i1 %tobool, i32 0, i32 %mul
|
|
call void @use_multi(i32 %shl.mask, i1 %tobool, i32 %mul)
|
|
ret i32 %cond
|
|
}
|
|
|
|
define i32 @neg_test_bits_not_match(i32 %x) {
|
|
; CHECK-LABEL: @neg_test_bits_not_match(
|
|
; CHECK-NEXT: [[SHL_MASK:%.*]] = and i32 [[X:%.*]], 1073741823
|
|
; CHECK-NEXT: [[TOBOOL:%.*]] = icmp eq i32 [[SHL_MASK]], 0
|
|
; CHECK-NEXT: [[MUL:%.*]] = shl i32 [[X]], 3
|
|
; CHECK-NEXT: [[COND:%.*]] = select i1 [[TOBOOL]], i32 0, i32 [[MUL]]
|
|
; CHECK-NEXT: ret i32 [[COND]]
|
|
;
|
|
%shl.mask = and i32 %x, 1073741823
|
|
%tobool = icmp eq i32 %shl.mask, 0
|
|
%mul = shl i32 %x, 3
|
|
%cond = select i1 %tobool, i32 0, i32 %mul
|
|
ret i32 %cond
|
|
}
|
|
|
|
define i32 @neg_test_icmp_non_equality(i32 %x) {
|
|
; CHECK-LABEL: @neg_test_icmp_non_equality(
|
|
; CHECK-NEXT: [[MUL:%.*]] = shl i32 [[X:%.*]], 2
|
|
; CHECK-NEXT: ret i32 [[MUL]]
|
|
;
|
|
%shl.mask = and i32 %x, 1073741823
|
|
%tobool = icmp slt i32 %shl.mask, 0
|
|
%mul = shl i32 %x, 2
|
|
%cond = select i1 %tobool, i32 0, i32 %mul
|
|
ret i32 %cond
|
|
}
|
|
|
|
define i32 @neg_test_select_non_zero_constant(i32 %x) {
|
|
; CHECK-LABEL: @neg_test_select_non_zero_constant(
|
|
; CHECK-NEXT: [[SHL_MASK:%.*]] = and i32 [[X:%.*]], 1073741823
|
|
; CHECK-NEXT: [[TOBOOL:%.*]] = icmp eq i32 [[SHL_MASK]], 0
|
|
; CHECK-NEXT: [[MUL:%.*]] = shl i32 [[X]], 2
|
|
; CHECK-NEXT: [[COND:%.*]] = select i1 [[TOBOOL]], i32 1, i32 [[MUL]]
|
|
; CHECK-NEXT: ret i32 [[COND]]
|
|
;
|
|
%shl.mask = and i32 %x, 1073741823
|
|
%tobool = icmp eq i32 %shl.mask, 0
|
|
%mul = shl i32 %x, 2
|
|
%cond = select i1 %tobool, i32 1, i32 %mul
|
|
ret i32 %cond
|
|
}
|
|
|
|
define i32 @neg_test_icmp_non_zero_constant(i32 %x) {
|
|
; CHECK-LABEL: @neg_test_icmp_non_zero_constant(
|
|
; CHECK-NEXT: [[SHL_MASK:%.*]] = and i32 [[X:%.*]], 1073741823
|
|
; CHECK-NEXT: [[TOBOOL:%.*]] = icmp eq i32 [[SHL_MASK]], 1
|
|
; CHECK-NEXT: [[MUL:%.*]] = shl i32 [[X]], 2
|
|
; CHECK-NEXT: [[COND:%.*]] = select i1 [[TOBOOL]], i32 0, i32 [[MUL]]
|
|
; CHECK-NEXT: ret i32 [[COND]]
|
|
;
|
|
%shl.mask = and i32 %x, 1073741823
|
|
%tobool = icmp eq i32 %shl.mask, 1
|
|
%mul = shl i32 %x, 2
|
|
%cond = select i1 %tobool, i32 0, i32 %mul
|
|
ret i32 %cond
|
|
}
|