Files
wasmtime/cranelift/filetests/filetests/egraph/vselect.clif
Afonso Bordado 76539ef9f2 cranelift: Optimize select+icmp into {s,u}{min,max} (#5546)
* cranelift: Optimize `select+icmp` into `{s,u}{min,max}`

* cranelift: Add generic egraph icmp reverse rule

* cranelift: Optimize `vselect+icmp` into `{s,u}{min,max}`

* cranelift: Optimize some `vselect+fcmp` into `f{min,max}_pseudo`

* cranelift: Add inverted forms of min/max rules
2023-02-15 15:06:21 -08:00

155 lines
3.2 KiB
Plaintext

test optimize
set opt_level=speed
set use_egraphs=true
target x86_64
target aarch64
target s390x
function %vselect_sgt_to_smax(i32x4, i32x4) -> i32x4 {
block0(v0: i32x4, v1: i32x4):
v2 = icmp sgt v0, v1
v3 = vselect v2, v0, v1
return v3
}
; check: block0(v0: i32x4, v1: i32x4):
; check: v4 = smax v0, v1
; check: return v4
; This tests an inverted vselect, where the operands are swapped.
function %vselect_sgt_to_smax(i32x4, i32x4) -> i32x4 {
block0(v0: i32x4, v1: i32x4):
v2 = icmp sgt v0, v1
v3 = vselect v2, v1, v0
return v3
}
; check: block0(v0: i32x4, v1: i32x4):
; check: v4 = smin v0, v1
; check: return v4
function %vselect_sge_to_smax(i32x4, i32x4) -> i32x4 {
block0(v0: i32x4, v1: i32x4):
v2 = icmp sge v0, v1
v3 = vselect v2, v0, v1
return v3
}
; check: block0(v0: i32x4, v1: i32x4):
; check: v4 = smax v0, v1
; check: return v4
function %vselect_ugt_to_umax(i32x4, i32x4) -> i32x4 {
block0(v0: i32x4, v1: i32x4):
v2 = icmp ugt v0, v1
v3 = vselect v2, v0, v1
return v3
}
; check: block0(v0: i32x4, v1: i32x4):
; check: v4 = umax v0, v1
; check: return v4
function %vselect_uge_to_umax(i32x4, i32x4) -> i32x4 {
block0(v0: i32x4, v1: i32x4):
v2 = icmp uge v0, v1
v3 = vselect v2, v0, v1
return v3
}
; check: block0(v0: i32x4, v1: i32x4):
; check: v4 = umax v0, v1
; check: return v4
function %vselect_slt_to_smin(i32x4, i32x4) -> i32x4 {
block0(v0: i32x4, v1: i32x4):
v2 = icmp slt v0, v1
v3 = vselect v2, v0, v1
return v3
}
; check: block0(v0: i32x4, v1: i32x4):
; check: v4 = smin v0, v1
; check: return v4
function %vselect_sle_to_smin(i32x4, i32x4) -> i32x4 {
block0(v0: i32x4, v1: i32x4):
v2 = icmp sle v0, v1
v3 = vselect v2, v0, v1
return v3
}
; check: block0(v0: i32x4, v1: i32x4):
; check: v4 = smin v0, v1
; check: return v4
function %vselect_ult_to_umin(i32x4, i32x4) -> i32x4 {
block0(v0: i32x4, v1: i32x4):
v2 = icmp ult v0, v1
v3 = vselect v2, v0, v1
return v3
}
; check: block0(v0: i32x4, v1: i32x4):
; check: v4 = umin v0, v1
; check: return v4
function %vselect_ule_to_umin(i32x4, i32x4) -> i32x4 {
block0(v0: i32x4, v1: i32x4):
v2 = icmp ule v0, v1
v3 = vselect v2, v0, v1
return v3
}
; check: block0(v0: i32x4, v1: i32x4):
; check: v4 = umin v0, v1
; check: return v4
function %vselect_with_different_regs_does_not_optimize(i32x4, i32x4, i32x4, i32x4) -> i32x4 {
block0(v0: i32x4, v1: i32x4, v2: i32x4, v3: i32x4):
v4 = icmp ule v0, v1
v5 = vselect v4, v2, v3
return v5
}
; check: block0(v0: i32x4, v1: i32x4, v2: i32x4, v3: i32x4):
; check: v4 = icmp ule v0, v1
; check: v5 = vselect v4, v2, v3
; check: return v5
function %vselect_fcmp_gt_to_fmax_pseudo(f32x4, f32x4) -> f32x4 {
block0(v0: f32x4, v1: f32x4):
v2 = fcmp gt v0, v1
v3 = vselect v2, v0, v1
return v3
}
; check: block0(v0: f32x4, v1: f32x4):
; check: v4 = fmax_pseudo v0, v1
; check: return v4
function %vselect_fcmp_lt_to_fmin_pseudo(f32x4, f32x4) -> f32x4 {
block0(v0: f32x4, v1: f32x4):
v2 = fcmp lt v0, v1
v3 = vselect v2, v0, v1
return v3
}
; check: block0(v0: f32x4, v1: f32x4):
; check: v4 = fmin_pseudo v0, v1
; check: return v4