llvm/test/CodeGen/LoongArch/lasx/fsqrt.ll

; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
; 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

;; fsqrt
define void @sqrt_v8f32(ptr %res, ptr %a0) nounwind {
; LA32-LABEL: sqrt_v8f32:
; LA32:       # %bb.0: # %entry
; LA32-NEXT:    addi.w $sp, $sp, -128
; LA32-NEXT:    st.w $ra, $sp, 124 # 4-byte Folded Spill
; LA32-NEXT:    st.w $fp, $sp, 120 # 4-byte Folded Spill
; LA32-NEXT:    addi.w $fp, $sp, 128
; LA32-NEXT:    bstrins.w $sp, $zero, 4, 0
; LA32-NEXT:    vld $vr0, $a1, 16
; LA32-NEXT:    vst $vr0, $sp, 48
; LA32-NEXT:    ld.w $a2, $a1, 12
; LA32-NEXT:    st.w $a2, $sp, 44
; LA32-NEXT:    ld.w $a2, $a1, 8
; LA32-NEXT:    st.w $a2, $sp, 40
; LA32-NEXT:    ld.w $a2, $a1, 4
; LA32-NEXT:    st.w $a2, $sp, 36
; LA32-NEXT:    ld.w $a1, $a1, 0
; LA32-NEXT:    st.w $a1, $sp, 32
; LA32-NEXT:    xvld $xr0, $sp, 32
; LA32-NEXT:    xvfsqrt.s $xr0, $xr0
; LA32-NEXT:    xvst $xr0, $sp, 64
; LA32-NEXT:    vld $vr0, $sp, 80
; LA32-NEXT:    vst $vr0, $a0, 16
; LA32-NEXT:    ld.w $a1, $sp, 76
; LA32-NEXT:    st.w $a1, $a0, 12
; LA32-NEXT:    ld.w $a1, $sp, 72
; LA32-NEXT:    st.w $a1, $a0, 8
; LA32-NEXT:    ld.w $a1, $sp, 68
; LA32-NEXT:    st.w $a1, $a0, 4
; LA32-NEXT:    ld.w $a1, $sp, 64
; LA32-NEXT:    st.w $a1, $a0, 0
; LA32-NEXT:    addi.w $sp, $fp, -128
; LA32-NEXT:    ld.w $fp, $sp, 120 # 4-byte Folded Reload
; LA32-NEXT:    ld.w $ra, $sp, 124 # 4-byte Folded Reload
; LA32-NEXT:    addi.w $sp, $sp, 128
; LA32-NEXT:    ret
;
; LA64-LABEL: sqrt_v8f32:
; LA64:       # %bb.0: # %entry
; LA64-NEXT:    xvld $xr0, $a1, 0
; LA64-NEXT:    xvfsqrt.s $xr0, $xr0
; LA64-NEXT:    xvst $xr0, $a0, 0
; LA64-NEXT:    ret
entry:
  %v0 = load <8 x float>, ptr %a0, align 16
  %sqrt = call <8 x float> @llvm.sqrt.v8f32 (<8 x float> %v0)
  store <8 x float> %sqrt, ptr %res, align 16
  ret void
}

