Skip to content

Commit 43206ac

Browse files
committed
Call OutputKernel in evalGemv
1 parent e204ecd commit 43206ac

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -549,6 +549,11 @@ struct TensorContractionEvaluatorBase
549549
internal::general_matrix_vector_product<Index,LhsScalar,LhsMapper,ColMajor,false,RhsScalar,RhsMapper,false>::run(
550550
rows, cols, lhs, rhs,
551551
buffer, resIncr, alpha);
552+
553+
typedef internal::blas_data_mapper<Scalar, Index, ColMajor> OutputMapper;
554+
m_output_kernel(OutputMapper(buffer, rows), m_tensor_contraction_params,
555+
static_cast<Index>(0), static_cast<Index>(0), rows,
556+
static_cast<Index>(1));
552557
}
553558

554559
template <bool lhs_inner_dim_contiguous, bool rhs_inner_dim_contiguous, bool rhs_inner_dim_reordered, int Alignment>

0 commit comments

Comments
 (0)