| // SPDX-FileCopyrightText: The Eigen Authors |
| // SPDX-License-Identifier: MPL-2.0 |
| |
| #ifndef EIGEN_NEON_BESSELFUNCTIONS_H |
| #define EIGEN_NEON_BESSELFUNCTIONS_H |
| |
| namespace Eigen { |
| namespace internal { |
| |
| #if EIGEN_ARCH_ARM64 && EIGEN_HAS_ARM64_FP16 |
| |
| #define NEON_HALF_TO_FLOAT_FUNCTIONS(METHOD) \ |
| template <> \ |
| EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet8hf METHOD<Packet8hf>(const Packet8hf& x) { \ |
| const Packet4f lo = METHOD<Packet4f>(vcvt_f32_f16(vget_low_f16(x))); \ |
| const Packet4f hi = METHOD<Packet4f>(vcvt_f32_f16(vget_high_f16(x))); \ |
| return vcombine_f16(vcvt_f16_f32(lo), vcvt_f16_f32(hi)); \ |
| } \ |
| \ |
| template <> \ |
| EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet4hf METHOD<Packet4hf>(const Packet4hf& x) { \ |
| return vcvt_f16_f32(METHOD<Packet4f>(vcvt_f32_f16(x))); \ |
| } |
| |
| NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_i0) |
| NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_i0e) |
| NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_i1) |
| NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_i1e) |
| NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_j0) |
| NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_j1) |
| NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_k0) |
| NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_k0e) |
| NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_k1) |
| NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_k1e) |
| NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_y0) |
| NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_y1) |
| |
| #undef NEON_HALF_TO_FLOAT_FUNCTIONS |
| #endif // EIGEN_ARCH_ARM64 && EIGEN_HAS_ARM64_FP16 |
| |
| EIGEN_INSTANTIATE_BESSEL_FUNCS_BF16(Packet4f, Packet4bf) |
| |
| } // namespace internal |
| } // namespace Eigen |
| |
| #endif // EIGEN_NEON_BESSELFUNCTIONS_H |
