Skip to content

Commit 6a16032

Browse files
authored
fix include for multiclass_nms3_kernel (PaddlePaddle#2220)
1 parent 0fc198f commit 6a16032

File tree

4 files changed

+2
-2
lines changed

4 files changed

+2
-2
lines changed

backends/iluvatar_gpu/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -357,6 +357,7 @@ file(
357357
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/moe_unpermute_kernel.cu
358358
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/momentum_kernel.cu
359359
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/mp_allreduce_sum_kernel.cu
360+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/multiclass_nms3_kernel.cu
360361
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/multiplex_grad_kernel.cu
361362
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/nonzero_kernel.cu
362363
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/pad3d_kernel.cu

backends/iluvatar_gpu/kernels/cuda_kernels/multiclass_nms3_kernel_register.cu

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

1515
#include "paddle/phi/core/kernel_registry.h"
16-
#include "paddle/phi/kernels/gpu/multiclass_nms3_kernel.cu" //NOLINT
1716
#include "paddle/phi/kernels/multiclass_nms3_kernel.h"
1817

1918
PD_CUSTOM_KERNEL_REGISTER(multiclass_nms3,

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/multiclass_nms3_kernel.cu
255256
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/mp_allreduce_sum_kernel.cu
256257
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/pow2_decay_with_linear_warmup_kernel.cu
257258
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/sigmoid_cross_entropy_with_logits_grad_kernel.cu

backends/metax_gpu/kernels/cuda_kernels/multiclass_nms3_kernel_register.cu

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

1515
#include "paddle/phi/core/kernel_registry.h"
16-
#include "paddle/phi/kernels/gpu/multiclass_nms3_kernel.cu" //NOLINT
1716
#include "paddle/phi/kernels/multiclass_nms3_kernel.h"
1817

1918
PD_CUSTOM_KERNEL_REGISTER(multiclass_nms3,

0 commit comments

Comments
 (0)