xref: /llvm-project/llvm/test/CodeGen/RISCV/double-select-icmp.ll (revision 576d81baa5cf1801bae0fd05892be34acde33c6a)

; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py

; RUN: llc -mtriple=riscv32 -mattr=+d -verify-machineinstrs < %s \
; RUN:   -target-abi=ilp32d | FileCheck %s
; RUN: llc -mtriple=riscv64 -mattr=+d -verify-machineinstrs < %s \
; RUN:   -target-abi=lp64d | FileCheck %s
; RUN: llc -mtriple=riscv32 -mattr=+zdinx -verify-machineinstrs < %s \
; RUN:   -target-abi=ilp32 | FileCheck -check-prefixes=RV32ZDINX %s
; RUN: llc -mtriple=riscv64 -mattr=+zdinx -verify-machineinstrs < %s \
; RUN:   -target-abi=lp64 | FileCheck -check-prefixes=RV64ZDINX %s

define double @select_icmp_eq(i32 signext %a, i32 signext %b, double %c, double %d) {
; CHECK-LABEL: select_icmp_eq:
; CHECK:       # %bb.0:
; CHECK-NEXT:    beq a0, a1, .LBB0_2
; CHECK-NEXT:  # %bb.1:
; CHECK-NEXT:    fmv.d fa0, fa1
; CHECK-NEXT:  .LBB0_2:
; CHECK-NEXT:    ret
;
; RV32ZDINX-LABEL: select_icmp_eq:
; RV32ZDINX:       # %bb.0:
; RV32ZDINX-NEXT:    beq a0, a1, .LBB0_2
; RV32ZDINX-NEXT:  # %bb.1:
; RV32ZDINX-NEXT:    mv a2, a4
; RV32ZDINX-NEXT:    mv a3, a5
; RV32ZDINX-NEXT:  .LBB0_2:
; RV32ZDINX-NEXT:    mv a0, a2
; RV32ZDINX-NEXT:    mv a1, a3
; RV32ZDINX-NEXT:    ret
;
; RV64ZDINX-LABEL: select_icmp_eq:
; RV64ZDINX:       # %bb.0:
; RV64ZDINX-NEXT:    beq a0, a1, .LBB0_2
; RV64ZDINX-NEXT:  # %bb.1:
; RV64ZDINX-NEXT:    mv a2, a3
; RV64ZDINX-NEXT:  .LBB0_2:
; RV64ZDINX-NEXT:    mv a0, a2
; RV64ZDINX-NEXT:    ret
  %1 = icmp eq i32 %a, %b
  %2 = select i1 %1, double %c, double %d
  ret double %2
}

define double @select_icmp_ne(i32 signext %a, i32 signext %b, double %c, double %d) {
; CHECK-LABEL: select_icmp_ne:
; CHECK:       # %bb.0:
; CHECK-NEXT:    bne a0, a1, .LBB1_2
; CHECK-NEXT:  # %bb.1:
; CHECK-NEXT:    fmv.d fa0, fa1
; CHECK-NEXT:  .LBB1_2:
; CHECK-NEXT:    ret
;
; RV32ZDINX-LABEL: select_icmp_ne:
; RV32ZDINX:       # %bb.0:
; RV32ZDINX-NEXT:    bne a0, a1, .LBB1_2
; RV32ZDINX-NEXT:  # %bb.1:
; RV32ZDINX-NEXT:    mv a2, a4
; RV32ZDINX-NEXT:    mv a3, a5
; RV32ZDINX-NEXT:  .LBB1_2:
; RV32ZDINX-NEXT:    mv a0, a2
; RV32ZDINX-NEXT:    mv a1, a3
; RV32ZDINX-NEXT:    ret
;
; RV64ZDINX-LABEL: select_icmp_ne:
; RV64ZDINX:       # %bb.0:
; RV64ZDINX-NEXT:    bne a0, a1, .LBB1_2
; RV64ZDINX-NEXT:  # %bb.1:
; RV64ZDINX-NEXT:    mv a2, a3
; RV64ZDINX-NEXT:  .LBB1_2:
; RV64ZDINX-NEXT:    mv a0, a2
; RV64ZDINX-NEXT:    ret
  %1 = icmp ne i32 %a, %b
  %2 = select i1 %1, double %c, double %d
  ret double %2
}

