mirror of
https://gitlab.com/libeigen/eigen.git
synced 2026-04-10 11:34:33 +08:00
Don't use the rational approximation to the logistic function on GPUs as it appears to be slightly slower.
This commit is contained in:
@@ -924,6 +924,7 @@ struct scalar_logistic_op {
|
||||
}
|
||||
};
|
||||
|
||||
#ifndef EIGEN_GPU_COMPILE_PHASE
|
||||
/** \internal
|
||||
* \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>
|
||||
struct functor_traits<scalar_logistic_op<T> > {
|
||||
|
||||
Reference in New Issue
Block a user