Skip to content

Commit 365ba4a

Browse files
authored
【CUDA Kernel No.24】ap_variadic算子Kernel修复 (PaddlePaddle#2093)
1 parent 18862cb commit 365ba4a

3 files changed

Lines changed: 3 additions & 2 deletions

File tree

backends/iluvatar_gpu/kernels/cuda_kernels/ap_variadic_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/ap/include/kernel_dispatch/ap_variadic_kernel.h"
1616
#include "paddle/phi/core/kernel_registry.h"
17-
#include "paddle/phi/kernels/gpu/ap_variadic_kernel.cu" //NOLINT
17+
#include "paddle/phi/kernels/gpu/ap_variadic_kernel.h"
1818

1919
#ifdef PADDLE_WITH_HIP
2020
PD_CUSTOM_KERNEL_REGISTER(ap_variadic,

backends/metax_gpu/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -151,6 +151,7 @@ file(
151151
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/angle_kernel.cu
152152
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/adamax_kernel.cu
153153
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/ap_facade_kernel.cu
154+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/ap_variadic_kernel.cu
154155
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/barrier_kernel.cu
155156
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/bincount_kernel.cu
156157
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/box_clip_kernel.cu

backends/metax_gpu/kernels/cuda_kernels/ap_variadic_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/ap/include/kernel_dispatch/ap_variadic_kernel.h"
1616
#include "paddle/phi/core/kernel_registry.h"
17-
#include "paddle/phi/kernels/gpu/ap_variadic_kernel.cu" //NOLINT
17+
#include "paddle/phi/kernels/gpu/ap_variadic_kernel.h"
1818

1919
#ifdef PADDLE_WITH_HIP
2020
PD_CUSTOM_KERNEL_REGISTER(ap_variadic,

0 commit comments

Comments
 (0)