Merge pull request #3660 from jacquesqiao/rowwise_add_gpu

register rowwise add gpu kernel
8 years ago · 8a63a8ab6f
parent 26c473ad1d 12864f1420
commit 8a63a8ab6f
1 changed files with 3 additions and 0 deletions
--- a/paddle/operators/rowwise_add_op.cu
+++ b/paddle/operators/rowwise_add_op.cu
@ -18,3 +18,6 @@
 namespace ops = paddle::operators;
 REGISTER_OP_GPU_KERNEL(
    rowwise_add, ops::RowwiseAddKernel<paddle::platform::GPUPlace, float>);
+REGISTER_OP_GPU_KERNEL(
+    rowwise_add_grad,
+    ops::RowwiseAddGradKernel<paddle::platform::GPUPlace, float>);