199 lines
6 KiB
LLVM
199 lines
6 KiB
LLVM
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 2
|
|
; RUN: llc -mtriple=riscv32 -mattr=+experimental-zfbfmin -verify-machineinstrs \
|
|
; RUN: -target-abi ilp32f < %s | FileCheck %s
|
|
; RUN: llc -mtriple=riscv64 -mattr=+experimental-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: fcvt.s.bf16 fa5, fa1
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; CHECK-NEXT: ret
|
|
; CHECK-NEXT: .LBB0_2:
|
|
; CHECK-NEXT: fcvt.s.bf16 fa5, fa0
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; 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: fcvt.s.bf16 fa5, fa1
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; CHECK-NEXT: ret
|
|
; CHECK-NEXT: .LBB1_2:
|
|
; CHECK-NEXT: fcvt.s.bf16 fa5, fa0
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; 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: fcvt.s.bf16 fa5, fa1
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; CHECK-NEXT: ret
|
|
; CHECK-NEXT: .LBB2_2:
|
|
; CHECK-NEXT: fcvt.s.bf16 fa5, fa0
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; 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: fcvt.s.bf16 fa5, fa1
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; CHECK-NEXT: ret
|
|
; CHECK-NEXT: .LBB3_2:
|
|
; CHECK-NEXT: fcvt.s.bf16 fa5, fa0
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; 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: fcvt.s.bf16 fa5, fa1
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; CHECK-NEXT: ret
|
|
; CHECK-NEXT: .LBB4_2:
|
|
; CHECK-NEXT: fcvt.s.bf16 fa5, fa0
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; 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: fcvt.s.bf16 fa5, fa1
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; CHECK-NEXT: ret
|
|
; CHECK-NEXT: .LBB5_2:
|
|
; CHECK-NEXT: fcvt.s.bf16 fa5, fa0
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; 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: fcvt.s.bf16 fa5, fa1
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; CHECK-NEXT: ret
|
|
; CHECK-NEXT: .LBB6_2:
|
|
; CHECK-NEXT: fcvt.s.bf16 fa5, fa0
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; 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: fcvt.s.bf16 fa5, fa1
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; CHECK-NEXT: ret
|
|
; CHECK-NEXT: .LBB7_2:
|
|
; CHECK-NEXT: fcvt.s.bf16 fa5, fa0
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; 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: fcvt.s.bf16 fa5, fa1
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; CHECK-NEXT: ret
|
|
; CHECK-NEXT: .LBB8_2:
|
|
; CHECK-NEXT: fcvt.s.bf16 fa5, fa0
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; 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: fcvt.s.bf16 fa5, fa1
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; CHECK-NEXT: ret
|
|
; CHECK-NEXT: .LBB9_2:
|
|
; CHECK-NEXT: fcvt.s.bf16 fa5, fa0
|
|
; CHECK-NEXT: fcvt.bf16.s fa0, fa5
|
|
; 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
|
|
}
|