From d35f5882615b66e7cb18d7f966dde38dfbb20410 Mon Sep 17 00:00:00 2001 From: phlrain Date: Thu, 10 Mar 2022 14:08:00 +0000 Subject: [PATCH] update --- .../phi/kernels/gpu/slice_grad_kernel.cu.cc | 32 +++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 paddle/phi/kernels/gpu/slice_grad_kernel.cu.cc diff --git a/paddle/phi/kernels/gpu/slice_grad_kernel.cu.cc b/paddle/phi/kernels/gpu/slice_grad_kernel.cu.cc new file mode 100644 index 00000000000..8a0dc895db1 --- /dev/null +++ b/paddle/phi/kernels/gpu/slice_grad_kernel.cu.cc @@ -0,0 +1,32 @@ +// Copyright (c) 2022 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/slice_grad_kernel.h" +#include "paddle/phi/kernels/impl/slice_grad_kernel_impl.h" + +#include "paddle/phi/backends/gpu/gpu_context.h" +#include "paddle/phi/core/kernel_registry.h" + +PD_REGISTER_KERNEL(slice_grad, + GPU, + ALL_LAYOUT, + phi::SliceGradRawKernel, + bool, + int, + int64_t, + float, + double, + phi::dtype::complex, + phi::dtype::complex, + phi::dtype::bfloat16) {} -- GitLab