diff options
| -rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h b/unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h index 3b8ec14f0..d9400163c 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h @@ -244,6 +244,7 @@ template <typename ResScalar, typename LhsScalar, typename RhsScalar, typename StorageIndex, typename OutputMapper, typename LhsMapper, typename RhsMapper> struct TensorContractionKernel { + EIGEN_DEVICE_FUNC TensorContractionKernel(StorageIndex m, StorageIndex k, StorageIndex n, StorageIndex bm, StorageIndex bk, StorageIndex bn) : m(m), k(k), n(n), bm(bm), bk(bk), bn(bn) {} |