Panagiotis Karouzakis c2e7fad446
[DemandedBits] Support non-constant shift amounts (#148880)
This patch adds support for the shift operators to handle non-constant
shift operands.

ashr proof -->https://alive2.llvm.org/ce/z/EN-siK
lshr proof --> https://alive2.llvm.org/ce/z/eeGzyB
shl proof --> https://alive2.llvm.org/ce/z/dpvbkq
2025-08-19 01:11:16 +08:00

199 lines
8.3 KiB
LLVM

; RUN: opt -S -disable-output -passes="print<demanded-bits>" < %s 2>&1 | FileCheck %s
define i8 @test_lshr_const_amount_4(i32 %a) {
; CHECK-LABEL: 'test_lshr_const_amount_4'
; CHECK-DAG: DemandedBits: 0xff for %lshr.t = trunc i32 %lshr to i8
; CHECK-DAG: DemandedBits: 0xff for %lshr in %lshr.t = trunc i32 %lshr to i8
; CHECK-DAG: DemandedBits: 0xff for %lshr = lshr i32 %a, 4
; CHECK-DAG: DemandedBits: 0xff0 for %a in %lshr = lshr i32 %a, 4
; CHECK-DAG: DemandedBits: 0xffffffff for 4 in %lshr = lshr i32 %a, 4
;
%lshr = lshr i32 %a, 4
%lshr.t = trunc i32 %lshr to i8
ret i8 %lshr.t
}
define i8 @test_lshr_const_amount_5(i32 %a) {
; CHECK-LABEL: 'test_lshr_const_amount_5'
; CHECK-DAG: DemandedBits: 0xff for %lshr = lshr i32 %a, 5
; CHECK-DAG: DemandedBits: 0x1fe0 for %a in %lshr = lshr i32 %a, 5
; CHECK-DAG: DemandedBits: 0xffffffff for 5 in %lshr = lshr i32 %a, 5
; CHECK-DAG: DemandedBits: 0xff for %lshr.t = trunc i32 %lshr to i8
; CHECK-DAG: DemandedBits: 0xff for %lshr in %lshr.t = trunc i32 %lshr to i8
;
%lshr = lshr i32 %a, 5
%lshr.t = trunc i32 %lshr to i8
ret i8 %lshr.t
}
define i8 @test_lshr_const_amount_8(i32 %a) {
; CHECK-LABEL: 'test_lshr_const_amount_8'
; CHECK-DAG: DemandedBits: 0xff for %lshr.t = trunc i32 %lshr to i8
; CHECK-DAG: DemandedBits: 0xff for %lshr in %lshr.t = trunc i32 %lshr to i8
; CHECK-DAG: DemandedBits: 0xff for %lshr = lshr i32 %a, 8
; CHECK-DAG: DemandedBits: 0xff00 for %a in %lshr = lshr i32 %a, 8
; CHECK-DAG: DemandedBits: 0xffffffff for 8 in %lshr = lshr i32 %a, 8
;
%lshr = lshr i32 %a, 8
%lshr.t = trunc i32 %lshr to i8
ret i8 %lshr.t
}
define i8 @test_lshr_const_amount_9(i32 %a) {
; CHECK-LABEL: 'test_lshr_const_amount_9'
; CHECK-DAG: DemandedBits: 0xff for %lshr.t = trunc i32 %lshr to i8
; CHECK-DAG: DemandedBits: 0xff for %lshr in %lshr.t = trunc i32 %lshr to i8
; CHECK-DAG: DemandedBits: 0xff for %lshr = lshr i32 %a, 9
; CHECK-DAG: DemandedBits: 0x1fe00 for %a in %lshr = lshr i32 %a, 9
; CHECK-DAG: DemandedBits: 0xffffffff for 9 in %lshr = lshr i32 %a, 9
;
%lshr = lshr i32 %a, 9
%lshr.t = trunc i32 %lshr to i8
ret i8 %lshr.t
}
define i8 @test_lshr(i32 %a, i32 %b) {
; CHECK-LABEL: 'test_lshr'
; CHECK-DAG: DemandedBits: 0xff for %lshr = lshr i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffffffff for %a in %lshr = lshr i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffffffff for %b in %lshr = lshr i32 %a, %b
; CHECK-DAG: DemandedBits: 0xff for %lshr.t = trunc i32 %lshr to i8
; CHECK-DAG: DemandedBits: 0xff for %lshr in %lshr.t = trunc i32 %lshr to i8
;
%lshr = lshr i32 %a, %b
%lshr.t = trunc i32 %lshr to i8
ret i8 %lshr.t
}
define i8 @test_lshr_range_1(i32 %a, i32 %b) {
; CHECK-LABEL: 'test_lshr_range_1'
; CHECK-DAG: DemandedBits: 0xff for %shl.t = trunc i32 %lshr to i8
; CHECK-DAG: DemandedBits: 0xff for %lshr in %shl.t = trunc i32 %lshr to i8
; CHECK-DAG: DemandedBits: 0xff for %lshr = lshr i32 %a, %b2
; CHECK-DAG: DemandedBits: 0x7ff for %a in %lshr = lshr i32 %a, %b2
; CHECK-DAG: DemandedBits: 0xffffffff for %b2 in %lshr = lshr i32 %a, %b2
; CHECK-DAG: DemandedBits: 0xffffffff for %b2 = and i32 %b, 3
; CHECK-DAG: DemandedBits: 0x3 for %b in %b2 = and i32 %b, 3
; CHECK-DAG: DemandedBits: 0xffffffff for 3 in %b2 = and i32 %b, 3
;
%b2 = and i32 %b, 3
%lshr = lshr i32 %a, %b2
%shl.t = trunc i32 %lshr to i8
ret i8 %shl.t
}
define i32 @test_lshr_range_2(i32 %a, i32 %b) {
; CHECK-LABEL: 'test_lshr_range_2'
; CHECK-DAG: DemandedBits: 0xffffffff for %lshr = lshr i32 %a, %b2
; CHECK-DAG: DemandedBits: 0xffffffff for %a in %lshr = lshr i32 %a, %b2
; CHECK-DAG: DemandedBits: 0xffffffff for %b2 in %lshr = lshr i32 %a, %b2
; CHECK-DAG: DemandedBits: 0xffffffff for %b2 = and i32 %b, 3
; CHECK-DAG: DemandedBits: 0x3 for %b in %b2 = and i32 %b, 3
; CHECK-DAG: DemandedBits: 0xffffffff for 3 in %b2 = and i32 %b, 3
;
%b2 = and i32 %b, 3
%lshr = lshr i32 %a, %b2
ret i32 %lshr
}
define i32 @test_lshr_range_3(i32 %a, i32 %b) {
; CHECK-LABEL: 'test_lshr_range_3'
; CHECK-DAG: DemandedBits: 0xffff for %lshr = lshr i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffffffff for %a in %lshr = lshr i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffffffff for %b in %lshr = lshr i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffffffff for %shl = shl i32 %lshr, 16
; CHECK-DAG: DemandedBits: 0xffff for %lshr in %shl = shl i32 %lshr, 16
; CHECK-DAG: DemandedBits: 0xffffffff for 16 in %shl = shl i32 %lshr, 16
;
%lshr = lshr i32 %a, %b
%shl = shl i32 %lshr, 16
ret i32 %shl
}
define i32 @test_lshr_range_4(i32 %a, i32 %b) {
; CHECK-LABEL: 'test_lshr_range_4'
; CHECK-DAG: DemandedBits: 0xffffff00 for %lshr = lshr i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffffff00 for %a in %lshr = lshr i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffffffff for %b in %lshr = lshr i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffffffff for %shr = ashr i32 %lshr, 8
; CHECK-DAG: DemandedBits: 0xffffff00 for %lshr in %shr = ashr i32 %lshr, 8
; CHECK-DAG: DemandedBits: 0xffffffff for 8 in %shr = ashr i32 %lshr, 8
%lshr = lshr i32 %a, %b
%shr = ashr i32 %lshr, 8
ret i32 %shr
}
define i32 @test_lshr_range_5(i32 %a, i32 %b) {
; CHECK-LABEL: 'test_lshr_range_5'
; CHECK-DAG: DemandedBits: 0xff for %1 = lshr i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffffffff for %a in %1 = lshr i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffffffff for %b in %1 = lshr i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffffffff for %2 = and i32 %1, 255
; CHECK-DAG: DemandedBits: 0xff for %1 in %2 = and i32 %1, 255
; CHECK-DAG: DemandedBits: 0xffffffff for 255 in %2 = and i32 %1, 255
;
%1 = lshr i32 %a, %b
%2 = and i32 %1, 255
ret i32 %2
}
define i32 @test_lshr_range_6(i32 %a, i32 %b) {
; CHECK-LABEL: 'test_lshr_range_6'
; CHECK-DAG: DemandedBits: 0xffff0000 for %lshr.1 = lshr i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffff0000 for %a in %lshr.1 = lshr i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffffffff for %b in %lshr.1 = lshr i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffffffff for %lshr.2 = lshr i32 %lshr.1, 16
; CHECK-DAG: DemandedBits: 0xffff0000 for %lshr.1 in %lshr.2 = lshr i32 %lshr.1, 16
; CHECK-DAG: DemandedBits: 0xffffffff for 16 in %lshr.2 = lshr i32 %lshr.1, 16
;
%lshr.1 = lshr i32 %a, %b
%lshr.2 = lshr i32 %lshr.1, 16
ret i32 %lshr.2
}
define i8 @test_lshr_var_amount(i32 %a, i32 %b){
; CHECK-LABEL: 'test_lshr_var_amount'
; CHECK-DAG: DemandedBits: 0xff for %4 = lshr i32 %1, %3
; CHECK-DAG: DemandedBits: 0xffffffff for %1 in %4 = lshr i32 %1, %3
; CHECK-DAG: DemandedBits: 0xffffffff for %3 in %4 = lshr i32 %1, %3
; CHECK-DAG: DemandedBits: 0xff for %5 = trunc i32 %4 to i8
; CHECK-DAG: DemandedBits: 0xff for %4 in %5 = trunc i32 %4 to i8
; CHECK-DAG: DemandedBits: 0xffffffff for %3 = zext i8 %2 to i32
; CHECK-DAG: DemandedBits: 0xff for %2 in %3 = zext i8 %2 to i32
; CHECK-DAG: DemandedBits: 0xffffffff for %1 = add nsw i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffffffff for %a in %1 = add nsw i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffffffff for %b in %1 = add nsw i32 %a, %b
; CHECK-DAG: DemandedBits: 0xff for %2 = trunc i32 %1 to i8
; CHECK-DAG: DemandedBits: 0xff for %1 in %2 = trunc i32 %1 to i8
;
%1 = add nsw i32 %a, %b
%2 = trunc i32 %1 to i8
%3 = zext i8 %2 to i32
%4 = lshr i32 %1, %3
%5 = trunc i32 %4 to i8
ret i8 %5
}
define i8 @test_lshr_var_amount_exact(i32 %a, i32 %b){
; CHECK-LABEL 'test_lshr_var_amount_nsw'
; CHECK-DAG: DemandedBits: 0xffffffff for %1 = add nsw i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffffffff for %a in %1 = add nsw i32 %a, %b
; CHECK-DAG: DemandedBits: 0xffffffff for %b in %1 = add nsw i32 %a, %b
; CHECK-DAG: DemandedBits: 0xff for %2 = trunc i32 %1 to i8
; CHECK-DAG: DemandedBits: 0xff for %1 in %2 = trunc i32 %1 to i8
; CHECK-DAG: DemandedBits: 0xffffffff for %3 = zext i8 %2 to i32
; CHECK-DAG: DemandedBits: 0xff for %2 in %3 = zext i8 %2 to i32
; CHECK-DAG: DemandedBits: 0xff for %4 = lshr exact i32 %1, %3
; CHECK-DAG: DemandedBits: 0xffffffff for %1 in %4 = lshr exact i32 %1, %3
; CHECK-DAG: DemandedBits: 0xffffffff for %3 in %4 = lshr exact i32 %1, %3
; CHECK-DAG: DemandedBits: 0xff for %5 = trunc i32 %4 to i8
; CHECK-DAG: DemandedBits: 0xff for %4 in %5 = trunc i32 %4 to i8
;
%1 = add nsw i32 %a, %b
%2 = trunc i32 %1 to i8
%3 = zext i8 %2 to i32
%4 = lshr exact i32 %1, %3
%5 = trunc i32 %4 to i8
ret i8 %5
}