[llvm] 1e0fe6a - [LoongArch] Add codegen support for fsqrt

via llvm-commits llvm-commits at lists.llvm.org
Fri Aug 19 19:45:19 PDT 2022


Author: gonglingqin
Date: 2022-08-20T10:31:24+08:00
New Revision: 1e0fe6a9475ab2cf0d09f3ddf5cc69761458e1e0

URL: https://github.com/llvm/llvm-project/commit/1e0fe6a9475ab2cf0d09f3ddf5cc69761458e1e0
DIFF: https://github.com/llvm/llvm-project/commit/1e0fe6a9475ab2cf0d09f3ddf5cc69761458e1e0.diff

LOG: [LoongArch] Add codegen support for fsqrt

Reviewed By: xen0n, SixWeining

Differential Revision: https://reviews.llvm.org/D132198

Added: 
    llvm/test/CodeGen/LoongArch/fsqrt.ll

Modified: 
    llvm/lib/Target/LoongArch/LoongArchFloat32InstrInfo.td
    llvm/lib/Target/LoongArch/LoongArchFloat64InstrInfo.td

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Target/LoongArch/LoongArchFloat32InstrInfo.td b/llvm/lib/Target/LoongArch/LoongArchFloat32InstrInfo.td
index f2b47d5818b4d..ab1b164503be1 100644
--- a/llvm/lib/Target/LoongArch/LoongArchFloat32InstrInfo.td
+++ b/llvm/lib/Target/LoongArch/LoongArchFloat32InstrInfo.td
@@ -144,6 +144,9 @@ def : PatFprFpr<fmul, FMUL_S, FPR32>;
 def : PatFprFpr<fdiv, FDIV_S, FPR32>;
 def : PatFpr<fneg, FNEG_S, FPR32>;
 def : PatFpr<fabs, FABS_S, FPR32>;
+def : PatFpr<fsqrt, FSQRT_S, FPR32>;
+
+def : Pat<(fdiv fpimm1, (fsqrt FPR32:$fj)), (FRSQRT_S FPR32:$fj)>;
 
 /// Setcc
 

diff  --git a/llvm/lib/Target/LoongArch/LoongArchFloat64InstrInfo.td b/llvm/lib/Target/LoongArch/LoongArchFloat64InstrInfo.td
index ad120eb2949f0..13ad87383c5d5 100644
--- a/llvm/lib/Target/LoongArch/LoongArchFloat64InstrInfo.td
+++ b/llvm/lib/Target/LoongArch/LoongArchFloat64InstrInfo.td
@@ -150,6 +150,9 @@ def : PatFprFpr<fmul, FMUL_D, FPR64>;
 def : PatFprFpr<fdiv, FDIV_D, FPR64>;
 def : PatFpr<fneg, FNEG_D, FPR64>;
 def : PatFpr<fabs, FABS_D, FPR64>;
+def : PatFpr<fsqrt, FSQRT_D, FPR64>;
+
+def : Pat<(fdiv fpimm1, (fsqrt FPR64:$fj)), (FRSQRT_D FPR64:$fj)>;
 
 /// Setcc
 

