adding EIGEN_DEVICE_FUNC attribute to fix some GPU unit tests that are broken in HIP mode

This commit is contained in:
Deven Desai 2018-08-27 23:04:08 +00:00
parent 7ec8b40ad9
commit 946c3e2544

View File

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