From 4319b2953fbc41615b746dc135915bcaef72328f Mon Sep 17 00:00:00 2001 From: Yaxun Sam Liu Date: Fri, 7 Sep 2018 15:50:05 -0400 Subject: [PATCH] Fix atomicAdd for hip-clang --- aten/src/THC/THCAtomics.cuh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aten/src/THC/THCAtomics.cuh b/aten/src/THC/THCAtomics.cuh index 485b5744f15255..c5c1766cb46ac7 100644 --- a/aten/src/THC/THCAtomics.cuh +++ b/aten/src/THC/THCAtomics.cuh @@ -137,7 +137,7 @@ static inline __device__ void atomicAdd(double *address, double val) { // Note: uses integer comparison to avoid hang in case of NaN (since NaN != NaN) } while (assumed != old); } -#elif !defined(__CUDA_ARCH__) && (CUDA_VERSION < 8000) || defined(__HIP_PLATFORM_HCC__) +#elif !defined(__CUDA_ARCH__) && (CUDA_VERSION < 8000) && !__HIP__ || defined(__HCC__) // This needs to be defined for the host side pass static inline __device__ void atomicAdd(double *address, double val) { } #endif