Merge pull request #3660 from jacquesqiao/rowwise_add_gpu

register rowwise add gpu kernel

Merge pull request #3660 from jacquesqiao/rowwise_add_gpu
register rowwise add gpu kernel
8a63a8ab · Qiao Longfei · GitHub · 26c473ad · 12864f14 · 8a63a8ab
显示空白变更内容
内联并排

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>);