aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorGravatar Eugene Zhulenev <ezhulenev@google.com>2019-03-11 09:43:33 -0700
committerGravatar Eugene Zhulenev <ezhulenev@google.com>2019-03-11 09:43:33 -0700
commit001f10e3c94608d74898e8911a8a42d3f0640a1a (patch)
tree3cd9d8d945ba322ec51769888f9e3d3a470e7c4f
parent899c16fa2cec396a7d75dff020f6f755834f5961 (diff)
Fix segfaults with cuda compilation
-rw-r--r--unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h3
1 files changed, 1 insertions, 2 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h
index 057e90e50..e2ff11129 100644
--- a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h
+++ b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h
@@ -317,7 +317,6 @@ 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 +325,7 @@ class TensorExecutor<Expression, GpuDevice, Vectorizable, Tileable> {
static void run(const Expression& expr, const GpuDevice& device);
};
-
+#if defined(EIGEN_GPUCC)
template <typename Evaluator, typename StorageIndex, bool Vectorizable>
struct EigenMetaKernelEval {
static __device__ EIGEN_ALWAYS_INLINE