shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
llvm-project/llvm/test/Transforms/SimplifyCFG/switch-umin.ll
Congcong Cai 31e5f86a3c
[SimplifyCFG] process prof data when remove case in umin (#182261) 
In #164097, we introduce a optimization for umin. But it does not handle
profile data correctly.
This PR remove profile data when remove cases.
Fixed: #181837
2026-02-20 07:03:27 +08:00

290 lines
7.0 KiB
LLVM
Raw Permalink Blame History

; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 6
; RUN: opt -S -passes=simplifycfg < %s | FileCheck %s
declare void @a()
declare void @b()
declare void @c()
declare void @d()
define void @switch_replace_default(i32 %x) {
; CHECK-LABEL: define void @switch_replace_default(
; CHECK-SAME: i32 [[X:%.*]]) {
; CHECK-NEXT: [[MIN:%.*]] = call i32 @llvm.umin.i32(i32 [[X]], i32 3)
; CHECK-NEXT: switch i32 [[X]], label %[[COMMON_RET:.*]] [
; CHECK-NEXT: i32 0, label %[[CASE0:.*]]
; CHECK-NEXT: i32 1, label %[[CASE1:.*]]
; CHECK-NEXT: i32 2, label %[[CASE2:.*]]
; CHECK-NEXT: ], !prof [[PROF0:![0-9]+]]
; CHECK: [[COMMON_RET]]:
; CHECK-NEXT: ret void
; CHECK: [[CASE0]]:
; CHECK-NEXT: call void @a()
; CHECK-NEXT: br label %[[COMMON_RET]]
; CHECK: [[CASE1]]:
; CHECK-NEXT: call void @b()
; CHECK-NEXT: br label %[[COMMON_RET]]
; CHECK: [[CASE2]]:
; CHECK-NEXT: call void @c()
; CHECK-NEXT: br label %[[COMMON_RET]]
;
%min = call i32 @llvm.umin.i32(i32 %x, i32 3)
switch i32 %min, label %unreachable [
i32 0, label %case0
i32 1, label %case1
i32 2, label %case2
i32 3, label %case3
], !prof !0
case0:
call void @a()
ret void
case1:
call void @b()
ret void
case2:
call void @c()
ret void
case3:
ret void
unreachable:
unreachable
}
define void @switch_replace_default_and_remove_dead_cases(i32 %x) {
; CHECK-LABEL: define void @switch_replace_default_and_remove_dead_cases(
; CHECK-SAME: i32 [[X:%.*]]) {
; CHECK-NEXT: [[MIN:%.*]] = call i32 @llvm.umin.i32(i32 [[X]], i32 3)
; CHECK-NEXT: switch i32 [[X]], label %[[COMMON_RET:.*]] [
; CHECK-NEXT: i32 2, label %[[CASE2:.*]]
; CHECK-NEXT: i32 1, label %[[CASE1:.*]]
; CHECK-NEXT: ]
; CHECK: [[COMMON_RET]]:
; CHECK-NEXT: ret void
; CHECK: [[CASE1]]:
; CHECK-NEXT: call void @b()
; CHECK-NEXT: br label %[[COMMON_RET]]
; CHECK: [[CASE2]]:
; CHECK-NEXT: call void @c()
; CHECK-NEXT: br label %[[COMMON_RET]]
;
%min = call i32 @llvm.umin.i32(i32 %x, i32 3)
switch i32 %min, label %unreachable [
i32 4, label %case4
i32 1, label %case1
i32 2, label %case2
i32 3, label %case3
]
case4:
call void @a()
ret void
case1:
call void @b()
ret void
case2:
call void @c()
ret void
case3:
ret void
unreachable:
unreachable
}
define void @switch_replace_default_when_holes(i32 %x) {
; CHECK-LABEL: define void @switch_replace_default_when_holes(
; CHECK-SAME: i32 [[X:%.*]]) {
; CHECK-NEXT: [[MIN:%.*]] = call i32 @llvm.umin.i32(i32 [[X]], i32 3)
; CHECK-NEXT: switch i32 [[X]], label %[[COMMON_RET:.*]] [
; CHECK-NEXT: i32 1, label %[[CASE1:.*]]
; CHECK-NEXT: i32 2, label %[[CASE2:.*]]
; CHECK-NEXT: ]
; CHECK: [[COMMON_RET]]:
; CHECK-NEXT: ret void
; CHECK: [[CASE1]]:
; CHECK-NEXT: call void @b()
; CHECK-NEXT: br label %[[COMMON_RET]]
; CHECK: [[CASE2]]:
; CHECK-NEXT: call void @c()
; CHECK-NEXT: br label %[[COMMON_RET]]
;
%min = call i32 @llvm.umin.i32(i32 %x, i32 3)
switch i32 %min, label %unreachable [
i32 1, label %case1
i32 2, label %case2
i32 3, label %case3
]
case1:
call void @b()
ret void
case2:
call void @c()
ret void
case3:
ret void
unreachable:
unreachable
}
define void @do_not_switch_replace_default(i32 %x, i32 %y) {
; CHECK-LABEL: define void @do_not_switch_replace_default(
; CHECK-SAME: i32 [[X:%.*]], i32 [[Y:%.*]]) {
; CHECK-NEXT: [[MIN:%.*]] = call i32 @llvm.umin.i32(i32 [[X]], i32 [[Y]])
; CHECK-NEXT: switch i32 [[MIN]], label %[[UNREACHABLE:.*]] [
; CHECK-NEXT: i32 0, label %[[CASE0:.*]]
; CHECK-NEXT: i32 1, label %[[CASE1:.*]]
; CHECK-NEXT: i32 2, label %[[CASE2:.*]]
; CHECK-NEXT: i32 3, label %[[COMMON_RET:.*]]
; CHECK-NEXT: ]
; CHECK: [[COMMON_RET]]:
; CHECK-NEXT: ret void
; CHECK: [[CASE0]]:
; CHECK-NEXT: call void @a()
; CHECK-NEXT: br label %[[COMMON_RET]]
; CHECK: [[CASE1]]:
; CHECK-NEXT: call void @b()
; CHECK-NEXT: br label %[[COMMON_RET]]
; CHECK: [[CASE2]]:
; CHECK-NEXT: call void @c()
; CHECK-NEXT: br label %[[COMMON_RET]]
; CHECK: [[UNREACHABLE]]:
; CHECK-NEXT: unreachable
;
%min = call i32 @llvm.umin.i32(i32 %x, i32 %y)
switch i32 %min, label %unreachable [
i32 0, label %case0
i32 1, label %case1
i32 2, label %case2
i32 3, label %case3
]
case0:
call void @a()
ret void
case1:
call void @b()
ret void
case2:
call void @c()
ret void
case3:
ret void
unreachable:
unreachable
}
define void @do_not_replace_switch_default_but_remove_dead_cases(i32 %x) {
; CHECK-LABEL: define void @do_not_replace_switch_default_but_remove_dead_cases(
; CHECK-SAME: i32 [[X:%.*]]) {
; CHECK-NEXT: [[MIN:%.*]] = call i32 @llvm.umin.i32(i32 [[X]], i32 3)
; CHECK-NEXT: switch i32 [[MIN]], label %[[CASE0:.*]] [
; CHECK-NEXT: i32 3, label %[[COMMON_RET:.*]]
; CHECK-NEXT: i32 1, label %[[CASE1:.*]]
; CHECK-NEXT: i32 2, label %[[CASE2:.*]]
; CHECK-NEXT: ]
; CHECK: [[COMMON_RET]]:
; CHECK-NEXT: ret void
; CHECK: [[CASE0]]:
; CHECK-NEXT: call void @a()
; CHECK-NEXT: br label %[[COMMON_RET]]
; CHECK: [[CASE1]]:
; CHECK-NEXT: call void @b()
; CHECK-NEXT: br label %[[COMMON_RET]]
; CHECK: [[CASE2]]:
; CHECK-NEXT: call void @c()
; CHECK-NEXT: br label %[[COMMON_RET]]
;
%min = call i32 @llvm.umin.i32(i32 %x, i32 3)
switch i32 %min, label %case0 [ ; default is reachable, therefore simplification not triggered
i32 0, label %case0
i32 1, label %case1
i32 2, label %case2
i32 3, label %case3
i32 4, label %case4
]
case0:
call void @a()
ret void
case1:
call void @b()
ret void
case2:
call void @c()
ret void
case3:
ret void
case4:
call void @d()
ret void
}
define void @switch_remove_dead_cases(i32 %x) {
; CHECK-LABEL: define void @switch_remove_dead_cases(
; CHECK-SAME: i32 [[X:%.*]]) {
; CHECK-NEXT: [[MIN:%.*]] = call i32 @llvm.umin.i32(i32 [[X]], i32 4)
; CHECK-NEXT: switch i32 [[X]], label %[[COMMON_RET:.*]] [
; CHECK-NEXT: i32 2, label %[[CASE_A:.*]]
; CHECK-NEXT: i32 3, label %[[CASE_B:.*]]
; CHECK-NEXT: ], !prof [[PROF1:![0-9]+]]
; CHECK: [[COMMON_RET]]:
; CHECK-NEXT: ret void
; CHECK: [[CASE_A]]:
; CHECK-NEXT: call void @a()
; CHECK-NEXT: br label %[[COMMON_RET]]
; CHECK: [[CASE_B]]:
; CHECK-NEXT: call void @b()
; CHECK-NEXT: br label %[[COMMON_RET]]
;
%min = call i32 @llvm.umin.i32(i32 %x, i32 4)
switch i32 %min, label %unreachable [
i32 2, label %case_a
i32 3, label %case_b
i32 4, label %case_ret
i32 5, label %case_ret
], !prof !1
case_a:
call void @a()
ret void
case_b:
call void @b()
ret void
case_ret:
ret void
unreachable:
unreachable
}
!0 = !{!"branch_weights", i32 1, i32 2, i32 3, i32 99, i32 5}
;.
; CHECK: [[PROF0]] = !{!"branch_weights", i32 5, i32 2, i32 3, i32 99}
;.
!1 = !{!"branch_weights", i32 11, i32 12, i32 13, i32 14, i32 15}
;.
; CHECK: [[PROF1]] = !{!"branch_weights", i32 14, i32 12, i32 13}
;.
Reference in New Issue View Git Blame Copy Permalink