define void @sqrt_v4f64(ptr %res, ptr %a0) nounwind {
; LA32-LABEL: sqrt_v4f64:
; LA32:       # %bb.0: # %entry
; LA32-NEXT:    addi.w $sp, $sp, -128
; LA32-NEXT:    st.w $ra, $sp, 124 # 4-byte Folded Spill
; LA32-NEXT:    st.w $fp, $sp, 120 # 4-byte Folded Spill
; LA32-NEXT:    addi.w $fp, $sp, 128
; LA32-NEXT:    bstrins.w $sp, $zero, 4, 0
; LA32-NEXT:    vld $vr0, $a1, 16
; LA32-NEXT:    vst $vr0, $sp, 48
; LA32-NEXT:    ld.w $a2, $a1, 12
; LA32-NEXT:    st.w $a2, $sp, 44
; LA32-NEXT:    ld.w $a2, $a1, 8
; LA32-NEXT:    st.w $a2, $sp, 40
; LA32-NEXT:    ld.w $a2, $a1, 4
; LA32-NEXT:    st.w $a2, $sp, 36
; LA32-NEXT:    ld.w $a1, $a1, 0
; LA32-NEXT:    st.w $a1, $sp, 32
; LA32-NEXT:    xvld $xr0, $sp, 32
; LA32-NEXT:    xvfsqrt.d $xr0, $xr0
; LA32-NEXT:    xvst $xr0, $sp, 64
; LA32-NEXT:    vld $vr0, $sp, 80
; LA32-NEXT:    vst $vr0, $a0, 16
; LA32-NEXT:    ld.w $a1, $sp, 76
; LA32-NEXT:    st.w $a1, $a0, 12
; LA32-NEXT:    ld.w $a1, $sp, 72
; LA32-NEXT:    st.w $a1, $a0, 8
; LA32-NEXT:    ld.w $a1, $sp, 68
; LA32-NEXT:    st.w $a1, $a0, 4
; LA32-NEXT:    ld.w $a1, $sp, 64
; LA32-NEXT:    st.w $a1, $a0, 0
; LA32-NEXT:    addi.w $sp, $fp, -128
; LA32-NEXT:    ld.w $fp, $sp, 120 # 4-byte Folded Reload
; LA32-NEXT:    ld.w $ra, $sp, 124 # 4-byte Folded Reload
; LA32-NEXT:    addi.w $sp, $sp, 128
; LA32-NEXT:    ret
;
; LA64-LABEL: sqrt_v4f64:
; LA64:       # %bb.0: # %entry
; LA64-NEXT:    xvld $xr0, $a1, 0
; LA64-NEXT:    xvfsqrt.d $xr0, $xr0
; LA64-NEXT:    xvst $xr0, $a0, 0
; LA64-NEXT:    ret
entry:
  %v0 = load <4 x double>, ptr %a0, align 16
  %sqrt = call <4 x double> @llvm.sqrt.v4f64 (<4 x double> %v0)
  store <4 x double> %sqrt, ptr %res, align 16
  ret void
}

;; 1.0 / (fsqrt vec)
define void @one_div_sqrt_v8f32(ptr %res, ptr %a0) nounwind {
; LA32-LABEL: one_div_sqrt_v8f32:
; LA32:       # %bb.0: # %entry
; LA32-NEXT:    addi.w $sp, $sp, -128
; LA32-NEXT:    st.w $ra, $sp, 124 # 4-byte Folded Spill
; LA32-NEXT:    st.w $fp, $sp, 120 # 4-byte Folded Spill
; LA32-NEXT:    addi.w $fp, $sp, 128
; LA32-NEXT:    bstrins.w $sp, $zero, 4, 0
; LA32-NEXT:    vld $vr0, $a1, 16
; LA32-NEXT:    vst $vr0, $sp, 48
; LA32-NEXT:    ld.w $a2, $a1, 12
; LA32-NEXT:    st.w $a2, $sp, 44
; LA32-NEXT:    ld.w $a2, $a1, 8
; LA32-NEXT:    st.w $a2, $sp, 40
; LA32-NEXT:    ld.w $a2, $a1, 4
; LA32-NEXT:    st.w $a2, $sp, 36
; LA32-NEXT:    ld.w $a1, $a1, 0
; LA32-NEXT:    st.w $a1, $sp, 32
; LA32-NEXT:    xvld $xr0, $sp, 32
; LA32-NEXT:    xvfrsqrt.s $xr0, $xr0
; LA32-NEXT:    xvst $xr0, $sp, 64
; LA32-NEXT:    vld $vr0, $sp, 80
; LA32-NEXT:    vst $vr0, $a0, 16
; LA32-NEXT:    ld.w $a1, $sp, 76
; LA32-NEXT:    st.w $a1, $a0, 12
; LA32-NEXT:    ld.w $a1, $sp, 72
; LA32-NEXT:    st.w $a1, $a0, 8
; LA32-NEXT:    ld.w $a1, $sp, 68
; LA32-NEXT:    st.w $a1, $a0, 4
; LA32-NEXT:    ld.w $a1, $sp, 64
; LA32-NEXT:    st.w $a1, $a0, 0
; LA32-NEXT:    addi.w $sp, $fp, -128
; LA32-NEXT:    ld.w $fp, $sp, 120 # 4-byte Folded Reload
; LA32-NEXT:    ld.w $ra, $sp, 124 # 4-byte Folded Reload
; LA32-NEXT:    addi.w $sp, $sp, 128
; LA32-NEXT:    ret
;
; LA64-LABEL: one_div_sqrt_v8f32:
; LA64:       # %bb.0: # %entry
; LA64-NEXT:    xvld $xr0, $a1, 0
; LA64-NEXT:    xvfrsqrt.s $xr0, $xr0
; LA64-NEXT:    xvst $xr0, $a0, 0
; LA64-NEXT:    ret
entry:
  %v0 = load <8 x float>, ptr %a0, align 16
  %sqrt = call <8 x float> @llvm.sqrt.v8f32 (<8 x float> %v0)
  %div = fdiv <8 x float> <float 1.0, float 1.0, float 1.0, float 1.0, float 1.0, float 1.0, float 1.0, float 1.0>, %sqrt
  store <8 x float> %div, ptr %res, align 16
  ret void
}

