diff --git a/paddle/phi/kernels/collect_fpn_proposals_kernel.h b/paddle/phi/kernels/collect_fpn_proposals_kernel.h new file mode 100644 index 0000000000000..401e577c2985a --- /dev/null +++ b/paddle/phi/kernels/collect_fpn_proposals_kernel.h @@ -0,0 +1,31 @@ +// Copyright (c) 2024 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. + +#pragma once + +#include +#include "paddle/phi/core/dense_tensor.h" + +namespace phi { +template +void GPUCollectFpnProposalsOpKernel( + const Context& dev_ctx, + const std::vector& multi_level_rois, + const std::vector& multi_level_scores, + const paddle::optional>& + multi_level_rois_num, + int post_nms_topn, + DenseTensor* fpn_rois_out, + DenseTensor* rois_num_out); +} // namespace phi diff --git a/paddle/phi/kernels/cpu/collect_fpn_proposals_kernel.cc b/paddle/phi/kernels/cpu/collect_fpn_proposals_kernel.cc index 31595b21090e6..54a4150ef7a63 100644 --- a/paddle/phi/kernels/cpu/collect_fpn_proposals_kernel.cc +++ b/paddle/phi/kernels/cpu/collect_fpn_proposals_kernel.cc @@ -12,6 +12,7 @@ // See the License for the specific language governing permissions and // limitations under the License. +#include "paddle/phi/kernels/collect_fpn_proposals_kernel.h" #include "paddle/phi/core/kernel_registry.h" #include "paddle/phi/kernels/impl/collect_fpn_proposals_kernel_impl.h" diff --git a/paddle/phi/kernels/gpu/collect_fpn_proposals_kernel.cu b/paddle/phi/kernels/gpu/collect_fpn_proposals_kernel.cu index f75694421d6a8..d39275ba3fc13 100644 --- a/paddle/phi/kernels/gpu/collect_fpn_proposals_kernel.cu +++ b/paddle/phi/kernels/gpu/collect_fpn_proposals_kernel.cu @@ -25,6 +25,7 @@ namespace cub = hipcub; #include "paddle/phi/core/allocator.h" #include "paddle/phi/core/kernel_registry.h" #include "paddle/phi/core/mixed_vector.h" +#include "paddle/phi/kernels/collect_fpn_proposals_kernel.h" #include "paddle/phi/kernels/funcs/concat_and_split_functor.h" #include "paddle/phi/kernels/funcs/detection/bbox_util.h" #include "paddle/phi/kernels/funcs/for_range.h"