diff options
author | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2014-10-09 15:39:13 -0700 |
---|---|---|
committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2014-10-09 15:39:13 -0700 |
commit | 498b7eed25bdb3b90f2fc45dd822c96aa08db2f8 (patch) | |
tree | 869cdd56d821bb2cc459346e24fabcd56ac69358 | |
parent | 767424af18a55604496f38dd4593542db97240a1 (diff) |
Rewrote the TensorBase::random method to support the generation of random number on gpu.
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorBase.h | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h b/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h index 90a9cc2c4..d4b7846a0 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h @@ -39,9 +39,14 @@ class TensorBase<Derived, ReadOnlyAccessors> } EIGEN_DEVICE_FUNC - EIGEN_STRONG_INLINE const TensorCwiseNullaryOp<internal::scalar_random_op<Scalar>, const Derived> + EIGEN_STRONG_INLINE const TensorCwiseNullaryOp<internal::UniformRandomGenerator<Scalar>, const Derived> random() const { - return TensorCwiseNullaryOp<internal::scalar_random_op<Scalar>, const Derived>(derived()); + return TensorCwiseNullaryOp<internal::UniformRandomGenerator<Scalar>, const Derived>(derived()); + } + template <typename RandomGenerator> EIGEN_DEVICE_FUNC + EIGEN_STRONG_INLINE const TensorCwiseNullaryOp<RandomGenerator, const Derived> + random() const { + return TensorCwiseNullaryOp<RandomGenerator, const Derived>(derived()); } // Coefficient-wise unary operators |