[Beignet] [PATCH 1/2] backend: fix tgamma error after restructure

[rander.wang]

Signed-off-by: rander.wang <rander.w...@intel.com>
---
 backend/src/libocl/tmpl/ocl_math_common.tmpl.cl | 56 ++++++++++++++-----------
 1 file changed, 31 insertions(+), 25 deletions(-)

diff --git a/backend/src/libocl/tmpl/ocl_math_common.tmpl.cl 
b/backend/src/libocl/tmpl/ocl_math_common.tmpl.cl
index 9d4100e..6b942db 100644
--- a/backend/src/libocl/tmpl/ocl_math_common.tmpl.cl
+++ b/backend/src/libocl/tmpl/ocl_math_common.tmpl.cl
@@ -2550,30 +2550,6 @@ OVERLOADABLE float __gen_ocl_internal_pow(float x, float 
y) {
   return bRet ? retVal:sn*z;
 }
 
-#define BODY \
-  if (isnan(x_abs) || isinf(x_abs)) { \
-    x_log2 = 0; \
-    return x_abs; \
-  } \
-  uint u = as_uint(x_abs); \
-  uint a = u & 0x7FFFFFFFu; \
-  if (a == 0) { \
-    x_log2 = 0; \
-    return x_abs; \
-  } \
-  if (a >= 0x800000) { \
-    x_log2 = (a >> 23) - 126; \
-    return as_float((u & (0x807FFFFFu)) | 0x3F000000); \
-  } \
-  int e = -126; \
-  while (a < 0x400000) { \
-    e --; \
-    a <<= 1; \
-  } \
-  a <<= 1; \
-  x_log2 = e; \
-  float x_mant = as_float((a & (0x807FFFFFu)) | (u & 0x80000000u) | 
0x3F000000);
-
 OVERLOADABLE float tgamma (float x)
 {
   /* based on glibc __ieee754_gammaf_r by Ulrich Drepper <drep...@cygnus.com> 
*/
@@ -2641,8 +2617,38 @@ OVERLOADABLE float tgamma (float x)
         float x_int = __gen_ocl_internal_round (x_abs);
         float x_frac = x_abs - x_int;
         int x_log2;
+        float x_mant;
+        bool skip = false;
+        if (isnan(x_abs) || isinf(x_abs)) {
+         x_log2 = 0;
+         x_mant = x_abs;
+         skip = true;
+        }
 
-        BODY
+        uint u = as_uint(x_abs);
+        uint a = u & 0x7FFFFFFFu;
+        if (a == 0) {
+         x_log2 = 0;
+         x_mant = x_abs;
+         skip = true;
+        }
+        if (a >= 0x800000 && !skip) {
+         x_log2 = (a >> 23) - 126;
+         x_mant = as_float((u & (0x807FFFFFu)) | 0x3F000000);
+         skip = true;
+        }
+
+        int e = -126;
+        if(!skip)
+        {
+               while (a < 0x400000) {
+                 e --;
+                 a <<= 1;
+               }
+               a <<= 1;
+               x_log2 = e;
+               x_mant = as_float((a & (0x807FFFFFu)) | (u & 0x80000000u) | 
0x3F000000);
+        }
 
         if (x_mant < M_SQRT1_2_F)
           {
-- 
2.7.4

_______________________________________________
Beignet mailing list
Beignet@lists.freedesktop.org
https://lists.freedesktop.org/mailman/listinfo/beignet