blob: 60df7ef011df23572f7ea7444d66a8a663098300 [file] [edit]
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 6
; RUN: llc -mtriple=riscv64 -mattr=+b,+f,+m < %s | FileCheck %s
define bfloat @fp8e5m2_to_bf16(i8 signext %x) nounwind {
; CHECK-LABEL: fp8e5m2_to_bf16:
; CHECK: # %bb.0:
; CHECK-NEXT: lui a4, 524288
; CHECK-NEXT: slli a3, a0, 57
; CHECK-NEXT: andi a2, a0, 3
; CHECK-NEXT: srli a3, a3, 59
; CHECK-NEXT: snez a1, a2
; CHECK-NEXT: seqz a5, a3
; CHECK-NEXT: and a5, a5, a1
; CHECK-NEXT: and a0, a0, a4
; CHECK-NEXT: bnez a5, .LBB0_2
; CHECK-NEXT: # %bb.1:
; CHECK-NEXT: addi a4, a3, 112
; CHECK-NEXT: slli a5, a2, 21
; CHECK-NEXT: slli a4, a4, 23
; CHECK-NEXT: or a5, a0, a5
; CHECK-NEXT: j .LBB0_3
; CHECK-NEXT: .LBB0_2:
; CHECK-NEXT: clz a4, a2
; CHECK-NEXT: li a5, 174
; CHECK-NEXT: sub a5, a5, a4
; CHECK-NEXT: not a6, a4
; CHECK-NEXT: slli a5, a5, 23
; CHECK-NEXT: binv a6, a2, a6
; CHECK-NEXT: addi a4, a4, -40
; CHECK-NEXT: or a5, a0, a5
; CHECK-NEXT: sll a4, a6, a4
; CHECK-NEXT: .LBB0_3:
; CHECK-NEXT: or a5, a5, a4
; CHECK-NEXT: or a6, a3, a2
; CHECK-NEXT: mv a4, a0
; CHECK-NEXT: beqz a6, .LBB0_5
; CHECK-NEXT: # %bb.4:
; CHECK-NEXT: mv a4, a5
; CHECK-NEXT: .LBB0_5:
; CHECK-NEXT: addi a3, a3, -31
; CHECK-NEXT: seqz a5, a2
; CHECK-NEXT: seqz a2, a3
; CHECK-NEXT: and a5, a2, a5
; CHECK-NEXT: beqz a5, .LBB0_7
; CHECK-NEXT: # %bb.6:
; CHECK-NEXT: lui a4, 522240
; CHECK-NEXT: or a4, a0, a4
; CHECK-NEXT: .LBB0_7:
; CHECK-NEXT: and a1, a2, a1
; CHECK-NEXT: lui a0, 523264
; CHECK-NEXT: bnez a1, .LBB0_9
; CHECK-NEXT: # %bb.8:
; CHECK-NEXT: mv a0, a4
; CHECK-NEXT: .LBB0_9:
; CHECK-NEXT: addi sp, sp, -16
; CHECK-NEXT: sd ra, 8(sp) # 8-byte Folded Spill
; CHECK-NEXT: sw a0, 4(sp)
; CHECK-NEXT: flw fa0, 4(sp)
; CHECK-NEXT: call __truncsfbf2
; CHECK-NEXT: fmv.x.w a0, fa0
; CHECK-NEXT: lui a1, 1048560
; CHECK-NEXT: or a0, a0, a1
; CHECK-NEXT: fmv.w.x fa0, a0
; CHECK-NEXT: ld ra, 8(sp) # 8-byte Folded Reload
; CHECK-NEXT: addi sp, sp, 16
; CHECK-NEXT: ret
%r = call bfloat @llvm.convert.from.arbitrary.fp(i8 %x, metadata !"Float8E5M2")
ret bfloat %r
}