From a6e39cd03a0b3970bd72d20e1e3dd40930461fdb Mon Sep 17 00:00:00 2001 From: Sandeep Kumar Date: Wed, 20 Jun 2018 06:54:57 -0700 Subject: [PATCH] use hip atomicAdd --- 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 e89bf424e30ae..cd8cb3f1046cf 100644 --- a/aten/src/THC/THCAtomics.cuh +++ b/aten/src/THC/THCAtomics.cuh @@ -135,7 +135,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) && !defined(__HIP_PLATFORM_HCC__) // This needs to be defined for the host side pass static inline __device__ void atomicAdd(double *address, double val) { } #endif