From 79d4129cce4055955e7a340822df31b6f8d25edc Mon Sep 17 00:00:00 2001
From: Eugene Zhulenev <ezhulenev@google.com>
Date: Wed, 18 Jul 2018 14:21:01 -0700
Subject: Specify default output kernel for TensorContractionOp

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

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

diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h b/unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h
index c7c443a59..0e69cd40c 100644
--- a/unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h
+++ b/unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h
@@ -182,7 +182,7 @@ struct NoOpOutputKernel {
       Index num_cols) const {}
 };
 
-template<typename Indices, typename LhsXprType, typename RhsXprType, typename OutputKernelType>
+template<typename Indices, typename LhsXprType, typename RhsXprType, typename OutputKernelType = const NoOpOutputKernel>
 class TensorContractionOp : public TensorBase<TensorContractionOp<Indices, LhsXprType, RhsXprType, OutputKernelType>, ReadOnlyAccessors>
 {
   public:
@@ -507,7 +507,7 @@ struct TensorContractionEvaluatorBase
   }
 
   template <bool lhs_inner_dim_contiguous, bool rhs_inner_dim_contiguous, bool rhs_inner_dim_reordered, int Alignment>
-  #if !defined(EIGEN_HIPCC)    
+  #if !defined(EIGEN_HIPCC)
   EIGEN_DEVICE_FUNC
   #endif
   void evalGemv(Scalar* buffer) const {
@@ -556,7 +556,7 @@ struct TensorContractionEvaluatorBase
   }
 
   template <bool lhs_inner_dim_contiguous, bool rhs_inner_dim_contiguous, bool rhs_inner_dim_reordered, int Alignment>
-  #if !defined(EIGEN_HIPCC)    
+  #if !defined(EIGEN_HIPCC)
   EIGEN_DEVICE_FUNC
   #endif
   void evalGemm(Scalar* buffer) const {
-- 
cgit v1.2.3