Use atomicAdd defined in paddle fromework (#26631)

test=develop

Use atomicAdd defined in paddle fromework (#26631)
test=develop
a004dfde · whs · GitHub · 02fc1fef · a004dfde
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

paddle/fluid/operators/affine_grid_op.cu paddle/fluid/operators/affine_grid_op.cu +2 -0

未找到文件。
--- 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 {