338dee0742
With this PR, if we have customized implementation for scalar or vector length = 2, we don't need to write new macros, e.g. https://github.com/intel/llvm/blob/fb18321705f6/libclc/clc/include/clc/clcmacro.h#L15 Undef __HALF_ONLY, __FLOAT_ONLY and __DOUBLE_ONLY at the end of clc/include/clc/math/gentype.inc llvm-diff shows no change to nvptx64--nvidiacl.bc and amdgcn--amdhsa.bc
83 lines
2.0 KiB
Common Lisp
83 lines
2.0 KiB
Common Lisp
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include <clc/clcmacro.h>
|
|
#include <clc/float/definitions.h>
|
|
#include <clc/internal/clc.h>
|
|
#include <clc/math/clc_exp.h>
|
|
#include <clc/math/clc_fabs.h>
|
|
#include <clc/math/clc_lgamma.h>
|
|
#include <clc/math/clc_sinpi.h>
|
|
#include <clc/math/math.h>
|
|
|
|
_CLC_OVERLOAD _CLC_DEF float __clc_tgamma(float x) {
|
|
const float pi = 3.1415926535897932384626433832795f;
|
|
float absx = __clc_fabs(x);
|
|
float lg = __clc_lgamma(absx);
|
|
float g = __clc_exp(lg);
|
|
|
|
if (x < 0.0f) {
|
|
float z = __clc_sinpi(x);
|
|
g = g * absx * z;
|
|
g = pi / g;
|
|
g = g == 0 ? INFINITY : g;
|
|
g = z == 0 ? FLT_NAN : g;
|
|
}
|
|
|
|
return g;
|
|
}
|
|
|
|
#define __FLOAT_ONLY
|
|
#define FUNCTION __clc_tgamma
|
|
#define __CLC_BODY <clc/shared/unary_def_scalarize.inc>
|
|
#include <clc/math/gentype.inc>
|
|
#undef FUNCTION
|
|
|
|
#ifdef cl_khr_fp64
|
|
|
|
#pragma OPENCL EXTENSION cl_khr_fp64 : enable
|
|
|
|
_CLC_OVERLOAD _CLC_DEF double __clc_tgamma(double x) {
|
|
const double pi = 3.1415926535897932384626433832795;
|
|
double absx = __clc_fabs(x);
|
|
double lg = __clc_lgamma(absx);
|
|
double g = __clc_exp(lg);
|
|
|
|
if (x < 0.0) {
|
|
double z = __clc_sinpi(x);
|
|
g = g * absx * z;
|
|
g = pi / g;
|
|
g = g == 0 ? INFINITY : g;
|
|
g = z == 0 ? DBL_NAN : g;
|
|
}
|
|
|
|
return g;
|
|
}
|
|
|
|
#define __DOUBLE_ONLY
|
|
#define FUNCTION __clc_tgamma
|
|
#define __CLC_BODY <clc/shared/unary_def_scalarize.inc>
|
|
#include <clc/math/gentype.inc>
|
|
#undef FUNCTION
|
|
|
|
#endif
|
|
|
|
#ifdef cl_khr_fp16
|
|
|
|
#include <clc/clc_convert.h>
|
|
|
|
#pragma OPENCL EXTENSION cl_khr_fp16 : enable
|
|
|
|
// Forward the half version of this builtin onto the float one
|
|
#define __HALF_ONLY
|
|
#define __CLC_FUNCTION __clc_tgamma
|
|
#define __CLC_BODY <clc/math/unary_def_via_fp32.inc>
|
|
#include <clc/math/gentype.inc>
|
|
|
|
#endif
|