Sasha Smundak | 0fc590b | 2020-10-07 08:11:59 -0700 | [diff] [blame] | 1 | /*===---- __clang_cuda_math.h - Device-side CUDA math support --------------=== |
| 2 | * |
| 3 | * Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| 4 | * See https://llvm.org/LICENSE.txt for license information. |
| 5 | * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| 6 | * |
| 7 | *===-----------------------------------------------------------------------=== |
| 8 | */ |
| 9 | #ifndef __CLANG_CUDA_MATH_H__ |
| 10 | #define __CLANG_CUDA_MATH_H__ |
| 11 | #ifndef __CUDA__ |
| 12 | #error "This file is for CUDA compilation only." |
| 13 | #endif |
| 14 | |
| 15 | #ifndef __OPENMP_NVPTX__ |
| 16 | #if CUDA_VERSION < 9000 |
| 17 | #error This file is intended to be used with CUDA-9+ only. |
| 18 | #endif |
| 19 | #endif |
| 20 | |
| 21 | // __DEVICE__ is a helper macro with common set of attributes for the wrappers |
| 22 | // we implement in this file. We need static in order to avoid emitting unused |
| 23 | // functions and __forceinline__ helps inlining these wrappers at -O1. |
| 24 | #pragma push_macro("__DEVICE__") |
| 25 | #ifdef __OPENMP_NVPTX__ |
| 26 | #if defined(__cplusplus) |
| 27 | #define __DEVICE__ static constexpr __attribute__((always_inline, nothrow)) |
| 28 | #else |
| 29 | #define __DEVICE__ static __attribute__((always_inline, nothrow)) |
| 30 | #endif |
| 31 | #else |
| 32 | #define __DEVICE__ static __device__ __forceinline__ |
| 33 | #endif |
| 34 | |
| 35 | // Specialized version of __DEVICE__ for functions with void return type. Needed |
| 36 | // because the OpenMP overlay requires constexpr functions here but prior to |
| 37 | // c++14 void return functions could not be constexpr. |
| 38 | #pragma push_macro("__DEVICE_VOID__") |
| 39 | #ifdef __OPENMP_NVPTX__ && defined(__cplusplus) && __cplusplus < 201402L |
| 40 | #define __DEVICE_VOID__ static __attribute__((always_inline, nothrow)) |
| 41 | #else |
| 42 | #define __DEVICE_VOID__ __DEVICE__ |
| 43 | #endif |
| 44 | |
| 45 | // libdevice provides fast low precision and slow full-recision implementations |
| 46 | // for some functions. Which one gets selected depends on |
| 47 | // __CLANG_CUDA_APPROX_TRANSCENDENTALS__ which gets defined by clang if |
| 48 | // -ffast-math or -fcuda-approx-transcendentals are in effect. |
| 49 | #pragma push_macro("__FAST_OR_SLOW") |
| 50 | #if defined(__CLANG_CUDA_APPROX_TRANSCENDENTALS__) |
| 51 | #define __FAST_OR_SLOW(fast, slow) fast |
| 52 | #else |
| 53 | #define __FAST_OR_SLOW(fast, slow) slow |
| 54 | #endif |
| 55 | |
| 56 | __DEVICE__ int abs(int __a) { return __nv_abs(__a); } |
| 57 | __DEVICE__ double fabs(double __a) { return __nv_fabs(__a); } |
| 58 | __DEVICE__ double acos(double __a) { return __nv_acos(__a); } |
| 59 | __DEVICE__ float acosf(float __a) { return __nv_acosf(__a); } |
| 60 | __DEVICE__ double acosh(double __a) { return __nv_acosh(__a); } |
| 61 | __DEVICE__ float acoshf(float __a) { return __nv_acoshf(__a); } |
| 62 | __DEVICE__ double asin(double __a) { return __nv_asin(__a); } |
| 63 | __DEVICE__ float asinf(float __a) { return __nv_asinf(__a); } |
| 64 | __DEVICE__ double asinh(double __a) { return __nv_asinh(__a); } |
| 65 | __DEVICE__ float asinhf(float __a) { return __nv_asinhf(__a); } |
| 66 | __DEVICE__ double atan(double __a) { return __nv_atan(__a); } |
| 67 | __DEVICE__ double atan2(double __a, double __b) { return __nv_atan2(__a, __b); } |
| 68 | __DEVICE__ float atan2f(float __a, float __b) { return __nv_atan2f(__a, __b); } |
| 69 | __DEVICE__ float atanf(float __a) { return __nv_atanf(__a); } |
| 70 | __DEVICE__ double atanh(double __a) { return __nv_atanh(__a); } |
| 71 | __DEVICE__ float atanhf(float __a) { return __nv_atanhf(__a); } |
| 72 | __DEVICE__ double cbrt(double __a) { return __nv_cbrt(__a); } |
| 73 | __DEVICE__ float cbrtf(float __a) { return __nv_cbrtf(__a); } |
| 74 | __DEVICE__ double ceil(double __a) { return __nv_ceil(__a); } |
| 75 | __DEVICE__ float ceilf(float __a) { return __nv_ceilf(__a); } |
| 76 | __DEVICE__ double copysign(double __a, double __b) { |
| 77 | return __nv_copysign(__a, __b); |
| 78 | } |
| 79 | __DEVICE__ float copysignf(float __a, float __b) { |
| 80 | return __nv_copysignf(__a, __b); |
| 81 | } |
| 82 | __DEVICE__ double cos(double __a) { return __nv_cos(__a); } |
| 83 | __DEVICE__ float cosf(float __a) { |
| 84 | return __FAST_OR_SLOW(__nv_fast_cosf, __nv_cosf)(__a); |
| 85 | } |
| 86 | __DEVICE__ double cosh(double __a) { return __nv_cosh(__a); } |
| 87 | __DEVICE__ float coshf(float __a) { return __nv_coshf(__a); } |
| 88 | __DEVICE__ double cospi(double __a) { return __nv_cospi(__a); } |
| 89 | __DEVICE__ float cospif(float __a) { return __nv_cospif(__a); } |
| 90 | __DEVICE__ double cyl_bessel_i0(double __a) { return __nv_cyl_bessel_i0(__a); } |
| 91 | __DEVICE__ float cyl_bessel_i0f(float __a) { return __nv_cyl_bessel_i0f(__a); } |
| 92 | __DEVICE__ double cyl_bessel_i1(double __a) { return __nv_cyl_bessel_i1(__a); } |
| 93 | __DEVICE__ float cyl_bessel_i1f(float __a) { return __nv_cyl_bessel_i1f(__a); } |
| 94 | __DEVICE__ double erf(double __a) { return __nv_erf(__a); } |
| 95 | __DEVICE__ double erfc(double __a) { return __nv_erfc(__a); } |
| 96 | __DEVICE__ float erfcf(float __a) { return __nv_erfcf(__a); } |
| 97 | __DEVICE__ double erfcinv(double __a) { return __nv_erfcinv(__a); } |
| 98 | __DEVICE__ float erfcinvf(float __a) { return __nv_erfcinvf(__a); } |
| 99 | __DEVICE__ double erfcx(double __a) { return __nv_erfcx(__a); } |
| 100 | __DEVICE__ float erfcxf(float __a) { return __nv_erfcxf(__a); } |
| 101 | __DEVICE__ float erff(float __a) { return __nv_erff(__a); } |
| 102 | __DEVICE__ double erfinv(double __a) { return __nv_erfinv(__a); } |
| 103 | __DEVICE__ float erfinvf(float __a) { return __nv_erfinvf(__a); } |
| 104 | __DEVICE__ double exp(double __a) { return __nv_exp(__a); } |
| 105 | __DEVICE__ double exp10(double __a) { return __nv_exp10(__a); } |
| 106 | __DEVICE__ float exp10f(float __a) { return __nv_exp10f(__a); } |
| 107 | __DEVICE__ double exp2(double __a) { return __nv_exp2(__a); } |
| 108 | __DEVICE__ float exp2f(float __a) { return __nv_exp2f(__a); } |
| 109 | __DEVICE__ float expf(float __a) { return __nv_expf(__a); } |
| 110 | __DEVICE__ double expm1(double __a) { return __nv_expm1(__a); } |
| 111 | __DEVICE__ float expm1f(float __a) { return __nv_expm1f(__a); } |
| 112 | __DEVICE__ float fabsf(float __a) { return __nv_fabsf(__a); } |
| 113 | __DEVICE__ double fdim(double __a, double __b) { return __nv_fdim(__a, __b); } |
| 114 | __DEVICE__ float fdimf(float __a, float __b) { return __nv_fdimf(__a, __b); } |
| 115 | __DEVICE__ double fdivide(double __a, double __b) { return __a / __b; } |
| 116 | __DEVICE__ float fdividef(float __a, float __b) { |
| 117 | #if __FAST_MATH__ && !__CUDA_PREC_DIV |
| 118 | return __nv_fast_fdividef(__a, __b); |
| 119 | #else |
| 120 | return __a / __b; |
| 121 | #endif |
| 122 | } |
| 123 | __DEVICE__ double floor(double __f) { return __nv_floor(__f); } |
| 124 | __DEVICE__ float floorf(float __f) { return __nv_floorf(__f); } |
| 125 | __DEVICE__ double fma(double __a, double __b, double __c) { |
| 126 | return __nv_fma(__a, __b, __c); |
| 127 | } |
| 128 | __DEVICE__ float fmaf(float __a, float __b, float __c) { |
| 129 | return __nv_fmaf(__a, __b, __c); |
| 130 | } |
| 131 | __DEVICE__ double fmax(double __a, double __b) { return __nv_fmax(__a, __b); } |
| 132 | __DEVICE__ float fmaxf(float __a, float __b) { return __nv_fmaxf(__a, __b); } |
| 133 | __DEVICE__ double fmin(double __a, double __b) { return __nv_fmin(__a, __b); } |
| 134 | __DEVICE__ float fminf(float __a, float __b) { return __nv_fminf(__a, __b); } |
| 135 | __DEVICE__ double fmod(double __a, double __b) { return __nv_fmod(__a, __b); } |
| 136 | __DEVICE__ float fmodf(float __a, float __b) { return __nv_fmodf(__a, __b); } |
| 137 | __DEVICE__ double frexp(double __a, int *__b) { return __nv_frexp(__a, __b); } |
| 138 | __DEVICE__ float frexpf(float __a, int *__b) { return __nv_frexpf(__a, __b); } |
| 139 | __DEVICE__ double hypot(double __a, double __b) { return __nv_hypot(__a, __b); } |
| 140 | __DEVICE__ float hypotf(float __a, float __b) { return __nv_hypotf(__a, __b); } |
| 141 | __DEVICE__ int ilogb(double __a) { return __nv_ilogb(__a); } |
| 142 | __DEVICE__ int ilogbf(float __a) { return __nv_ilogbf(__a); } |
| 143 | __DEVICE__ double j0(double __a) { return __nv_j0(__a); } |
| 144 | __DEVICE__ float j0f(float __a) { return __nv_j0f(__a); } |
| 145 | __DEVICE__ double j1(double __a) { return __nv_j1(__a); } |
| 146 | __DEVICE__ float j1f(float __a) { return __nv_j1f(__a); } |
| 147 | __DEVICE__ double jn(int __n, double __a) { return __nv_jn(__n, __a); } |
| 148 | __DEVICE__ float jnf(int __n, float __a) { return __nv_jnf(__n, __a); } |
| 149 | #if defined(__LP64__) || defined(_WIN64) |
| 150 | __DEVICE__ long labs(long __a) { return __nv_llabs(__a); }; |
| 151 | #else |
| 152 | __DEVICE__ long labs(long __a) { return __nv_abs(__a); }; |
| 153 | #endif |
| 154 | __DEVICE__ double ldexp(double __a, int __b) { return __nv_ldexp(__a, __b); } |
| 155 | __DEVICE__ float ldexpf(float __a, int __b) { return __nv_ldexpf(__a, __b); } |
| 156 | __DEVICE__ double lgamma(double __a) { return __nv_lgamma(__a); } |
| 157 | __DEVICE__ float lgammaf(float __a) { return __nv_lgammaf(__a); } |
| 158 | __DEVICE__ long long llabs(long long __a) { return __nv_llabs(__a); } |
| 159 | __DEVICE__ long long llmax(long long __a, long long __b) { |
| 160 | return __nv_llmax(__a, __b); |
| 161 | } |
| 162 | __DEVICE__ long long llmin(long long __a, long long __b) { |
| 163 | return __nv_llmin(__a, __b); |
| 164 | } |
| 165 | __DEVICE__ long long llrint(double __a) { return __nv_llrint(__a); } |
| 166 | __DEVICE__ long long llrintf(float __a) { return __nv_llrintf(__a); } |
| 167 | __DEVICE__ long long llround(double __a) { return __nv_llround(__a); } |
| 168 | __DEVICE__ long long llroundf(float __a) { return __nv_llroundf(__a); } |
Pirama Arumuga Nainar | 494f645 | 2021-12-02 10:42:14 -0800 | [diff] [blame] | 169 | __DEVICE__ double round(double __a) { return __nv_round(__a); } |
| 170 | __DEVICE__ float roundf(float __a) { return __nv_roundf(__a); } |
Sasha Smundak | 0fc590b | 2020-10-07 08:11:59 -0700 | [diff] [blame] | 171 | __DEVICE__ double log(double __a) { return __nv_log(__a); } |
| 172 | __DEVICE__ double log10(double __a) { return __nv_log10(__a); } |
| 173 | __DEVICE__ float log10f(float __a) { return __nv_log10f(__a); } |
| 174 | __DEVICE__ double log1p(double __a) { return __nv_log1p(__a); } |
| 175 | __DEVICE__ float log1pf(float __a) { return __nv_log1pf(__a); } |
| 176 | __DEVICE__ double log2(double __a) { return __nv_log2(__a); } |
| 177 | __DEVICE__ float log2f(float __a) { |
| 178 | return __FAST_OR_SLOW(__nv_fast_log2f, __nv_log2f)(__a); |
| 179 | } |
| 180 | __DEVICE__ double logb(double __a) { return __nv_logb(__a); } |
| 181 | __DEVICE__ float logbf(float __a) { return __nv_logbf(__a); } |
| 182 | __DEVICE__ float logf(float __a) { |
| 183 | return __FAST_OR_SLOW(__nv_fast_logf, __nv_logf)(__a); |
| 184 | } |
| 185 | #if defined(__LP64__) || defined(_WIN64) |
| 186 | __DEVICE__ long lrint(double __a) { return llrint(__a); } |
| 187 | __DEVICE__ long lrintf(float __a) { return __float2ll_rn(__a); } |
| 188 | __DEVICE__ long lround(double __a) { return llround(__a); } |
| 189 | __DEVICE__ long lroundf(float __a) { return llroundf(__a); } |
| 190 | #else |
| 191 | __DEVICE__ long lrint(double __a) { return (long)rint(__a); } |
| 192 | __DEVICE__ long lrintf(float __a) { return __float2int_rn(__a); } |
| 193 | __DEVICE__ long lround(double __a) { return round(__a); } |
| 194 | __DEVICE__ long lroundf(float __a) { return roundf(__a); } |
| 195 | #endif |
| 196 | __DEVICE__ int max(int __a, int __b) { return __nv_max(__a, __b); } |
| 197 | __DEVICE__ int min(int __a, int __b) { return __nv_min(__a, __b); } |
| 198 | __DEVICE__ double modf(double __a, double *__b) { return __nv_modf(__a, __b); } |
| 199 | __DEVICE__ float modff(float __a, float *__b) { return __nv_modff(__a, __b); } |
Sasha Smundak | 4b1f33a | 2021-01-11 15:05:07 -0800 | [diff] [blame] | 200 | __DEVICE__ double nearbyint(double __a) { return __builtin_nearbyint(__a); } |
| 201 | __DEVICE__ float nearbyintf(float __a) { return __builtin_nearbyintf(__a); } |
Sasha Smundak | 0fc590b | 2020-10-07 08:11:59 -0700 | [diff] [blame] | 202 | __DEVICE__ double nextafter(double __a, double __b) { |
| 203 | return __nv_nextafter(__a, __b); |
| 204 | } |
| 205 | __DEVICE__ float nextafterf(float __a, float __b) { |
| 206 | return __nv_nextafterf(__a, __b); |
| 207 | } |
| 208 | __DEVICE__ double norm(int __dim, const double *__t) { |
| 209 | return __nv_norm(__dim, __t); |
| 210 | } |
| 211 | __DEVICE__ double norm3d(double __a, double __b, double __c) { |
| 212 | return __nv_norm3d(__a, __b, __c); |
| 213 | } |
| 214 | __DEVICE__ float norm3df(float __a, float __b, float __c) { |
| 215 | return __nv_norm3df(__a, __b, __c); |
| 216 | } |
| 217 | __DEVICE__ double norm4d(double __a, double __b, double __c, double __d) { |
| 218 | return __nv_norm4d(__a, __b, __c, __d); |
| 219 | } |
| 220 | __DEVICE__ float norm4df(float __a, float __b, float __c, float __d) { |
| 221 | return __nv_norm4df(__a, __b, __c, __d); |
| 222 | } |
| 223 | __DEVICE__ double normcdf(double __a) { return __nv_normcdf(__a); } |
| 224 | __DEVICE__ float normcdff(float __a) { return __nv_normcdff(__a); } |
| 225 | __DEVICE__ double normcdfinv(double __a) { return __nv_normcdfinv(__a); } |
| 226 | __DEVICE__ float normcdfinvf(float __a) { return __nv_normcdfinvf(__a); } |
| 227 | __DEVICE__ float normf(int __dim, const float *__t) { |
| 228 | return __nv_normf(__dim, __t); |
| 229 | } |
| 230 | __DEVICE__ double pow(double __a, double __b) { return __nv_pow(__a, __b); } |
| 231 | __DEVICE__ float powf(float __a, float __b) { return __nv_powf(__a, __b); } |
| 232 | __DEVICE__ double powi(double __a, int __b) { return __nv_powi(__a, __b); } |
| 233 | __DEVICE__ float powif(float __a, int __b) { return __nv_powif(__a, __b); } |
| 234 | __DEVICE__ double rcbrt(double __a) { return __nv_rcbrt(__a); } |
| 235 | __DEVICE__ float rcbrtf(float __a) { return __nv_rcbrtf(__a); } |
| 236 | __DEVICE__ double remainder(double __a, double __b) { |
| 237 | return __nv_remainder(__a, __b); |
| 238 | } |
| 239 | __DEVICE__ float remainderf(float __a, float __b) { |
| 240 | return __nv_remainderf(__a, __b); |
| 241 | } |
| 242 | __DEVICE__ double remquo(double __a, double __b, int *__c) { |
| 243 | return __nv_remquo(__a, __b, __c); |
| 244 | } |
| 245 | __DEVICE__ float remquof(float __a, float __b, int *__c) { |
| 246 | return __nv_remquof(__a, __b, __c); |
| 247 | } |
| 248 | __DEVICE__ double rhypot(double __a, double __b) { |
| 249 | return __nv_rhypot(__a, __b); |
| 250 | } |
| 251 | __DEVICE__ float rhypotf(float __a, float __b) { |
| 252 | return __nv_rhypotf(__a, __b); |
| 253 | } |
Sasha Smundak | 4b1f33a | 2021-01-11 15:05:07 -0800 | [diff] [blame] | 254 | // __nv_rint* in libdevice is buggy and produces incorrect results. |
| 255 | __DEVICE__ double rint(double __a) { return __builtin_rint(__a); } |
| 256 | __DEVICE__ float rintf(float __a) { return __builtin_rintf(__a); } |
Sasha Smundak | 0fc590b | 2020-10-07 08:11:59 -0700 | [diff] [blame] | 257 | __DEVICE__ double rnorm(int __a, const double *__b) { |
| 258 | return __nv_rnorm(__a, __b); |
| 259 | } |
| 260 | __DEVICE__ double rnorm3d(double __a, double __b, double __c) { |
| 261 | return __nv_rnorm3d(__a, __b, __c); |
| 262 | } |
| 263 | __DEVICE__ float rnorm3df(float __a, float __b, float __c) { |
| 264 | return __nv_rnorm3df(__a, __b, __c); |
| 265 | } |
| 266 | __DEVICE__ double rnorm4d(double __a, double __b, double __c, double __d) { |
| 267 | return __nv_rnorm4d(__a, __b, __c, __d); |
| 268 | } |
| 269 | __DEVICE__ float rnorm4df(float __a, float __b, float __c, float __d) { |
| 270 | return __nv_rnorm4df(__a, __b, __c, __d); |
| 271 | } |
| 272 | __DEVICE__ float rnormf(int __dim, const float *__t) { |
| 273 | return __nv_rnormf(__dim, __t); |
| 274 | } |
Sasha Smundak | 0fc590b | 2020-10-07 08:11:59 -0700 | [diff] [blame] | 275 | __DEVICE__ double rsqrt(double __a) { return __nv_rsqrt(__a); } |
| 276 | __DEVICE__ float rsqrtf(float __a) { return __nv_rsqrtf(__a); } |
| 277 | __DEVICE__ double scalbn(double __a, int __b) { return __nv_scalbn(__a, __b); } |
| 278 | __DEVICE__ float scalbnf(float __a, int __b) { return __nv_scalbnf(__a, __b); } |
| 279 | __DEVICE__ double scalbln(double __a, long __b) { |
| 280 | if (__b > INT_MAX) |
| 281 | return __a > 0 ? HUGE_VAL : -HUGE_VAL; |
| 282 | if (__b < INT_MIN) |
| 283 | return __a > 0 ? 0.0 : -0.0; |
| 284 | return scalbn(__a, (int)__b); |
| 285 | } |
| 286 | __DEVICE__ float scalblnf(float __a, long __b) { |
| 287 | if (__b > INT_MAX) |
| 288 | return __a > 0 ? HUGE_VALF : -HUGE_VALF; |
| 289 | if (__b < INT_MIN) |
| 290 | return __a > 0 ? 0.f : -0.f; |
| 291 | return scalbnf(__a, (int)__b); |
| 292 | } |
| 293 | __DEVICE__ double sin(double __a) { return __nv_sin(__a); } |
| 294 | __DEVICE_VOID__ void sincos(double __a, double *__s, double *__c) { |
| 295 | return __nv_sincos(__a, __s, __c); |
| 296 | } |
| 297 | __DEVICE_VOID__ void sincosf(float __a, float *__s, float *__c) { |
| 298 | return __FAST_OR_SLOW(__nv_fast_sincosf, __nv_sincosf)(__a, __s, __c); |
| 299 | } |
| 300 | __DEVICE_VOID__ void sincospi(double __a, double *__s, double *__c) { |
| 301 | return __nv_sincospi(__a, __s, __c); |
| 302 | } |
| 303 | __DEVICE_VOID__ void sincospif(float __a, float *__s, float *__c) { |
| 304 | return __nv_sincospif(__a, __s, __c); |
| 305 | } |
| 306 | __DEVICE__ float sinf(float __a) { |
| 307 | return __FAST_OR_SLOW(__nv_fast_sinf, __nv_sinf)(__a); |
| 308 | } |
| 309 | __DEVICE__ double sinh(double __a) { return __nv_sinh(__a); } |
| 310 | __DEVICE__ float sinhf(float __a) { return __nv_sinhf(__a); } |
| 311 | __DEVICE__ double sinpi(double __a) { return __nv_sinpi(__a); } |
| 312 | __DEVICE__ float sinpif(float __a) { return __nv_sinpif(__a); } |
| 313 | __DEVICE__ double sqrt(double __a) { return __nv_sqrt(__a); } |
| 314 | __DEVICE__ float sqrtf(float __a) { return __nv_sqrtf(__a); } |
| 315 | __DEVICE__ double tan(double __a) { return __nv_tan(__a); } |
| 316 | __DEVICE__ float tanf(float __a) { return __nv_tanf(__a); } |
| 317 | __DEVICE__ double tanh(double __a) { return __nv_tanh(__a); } |
| 318 | __DEVICE__ float tanhf(float __a) { return __nv_tanhf(__a); } |
| 319 | __DEVICE__ double tgamma(double __a) { return __nv_tgamma(__a); } |
| 320 | __DEVICE__ float tgammaf(float __a) { return __nv_tgammaf(__a); } |
| 321 | __DEVICE__ double trunc(double __a) { return __nv_trunc(__a); } |
| 322 | __DEVICE__ float truncf(float __a) { return __nv_truncf(__a); } |
| 323 | __DEVICE__ unsigned long long ullmax(unsigned long long __a, |
| 324 | unsigned long long __b) { |
| 325 | return __nv_ullmax(__a, __b); |
| 326 | } |
| 327 | __DEVICE__ unsigned long long ullmin(unsigned long long __a, |
| 328 | unsigned long long __b) { |
| 329 | return __nv_ullmin(__a, __b); |
| 330 | } |
| 331 | __DEVICE__ unsigned int umax(unsigned int __a, unsigned int __b) { |
| 332 | return __nv_umax(__a, __b); |
| 333 | } |
| 334 | __DEVICE__ unsigned int umin(unsigned int __a, unsigned int __b) { |
| 335 | return __nv_umin(__a, __b); |
| 336 | } |
| 337 | __DEVICE__ double y0(double __a) { return __nv_y0(__a); } |
| 338 | __DEVICE__ float y0f(float __a) { return __nv_y0f(__a); } |
| 339 | __DEVICE__ double y1(double __a) { return __nv_y1(__a); } |
| 340 | __DEVICE__ float y1f(float __a) { return __nv_y1f(__a); } |
| 341 | __DEVICE__ double yn(int __a, double __b) { return __nv_yn(__a, __b); } |
| 342 | __DEVICE__ float ynf(int __a, float __b) { return __nv_ynf(__a, __b); } |
| 343 | |
| 344 | #pragma pop_macro("__DEVICE__") |
| 345 | #pragma pop_macro("__DEVICE_VOID__") |
| 346 | #pragma pop_macro("__FAST_OR_SLOW") |
| 347 | |
| 348 | #endif // __CLANG_CUDA_DEVICE_FUNCTIONS_H__ |