diff --git a/paddle/fluid/operators/one_hot_v2_op_mlu.cc b/paddle/fluid/operators/one_hot_v2_op_mlu.cc index f574cc525f142b0e65bc357fe602579ff466d019..21e8975e37dcb33a7c1cd81fca6a0f8c29f8a533 100644 --- a/paddle/fluid/operators/one_hot_v2_op_mlu.cc +++ b/paddle/fluid/operators/one_hot_v2_op_mlu.cc @@ -97,4 +97,6 @@ class OneHotV2MLUKernel : public framework::OpKernel { namespace ops = paddle::operators; namespace plat = paddle::platform; -REGISTER_OP_MLU_KERNEL(one_hot_v2, ops::OneHotV2MLUKernel); +REGISTER_OP_MLU_KERNEL(one_hot_v2, + ops::OneHotV2MLUKernel, + ops::OneHotV2MLUKernel);