diff options
| author |  Rasmus Munk Larsen <rmlarsen@google.com> | 2019-12-03 12:08:22 -0800 |
|---|---|---|
| committer | Rasmus Munk Larsen <rmlarsen@google.com> | 2019-12-03 12:08:22 -0800 |
| commit | ead81559c85d6ec7903a47d0c34867f9a87781d7 (patch) | |
| tree | 94f811133885c26df000acae7b1ea240cc03e3a2 /unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h | |
| parent | 6358599ecbe23a8ba1829dea21e532aa85aab932 (diff) | |
Use EIGEN_DEVICE_FUNC macro instead of __device__.
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h')
| -rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h index b83174ab7..ca056e96e 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h @@ -582,7 +582,7 @@ class TensorExecutor<Expression, GpuDevice, Vectorizable, Tiling> { #if defined(EIGEN_GPUCC) template <typename Evaluator, typename StorageIndex, bool Vectorizable> struct EigenMetaKernelEval { - static __device__ EIGEN_ALWAYS_INLINE + static EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void run(Evaluator& eval, StorageIndex firstIdx, StorageIndex lastIdx, StorageIndex step_size) { for (StorageIndex i = firstIdx; i < lastIdx; i += step_size) { eval.evalScalar(i); @@ -592,7 +592,7 @@ struct EigenMetaKernelEval { template <typename Evaluator, typename StorageIndex> struct EigenMetaKernelEval<Evaluator, StorageIndex, true> { - static __device__ EIGEN_ALWAYS_INLINE + static EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void run(Evaluator& eval, StorageIndex firstIdx, StorageIndex lastIdx, StorageIndex step_size) { const StorageIndex PacketSize = unpacket_traits<typename Evaluator::PacketReturnType>::size; const StorageIndex vectorized_size = (lastIdx / PacketSize) * PacketSize; |