diff options
author | Deven Desai <deven.desai.amd@gmail.com> | 2018-08-27 23:04:08 +0000 |
---|---|---|
committer | Deven Desai <deven.desai.amd@gmail.com> | 2018-08-27 23:04:08 +0000 |
commit | 946c3e2544da35ed0539e433cc906c93c9bda3ed (patch) | |
tree | 9e87a7711fc266ebd8b6580e162db2cfb57f8e83 /unsupported/Eigen/CXX11/src/Tensor/TensorBlock.h | |
parent | 7ec8b40ad90bcc63d36fc5a36b517d4a07c60b4b (diff) |
adding EIGEN_DEVICE_FUNC attribute to fix some GPU unit tests that are broken in HIP mode
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorBlock.h')
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorBlock.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorBlock.h b/unsupported/Eigen/CXX11/src/Tensor/TensorBlock.h index 4cb49e9b8..9b92b6179 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorBlock.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorBlock.h @@ -79,7 +79,7 @@ struct TensorOpResourceRequirements { }; // Tries to merge multiple resource requirements. -EIGEN_STRONG_INLINE void MergeResourceRequirements( +EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void MergeResourceRequirements( const std::vector<TensorOpResourceRequirements>& resources, TensorBlockShapeType* block_shape, Index* block_total_size) { if (resources.empty()) { |