1 files changed, 47 insertions, 18 deletions
diff --git a/llvm/test/CodeGen/LoongArch/lasx/issue107355.ll b/llvm/test/CodeGen/LoongArch/lasx/issue107355.ll
index 506b5c1232f2..249414f5b1bb 100644
--- a/llvm/test/CodeGen/LoongArch/lasx/issue107355.ll
+++ b/llvm/test/CodeGen/LoongArch/lasx/issue107355.ll
@@ -1,5 +1,6 @@
 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 5
-; RUN: llc --mtriple=loongarch64 --mattr=+lasx < %s | FileCheck %s
+; RUN: llc --mtriple=loongarch32 --mattr=+32s,+lasx < %s | FileCheck %s --check-prefix=LA32
+; RUN: llc --mtriple=loongarch64 --mattr=+lasx < %s | FileCheck %s --check-prefix=LA64
 
 ;; Without this patch(codegen for concat_vectors), the test will hang.
 @g_156 = external global [12 x i32]
@@ -7,23 +8,51 @@
 @g_813 = external global i32
 
 define void @foo() {
-; CHECK-LABEL: foo:
-; CHECK:       # %bb.0: # %entry
-; CHECK-NEXT:    pcalau12i $a0, %got_pc_hi20(g_156)
-; CHECK-NEXT:    ld.d $a0, $a0, %got_pc_lo12(g_156)
-; CHECK-NEXT:    pcalau12i $a1, %got_pc_hi20(g_490)
-; CHECK-NEXT:    ld.d $a1, $a1, %got_pc_lo12(g_490)
-; CHECK-NEXT:    ld.w $a2, $a0, 24
-; CHECK-NEXT:    pcalau12i $a3, %got_pc_hi20(g_813)
-; CHECK-NEXT:    ld.d $a3, $a3, %got_pc_lo12(g_813)
-; CHECK-NEXT:    st.w $zero, $a1, 0
-; CHECK-NEXT:    st.w $a2, $a3, 0
-; CHECK-NEXT:    xvrepli.b $xr0, 0
-; CHECK-NEXT:    xvst $xr0, $a0, 0
-; CHECK-NEXT:    vrepli.b $vr0, 0
-; CHECK-NEXT:    vst $vr0, $a0, 32
-; CHECK-NEXT:    st.w $zero, $a0, 20
-; CHECK-NEXT:    ret
+; LA32-LABEL: foo:
+; LA32:       # %bb.0: # %entry
+; LA32-NEXT:    addi.w $sp, $sp, -16
+; LA32-NEXT:    .cfi_def_cfa_offset 16
+; LA32-NEXT:    st.w $ra, $sp, 12 # 4-byte Folded Spill
+; LA32-NEXT:    st.w $fp, $sp, 8 # 4-byte Folded Spill
+; LA32-NEXT:    .cfi_offset 1, -4
+; LA32-NEXT:    .cfi_offset 22, -8
+; LA32-NEXT:    pcalau12i $a0, %got_pc_hi20(g_156)
+; LA32-NEXT:    ld.w $fp, $a0, %got_pc_lo12(g_156)
+; LA32-NEXT:    pcalau12i $a0, %got_pc_hi20(g_490)
+; LA32-NEXT:    ld.w $a0, $a0, %got_pc_lo12(g_490)
+; LA32-NEXT:    ld.w $a1, $fp, 24
+; LA32-NEXT:    pcalau12i $a2, %got_pc_hi20(g_813)
+; LA32-NEXT:    ld.w $a2, $a2, %got_pc_lo12(g_813)
+; LA32-NEXT:    st.w $zero, $fp, 20
+; LA32-NEXT:    st.w $zero, $a0, 0
+; LA32-NEXT:    st.w $a1, $a2, 0
+; LA32-NEXT:    ori $a2, $zero, 48
+; LA32-NEXT:    move $a0, $fp
+; LA32-NEXT:    move $a1, $zero
+; LA32-NEXT:    bl memset
+; LA32-NEXT:    st.w $zero, $fp, 20
+; LA32-NEXT:    ld.w $fp, $sp, 8 # 4-byte Folded Reload
+; LA32-NEXT:    ld.w $ra, $sp, 12 # 4-byte Folded Reload
+; LA32-NEXT:    addi.w $sp, $sp, 16
+; LA32-NEXT:    ret
+;
+; LA64-LABEL: foo:
+; LA64:       # %bb.0: # %entry
+; LA64-NEXT:    pcalau12i $a0, %got_pc_hi20(g_156)
+; LA64-NEXT:    ld.d $a0, $a0, %got_pc_lo12(g_156)
+; LA64-NEXT:    pcalau12i $a1, %got_pc_hi20(g_490)
+; LA64-NEXT:    ld.d $a1, $a1, %got_pc_lo12(g_490)
+; LA64-NEXT:    ld.w $a2, $a0, 24
+; LA64-NEXT:    pcalau12i $a3, %got_pc_hi20(g_813)
+; LA64-NEXT:    ld.d $a3, $a3, %got_pc_lo12(g_813)
+; LA64-NEXT:    st.w $zero, $a1, 0
+; LA64-NEXT:    st.w $a2, $a3, 0
+; LA64-NEXT:    xvrepli.b $xr0, 0
+; LA64-NEXT:    xvst $xr0, $a0, 0
+; LA64-NEXT:    vrepli.b $vr0, 0
+; LA64-NEXT:    vst $vr0, $a0, 32
+; LA64-NEXT:    st.w $zero, $a0, 20
+; LA64-NEXT:    ret
 entry:
   store i32 0, ptr getelementptr inbounds (i8, ptr @g_156, i64 20), align 4
   store i32 0, ptr @g_490, align 4