diff options
| author |  Deven Desai <deven.desai.amd@gmail.com> | 2019-04-08 13:45:08 +0000 |
|---|---|---|
| committer | Deven Desai <deven.desai.amd@gmail.com> | 2019-04-08 13:45:08 +0000 |
| commit | 66a885b61ee73a751cf07ba0e3c91919dc95b5b9 (patch) | |
| tree | 5bc68089273a82bd56ea1a92492d548464d850b4 /unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h | |
| parent | 629ddebd153c0c51c363a6d389aacb135719f8ed (diff) | |
adding EIGEN_DEVICE_FUNC to the recently added TensorContractionKernel constructor. Not having the EIGEN_DEVICE_FUNC attribute on it was leading to compiler errors when compiling Eigen in the ROCm/HIP path
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h')
| -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) {} |