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