add register of VITERBI_DECODE (#51318)

paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc

@@ -105,7 +105,6 @@ static std::set<std::string> OpsNeedSetOutputDtypeWhenRegisterPhiKernel = {
     "unique",
     "unique_consecutive_flattened_tensor",
     "unique_raw",
-    "viterbi_decode",
     "viterbi_devode",
     "yolo_loss"};
 

paddle/phi/kernels/cpu/viterbi_decode_kernel.cc

@@ -318,4 +318,6 @@ void ViterbiDecodeKernel(const Context& dev_ctx,
 }  // namespace phi
 
 PD_REGISTER_KERNEL(
-    viterbi_decode, CPU, ALL_LAYOUT, phi::ViterbiDecodeKernel, float, double) {}
+    viterbi_decode, CPU, ALL_LAYOUT, phi::ViterbiDecodeKernel, float, double) {
+  kernel->OutputAt(1).SetDataType(paddle::experimental::DataType::INT64);
+}

paddle/phi/kernels/gpu/viterbi_decode_kernel.cu

@@ -397,4 +397,6 @@ void ViterbiDecodeKernel(const Context& dev_ctx,
 }  // namespace phi
 
 PD_REGISTER_KERNEL(
-    viterbi_decode, GPU, ALL_LAYOUT, phi::ViterbiDecodeKernel, float, double) {}
+    viterbi_decode, GPU, ALL_LAYOUT, phi::ViterbiDecodeKernel, float, double) {
+  kernel->OutputAt(1).SetDataType(paddle::experimental::DataType::INT64);
+}