Home | History | Annotate | Line # | Download | only in openmp_wrappers 
      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 #endif
     79