Correct bug #164

ARM-software · Mar 6, 2024 · 0d49aa6 · 0d49aa6
1 parent a4e82f8
commit 0d49aa6
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/Source/FastMathFunctions/arm_sqrt_q15.c b/Source/FastMathFunctions/arm_sqrt_q15.c
@@ -115,7 +115,14 @@ arm_status arm_sqrt_q15(
   {
     *pOut = 0;
 
-    return (ARM_MATH_ARGUMENT_ERROR);
+    if (number==0)
+    {
+       return (ARM_MATH_SUCCESS);
+    }
+    else
+    {
+       return (ARM_MATH_ARGUMENT_ERROR);
+    }
   }
 }
 

diff --git a/Source/FastMathFunctions/arm_sqrt_q31.c b/Source/FastMathFunctions/arm_sqrt_q31.c
@@ -118,7 +118,14 @@ arm_status arm_sqrt_q31(
   {
     *pOut = 0;
 
-    return (ARM_MATH_ARGUMENT_ERROR);
+    if (number==0)
+    {
+       return (ARM_MATH_SUCCESS);
+    }
+    else
+    {
+       return (ARM_MATH_ARGUMENT_ERROR);
+    }
   }
 }