Modify the registration information of the interpolate kernel (#52163)

3b055199 · csy0225 · GitHub · 8c888eea · 3b055199
显示空白变更内容
内联并排

Showing with 2 addition and 0 deletion

paddle/phi/kernels/xpu/interpolate_kernel.cc paddle/phi/kernels/xpu/interpolate_kernel.cc +2 -0

未找到文件。
--- a/paddle/phi/kernels/xpu/interpolate_kernel.cc
+++ b/paddle/phi/kernels/xpu/interpolate_kernel.cc
@@ -223,11 +223,13 @@ void NearestInterpKernel(

 PD_REGISTER_KERNEL(
    bilinear_interp, XPU, ALL_LAYOUT, phi::BilinearInterpKernel, float) {
+  kernel->InputAt(1).SetBackend(phi::Backend::ALL_BACKEND);
  kernel->InputAt(2).SetBackend(phi::Backend::ALL_BACKEND);
  kernel->InputAt(3).SetBackend(phi::Backend::ALL_BACKEND);
 }
 PD_REGISTER_KERNEL(
    nearest_interp, XPU, ALL_LAYOUT, phi::NearestInterpKernel, float) {
+  kernel->InputAt(1).SetBackend(phi::Backend::ALL_BACKEND);
  kernel->InputAt(2).SetBackend(phi::Backend::ALL_BACKEND);
  kernel->InputAt(3).SetBackend(phi::Backend::ALL_BACKEND);
 }