update

7 years ago · 948628563f
parent 04c559e3aa
commit 948628563f
1 changed files with 2 additions and 1 deletions
--- a/paddle/fluid/operators/split_byref_op.cu.cc
+++ b/paddle/fluid/operators/split_byref_op.cu.cc
@ -15,4 +15,5 @@ limitations under the License. */
 #include "paddle/fluid/operators/split_byref_op.h"
 namespace ops = paddle::operators;
 REGISTER_OP_CUDA_KERNEL(
-    split, ops::SplitByrefOpKernel<paddle::platform::CUDADeviceContext, float>);
+    split_byref,
+    ops::SplitByrefOpKernel<paddle::platform::CUDADeviceContext, float>);