[ILUVATAR_GPU] Add c_softmax_with_cross_entropy_grad_kernel (#1866)

Author: tianyuzhou668

README.md

@@ -24,6 +24,7 @@
 - [飞桨自定义接入硬件后端(燧原GCU)](backends/gcu/README_cn.md)
 - [飞桨自定义接入硬件后端(太初SDAA)](backends/sdaa/README_cn.md)
 - [飞桨自定义接入硬件后端(沐曦METAX)](backends/metax_gpu/README_cn.md)
+- [飞桨自定义接入硬件后端(天数ILUVATAR)](backends/iluvatar_gpu/README_cn.md)
 
 ## 版权和许可证
 

backends/iluvatar_gpu/CMakeLists.txt

@@ -119,6 +119,7 @@ file(
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/c_embedding_grad_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/c_embedding_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/c_identity_kernel.cu
+  ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/c_softmax_with_cross_entropy_grad_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/cast_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/clip_grad_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/clip_kernel.cu

backends/iluvatar_gpu/kernels/cuda_kernels/c_softmax_with_cross_entropy_grad_kernel_register.cc

@@ -0,0 +1,23 @@
+// Copyright (c) 2024 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/core/kernel_registry.h"
+#include "paddle/phi/kernels/gpu/c_softmax_with_cross_entropy_grad_kernel.cu"  // NOLINT
+
+PD_CUSTOM_KERNEL_REGISTER(c_softmax_with_cross_entropy_grad,
+                          iluvatar_gpu,
+                          ALL_LAYOUT,
+                          phi::CSoftmaxWithCrossEntropyGradKernel,
+                          float,
+                          phi::dtype::float16) {}