9a844a36eb
Instead of only trying to constant fold the select arms, try to simplify them. This subsumes https://github.com/llvm/llvm-project/pull/115969 which implements this for extractvalue only. This is still fairly limited in that we will usually only call FoldOpIntoSelect in the first place if we have a constant operand. This can be relaxed in the future if worthwhile.
143 lines
5.8 KiB
LLVM
143 lines
5.8 KiB
LLVM
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 4
|
|
; RUN: opt < %s -passes=instcombine -S | FileCheck %s
|
|
|
|
define i64 @test_select_agg_constant_agg(i64 %val, i1 %cond) {
|
|
; CHECK-LABEL: define i64 @test_select_agg_constant_agg(
|
|
; CHECK-SAME: i64 [[VAL:%.*]], i1 [[COND:%.*]]) {
|
|
; CHECK-NEXT: entry:
|
|
; CHECK-NEXT: [[RET:%.*]] = zext i1 [[COND]] to i64
|
|
; CHECK-NEXT: ret i64 [[RET]]
|
|
;
|
|
entry:
|
|
%sel = select i1 %cond, { i64, i64 } {i64 1, i64 2}, { i64, i64 } {i64 0, i64 3}
|
|
%ret = extractvalue { i64, i64 } %sel, 0
|
|
ret i64 %ret
|
|
}
|
|
|
|
define void @test_select_agg_constant_agg_multiuse(i64 %val, i1 %cond) {
|
|
; CHECK-LABEL: define void @test_select_agg_constant_agg_multiuse(
|
|
; CHECK-SAME: i64 [[VAL:%.*]], i1 [[COND:%.*]]) {
|
|
; CHECK-NEXT: entry:
|
|
; CHECK-NEXT: [[RET:%.*]] = zext i1 [[COND]] to i64
|
|
; CHECK-NEXT: call void @use(i64 [[RET]])
|
|
; CHECK-NEXT: [[V1:%.*]] = select i1 [[COND]], i64 2, i64 3
|
|
; CHECK-NEXT: call void @use(i64 [[V1]])
|
|
; CHECK-NEXT: ret void
|
|
;
|
|
entry:
|
|
%sel = select i1 %cond, { i64, i64 } {i64 1, i64 2}, { i64, i64 } {i64 0, i64 3}
|
|
%v0 = extractvalue { i64, i64 } %sel, 0
|
|
call void @use(i64 %v0)
|
|
%v1 = extractvalue { i64, i64 } %sel, 1
|
|
call void @use(i64 %v1)
|
|
ret void
|
|
}
|
|
|
|
; TODO: it can be folded to zext i1 %cond to i64
|
|
define i64 @test_select_agg_constant(i64 %val, i1 %cond) {
|
|
; CHECK-LABEL: define i64 @test_select_agg_constant(
|
|
; CHECK-SAME: i64 [[VAL:%.*]], i1 [[COND:%.*]]) {
|
|
; CHECK-NEXT: entry:
|
|
; CHECK-NEXT: [[A:%.*]] = insertvalue { i64, i64 } { i64 1, i64 poison }, i64 [[VAL]], 1
|
|
; CHECK-NEXT: [[B:%.*]] = insertvalue { i64, i64 } { i64 0, i64 poison }, i64 [[VAL]], 1
|
|
; CHECK-NEXT: [[SEL:%.*]] = select i1 [[COND]], { i64, i64 } [[A]], { i64, i64 } [[B]]
|
|
; CHECK-NEXT: [[RET:%.*]] = extractvalue { i64, i64 } [[SEL]], 0
|
|
; CHECK-NEXT: ret i64 [[RET]]
|
|
;
|
|
entry:
|
|
%a = insertvalue { i64, i64 } { i64 1, i64 poison }, i64 %val, 1
|
|
%b = insertvalue { i64, i64 } { i64 0, i64 poison }, i64 %val, 1
|
|
%sel = select i1 %cond, { i64, i64 } %a, { i64, i64 } %b
|
|
%ret = extractvalue { i64, i64 } %sel, 0
|
|
ret i64 %ret
|
|
}
|
|
|
|
define void @test_select_agg_multiuse(i1 %cond, i64 %v1, i64 %v2, i64 %v3, i64 %v4) {
|
|
; CHECK-LABEL: define void @test_select_agg_multiuse(
|
|
; CHECK-SAME: i1 [[COND:%.*]], i64 [[V1:%.*]], i64 [[V2:%.*]], i64 [[V3:%.*]], i64 [[V4:%.*]]) {
|
|
; CHECK-NEXT: entry:
|
|
; CHECK-NEXT: [[X:%.*]] = select i1 [[COND]], i64 [[V1]], i64 [[V3]]
|
|
; CHECK-NEXT: call void @use(i64 [[X]])
|
|
; CHECK-NEXT: [[Y:%.*]] = select i1 [[COND]], i64 [[V2]], i64 [[V4]]
|
|
; CHECK-NEXT: call void @use(i64 [[Y]])
|
|
; CHECK-NEXT: ret void
|
|
;
|
|
entry:
|
|
%a0 = insertvalue { i64, i64 } poison, i64 %v1, 0
|
|
%a1 = insertvalue { i64, i64 } %a0, i64 %v2, 1
|
|
%b0 = insertvalue { i64, i64 } poison, i64 %v3, 0
|
|
%b1 = insertvalue { i64, i64 } %b0, i64 %v4, 1
|
|
%sel = select i1 %cond, { i64, i64 } %a1, { i64, i64 } %b1
|
|
%x = extractvalue { i64, i64 } %sel, 0
|
|
call void @use(i64 %x)
|
|
%y = extractvalue { i64, i64 } %sel, 1
|
|
call void @use(i64 %y)
|
|
ret void
|
|
}
|
|
|
|
declare void @use(i64)
|
|
|
|
define i64 @test_extract_select_insert_left(ptr %p1, i64 %v) {
|
|
; CHECK-LABEL: define i64 @test_extract_select_insert_left(
|
|
; CHECK-SAME: ptr [[P1:%.*]], i64 [[V:%.*]]) {
|
|
; CHECK-NEXT: [[CALL:%.*]] = call { ptr, i64 } @foo()
|
|
; CHECK-NEXT: [[ELM1:%.*]] = extractvalue { ptr, i64 } [[CALL]], 1
|
|
; CHECK-NEXT: [[CMP:%.*]] = icmp eq i64 [[ELM1]], [[V]]
|
|
; CHECK-NEXT: [[TMP1:%.*]] = extractvalue { ptr, i64 } [[CALL]], 1
|
|
; CHECK-NEXT: [[RES:%.*]] = select i1 [[CMP]], i64 4294967294, i64 [[TMP1]]
|
|
; CHECK-NEXT: ret i64 [[RES]]
|
|
;
|
|
%call = call { ptr, i64 } @foo()
|
|
%elm1 = extractvalue { ptr, i64 } %call, 1
|
|
%cmp = icmp eq i64 %elm1, %v
|
|
%fca0 = insertvalue { ptr, i64 } poison, ptr %p1, 0
|
|
%fca1 = insertvalue { ptr, i64 } %fca0, i64 4294967294, 1
|
|
%select = select i1 %cmp, { ptr, i64 } %fca1, { ptr, i64 } %call
|
|
%res = extractvalue { ptr, i64 } %select, 1
|
|
ret i64 %res
|
|
}
|
|
|
|
define i64 @test_extract_select_insert_right(ptr %p1, i64 %v) {
|
|
; CHECK-LABEL: define i64 @test_extract_select_insert_right(
|
|
; CHECK-SAME: ptr [[P1:%.*]], i64 [[V:%.*]]) {
|
|
; CHECK-NEXT: [[CALL:%.*]] = call { ptr, i64 } @foo()
|
|
; CHECK-NEXT: [[ELM1:%.*]] = extractvalue { ptr, i64 } [[CALL]], 1
|
|
; CHECK-NEXT: [[CMP:%.*]] = icmp eq i64 [[ELM1]], [[V]]
|
|
; CHECK-NEXT: [[TMP1:%.*]] = extractvalue { ptr, i64 } [[CALL]], 1
|
|
; CHECK-NEXT: [[RES:%.*]] = select i1 [[CMP]], i64 [[TMP1]], i64 4294967294
|
|
; CHECK-NEXT: ret i64 [[RES]]
|
|
;
|
|
%call = call { ptr, i64 } @foo()
|
|
%elm1 = extractvalue { ptr, i64 } %call, 1
|
|
%cmp = icmp eq i64 %elm1, %v
|
|
%fca0 = insertvalue { ptr, i64 } poison, ptr %p1, 0
|
|
%fca1 = insertvalue { ptr, i64 } %fca0, i64 4294967294, 1
|
|
%select = select i1 %cmp, { ptr, i64 } %call, { ptr, i64 } %fca1
|
|
%res = extractvalue { ptr, i64 } %select, 1
|
|
ret i64 %res
|
|
}
|
|
|
|
define i64 @test_extract_select_insert_negative(ptr %p1, i64 %v) {
|
|
; CHECK-LABEL: define i64 @test_extract_select_insert_negative(
|
|
; CHECK-SAME: ptr [[P1:%.*]], i64 [[V:%.*]]) {
|
|
; CHECK-NEXT: [[CALL:%.*]] = call { ptr, i64 } @foo()
|
|
; CHECK-NEXT: [[CALL2:%.*]] = call { ptr, i64 } @foo()
|
|
; CHECK-NEXT: [[ELM1:%.*]] = extractvalue { ptr, i64 } [[CALL]], 1
|
|
; CHECK-NEXT: [[CMP:%.*]] = icmp eq i64 [[ELM1]], [[V]]
|
|
; CHECK-NEXT: [[FCA1:%.*]] = insertvalue { ptr, i64 } [[CALL2]], ptr [[P1]], 0
|
|
; CHECK-NEXT: [[SELECT:%.*]] = select i1 [[CMP]], { ptr, i64 } [[CALL]], { ptr, i64 } [[FCA1]]
|
|
; CHECK-NEXT: [[RES:%.*]] = extractvalue { ptr, i64 } [[SELECT]], 1
|
|
; CHECK-NEXT: ret i64 [[RES]]
|
|
;
|
|
%call = call { ptr, i64 } @foo()
|
|
%call2 = call { ptr, i64 } @foo()
|
|
%elm1 = extractvalue { ptr, i64 } %call, 1
|
|
%cmp = icmp eq i64 %elm1, %v
|
|
%fca1 = insertvalue { ptr, i64 } %call2, ptr %p1, 0
|
|
%select = select i1 %cmp, { ptr, i64 } %call, { ptr, i64 } %fca1
|
|
%res = extractvalue { ptr, i64 } %select, 1
|
|
ret i64 %res
|
|
}
|
|
|
|
declare { ptr, i64 } @foo()
|