133 lines · plain
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-exception6 *7 *===-----------------------------------------------------------------------===8 */9 10#ifndef __CLANG_OPENMP_CMATH_H__11#define __CLANG_OPENMP_CMATH_H__12 13#ifndef _OPENMP14#error "This file is for OpenMP compilation only."15#endif16 17#include_next <cmath>18 19// Make sure we include our math.h overlay, it probably happend already but we20// 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 variant77 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 absent91#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 variant130#endif // __AMDGCN__131 132#endif133