GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-11 19:29:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Deven Desai 2019-04-08 13:45:08 +00:00
parent 629ddebd15
commit 66a885b61e
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h
View File

@ -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) {}