shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
llvm-project/llvm/test/Transforms/InstSimplify/select-icmp.ll
hanbeom a3c9b72651
[IR] Fix canReplacePointersIfEqual to properly validate vector pointers (#174142) 
Previously, `canReplacePointersIfEqual` unconditionally returned
`true` for vectors of pointers (e.g., `<2 x ptr>`) because it only
checked for scalar pointer types.

This resulted in a failure to perform appropriate verification for
these types. This patch fixes the logic to ensure they are properly
validated.

Fixes https://github.com/llvm/llvm-project/issues/174045
2026-01-08 09:30:11 +09:00

312 lines
11 KiB
LLVM
Executable File
Raw Blame History

; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
; RUN: opt < %s -passes=instsimplify -S | FileCheck %s
; TODO: https://alive2.llvm.org/ce/z/3ybZRl
define i32 @pr54735_slt(i32 %x, i32 %y) {
; CHECK-LABEL: @pr54735_slt(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[CMP:%.*]] = icmp slt i32 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: br i1 [[CMP]], label [[COND_TRUE:%.*]], label [[COND_END:%.*]]
; CHECK: cond.true:
; CHECK-NEXT: [[SUB:%.*]] = sub nsw i32 [[X]], [[Y]]
; CHECK-NEXT: [[ADD:%.*]] = add nsw i32 [[SUB]], 1
; CHECK-NEXT: [[NEG:%.*]] = xor i32 [[SUB]], -1
; CHECK-NEXT: [[ABSCOND:%.*]] = icmp sle i32 [[SUB]], -1
; CHECK-NEXT: [[ABS:%.*]] = select i1 [[ABSCOND]], i32 [[NEG]], i32 [[ADD]]
; CHECK-NEXT: ret i32 [[ABS]]
; CHECK: cond.end:
; CHECK-NEXT: ret i32 0
;
entry:
%cmp = icmp slt i32 %x, %y ; x<y ? abs (x-y+1): 0
br i1 %cmp, label %cond.true, label %cond.end
cond.true: ; preds = %entry
%sub = sub nsw i32 %x, %y
%add = add nsw i32 %sub, 1
%neg = xor i32 %sub, -1 ; sub nsw i32 0, %add
%abscond = icmp sle i32 %sub, -1
%abs = select i1 %abscond, i32 %neg, i32 %add
ret i32 %abs
cond.end: ; preds = %entry, %cond.true
ret i32 0
}
; https://alive2.llvm.org/ce/z/fTTsdT
define i32 @pr54735_sgt(i32 %x, i32 %y) {
; CHECK-LABEL: @pr54735_sgt(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[CMP:%.*]] = icmp sgt i32 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: br i1 [[CMP]], label [[COND_TRUE:%.*]], label [[COND_END:%.*]]
; CHECK: cond.true:
; CHECK-NEXT: [[SUB:%.*]] = sub nsw i32 [[X]], [[Y]]
; CHECK-NEXT: [[ADD:%.*]] = add nsw i32 [[SUB]], 1
; CHECK-NEXT: ret i32 [[ADD]]
; CHECK: cond.end:
; CHECK-NEXT: ret i32 0
;
entry:
%cmp = icmp sgt i32 %x, %y ; x>y ? abs (x-y+1): 0
br i1 %cmp, label %cond.true, label %cond.end
cond.true: ; preds = %entry
%sub = sub nsw i32 %x, %y
%add = add nsw i32 %sub, 1
%neg = xor i32 %sub, -1 ; sub nsw i32 0, %add
%abscond = icmp slt i32 %sub, -1
%abs = select i1 %abscond, i32 %neg, i32 %add
ret i32 %abs
cond.end: ; preds = %entry, %cond.true
ret i32 0
}
; https://alive2.llvm.org/ce/z/k9v75c
define i32 @pr54735_sge(i32 %x, i32 %y) {
; CHECK-LABEL: @pr54735_sge(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[CMP:%.*]] = icmp sge i32 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: br i1 [[CMP]], label [[COND_TRUE:%.*]], label [[COND_END:%.*]]
; CHECK: cond.true:
; CHECK-NEXT: [[SUB:%.*]] = sub nsw i32 [[X]], [[Y]]
; CHECK-NEXT: [[ADD:%.*]] = add nsw i32 [[SUB]], 1
; CHECK-NEXT: ret i32 [[ADD]]
; CHECK: cond.end:
; CHECK-NEXT: ret i32 0
;
entry:
%cmp = icmp sge i32 %x, %y ; x>y ? abs (x-y+1): 0
br i1 %cmp, label %cond.true, label %cond.end
cond.true: ; preds = %entry
%sub = sub nsw i32 %x, %y
%add = add nsw i32 %sub, 1
%neg = xor i32 %sub, -1 ; sub nsw i32 0, %add
%abscond = icmp slt i32 %sub, -1
%abs = select i1 %abscond, i32 %neg, i32 %add
ret i32 %abs
cond.end: ; preds = %entry, %cond.true
ret i32 0
}
; Negative test: https://alive2.llvm.org/ce/z/oZyu4M
define i8 @pr54735_without_nsw (i8 %x, i8 %y) {
; CHECK-LABEL: @pr54735_without_nsw(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[CMP:%.*]] = icmp sgt i8 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: br i1 [[CMP]], label [[COND_TRUE:%.*]], label [[COND_END:%.*]]
; CHECK: cond.true:
; CHECK-NEXT: [[SUB:%.*]] = sub i8 [[X]], [[Y]]
; CHECK-NEXT: [[ADD:%.*]] = add i8 [[SUB]], 1
; CHECK-NEXT: [[NEG:%.*]] = xor i8 [[SUB]], -1
; CHECK-NEXT: [[ABSCOND:%.*]] = icmp slt i8 [[SUB]], -1
; CHECK-NEXT: [[ABS:%.*]] = select i1 [[ABSCOND]], i8 [[NEG]], i8 [[ADD]]
; CHECK-NEXT: ret i8 [[ABS]]
; CHECK: cond.end:
; CHECK-NEXT: ret i8 0
;
entry:
%cmp = icmp sgt i8 %x, %y
br i1 %cmp, label %cond.true, label %cond.end
cond.true: ; preds = %entry
%sub = sub i8 %x, %y
%add = add i8 %sub, 1
%neg = xor i8 %sub, -1
%abscond = icmp slt i8 %sub, -1
%abs = select i1 %abscond, i8 %neg, i8 %add
ret i8 %abs
cond.end: ; preds = %entry, %cond.true
ret i8 0
}
define i32 @pr54735_sle(i32 %x, i32 %y) {
; CHECK-LABEL: @pr54735_sle(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[CMP:%.*]] = icmp sle i32 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: br i1 [[CMP]], label [[COND_TRUE:%.*]], label [[COND_END:%.*]]
; CHECK: cond.true:
; CHECK-NEXT: [[SUB:%.*]] = sub nsw i32 [[X]], [[Y]]
; CHECK-NEXT: [[ADD:%.*]] = add nsw i32 [[SUB]], 1
; CHECK-NEXT: [[NEG:%.*]] = xor i32 [[SUB]], -1
; CHECK-NEXT: [[ABSCOND:%.*]] = icmp slt i32 [[SUB]], -1
; CHECK-NEXT: [[ABS:%.*]] = select i1 [[ABSCOND]], i32 [[NEG]], i32 [[ADD]]
; CHECK-NEXT: ret i32 [[ABS]]
; CHECK: cond.end:
; CHECK-NEXT: ret i32 0
;
entry:
%cmp = icmp sle i32 %x, %y ; x<=y ? abs (x-y+1): 0
br i1 %cmp, label %cond.true, label %cond.end
cond.true: ; preds = %entry
%sub = sub nsw i32 %x, %y
%add = add nsw i32 %sub, 1
%neg = xor i32 %sub, -1 ; sub nsw i32 0, %add
%abscond = icmp slt i32 %sub, -1
%abs = select i1 %abscond, i32 %neg, i32 %add
ret i32 %abs
cond.end: ; preds = %entry, %cond.true
ret i32 0
}
; https://alive2.llvm.org/ce/z/pp9zJi
define i32 @pr54735_slt_neg(i32 %x, i32 %y) {
; CHECK-LABEL: @pr54735_slt_neg(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[CMP:%.*]] = icmp slt i32 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: br i1 [[CMP]], label [[COND_TRUE:%.*]], label [[COND_END:%.*]]
; CHECK: cond.true:
; CHECK-NEXT: [[SUB:%.*]] = sub nsw i32 [[X]], [[Y]]
; CHECK-NEXT: [[NEG:%.*]] = xor i32 [[SUB]], 12
; CHECK-NEXT: ret i32 [[NEG]]
; CHECK: cond.end:
; CHECK-NEXT: ret i32 0
;
entry:
%cmp = icmp slt i32 %x, %y ; x<y ? abs (x-y-12): 0
br i1 %cmp, label %cond.true, label %cond.end
cond.true: ; preds = %entry
%sub = sub nsw i32 %x, %y
%add = add nsw i32 %sub, -12 ; %sub - 12
%neg = xor i32 %sub, 12 ; 12 - %sub
%abscond = icmp sle i32 %sub, 12
%abs = select i1 %abscond, i32 %neg, i32 %add
ret i32 %abs
cond.end: ; preds = %entry, %cond.true
ret i32 0
}
; https://alive2.llvm.org/ce/z/9P6grR
define i32 @pr54735_sle_neg(i32 %x, i32 %y) {
; CHECK-LABEL: @pr54735_sle_neg(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[CMP:%.*]] = icmp sle i32 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: br i1 [[CMP]], label [[COND_TRUE:%.*]], label [[COND_END:%.*]]
; CHECK: cond.true:
; CHECK-NEXT: [[SUB:%.*]] = sub nsw i32 [[X]], [[Y]]
; CHECK-NEXT: [[NEG:%.*]] = xor i32 [[SUB]], 12
; CHECK-NEXT: ret i32 [[NEG]]
; CHECK: cond.end:
; CHECK-NEXT: ret i32 0
;
entry:
%cmp = icmp sle i32 %x, %y ; x<=y ? abs (x-y-12): 0
br i1 %cmp, label %cond.true, label %cond.end
cond.true: ; preds = %entry
%sub = sub nsw i32 %x, %y
%add = add nsw i32 %sub, -12
%neg = xor i32 %sub, 12 ; %sub - 12
%abscond = icmp sle i32 %sub, 12
%abs = select i1 %abscond, i32 %neg, i32 %add
ret i32 %abs
cond.end: ; preds = %entry, %cond.true
ret i32 0
}
; Negative test: https://alive2.llvm.org/ce/z/Yqv4x2
define i8 @pr54735_unexpect_const (i8 %x, i8 %y) {
; CHECK-LABEL: @pr54735_unexpect_const(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[CMP:%.*]] = icmp sgt i8 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: br i1 [[CMP]], label [[COND_TRUE:%.*]], label [[COND_END:%.*]]
; CHECK: cond.true:
; CHECK-NEXT: [[SUB:%.*]] = sub i8 [[X]], [[Y]]
; CHECK-NEXT: [[ADD:%.*]] = add i8 [[SUB]], 2
; CHECK-NEXT: [[NEG:%.*]] = xor i8 [[SUB]], -1
; CHECK-NEXT: [[ABSCOND:%.*]] = icmp slt i8 [[SUB]], -2
; CHECK-NEXT: [[ABS:%.*]] = select i1 [[ABSCOND]], i8 [[NEG]], i8 [[ADD]]
; CHECK-NEXT: ret i8 [[ABS]]
; CHECK: cond.end:
; CHECK-NEXT: ret i8 0
;
entry:
%cmp = icmp sgt i8 %x, %y ; x>y ? abs (x-y+2): 0
br i1 %cmp, label %cond.true, label %cond.end
cond.true: ; preds = %entry
%sub = sub i8 %x, %y
%add = add i8 %sub, 2 ; x-y+2
%neg = xor i8 %sub, -1 ; y-x-1
%neg1 = sub i8 %neg, 1 ; y-x-2
%abscond = icmp slt i8 %sub, -2
%abs = select i1 %abscond, i8 %neg, i8 %add
ret i8 %abs
cond.end: ; preds = %entry, %cond.true
ret i8 0
}
define ptr @icmp_ptr_eq_replace(ptr %a, ptr %b) {
; CHECK-LABEL: @icmp_ptr_eq_replace(
; CHECK-NEXT: [[CMP:%.*]] = icmp eq ptr [[A:%.*]], [[B1:%.*]]
; CHECK-NEXT: [[B:%.*]] = select i1 [[CMP]], ptr [[A]], ptr [[B1]]
; CHECK-NEXT: ret ptr [[B]]
;
%cmp = icmp eq ptr %a, %b
%sel = select i1 %cmp, ptr %a, ptr %b
ret ptr %sel
}
define <2 x ptr> @icmp_vector_ptr_eq_replace(<2 x ptr> %a, <2 x ptr> %b) {
; CHECK-LABEL: @icmp_vector_ptr_eq_replace(
; CHECK-NEXT: [[CMP:%.*]] = icmp eq <2 x ptr> [[A:%.*]], [[B1:%.*]]
; CHECK-NEXT: [[B:%.*]] = select <2 x i1> [[CMP]], <2 x ptr> [[A]], <2 x ptr> [[B1]]
; CHECK-NEXT: ret <2 x ptr> [[B]]
;
%cmp = icmp eq <2 x ptr> %a, %b
%sel = select <2 x i1> %cmp, <2 x ptr> %a, <2 x ptr> %b
ret <2 x ptr> %sel
}
define ptr @icmp_ptr_eq_replace_null(ptr %a) {
; CHECK-LABEL: @icmp_ptr_eq_replace_null(
; CHECK-NEXT: ret ptr [[A:%.*]]
;
%cmp = icmp eq ptr %a, null
%sel = select i1 %cmp, ptr null, ptr %a
ret ptr %sel
}
define ptr @ptr_eq_replace_same_underlying_object(ptr %st, i64 %i, i64 %j) {
; CHECK-LABEL: @ptr_eq_replace_same_underlying_object(
; CHECK-NEXT: [[B:%.*]] = getelementptr inbounds i8, ptr [[ST:%.*]], i64 [[J:%.*]]
; CHECK-NEXT: ret ptr [[B]]
;
%a = getelementptr inbounds i8, ptr %st, i64 %i
%b = getelementptr inbounds i8, ptr %st, i64 %j
%cmp = icmp eq ptr %a, %b
%sel = select i1 %cmp, ptr %a, ptr %b
ret ptr %sel
}
define ptr @ptr_eq_replace_constant(ptr %a) {
; CHECK-LABEL: @ptr_eq_replace_constant(
; CHECK-NEXT: [[CMP:%.*]] = icmp eq ptr [[A:%.*]], inttoptr (i64 42 to ptr)
; CHECK-NEXT: [[SEL:%.*]] = select i1 [[CMP]], ptr inttoptr (i64 42 to ptr), ptr [[A]]
; CHECK-NEXT: ret ptr [[SEL]]
;
%cmp = icmp eq ptr %a, inttoptr (i64 42 to ptr)
%sel = select i1 %cmp, ptr inttoptr (i64 42 to ptr), ptr %a
ret ptr %sel
}
define <2 x ptr> @ptr_eq_replace_vector_constant(<2 x ptr> %a) {
; CHECK-LABEL: @ptr_eq_replace_vector_constant(
; CHECK-NEXT: [[CMP:%.*]] = icmp eq <2 x ptr> [[A:%.*]], <ptr inttoptr (i64 42 to ptr), ptr inttoptr (i64 88 to ptr)>
; CHECK-NEXT: [[SEL:%.*]] = select <2 x i1> [[CMP]], <2 x ptr> [[A]], <2 x ptr> <ptr inttoptr (i64 42 to ptr), ptr inttoptr (i64 88 to ptr)>
; CHECK-NEXT: ret <2 x ptr> [[SEL]]
;
%cmp = icmp eq <2 x ptr> %a, <ptr inttoptr (i64 42 to ptr), ptr inttoptr (i64 88 to ptr)>
%sel = select <2 x i1> %cmp, <2 x ptr> %a, <2 x ptr> <ptr inttoptr (i64 42 to ptr), ptr inttoptr (i64 88 to ptr)>
ret <2 x ptr> %sel
}
Reference in New Issue View Git Blame Copy Permalink