Use atomicAdd defined in paddle framework

test=develop
PaddlePaddle · Aug 24, 2020 · b68a2cd · b68a2cd · paddle-bot-old · Aug 24, 2020
1 parent 924734f
commit b68a2cd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/paddle/fluid/operators/grid_sampler_op.cu b/paddle/fluid/operators/grid_sampler_op.cu
@@ -16,6 +16,7 @@ limitations under the License. */
 #include "paddle/fluid/framework/op_registry.h"
 #include "paddle/fluid/operators/grid_sampler_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 {