15f757f3fSDimitry Andric/*===-- __clang_openmp_device_functions.h - OpenMP math declares -*- c++ -*-=== 20b57cec5SDimitry Andric * 30b57cec5SDimitry Andric * Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 40b57cec5SDimitry Andric * See https://llvm.org/LICENSE.txt for license information. 50b57cec5SDimitry Andric * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 60b57cec5SDimitry Andric * 70b57cec5SDimitry Andric *===-----------------------------------------------------------------------=== 80b57cec5SDimitry Andric */ 90b57cec5SDimitry Andric 105ffd83dbSDimitry Andric#ifndef __CLANG_OPENMP_CMATH_H__ 115ffd83dbSDimitry Andric#define __CLANG_OPENMP_CMATH_H__ 120b57cec5SDimitry Andric 135ffd83dbSDimitry Andric#ifndef _OPENMP 145ffd83dbSDimitry Andric#error "This file is for OpenMP compilation only." 155ffd83dbSDimitry Andric#endif 165ffd83dbSDimitry Andric 170b57cec5SDimitry Andric#include_next <cmath> 185ffd83dbSDimitry Andric 195ffd83dbSDimitry Andric// Make sure we include our math.h overlay, it probably happend already but we 205ffd83dbSDimitry Andric// need to be sure. 215ffd83dbSDimitry Andric#include <math.h> 225ffd83dbSDimitry Andric 235ffd83dbSDimitry Andric// We (might) need cstdlib because __clang_cuda_cmath.h below declares `abs` 245ffd83dbSDimitry Andric// which might live in cstdlib. 255ffd83dbSDimitry Andric#include <cstdlib> 265ffd83dbSDimitry Andric 27e8d8bef9SDimitry Andric// We need limits because __clang_cuda_cmath.h below uses `std::numeric_limit`. 28e8d8bef9SDimitry Andric#include <limits> 29e8d8bef9SDimitry Andric 305ffd83dbSDimitry Andric#pragma omp begin declare variant match( \ 31e8d8bef9SDimitry Andric device = {arch(nvptx, nvptx64)}, implementation = {extension(match_any, allow_templates)}) 325ffd83dbSDimitry Andric 335ffd83dbSDimitry Andric#define __CUDA__ 345ffd83dbSDimitry Andric#define __OPENMP_NVPTX__ 355ffd83dbSDimitry Andric#include <__clang_cuda_cmath.h> 365ffd83dbSDimitry Andric#undef __OPENMP_NVPTX__ 375ffd83dbSDimitry Andric#undef __CUDA__ 385ffd83dbSDimitry Andric 395ffd83dbSDimitry Andric// Overloads not provided by the CUDA wrappers but by the CUDA system headers. 405ffd83dbSDimitry Andric// Since we do not include the latter we define them ourselves. 415ffd83dbSDimitry Andric#define __DEVICE__ static constexpr __attribute__((always_inline, nothrow)) 425ffd83dbSDimitry Andric 435ffd83dbSDimitry Andric__DEVICE__ float acosh(float __x) { return ::acoshf(__x); } 445ffd83dbSDimitry Andric__DEVICE__ float asinh(float __x) { return ::asinhf(__x); } 455ffd83dbSDimitry Andric__DEVICE__ float atanh(float __x) { return ::atanhf(__x); } 465ffd83dbSDimitry Andric__DEVICE__ float cbrt(float __x) { return ::cbrtf(__x); } 475ffd83dbSDimitry Andric__DEVICE__ float erf(float __x) { return ::erff(__x); } 485ffd83dbSDimitry Andric__DEVICE__ float erfc(float __x) { return ::erfcf(__x); } 495ffd83dbSDimitry Andric__DEVICE__ float exp2(float __x) { return ::exp2f(__x); } 505ffd83dbSDimitry Andric__DEVICE__ float expm1(float __x) { return ::expm1f(__x); } 515ffd83dbSDimitry Andric__DEVICE__ float fdim(float __x, float __y) { return ::fdimf(__x, __y); } 525ffd83dbSDimitry Andric__DEVICE__ float hypot(float __x, float __y) { return ::hypotf(__x, __y); } 535ffd83dbSDimitry Andric__DEVICE__ int ilogb(float __x) { return ::ilogbf(__x); } 545ffd83dbSDimitry Andric__DEVICE__ float lgamma(float __x) { return ::lgammaf(__x); } 555ffd83dbSDimitry Andric__DEVICE__ long long int llrint(float __x) { return ::llrintf(__x); } 565ffd83dbSDimitry Andric__DEVICE__ long long int llround(float __x) { return ::llroundf(__x); } 575ffd83dbSDimitry Andric__DEVICE__ float log1p(float __x) { return ::log1pf(__x); } 585ffd83dbSDimitry Andric__DEVICE__ float log2(float __x) { return ::log2f(__x); } 595ffd83dbSDimitry Andric__DEVICE__ float logb(float __x) { return ::logbf(__x); } 605ffd83dbSDimitry Andric__DEVICE__ long int lrint(float __x) { return ::lrintf(__x); } 615ffd83dbSDimitry Andric__DEVICE__ long int lround(float __x) { return ::lroundf(__x); } 625ffd83dbSDimitry Andric__DEVICE__ float nextafter(float __x, float __y) { 635ffd83dbSDimitry Andric return ::nextafterf(__x, __y); 645ffd83dbSDimitry Andric} 655ffd83dbSDimitry Andric__DEVICE__ float remainder(float __x, float __y) { 665ffd83dbSDimitry Andric return ::remainderf(__x, __y); 675ffd83dbSDimitry Andric} 685ffd83dbSDimitry Andric__DEVICE__ float scalbln(float __x, long int __y) { 695ffd83dbSDimitry Andric return ::scalblnf(__x, __y); 705ffd83dbSDimitry Andric} 715ffd83dbSDimitry Andric__DEVICE__ float scalbn(float __x, int __y) { return ::scalbnf(__x, __y); } 725ffd83dbSDimitry Andric__DEVICE__ float tgamma(float __x) { return ::tgammaf(__x); } 735ffd83dbSDimitry Andric 745ffd83dbSDimitry Andric#undef __DEVICE__ 755ffd83dbSDimitry Andric 765ffd83dbSDimitry Andric#pragma omp end declare variant 775ffd83dbSDimitry Andric 7869ade1e0SDimitry Andric#ifdef __AMDGCN__ 7969ade1e0SDimitry Andric#pragma omp begin declare variant match(device = {arch(amdgcn)}) 8069ade1e0SDimitry Andric 8169ade1e0SDimitry Andric#pragma push_macro("__constant__") 8269ade1e0SDimitry Andric#define __constant__ __attribute__((constant)) 8369ade1e0SDimitry Andric#define __OPENMP_AMDGCN__ 8469ade1e0SDimitry Andric 8569ade1e0SDimitry Andric#include <__clang_hip_cmath.h> 8669ade1e0SDimitry Andric 8769ade1e0SDimitry Andric#pragma pop_macro("__constant__") 8869ade1e0SDimitry Andric#undef __OPENMP_AMDGCN__ 8969ade1e0SDimitry Andric 9069ade1e0SDimitry Andric// Define overloads otherwise which are absent 9169ade1e0SDimitry Andric#define __DEVICE__ static constexpr __attribute__((always_inline, nothrow)) 9269ade1e0SDimitry Andric 9369ade1e0SDimitry Andric__DEVICE__ float acos(float __x) { return ::acosf(__x); } 9469ade1e0SDimitry Andric__DEVICE__ float acosh(float __x) { return ::acoshf(__x); } 9569ade1e0SDimitry Andric__DEVICE__ float asin(float __x) { return ::asinf(__x); } 9669ade1e0SDimitry Andric__DEVICE__ float asinh(float __x) { return ::asinhf(__x); } 9769ade1e0SDimitry Andric__DEVICE__ float atan(float __x) { return ::atanf(__x); } 9869ade1e0SDimitry Andric__DEVICE__ float atan2(float __x, float __y) { return ::atan2f(__x, __y); } 9969ade1e0SDimitry Andric__DEVICE__ float atanh(float __x) { return ::atanhf(__x); } 10069ade1e0SDimitry Andric__DEVICE__ float cbrt(float __x) { return ::cbrtf(__x); } 10169ade1e0SDimitry Andric__DEVICE__ float cosh(float __x) { return ::coshf(__x); } 10269ade1e0SDimitry Andric__DEVICE__ float erf(float __x) { return ::erff(__x); } 10369ade1e0SDimitry Andric__DEVICE__ float erfc(float __x) { return ::erfcf(__x); } 10469ade1e0SDimitry Andric__DEVICE__ float exp2(float __x) { return ::exp2f(__x); } 10569ade1e0SDimitry Andric__DEVICE__ float expm1(float __x) { return ::expm1f(__x); } 10669ade1e0SDimitry Andric__DEVICE__ float fdim(float __x, float __y) { return ::fdimf(__x, __y); } 10769ade1e0SDimitry Andric__DEVICE__ float hypot(float __x, float __y) { return ::hypotf(__x, __y); } 10869ade1e0SDimitry Andric__DEVICE__ int ilogb(float __x) { return ::ilogbf(__x); } 10969ade1e0SDimitry Andric__DEVICE__ float ldexp(float __arg, int __exp) { 11069ade1e0SDimitry Andric return ::ldexpf(__arg, __exp); 11169ade1e0SDimitry Andric} 11269ade1e0SDimitry Andric__DEVICE__ float lgamma(float __x) { return ::lgammaf(__x); } 11369ade1e0SDimitry Andric__DEVICE__ float log1p(float __x) { return ::log1pf(__x); } 11469ade1e0SDimitry Andric__DEVICE__ float logb(float __x) { return ::logbf(__x); } 11569ade1e0SDimitry Andric__DEVICE__ float nextafter(float __x, float __y) { 11669ade1e0SDimitry Andric return ::nextafterf(__x, __y); 11769ade1e0SDimitry Andric} 11869ade1e0SDimitry Andric__DEVICE__ float remainder(float __x, float __y) { 11969ade1e0SDimitry Andric return ::remainderf(__x, __y); 12069ade1e0SDimitry Andric} 12169ade1e0SDimitry Andric__DEVICE__ float scalbn(float __x, int __y) { return ::scalbnf(__x, __y); } 12269ade1e0SDimitry Andric__DEVICE__ float sinh(float __x) { return ::sinhf(__x); } 12369ade1e0SDimitry Andric__DEVICE__ float tan(float __x) { return ::tanf(__x); } 12469ade1e0SDimitry Andric__DEVICE__ float tanh(float __x) { return ::tanhf(__x); } 12569ade1e0SDimitry Andric__DEVICE__ float tgamma(float __x) { return ::tgammaf(__x); } 12669ade1e0SDimitry Andric 12769ade1e0SDimitry Andric#undef __DEVICE__ 12869ade1e0SDimitry Andric 12969ade1e0SDimitry Andric#pragma omp end declare variant 13069ade1e0SDimitry Andric#endif // __AMDGCN__ 13169ade1e0SDimitry Andric 1320b57cec5SDimitry Andric#endif 133