define double @select_icmp_ugt(i32 signext %a, i32 signext %b, double %c, double %d) {
; CHECK-LABEL: select_icmp_ugt:
; CHECK:       # %bb.0:
; CHECK-NEXT:    bltu a1, a0, .LBB2_2
; CHECK-NEXT:  # %bb.1:
; CHECK-NEXT:    fmv.d fa0, fa1
; CHECK-NEXT:  .LBB2_2:
; CHECK-NEXT:    ret
;
; RV32ZDINX-LABEL: select_icmp_ugt:
; RV32ZDINX:       # %bb.0:
; RV32ZDINX-NEXT:    bltu a1, a0, .LBB2_2
; RV32ZDINX-NEXT:  # %bb.1:
; RV32ZDINX-NEXT:    mv a2, a4
; RV32ZDINX-NEXT:    mv a3, a5
; RV32ZDINX-NEXT:  .LBB2_2:
; RV32ZDINX-NEXT:    mv a0, a2
; RV32ZDINX-NEXT:    mv a1, a3
; RV32ZDINX-NEXT:    ret
;
; RV64ZDINX-LABEL: select_icmp_ugt:
; RV64ZDINX:       # %bb.0:
; RV64ZDINX-NEXT:    bltu a1, a0, .LBB2_2
; RV64ZDINX-NEXT:  # %bb.1:
; RV64ZDINX-NEXT:    mv a2, a3
; RV64ZDINX-NEXT:  .LBB2_2:
; RV64ZDINX-NEXT:    mv a0, a2
; RV64ZDINX-NEXT:    ret
  %1 = icmp ugt i32 %a, %b
  %2 = select i1 %1, double %c, double %d
  ret double %2
}

define double @select_icmp_uge(i32 signext %a, i32 signext %b, double %c, double %d) {
; CHECK-LABEL: select_icmp_uge:
; CHECK:       # %bb.0:
; CHECK-NEXT:    bgeu a0, a1, .LBB3_2
; CHECK-NEXT:  # %bb.1:
; CHECK-NEXT:    fmv.d fa0, fa1
; CHECK-NEXT:  .LBB3_2:
; CHECK-NEXT:    ret
;
; RV32ZDINX-LABEL: select_icmp_uge:
; RV32ZDINX:       # %bb.0:
; RV32ZDINX-NEXT:    bgeu a0, a1, .LBB3_2
; RV32ZDINX-NEXT:  # %bb.1:
; RV32ZDINX-NEXT:    mv a2, a4
; RV32ZDINX-NEXT:    mv a3, a5
; RV32ZDINX-NEXT:  .LBB3_2:
; RV32ZDINX-NEXT:    mv a0, a2
; RV32ZDINX-NEXT:    mv a1, a3
; RV32ZDINX-NEXT:    ret
;
; RV64ZDINX-LABEL: select_icmp_uge:
; RV64ZDINX:       # %bb.0:
; RV64ZDINX-NEXT:    bgeu a0, a1, .LBB3_2
; RV64ZDINX-NEXT:  # %bb.1:
; RV64ZDINX-NEXT:    mv a2, a3
; RV64ZDINX-NEXT:  .LBB3_2:
; RV64ZDINX-NEXT:    mv a0, a2
; RV64ZDINX-NEXT:    ret
  %1 = icmp uge i32 %a, %b
  %2 = select i1 %1, double %c, double %d
  ret double %2
}

