Fix placement of "#if defined(EIGEN_GPUCC)" guard region.

Found with -Wundefined-func-template.

Author: tkoeppe@google.com
This commit is contained in:
Rasmus Munk Larsen 2019-03-06 11:40:06 -08:00
parent 4d808e834a
commit cc407c9d4d

View File

@ -317,6 +317,7 @@ class TensorExecutor<Expression, ThreadPoolDevice, Vectorizable, /*Tileable*/ tr
// GPU: the evaluation of the expression is offloaded to a GPU.
#if defined(EIGEN_USE_GPU)
#if defined(EIGEN_GPUCC)
template <typename Expression, bool Vectorizable, bool Tileable>
class TensorExecutor<Expression, GpuDevice, Vectorizable, Tileable> {
@ -326,7 +327,6 @@ class TensorExecutor<Expression, GpuDevice, Vectorizable, Tileable> {
};
#if defined(EIGEN_GPUCC)
template <typename Evaluator, typename StorageIndex, bool Vectorizable>
struct EigenMetaKernelEval {
static __device__ EIGEN_ALWAYS_INLINE