Skip to content

Commit 6595afc

Browse files
【CUDA Kernel No.20】affine_channel算子Kernel修复 (#2016)
1 parent 4dfcb29 commit 6595afc

File tree

3 files changed

+3
-2
lines changed

3 files changed

+3
-2
lines changed

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

1818
PD_CUSTOM_KERNEL_REGISTER(affine_channel,
1919
iluvatar_gpu,

backends/metax_gpu/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -603,6 +603,7 @@ file(
603603
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/fusion/gpu/fused_softmax_mask_kernel.cu
604604
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/fusion/gpu/fusion_group_kernel.cu
605605
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/fusion/gpu/skip_layernorm_kernel.cu
606+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/affine_channel_kernel.cu
606607
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/fusion/gpu/fused_layernorm_kernel.cu
607608
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/fusion/gpu/fused_embedding_eltwise_layernorm_kernel.cu
608609
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/fusion/gpu/fused_stack_transpose_quant_kernel.cu

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

1818
PD_CUSTOM_KERNEL_REGISTER(affine_channel,
1919
metax_gpu,

0 commit comments

Comments
 (0)