diff options
author | Eugene Zhulenev <ezhulenev@google.com> | 2018-07-18 14:34:22 -0700 |
---|---|---|
committer | Eugene Zhulenev <ezhulenev@google.com> | 2018-07-18 14:34:22 -0700 |
commit | e3c2d61739ebf09ad4bf86752a9878b7cf4e045f (patch) | |
tree | 60d1fd1178ef3fa21b94fdeb416f3ea8bcacfa61 | |
parent | 086ded5c85457f813cddaec03f6a53c561f8652d (diff) |
Assert that no output kernel is defined for GPU contraction
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorContractionGpu.h | 3 |
1 files changed, 3 insertions, 0 deletions
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; |