-
Notifications
You must be signed in to change notification settings - Fork 61
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
98 changed files
with
1,741 additions
and
3,919 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,76 @@ | ||
| //================================================================================================== | ||
| /* | ||
| EVE - Expressive Vector Engine | ||
| Copyright : EVE Project Contributors | ||
| SPDX-License-Identifier: BSL-1.0 | ||
| */ | ||
| //================================================================================================== | ||
| #pragma once | ||
| #include <eve/module/bessel/detail/kernel_bessel_y0.hpp> | ||
| #include <eve/module/bessel/detail/kernel_bessel_y1.hpp> | ||
| #include <eve/module/bessel/detail/kernel_bessel_y.hpp> | ||
| #include <eve/module/core.hpp> | ||
| #include <eve/module/math.hpp> | ||
| #include <eve/as_element.hpp> | ||
|
|
||
| namespace eve::detail | ||
| { | ||
| // T is always floating | ||
| template < typename I, typename T > constexpr EVE_FORCEINLINE | ||
| T cb_y(I nu, T x) noexcept | ||
| { | ||
| if constexpr(floating_value<I>) | ||
| { | ||
| if constexpr(scalar_floating_value<I>) | ||
| { | ||
| auto flintx = is_flint(nu); | ||
| return flintx ? kernel_bessel_y_int(nu, x) : kernel_bessel_y_flt(T(nu), x); | ||
| } | ||
| else if constexpr( scalar_value<T> ) | ||
| { | ||
| auto flintx = is_flint(nu); | ||
| return flintx ? kernel_bessel_y_int(nu, x) : kernel_bessel_y_flt(nu, x); | ||
| } | ||
| else // simd | ||
| { | ||
| auto flint_nu = is_flint(nu); | ||
| if( eve::all(flint_nu) ) return kernel_bessel_y_int(nu, x); | ||
| else if( eve::none(flint_nu) ) return kernel_bessel_y_flt(nu, x); | ||
| else | ||
| { | ||
| auto nu_int = if_else(flint_nu, nu, zero); | ||
| auto nu_flt = if_else(flint_nu, T(0.5), nu); | ||
| return if_else(flint_nu, kernel_bessel_y_int(nu_int, x), kernel_bessel_y_flt(nu_flt, x)); | ||
| } | ||
| } | ||
| } | ||
| else constexpr(integral_value<I>) | ||
| { | ||
| if constexpr(simd_value<I> && scalar_value<T>) | ||
| { | ||
| using c_t = wide<T, cardinal_t<I>>; | ||
| return cb_yn(nu, c_t(x)); | ||
| } | ||
| else if constexpr(scalar_value<I> && simd_value<T>) | ||
| { | ||
| using i_t = wide<I, cardinal_t<T>>; | ||
| return kernel_bessel_y_int(i_t(nu), x); | ||
| } | ||
| else if constexpr(simd_value<I> && simd_value<T>) | ||
| { | ||
| auto n = convert(nu, as_element(x)); | ||
| return kernel_bessel_y_int(n, x); | ||
| } | ||
| else return kernel_bessel_y_int(T(nu), x); | ||
| } | ||
| } | ||
|
|
||
| // T is always floating | ||
| template < typename I, typename T > constexpr EVE_FORCEINLINE | ||
| T sb_y(I nu, T x) noexcept | ||
| { | ||
| using elt_t = element_type_t<T>; | ||
| if constexpr( integral_value<I> ) return sph_bessel_yn(convert(n, as<elt_t>()), x); | ||
| else return cyl_bessel_yn(n + half(as(x)), x) * rsqrt(2 * x * inv_pi(as(x))); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,79 @@ | ||
| //================================================================================================== | ||
| /* | ||
| EVE - Expressive Vector Engine | ||
| Copyright : EVE Project Contributors | ||
| SPDX-License-Identifier: BSL-1.0 | ||
| */ | ||
| //================================================================================================== | ||
| #pragma once | ||
| #include <eve/module/bessel/detail/kernel_bessel_k0.hpp> | ||
| #include <eve/module/bessel/detail/kernel_bessel_k1.hpp> | ||
| #include <eve/module/bessel/detail/kernel_bessel_k.hpp> | ||
| #include <eve/module/core.hpp> | ||
| #include <eve/module/math.hpp> | ||
| #include <eve/as_element.hpp> | ||
|
|
||
| namespace eve::detail | ||
| { | ||
| // T is always floating | ||
| template < typename I, typename T > constexpr EVE_FORCEINLINE | ||
| T cb_k(I nu, T x) noexcept | ||
| { | ||
| if constexpr(floating_value<I>) | ||
| { | ||
| if constexpr(scalar_value<I> && simd_value<T>) | ||
| { | ||
| auto flintx = is_flint(nu); | ||
| return flintx ? kernel_bessel_k_int(nu, x) : kernel_bessel_k_flt(T(nu), x); | ||
| } | ||
| else if constexpr( scalar_value<T> ) | ||
| { | ||
| auto flintx = is_flint(nu); | ||
| return flintx ? kernel_bessel_k_int(nu, x) : kernel_bessel_k_flt(nu, x); | ||
| } | ||
| else // simd | ||
| { | ||
| auto flint_nu = is_flint(nu); | ||
| if( eve::all(flint_nu) ) return kernel_bessel_k_int(nu, x); | ||
| else if( eve::none(flint_nu) ) return kernel_bessel_k_flt(nu, x); | ||
| else | ||
| { | ||
| auto nu_int = if_else(flint_nu, nu, zero); | ||
| auto nu_flt = if_else(flint_nu, T(0.5), nu); | ||
| return if_else(flint_nu, kernel_bessel_k_int(nu_int, x), kernel_bessel_k_flt(nu_flt, x)); | ||
| } | ||
| } | ||
| } | ||
| else //if constexpr(integral_value<I>) | ||
| { | ||
| if constexpr(simd_value<I> && scalar_value<T>) | ||
| { | ||
| using c_t = wide<T, cardinal_t<I>>; | ||
| return cyl_bessel_kn(nu, c_t(x)); | ||
| } | ||
| else if constexpr(scalar_value<I> && scalar_value<T>) | ||
| { | ||
| return kernel_bessel_k_int(nu, x); | ||
| } | ||
| else if constexpr(scalar_value<I> && simd_value<T>) | ||
| { | ||
| using i_t = wide<I, cardinal_t<T>>; | ||
| return kernel_bessel_k_int(i_t(nu), x); | ||
| } | ||
| else if constexpr(simd_value<I> && simd_value<T>) | ||
| { | ||
| auto n = convert(nu, as_element(x)); | ||
| return kernel_bessel_k_int(n, x); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| // T is always floating | ||
| template < typename I, typename T > constexpr EVE_FORCEINLINE | ||
| T sb_k(I n, T x) noexcept | ||
| { | ||
| using elt_t = element_type_t<T>; | ||
| if constexpr( integral_value<I> ) return sb_k(convert(n, as<elt_t>()), x); | ||
| else return cb_k(n + half(as(x)), x) * rsqrt(2 * x * inv_pi(as(x))); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,76 @@ | ||
| //================================================================================================== | ||
| /* | ||
| EVE - Expressive Vector Engine | ||
| Copyright : EVE Project Contributors | ||
| SPDX-License-Identifier: BSL-1.0 | ||
| */ | ||
| //================================================================================================== | ||
| #pragma once | ||
| #include <eve/module/bessel/detail/kernel_bessel_y0.hpp> | ||
| #include <eve/module/bessel/detail/kernel_bessel_y1.hpp> | ||
| #include <eve/module/bessel/detail/kernel_bessel_y.hpp> | ||
| #include <eve/module/core.hpp> | ||
| #include <eve/module/math.hpp> | ||
| #include <eve/as_element.hpp> | ||
|
|
||
| namespace eve::detail | ||
| { | ||
| // T is always floating | ||
| template < typename I, typename T > constexpr EVE_FORCEINLINE | ||
| T cb_y(I nu, T x) noexcept | ||
| { | ||
| if constexpr(floating_value<I>) | ||
| { | ||
| if constexpr(floating_scalar_value<I>) | ||
| { | ||
| auto flintx = is_flint(nu); | ||
| return flintx ? kernel_bessel_y_int(nu, x) : kernel_bessel_y_flt(T(nu), x); | ||
| } | ||
| else if constexpr( scalar_value<T> ) | ||
| { | ||
| auto flintx = is_flint(nu); | ||
| return flintx ? kernel_bessel_y_int(nu, x) : kernel_bessel_y_flt(nu, x); | ||
| } | ||
| else // simd | ||
| { | ||
| auto flint_nu = is_flint(nu); | ||
| if( eve::all(flint_nu) ) return kernel_bessel_y_int(nu, x); | ||
| else if( eve::none(flint_nu) ) return kernel_bessel_y_flt(nu, x); | ||
| else | ||
| { | ||
| auto nu_int = if_else(flint_nu, nu, zero); | ||
| auto nu_flt = if_else(flint_nu, T(0.5), nu); | ||
| return if_else(flint_nu, kernel_bessel_y_int(nu_int, x), kernel_bessel_y_flt(nu_flt, x)); | ||
| } | ||
| } | ||
| } | ||
| else //if constexpr(integral_value<I>) | ||
| { | ||
| if constexpr(simd_value<I> && scalar_value<T>) | ||
| { | ||
| using c_t = wide<T, cardinal_t<I>>; | ||
| return cb_yn(nu, c_t(x)); | ||
| } | ||
| else if constexpr(scalar_value<I> && simd_value<T>) | ||
| { | ||
| using i_t = wide<I, cardinal_t<T>>; | ||
| return kernel_bessel_y_int(i_t(nu), x); | ||
| } | ||
| else if constexpr(simd_value<I> && simd_value<T>) | ||
| { | ||
| auto n = convert(nu, as_element(x)); | ||
| return kernel_bessel_y_int(n, x); | ||
| } | ||
| else return kernel_bessel_y_int(T(nu), x); | ||
| } | ||
| } | ||
|
|
||
| // T is always floating | ||
| template < typename I, typename T > constexpr EVE_FORCEINLINE | ||
| T sb_y(I n, T x) noexcept | ||
| { | ||
| using elt_t = element_type_t<T>; | ||
| if constexpr( integral_value<I> ) return sb_y(convert(n, as<elt_t>()), x); | ||
| else return cb_y(n + half(as(x)), x) * rsqrt(2 * x * inv_pi(as(x))); | ||
| } | ||
| } |
Oops, something went wrong.