| ; 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 |
| } |