Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 1 年 前同步成功

通知 2299
Star 20931
Fork 5422
P
Paddle
提交 c9dc794e 编写于 作者: F Fan Yang 提交者： Yu Yang
浏览文件
操作

fixed build issue of double definition of atomicAdd on modern GPUs

上级 0b02a221
隐藏空白更改
内联 并排
Showing with 4 addition and 0 deletion
paddle/cuda/include/hl_device_functions.cuh 100644 → 100755
浏览文件 @ c9dc794e
...@@ -16,6 +16,8 @@ limitations under the License. */ ...@@ -16,6 +16,8 @@ limitations under the License. */
#ifndef HL_DEVICE_FUNCTIONS_CUH_ #ifndef HL_DEVICE_FUNCTIONS_CUH_
#define HL_DEVICE_FUNCTIONS_CUH_ #define HL_DEVICE_FUNCTIONS_CUH_
#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ < 600
namespace hppl { namespace hppl {
static __inline__ __device__ double atomicAdd(double* address, double val) { static __inline__ __device__ double atomicAdd(double* address, double val) {
...@@ -40,4 +42,6 @@ static __inline__ __device__ double atomicAdd(double* address, double val) { ...@@ -40,4 +42,6 @@ static __inline__ __device__ double atomicAdd(double* address, double val) {
using hppl::atomicAdd; using hppl::atomicAdd;
#endif #endif
#endif
#endif /* HL_DEVICE_FUNCTIONS_CUH_ */ #endif /* HL_DEVICE_FUNCTIONS_CUH_ */
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或