Use atomicAdd defined in paddle framework (#26628)

test=develop
5 years ago · 786373ba29
parent 1f82c0cd62
commit 786373ba29
1 changed files with 1 additions and 0 deletions
--- 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 {