diff --git a/llvm/test/Transforms/InstCombine/ctpop-pow2.ll b/llvm/test/Transforms/InstCombine/ctpop-pow2.ll new file mode 100644 --- /dev/null +++ b/llvm/test/Transforms/InstCombine/ctpop-pow2.ll @@ -0,0 +1,127 @@ +; NOTE: Assertions have been autogenerated by utils/update_test_checks.py +; RUN: opt < %s -passes=instcombine -S | FileCheck %s + +declare i64 @llvm.ctpop.i64(i64) +declare i32 @llvm.ctpop.i32(i32) +declare i16 @llvm.ctpop.i16(i16) +declare i8 @llvm.ctpop.i8(i8) +declare void @llvm.assume(i1) + +define i64 @ctpop_1_shl(i64 %x) { +; CHECK-LABEL: @ctpop_1_shl( +; CHECK-NEXT: [[V:%.*]] = shl nuw i64 1, [[X:%.*]] +; CHECK-NEXT: [[CNT:%.*]] = call i64 @llvm.ctpop.i64(i64 [[V]]), !range [[RNG0:![0-9]+]] +; CHECK-NEXT: ret i64 [[CNT]] +; + %v = shl i64 1, %x + %cnt = call i64 @llvm.ctpop.i64(i64 %v) + ret i64 %cnt +} + +define i32 @ctpop_imin_lshr(i32 %x) { +; CHECK-LABEL: @ctpop_imin_lshr( +; CHECK-NEXT: [[V:%.*]] = lshr i32 -2147483648, [[X:%.*]] +; CHECK-NEXT: [[CNT:%.*]] = call i32 @llvm.ctpop.i32(i32 [[V]]), !range [[RNG1:![0-9]+]] +; CHECK-NEXT: ret i32 [[CNT]] +; + %v = lshr i32 2147483648, %x + %cnt = call i32 @llvm.ctpop.i32(i32 %v) + ret i32 %cnt +} + +define i16 @ctpop_x_and_negx(i16 %x) { +; CHECK-LABEL: @ctpop_x_and_negx( +; CHECK-NEXT: [[V0:%.*]] = sub i16 0, [[X:%.*]] +; CHECK-NEXT: [[V1:%.*]] = and i16 [[V0]], [[X]] +; CHECK-NEXT: [[CNT:%.*]] = call i16 @llvm.ctpop.i16(i16 [[V1]]), !range [[RNG2:![0-9]+]] +; CHECK-NEXT: ret i16 [[CNT]] +; + %v0 = sub i16 0, %x + %v1 = and i16 %x, %v0 + %cnt = call i16 @llvm.ctpop.i16(i16 %v1) + ret i16 %cnt +} + +define i8 @ctpop_x_nz_and_negx(i8 %x) { +; CHECK-LABEL: @ctpop_x_nz_and_negx( +; CHECK-NEXT: [[X1:%.*]] = or i8 [[X:%.*]], 1 +; CHECK-NEXT: [[V0:%.*]] = sub nsw i8 0, [[X1]] +; CHECK-NEXT: [[V1:%.*]] = and i8 [[X1]], [[V0]] +; CHECK-NEXT: [[CNT:%.*]] = call i8 @llvm.ctpop.i8(i8 [[V1]]), !range [[RNG3:![0-9]+]] +; CHECK-NEXT: ret i8 [[CNT]] +; + %x1 = or i8 %x, 1 + %v0 = sub i8 0, %x1 + %v1 = and i8 %x1, %v0 + %cnt = call i8 @llvm.ctpop.i8(i8 %v1) + ret i8 %cnt +} + +define i32 @ctpop_2_shl(i32 %x) { +; CHECK-LABEL: @ctpop_2_shl( +; CHECK-NEXT: [[V:%.*]] = shl i32 2, [[X:%.*]] +; CHECK-NEXT: [[CNT:%.*]] = call i32 @llvm.ctpop.i32(i32 [[V]]), !range [[RNG4:![0-9]+]] +; CHECK-NEXT: ret i32 [[CNT]] +; + %v = shl i32 2, %x + %cnt = call i32 @llvm.ctpop.i32(i32 %v) + ret i32 %cnt +} + +define i16 @ctpop_imin_plus1_lshr(i16 %x) { +; CHECK-LABEL: @ctpop_imin_plus1_lshr( +; CHECK-NEXT: [[V:%.*]] = lshr i16 -32767, [[X:%.*]] +; CHECK-NEXT: [[CNT:%.*]] = call i16 @llvm.ctpop.i16(i16 [[V]]), !range [[RNG2]] +; CHECK-NEXT: ret i16 [[CNT]] +; + %v = lshr i16 32769, %x + %cnt = call i16 @llvm.ctpop.i16(i16 %v) + ret i16 %cnt +} + + +define i32 @ctpop_2_shl_nz(i32 %x) { +; CHECK-LABEL: @ctpop_2_shl_nz( +; CHECK-NEXT: [[XA30:%.*]] = and i32 [[X:%.*]], 30 +; CHECK-NEXT: [[V:%.*]] = shl i32 2, [[XA30]] +; CHECK-NEXT: [[CNT:%.*]] = call i32 @llvm.ctpop.i32(i32 [[V]]), !range [[RNG5:![0-9]+]] +; CHECK-NEXT: ret i32 [[CNT]] +; + %xa30 = and i32 30, %x + %v = shl i32 2, %xa30 + %cnt = call i32 @llvm.ctpop.i32(i32 %v) + ret i32 %cnt +} + +define i8 @ctpop_imin_plus1_lshr_nz(i8 %x) { +; CHECK-LABEL: @ctpop_imin_plus1_lshr_nz( +; CHECK-NEXT: [[CMP:%.*]] = icmp ne i8 [[X:%.*]], 0 +; CHECK-NEXT: call void @llvm.assume(i1 [[CMP]]) +; CHECK-NEXT: [[V:%.*]] = lshr i8 -127, [[X]] +; CHECK-NEXT: [[CNT:%.*]] = call i8 @llvm.ctpop.i8(i8 [[V]]), !range [[RNG6:![0-9]+]] +; CHECK-NEXT: ret i8 [[CNT]] +; + %cmp = icmp ne i8 %x, 0 + call void @llvm.assume(i1 %cmp) + %v = lshr i8 129, %x + %cnt = call i8 @llvm.ctpop.i8(i8 %v) + ret i8 %cnt +} + + +define i64 @ctpop_x_and_negx_nz(i64 %x) { +; CHECK-LABEL: @ctpop_x_and_negx_nz( +; CHECK-NEXT: [[V0:%.*]] = sub i64 0, [[X:%.*]] +; CHECK-NEXT: [[V1:%.*]] = and i64 [[V0]], [[X]] +; CHECK-NEXT: [[CMP:%.*]] = icmp ne i64 [[V1]], 0 +; CHECK-NEXT: call void @llvm.assume(i1 [[CMP]]) +; CHECK-NEXT: [[CNT:%.*]] = call i64 @llvm.ctpop.i64(i64 [[V1]]), !range [[RNG0]] +; CHECK-NEXT: ret i64 [[CNT]] +; + %v0 = sub i64 0, %x + %v1 = and i64 %x, %v0 + %cmp = icmp ne i64 %v1, 0 + call void @llvm.assume(i1 %cmp) + %cnt = call i64 @llvm.ctpop.i64(i64 %v1) + ret i64 %cnt +}