Fix GPU build for ptanh_float.

This commit is contained in:
Antonio Sánchez
2024-02-20 16:08:50 +00:00
parent b14c5d0fa1
commit 6b365e74d6
2 changed files with 2 additions and 2 deletions

View File

@@ -939,7 +939,7 @@ EIGEN_DEFINE_FUNCTION_ALLOWING_MULTIPLE_DEFINITIONS Packet patan_double(const Pa
This implementation works on both scalars and packets.
*/
template <typename T>
T ptanh_float(const T& a_x) {
EIGEN_DEFINE_FUNCTION_ALLOWING_MULTIPLE_DEFINITIONS T ptanh_float(const T& a_x) {
// Clamp the inputs to the range [-c, c]
#ifdef EIGEN_VECTORIZE_FMA
const T plus_clamp = pset1<T>(7.99881172180175781f);