Skip to content

Commit ff29595

Browse files
【CUDA Kernel No.15】fusion_group算子Kernel修复 (PaddlePaddle#2013)
Co-authored-by: Tao Luo <luotao02@baidu.com>
1 parent 365ba4a commit ff29595

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

backends/iluvatar_gpu/kernels/cuda_kernels/fusion_group_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/core/utils/data_type.h"
17-
#include "paddle/phi/kernels/fusion/gpu/fusion_group_kernel.cu" //NOLINT
17+
#include "paddle/phi/kernels/fusion/gpu/fusion_group_kernel.h"
1818

1919
PD_CUSTOM_KERNEL_REGISTER(fusion_group,
2020
iluvatar_gpu,

backends/metax_gpu/kernels/cuda_kernels/fusion_group_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/core/utils/data_type.h"
17-
#include "paddle/phi/kernels/fusion/gpu/fusion_group_kernel.cu" //NOLINT
17+
#include "paddle/phi/kernels/fusion/gpu/fusion_group_kernel.h"
1818

1919
PD_CUSTOM_KERNEL_REGISTER(fusion_group,
2020
metax_gpu,

0 commit comments

Comments
 (0)