define void @one_div_sqrt_v4f64(ptr %res, ptr %a0) nounwind {
; LA32-LABEL: one_div_sqrt_v4f64:
; LA32:       # %bb.0: # %entry
; LA32-NEXT:    addi.w $sp, $sp, -128
; LA32-NEXT:    st.w $ra, $sp, 124 # 4-byte Folded Spill
; LA32-NEXT:    st.w $fp, $sp, 120 # 4-byte Folded Spill
; LA32-NEXT:    addi.w $fp, $sp, 128
; LA32-NEXT:    bstrins.w $sp, $zero, 4, 0
; LA32-NEXT:    vld $vr0, $a1, 16
; LA32-NEXT:    vst $vr0, $sp, 48
; LA32-NEXT:    ld.w $a2, $a1, 12
; LA32-NEXT:    st.w $a2, $sp, 44
; LA32-NEXT:    ld.w $a2, $a1, 8
; LA32-NEXT:    st.w $a2, $sp, 40
; LA32-NEXT:    ld.w $a2, $a1, 4
; LA32-NEXT:    st.w $a2, $sp, 36
; LA32-NEXT:    ld.w $a1, $a1, 0
; LA32-NEXT:    st.w $a1, $sp, 32
; LA32-NEXT:    xvld $xr0, $sp, 32
; LA32-NEXT:    xvfsqrt.d $xr0, $xr0
; LA32-NEXT:    xvldi $xr1, -912
; LA32-NEXT:    xvfdiv.d $xr0, $xr1, $xr0
; LA32-NEXT:    xvst $xr0, $sp, 64
; LA32-NEXT:    vld $vr0, $sp, 80
; LA32-NEXT:    vst $vr0, $a0, 16
; LA32-NEXT:    ld.w $a1, $sp, 76
; LA32-NEXT:    st.w $a1, $a0, 12
; LA32-NEXT:    ld.w $a1, $sp, 72
; LA32-NEXT:    st.w $a1, $a0, 8
; LA32-NEXT:    ld.w $a1, $sp, 68
; LA32-NEXT:    st.w $a1, $a0, 4
; LA32-NEXT:    ld.w $a1, $sp, 64
; LA32-NEXT:    st.w $a1, $a0, 0
; LA32-NEXT:    addi.w $sp, $fp, -128
; LA32-NEXT:    ld.w $fp, $sp, 120 # 4-byte Folded Reload
; LA32-NEXT:    ld.w $ra, $sp, 124 # 4-byte Folded Reload
; LA32-NEXT:    addi.w $sp, $sp, 128
; LA32-NEXT:    ret
;
; LA64-LABEL: one_div_sqrt_v4f64:
; LA64:       # %bb.0: # %entry
; LA64-NEXT:    xvld $xr0, $a1, 0
; LA64-NEXT:    xvfrsqrt.d $xr0, $xr0
; LA64-NEXT:    xvst $xr0, $a0, 0
; LA64-NEXT:    ret
entry:
  %v0 = load <4 x double>, ptr %a0, align 16
  %sqrt = call <4 x double> @llvm.sqrt.v4f64 (<4 x double> %v0)
  %div = fdiv <4 x double> <double 1.0, double 1.0, double 1.0, double 1.0>, %sqrt
  store <4 x double> %div, ptr %res, align 16
  ret void
}

declare <8 x float> @llvm.sqrt.v8f32(<8 x float>)
declare <4 x double> @llvm.sqrt.v4f64(<4 x double>)