define double @select_icmp_ult(i32 signext %a, i32 signext %b, double %c, double %d) {
; CHECK-LABEL: select_icmp_ult:
; CHECK:       # %bb.0:
; CHECK-NEXT:    bltu a0, a1, .LBB4_2
; CHECK-NEXT:  # %bb.1:
; CHECK-NEXT:    fmv.d fa0, fa1
; CHECK-NEXT:  .LBB4_2:
; CHECK-NEXT:    ret
;
; RV32ZDINX-LABEL: select_icmp_ult:
; RV32ZDINX:       # %bb.0:
; RV32ZDINX-NEXT:    bltu a0, a1, .LBB4_2
; RV32ZDINX-NEXT:  # %bb.1:
; RV32ZDINX-NEXT:    mv a2, a4
; RV32ZDINX-NEXT:    mv a3, a5
; RV32ZDINX-NEXT:  .LBB4_2:
; RV32ZDINX-NEXT:    mv a0, a2
; RV32ZDINX-NEXT:    mv a1, a3
; RV32ZDINX-NEXT:    ret
;
; RV64ZDINX-LABEL: select_icmp_ult:
; RV64ZDINX:       # %bb.0:
; RV64ZDINX-NEXT:    bltu a0, a1, .LBB4_2
; RV64ZDINX-NEXT:  # %bb.1:
; RV64ZDINX-NEXT:    mv a2, a3
; RV64ZDINX-NEXT:  .LBB4_2:
; RV64ZDINX-NEXT:    mv a0, a2
; RV64ZDINX-NEXT:    ret
  %1 = icmp ult i32 %a, %b
  %2 = select i1 %1, double %c, double %d
  ret double %2
}

define double @select_icmp_ule(i32 signext %a, i32 signext %b, double %c, double %d) {
; CHECK-LABEL: select_icmp_ule:
; CHECK:       # %bb.0:
; CHECK-NEXT:    bgeu a1, a0, .LBB5_2
; CHECK-NEXT:  # %bb.1:
; CHECK-NEXT:    fmv.d fa0, fa1
; CHECK-NEXT:  .LBB5_2:
; CHECK-NEXT:    ret
;
; RV32ZDINX-LABEL: select_icmp_ule:
; RV32ZDINX:       # %bb.0:
; RV32ZDINX-NEXT:    bgeu a1, a0, .LBB5_2
; RV32ZDINX-NEXT:  # %bb.1:
; RV32ZDINX-NEXT:    mv a2, a4
; RV32ZDINX-NEXT:    mv a3, a5
; RV32ZDINX-NEXT:  .LBB5_2:
; RV32ZDINX-NEXT:    mv a0, a2
; RV32ZDINX-NEXT:    mv a1, a3
; RV32ZDINX-NEXT:    ret
;
; RV64ZDINX-LABEL: select_icmp_ule:
; RV64ZDINX:       # %bb.0:
; RV64ZDINX-NEXT:    bgeu a1, a0, .LBB5_2
; RV64ZDINX-NEXT:  # %bb.1:
; RV64ZDINX-NEXT:    mv a2, a3
; RV64ZDINX-NEXT:  .LBB5_2:
; RV64ZDINX-NEXT:    mv a0, a2
; RV64ZDINX-NEXT:    ret
  %1 = icmp ule i32 %a, %b
  %2 = select i1 %1, double %c, double %d
  ret double %2
}

