From 12864f142073b4a280120e4d9b3abe4e2483ca32 Mon Sep 17 00:00:00 2001 From: qiaolongfei Date: Wed, 23 Aug 2017 22:51:35 -0700 Subject: [PATCH] register rowwise add gpu kernel --- paddle/operators/rowwise_add_op.cu | 3 +++ 1 file changed, 3 insertions(+) diff --git a/paddle/operators/rowwise_add_op.cu b/paddle/operators/rowwise_add_op.cu index cbc61ad3e11..4a57f64c890 100644 --- 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); +REGISTER_OP_GPU_KERNEL( + rowwise_add_grad, + ops::RowwiseAddGradKernel); -- GitLab