From c37aeeba63b10ca0f60d10fbe90f8e2e2d5e4ad3 Mon Sep 17 00:00:00 2001 From: Le-soleile <3516093767@qq.com> Date: Sun, 28 Sep 2025 07:09:13 +0000 Subject: [PATCH] Change include --- .../kernels/cuda_kernels/set_value_kernel_register.cc | 2 +- .../kernels/cuda_kernels/set_value_kernel_register.cu | 5 ++--- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/set_value_kernel_register.cc b/backends/iluvatar_gpu/kernels/cuda_kernels/set_value_kernel_register.cc index 077c2d9a1d0..79b72334c74 100644 --- a/backends/iluvatar_gpu/kernels/cuda_kernels/set_value_kernel_register.cc +++ b/backends/iluvatar_gpu/kernels/cuda_kernels/set_value_kernel_register.cc @@ -13,7 +13,7 @@ // limitations under the License. #include "paddle/phi/core/kernel_registry.h" -#include "paddle/phi/kernels/gpu/set_value_kernel.cu" // NOLINT +#include "paddle/phi/kernels/set_value_kernel.h" PD_CUSTOM_KERNEL_REGISTER(set_value, iluvatar_gpu, diff --git a/backends/metax_gpu/kernels/cuda_kernels/set_value_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/set_value_kernel_register.cu index 92d76abf9e2..73d1e8d07dd 100644 --- a/backends/metax_gpu/kernels/cuda_kernels/set_value_kernel_register.cu +++ b/backends/metax_gpu/kernels/cuda_kernels/set_value_kernel_register.cu @@ -12,11 +12,10 @@ // 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. -// clang-format off + #include "paddle/phi/core/kernel_registry.h" -#include "paddle/phi/kernels/gpu/set_value_kernel.cu" // NOLINT #include "paddle/phi/kernels/set_value_kernel.h" -// clang-format on + PD_CUSTOM_KERNEL_REGISTER(set_value, metax_gpu, ALL_LAYOUT,