define double @select_icmp_sgt(i32 signext %a, i32 signext %b, double %c, double %d) {
; CHECK-LABEL: select_icmp_sgt:
; CHECK:       # %bb.0:
; CHECK-NEXT:    blt a1, a0, .LBB6_2
; CHECK-NEXT:  # %bb.1:
; CHECK-NEXT:    fmv.d fa0, fa1
; CHECK-NEXT:  .LBB6_2:
; CHECK-NEXT:    ret
;
; RV32ZDINX-LABEL: select_icmp_sgt:
; RV32ZDINX:       # %bb.0:
; RV32ZDINX-NEXT:    blt a1, a0, .LBB6_2
; RV32ZDINX-NEXT:  # %bb.1:
; RV32ZDINX-NEXT:    mv a2, a4
; RV32ZDINX-NEXT:    mv a3, a5
; RV32ZDINX-NEXT:  .LBB6_2:
; RV32ZDINX-NEXT:    mv a0, a2
; RV32ZDINX-NEXT:    mv a1, a3
; RV32ZDINX-NEXT:    ret
;
; RV64ZDINX-LABEL: select_icmp_sgt:
; RV64ZDINX:       # %bb.0:
; RV64ZDINX-NEXT:    blt a1, a0, .LBB6_2
; RV64ZDINX-NEXT:  # %bb.1:
; RV64ZDINX-NEXT:    mv a2, a3
; RV64ZDINX-NEXT:  .LBB6_2:
; RV64ZDINX-NEXT:    mv a0, a2
; RV64ZDINX-NEXT:    ret
  %1 = icmp sgt i32 %a, %b
  %2 = select i1 %1, double %c, double %d
  ret double %2
}

define double @select_icmp_sge(i32 signext %a, i32 signext %b, double %c, double %d) {
; CHECK-LABEL: select_icmp_sge:
; CHECK:       # %bb.0:
; CHECK-NEXT:    bge a0, a1, .LBB7_2
; CHECK-NEXT:  # %bb.1:
; CHECK-NEXT:    fmv.d fa0, fa1
; CHECK-NEXT:  .LBB7_2:
; CHECK-NEXT:    ret
;
; RV32ZDINX-LABEL: select_icmp_sge:
; RV32ZDINX:       # %bb.0:
; RV32ZDINX-NEXT:    bge a0, a1, .LBB7_2
; RV32ZDINX-NEXT:  # %bb.1:
; RV32ZDINX-NEXT:    mv a2, a4
; RV32ZDINX-NEXT:    mv a3, a5
; RV32ZDINX-NEXT:  .LBB7_2:
; RV32ZDINX-NEXT:    mv a0, a2
; RV32ZDINX-NEXT:    mv a1, a3
; RV32ZDINX-NEXT:    ret
;
; RV64ZDINX-LABEL: select_icmp_sge:
; RV64ZDINX:       # %bb.0:
; RV64ZDINX-NEXT:    bge a0, a1, .LBB7_2
; RV64ZDINX-NEXT:  # %bb.1:
; RV64ZDINX-NEXT:    mv a2, a3
; RV64ZDINX-NEXT:  .LBB7_2:
; RV64ZDINX-NEXT:    mv a0, a2
; RV64ZDINX-NEXT:    ret
  %1 = icmp sge i32 %a, %b
  %2 = select i1 %1, double %c, double %d
  ret double %2
}

define double @select_icmp_slt(i32 signext %a, i32 signext %b, double %c, double %d) {
; CHECK-LABEL: select_icmp_slt:
; CHECK:       # %bb.0:
; CHECK-NEXT:    blt a0, a1, .LBB8_2
; CHECK-NEXT:  # %bb.1:
; CHECK-NEXT:    fmv.d fa0, fa1
; CHECK-NEXT:  .LBB8_2:
; CHECK-NEXT:    ret
;
; RV32ZDINX-LABEL: select_icmp_slt:
; RV32ZDINX:       # %bb.0:
; RV32ZDINX-NEXT:    blt a0, a1, .LBB8_2
; RV32ZDINX-NEXT:  # %bb.1:
; RV32ZDINX-NEXT:    mv a2, a4
; RV32ZDINX-NEXT:    mv a3, a5
; RV32ZDINX-NEXT:  .LBB8_2:
; RV32ZDINX-NEXT:    mv a0, a2
; RV32ZDINX-NEXT:    mv a1, a3
; RV32ZDINX-NEXT:    ret
;
; RV64ZDINX-LABEL: select_icmp_slt:
; RV64ZDINX:       # %bb.0:
; RV64ZDINX-NEXT:    blt a0, a1, .LBB8_2
; RV64ZDINX-NEXT:  # %bb.1:
; RV64ZDINX-NEXT:    mv a2, a3
; RV64ZDINX-NEXT:  .LBB8_2:
; RV64ZDINX-NEXT:    mv a0, a2
; RV64ZDINX-NEXT:    ret
  %1 = icmp slt i32 %a, %b
  %2 = select i1 %1, double %c, double %d
  ret double %2
}

