diff options
author | Deven Desai <deven.desai.amd@gmail.com> | 2018-07-17 14:16:48 -0400 |
---|---|---|
committer | Deven Desai <deven.desai.amd@gmail.com> | 2018-07-17 14:16:48 -0400 |
commit | f124f0796533081cb7b061c4aa8667df6ed58863 (patch) | |
tree | 20efd29fdfaf3da79cad39f468d330927b62ebc2 /unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h | |
parent | 82f0ce27261df3b21037d93d4595655b3df754a6 (diff) |
applying EIGEN_DECLARE_TEST to *gpu* tests
Also, a few minor fixes for GPU tests running in HIP mode.
1. Adding an include for hip/hip_runtime.h in the Macros.h file
For HIP __host__ and __device__ are macros which are defined in hip headers.
Their definitions need to be included before their use in the file.
2. Fixing the compile failure in TensorContractionGpu introduced by the commit to
"Fuse computations into the Tensor contractions using output kernel"
3. Fixing a HIP/clang specific compile error by making the struct-member assignment explicit
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h')
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h b/unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h index 0fbffa34c..c7c443a59 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h @@ -449,8 +449,7 @@ struct TensorContractionEvaluatorBase // tensor dimensions (i, j) into the original tensor dimensions. // TODO(ezhulenev): Add parameters required to infer output tensor index for // more complex contractions than 2x2 on internal dimension. - m_tensor_contraction_params = { - /**swapped_arguments=*/static_cast<int>(Layout) == RowMajor}; + m_tensor_contraction_params.swapped_arguments = static_cast<int>(Layout) == RowMajor; } EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; } |