cmath (5251B) - Raw
1 /*===-- __clang_openmp_device_functions.h - OpenMP math declares -*- c++ -*-=== 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 10 #ifndef __CLANG_OPENMP_CMATH_H__ 11 #define __CLANG_OPENMP_CMATH_H__ 12 13 #ifndef _OPENMP 14 #error "This file is for OpenMP compilation only." 15 #endif 16 17 #include_next <cmath> 18 19 // Make sure we include our math.h overlay, it probably happend already but we 20 // need to be sure. 21 #include <math.h> 22 23 // We (might) need cstdlib because __clang_cuda_cmath.h below declares `abs` 24 // which might live in cstdlib. 25 #include <cstdlib> 26 27 // We need limits because __clang_cuda_cmath.h below uses `std::numeric_limit`. 28 #include <limits> 29 30 #pragma omp begin declare variant match( \ 31 device = {arch(nvptx, nvptx64)}, implementation = {extension(match_any, allow_templates)}) 32 33 #define __CUDA__ 34 #define __OPENMP_NVPTX__ 35 #include <__clang_cuda_cmath.h> 36 #undef __OPENMP_NVPTX__ 37 #undef __CUDA__ 38 39 // Overloads not provided by the CUDA wrappers but by the CUDA system headers. 40 // Since we do not include the latter we define them ourselves. 41 #define __DEVICE__ static constexpr __attribute__((always_inline, nothrow)) 42 43 __DEVICE__ float acosh(float __x) { return ::acoshf(__x); } 44 __DEVICE__ float asinh(float __x) { return ::asinhf(__x); } 45 __DEVICE__ float atanh(float __x) { return ::atanhf(__x); } 46 __DEVICE__ float cbrt(float __x) { return ::cbrtf(__x); } 47 __DEVICE__ float erf(float __x) { return ::erff(__x); } 48 __DEVICE__ float erfc(float __x) { return ::erfcf(__x); } 49 __DEVICE__ float exp2(float __x) { return ::exp2f(__x); } 50 __DEVICE__ float expm1(float __x) { return ::expm1f(__x); } 51 __DEVICE__ float fdim(float __x, float __y) { return ::fdimf(__x, __y); } 52 __DEVICE__ float hypot(float __x, float __y) { return ::hypotf(__x, __y); } 53 __DEVICE__ int ilogb(float __x) { return ::ilogbf(__x); } 54 __DEVICE__ float lgamma(float __x) { return ::lgammaf(__x); } 55 __DEVICE__ long long int llrint(float __x) { return ::llrintf(__x); } 56 __DEVICE__ long long int llround(float __x) { return ::llroundf(__x); } 57 __DEVICE__ float log1p(float __x) { return ::log1pf(__x); } 58 __DEVICE__ float log2(float __x) { return ::log2f(__x); } 59 __DEVICE__ float logb(float __x) { return ::logbf(__x); } 60 __DEVICE__ long int lrint(float __x) { return ::lrintf(__x); } 61 __DEVICE__ long int lround(float __x) { return ::lroundf(__x); } 62 __DEVICE__ float nextafter(float __x, float __y) { 63 return ::nextafterf(__x, __y); 64 } 65 __DEVICE__ float remainder(float __x, float __y) { 66 return ::remainderf(__x, __y); 67 } 68 __DEVICE__ float scalbln(float __x, long int __y) { 69 return ::scalblnf(__x, __y); 70 } 71 __DEVICE__ float scalbn(float __x, int __y) { return ::scalbnf(__x, __y); } 72 __DEVICE__ float tgamma(float __x) { return ::tgammaf(__x); } 73 74 #undef __DEVICE__ 75 76 #pragma omp end declare variant 77 78 #ifdef __AMDGCN__ 79 #pragma omp begin declare variant match(device = {arch(amdgcn)}) 80 81 #pragma push_macro("__constant__") 82 #define __constant__ __attribute__((constant)) 83 #define __OPENMP_AMDGCN__ 84 85 #include <__clang_hip_cmath.h> 86 87 #pragma pop_macro("__constant__") 88 #undef __OPENMP_AMDGCN__ 89 90 // Define overloads otherwise which are absent 91 #define __DEVICE__ static constexpr __attribute__((always_inline, nothrow)) 92 93 __DEVICE__ float acos(float __x) { return ::acosf(__x); } 94 __DEVICE__ float acosh(float __x) { return ::acoshf(__x); } 95 __DEVICE__ float asin(float __x) { return ::asinf(__x); } 96 __DEVICE__ float asinh(float __x) { return ::asinhf(__x); } 97 __DEVICE__ float atan(float __x) { return ::atanf(__x); } 98 __DEVICE__ float atan2(float __x, float __y) { return ::atan2f(__x, __y); } 99 __DEVICE__ float atanh(float __x) { return ::atanhf(__x); } 100 __DEVICE__ float cbrt(float __x) { return ::cbrtf(__x); } 101 __DEVICE__ float cosh(float __x) { return ::coshf(__x); } 102 __DEVICE__ float erf(float __x) { return ::erff(__x); } 103 __DEVICE__ float erfc(float __x) { return ::erfcf(__x); } 104 __DEVICE__ float exp2(float __x) { return ::exp2f(__x); } 105 __DEVICE__ float expm1(float __x) { return ::expm1f(__x); } 106 __DEVICE__ float fdim(float __x, float __y) { return ::fdimf(__x, __y); } 107 __DEVICE__ float hypot(float __x, float __y) { return ::hypotf(__x, __y); } 108 __DEVICE__ int ilogb(float __x) { return ::ilogbf(__x); } 109 __DEVICE__ float ldexp(float __arg, int __exp) { 110 return ::ldexpf(__arg, __exp); 111 } 112 __DEVICE__ float lgamma(float __x) { return ::lgammaf(__x); } 113 __DEVICE__ float log1p(float __x) { return ::log1pf(__x); } 114 __DEVICE__ float logb(float __x) { return ::logbf(__x); } 115 __DEVICE__ float nextafter(float __x, float __y) { 116 return ::nextafterf(__x, __y); 117 } 118 __DEVICE__ float remainder(float __x, float __y) { 119 return ::remainderf(__x, __y); 120 } 121 __DEVICE__ float scalbn(float __x, int __y) { return ::scalbnf(__x, __y); } 122 __DEVICE__ float sinh(float __x) { return ::sinhf(__x); } 123 __DEVICE__ float tan(float __x) { return ::tanf(__x); } 124 __DEVICE__ float tanh(float __x) { return ::tanhf(__x); } 125 __DEVICE__ float tgamma(float __x) { return ::tgammaf(__x); } 126 127 #undef __DEVICE__ 128 129 #pragma omp end declare variant 130 #endif // __AMDGCN__ 131 132 #endif