mirror of
https://github.com/boostorg/math.git
synced 2026-01-19 04:22:09 +00:00
Fix CUDA support with logistic distribution
This commit is contained in:
@@ -15,7 +15,7 @@ namespace boost {
|
|||||||
namespace math {
|
namespace math {
|
||||||
|
|
||||||
template <typename RealType, typename Policy>
|
template <typename RealType, typename Policy>
|
||||||
RealType logistic_sigmoid(RealType x, const Policy&)
|
BOOST_MATH_GPU_ENABLED RealType logistic_sigmoid(RealType x, const Policy&)
|
||||||
{
|
{
|
||||||
BOOST_MATH_STD_USING
|
BOOST_MATH_STD_USING
|
||||||
|
|
||||||
@@ -35,7 +35,7 @@ RealType logistic_sigmoid(RealType x, const Policy&)
|
|||||||
}
|
}
|
||||||
|
|
||||||
template <typename RealType>
|
template <typename RealType>
|
||||||
RealType logistic_sigmoid(RealType x)
|
BOOST_MATH_GPU_ENABLED RealType logistic_sigmoid(RealType x)
|
||||||
{
|
{
|
||||||
return logistic_sigmoid(x, policies::policy<>());
|
return logistic_sigmoid(x, policies::policy<>());
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -17,7 +17,7 @@ namespace boost {
|
|||||||
namespace math {
|
namespace math {
|
||||||
|
|
||||||
template <typename RealType, typename Policy>
|
template <typename RealType, typename Policy>
|
||||||
RealType logit(RealType p, const Policy&)
|
BOOST_MATH_GPU_ENABLED RealType logit(RealType p, const Policy&)
|
||||||
{
|
{
|
||||||
BOOST_MATH_STD_USING
|
BOOST_MATH_STD_USING
|
||||||
using std::atanh;
|
using std::atanh;
|
||||||
@@ -45,7 +45,7 @@ RealType logit(RealType p, const Policy&)
|
|||||||
}
|
}
|
||||||
|
|
||||||
template <typename RealType>
|
template <typename RealType>
|
||||||
RealType logit(RealType p)
|
BOOST_MATH_GPU_ENABLED RealType logit(RealType p)
|
||||||
{
|
{
|
||||||
return logit(p, policies::policy<>());
|
return logit(p, policies::policy<>());
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user