1/* 2 * Copyright (c) 2014,2015 Advanced Micro Devices, Inc. 3 * 4 * Permission is hereby granted, free of charge, to any person obtaining a copy 5 * of this software and associated documentation files (the "Software"), to deal 6 * in the Software without restriction, including without limitation the rights 7 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 8 * copies of the Software, and to permit persons to whom the Software is 9 * furnished to do so, subject to the following conditions: 10 * 11 * The above copyright notice and this permission notice shall be included in 12 * all copies or substantial portions of the Software. 13 * 14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 17 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 18 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 19 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 20 * THE SOFTWARE. 21 */ 22 23#include <clc/clc.h> 24#include <clc/clcmacro.h> 25#include <clc/math/math.h> 26 27_CLC_OVERLOAD _CLC_DEF float atanpi(float x) { 28 const float pi = 3.1415926535897932f; 29 30 uint ux = as_uint(x); 31 uint aux = ux & EXSIGNBIT_SP32; 32 uint sx = ux ^ aux; 33 34 float xbypi = MATH_DIVIDE(x, pi); 35 float shalf = as_float(sx | as_uint(0.5f)); 36 37 float v = as_float(aux); 38 39 // Return for NaN 40 float ret = x; 41 42 // 2^26 <= |x| <= Inf => atan(x) is close to piby2 43 ret = aux <= PINFBITPATT_SP32 ? shalf : ret; 44 45 // Reduce arguments 2^-19 <= |x| < 2^26 46 47 // 39/16 <= x < 2^26 48 x = -MATH_RECIP(v); 49 float c = 1.57079632679489655800f; // atan(infinity) 50 51 // 19/16 <= x < 39/16 52 int l = aux < 0x401c0000; 53 float xx = MATH_DIVIDE(v - 1.5f, mad(v, 1.5f, 1.0f)); 54 x = l ? xx : x; 55 c = l ? 9.82793723247329054082e-01f : c; // atan(1.5) 56 57 // 11/16 <= x < 19/16 58 l = aux < 0x3f980000U; 59 xx = MATH_DIVIDE(v - 1.0f, 1.0f + v); 60 x = l ? xx : x; 61 c = l ? 7.85398163397448278999e-01f : c; // atan(1) 62 63 // 7/16 <= x < 11/16 64 l = aux < 0x3f300000; 65 xx = MATH_DIVIDE(mad(v, 2.0f, -1.0f), 2.0f + v); 66 x = l ? xx : x; 67 c = l ? 4.63647609000806093515e-01f : c; // atan(0.5) 68 69 // 2^-19 <= x < 7/16 70 l = aux < 0x3ee00000; 71 x = l ? v : x; 72 c = l ? 0.0f : c; 73 74 // Core approximation: Remez(2,2) on [-7/16,7/16] 75 76 float s = x * x; 77 float a = mad(s, 78 mad(s, 0.470677934286149214138357545549e-2f, 0.192324546402108583211697690500f), 79 0.296528598819239217902158651186f); 80 81 float b = mad(s, 82 mad(s, 0.299309699959659728404442796915f, 0.111072499995399550138837673349e1f), 83 0.889585796862432286486651434570f); 84 85 float q = x * s * MATH_DIVIDE(a, b); 86 87 float z = c - (q - x); 88 z = MATH_DIVIDE(z, pi); 89 float zs = as_float(sx | as_uint(z)); 90 91 ret = aux < 0x4c800000 ? zs : ret; 92 93 // |x| < 2^-19 94 ret = aux < 0x36000000 ? xbypi : ret; 95 return ret; 96} 97 98_CLC_UNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, float, atanpi, float) 99 100#ifdef cl_khr_fp64 101#pragma OPENCL EXTENSION cl_khr_fp64 : enable 102 103_CLC_OVERLOAD _CLC_DEF double atanpi(double x) { 104 const double pi = 0x1.921fb54442d18p+1; 105 106 double v = fabs(x); 107 108 // 2^56 > v > 39/16 109 double a = -1.0; 110 double b = v; 111 // (chi + clo) = arctan(infinity) 112 double chi = 1.57079632679489655800e+00; 113 double clo = 6.12323399573676480327e-17; 114 115 double ta = v - 1.5; 116 double tb = 1.0 + 1.5 * v; 117 int l = v <= 0x1.38p+1; // 39/16 > v > 19/16 118 a = l ? ta : a; 119 b = l ? tb : b; 120 // (chi + clo) = arctan(1.5) 121 chi = l ? 9.82793723247329054082e-01 : chi; 122 clo = l ? 1.39033110312309953701e-17 : clo; 123 124 ta = v - 1.0; 125 tb = 1.0 + v; 126 l = v <= 0x1.3p+0; // 19/16 > v > 11/16 127 a = l ? ta : a; 128 b = l ? tb : b; 129 // (chi + clo) = arctan(1.) 130 chi = l ? 7.85398163397448278999e-01 : chi; 131 clo = l ? 3.06161699786838240164e-17 : clo; 132 133 ta = 2.0 * v - 1.0; 134 tb = 2.0 + v; 135 l = v <= 0x1.6p-1; // 11/16 > v > 7/16 136 a = l ? ta : a; 137 b = l ? tb : b; 138 // (chi + clo) = arctan(0.5) 139 chi = l ? 4.63647609000806093515e-01 : chi; 140 clo = l ? 2.26987774529616809294e-17 : clo; 141 142 l = v <= 0x1.cp-2; // v < 7/16 143 a = l ? v : a; 144 b = l ? 1.0 : b;; 145 chi = l ? 0.0 : chi; 146 clo = l ? 0.0 : clo; 147 148 // Core approximation: Remez(4,4) on [-7/16,7/16] 149 double r = a / b; 150 double s = r * r; 151 double qn = fma(s, 152 fma(s, 153 fma(s, 154 fma(s, 0.142316903342317766e-3, 155 0.304455919504853031e-1), 156 0.220638780716667420e0), 157 0.447677206805497472e0), 158 0.268297920532545909e0); 159 160 double qd = fma(s, 161 fma(s, 162 fma(s, 163 fma(s, 0.389525873944742195e-1, 164 0.424602594203847109e0), 165 0.141254259931958921e1), 166 0.182596787737507063e1), 167 0.804893761597637733e0); 168 169 double q = r * s * qn / qd; 170 r = (chi - ((q - clo) - r)) / pi; 171 double vp = v / pi; 172 173 double z = isnan(x) ? x : 0.5; 174 z = v <= 0x1.0p+56 ? r : z; 175 z = v < 0x1.0p-26 ? vp : z; 176 return x == v ? z : -z; 177} 178 179_CLC_UNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, double, atanpi, double) 180 181#endif 182 183#ifdef cl_khr_fp16 184 185#pragma OPENCL EXTENSION cl_khr_fp16 : enable 186 187_CLC_DEFINE_UNARY_BUILTIN_FP16(atanpi) 188 189#endif 190