shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
llvm-project/llvm/test/Transforms/SCCP/uscmp.ll
Yingwei Zheng 9e587ce6f0
[SCCP] Simplify [us]cmp(X, Y) into X - Y (#144717) 
If the difference between [us]cmp's operands is not greater than 1, we
can simplify it into `X - Y`.
Alive2: https://alive2.llvm.org/ce/z/JS55so
llvm-opt-benchmark diff:
https://github.com/dtcxzyw/llvm-opt-benchmark/pull/2464/files
2025-07-20 15:01:44 +08:00

186 lines
6.0 KiB
LLVM
Raw Permalink Blame History

; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5
; RUN: opt -passes=sccp -S < %s | FileCheck %s
define i32 @scmp_to_sub(i32 range(i32 -1, 2) %a) {
; CHECK-LABEL: define i32 @scmp_to_sub(
; CHECK-SAME: i32 range(i32 -1, 2) [[A:%.*]]) {
; CHECK-NEXT: [[SCMP:%.*]] = sub nsw i32 [[A]], 0
; CHECK-NEXT: ret i32 [[SCMP]]
;
%scmp = call i32 @llvm.scmp(i32 %a, i32 0)
ret i32 %scmp
}
define i32 @scmp_zext_to_sub(i1 %a, i1 %b) {
; CHECK-LABEL: define i32 @scmp_zext_to_sub(
; CHECK-SAME: i1 [[A:%.*]], i1 [[B:%.*]]) {
; CHECK-NEXT: [[ZEXT_A:%.*]] = zext i1 [[A]] to i32
; CHECK-NEXT: [[ZEXT_B:%.*]] = zext i1 [[B]] to i32
; CHECK-NEXT: [[SCMP:%.*]] = sub nsw i32 [[ZEXT_A]], [[ZEXT_B]]
; CHECK-NEXT: ret i32 [[SCMP]]
;
%zext_a = zext i1 %a to i32
%zext_b = zext i1 %b to i32
%scmp = call i32 @llvm.scmp(i32 %zext_a, i32 %zext_b)
ret i32 %scmp
}
define i8 @scmp_to_sub_trunc(i32 range(i32 -1, 2) %a) {
; CHECK-LABEL: define i8 @scmp_to_sub_trunc(
; CHECK-SAME: i32 range(i32 -1, 2) [[A:%.*]]) {
; CHECK-NEXT: [[SCMP1:%.*]] = sub nsw i32 [[A]], 0
; CHECK-NEXT: [[SCMP:%.*]] = trunc i32 [[SCMP1]] to i8
; CHECK-NEXT: ret i8 [[SCMP]]
;
%scmp = call i8 @llvm.scmp(i32 %a, i32 0)
ret i8 %scmp
}
define i64 @scmp_to_sub_sext(i32 range(i32 -1, 2) %a) {
; CHECK-LABEL: define i64 @scmp_to_sub_sext(
; CHECK-SAME: i32 range(i32 -1, 2) [[A:%.*]]) {
; CHECK-NEXT: [[SCMP1:%.*]] = sub nsw i32 [[A]], 0
; CHECK-NEXT: [[SCMP:%.*]] = sext i32 [[SCMP1]] to i64
; CHECK-NEXT: ret i64 [[SCMP]]
;
%scmp = call i64 @llvm.scmp(i32 %a, i32 0)
ret i64 %scmp
}
define i32 @scmp_to_sub_small_range(i32 range(i32 -1, 1) %a) {
; CHECK-LABEL: define i32 @scmp_to_sub_small_range(
; CHECK-SAME: i32 range(i32 -1, 1) [[A:%.*]]) {
; CHECK-NEXT: [[SCMP:%.*]] = sub nsw i32 [[A]], 0
; CHECK-NEXT: ret i32 [[SCMP]]
;
%scmp = call i32 @llvm.scmp(i32 %a, i32 0)
ret i32 %scmp
}
define i32 @ucmp_to_sub(i32 range(i32 0, 3) %a) {
; CHECK-LABEL: define i32 @ucmp_to_sub(
; CHECK-SAME: i32 range(i32 0, 3) [[A:%.*]]) {
; CHECK-NEXT: [[SCMP:%.*]] = sub i32 [[A]], 1
; CHECK-NEXT: ret i32 [[SCMP]]
;
%ucmp = call i32 @llvm.ucmp(i32 %a, i32 1)
ret i32 %ucmp
}
define i8 @ucmp_to_sub_trunc(i32 range(i32 0, 3) %a) {
; CHECK-LABEL: define i8 @ucmp_to_sub_trunc(
; CHECK-SAME: i32 range(i32 0, 3) [[A:%.*]]) {
; CHECK-NEXT: [[UCMP1:%.*]] = sub i32 [[A]], 1
; CHECK-NEXT: [[UCMP:%.*]] = trunc i32 [[UCMP1]] to i8
; CHECK-NEXT: ret i8 [[UCMP]]
;
%ucmp = call i8 @llvm.ucmp(i32 %a, i32 1)
ret i8 %ucmp
}
define i64 @ucmp_to_sub_sext(i32 range(i32 0, 3) %a) {
; CHECK-LABEL: define i64 @ucmp_to_sub_sext(
; CHECK-SAME: i32 range(i32 0, 3) [[A:%.*]]) {
; CHECK-NEXT: [[UCMP1:%.*]] = sub i32 [[A]], 1
; CHECK-NEXT: [[UCMP:%.*]] = sext i32 [[UCMP1]] to i64
; CHECK-NEXT: ret i64 [[UCMP]]
;
%ucmp = call i64 @llvm.ucmp(i32 %a, i32 1)
ret i64 %ucmp
}
; TODO: we can fold this into %a.
define i32 @ucmp_to_sub_small_range(i32 range(i32 0, 2) %a) {
; CHECK-LABEL: define i32 @ucmp_to_sub_small_range(
; CHECK-SAME: i32 range(i32 0, 2) [[A:%.*]]) {
; CHECK-NEXT: [[UCMP:%.*]] = call i32 @llvm.ucmp.i32.i32(i32 [[A]], i32 0)
; CHECK-NEXT: ret i32 [[UCMP]]
;
%ucmp = call i32 @llvm.ucmp(i32 %a, i32 0)
ret i32 %ucmp
}
define i32 @scmp_to_sub_large_range(i32 range(i32 -1, 3) %a) {
; CHECK-LABEL: define i32 @scmp_to_sub_large_range(
; CHECK-SAME: i32 range(i32 -1, 3) [[A:%.*]]) {
; CHECK-NEXT: [[SCMP:%.*]] = call i32 @llvm.scmp.i32.i32(i32 [[A]], i32 0)
; CHECK-NEXT: ret i32 [[SCMP]]
;
%scmp = call i32 @llvm.scmp(i32 %a, i32 0)
ret i32 %scmp
}
define i32 @ucmp_to_sub_large_range(i32 range(i32 -1, 3) %a) {
; CHECK-LABEL: define i32 @ucmp_to_sub_large_range(
; CHECK-SAME: i32 range(i32 -1, 3) [[A:%.*]]) {
; CHECK-NEXT: [[SCMP:%.*]] = call i32 @llvm.ucmp.i32.i32(i32 [[A]], i32 0)
; CHECK-NEXT: ret i32 [[SCMP]]
;
%ucmp = call i32 @llvm.ucmp(i32 %a, i32 0)
ret i32 %ucmp
}
define i32 @scmp_to_sub_wrap(i8 range(i8 127, -126) %a) {
; CHECK-LABEL: define i32 @scmp_to_sub_wrap(
; CHECK-SAME: i8 range(i8 127, -126) [[A:%.*]]) {
; CHECK-NEXT: [[SCMP:%.*]] = call i32 @llvm.scmp.i32.i8(i8 [[A]], i8 -128)
; CHECK-NEXT: ret i32 [[SCMP]]
;
%scmp = call i32 @llvm.scmp(i8 %a, i8 -128)
ret i32 %scmp
}
define i32 @ucmp_to_sub_wrap(i8 range(i8 -1, 2) %a) {
; CHECK-LABEL: define i32 @ucmp_to_sub_wrap(
; CHECK-SAME: i8 range(i8 -1, 2) [[A:%.*]]) {
; CHECK-NEXT: [[UCMP:%.*]] = call i32 @llvm.ucmp.i32.i8(i8 [[A]], i8 0)
; CHECK-NEXT: ret i32 [[UCMP]]
;
%ucmp = call i32 @llvm.ucmp(i8 %a, i8 0)
ret i32 %ucmp
}
; It is incorrect to convert a ucmp into sub when the input type is i1.
define i32 @ucmp_to_sub_i1_rhs_const(i1 %a) {
; CHECK-LABEL: define i32 @ucmp_to_sub_i1_rhs_const(
; CHECK-SAME: i1 [[A:%.*]]) {
; CHECK-NEXT: [[SCMP:%.*]] = call i32 @llvm.ucmp.i32.i1(i1 [[A]], i1 false)
; CHECK-NEXT: ret i32 [[SCMP]]
;
%ucmp = call i32 @llvm.ucmp(i1 %a, i1 false)
ret i32 %ucmp
}
; It is incorrect to convert a ucmp into sub when the input type is i1.
define i32 @ucmp_to_sub_i1_lhs_const(i1 %a) {
; CHECK-LABEL: define i32 @ucmp_to_sub_i1_lhs_const(
; CHECK-SAME: i1 [[A:%.*]]) {
; CHECK-NEXT: [[SCMP:%.*]] = call i32 @llvm.ucmp.i32.i1(i1 false, i1 [[A]])
; CHECK-NEXT: ret i32 [[SCMP]]
;
%ucmp = call i32 @llvm.ucmp(i1 false, i1 %a)
ret i32 %ucmp
}
; It is incorrect to convert a ucmp into sub when the input type is i1.
define i32 @ucmp_to_sub_i1(i1 %a, i1 %b) {
; CHECK-LABEL: define i32 @ucmp_to_sub_i1(
; CHECK-SAME: i1 [[A:%.*]], i1 [[B:%.*]]) {
; CHECK-NEXT: [[SCMP:%.*]] = call i32 @llvm.ucmp.i32.i1(i1 [[A]], i1 [[B]])
; CHECK-NEXT: ret i32 [[SCMP]]
;
%ucmp = call i32 @llvm.ucmp(i1 %a, i1 %b)
ret i32 %ucmp
}
; It is incorrect to convert a scmp into sub when the input type is i1.
define i32 @scmp_to_sub_i1_rhs_const(i1 %a) {
; CHECK-LABEL: define i32 @scmp_to_sub_i1_rhs_const(
; CHECK-SAME: i1 [[A:%.*]]) {
; CHECK-NEXT: [[SCMP:%.*]] = call i32 @llvm.scmp.i32.i1(i1 [[A]], i1 false)
; CHECK-NEXT: ret i32 [[SCMP]]
;
%scmp = call i32 @llvm.scmp(i1 %a, i1 false)
ret i32 %scmp
}
Reference in New Issue View Git Blame Copy Permalink