Fraser Cormack 13a313fe58
[libclc] Move sinpi/cospi/tanpi to the CLC library (#133889)
Additionally, these builtins are now vectorized.

This also moves the native_recip and native_divide builtins as they are
used by the tanpi builtin.
2025-04-01 12:03:21 +01:00

133 lines
3.4 KiB
C++

//===----------------------------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
#if __CLC_FPSIZE == 32
_CLC_DEF _CLC_OVERLOAD __CLC_GENTYPE __clc_tanpi(__CLC_GENTYPE x) {
__CLC_INTN ix = __CLC_AS_INTN(x);
__CLC_INTN xsgn = ix & (__CLC_INTN)SIGNBIT_SP32;
__CLC_INTN xnsgn = xsgn ^ (__CLC_INTN)SIGNBIT_SP32;
ix ^= xsgn;
__CLC_GENTYPE absx = __clc_fabs(x);
__CLC_INTN iax = __CLC_CONVERT_INTN(absx);
__CLC_GENTYPE r = absx - __CLC_CONVERT_GENTYPE(iax);
__CLC_INTN xodd = xsgn ^ __CLC_AS_INTN((iax & 0x1) != 0 ? SIGNBIT_SP32 : 0);
// Initialize with return for +-Inf and NaN
__CLC_INTN ir = QNANBITPATT_SP32;
// 2^24 <= |x| < Inf, the result is always even integer
ir = ix < PINFBITPATT_SP32 ? xsgn : ir;
// 2^23 <= |x| < 2^24, the result is always integer
ir = ix < 0x4b800000 ? xodd : ir;
// 0x1.0p-7 <= |x| < 2^23, result depends on which 0.25 interval
// r < 1.0
__CLC_GENTYPE a = 1.0f - r;
__CLC_INTN e = 0;
__CLC_INTN s = xnsgn;
// r <= 0.75
__CLC_INTN c = r <= 0.75f;
a = c ? r - 0.5f : a;
e = c ? 1 : e;
s = c ? xsgn : s;
// r < 0.5
c = r < 0.5f;
a = c ? 0.5f - r : a;
s = c ? xnsgn : s;
// 0 < r <= 0.25
c = r <= 0.25f;
a = c ? r : a;
e = c ? 0 : e;
s = c ? xsgn : s;
__CLC_GENTYPE t = __clc_tanf_piby4(a * M_PI_F, 0);
__CLC_GENTYPE tr = -__clc_native_recip(t);
__CLC_INTN jr = s ^ __CLC_AS_INTN(e != 0 ? tr : t);
jr = r == 0.5f ? xodd | 0x7f800000 : jr;
ir = ix < 0x4b000000 ? jr : ir;
return __CLC_AS_GENTYPE(ir);
}
#elif __CLC_FPSIZE == 64
_CLC_DEF _CLC_OVERLOAD __CLC_GENTYPE __clc_tanpi(__CLC_GENTYPE x) {
__CLC_LONGN ix = __CLC_AS_LONGN(x);
__CLC_LONGN xsgn = ix & (__CLC_LONGN)0x8000000000000000L;
__CLC_LONGN xnsgn = xsgn ^ (__CLC_LONGN)0x8000000000000000L;
ix ^= xsgn;
__CLC_GENTYPE absx = __clc_fabs(x);
__CLC_LONGN iax = __CLC_CONVERT_LONGN(absx);
__CLC_GENTYPE r = absx - __CLC_CONVERT_GENTYPE(iax);
__CLC_LONGN xodd =
xsgn ^ __CLC_AS_LONGN((iax & 0x1) != 0 ? 0x8000000000000000L : 0L);
// Initialize with return for +-Inf and NaN
__CLC_LONGN ir = QNANBITPATT_DP64;
// 2^53 <= |x| < Inf, the result is always even integer
ir = ix < PINFBITPATT_DP64 ? xsgn : ir;
// 2^52 <= |x| < 2^53, the result is always integer
ir = ix < 0x4340000000000000L ? xodd : ir;
// 0x1.0p-14 <= |x| < 2^53, result depends on which 0.25 interval
// r < 1.0
__CLC_GENTYPE a = 1.0 - r;
__CLC_LONGN e = 0;
__CLC_LONGN s = xnsgn;
// r <= 0.75
__CLC_LONGN c = r <= 0.75;
__CLC_GENTYPE t = r - 0.5;
a = c ? t : a;
e = c ? 1 : e;
s = c ? xsgn : s;
// r < 0.5
c = r < 0.5;
t = 0.5 - r;
a = c ? t : a;
s = c ? xnsgn : s;
// r <= 0.25
c = r <= 0.25;
a = c ? r : a;
e = c ? 0 : e;
s = c ? xsgn : s;
__CLC_GENTYPE api = a * M_PI;
__CLC_GENTYPE lo, hi;
__clc_tan_piby4(api, 0.0, &lo, &hi);
__CLC_LONGN jr = s ^ __CLC_AS_LONGN(e != 0 ? hi : lo);
__CLC_LONGN si = xodd | 0x7ff0000000000000L;
jr = r == 0.5 ? si : jr;
ir = ix < 0x4330000000000000L ? jr : ir;
return __CLC_AS_GENTYPE(ir);
}
#elif __CLC_FPSIZE == 16
_CLC_OVERLOAD _CLC_DEF __CLC_GENTYPE __clc_tanpi(__CLC_GENTYPE x) {
return __CLC_CONVERT_GENTYPE(__clc_tanpi(__CLC_CONVERT_FLOATN(x)));
}
#endif