; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 6
; RUN: llc --mtriple=loongarch32 --mattr=+32s,+lsx < %s | FileCheck %s --check-prefixes=CHECK,LA32
; RUN: llc --mtriple=loongarch64 --mattr=+lsx < %s | FileCheck %s --check-prefixes=CHECK,LA64

define i8 @extractelt_add_v16i8(ptr %p) {
; CHECK-LABEL: extractelt_add_v16i8:
; CHECK:       # %bb.0: # %entry
; CHECK-NEXT:    vld $vr0, $a0, 0
; CHECK-NEXT:    vaddi.bu $vr0, $vr0, 13
; CHECK-NEXT:    vpickve2gr.b $a0, $vr0, 2
; CHECK-NEXT:    ret
entry:
  %x = load <16 x i8>, ptr %p
  %add = add <16 x i8> %x, <i8 11, i8 12, i8 13, i8 14, i8 11, i8 12, i8 13, i8 14, i8 11, i8 12, i8 13, i8 14, i8 11, i8 12, i8 13, i8 14>
  %ext = extractelement <16 x i8> %add, i32 2
  ret i8 %ext
}

define i16 @extractelt_add_v8i16(ptr %p) {
; CHECK-LABEL: extractelt_add_v8i16:
; CHECK:       # %bb.0: # %entry
; CHECK-NEXT:    vld $vr0, $a0, 0
; CHECK-NEXT:    vaddi.hu $vr0, $vr0, 13
; CHECK-NEXT:    vpickve2gr.h $a0, $vr0, 2
; CHECK-NEXT:    ret
entry:
  %x = load <8 x i16>, ptr %p
  %add = add <8 x i16> %x, <i16 11, i16 12, i16 13, i16 14, i16 11, i16 12, i16 13, i16 14>
  %ext = extractelement <8 x i16> %add, i32 2
  ret i16 %ext
}

define i32 @extractelt_add_v4i32(ptr %p) {
; LA32-LABEL: extractelt_add_v4i32:
; LA32:       # %bb.0: # %entry
; LA32-NEXT:    ld.w $a0, $a0, 8
; LA32-NEXT:    addi.w $a0, $a0, 13
; LA32-NEXT:    ret
;
; LA64-LABEL: extractelt_add_v4i32:
; LA64:       # %bb.0: # %entry
; LA64-NEXT:    vld $vr0, $a0, 0
; LA64-NEXT:    vaddi.wu $vr0, $vr0, 13
; LA64-NEXT:    vpickve2gr.w $a0, $vr0, 2
; LA64-NEXT:    ret
entry:
  %x = load <4 x i32>, ptr %p
  %add = add <4 x i32> %x, <i32 11, i32 12, i32 13, i32 14>
  %ext = extractelement <4 x i32> %add, i32 2
  ret i32 %ext
}

define i64 @extractelt_add_v2i64(ptr %p) {
; LA32-LABEL: extractelt_add_v2i64:
; LA32:       # %bb.0: # %entry
; LA32-NEXT:    vld $vr0, $a0, 0
; LA32-NEXT:    vaddi.du $vr0, $vr0, 12
; LA32-NEXT:    vpickve2gr.w $a0, $vr0, 2
; LA32-NEXT:    vpickve2gr.w $a1, $vr0, 3
; LA32-NEXT:    ret
;
; LA64-LABEL: extractelt_add_v2i64:
; LA64:       # %bb.0: # %entry
; LA64-NEXT:    ld.d $a0, $a0, 8
; LA64-NEXT:    addi.d $a0, $a0, 12
; LA64-NEXT:    ret
entry:
  %x = load <2 x i64>, ptr %p
  %add = add <2 x i64> %x, <i64 11, i64 12>
  %ext = extractelement <2 x i64> %add, i32 1
  ret i64 %ext
}

define float @extractelt_fadd_v4f32(ptr %p) {
; CHECK-LABEL: extractelt_fadd_v4f32:
; CHECK:       # %bb.0: # %entry
; CHECK-NEXT:    fld.s $fa0, $a0, 8
; CHECK-NEXT:    vldi $vr1, -1238
; CHECK-NEXT:    fadd.s $fa0, $fa0, $fa1
; CHECK-NEXT:    ret
entry:
  %x = load <4 x float>, ptr %p
  %add = fadd <4 x float> %x, <float 11.0, float 12.0, float 13.0, float 14.0>
  %ext = extractelement <4 x float> %add, i32 2
  ret float %ext
}

define double @extractelt_fadd_v2f64(ptr %p) {
; CHECK-LABEL: extractelt_fadd_v2f64:
; CHECK:       # %bb.0: # %entry
; CHECK-NEXT:    fld.d $fa0, $a0, 8
; CHECK-NEXT:    vldi $vr1, -984
; CHECK-NEXT:    fadd.d $fa0, $fa0, $fa1
; CHECK-NEXT:    ret
entry:
  %x = load <2 x double>, ptr %p
  %add = fadd <2 x double> %x, <double 11.0, double 12.0>
  %ext = extractelement <2 x double> %add, i32 1
  ret double %ext
}