From 520f48d6367b519aa5c3ec0d4073fdc380c458b1 Mon Sep 17 00:00:00 2001 From: zhangyikun02 <48021248+zhangyk0314@users.noreply.github.com> Date: Wed, 1 Feb 2023 14:49:29 +0800 Subject: [PATCH] support grid_sampler_grad op for XPU (#49857) --- paddle/phi/backends/xpu/xpu2_op_list.cc | 1 + .../kernels/xpu/grid_sample_grad_kernel.cc | 87 +++++++++++++++++++ .../fluid/tests/unittests/op_test_xpu.py | 7 +- .../unittests/xpu/test_grid_sampler_op_xpu.py | 1 + 4 files changed, 95 insertions(+), 1 deletion(-) create mode 100644 paddle/phi/kernels/xpu/grid_sample_grad_kernel.cc diff --git a/paddle/phi/backends/xpu/xpu2_op_list.cc b/paddle/phi/backends/xpu/xpu2_op_list.cc index 67ac2b17a7..b3635652ff 100644 --- a/paddle/phi/backends/xpu/xpu2_op_list.cc +++ b/paddle/phi/backends/xpu/xpu2_op_list.cc @@ -313,6 +313,7 @@ XPUOpMap& get_kl2_ops() { XPUKernelSet({phi::DataType::INT64, phi::DataType::INT32, phi::DataType::FLOAT32})}, + {"grid_sampler_grad", XPUKernelSet({phi::DataType::FLOAT32})}, {"grid_sampler", XPUKernelSet({phi::DataType::FLOAT32})}, {"hard_sigmoid_grad", XPUKernelSet({phi::DataType::FLOAT32})}, {"hard_sigmoid", XPUKernelSet({phi::DataType::FLOAT32})}, diff --git a/paddle/phi/kernels/xpu/grid_sample_grad_kernel.cc b/paddle/phi/kernels/xpu/grid_sample_grad_kernel.cc new file mode 100644 index 0000000000..86e78b4b15 --- /dev/null +++ b/paddle/phi/kernels/xpu/grid_sample_grad_kernel.cc @@ -0,0 +1,87 @@ +// Copyright (c) 2023 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/grid_sample_grad_kernel.h" + +#include "paddle/phi/backends/xpu/enforce_xpu.h" +#include "paddle/phi/core/kernel_registry.h" + +namespace phi { + +template +void GridSampleGradKernel(const Context& dev_ctx, + const DenseTensor& x, + const DenseTensor& grid, + const DenseTensor& out_grid, + const std::string& mode, + const std::string& padding_mode, + bool align_corners, + DenseTensor* x_grad, + DenseTensor* grid_grad) { + PADDLE_ENFORCE_EQ( + x.dims().size(), + 4, + phi::errors::InvalidArgument( + ("XPU is only support input_dims == 4 in grid_sample_grad op."))); + + const int64_t n = grid.dims()[0]; + const int64_t out_h = grid.dims()[1]; + const int64_t out_w = grid.dims()[2]; + const int64_t c = x.dims()[1]; + const int64_t in_h = x.dims()[2]; + const int64_t in_w = x.dims()[3]; + + x_grad->Resize({n, c, in_h, in_w}); + T* x_grad_ptr = dev_ctx.template Alloc(x_grad); + + T* grid_grad_ptr = nullptr; + if (grid_grad != nullptr) { + grid_grad->Resize({n, out_h, out_w, 2}); + grid_grad_ptr = dev_ctx.template Alloc(grid_grad); + } + + bool is_nearest = false; + if (mode == "nearest") { + is_nearest = true; + } + int64_t padding_mode_type = 0; + if (padding_mode == "border") { + padding_mode_type = 1; + } else if (padding_mode == "reflection") { + padding_mode_type = 2; + } + + int r = xpu::grid_sample_grad(dev_ctx.x_context(), + x.data(), + grid.data(), + out_grid.data(), + x_grad_ptr, + grid_grad_ptr, + n, + c, + in_h, + in_w, + out_h, + out_w, + is_nearest, + align_corners, + padding_mode_type, + true); + PADDLE_ENFORCE_XDNN_SUCCESS(r, "grid_sample_grad"); +} + +} // namespace phi + +PD_REGISTER_KERNEL( + grid_sample_grad, XPU, ALL_LAYOUT, phi::GridSampleGradKernel, float) {} diff --git a/python/paddle/fluid/tests/unittests/op_test_xpu.py b/python/paddle/fluid/tests/unittests/op_test_xpu.py index a7cb066db7..c6a76c5563 100644 --- a/python/paddle/fluid/tests/unittests/op_test_xpu.py +++ b/python/paddle/fluid/tests/unittests/op_test_xpu.py @@ -35,6 +35,7 @@ class XPUOpTest(OpTest): '''Fix random seeds to remove randomness from tests''' cls.use_xpu = True cls.use_mkldnn = False + cls.epsilon_xpu2xpu = 0.00000001 super().setUpClass() @classmethod @@ -212,7 +213,11 @@ class XPUOpTest(OpTest): user_defined_grad_outputs=user_defined_grad_outputs, ) self._assert_is_close( - a1, a2, inputs_to_check, 0.00000001, "Gradient Check On two xpu" + a1, + a2, + inputs_to_check, + self.epsilon_xpu2xpu, + "Gradient Check On two xpu", ) self._assert_is_close( a1, diff --git a/python/paddle/fluid/tests/unittests/xpu/test_grid_sampler_op_xpu.py b/python/paddle/fluid/tests/unittests/xpu/test_grid_sampler_op_xpu.py index 3b2deaf439..c92ddc9531 100644 --- a/python/paddle/fluid/tests/unittests/xpu/test_grid_sampler_op_xpu.py +++ b/python/paddle/fluid/tests/unittests/xpu/test_grid_sampler_op_xpu.py @@ -170,6 +170,7 @@ class XPUTestGridSamplerOP(XPUOpTestWrapper): self.place = paddle.XPUPlace(0) self.init_dtype() self.op_type = 'grid_sampler' + self.epsilon_xpu2xpu = 0.000001 self.use_cudnn = False self.align_corners = True -- GitLab