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 13
13
// limitations under the License.
14
14
15
15
#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 "
17
17
18
18
PD_CUSTOM_KERNEL_REGISTER (affine_channel,
19
19
iluvatar_gpu,
Original file line number Diff line number Diff line change @@ -603,6 +603,7 @@ file(
603
603
${PADDLE_SOURCE_DIR} /paddle/phi/kernels/fusion/gpu/fused_softmax_mask_kernel.cu
604
604
${PADDLE_SOURCE_DIR} /paddle/phi/kernels/fusion/gpu/fusion_group_kernel.cu
605
605
${PADDLE_SOURCE_DIR} /paddle/phi/kernels/fusion/gpu/skip_layernorm_kernel.cu
606
+ ${PADDLE_SOURCE_DIR} /paddle/phi/kernels/gpu/affine_channel_kernel.cu
606
607
${PADDLE_SOURCE_DIR} /paddle/phi/kernels/fusion/gpu/fused_layernorm_kernel.cu
607
608
${PADDLE_SOURCE_DIR} /paddle/phi/kernels/fusion/gpu/fused_embedding_eltwise_layernorm_kernel.cu
608
609
${PADDLE_SOURCE_DIR} /paddle/phi/kernels/fusion/gpu/fused_stack_transpose_quant_kernel.cu
Original file line number Diff line number Diff line change 13
13
// limitations under the License.
14
14
15
15
#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 "
17
17
18
18
PD_CUSTOM_KERNEL_REGISTER (affine_channel,
19
19
metax_gpu,
You can’t perform that action at this time.
0 commit comments