diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/fused_bias_residual_layernorm_kernel_register.cu b/backends/iluvatar_gpu/kernels/cuda_kernels/fused_bias_residual_layernorm_kernel_register.cu
index 8c4bbf96170..f6018d573d9 100755
--- a/backends/iluvatar_gpu/kernels/cuda_kernels/fused_bias_residual_layernorm_kernel_register.cu
+++ b/backends/iluvatar_gpu/kernels/cuda_kernels/fused_bias_residual_layernorm_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/fusion/gpu/fused_layernorm_kernel.cu"  //NOLINT
+#include "paddle/phi/kernels/fused_layernorm_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(fused_bias_residual_layernorm,
                           iluvatar_gpu,
diff --git a/backends/metax_gpu/kernels/fusion/fused_layernorm_kernel_register.cu b/backends/metax_gpu/kernels/fusion/fused_layernorm_kernel_register.cu
index 955d52053fb..bdb809a2149 100644
--- a/backends/metax_gpu/kernels/fusion/fused_layernorm_kernel_register.cu
+++ b/backends/metax_gpu/kernels/fusion/fused_layernorm_kernel_register.cu
@@ -13,9 +13,9 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
+#include "paddle/phi/kernels/fused_layernorm_kernel.h"
 #include "paddle/phi/kernels/fusion/gpu/attention_layer.norm.h"
 #include "paddle/phi/kernels/fusion/gpu/fused_dropout_helper.h"
-#include "paddle/phi/kernels/fusion/gpu/fused_layernorm_kernel.cu"  //NOLINT
 
 #ifndef PADDLE_WITH_HIP
 #if CUDNN_VERSION_MIN(8, 1, 0)