diff options
| author |  Benoit Steiner <benoit.steiner.goog@gmail.com> | 2015-02-10 14:29:47 -0800 |
|---|---|---|
| committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2015-02-10 14:29:47 -0800 |
| commit | f669f5656ab550010c5dd92ce2da7d3fab07babd (patch) | |
| tree | 66a0fdee870f1dd927dc1515717a8f6173db146e /unsupported/Eigen/CXX11/src/Tensor/Tensor.h | |
| parent | fefec723aa44703c1b7884b2ccfa73877a58f500 (diff) | |
Marked a few functions as EIGEN_DEVICE_FUNC to enable the use of tensors in cuda kernels.
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/Tensor.h')
| -rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/Tensor.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/Tensor.h b/unsupported/Eigen/CXX11/src/Tensor/Tensor.h index 0e8a4b8d6..037219f23 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/Tensor.h +++ b/unsupported/Eigen/CXX11/src/Tensor/Tensor.h @@ -350,7 +350,7 @@ class Tensor : public TensorBase<Tensor<Scalar_, NumIndices_, Options_> > } #endif - void resize(const array<Index, NumIndices>& dimensions) + EIGEN_DEVICE_FUNC void resize(const array<Index, NumIndices>& dimensions) { std::size_t i; Index size = Index(1); @@ -367,7 +367,7 @@ class Tensor : public TensorBase<Tensor<Scalar_, NumIndices_, Options_> > #endif } - void resize(const DSizes<Index, NumIndices>& dimensions) { + EIGEN_DEVICE_FUNC void resize(const DSizes<Index, NumIndices>& dimensions) { array<Index, NumIndices> dims; for (std::size_t i = 0; i < NumIndices; ++i) { dims[i] = dimensions[i]; |