From a004dfde3d81b1aa7c4721203266147338a0b2ea Mon Sep 17 00:00:00 2001
From: whs <wanghaoshuang@baidu.com>
Date: Tue, 25 Aug 2020 12:51:49 +0800
Subject: [PATCH] Use atomicAdd defined in paddle fromework (#26631)

test=develop
---
 paddle/fluid/operators/affine_grid_op.cu | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/paddle/fluid/operators/affine_grid_op.cu b/paddle/fluid/operators/affine_grid_op.cu
index eca8246533f..66e3c741e5d 100644
--- a/paddle/fluid/operators/affine_grid_op.cu
+++ b/paddle/fluid/operators/affine_grid_op.cu
@@ -15,7 +15,9 @@ limitations under the License. */
 #include "paddle/fluid/framework/op_registry.h"
 #include "paddle/fluid/operators/affine_grid_op.h"
 #include "paddle/fluid/platform/cuda_device_function.h"
+#include "paddle/fluid/platform/cuda_primitives.h"
 #include "paddle/fluid/platform/gpu_info.h"
+
 namespace paddle {
 namespace operators {
 
-- 
GitLab