From 1301d744f8dd1215336371046bd7c68b62dac810 Mon Sep 17 00:00:00 2001 From: Benoit Steiner Date: Thu, 22 Sep 2016 19:04:44 -0700 Subject: Made the gaussian generator usable on GPU --- unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h') diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h b/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h index 1ba8d6328..760074622 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h @@ -895,7 +895,7 @@ class GaussianGenerator { } } - T operator()(const array& coordinates) const { + EIGEN_DEVICE_FUNC T operator()(const array& coordinates) const { T tmp = T(0); for (size_t i = 0; i < NumDims; ++i) { T offset = coordinates[i] - m_means[i]; -- cgit v1.2.3