Fix unittest

8 years ago · 122e83e36c
parent 73a9f0f25d
commit 122e83e36c
1 changed files with 2 additions and 1 deletions
--- a/paddle/operators/rowwise_add_op.cu
+++ b/paddle/operators/rowwise_add_op.cu
@ -2,4 +2,5 @@
 #include <paddle/operators/rowwise_add_op.h>

 REGISTER_OP_GPU_KERNEL(
-    mul, paddle::operators::RowWiseAddKernel<paddle::platform ::GPUPlace>);
+    rowwise_add,
+    paddle::operators::RowWiseAddKernel<paddle::platform ::GPUPlace>);