diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/set_value_kernel_register.cc b/backends/iluvatar_gpu/kernels/cuda_kernels/set_value_kernel_register.cc
index 077c2d9a1d0..79b72334c74 100644
--- a/backends/iluvatar_gpu/kernels/cuda_kernels/set_value_kernel_register.cc
+++ b/backends/iluvatar_gpu/kernels/cuda_kernels/set_value_kernel_register.cc
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/set_value_kernel.cu"  // NOLINT
+#include "paddle/phi/kernels/set_value_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(set_value,
                           iluvatar_gpu,
diff --git a/backends/metax_gpu/kernels/cuda_kernels/set_value_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/set_value_kernel_register.cu
index 92d76abf9e2..73d1e8d07dd 100644
--- a/backends/metax_gpu/kernels/cuda_kernels/set_value_kernel_register.cu
+++ b/backends/metax_gpu/kernels/cuda_kernels/set_value_kernel_register.cu
@@ -12,11 +12,10 @@
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.
-// clang-format off
+
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/set_value_kernel.cu" // NOLINT
 #include "paddle/phi/kernels/set_value_kernel.h"
-// clang-format on
+
 PD_CUSTOM_KERNEL_REGISTER(set_value,
                           metax_gpu,
                           ALL_LAYOUT,