devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Made the gaussian generator usable on GPU

Browse Source
This commit is contained in:
Benoit Steiner
2016-09-22 19:04:44 -07:00
parent 2a69290ddb
commit 1301d744f8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h
View File

@@ -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];