1c874bbbd6
Select only needs branches and moves so we don't need to promote it. Promoting would canonicalize NaNs which select shouldn't do.
160 lines
4.6 KiB
LLVM
160 lines
4.6 KiB
LLVM
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 2
|
|
; RUN: llc -mtriple=riscv32 -mattr=+zfbfmin -verify-machineinstrs \
|
|
; RUN: -target-abi ilp32f < %s | FileCheck %s
|
|
; RUN: llc -mtriple=riscv64 -mattr=+zfbfmin -verify-machineinstrs \
|
|
; RUN: -target-abi lp64f < %s | FileCheck %s
|
|
|
|
define bfloat @select_icmp_eq(i32 signext %a, i32 signext %b, bfloat %c, bfloat %d) {
|
|
; CHECK-LABEL: select_icmp_eq:
|
|
; CHECK: # %bb.0:
|
|
; CHECK-NEXT: beq a0, a1, .LBB0_2
|
|
; CHECK-NEXT: # %bb.1:
|
|
; CHECK-NEXT: fmv.s fa0, fa1
|
|
; CHECK-NEXT: .LBB0_2:
|
|
; CHECK-NEXT: ret
|
|
%1 = icmp eq i32 %a, %b
|
|
%2 = select i1 %1, bfloat %c, bfloat %d
|
|
ret bfloat %2
|
|
}
|
|
|
|
define bfloat @select_icmp_ne(i32 signext %a, i32 signext %b, bfloat %c, bfloat %d) {
|
|
; CHECK-LABEL: select_icmp_ne:
|
|
; CHECK: # %bb.0:
|
|
; CHECK-NEXT: bne a0, a1, .LBB1_2
|
|
; CHECK-NEXT: # %bb.1:
|
|
; CHECK-NEXT: fmv.s fa0, fa1
|
|
; CHECK-NEXT: .LBB1_2:
|
|
; CHECK-NEXT: ret
|
|
%1 = icmp ne i32 %a, %b
|
|
%2 = select i1 %1, bfloat %c, bfloat %d
|
|
ret bfloat %2
|
|
}
|
|
|
|
define bfloat @select_icmp_ugt(i32 signext %a, i32 signext %b, bfloat %c, bfloat %d) {
|
|
; CHECK-LABEL: select_icmp_ugt:
|
|
; CHECK: # %bb.0:
|
|
; CHECK-NEXT: bltu a1, a0, .LBB2_2
|
|
; CHECK-NEXT: # %bb.1:
|
|
; CHECK-NEXT: fmv.s fa0, fa1
|
|
; CHECK-NEXT: .LBB2_2:
|
|
; CHECK-NEXT: ret
|
|
%1 = icmp ugt i32 %a, %b
|
|
%2 = select i1 %1, bfloat %c, bfloat %d
|
|
ret bfloat %2
|
|
}
|
|
|
|
define bfloat @select_icmp_uge(i32 signext %a, i32 signext %b, bfloat %c, bfloat %d) {
|
|
; CHECK-LABEL: select_icmp_uge:
|
|
; CHECK: # %bb.0:
|
|
; CHECK-NEXT: bgeu a0, a1, .LBB3_2
|
|
; CHECK-NEXT: # %bb.1:
|
|
; CHECK-NEXT: fmv.s fa0, fa1
|
|
; CHECK-NEXT: .LBB3_2:
|
|
; CHECK-NEXT: ret
|
|
%1 = icmp uge i32 %a, %b
|
|
%2 = select i1 %1, bfloat %c, bfloat %d
|
|
ret bfloat %2
|
|
}
|
|
|
|
define bfloat @select_icmp_ult(i32 signext %a, i32 signext %b, bfloat %c, bfloat %d) {
|
|
; CHECK-LABEL: select_icmp_ult:
|
|
; CHECK: # %bb.0:
|
|
; CHECK-NEXT: bltu a0, a1, .LBB4_2
|
|
; CHECK-NEXT: # %bb.1:
|
|
; CHECK-NEXT: fmv.s fa0, fa1
|
|
; CHECK-NEXT: .LBB4_2:
|
|
; CHECK-NEXT: ret
|
|
%1 = icmp ult i32 %a, %b
|
|
%2 = select i1 %1, bfloat %c, bfloat %d
|
|
ret bfloat %2
|
|
}
|
|
|
|
define bfloat @select_icmp_ule(i32 signext %a, i32 signext %b, bfloat %c, bfloat %d) {
|
|
; CHECK-LABEL: select_icmp_ule:
|
|
; CHECK: # %bb.0:
|
|
; CHECK-NEXT: bgeu a1, a0, .LBB5_2
|
|
; CHECK-NEXT: # %bb.1:
|
|
; CHECK-NEXT: fmv.s fa0, fa1
|
|
; CHECK-NEXT: .LBB5_2:
|
|
; CHECK-NEXT: ret
|
|
%1 = icmp ule i32 %a, %b
|
|
%2 = select i1 %1, bfloat %c, bfloat %d
|
|
ret bfloat %2
|
|
}
|
|
|
|
define bfloat @select_icmp_sgt(i32 signext %a, i32 signext %b, bfloat %c, bfloat %d) {
|
|
; CHECK-LABEL: select_icmp_sgt:
|
|
; CHECK: # %bb.0:
|
|
; CHECK-NEXT: blt a1, a0, .LBB6_2
|
|
; CHECK-NEXT: # %bb.1:
|
|
; CHECK-NEXT: fmv.s fa0, fa1
|
|
; CHECK-NEXT: .LBB6_2:
|
|
; CHECK-NEXT: ret
|
|
%1 = icmp sgt i32 %a, %b
|
|
%2 = select i1 %1, bfloat %c, bfloat %d
|
|
ret bfloat %2
|
|
}
|
|
|
|
define bfloat @select_icmp_sge(i32 signext %a, i32 signext %b, bfloat %c, bfloat %d) {
|
|
; CHECK-LABEL: select_icmp_sge:
|
|
; CHECK: # %bb.0:
|
|
; CHECK-NEXT: bge a0, a1, .LBB7_2
|
|
; CHECK-NEXT: # %bb.1:
|
|
; CHECK-NEXT: fmv.s fa0, fa1
|
|
; CHECK-NEXT: .LBB7_2:
|
|
; CHECK-NEXT: ret
|
|
%1 = icmp sge i32 %a, %b
|
|
%2 = select i1 %1, bfloat %c, bfloat %d
|
|
ret bfloat %2
|
|
}
|
|
|
|
define bfloat @select_icmp_slt(i32 signext %a, i32 signext %b, bfloat %c, bfloat %d) {
|
|
; CHECK-LABEL: select_icmp_slt:
|
|
; CHECK: # %bb.0:
|
|
; CHECK-NEXT: blt a0, a1, .LBB8_2
|
|
; CHECK-NEXT: # %bb.1:
|
|
; CHECK-NEXT: fmv.s fa0, fa1
|
|
; CHECK-NEXT: .LBB8_2:
|
|
; CHECK-NEXT: ret
|
|
%1 = icmp slt i32 %a, %b
|
|
%2 = select i1 %1, bfloat %c, bfloat %d
|
|
ret bfloat %2
|
|
}
|
|
|
|
define bfloat @select_icmp_sle(i32 signext %a, i32 signext %b, bfloat %c, bfloat %d) {
|
|
; CHECK-LABEL: select_icmp_sle:
|
|
; CHECK: # %bb.0:
|
|
; CHECK-NEXT: bge a1, a0, .LBB9_2
|
|
; CHECK-NEXT: # %bb.1:
|
|
; CHECK-NEXT: fmv.s fa0, fa1
|
|
; CHECK-NEXT: .LBB9_2:
|
|
; CHECK-NEXT: ret
|
|
%1 = icmp sle i32 %a, %b
|
|
%2 = select i1 %1, bfloat %c, bfloat %d
|
|
ret bfloat %2
|
|
}
|
|
|
|
define bfloat @select_icmp_slt_one(i32 signext %a) {
|
|
; CHECK-LABEL: select_icmp_slt_one:
|
|
; CHECK: # %bb.0:
|
|
; CHECK-NEXT: slti a0, a0, 1
|
|
; CHECK-NEXT: fcvt.s.w fa5, a0
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; CHECK-NEXT: ret
|
|
%1 = icmp slt i32 %a, 1
|
|
%2 = select i1 %1, bfloat 1.000000e+00, bfloat 0.000000e+00
|
|
ret bfloat %2
|
|
}
|
|
|
|
define bfloat @select_icmp_sgt_zero(i32 signext %a) {
|
|
; CHECK-LABEL: select_icmp_sgt_zero:
|
|
; CHECK: # %bb.0:
|
|
; CHECK-NEXT: slti a0, a0, 1
|
|
; CHECK-NEXT: fcvt.s.w fa5, a0
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; CHECK-NEXT: ret
|
|
%1 = icmp sgt i32 %a, 0
|
|
%2 = select i1 %1, bfloat 0.000000e+00, bfloat 1.000000e+00
|
|
ret bfloat %2
|
|
}
|