register rowwise add gpu kernel

12864f14 · qiaolongfei · 26c473ad · 12864f14
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

paddle/operators/rowwise_add_op.cu paddle/operators/rowwise_add_op.cu +3 -0

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