From 5cb18e5b5eac0fa18cd91bee117696540d0a063e Mon Sep 17 00:00:00 2001 From: Benoit Steiner Date: Wed, 11 Nov 2015 14:36:33 -0800 Subject: [PATCH] Fixed CUDA compilation errors --- unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h | 2 +- unsupported/Eigen/CXX11/src/Tensor/TensorReduction.h | 6 ++++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h index 95fc9fec6..bc06ca1f0 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h @@ -229,7 +229,7 @@ inline void TensorExecutor::run(const Expression& /*static*/ template -inline void TensorExecutor::run(const Expression& expr, const GpuDevice& device) +inline void TensorExecutor::run(const Expression& expr, const GpuDevice& device) { TensorEvaluator evaluator(expr, device); const bool needs_assign = evaluator.evalSubExprsIfNeeded(NULL); diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorReduction.h b/unsupported/Eigen/CXX11/src/Tensor/TensorReduction.h index d4e88fabd..c1cdb98a4 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorReduction.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorReduction.h @@ -336,6 +336,12 @@ struct FullReducer { }; #endif + +#if defined(EIGEN_USE_GPU) && defined(__CUDACC__) +template +__global__ void FullReductionKernel(R, const S, I, typename S::CoeffReturnType*); +#endif + } // end namespace internal