diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/stack_grad_kernel_register.cu b/backends/iluvatar_gpu/kernels/cuda_kernels/stack_grad_kernel_register.cu
index 6a39776e11..d234da72a3 100644
--- a/backends/iluvatar_gpu/kernels/cuda_kernels/stack_grad_kernel_register.cu
+++ b/backends/iluvatar_gpu/kernels/cuda_kernels/stack_grad_kernel_register.cu
@@ -12,7 +12,8 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-#include "paddle/phi/kernels/gpu/stack_grad_kernel.cu"  // NOLINT
+#include "paddle/phi/core/kernel_registry.h"
+#include "paddle/phi/kernels/stack_grad_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(stack_grad,
                           iluvatar_gpu,
diff --git a/backends/metax_gpu/kernels/cuda_kernels/stack_grad_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/stack_grad_kernel_register.cu
index feee99f383..42c44568f9 100644
--- a/backends/metax_gpu/kernels/cuda_kernels/stack_grad_kernel_register.cu
+++ b/backends/metax_gpu/kernels/cuda_kernels/stack_grad_kernel_register.cu
@@ -12,7 +12,8 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-#include "paddle/phi/kernels/gpu/stack_grad_kernel.cu"  // NOLINT
+#include "paddle/phi/core/kernel_registry.h"
+#include "paddle/phi/kernels/stack_grad_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(stack_grad,
                           metax_gpu,