151 lines · plain
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py2; RUN: opt < %s -passes=instcombine -S | FileCheck %s3 4declare i64 @llvm.ctpop.i64(i64)5declare i32 @llvm.ctpop.i32(i32)6declare i16 @llvm.ctpop.i16(i16)7declare i8 @llvm.ctpop.i8(i8)8declare <2 x i32> @llvm.ctpop.v2i32(<2 x i32>)9declare <2 x i64> @llvm.ctpop.v2i64(<2 x i64>)10declare void @llvm.assume(i1)11 12define i16 @ctpop_x_and_negx(i16 %x) {13; CHECK-LABEL: @ctpop_x_and_negx(14; CHECK-NEXT: [[V0:%.*]] = sub i16 0, [[X:%.*]]15; CHECK-NEXT: [[V1:%.*]] = and i16 [[X]], [[V0]]16; CHECK-NEXT: [[TMP1:%.*]] = icmp ne i16 [[V1]], 017; CHECK-NEXT: [[CNT:%.*]] = zext i1 [[TMP1]] to i1618; CHECK-NEXT: ret i16 [[CNT]]19;20 %v0 = sub i16 0, %x21 %v1 = and i16 %x, %v022 %cnt = call i16 @llvm.ctpop.i16(i16 %v1)23 ret i16 %cnt24}25 26define i8 @ctpop_x_nz_and_negx(i8 %x) {27; CHECK-LABEL: @ctpop_x_nz_and_negx(28; CHECK-NEXT: ret i8 129;30 %x1 = or i8 %x, 131 %v0 = sub i8 0, %x132 %v1 = and i8 %x1, %v033 %cnt = call i8 @llvm.ctpop.i8(i8 %v1)34 ret i8 %cnt35}36 37define i32 @ctpop_2_shl(i32 %x) {38; CHECK-LABEL: @ctpop_2_shl(39; CHECK-NEXT: [[TMP1:%.*]] = icmp ult i32 [[X:%.*]], 3140; CHECK-NEXT: [[CNT:%.*]] = zext i1 [[TMP1]] to i3241; CHECK-NEXT: ret i32 [[CNT]]42;43 %v = shl i32 2, %x44 %cnt = call i32 @llvm.ctpop.i32(i32 %v)45 ret i32 %cnt46}47 48define i32 @ctpop_2_shl_nz(i32 %x) {49; CHECK-LABEL: @ctpop_2_shl_nz(50; CHECK-NEXT: ret i32 151;52 %xa30 = and i32 30, %x53 %v = shl i32 2, %xa3054 %cnt = call i32 @llvm.ctpop.i32(i32 %v)55 ret i32 %cnt56}57 58define i8 @ctpop_imin_plus1_lshr_nz(i8 %x) {59; CHECK-LABEL: @ctpop_imin_plus1_lshr_nz(60; CHECK-NEXT: [[CMP:%.*]] = icmp ne i8 [[X:%.*]], 061; CHECK-NEXT: call void @llvm.assume(i1 [[CMP]])62; CHECK-NEXT: [[V:%.*]] = lshr i8 -127, [[X]]63; CHECK-NEXT: [[CNT:%.*]] = call range(i8 1, 9) i8 @llvm.ctpop.i8(i8 [[V]])64; CHECK-NEXT: ret i8 [[CNT]]65;66 %cmp = icmp ne i8 %x, 067 call void @llvm.assume(i1 %cmp)68 %v = lshr i8 129, %x69 %cnt = call i8 @llvm.ctpop.i8(i8 %v)70 ret i8 %cnt71}72 73 74define i64 @ctpop_x_and_negx_nz(i64 %x) {75; CHECK-LABEL: @ctpop_x_and_negx_nz(76; CHECK-NEXT: [[V0:%.*]] = sub i64 0, [[X:%.*]]77; CHECK-NEXT: [[V1:%.*]] = and i64 [[X]], [[V0]]78; CHECK-NEXT: [[CMP:%.*]] = icmp ne i64 [[V1]], 079; CHECK-NEXT: call void @llvm.assume(i1 [[CMP]])80; CHECK-NEXT: ret i64 181;82 %v0 = sub i64 0, %x83 %v1 = and i64 %x, %v084 %cmp = icmp ne i64 %v1, 085 call void @llvm.assume(i1 %cmp)86 %cnt = call i64 @llvm.ctpop.i64(i64 %v1)87 ret i64 %cnt88}89 90define <2 x i32> @ctpop_shl2_1_vec(<2 x i32> %x) {91; CHECK-LABEL: @ctpop_shl2_1_vec(92; CHECK-NEXT: [[SHL:%.*]] = shl <2 x i32> <i32 2, i32 1>, [[X:%.*]]93; CHECK-NEXT: [[TMP1:%.*]] = icmp ne <2 x i32> [[SHL]], zeroinitializer94; CHECK-NEXT: [[CNT:%.*]] = zext <2 x i1> [[TMP1]] to <2 x i32>95; CHECK-NEXT: ret <2 x i32> [[CNT]]96;97 %shl = shl <2 x i32> <i32 2 ,i32 1>, %x98 %cnt = call <2 x i32> @llvm.ctpop.v2i32(<2 x i32> %shl)99 ret <2 x i32> %cnt100}101 102 103define <2 x i32> @ctpop_lshr_intmin_intmin_plus1_vec_nz(<2 x i32> %x) {104; CHECK-LABEL: @ctpop_lshr_intmin_intmin_plus1_vec_nz(105; CHECK-NEXT: [[X1:%.*]] = or <2 x i32> [[X:%.*]], splat (i32 1)106; CHECK-NEXT: [[SHR:%.*]] = lshr <2 x i32> <i32 -2147483648, i32 -2147483647>, [[X1]]107; CHECK-NEXT: [[CNT:%.*]] = call range(i32 1, 17) <2 x i32> @llvm.ctpop.v2i32(<2 x i32> [[SHR]])108; CHECK-NEXT: ret <2 x i32> [[CNT]]109;110 %x1 = or <2 x i32> %x, <i32 1 ,i32 1>111 %shr = lshr <2 x i32> <i32 2147483648 ,i32 2147483649>, %x1112 %cnt = call <2 x i32> @llvm.ctpop.v2i32(<2 x i32> %shr)113 ret <2 x i32> %cnt114}115 116 117define <2 x i32> @ctpop_shl2_1_vec_nz(<2 x i32> %x) {118; CHECK-LABEL: @ctpop_shl2_1_vec_nz(119; CHECK-NEXT: ret <2 x i32> splat (i32 1)120;121 %and = and <2 x i32> %x, <i32 15 ,i32 15>122 %shl = shl <2 x i32> <i32 2 ,i32 1>, %and123 %cnt = call <2 x i32> @llvm.ctpop.v2i32(<2 x i32> %shl)124 ret <2 x i32> %cnt125}126 127define <2 x i64> @ctpop_x_and_negx_vec(<2 x i64> %x) {128; CHECK-LABEL: @ctpop_x_and_negx_vec(129; CHECK-NEXT: [[SUB:%.*]] = sub <2 x i64> zeroinitializer, [[X:%.*]]130; CHECK-NEXT: [[AND:%.*]] = and <2 x i64> [[X]], [[SUB]]131; CHECK-NEXT: [[TMP1:%.*]] = icmp ne <2 x i64> [[AND]], zeroinitializer132; CHECK-NEXT: [[CNT:%.*]] = zext <2 x i1> [[TMP1]] to <2 x i64>133; CHECK-NEXT: ret <2 x i64> [[CNT]]134;135 %sub = sub <2 x i64> <i64 0 ,i64 0>, %x136 %and = and <2 x i64> %sub, %x137 %cnt = call <2 x i64> @llvm.ctpop.v2i64(<2 x i64> %and)138 ret <2 x i64> %cnt139}140 141define <2 x i32> @ctpop_x_and_negx_vec_nz(<2 x i32> %x) {142; CHECK-LABEL: @ctpop_x_and_negx_vec_nz(143; CHECK-NEXT: ret <2 x i32> splat (i32 1)144;145 %x1 = or <2 x i32> %x, <i32 1 ,i32 1>146 %sub = sub <2 x i32> <i32 0 ,i32 0>, %x1147 %and = and <2 x i32> %sub, %x1148 %cnt = call <2 x i32> @llvm.ctpop.v2i32(<2 x i32> %and)149 ret <2 x i32> %cnt150}151