diff --git a/paddle/cuda/include/hl_device_functions.cuh b/paddle/cuda/include/hl_device_functions.cuh index 408ff35d963c65bd995b688dbc76b7520647a03f..27e3f450c5c1c987f4b3fbc2eb1e52a1696c7320 100644 --- a/paddle/cuda/include/hl_device_functions.cuh +++ b/paddle/cuda/include/hl_device_functions.cuh @@ -36,6 +36,8 @@ static __inline__ __device__ double atomicAdd(double* address, double val) { } // namespace hppl +#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ < 600 using hppl::atomicAdd; +#endif #endif /* HL_DEVICE_FUNCTIONS_CUH_ */