Skip to content

Commit 18862cb

Browse files
include .h file for c_concat_kernel (PaddlePaddle#2052)
Co-authored-by: Yuqiang Ge <143453447+YqGe585@users.noreply.github.com>
1 parent 46580d8 commit 18862cb

3 files changed

Lines changed: 3 additions & 2 deletions

File tree

backends/iluvatar_gpu/kernels/cuda_kernels/c_concat_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/c_concat_kernel.cu" //NOLINT
16+
#include "paddle/phi/kernels/c_concat_kernel.h"
1717

1818
PD_CUSTOM_KERNEL_REGISTER(c_concat,
1919
iluvatar_gpu,

backends/metax_gpu/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -154,6 +154,7 @@ file(
154154
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/barrier_kernel.cu
155155
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/bincount_kernel.cu
156156
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/box_clip_kernel.cu
157+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/c_concat_kernel.cu
157158
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/c_embedding_kernel.cu
158159
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/c_embedding_grad_kernel.cu
159160
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/cast_kernel.cu

backends/metax_gpu/kernels/cuda_kernels/c_concat_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/c_concat_kernel.cu" //NOLINT
16+
#include "paddle/phi/kernels/c_concat_kernel.h"
1717

1818
PD_CUSTOM_KERNEL_REGISTER(c_concat,
1919
metax_gpu,

0 commit comments

Comments
 (0)