GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-09-13 18:03:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Don't use the rational approximation to the logistic function on GPUs as it appears to be slightly slower.

Browse Source
This commit is contained in:
Rasmus Munk Larsen 2020-01-09 00:04:26 +00:00
parent 4217a9f090
commit e6fcee995b
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Eigen/src/Core/functors/UnaryFunctors.h
View File

@ -924,6 +924,7 @@ struct scalar_logistic_op {
} }
}; };
#ifndef EIGEN_GPU_COMPILE_PHASE
/** \internal /** \internal
* \brief Template specialization of the logistic function for float. * \brief Template specialization of the logistic function for float.
* *
@ -1007,6 +1008,7 @@ struct scalar_logistic_op<float> {
} }
} }
}; };
#endif // #ifndef EIGEN_GPU_COMPILE_PHASE
template <typename T> template <typename T>
struct functor_traits<scalar_logistic_op<T> > { struct functor_traits<scalar_logistic_op<T> > {