diff options
author | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-09-22 19:04:44 -0700 |
---|---|---|
committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-09-22 19:04:44 -0700 |
commit | 1301d744f8dd1215336371046bd7c68b62dac810 (patch) | |
tree | e910997ea3bad38a1ede868915053fc088da9143 /unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h | |
parent | 2a69290ddb165b7103c87ba8f5b257eca23f62aa (diff) |
Made the gaussian generator usable on GPU
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h')
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h | 2 |
1 files changed, 1 insertions, 1 deletions
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<Index, NumDims>& coordinates) const { + EIGEN_DEVICE_FUNC T operator()(const array<Index, NumDims>& coordinates) const { T tmp = T(0); for (size_t i = 0; i < NumDims; ++i) { T offset = coordinates[i] - m_means[i]; |