From 76dd89fdc4ce29221c0107b33e31676b8613c5d0 Mon Sep 17 00:00:00 2001 From: youge325 Date: Fri, 3 Oct 2025 19:56:27 +0800 Subject: [PATCH] =?UTF-8?q?=E3=80=90CUDA=20Kernel=20No.21=E3=80=91ap=5Ffac?= =?UTF-8?q?ade=E7=AE=97=E5=AD=90Kernel=E4=BF=AE=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- paddle/phi/kernels/gpu/ap_facade_kernel.cu | 1 + paddle/phi/kernels/gpu/ap_facade_kernel.h | 33 ++++++++++++++++++++++ 2 files changed, 34 insertions(+) create mode 100644 paddle/phi/kernels/gpu/ap_facade_kernel.h diff --git a/paddle/phi/kernels/gpu/ap_facade_kernel.cu b/paddle/phi/kernels/gpu/ap_facade_kernel.cu index 42e045646aa245..5151c93e2b2184 100644 --- a/paddle/phi/kernels/gpu/ap_facade_kernel.cu +++ b/paddle/phi/kernels/gpu/ap_facade_kernel.cu @@ -12,6 +12,7 @@ // See the License for the specific language governing permissions and // limitations under the License. +#include "paddle/phi/kernels/gpu/ap_facade_kernel.h" #include "paddle/common/enforce.h" #include "paddle/phi/backends/gpu/gpu_context.h" #include "paddle/phi/core/dense_tensor.h" diff --git a/paddle/phi/kernels/gpu/ap_facade_kernel.h b/paddle/phi/kernels/gpu/ap_facade_kernel.h new file mode 100644 index 00000000000000..c1c016e7abfa9c --- /dev/null +++ b/paddle/phi/kernels/gpu/ap_facade_kernel.h @@ -0,0 +1,33 @@ +// 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. + +#pragma once + +#include +#include +#include "paddle/phi/core/dense_tensor.h" + +namespace phi { + +template +void ApFacadeKernel(const Context& dev_ctx, + const paddle::optional>& xs, + int64_t num_outputs, + const std::string& custom_op_name, + const std::string& infer_meta_func_name, + const std::string& infer_symbolic_func_name, + const std::string& serialized_attributes, + std::vector outs); + +} // namespace phi