[LoongArch] Pre-commit tests for optimizing insert extracted fp elements

Author: zhaoqi5

llvm/test/CodeGen/LoongArch/lasx/ir-instruction/insert-extract-element.ll

@@ -0,0 +1,30 @@
+; 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
+
+define <8 x float> @insert_extract_v8f32(<8 x float> %a) nounwind {
+; CHECK-LABEL: insert_extract_v8f32:
+; CHECK:       # %bb.0: # %entry
+; CHECK-NEXT:    xvpickve2gr.w $a0, $xr0, 7
+; CHECK-NEXT:    movgr2fr.w $fa1, $a0
+; CHECK-NEXT:    movfr2gr.s $a0, $fa1
+; CHECK-NEXT:    xvinsgr2vr.w $xr0, $a0, 1
+; CHECK-NEXT:    ret
+entry:
+  %b = extractelement <8 x float> %a, i32 7
+  %c = insertelement <8 x float> %a, float %b, i32 1
+  ret <8 x float> %c
+}
+
+define <4 x double> @insert_extract_v4f64(<4 x double> %a) nounwind {
+; CHECK-LABEL: insert_extract_v4f64:
+; CHECK:       # %bb.0: # %entry
+; CHECK-NEXT:    xvpickve2gr.d $a0, $xr0, 3
+; CHECK-NEXT:    movgr2fr.d $fa1, $a0
+; CHECK-NEXT:    movfr2gr.d $a0, $fa1
+; CHECK-NEXT:    xvinsgr2vr.d $xr0, $a0, 1
+; CHECK-NEXT:    ret
+entry:
+  %b = extractelement <4 x double> %a, i32 3
+  %c = insertelement <4 x double> %a, double %b, i32 1
+  ret <4 x double> %c
+}

llvm/test/CodeGen/LoongArch/lsx/ir-instruction/insert-extract-element.ll

@@ -0,0 +1,28 @@
+; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 5
+; RUN: llc --mtriple=loongarch64 --mattr=+lsx < %s | FileCheck %s
+
+define <4 x float> @insert_extract_v4f32(<4 x float> %a) nounwind {
+; CHECK-LABEL: insert_extract_v4f32:
+; CHECK:       # %bb.0: # %entry
+; CHECK-NEXT:    vreplvei.w $vr1, $vr0, 3
+; CHECK-NEXT:    movfr2gr.s $a0, $fa1
+; CHECK-NEXT:    vinsgr2vr.w $vr0, $a0, 0
+; CHECK-NEXT:    ret
+entry:
+  %b = extractelement <4 x float> %a, i32 3
+  %c = insertelement <4 x float> %a, float %b, i32 0
+  ret <4 x float> %c
+}
+
+define <2 x double> @insert_extract_v2f64(<2 x double> %a) nounwind {
+; CHECK-LABEL: insert_extract_v2f64:
+; CHECK:       # %bb.0: # %entry
+; CHECK-NEXT:    vreplvei.d $vr1, $vr0, 1
+; CHECK-NEXT:    movfr2gr.d $a0, $fa1
+; CHECK-NEXT:    vinsgr2vr.d $vr0, $a0, 0
+; CHECK-NEXT:    ret
+entry:
+  %b = extractelement <2 x double> %a, i32 1
+  %c = insertelement <2 x double> %a, double %b, i32 0
+  ret <2 x double> %c
+}