Skip to content

Commit 3a76527

Browse files
authored
【CUDA Kernel No.49】elementwise_grad_kernel.cu算子Kernel修复 (#2129)
* Add elementwise_grad_kernel.cu to CMakeLists.txt for GPU backend * Fix: Replace .cu include with .h in elementwise_grad_kernel_register.cu
1 parent e95493f commit 3a76527

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

backends/metax_gpu/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -661,6 +661,7 @@ file(
661661
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/rms_norm_kernel.cu
662662
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/lars_momentum_kernel.cu
663663
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/partial_sum_kernel.cu
664+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/elementwise_grad_kernel.cu
664665
${PADDLE_SOURCE_DIR}/paddle/phi/backends/gpu/gpu_info.cc
665666
# ############################################################################
666667
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/selected_rows/gpu/adamw_kernel.cu

backends/metax_gpu/kernels/cuda_kernels/elementwise_grad_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
// limitations under the License.
1414

1515
#include "paddle/phi/core/kernel_registry.h"
16-
#include "paddle/phi/kernels/gpu/elementwise_grad_kernel.cu" // NOLINT
16+
#include "paddle/phi/kernels/elementwise_grad_kernel.h"
1717

1818
PD_CUSTOM_KERNEL_REGISTER(fmax_grad,
1919
metax_gpu,

0 commit comments

Comments
 (0)