diff  --git a/llvm/test/CodeGen/LoongArch/fsqrt.ll b/llvm/test/CodeGen/LoongArch/fsqrt.ll
new file mode 100644
index 0000000000000..296c3c01bd8a1
--- /dev/null
+++ b/llvm/test/CodeGen/LoongArch/fsqrt.ll
@@ -0,0 +1,130 @@
+; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
+; RUN: llc --mtriple=loongarch32 --mattr=+f,-d < %s | FileCheck %s --check-prefix=LA32F
+; RUN: llc --mtriple=loongarch32 --mattr=+d < %s | FileCheck %s --check-prefix=LA32D
+; RUN: llc --mtriple=loongarch64 --mattr=+f,-d < %s | FileCheck %s --check-prefix=LA64F
+; RUN: llc --mtriple=loongarch64 --mattr=+d < %s | FileCheck %s --check-prefix=LA64D
+
+declare float @llvm.sqrt.f32(float)
+declare double @llvm.sqrt.f64(double)
+
+define float @fsqrt_f32(float %a) nounwind {
+; LA32F-LABEL: fsqrt_f32:
+; LA32F:       # %bb.0:
+; LA32F-NEXT:    fsqrt.s $fa0, $fa0
+; LA32F-NEXT:    ret
+;
+; LA32D-LABEL: fsqrt_f32:
+; LA32D:       # %bb.0:
+; LA32D-NEXT:    fsqrt.s $fa0, $fa0
+; LA32D-NEXT:    ret
+;
+; LA64F-LABEL: fsqrt_f32:
+; LA64F:       # %bb.0:
+; LA64F-NEXT:    fsqrt.s $fa0, $fa0
+; LA64F-NEXT:    ret
+;
+; LA64D-LABEL: fsqrt_f32:
+; LA64D:       # %bb.0:
+; LA64D-NEXT:    fsqrt.s $fa0, $fa0
+; LA64D-NEXT:    ret
+  %1 = call float @llvm.sqrt.f32(float %a)
+  ret float %1
+}
+
+define double @fsqrt_f64(double %a) nounwind {
+; LA32F-LABEL: fsqrt_f64:
+; LA32F:       # %bb.0:
+; LA32F-NEXT:    addi.w $sp, $sp, -16
+; LA32F-NEXT:    st.w $ra, $sp, 12 # 4-byte Folded Spill
+; LA32F-NEXT:    bl sqrt
+; LA32F-NEXT:    ld.w $ra, $sp, 12 # 4-byte Folded Reload
+; LA32F-NEXT:    addi.w $sp, $sp, 16
+; LA32F-NEXT:    ret
+;
+; LA32D-LABEL: fsqrt_f64:
+; LA32D:       # %bb.0:
+; LA32D-NEXT:    fsqrt.d $fa0, $fa0
+; LA32D-NEXT:    ret
+;
+; LA64F-LABEL: fsqrt_f64:
+; LA64F:       # %bb.0:
+; LA64F-NEXT:    addi.d $sp, $sp, -16
+; LA64F-NEXT:    st.d $ra, $sp, 8 # 8-byte Folded Spill
+; LA64F-NEXT:    bl sqrt
+; LA64F-NEXT:    ld.d $ra, $sp, 8 # 8-byte Folded Reload
+; LA64F-NEXT:    addi.d $sp, $sp, 16
+; LA64F-NEXT:    ret
+;
+; LA64D-LABEL: fsqrt_f64:
+; LA64D:       # %bb.0:
+; LA64D-NEXT:    fsqrt.d $fa0, $fa0
+; LA64D-NEXT:    ret
+  %1 = call double @llvm.sqrt.f64(double %a)
+  ret double %1
+}
+
+define float @frsqrt_f32(float %a) nounwind {
+; LA32F-LABEL: frsqrt_f32:
+; LA32F:       # %bb.0:
+; LA32F-NEXT:    frsqrt.s $fa0, $fa0
+; LA32F-NEXT:    ret
+;
+; LA32D-LABEL: frsqrt_f32:
+; LA32D:       # %bb.0:
+; LA32D-NEXT:    frsqrt.s $fa0, $fa0
+; LA32D-NEXT:    ret
+;
+; LA64F-LABEL: frsqrt_f32:
+; LA64F:       # %bb.0:
+; LA64F-NEXT:    frsqrt.s $fa0, $fa0
+; LA64F-NEXT:    ret
+;
+; LA64D-LABEL: frsqrt_f32:
+; LA64D:       # %bb.0:
+; LA64D-NEXT:    frsqrt.s $fa0, $fa0
+; LA64D-NEXT:    ret
+  %1 = call float @llvm.sqrt.f32(float %a)
+  %2 = fdiv float 1.0, %1
+  ret float %2
+}
+
+define double @frsqrt_f64(double %a) nounwind {
+; LA32F-LABEL: frsqrt_f64:
+; LA32F:       # %bb.0:
+; LA32F-NEXT:    addi.w $sp, $sp, -16
+; LA32F-NEXT:    st.w $ra, $sp, 12 # 4-byte Folded Spill
+; LA32F-NEXT:    bl sqrt
+; LA32F-NEXT:    move $a2, $a0
+; LA32F-NEXT:    move $a3, $a1
+; LA32F-NEXT:    lu12i.w $a1, 261888
+; LA32F-NEXT:    move $a0, $zero
+; LA32F-NEXT:    bl __divdf3
+; LA32F-NEXT:    ld.w $ra, $sp, 12 # 4-byte Folded Reload
+; LA32F-NEXT:    addi.w $sp, $sp, 16
+; LA32F-NEXT:    ret
+;
+; LA32D-LABEL: frsqrt_f64:
+; LA32D:       # %bb.0:
+; LA32D-NEXT:    frsqrt.d $fa0, $fa0
+; LA32D-NEXT:    ret
+;
+; LA64F-LABEL: frsqrt_f64:
+; LA64F:       # %bb.0:
+; LA64F-NEXT:    addi.d $sp, $sp, -16
+; LA64F-NEXT:    st.d $ra, $sp, 8 # 8-byte Folded Spill
+; LA64F-NEXT:    bl sqrt
+; LA64F-NEXT:    move $a1, $a0
+; LA64F-NEXT:    lu52i.d $a0, $zero, 1023
+; LA64F-NEXT:    bl __divdf3
+; LA64F-NEXT:    ld.d $ra, $sp, 8 # 8-byte Folded Reload
+; LA64F-NEXT:    addi.d $sp, $sp, 16
+; LA64F-NEXT:    ret
+;
+; LA64D-LABEL: frsqrt_f64:
+; LA64D:       # %bb.0:
+; LA64D-NEXT:    frsqrt.d $fa0, $fa0
+; LA64D-NEXT:    ret
+  %1 = call double @llvm.sqrt.f64(double %a)
+  %2 = fdiv double 1.0, %1
+  ret double %2
+}


        


More information about the llvm-commits mailing list