From 66a885b61ee73a751cf07ba0e3c91919dc95b5b9 Mon Sep 17 00:00:00 2001
From: Deven Desai <deven.desai.amd@gmail.com>
Date: Mon, 8 Apr 2019 13:45:08 +0000
Subject: 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

---
 unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h | 1 +
 1 file changed, 1 insertion(+)

(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 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) {}
-- 
cgit v1.2.3