define double @select_icmp_sle(i32 signext %a, i32 signext %b, double %c, double %d) {
; CHECK-LABEL: select_icmp_sle:
; CHECK:       # %bb.0:
; CHECK-NEXT:    bge a1, a0, .LBB9_2
; CHECK-NEXT:  # %bb.1:
; CHECK-NEXT:    fmv.d fa0, fa1
; CHECK-NEXT:  .LBB9_2:
; CHECK-NEXT:    ret
;
; RV32ZDINX-LABEL: select_icmp_sle:
; RV32ZDINX:       # %bb.0:
; RV32ZDINX-NEXT:    bge a1, a0, .LBB9_2
; RV32ZDINX-NEXT:  # %bb.1:
; RV32ZDINX-NEXT:    mv a2, a4
; RV32ZDINX-NEXT:    mv a3, a5
; RV32ZDINX-NEXT:  .LBB9_2:
; RV32ZDINX-NEXT:    mv a0, a2
; RV32ZDINX-NEXT:    mv a1, a3
; RV32ZDINX-NEXT:    ret
;
; RV64ZDINX-LABEL: select_icmp_sle:
; RV64ZDINX:       # %bb.0:
; RV64ZDINX-NEXT:    bge a1, a0, .LBB9_2
; RV64ZDINX-NEXT:  # %bb.1:
; RV64ZDINX-NEXT:    mv a2, a3
; RV64ZDINX-NEXT:  .LBB9_2:
; RV64ZDINX-NEXT:    mv a0, a2
; RV64ZDINX-NEXT:    ret
  %1 = icmp sle i32 %a, %b
  %2 = select i1 %1, double %c, double %d
  ret double %2
}

define double @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.d.w fa0, a0
; CHECK-NEXT:    ret
;
; RV32ZDINX-LABEL: select_icmp_slt_one:
; RV32ZDINX:       # %bb.0:
; RV32ZDINX-NEXT:    slti a0, a0, 1
; RV32ZDINX-NEXT:    fcvt.d.w a0, a0
; RV32ZDINX-NEXT:    ret
;
; RV64ZDINX-LABEL: select_icmp_slt_one:
; RV64ZDINX:       # %bb.0:
; RV64ZDINX-NEXT:    slti a0, a0, 1
; RV64ZDINX-NEXT:    fcvt.d.w a0, a0
; RV64ZDINX-NEXT:    ret
  %1 = icmp slt i32 %a, 1
  %2 = select i1 %1, double 1.000000e+00, double 0.000000e+00
  ret double %2
}

define double @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.d.w fa0, a0
; CHECK-NEXT:    ret
;
; RV32ZDINX-LABEL: select_icmp_sgt_zero:
; RV32ZDINX:       # %bb.0:
; RV32ZDINX-NEXT:    slti a0, a0, 1
; RV32ZDINX-NEXT:    fcvt.d.w a0, a0
; RV32ZDINX-NEXT:    ret
;
; RV64ZDINX-LABEL: select_icmp_sgt_zero:
; RV64ZDINX:       # %bb.0:
; RV64ZDINX-NEXT:    slti a0, a0, 1
; RV64ZDINX-NEXT:    fcvt.d.w a0, a0
; RV64ZDINX-NEXT:    ret
  %1 = icmp sgt i32 %a, 0
  %2 = select i1 %1, double 0.000000e+00, double 1.000000e+00
  ret double %2
}