Skip to content

Commit cf85f6b

Browse files
authored
Enable atomicAddNoRet() for all gfx targets (#992)
1 parent 539d476 commit cf85f6b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

aten/src/THC/THCAtomics.cuh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -301,7 +301,7 @@ static inline __device__ void gpuAtomicAddNoReturn(at::BFloat16 *address, at::BF
301301
static inline __device__ void gpuAtomicAddNoReturn(double *address, double val) { gpuAtomicAdd(address, val); }
302302

303303
/* Special case fp32 atomic. */
304-
#if defined(__HIP_PLATFORM_HCC__) && defined(__gfx908__)
304+
#if defined(__HIP_PLATFORM_HCC__)
305305
static inline __device__ void gpuAtomicAddNoReturn(float *address, float val) { atomicAddNoRet(address, val); }
306306
#else
307307
static inline __device__ void gpuAtomicAddNoReturn(float *address, float val) { gpuAtomicAdd(address, val); }

0 commit comments

Comments
 (0)