mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-05-01 08:14:10 +08:00
Made it possible to run the scalar_binary_pow_op functor on GPU
This commit is contained in:
parent
c569cfe12a
commit
3628f7655d
@ -496,7 +496,7 @@ template<typename Scalar, bool IsInteger>
|
|||||||
struct pow_default_impl
|
struct pow_default_impl
|
||||||
{
|
{
|
||||||
typedef Scalar retval;
|
typedef Scalar retval;
|
||||||
static inline Scalar run(const Scalar& x, const Scalar& y)
|
static EIGEN_DEVICE_FUNC inline Scalar run(const Scalar& x, const Scalar& y)
|
||||||
{
|
{
|
||||||
EIGEN_USING_STD_MATH(pow);
|
EIGEN_USING_STD_MATH(pow);
|
||||||
return pow(x, y);
|
return pow(x, y);
|
||||||
@ -506,7 +506,7 @@ struct pow_default_impl
|
|||||||
template<typename Scalar>
|
template<typename Scalar>
|
||||||
struct pow_default_impl<Scalar, true>
|
struct pow_default_impl<Scalar, true>
|
||||||
{
|
{
|
||||||
static inline Scalar run(Scalar x, Scalar y)
|
static EIGEN_DEVICE_FUNC inline Scalar run(Scalar x, Scalar y)
|
||||||
{
|
{
|
||||||
Scalar res(1);
|
Scalar res(1);
|
||||||
eigen_assert(!NumTraits<Scalar>::IsSigned || y >= 0);
|
eigen_assert(!NumTraits<Scalar>::IsSigned || y >= 0);
|
||||||
|
Loading…
x
Reference in New Issue
Block a user