File tree Expand file tree Collapse file tree 3 files changed +3
-2
lines changed
iluvatar_gpu/kernels/cuda_kernels Expand file tree Collapse file tree 3 files changed +3
-2
lines changed Original file line number Diff line number Diff line change 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
1818PD_CUSTOM_KERNEL_REGISTER (affine_channel,
1919 iluvatar_gpu,
Original file line number Diff line number Diff 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
Original file line number Diff line number Diff line change 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
1818PD_CUSTOM_KERNEL_REGISTER (affine_channel,
1919 metax_gpu,
You can’t perform that action at this time.
0 commit comments