llvm-project/llvm/test/CodeGen/LoongArch/annotate-tablejump.ll
hev 746c682c4a
[LoongArch] Introduce 32s target feature for LA32S ISA extensions (#139695)
According to the offical LoongArch reference manual, the 32-bit
LoongArch is divied into two variants: the Reduced version (LA32R) and
Standard version (LA32S). LA32S extends LA32R by adding additional
instructions, and the 64-bit version (LA64) fully includes the LA32S
instruction set.

This patch introduces a new target feature `32s` for the LoongArch
backend, enabling support for instructions specific to the LA32S
variant.

The LA32S exntension includes the following additional instructions:

- ALSL.W
- {AND,OR}N
- B{EQ,NE}Z
- BITREV.{4B,W}
- BSTR{INS,PICK}.W
- BYTEPICK.W
- CL{O,Z}.W
- CPUCFG
- CT{O,Z}.W
- EXT.W,{B,H}
- F{LD,ST}X.{D,S}
- MASK{EQ,NE}Z
- PC{ADDI,ALAU12I}
- REVB.2H
- ROTR{I},W

Additionally, LA32R defines three new instruction aliases:

- RDCNTID.W RJ => RDTIMEL.W ZERO, RJ
- RDCNTVH.W RD => RDTIMEH.W RD, ZERO
- RDCNTVL.W RD => RDTIMEL.W RD, ZERO
2025-05-20 18:28:08 +08:00

134 lines
4.1 KiB
LLVM

; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 5
; RUN: llc --mtriple=loongarch32 -mattr=+32s,+d \
; RUN: --min-jump-table-entries=4 < %s \
; RUN: --loongarch-annotate-tablejump \
; RUN: | FileCheck %s --check-prefix=LA32-JT
; RUN: llc --mtriple=loongarch64 -mattr=+d \
; RUN: --min-jump-table-entries=4 < %s \
; RUN: --loongarch-annotate-tablejump \
; RUN: | FileCheck %s --check-prefix=LA64-JT
define void @switch_4_arms(i32 %in, ptr %out) nounwind {
; LA32-JT-LABEL: switch_4_arms:
; LA32-JT: # %bb.0: # %entry
; LA32-JT-NEXT: addi.w $a3, $a0, -1
; LA32-JT-NEXT: ori $a2, $zero, 3
; LA32-JT-NEXT: bltu $a2, $a3, .LBB0_7
; LA32-JT-NEXT: # %bb.1: # %entry
; LA32-JT-NEXT: pcalau12i $a4, %pc_hi20(.LJTI0_0)
; LA32-JT-NEXT: addi.w $a4, $a4, %pc_lo12(.LJTI0_0)
; LA32-JT-NEXT: alsl.w $a3, $a3, $a4, 2
; LA32-JT-NEXT: ld.w $a3, $a3, 0
; LA32-JT-NEXT: .Ljrtb_0:
; LA32-JT-NEXT: jr $a3
; LA32-JT-NEXT: .LBB0_2: # %bb1
; LA32-JT-NEXT: ori $a3, $zero, 4
; LA32-JT-NEXT: b .LBB0_6
; LA32-JT-NEXT: .LBB0_3: # %bb2
; LA32-JT-NEXT: ori $a3, $zero, 3
; LA32-JT-NEXT: b .LBB0_6
; LA32-JT-NEXT: .LBB0_4: # %bb3
; LA32-JT-NEXT: ori $a3, $zero, 2
; LA32-JT-NEXT: b .LBB0_6
; LA32-JT-NEXT: .LBB0_5: # %bb4
; LA32-JT-NEXT: ori $a3, $zero, 1
; LA32-JT-NEXT: .LBB0_6: # %exit
; LA32-JT-NEXT: st.w $a3, $a1, 0
; LA32-JT-NEXT: .LBB0_7: # %exit
; LA32-JT-NEXT: addi.w $a3, $a0, -5
; LA32-JT-NEXT: bltu $a2, $a3, .LBB0_9
; LA32-JT-NEXT: # %bb.8: # %exit
; LA32-JT-NEXT: pcalau12i $a4, %pc_hi20(.LJTI0_1)
; LA32-JT-NEXT: addi.w $a4, $a4, %pc_lo12(.LJTI0_1)
; LA32-JT-NEXT: alsl.w $a3, $a3, $a4, 2
; LA32-JT-NEXT: ld.w $a3, $a3, 0
; LA32-JT-NEXT: .Ljrtb_1:
; LA32-JT-NEXT: jr $a3
; LA32-JT-NEXT: .LBB0_9: # %exit2
; LA32-JT-NEXT: ret
;
; LA64-JT-LABEL: switch_4_arms:
; LA64-JT: # %bb.0: # %entry
; LA64-JT-NEXT: addi.w $a0, $a0, 0
; LA64-JT-NEXT: addi.d $a3, $a0, -1
; LA64-JT-NEXT: ori $a2, $zero, 3
; LA64-JT-NEXT: bltu $a2, $a3, .LBB0_7
; LA64-JT-NEXT: # %bb.1: # %entry
; LA64-JT-NEXT: slli.d $a3, $a3, 3
; LA64-JT-NEXT: pcalau12i $a4, %pc_hi20(.LJTI0_0)
; LA64-JT-NEXT: addi.d $a4, $a4, %pc_lo12(.LJTI0_0)
; LA64-JT-NEXT: ldx.d $a3, $a4, $a3
; LA64-JT-NEXT: .Ljrtb_0:
; LA64-JT-NEXT: jr $a3
; LA64-JT-NEXT: .LBB0_2: # %bb1
; LA64-JT-NEXT: ori $a3, $zero, 4
; LA64-JT-NEXT: b .LBB0_6
; LA64-JT-NEXT: .LBB0_3: # %bb2
; LA64-JT-NEXT: ori $a3, $zero, 3
; LA64-JT-NEXT: b .LBB0_6
; LA64-JT-NEXT: .LBB0_4: # %bb3
; LA64-JT-NEXT: ori $a3, $zero, 2
; LA64-JT-NEXT: b .LBB0_6
; LA64-JT-NEXT: .LBB0_5: # %bb4
; LA64-JT-NEXT: ori $a3, $zero, 1
; LA64-JT-NEXT: .LBB0_6: # %exit
; LA64-JT-NEXT: st.w $a3, $a1, 0
; LA64-JT-NEXT: .LBB0_7: # %exit
; LA64-JT-NEXT: addi.d $a3, $a0, -5
; LA64-JT-NEXT: bltu $a2, $a3, .LBB0_9
; LA64-JT-NEXT: # %bb.8: # %exit
; LA64-JT-NEXT: slli.d $a3, $a3, 3
; LA64-JT-NEXT: pcalau12i $a4, %pc_hi20(.LJTI0_1)
; LA64-JT-NEXT: addi.d $a4, $a4, %pc_lo12(.LJTI0_1)
; LA64-JT-NEXT: ldx.d $a3, $a4, $a3
; LA64-JT-NEXT: .Ljrtb_1:
; LA64-JT-NEXT: jr $a3
; LA64-JT-NEXT: .LBB0_9: # %exit2
; LA64-JT-NEXT: ret
entry:
switch i32 %in, label %exit [
i32 1, label %bb1
i32 2, label %bb2
i32 3, label %bb3
i32 4, label %bb4
]
bb1:
store i32 4, ptr %out
br label %exit
bb2:
store i32 3, ptr %out
br label %exit
bb3:
store i32 2, ptr %out
br label %exit
bb4:
store i32 1, ptr %out
br label %exit
exit:
switch i32 %in, label %exit2 [
i32 5, label %bb1
i32 6, label %bb2
i32 7, label %bb3
i32 8, label %bb4
]
exit2:
ret void
}
; UTC_ARGS: --disable
; LA32-JT-LABEL: .LJTI0_0:
; LA32-JT: .section .discard.tablejump_annotate,"",@progbits
; LA32-JT-NEXT: .word .Ljrtb_0
; LA32-JT-NEXT: .word .LJTI0_0
; LA32-JT-NEXT: .word .Ljrtb_1
; LA32-JT-NEXT: .word .LJTI0_1
; UTC_ARGS: --disable
; LA64-JT-LABEL: .LJTI0_0:
; LA64-JT: .section .discard.tablejump_annotate,"",@progbits
; LA64-JT-NEXT: .dword .Ljrtb_0
; LA64-JT-NEXT: .dword .LJTI0_0
; LA64-JT-NEXT: .dword .Ljrtb_1
; LA64-JT-NEXT: .dword .LJTI0_1