File tree Expand file tree Collapse file tree 3 files changed +3
-2
lines changed
iluvatar_gpu/kernels/cuda_kernels Expand file tree Collapse file tree 3 files changed +3
-2
lines changed Original file line number Diff line number Diff line change 1414
1515#include " paddle/phi/core/kernel_registry.h"
1616#include " paddle/phi/kernels/all_reduce_kernel.h"
17- #include " paddle/phi/kernels/gpu/ mp_allreduce_sum_kernel.cu " // NOLINT
17+ #include " paddle/phi/kernels/mp_allreduce_sum_kernel.h " // NOLINT
1818PD_CUSTOM_KERNEL_REGISTER (mp_allreduce_sum,
1919 iluvatar_gpu,
2020 ALL_LAYOUT,
Original file line number Diff line number Diff line change @@ -252,6 +252,7 @@ file(
252252 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/mean_all_grad_kernel.cu
253253 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/multiplex_kernel.cu
254254 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/multiplex_grad_kernel.cu
255+ ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/mp_allreduce_sum_kernel.cu
255256 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/pow2_decay_with_linear_warmup_kernel.cu
256257 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/sigmoid_cross_entropy_with_logits_grad_kernel.cu
257258 ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/top_k_kernel.cu
Original file line number Diff line number Diff line change 1414
1515#include " paddle/phi/core/kernel_registry.h"
1616#include " paddle/phi/kernels/all_reduce_kernel.h"
17- #include " paddle/phi/kernels/gpu/ mp_allreduce_sum_kernel.cu " // NOLINT
17+ #include " paddle/phi/kernels/mp_allreduce_sum_kernel.h " // NOLINT
1818PD_CUSTOM_KERNEL_REGISTER (mp_allreduce_sum,
1919 metax_gpu,
2020 ALL_LAYOUT,
You can’t perform that action at this time.
0 commit comments