diff --git a/paddle/fluid/operators/lookup_table_v2_op.cc b/paddle/fluid/operators/lookup_table_v2_op.cc index 47a00a93a647253305080bde2d8c98eb735513d6..48ae080783d112c7e11daebe984de70925f5bbe2 100644 --- a/paddle/fluid/operators/lookup_table_v2_op.cc +++ b/paddle/fluid/operators/lookup_table_v2_op.cc @@ -203,14 +203,6 @@ REGISTER_OPERATOR(lookup_table_v2_grad, ops::LookupTableV2OpGrad, ops::LookupTableV2GradOpNoBufferVarsInferer, ops::LookupTableV2OpGradVarTypeInference); -REGISTER_OP_CPU_KERNEL(lookup_table_v2, ops::LookupTableV2Kernel, - ops::LookupTableV2Kernel, - ops::LookupTableV2Kernel); -REGISTER_OP_CPU_KERNEL( - lookup_table_v2_grad, ops::LookupTableV2GradKernel, - ops::LookupTableV2GradKernel, - ops::LookupTableV2GradKernel); - /* ========================== register checkpoint ===========================*/ REGISTER_OP_VERSION(lookup_table_v2) .AddCheckpoint( diff --git a/paddle/fluid/operators/lookup_table_v2_op.cu b/paddle/fluid/operators/lookup_table_v2_op.cu index d40b2643785706e843dbd9812e74ca0aa134f7b5..74d089e23a82c6ea15988d9c3c0c3e5b42da8b2e 100644 --- a/paddle/fluid/operators/lookup_table_v2_op.cu +++ b/paddle/fluid/operators/lookup_table_v2_op.cu @@ -235,13 +235,3 @@ class LookupTableV2GradCUDAKernel : public framework::OpKernel { } // namespace operators } // namespace paddle - -namespace ops = paddle::operators; -namespace plat = paddle::platform; -REGISTER_OP_CUDA_KERNEL(lookup_table_v2, ops::LookupTableV2CUDAKernel, - ops::LookupTableV2CUDAKernel, - ops::LookupTableV2CUDAKernel); -REGISTER_OP_CUDA_KERNEL(lookup_table_v2_grad, - ops::LookupTableV2GradCUDAKernel, - ops::LookupTableV2GradCUDAKernel, - ops::LookupTableV2GradCUDAKernel);