Skip to content

Commit d699475

Browse files
LiaoYFBHYqGe585
andauthored
CUDA Kernel No.52 (#2240)
* Add elementwise_grad_kernel.cu to CMakeLists.txt for GPU backend * Fix: Replace .cu include with .h in elementwise_grad_kernel_register.cu * fix: Correct .cu include and CMake for embedding_with_scaled_gradient_grad kernel. * FIX(iluvatar_gpu): Correct include path for embedding_grad kernel registration file * FIX(iluvatar_gpu): Include missing headers for embedding_grad kernel registration. * Change included .cu files to .h * Fix flip_kernel include path and clean up CMakeLists redundancy --------- Co-authored-by: Yuqiang Ge <143453447+YqGe585@users.noreply.github.com>
1 parent c0f59da commit d699475

2 files changed

Lines changed: 1 addition & 3 deletions

File tree

backends/iluvatar_gpu/CMakeLists.txt

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -327,7 +327,6 @@ file(
327327
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/cvm_kernel.cu
328328
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/embedding_with_scaled_gradient_grad_kernel.cu
329329
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/exponential_kernel.cu
330-
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/flip_kernel.cu
331330
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/gather_grad_kernel.cu
332331
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/gelu_grad_kernel.cu
333332
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/global_gather_kernel.cu
@@ -901,7 +900,6 @@ file(
901900
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/eigh_grad_kernel.cu
902901
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/eigvalsh_grad_kernel.cu
903902
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/exponential_kernel.cu
904-
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/flip_kernel.cu
905903
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/gather_tree_kernel.cu
906904
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/graph_reindex_kernel.cu
907905
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/graph_sample_neighbors_kernel.cu

backends/metax_gpu/kernels/cuda_kernels/flip_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/flip_kernel.cu" //NOLINT
16+
#include "paddle/phi/kernels/flip_kernel.h" //NOLINT
1717
PD_CUSTOM_KERNEL_REGISTER(flip,
1818
metax_gpu,
1919
ALL_LAYOUT,

0 commit comments

Comments
 (0)