From e3c2d61739ebf09ad4bf86752a9878b7cf4e045f Mon Sep 17 00:00:00 2001
From: Eugene Zhulenev <ezhulenev@google.com>
Date: Wed, 18 Jul 2018 14:34:22 -0700
Subject: Assert that no output kernel is defined for GPU contraction

---
 unsupported/Eigen/CXX11/src/Tensor/TensorContractionGpu.h | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorContractionGpu.h')

diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorContractionGpu.h b/unsupported/Eigen/CXX11/src/Tensor/TensorContractionGpu.h
index 6d3aa24c8..b5e186d21 100644
--- a/unsupported/Eigen/CXX11/src/Tensor/TensorContractionGpu.h
+++ b/unsupported/Eigen/CXX11/src/Tensor/TensorContractionGpu.h
@@ -1219,6 +1219,9 @@ template<typename Indices, typename LeftArgType, typename RightArgType, typename
 struct TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgType, OutputKernelType>, GpuDevice> :
     public TensorContractionEvaluatorBase<TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgType, OutputKernelType>, GpuDevice> > {
 
+  static_assert(std::is_same<OutputKernelType, const NoOpOutputKernel>::value,
+                "GPU tensor contraction does not support output kernels.");
+
   typedef GpuDevice Device;
 
   typedef TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgType, OutputKernelType>, Device> Self;
-- 
cgit v1.2.3