Skip to content

Commit 6940574

Browse files
authored
fix include for mp_allreduce_sum_kernel (PaddlePaddle#2218)
1 parent 3d3a0ce commit 6940574

File tree

3 files changed

+3
-2
lines changed

3 files changed

+3
-2
lines changed

backends/iluvatar_gpu/kernels/cuda_kernels/mp_allreduce_sum_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
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
1818
PD_CUSTOM_KERNEL_REGISTER(mp_allreduce_sum,
1919
iluvatar_gpu,
2020
ALL_LAYOUT,

backends/metax_gpu/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff 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

backends/metax_gpu/kernels/cuda_kernels/mp_allreduce_sum_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
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
1818
PD_CUSTOM_KERNEL_REGISTER(mp_allreduce_sum,
1919
metax_gpu,
2020
ALL_LAYOUT,

0 commit comments

Comments
 (0)