add backends/iluvatar_gpu/kernels/cuda_kernels/collect_fpn_proposals_kernel_register.cu

Author: youge325

backends/iluvatar_gpu/kernels/cuda_kernels/collect_fpn_proposals_kernel_register.cu

@@ -0,0 +1,25 @@
+// Copyright (c) 2025 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "paddle/phi/kernels/gpu/collect_fpn_proposals_kernel.h"
+
+PD_CUSTOM_KERNEL_REGISTER(collect_fpn_proposals,
+                          iluvatar_gpu,
+                          ALL_LAYOUT,
+                          phi::GPUCollectFpnProposalsOpKernel,
+                          float,
+                          double) {
+  kernel->InputAt(2).SetDataType(phi::DataType::INT32);
+  kernel->OutputAt(1).SetDataType(phi::DataType::INT32);
+}