DSP: Optimize sqrt using sqrt intrinsic function

Signed-off-by: Huaqi Fang <578567190@qq.com>
Nuclei-Software · Jul 31, 2020 · c819040 · c819040
1 parent 9ba880a
commit c819040
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/NMSIS/DSP/Include/riscv_math.h b/NMSIS/DSP/Include/riscv_math.h
@@ -5981,10 +5981,14 @@ __STATIC_FORCEINLINE riscv_status riscv_sqrt_f32(
   float32_t in,
   float32_t * pOut)
   {
+
     if (in >= 0.0f)
     {
-
+#if defined ( __riscv_flen )
+      __ASM volatile("fsqrt.s %0, %1" : "=f"(*pOut) : "f"(in));
+#else
       *pOut = sqrtf(in);
+#endif /*__riscv_flen*/
 
       return (RISCV_MATH_SUCCESS);
     }