diff options
| author |  Deven Desai <deven.desai.amd@gmail.com> | 2020-03-11 23:06:56 +0000 |
|---|---|---|
| committer | Deven Desai <deven.desai.amd@gmail.com> | 2020-03-12 01:06:24 +0000 |
| commit | 7158ed4e0e34d40cd0f358a3bf69a5c30d8d0f83 (patch) | |
| tree | 6ee1f2ce81b3e442210564b283fdf9e953ff0306 /Eigen/src/Core/arch/Default/Half.h | |
| parent | d53ae40f7bcfb948b85b893acf305cdebcba3ba8 (diff) | |
Fixing HIP breakage caused by the recent commit that introduces Packet4h2 as the Eigen::Half packet type
Diffstat (limited to 'Eigen/src/Core/arch/Default/Half.h')
| -rw-r--r-- | Eigen/src/Core/arch/Default/Half.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/Eigen/src/Core/arch/Default/Half.h b/Eigen/src/Core/arch/Default/Half.h index 9f4c1ebdf..cfd0bdc06 100644 --- a/Eigen/src/Core/arch/Default/Half.h +++ b/Eigen/src/Core/arch/Default/Half.h @@ -706,7 +706,7 @@ struct hash<Eigen::half> { // Add the missing shfl_xor intrinsic #if (defined(EIGEN_CUDA_ARCH) && EIGEN_CUDA_ARCH >= 300) || \ - defined(EIGEN_HIP_DEVICE_COMPILE) + defined(EIGEN_HIPCC) __device__ EIGEN_STRONG_INLINE Eigen::half __shfl_xor(Eigen::half var, int laneMask, int width=warpSize) { #if (EIGEN_CUDA_SDK_VER < 90000) || \ @@ -720,7 +720,7 @@ __device__ EIGEN_STRONG_INLINE Eigen::half __shfl_xor(Eigen::half var, int laneM // ldg() has an overload for __half_raw, but we also need one for Eigen::half. #if (defined(EIGEN_CUDA_ARCH) && EIGEN_CUDA_ARCH >= 350) || \ - defined(EIGEN_HIP_DEVICE_COMPILE) + defined(EIGEN_HIPCC) EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Eigen::half __ldg(const Eigen::half* ptr) { return Eigen::half_impl::raw_uint16_to_half( __ldg(reinterpret_cast<const unsigned short*>(ptr))); |